whoosh 1.0.1

data/lib/whoosh/streaming/websocket.rb

@@ -0,0 +1,51 @@
+# lib/whoosh/streaming/websocket.rb
+# frozen_string_literal: true
+
+require "json"
+
+module Whoosh
+  module Streaming
+    class WebSocket
+      def initialize(io)
+        @io = io
+        @closed = false
+        @on_message = nil
+        @on_close = nil
+      end
+
+      def send(data)
+        return if @closed
+        formatted = data.is_a?(String) ? data : JSON.generate(data)
+        @io.write(formatted + "\n")
+        @io.flush if @io.respond_to?(:flush)
+      rescue IOError, Errno::EPIPE
+        @closed = true
+      end
+
+      def on_message(&block)
+        @on_message = block
+      end
+
+      def on_close(&block)
+        @on_close = block
+      end
+
+      def trigger_message(msg)
+        @on_message&.call(msg)
+      end
+
+      def trigger_close
+        @on_close&.call
+        @closed = true
+      end
+
+      def close
+        @closed = true
+      end
+
+      def closed?
+        @closed
+      end
+    end
+  end
+end

data/lib/whoosh/test.rb

@@ -0,0 +1,70 @@
+# lib/whoosh/test.rb
+# frozen_string_literal: true
+
+require "rack/test"
+require "json"
+
+module Whoosh
+  module Test
+    include Rack::Test::Methods
+
+    # Override in your test class to return the Rack app
+    def app
+      raise NotImplementedError, "Define #app in your test class to return a Rack app"
+    end
+
+    # --- Request helpers ---
+
+    def post_json(path, body = {}, headers: {})
+      merged_headers = { "CONTENT_TYPE" => "application/json" }.merge(headers)
+      post path, JSON.generate(body), merged_headers
+    end
+
+    def put_json(path, body = {}, headers: {})
+      merged_headers = { "CONTENT_TYPE" => "application/json" }.merge(headers)
+      put path, JSON.generate(body), merged_headers
+    end
+
+    def patch_json(path, body = {}, headers: {})
+      merged_headers = { "CONTENT_TYPE" => "application/json" }.merge(headers)
+      patch path, JSON.generate(body), merged_headers
+    end
+
+    def get_with_auth(path, key:, header: "HTTP_X_API_KEY")
+      get path, {}, { header => key }
+    end
+
+    def post_with_auth(path, body = {}, key:, header: "HTTP_X_API_KEY")
+      post path, JSON.generate(body), { "CONTENT_TYPE" => "application/json", header => key }
+    end
+
+    # --- Response assertions ---
+
+    def assert_response(expected_status)
+      expect(last_response.status).to eq(expected_status)
+    end
+
+    def assert_json(expected_hash)
+      body = JSON.parse(last_response.body)
+      expected_hash.each do |key, value|
+        expect(body[key.to_s]).to eq(value)
+      end
+    end
+
+    def assert_json_path(path, expected_value)
+      body = JSON.parse(last_response.body)
+      keys = path.to_s.split(".")
+      value = keys.reduce(body) { |h, k| h.is_a?(Hash) ? h[k] : nil }
+      expect(value).to eq(expected_value)
+    end
+
+    def assert_json_includes(key)
+      body = JSON.parse(last_response.body)
+      expect(body).to have_key(key.to_s)
+    end
+
+    def response_json
+      JSON.parse(last_response.body)
+    end
+  end
+end

data/lib/whoosh/types.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+require "dry-types"
+
+module Whoosh
+  module Types
+    include Dry.Types()
+
+    Bool = Types::Params::Bool
+  end
+end

data/lib/whoosh/uploaded_file.rb

@@ -0,0 +1,47 @@
+# lib/whoosh/uploaded_file.rb
+# frozen_string_literal: true
+
+require "base64"
+
+module Whoosh
+  class UploadedFile
+    attr_reader :filename, :content_type
+
+    def initialize(rack_hash, storage: nil)
+      @filename = rack_hash[:filename]
+      @content_type = rack_hash[:type]
+      @tempfile = rack_hash[:tempfile]
+      @storage = storage
+    end
+
+    def size
+      @tempfile.size
+    end
+
+    def read
+      @tempfile.rewind
+      @tempfile.read
+    end
+
+    def read_text
+      read.force_encoding("UTF-8")
+    end
+
+    def to_base64
+      Base64.strict_encode64(read)
+    end
+
+    def save(prefix = "")
+      raise Errors::DependencyError, "No storage adapter configured" unless @storage
+      @storage.save(self, prefix)
+    end
+
+    def validate!(types: nil, max_size: nil)
+      errors = []
+      errors << { field: "file", message: "No file uploaded" } if @filename.nil? || @filename.empty?
+      errors << { field: "file", message: "File type #{@content_type} not allowed" } if types && !types.include?(@content_type)
+      errors << { field: "file", message: "File too large (#{size} bytes > #{max_size})" } if max_size && size > max_size
+      raise Errors::ValidationError.new(errors) unless errors.empty?
+    end
+  end
+end

