Merge pull request #1 from Mocksi/add-eye-candy

create UI

Author: elg0nz

Gemfile

@@ -2,11 +2,11 @@
 
 source 'https://rubygems.org'
 
-gem 'rack'
-gem 'puma'
-gem 'thor'
 gem 'json' 
 gem 'httpx', '~> 1.3'
+gem "puma", ">= 6.4.3"
+gem 'rack'
+gem 'thor'
 
 group :development, :test do
   gem 'rspec', '~> 3.13'

Gemfile.lock

@@ -6,8 +6,8 @@ GEM
     httpx (1.3.0)
       http-2 (>= 1.0.0)
     json (2.7.1)
-    nio4r (2.7.1)
-    puma (6.4.2)
+    nio4r (2.7.3)
+    puma (6.4.3)
       nio4r (~> 2.0)
     rack (2.2.9)
     rack-test (1.1.0)
@@ -34,7 +34,7 @@ PLATFORMS
 DEPENDENCIES
   httpx (~> 1.3)
   json
-  puma
+  puma (>= 6.4.3)
   rack
   rack-test (~> 1.1)
   rspec (~> 3.13)

lib/command_executor.rb

@@ -4,12 +4,11 @@
 
 class CommandExecutor
   attr_reader :logger, :client_uuid, :endpoint_url
-  REACTOR_URL = ENV['MOCKSI_REACTOR_URL'] || "https://crowllectordb.onrender.com/api/v1/reactor"
 
   def initialize(logger, client_uuid)
     @logger = logger
     @client_uuid = client_uuid
-    @endpoint_url = REACTOR_URL
+    @endpoint_url = Hawksi.configuration.reactor_url
   end
 
   def execute_command(command, params)

lib/file_uploader.rb

@@ -3,9 +3,6 @@
 
 class FileUploader
   # FIXME: use a base URL for the upload and process URLs
-  UPLOAD_URL =  ENV['MOCKSI_UPLOAD_URL'] || "https://crowllectordb.onrender.com/api/v1/upload"
-  PROCESS_URL = ENV['MOCKSI_PROCESS_URL'] || "https://crowllectordb.onrender.com/api/v1/process"
-
   def initialize(logger, client_uuid)
     @logger = logger
     @client_uuid = client_uuid
@@ -19,10 +16,15 @@ def upload_files(tar_gz_files)
 
   def process_files
     HTTPX.wrap do |client|
-      response = client.post(PROCESS_URL, headers: { "x-client-id" => @client_uuid })
+      response = begin
+        client.post(Hawksi.configuration.process_url, headers: { "x-client-id" => @client_uuid })
+      rescue => e
+        @logger.error "Failed to process files. Error: #{e.message}"
+      end
+
       if response.is_a?(HTTPX::Response)
         @logger.info "Processing uploaded files. Status: #{response.status}"
-      else
+      elsif response.is_a?(HTTPX::ErrorResponse)
         @logger.error "Failed to process files. Error: #{response.error}"
       end
     end
@@ -58,7 +60,7 @@ def upload_file(tar_gz_file)
 
   def post_file(client, tar_gz_file)
     filename = File.basename(tar_gz_file)
