schematichq 1.4.0

data/lib/schematic/entitlements/types/list_feature_companies_params.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      # Input parameters
+      class ListFeatureCompaniesParams < Internal::Types::Model
+        field :feature_id, -> { String }, optional: true, nullable: false
+        field :limit, -> { Integer }, optional: true, nullable: false
+        field :offset, -> { Integer }, optional: true, nullable: false
+        field :q, -> { String }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_feature_companies_request.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class ListFeatureCompaniesRequest < Internal::Types::Model
+        field :feature_id, -> { String }, optional: false, nullable: false
+        field :q, -> { String }, optional: true, nullable: false
+        field :limit, -> { Integer }, optional: true, nullable: false
+        field :offset, -> { Integer }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_feature_companies_response.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class ListFeatureCompaniesResponse < Internal::Types::Model
+        field :data, -> { Internal::Types::Array[Schematic::Types::FeatureCompanyResponseData] }, optional: false, nullable: false
+        field :params, -> { Schematic::Entitlements::Types::ListFeatureCompaniesParams }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_feature_usage_params.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      # Input parameters
+      class ListFeatureUsageParams < Internal::Types::Model
+        field :company_id, -> { String }, optional: true, nullable: false
+        field :company_keys, -> { Internal::Types::Hash[String, String] }, optional: true, nullable: false
+        field :feature_ids, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+        field :include_usage_aggregation, -> { Internal::Types::Boolean }, optional: true, nullable: false
+        field :limit, -> { Integer }, optional: true, nullable: false
+        field :offset, -> { Integer }, optional: true, nullable: false
+        field :q, -> { String }, optional: true, nullable: false
+        field :without_negative_entitlements, -> { Internal::Types::Boolean }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_feature_usage_request.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class ListFeatureUsageRequest < Internal::Types::Model
+        field :company_id, -> { String }, optional: true, nullable: false
+        field :company_keys, -> { Internal::Types::Hash[String, String] }, optional: true, nullable: false
+        field :feature_ids, -> { String }, optional: true, nullable: false
+        field :include_usage_aggregation, -> { Internal::Types::Boolean }, optional: true, nullable: false
+        field :q, -> { String }, optional: true, nullable: false
+        field :without_negative_entitlements, -> { Internal::Types::Boolean }, optional: true, nullable: false
+        field :limit, -> { Integer }, optional: true, nullable: false
+        field :offset, -> { Integer }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_feature_usage_response.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class ListFeatureUsageResponse < Internal::Types::Model
+        field :data, -> { Internal::Types::Array[Schematic::Types::FeatureUsageResponseData] }, optional: false, nullable: false
+        field :params, -> { Schematic::Entitlements::Types::ListFeatureUsageParams }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_feature_users_params.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      # Input parameters
+      class ListFeatureUsersParams < Internal::Types::Model
+        field :feature_id, -> { String }, optional: true, nullable: false
+        field :limit, -> { Integer }, optional: true, nullable: false
+        field :offset, -> { Integer }, optional: true, nullable: false
+        field :q, -> { String }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_feature_users_request.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class ListFeatureUsersRequest < Internal::Types::Model
+        field :feature_id, -> { String }, optional: false, nullable: false
+        field :q, -> { String }, optional: true, nullable: false
+        field :limit, -> { Integer }, optional: true, nullable: false
+        field :offset, -> { Integer }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_feature_users_response.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class ListFeatureUsersResponse < Internal::Types::Model
+        field :data, -> { Internal::Types::Array[Schematic::Types::FeatureCompanyUserResponseData] }, optional: false, nullable: false
+        field :params, -> { Schematic::Entitlements::Types::ListFeatureUsersParams }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_plan_entitlements_params.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      # Input parameters
+      class ListPlanEntitlementsParams < Internal::Types::Model
+        field :feature_id, -> { String }, optional: true, nullable: false
+        field :feature_ids, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+        field :ids, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+        field :limit, -> { Integer }, optional: true, nullable: false
+        field :offset, -> { Integer }, optional: true, nullable: false
+        field :plan_id, -> { String }, optional: true, nullable: false
+        field :plan_ids, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+        field :plan_version_id, -> { String }, optional: true, nullable: false
+        field :plan_version_ids, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+        field :q, -> { String }, optional: true, nullable: false
+        field :with_metered_products, -> { Internal::Types::Boolean }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_plan_entitlements_request.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class ListPlanEntitlementsRequest < Internal::Types::Model
+        field :feature_id, -> { String }, optional: true, nullable: false
+        field :feature_ids, -> { String }, optional: true, nullable: false
+        field :ids, -> { String }, optional: true, nullable: false
+        field :plan_id, -> { String }, optional: true, nullable: false
+        field :plan_ids, -> { String }, optional: true, nullable: false
+        field :plan_version_id, -> { String }, optional: true, nullable: false
+        field :plan_version_ids, -> { String }, optional: true, nullable: false
+        field :q, -> { String }, optional: true, nullable: false
+        field :with_metered_products, -> { Internal::Types::Boolean }, optional: true, nullable: false
+        field :limit, -> { Integer }, optional: true, nullable: false
+        field :offset, -> { Integer }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/list_plan_entitlements_response.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class ListPlanEntitlementsResponse < Internal::Types::Model
+        field :data, -> { Internal::Types::Array[Schematic::Types::PlanEntitlementResponseData] }, optional: false, nullable: false
+        field :params, -> { Schematic::Entitlements::Types::ListPlanEntitlementsParams }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/update_company_override_request_body.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class UpdateCompanyOverrideRequestBody < Internal::Types::Model
+        field :company_override_id, -> { String }, optional: false, nullable: false
+        field :credit_consumption_rate, -> { Integer }, optional: true, nullable: false
+        field :expiration_date, -> { String }, optional: true, nullable: false
+        field :metric_period, -> { Schematic::Entitlements::Types::UpdateCompanyOverrideRequestBodyMetricPeriod }, optional: true, nullable: false
+        field :metric_period_month_reset, -> { Schematic::Entitlements::Types::UpdateCompanyOverrideRequestBodyMetricPeriodMonthReset }, optional: true, nullable: false
+        field :note, -> { String }, optional: true, nullable: false
+        field :value_bool, -> { Internal::Types::Boolean }, optional: true, nullable: false
+        field :value_credit_id, -> { String }, optional: true, nullable: false
+        field :value_numeric, -> { Integer }, optional: true, nullable: false
+        field :value_trait_id, -> { String }, optional: true, nullable: false
+        field :value_type, -> { Schematic::Types::EntitlementValueType }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/update_company_override_request_body_metric_period.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      module UpdateCompanyOverrideRequestBodyMetricPeriod
+        extend Schematic::Internal::Types::Enum
+
+        ALL_TIME = "all_time"
+        CURRENT_MONTH = "current_month"
+        CURRENT_WEEK = "current_week"
+        CURRENT_DAY = "current_day"
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/update_company_override_request_body_metric_period_month_reset.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      module UpdateCompanyOverrideRequestBodyMetricPeriodMonthReset
+        extend Schematic::Internal::Types::Enum
+
+        FIRST_OF_MONTH = "first_of_month"
+        BILLING_CYCLE = "billing_cycle"
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/update_company_override_response.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class UpdateCompanyOverrideResponse < Internal::Types::Model
+        field :data, -> { Schematic::Types::CompanyOverrideResponseData }, optional: false, nullable: false
+        field :params, -> { Internal::Types::Hash[String, Object] }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/update_plan_entitlement_request_body.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class UpdatePlanEntitlementRequestBody < Internal::Types::Model
+        field :plan_entitlement_id, -> { String }, optional: false, nullable: false
+        field :billing_product_id, -> { String }, optional: true, nullable: false
+        field :billing_threshold, -> { Integer }, optional: true, nullable: false
+        field :credit_consumption_rate, -> { Integer }, optional: true, nullable: false
+        field :currency, -> { String }, optional: true, nullable: false
+        field :currency_prices, -> { Internal::Types::Array[Schematic::Types::CurrencyPriceRequestBody] }, optional: true, nullable: false
+        field :metric_period, -> { Schematic::Entitlements::Types::UpdatePlanEntitlementRequestBodyMetricPeriod }, optional: true, nullable: false
+        field :metric_period_month_reset, -> { Schematic::Entitlements::Types::UpdatePlanEntitlementRequestBodyMetricPeriodMonthReset }, optional: true, nullable: false
+        field :monthly_metered_price_id, -> { String }, optional: true, nullable: false
+        field :monthly_price_tiers, -> { Internal::Types::Array[Schematic::Types::CreatePriceTierRequestBody] }, optional: true, nullable: false
+        field :monthly_unit_price, -> { Integer }, optional: true, nullable: false
+        field :monthly_unit_price_decimal, -> { String }, optional: true, nullable: false
+        field :overage_billing_product_id, -> { String }, optional: true, nullable: false
+        field :price_behavior, -> { Schematic::Types::EntitlementPriceBehavior }, optional: true, nullable: false
+        field :price_tiers, -> { Internal::Types::Array[Schematic::Types::CreatePriceTierRequestBody] }, optional: true, nullable: false
+        field :soft_limit, -> { Integer }, optional: true, nullable: false
+        field :tier_mode, -> { Schematic::Types::BillingTiersMode }, optional: true, nullable: false
+        field :value_bool, -> { Internal::Types::Boolean }, optional: true, nullable: false
+        field :value_credit_id, -> { String }, optional: true, nullable: false
+        field :value_numeric, -> { Integer }, optional: true, nullable: false
+        field :value_trait_id, -> { String }, optional: true, nullable: false
+        field :value_type, -> { Schematic::Types::EntitlementValueType }, optional: false, nullable: false
+        field :yearly_metered_price_id, -> { String }, optional: true, nullable: false
+        field :yearly_price_tiers, -> { Internal::Types::Array[Schematic::Types::CreatePriceTierRequestBody] }, optional: true, nullable: false
+        field :yearly_unit_price, -> { Integer }, optional: true, nullable: false
+        field :yearly_unit_price_decimal, -> { String }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/update_plan_entitlement_request_body_metric_period.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      module UpdatePlanEntitlementRequestBodyMetricPeriod
+        extend Schematic::Internal::Types::Enum
+
+        ALL_TIME = "all_time"
+        CURRENT_MONTH = "current_month"
+        CURRENT_WEEK = "current_week"
+        CURRENT_DAY = "current_day"
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/update_plan_entitlement_request_body_metric_period_month_reset.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      module UpdatePlanEntitlementRequestBodyMetricPeriodMonthReset
+        extend Schematic::Internal::Types::Enum
+
+        FIRST_OF_MONTH = "first_of_month"
+        BILLING_CYCLE = "billing_cycle"
+      end
+    end
+  end
+end