data/lib/whoosh/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Whoosh
+  VERSION = "1.0.1"
+end

data/lib/whoosh.rb

@@ -0,0 +1,86 @@
+# frozen_string_literal: true
+
+require_relative "whoosh/version"
+
+module Whoosh
+  autoload :App,                 "whoosh/app"
+  autoload :Cache,               "whoosh/cache"
+  autoload :EnvLoader,           "whoosh/env_loader"
+  autoload :Instrumentation,     "whoosh/instrumentation"
+  autoload :Config,              "whoosh/config"
+  autoload :Database,            "whoosh/database"
+  autoload :DependencyInjection, "whoosh/dependency_injection"
+  autoload :Endpoint,            "whoosh/endpoint"
+  autoload :Errors,              "whoosh/errors"
+  autoload :Logger,              "whoosh/logger"
+  autoload :Request,             "whoosh/request"
+  autoload :Response,            "whoosh/response"
+  autoload :Router,              "whoosh/router"
+  autoload :Schema,              "whoosh/schema"
+  autoload :Shutdown,            "whoosh/shutdown"
+  autoload :Test,                "whoosh/test"
+  autoload :Types,               "whoosh/types"
+  autoload :HTTP,                "whoosh/http"
+  autoload :Storage,             "whoosh/storage"
+  autoload :UploadedFile,        "whoosh/uploaded_file"
+  autoload :Performance,         "whoosh/performance"
+  autoload :Job,                 "whoosh/job"
+  autoload :Jobs,                "whoosh/jobs"
+  autoload :Metrics,             "whoosh/metrics"
+  autoload :Paginate,            "whoosh/paginate"
+
+  module Auth
+    autoload :ApiKey,         "whoosh/auth/api_key"
+    autoload :Jwt,            "whoosh/auth/jwt"
+    autoload :OAuth2,         "whoosh/auth/oauth2"
+    autoload :RateLimiter,    "whoosh/auth/rate_limiter"
+    autoload :TokenTracker,   "whoosh/auth/token_tracker"
+    autoload :AccessControl,  "whoosh/auth/access_control"
+  end
+
+  module MCP
+    autoload :Server,        "whoosh/mcp/server"
+    autoload :Client,        "whoosh/mcp/client"
+    autoload :ClientManager, "whoosh/mcp/client_manager"
+    autoload :Protocol,      "whoosh/mcp/protocol"
+
+    module Transport
+      autoload :SSE,   "whoosh/mcp/transport/sse"
+      autoload :Stdio, "whoosh/mcp/transport/stdio"
+    end
+  end
+
+  module Middleware
+    autoload :Stack,           "whoosh/middleware/stack"
+    autoload :Cors,            "whoosh/middleware/cors"
+    autoload :RequestLogger,   "whoosh/middleware/request_logger"
+    autoload :SecurityHeaders, "whoosh/middleware/security_headers"
+    autoload :RequestLimit,    "whoosh/middleware/request_limit"
+    autoload :PluginHooks,     "whoosh/middleware/plugin_hooks"
+  end
+
+  module Streaming
+    autoload :SSE,        "whoosh/streaming/sse"
+    autoload :WebSocket,  "whoosh/streaming/websocket"
+    autoload :LlmStream,  "whoosh/streaming/llm_stream"
+    autoload :StreamBody, "whoosh/streaming/stream_body"
+  end
+
+  module OpenAPI
+    autoload :Generator,       "whoosh/openapi/generator"
+    autoload :UI,              "whoosh/openapi/ui"
+    autoload :SchemaConverter, "whoosh/openapi/schema_converter"
+  end
+
+  module Serialization
+    autoload :Json,       "whoosh/serialization/json"
+    autoload :Msgpack,    "whoosh/serialization/msgpack"
+    autoload :Protobuf,   "whoosh/serialization/protobuf"
+    autoload :Negotiator, "whoosh/serialization/negotiator"
+  end
+
+  module Plugins
+    autoload :Registry, "whoosh/plugins/registry"
+    autoload :Base,     "whoosh/plugins/base"
+  end
+end

