woods 1.6.4 → 2.0.0.beta1

data/lib/woods/formatting/generic_adapter.rb

@@ -1,56 +0,0 @@
-# frozen_string_literal: true
-
-require_relative 'base'
-
-module Woods
-  module Formatting
-    # Formats assembled context as plain text for generic LLM consumption.
-    #
-    # Produces plain text with:
-    # - `=== CODEBASE CONTEXT ===` header
-    # - Token usage line
-    # - Content separated by `---` dividers
-    # - Sources in bracket notation
-    #
-    # @example
-    #   adapter = GenericAdapter.new
-    #   text = adapter.format(assembled_context)
-    #   # => "=== CODEBASE CONTEXT ===\nTokens: 42 / 8000\n..."
-    #
-    class GenericAdapter < Base
-      # Format assembled context as plain text.
-      #
-      # @param assembled_context [Woods::Retrieval::AssembledContext]
-      # @return [String] Plain text formatted context
-      def format(assembled_context)
-        parts = []
-        parts << '=== CODEBASE CONTEXT ==='
-        parts << "Tokens: #{assembled_context.tokens_used} / #{assembled_context.budget}"
-        parts << '---'
-        parts << assembled_context.context
-        parts.concat(format_sources(assembled_context.sources))
-        parts.join("\n")
-      end
-
-      private
-
-      # Format sources in bracket notation.
-      #
-      # @param sources [Array<Hash>]
-      # @return [Array<String>] Lines to append
-      def format_sources(sources)
-        return [] if sources.empty?
-
-        lines = []
-        lines << '---'
-        sources.each do |source|
-          identifier = source[:identifier]
-          type = source[:type]
-          score = source[:score]
-          lines << "[Source: #{identifier} (#{type}) \u2014 score: #{score}]"
-        end
-        lines
-      end
-    end
-  end
-end

data/lib/woods/formatting/gpt_adapter.rb

@@ -1,64 +0,0 @@
-# frozen_string_literal: true
-
-require_relative 'base'
-
-module Woods
-  module Formatting
-    # Formats assembled context as Markdown for GPT models.
-    #
-    # Produces Markdown with:
-    # - `## Codebase Context` heading
-    # - Token usage in bold
-    # - Content in a fenced Ruby code block
-    # - Sources as a bullet list
-    #
-    # @example
-    #   adapter = GptAdapter.new
-    #   md = adapter.format(assembled_context)
-    #   # => "## Codebase Context\n\n**Tokens:** 42/8000\n..."
-    #
-    class GptAdapter < Base
-      # Format assembled context as Markdown for GPT.
-      #
-      # @param assembled_context [Woods::Retrieval::AssembledContext]
-      # @return [String] Markdown-formatted context
-      def format(assembled_context)
-        parts = []
-        parts << '## Codebase Context'
-        parts << ''
-        parts << "**Tokens:** #{assembled_context.tokens_used}/#{assembled_context.budget}"
-        parts << ''
-        parts << '---'
-        parts << ''
-        parts << '```ruby'
-        parts << assembled_context.context
-        parts << '```'
-        parts.concat(format_sources(assembled_context.sources))
-        parts.join("\n")
-      end
-
-      private
-
-      # Format sources as a Markdown bullet list.
-      #
-      # @param sources [Array<Hash>]
-      # @return [Array<String>] Lines to append
-      def format_sources(sources)
-        return [] if sources.empty?
-
-        lines = []
-        lines << ''
-        lines << '### Sources'
-        lines << ''
-        sources.each do |source|
-          identifier = source[:identifier]
-          type = source[:type]
-          score = source[:score]
-          file_path = source[:file_path]
-          lines << "- **#{identifier}** (#{type}) \u2014 score: #{score}, file: #{file_path}"
-        end
-        lines
-      end
-    end
-  end
-end

data/lib/woods/mcp/http_transport_options.rb

@@ -1,15 +0,0 @@
-# frozen_string_literal: true
-
-require_relative 'origin_policy'
-
-module Woods
-  module MCP
-    # Keep the SDK DNS-rebinding guard aligned with Woods' explicit allowlist.
-    # Empty configuration retains the SDK's strict loopback defaults.
-    module HttpTransportOptions
-      def self.for(origins)
-        OriginPolicy.new(allowed_origins: origins).transport_options
-      end
-    end
-  end
-end

data/lib/woods/mcp/origin_policy.rb