data/lib/schematic/entitlements/types/update_plan_entitlement_response.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Entitlements
+    module Types
+      class UpdatePlanEntitlementResponse < Internal::Types::Model
+        field :data, -> { Schematic::Types::PlanEntitlementResponseData }, optional: false, nullable: false
+        field :params, -> { Internal::Types::Hash[String, Object] }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/environment.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module Schematic
+  class Environment
+    DEFAULT = "https://api.schematichq.com"
+  end
+end

data/lib/schematic/errors/api_error.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Errors
+    class ApiError < StandardError
+    end
+  end
+end

data/lib/schematic/errors/client_error.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Errors
+    class ClientError < ResponseError
+    end
+
+    class UnauthorizedError < ClientError
+    end
+
+    class ForbiddenError < ClientError
+    end
+
+    class NotFoundError < ClientError
+    end
+  end
+end

data/lib/schematic/errors/redirect_error.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Errors
+    class RedirectError < ResponseError
+    end
+  end
+end

data/lib/schematic/errors/response_error.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Errors
+    class ResponseError < ApiError
+      attr_reader :code
+
+      def initialize(msg, code:)
+        @code = code
+        super(msg)
+      end
+
+      def inspect
+        "#<#{self.class.name} @code=#{code} @body=#{message}>"
+      end
+
+      # Returns the most appropriate error class for the given code.
+      #
+      # @return [Class]
+      def self.subclass_for_code(code)
+        case code
+        when 300..399
+          RedirectError
+        when 401
+          UnauthorizedError
+        when 403
+          ForbiddenError
+        when 404
+          NotFoundError
+        when 400..499
+          ClientError
+        when 503
+          ServiceUnavailableError
+        when 500..599
+          ServerError
+        else
+          ResponseError
+        end
+      end
+    end
+  end
+end