-    client.post("#{UPLOAD_URL}?filename=#{filename}",
+    client.post("#{Hawksi.configuration.upload_url}?filename=#{filename}",
                 headers: { "x-client-id" => @client_uuid },
                 body: File.read(tar_gz_file))
   rescue => e

lib/hawksi.rb

@@ -1,2 +1,16 @@
+require 'hawksi/configuration'
 require 'cli'
-require 'request_interceptor'
+require 'request_interceptor'
+
+
+BANNER = <<~BANNER
+                      _        _
+  /\  /\__ ___      _| | _____(_)
+ / /_/ / _` \ \ /\ / / |/ / __| |
+/ __  / (_| |\ V  V /|   <\__ \ |
+\/ /_/ \__,_| \_/\_/ |_|\_\___/_|
+
+BANNER
+
+
+puts BANNER

lib/hawksi/configuration.rb

@@ -0,0 +1,29 @@
+module Hawksi
+  class Configuration
+    attr_accessor :mocksi_server, :reactor_url, :upload_url, :process_url
+
+    def initialize
+      @mocksi_server = get_env('MOCKSI_SERVER') || "https://app.mocksi.ai"
+      @reactor_url = get_env('MOCKSI_REACTOR_URL') || "https://api.mocksi.ai/api/v1/reactor"
+      @upload_url = get_env('MOCKSI_UPLOAD_URL') || "https://api.mocksi.ai/api/v1/upload"
+      @process_url = get_env('MOCKSI_PROCESS_URL') || "https://api.mocksi.ai/api/v1/process"
+    end
+  
+    private
+
+    def get_env(key)
+      env_value = ENV.fetch(key, '').strip
+      env_value.empty? ? nil : env_value
+    end
+  end
+
+  class << self
+    def configuration
+      @configuration ||= Configuration.new
+    end
+
+    def configure
+      yield(configuration)
+    end
+  end
+end

lib/mocksi_handler.rb

@@ -0,0 +1,105 @@
+require 'httpx'
+
+HAWK_SVG = <<~SVG
+  <svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" viewBox="0 0 16 16">
+    <text x="9" y="8" font-size="8" text-anchor="middle" font-family="sans-serif">Hwk</text>
+  </svg>
+SVG
+
+module MocksiHandler
+  class << self
+    def handle(request)
+      if request.path == '/favicon.ico'
+        return [200, { 'Content-Type' => 'image/svg+xml' }, [HAWK_SVG]]
+      end
+
+      begin
+        # Get the mocksi server URL from configuration
+        mocksi_server_url = Hawksi.configuration.mocksi_server
+        raise "Mocksi server URL not configured" if mocksi_server_url.nil? || mocksi_server_url.empty?
+
+        # Prepare the full URL (mocksi_server + request path + query string)
+        target_uri = URI.join(mocksi_server_url, request.fullpath)
+
+        # Prepare headers from the request
+        headers = {}
+        request.env.each do |key, value|
+          if key.start_with?('HTTP_')
+            header_key = key.sub(/^HTTP_/, '').split('_').map(&:capitalize).join('-')
+            headers[header_key] = value
+          end
+          ## Yay for Rack's weirdness. See https://github.com/rack/rack/issues/1311
+          if key == 'CONTENT_TYPE'
+            headers['Content-Type'] = value
+          end
+          if key == 'CONTENT_LENGTH'
+            headers['Content-Length'] = value
+          end
+        end
+
+        # Forward the cookies
+        if request.cookies.any?
+          headers['Cookie'] = request.cookies.map { |k, v| "#{k}=#{v}" }.join('; ')
+        end
+
+        # Initialize httpx with headers
+        http_client = HTTPX.with(headers: headers)
+
+        # Forward the body content if it's a POST or PUT request
+        body = nil
+        if %w[POST PUT].include?(request.request_method)
+          request.body.rewind
+          body = request.body.read
+        end
+
+        # Make the HTTP request using the appropriate method
+        response = http_client.request(request.request_method.downcase.to_sym, target_uri, body: body)
+
+        # Clone headers to allow modification
+        response_headers = response.headers.dup
+
+        # Check for chunked transfer encoding and remove it
+        if response_headers["transfer-encoding"]&.include?("chunked")
+          response_headers.delete("transfer-encoding")
+        end
+
+        # Handle gzip or deflate content-encoding if present
+        response_body = response.body.to_s
+        if response_headers["content-encoding"]&.include?("gzip")
+          response_body = safe_decompress_gzip(response_body)
+          response_headers.delete("content-encoding") # Remove content-encoding since the content is decompressed
+        elsif response_headers["content-encoding"]&.include?("deflate")
+          response_body = decompress_deflate(response_body)
+          response_headers.delete("content-encoding") # Remove content-encoding since the content is decompressed
+        end
+
+        # Return the response in a format compatible with Rack
+        [response.status, response_headers, [response_body]]
+      rescue => e
+        # Handle any errors that occur during the reverse proxy operation
+        [500, { 'Content-Type' => 'text/plain' }, ["Error: #{e.message}"]]
+      end
+    end
+
+    private
+
+    # Helper method to safely decompress gzip content, returning the original body if it's not in gzip format
+    def safe_decompress_gzip(body)
+      io = StringIO.new(body)
+      begin
+        gzip_reader = Zlib::GzipReader.new(io)
+        decompressed_body = gzip_reader.read
+        gzip_reader.close
+        decompressed_body
+      rescue Zlib::GzipFile::Error
+        # If the body is not actually gzip, return the original body
+        body
+      end
+    end
+
+    # Helper method to decompress deflate content
+    def decompress_deflate(body)
+      Zlib::Inflate.inflate(body)
+    end
+  end
+end

lib/request_interceptor.rb

@@ -1,7 +1,8 @@
 require 'json'
 require 'logger'
 require 'digest'
-require_relative '../lib/file_storage'
+require_relative './file_storage'
+require_relative './mocksi_handler'
 
 module Hawksi
   class RequestInterceptor
@@ -13,6 +14,14 @@ def initialize(app, logger: Logger.new('hawksi.log'), storage: FileStorage)
 
     def call(env)
       request = Rack::Request.new(env)
+
+      if request.path.end_with?('/favicon.ico')
+        return MocksiHandler.handle(request)
+      end
+
+      if request.path.start_with?('/mocksi') || request.path.start_with?('/_') || request.path.start_with?('/api')
+        return MocksiHandler.handle(request)
+      end
       request_hash = generate_request_hash(request)  # Generate a hash of the request
       log_request(request, request_hash)
 

spec/lib/hawksi/configuration_spec.rb

@@ -0,0 +1,81 @@
+
+require 'spec_helper'
+require 'hawksi/configuration'
+
+RSpec.describe Hawksi::Configuration do
+  describe '.configuration' do
+    it 'returns a Configuration instance' do
+      expect(Hawksi.configuration).to be_an_instance_of(Hawksi::Configuration)
+    end
+
+    it 'memoizes the configuration instance' do
+      config1 = Hawksi.configuration
+      config2 = Hawksi.configuration
+      expect(config1).to eq(config2)
+    end
+  end
+
+  describe '.configure' do
+    it 'yields the configuration instance' do
+      expect { |b| Hawksi.configure(&b) }.to yield_with_args(Hawksi.configuration)
+    end
+  end
+
+  describe '#initialize' do
+    let(:config) { Hawksi::Configuration.new }
+
+    it 'sets default values for URLs' do
+      expect(config.instance_variable_get(:@mocksi_server)).to eq('https://app.mocksi.ai')
+      expect(config.instance_variable_get(:@reactor_url)).to eq('https://api.mocksi.ai/api/v1/reactor')
+      expect(config.instance_variable_get(:@upload_url)).to eq('https://api.mocksi.ai/api/v1/upload')
+      expect(config.instance_variable_get(:@process_url)).to eq('https://api.mocksi.ai/api/v1/process')
+    end
+
+    context 'when environment variables are set' do
+      before do
+        ENV['MOCKSI_SERVER'] = 'https://custom.mocksi.ai'
+        ENV['MOCKSI_REACTOR_URL'] = 'https://custom.mocksi.ai/reactor'
+        ENV['MOCKSI_UPLOAD_URL'] = 'https://custom.mocksi.ai/upload'
+        ENV['MOCKSI_PROCESS_URL'] = 'https://custom.mocksi.ai/process'
+      end
+
+      after do
+        ENV.delete('MOCKSI_SERVER')
+        ENV.delete('MOCKSI_REACTOR_URL')
+        ENV.delete('MOCKSI_UPLOAD_URL')
+        ENV.delete('MOCKSI_PROCESS_URL')
+      end
+
+      it 'uses environment variables for URLs' do
+        config = Hawksi::Configuration.new
+        expect(config.instance_variable_get(:@mocksi_server)).to eq('https://custom.mocksi.ai')
+        expect(config.instance_variable_get(:@reactor_url)).to eq('https://custom.mocksi.ai/reactor')
+        expect(config.instance_variable_get(:@upload_url)).to eq('https://custom.mocksi.ai/upload')
+        expect(config.instance_variable_get(:@process_url)).to eq('https://custom.mocksi.ai/process')
+      end
+    end
+    context 'when a value is set to an empty string' do
+      before do
+        ENV['MOCKSI_SERVER'] = ''
+        ENV['MOCKSI_REACTOR_URL'] = ''
+        ENV['MOCKSI_UPLOAD_URL'] = ''
+        ENV['MOCKSI_PROCESS_URL'] = ''
+      end
+
+      after do
+        ENV.delete('MOCKSI_SERVER')
+        ENV.delete('MOCKSI_REACTOR_URL')
+        ENV.delete('MOCKSI_UPLOAD_URL')
+        ENV.delete('MOCKSI_PROCESS_URL')
+      end
+
+      it 'uses default values for empty string environment variables' do
+        config = Hawksi::Configuration.new
+        expect(config.instance_variable_get(:@mocksi_server)).to eq('https://app.mocksi.ai')
+        expect(config.instance_variable_get(:@reactor_url)).to eq('https://api.mocksi.ai/api/v1/reactor')
+        expect(config.instance_variable_get(:@upload_url)).to eq('https://api.mocksi.ai/api/v1/upload')
+        expect(config.instance_variable_get(:@process_url)).to eq('https://api.mocksi.ai/api/v1/process')
+      end
+    end
+  end
+end