metadata

@@ -0,0 +1,265 @@
+--- !ruby/object:Gem::Specification
+name: whoosh
+version: !ruby/object:Gem::Version
+  version: 1.0.1
+platform: ruby
+authors:
+- Johannes Dwi Cahyo
+bindir: exe
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: base64
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
+- !ruby/object:Gem::Dependency
+  name: rack
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: dry-schema
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.13'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.13'
+- !ruby/object:Gem::Dependency
+  name: dry-types
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rackup
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+- !ruby/object:Gem::Dependency
+  name: webrick
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.13'
+- !ruby/object:Gem::Dependency
+  name: rack-test
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: benchmark-ips
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.13'
+description: A fast, secure Ruby API framework inspired by FastAPI with built-in MCP
+  support, auto-generated OpenAPI docs, and seamless AI gem ecosystem integration.
+executables:
+- whoosh
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- exe/whoosh
+- lib/whoosh.rb
+- lib/whoosh/app.rb
+- lib/whoosh/auth/access_control.rb
+- lib/whoosh/auth/api_key.rb
+- lib/whoosh/auth/jwt.rb
+- lib/whoosh/auth/oauth2.rb
+- lib/whoosh/auth/rate_limiter.rb
+- lib/whoosh/auth/token_tracker.rb
+- lib/whoosh/cache.rb
+- lib/whoosh/cache/memory_store.rb
+- lib/whoosh/cache/redis_store.rb
+- lib/whoosh/cli/generators.rb
+- lib/whoosh/cli/main.rb
+- lib/whoosh/cli/project_generator.rb
+- lib/whoosh/config.rb
+- lib/whoosh/database.rb
+- lib/whoosh/dependency_injection.rb
+- lib/whoosh/endpoint.rb
+- lib/whoosh/env_loader.rb
+- lib/whoosh/errors.rb
+- lib/whoosh/http.rb
+- lib/whoosh/http/response.rb
+- lib/whoosh/instrumentation.rb
+- lib/whoosh/job.rb
+- lib/whoosh/jobs.rb
+- lib/whoosh/jobs/memory_backend.rb
+- lib/whoosh/jobs/worker.rb
+- lib/whoosh/logger.rb
+- lib/whoosh/mcp/client.rb
+- lib/whoosh/mcp/client_manager.rb
+- lib/whoosh/mcp/protocol.rb
+- lib/whoosh/mcp/server.rb
+- lib/whoosh/mcp/transport/sse.rb
+- lib/whoosh/mcp/transport/stdio.rb
+- lib/whoosh/metrics.rb
+- lib/whoosh/middleware/cors.rb
+- lib/whoosh/middleware/plugin_hooks.rb
+- lib/whoosh/middleware/request_limit.rb
+- lib/whoosh/middleware/request_logger.rb
+- lib/whoosh/middleware/security_headers.rb
+- lib/whoosh/middleware/stack.rb
+- lib/whoosh/openapi/generator.rb
+- lib/whoosh/openapi/schema_converter.rb
+- lib/whoosh/openapi/ui.rb
+- lib/whoosh/paginate.rb
+- lib/whoosh/performance.rb
+- lib/whoosh/plugins/base.rb
+- lib/whoosh/plugins/registry.rb
+- lib/whoosh/request.rb
+- lib/whoosh/response.rb
+- lib/whoosh/router.rb
+- lib/whoosh/schema.rb
+- lib/whoosh/serialization/json.rb
+- lib/whoosh/serialization/msgpack.rb
+- lib/whoosh/serialization/negotiator.rb
+- lib/whoosh/serialization/protobuf.rb
+- lib/whoosh/shutdown.rb
+- lib/whoosh/storage.rb
+- lib/whoosh/storage/local.rb
+- lib/whoosh/storage/s3.rb
+- lib/whoosh/streaming/llm_stream.rb
+- lib/whoosh/streaming/sse.rb
+- lib/whoosh/streaming/stream_body.rb
+- lib/whoosh/streaming/websocket.rb
+- lib/whoosh/test.rb
+- lib/whoosh/types.rb
+- lib/whoosh/uploaded_file.rb
+- lib/whoosh/version.rb
+homepage: https://github.com/johannesdwicahyo/whoosh
+licenses:
+- MIT
+metadata: {}
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.4.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: AI-first Ruby API framework
+test_files: []