data/lib/schematic/errors/server_error.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Errors
+    class ServerError < ResponseError
+    end
+
+    class ServiceUnavailableError < ApiError
+    end
+  end
+end

data/lib/schematic/errors/timeout_error.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Errors
+    class TimeoutError < ApiError
+    end
+  end
+end

data/lib/schematic/event_buffer.rb

@@ -0,0 +1,180 @@
+# frozen_string_literal: true
+
+require "json"
+require "net/http"
+require "uri"
+
+module Schematic
+  class EventBuffer
+    DEFAULT_FLUSH_INTERVAL = 5.0 # seconds (canonical Go value)
+    DEFAULT_MAX_BATCH_SIZE = 100
+    DEFAULT_MAX_RETRIES = 3
+    DEFAULT_INITIAL_RETRY_DELAY = 1.0 # seconds
+    JITTER_FACTOR = 0.25
+    DEFAULT_CAPTURE_BASE_URL = "https://c.schematichq.com"
+
+    def initialize(api_key:, logger:, interval: DEFAULT_FLUSH_INTERVAL, max_batch_size: DEFAULT_MAX_BATCH_SIZE,
+                   offline: false, capture_base_url: DEFAULT_CAPTURE_BASE_URL)
+      @api_key = api_key
+      @logger = logger
+      @interval = interval
+      @max_batch_size = max_batch_size
+      @offline = offline
+      @capture_url = URI.parse("#{capture_base_url}/batch")
+      @events = []
+      @mutex = Mutex.new
+      @stopped = false
+      @flushing = false
+      @flush_done = ConditionVariable.new
+
+      start_periodic_flush unless @offline
+    end
+
+    def push(event)
+      return if @offline
+
+      should_flush = false
+      @mutex.synchronize do
+        return if @stopped
+
+        @events << event
+        should_flush = @events.size >= @max_batch_size
+      end
+
+      flush if should_flush
+    end
+
+    def flush
+      events_to_send = nil
+      did_set_flushing = false
+      @mutex.synchronize do
+        return if @flushing || @events.empty?
+
+        @flushing = true
+        did_set_flushing = true
+        events_to_send = @events.dup
+        @events.clear
+      end
+
+      return unless events_to_send&.any?
+
+      send_batch(events_to_send)
+
+      # Events may have accumulated while we were sending. Drain them so
+      # a size-triggered flush that lost the race doesn't have to wait for
+      # the next periodic interval.
+      drain_pending
+    ensure
+      if did_set_flushing
+        @mutex.synchronize do
+          @flushing = false
+          @flush_done.broadcast
+        end
+      end
+    end
+
+    def stop
+      @mutex.synchronize do
+        @stopped = true
+
+        # Wait for any in-flight flush to complete before our final flush,
+        # so we don't skip events that arrived during the in-flight batch.
+        @flush_done.wait(@mutex, 30) if @flushing
+      end
+
+      flush
+
+      @flush_thread&.join(5)
+    end
+
+    private
+
+    def start_periodic_flush
+      @flush_thread = Thread.new do
+        loop do
+          sleep(@interval)
+          break if @stopped
+
+          begin
+            flush
+          rescue StandardError => e
+            @logger.error("Event flush error: #{e.message}")
+          end
+        end
+      end
+      @flush_thread.abort_on_exception = false
+    end
+
+    def drain_pending
+      events_to_send = @mutex.synchronize do
+        return if @events.empty?
+
+        batch = @events.dup
+        @events.clear
+        batch
+      end
+
+      send_batch(events_to_send) if events_to_send&.any?
+    end
+
+    def send_batch(events)
+      payload = {
+        events: events.map { |e| event_to_capture_payload(e) }
+      }
+
+      attempts = 0
+      while attempts <= DEFAULT_MAX_RETRIES
+        begin
+          post_to_capture_service(payload)
+          @logger.debug("Flushed #{events.size} events to capture service")
+          return
+        rescue StandardError => e
+          attempts += 1
+          if attempts > DEFAULT_MAX_RETRIES
+            @logger.error("Failed to send event batch after #{DEFAULT_MAX_RETRIES} retries: #{e.message}")
+            return
+          end
+
+          delay = calculate_backoff(attempts - 1)
+          @logger.warn("Event batch send failed (attempt #{attempts}/#{DEFAULT_MAX_RETRIES}), retrying in #{delay.round(2)}s: #{e.message}")
+          sleep(delay)
+        end
+      end
+    end
+
+    def post_to_capture_service(payload)
+      http = Net::HTTP.new(@capture_url.host, @capture_url.port)
+      http.use_ssl = @capture_url.scheme == "https"
+      http.open_timeout = 10
+      http.read_timeout = 10
+
+      request = Net::HTTP::Post.new(@capture_url.path)
+      request["Content-Type"] = "application/json"
+      request["X-Schematic-Api-Key"] = @api_key
+      request.body = JSON.generate(payload)
+
+      response = http.request(request)
+      return if response.is_a?(Net::HTTPSuccess)
+
+      raise "capture service returned #{response.code}: #{response.body}"
+    end
+
+    def calculate_backoff(attempt)
+      base_delay = DEFAULT_INITIAL_RETRY_DELAY * (2**attempt)
+      jitter = base_delay * JITTER_FACTOR * ((rand * 2) - 1) # ±25%
+      base_delay + jitter
+    end
+
+    # Transform to capture service payload format (differs from Fern API):
+    # - Field is `type` not `event_type`
+    # - Each event includes `api_key`
+    def event_to_capture_payload(event)
+      {
+        api_key: @api_key,
+        type: event[:event_type],
+        body: event[:body],
+        sent_at: event[:sent_at] || Time.now.utc.iso8601
+      }
+    end
+  end
+end