@@ -1,113 +0,0 @@
-# frozen_string_literal: true
-
-require 'uri'
-require_relative '../util/host_guard'
-
-module Woods
-  module MCP
-    # One immutable policy for the outer Rack guard and the legacy MCP SDK.
-    # Keep SDK rebinding protection active and pass original request headers.
-    class OriginPolicy
-      LOOPBACK_HOSTS = %w[localhost 127.0.0.1 ::1].freeze
-      DEFAULT_ORIGINS = %w[
-        http://localhost http://127.0.0.1 http://[::1]
-        https://localhost https://127.0.0.1 https://[::1]
-      ].freeze
-
-      attr_reader :transport_options
-
-      def initialize(allowed_origins: nil)
-        entries = Array(allowed_origins).compact.reject do |entry|
-          entry.to_s.valid_encoding? && entry.to_s.strip.empty?
-        end
-        @explicit = entries.map { |entry| configured_origin(entry) }.uniq.freeze
-        @allowed = @explicit.empty? ? DEFAULT_ORIGINS : @explicit
-        @hosts = @explicit.map do |origin|
-          host = authority(origin)
-          (host.end_with?(']') ? host[1...-1] : host).freeze
-        end.uniq.freeze
-        @transport_options = if @explicit.empty?
-                               {}.freeze
-                             else
-                               { allowed_origins: transport_origins, allowed_hosts: @hosts }.freeze
-                             end
-        freeze
-      end
-
-      def host_allowed?(host)
-        return true if host.nil?
-        return false unless host.is_a?(String) && host.valid_encoding?
-
-        parsed = parsed_origin("http://#{host}")
-        return false unless parsed
-
-        hostname = parsed.hostname.downcase
-        return false if Util::HostGuard.suspicious_numeric_host?(hostname)
-        return true if LOOPBACK_HOSTS.include?(hostname)
-
-        @hosts.include?(host.downcase) || @hosts.include?(hostname)
-      end
-
-      def origin_allowed?(origin, host:)
-        return true if origin.nil?
-        return false unless parsed_origin(origin)
-
-        normalized = normalized_origin(origin)
-        return true if @explicit.include?(normalized)
-        return false unless @allowed.include?(normalized) || @allowed.include?(normalized.sub(/:\d+\z/, ''))
-        return false unless host.is_a?(String) && host.valid_encoding?
-
-        default_port = normalized.start_with?('https://') ? ':443' : ':80'
-        authority(normalized).delete_suffix(default_port) == host.downcase.delete_suffix(default_port)
-      end
-
-      private
-
-      def configured_origin(entry)
-        raw = entry.to_s
-        value = raw.strip.downcase.delete_suffix('/') if raw.valid_encoding?
-        unless parsed_origin(value)
-          label = raw.b.byteslice(0, 160).inspect
-          raise ArgumentError, "Invalid MCP allowed origin #{label}: expected http(s)://host[:port] without a path"
-        end
-
-        normalized_origin(value).freeze
-      end
-
-      def normalized_origin(origin)
-        value = origin.downcase
-        value.delete_suffix(value.start_with?('https://') ? ':443' : ':80')
-      end
-
-      # SDK 0.23 and 0.25 compare explicit entries literally. Both default-port
-      # spellings are needed for cross-origin requests; same-authority matching
-      # is otherwise performed independently by both layers.
-      def transport_origins
-        @explicit.flat_map do |origin|
-          parsed = parsed_origin(origin)
-          port = parsed.scheme == 'https' ? 443 : 80
-          parsed.port == port ? [origin, "#{origin}:#{port}"] : [origin]
-        end.uniq.map(&:freeze).freeze
-      end
-
-      def authority(origin)
-        origin.sub(%r{\Ahttps?://}, '')
-      end
-
-      def parsed_origin(origin)
-        return unless origin.is_a?(String) && origin.valid_encoding? && origin.ascii_only?
-        return if origin.match?(/[[:space:][:cntrl:]]/)
-
-        parsed = URI.parse(origin)
-        return unless %w[http https].include?(parsed.scheme&.downcase)
-        return unless parsed.host && !parsed.host.empty?
-        return unless parsed.path.to_s.empty? && !parsed.userinfo && !parsed.query && !parsed.fragment
-        return unless parsed.port&.between?(1, 65_535)
-
-        parsed
-      rescue URI::InvalidURIError, ArgumentError
-        nil
-      end
-    end
-  end
-end

data/lib/woods/notion/mapper.rb

@@ -1,40 +0,0 @@
-# frozen_string_literal: true
-
-require_relative 'mappers/shared'
-require_relative 'mappers/model_mapper'
-require_relative 'mappers/column_mapper'
-require_relative 'mappers/migration_mapper'
-
-module Woods
-  module Notion
-    # Dispatcher for Notion mappers. Returns the appropriate mapper for a unit type.
-    #
-    # @example
-    #   mapper = Mapper.for("model")
-    #   properties = mapper.map(unit_data)
-    #
-    class Mapper
-      REGISTRY = {
-        'model' => Mappers::ModelMapper,
-        'column' => Mappers::ColumnMapper,
-        'migration' => Mappers::MigrationMapper
-      }.freeze
-
-      # Get a mapper instance for a unit type.
-      #
-      # @param type [String] Unit type name (e.g. "model", "column", "migration")
-      # @return [Object, nil] Mapper instance or nil if type is not supported
-      def self.for(type)
-        klass = REGISTRY[type]
-        klass&.new
-      end
-
-      # List all supported unit types.
-      #
-      # @return [Array<String>]
-      def self.supported_types
-        REGISTRY.keys
-      end
-    end
-  end
-end

data/lib/woods/observability/health_check.rb

@@ -1,79 +0,0 @@
-# frozen_string_literal: true
-
-module Woods
-  module Observability
-    # Probes configured components and reports overall system health.
-    #
-    # Checks vector store, metadata store, and embedding provider by calling
-    # lightweight operations on each. Components that are nil are reported
-    # as :not_configured and do not affect the overall healthy? status.
-    #
-    # @example
-    #   check = HealthCheck.new(
-    #     vector_store: vector_store,
-    #     metadata_store: metadata_store,
-    #     embedding_provider: provider
-    #   )
-    #   status = check.run
-    #   status.healthy?    # => true
-    #   status.components  # => { vector_store: :ok, metadata_store: :ok, embedding_provider: :ok }
-    #
-    class HealthCheck
-      # Value object representing the result of a health check.
-      HealthStatus = Struct.new(:healthy?, :components, keyword_init: true)
-
-      # @param vector_store [Object, nil] Vector store adapter (must respond to #count)
-      # @param metadata_store [Object, nil] Metadata store adapter (must respond to #count)
-      # @param embedding_provider [Object, nil] Embedding provider (must respond to #embed)
-      def initialize(vector_store: nil, metadata_store: nil, embedding_provider: nil)
-        @vector_store = vector_store
-        @metadata_store = metadata_store
-        @embedding_provider = embedding_provider
-      end
-
-      # Run health probes on all configured components.
-      #
-      # @return [HealthStatus] Result with healthy? flag and per-component status
-      def run
-        components = {
-          vector_store: probe_store(@vector_store),
-          metadata_store: probe_store(@metadata_store),
-          embedding_provider: probe_provider(@embedding_provider)
-        }
-
-        all_healthy = components.values.all? { |status| %i[ok not_configured].include?(status) }
-
-        HealthStatus.new(healthy?: all_healthy, components: components)
-      end
-
-      private
-
-      # Probe a store component by calling #count.
-      #
-      # @param store [Object, nil] Store adapter
-      # @return [Symbol] :ok, :error, or :not_configured
-      def probe_store(store)
-        return :not_configured if store.nil?
-
-        store.count
-        :ok
-      rescue StandardError
-        :error
-      end
-
-      # Probe an embedding provider by checking its capabilities without making network calls.
-      #
-      # @param provider [Object, nil] Embedding provider
-      # @return [Symbol] :ok, :error, or :not_configured
-      def probe_provider(provider)
-        return :not_configured if provider.nil?
-
-        if provider.respond_to?(:embed) && provider.respond_to?(:dimensions)
-          :ok
-        else
-          :error
-        end
-      end
-    end
-  end
-end

data/lib/woods/observability/instrumentation.rb

@@ -1,34 +0,0 @@
-# frozen_string_literal: true
-
-module Woods
-  module Observability
-    # Lightweight instrumentation wrapper that delegates to ActiveSupport::Notifications
-    # when available, and falls back to a simple yield otherwise.
-    #
-    # @example
-    #   Instrumentation.instrument('woods.extraction', unit: 'User') do
-    #     extract_unit(user_model)
-    #   end
-    #
-    module Instrumentation
-      module_function
-
-      # Instrument a block of code with an event name and payload.
-      #
-      # Delegates to ActiveSupport::Notifications.instrument when available.
-      # Otherwise, yields the block directly.
-      #
-      # @param event [String] Event name (e.g., 'woods.extraction')
-      # @param payload [Hash] Additional data to include with the event
-      # @yield [payload] The block to instrument
-      # @return [Object] The return value of the block
-      def instrument(event, payload = {}, &block)
-        if defined?(ActiveSupport::Notifications)
-          ActiveSupport::Notifications.instrument(event, payload, &block)
-        elsif block
-          yield payload
-        end
-      end
-    end
-  end
-end