schematichq 1.5.2 → 1.5.4

data/lib/schematic/credits/leases/wire_client.rb

@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+require "securerandom"
+require "time"
+
+module Schematic
+  module Credits
+    module Leases
+      # What the server says a lease is after an acquire or an extend. It is
+      # also what installs a lease into a store: the local balance is derived,
+      # never supplied.
+      LeaseGrant = Struct.new(:lease_id, :company_id, :credit_type_id, :granted_amount, :expires_at,
+                              keyword_init: true)
+
+      # The lease lifecycle over the generated API client. Separated from
+      # LeaseManager so tests and the conformance runner can script the wire.
+      #
+      # Both calls take the client's default retry policy. Acquire is safe to
+      # retry because the server hands back the slot's existing active lease
+      # rather than opening a second one. An extend is an increment, so a retry
+      # after a lost response would grant the tranche twice; the request carries
+      # an idempotency key minted once per extend to collapse them.
+      class ApiWireClient
+        def initialize(credits_client:)
+          @credits = credits_client
+        end
+
+        def acquire(company_id:, credit_type_id:, requested_amount:, expires_at:, request_options: {})
+          response = @credits.acquire_credit_lease(
+            request_options: request_options,
+            company_id: company_id,
+            credit_type_id: credit_type_id,
+            # Rounded up, not to nearest: the API takes a whole number, and a
+            # shortfall of 10.4 asked for as 10 leaves the retried reserve short
+            # by the same fraction every time.
+            requested_amount: Leases.wire_quantity(requested_amount),
+            expires_at: expires_at.utc.iso8601
+          )
+          grant_from(response)
+        end
+
+        def extend(lease_id:, additional_amount:, expires_at:, idempotency_key: nil, request_options: {})
+          body = {
+            lease_id: lease_id,
+            additional_amount: Leases.wire_quantity(additional_amount),
+            expires_at: expires_at.utc.iso8601,
+            # The key is minted once here, outside the retry loop, so every
+            # attempt of this extend carries the same one and the server folds
+            # them into a single increment. A caller that already minted one
+            # keeps it: the manager does, so its single-flight followers settle
+            # against the same key.
+            idempotency_key: idempotency_key || SecureRandom.uuid
+          }
+          grant_from(@credits.extend_credit_lease(request_options: request_options, **body))
+        end
+
+        def release(lease_id:, request_options: {})
+          @credits.release_credit_lease(request_options: request_options, lease_id: lease_id)
+          nil
+        end
+
+        private
+
+        def grant_from(response)
+          data = response&.data
+          raise "credit lease response carried no data" if data.nil?
+
+          LeaseGrant.new(
+            lease_id: data.id,
+            company_id: data.company_id,
+            credit_type_id: data.credit_type_id,
+            granted_amount: data.granted_amount.to_f,
+            expires_at: parse_time(data.expires_at)
+          )
+        end
+
+        def parse_time(value)
+          Leases.parse_api_time(value)
+        end
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/acquire_credit_lease_request_body.rb

@@ -10,7 +10,9 @@ module Schematic
 
         field :expires_at, -> { String }, optional: true, nullable: false
 
-        field :requested_amount, -> { Integer }, optional: false, nullable: false
+        field :requested_amount, -> { Float }, optional: false, nullable: false
+
+        field :user_id, -> { String }, optional: true, nullable: false
       end
     end
   end

data/lib/schematic/credits/types/create_billing_credit_request_body.rb

@@ -6,7 +6,7 @@ module Schematic
       class CreateBillingCreditRequestBody < Internal::Types::Model
         field :burn_strategy, -> { Schematic::Types::BillingCreditBurnStrategy }, optional: true, nullable: false
 
-        field :currency, -> { String }, optional: false, nullable: false
+        field :currency, -> { Schematic::Types::Currency }, optional: false, nullable: false
 
         field :currency_prices, -> { Internal::Types::Array[Schematic::Types::CreditCurrencyPriceRequestBody] }, optional: true, nullable: false
 

data/lib/schematic/credits/types/create_company_credit_grant.rb

@@ -12,7 +12,7 @@ module Schematic
 
         field :credit_id, -> { String }, optional: false, nullable: false
 
-        field :currency, -> { String }, optional: true, nullable: false
+        field :currency, -> { Schematic::Types::Currency }, optional: true, nullable: false
 
         field :expires_at, -> { String }, optional: true, nullable: false
 

data/lib/schematic/credits/types/create_credit_bundle_request_body.rb

@@ -12,7 +12,7 @@ module Schematic
 
         field :credit_id, -> { String }, optional: false, nullable: false
 
-        field :currency, -> { String }, optional: false, nullable: false
+        field :currency, -> { Schematic::Types::Currency }, optional: false, nullable: false
 
         field :currency_prices, -> { Internal::Types::Array[Schematic::Types::CreditBundleCurrencyPriceRequestBody] }, optional: true, nullable: false
 

data/lib/schematic/credits/types/create_credit_spend_policy_request_body.rb

@@ -10,9 +10,13 @@ module Schematic
 
         field :label, -> { String }, optional: true, nullable: false
 
-        field :max_per_draw, -> { Integer }, optional: false, nullable: false
+        field :max_per_draw, -> { Float }, optional: true, nullable: false
 
         field :user_id, -> { String }, optional: true, nullable: false
+
+        field :window_amount, -> { Float }, optional: true, nullable: false
+
+        field :window_unit, -> { Schematic::Types::CreditSpendWindowUnit }, optional: true, nullable: false
       end
     end
   end

data/lib/schematic/credits/types/extend_credit_lease_request_body.rb

@@ -6,11 +6,13 @@ module Schematic
       class ExtendCreditLeaseRequestBody < Internal::Types::Model
         field :lease_id, -> { String }, optional: false, nullable: false
 
-        field :additional_amount, -> { Integer }, optional: false, nullable: false
+        field :additional_amount, -> { Float }, optional: false, nullable: false
 
         field :expires_at, -> { String }, optional: true, nullable: false
 
         field :idempotency_key, -> { String }, optional: true, nullable: false
+
+        field :user_id, -> { String }, optional: true, nullable: false
       end
     end
   end

data/lib/schematic/credits/types/get_credit_balance_time_series_params.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      # Input parameters
+      class GetCreditBalanceTimeSeriesParams < Internal::Types::Model
+        field :billing_credit_id, -> { String }, optional: true, nullable: false
+
+        field :company_id, -> { String }, optional: true, nullable: false
+
+        field :end_time, -> { String }, optional: true, nullable: false
+
+        field :start_time, -> { String }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/get_credit_balance_time_series_request.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class GetCreditBalanceTimeSeriesRequest < Internal::Types::Model
+        field :billing_credit_id, -> { String }, optional: false, nullable: false
+
+        field :company_id, -> { String }, optional: false, nullable: false
+
+        field :end_time, -> { String }, optional: false, nullable: false
+
+        field :start_time, -> { String }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/get_credit_balance_time_series_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class GetCreditBalanceTimeSeriesResponse < Internal::Types::Model
+        field :data, -> { Schematic::Types::CreditBalanceTimeSeriesResponseData }, optional: false, nullable: false
+
+        field :params, -> { Schematic::Credits::Types::GetCreditBalanceTimeSeriesParams }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/get_credit_spend_policy_usage_params.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      # Input parameters
+      class GetCreditSpendPolicyUsageParams < Internal::Types::Model
+        field :billing_credit_id, -> { String }, optional: true, nullable: false
+
+        field :company_id, -> { String }, optional: true, nullable: false
+
+        field :user_ids, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/get_credit_spend_policy_usage_request.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class GetCreditSpendPolicyUsageRequest < Internal::Types::Model
+        field :billing_credit_id, -> { String }, optional: true, nullable: false
+
+        field :company_id, -> { String }, optional: false, nullable: false
+
+        field :user_ids, -> { String }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/get_credit_spend_policy_usage_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class GetCreditSpendPolicyUsageResponse < Internal::Types::Model
+        field :data, -> { Internal::Types::Array[Schematic::Types::CreditSpendPolicyResponseData] }, optional: false, nullable: false
+
+        field :params, -> { Schematic::Credits::Types::GetCreditSpendPolicyUsageParams }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/reserve_credits_request_body.rb

@@ -4,7 +4,7 @@ module Schematic
   module Credits
     module Types
       class ReserveCreditsRequestBody < Internal::Types::Model
-        field :amount, -> { Integer }, optional: false, nullable: false
+        field :amount, -> { Float }, optional: false, nullable: false
 
         field :company_id, -> { String }, optional: false, nullable: false
 

data/lib/schematic/credits/types/update_credit_spend_policy_request_body.rb

@@ -8,7 +8,11 @@ module Schematic
 
         field :label, -> { String }, optional: true, nullable: false
 
-        field :max_per_draw, -> { Integer }, optional: true, nullable: false
+        field :max_per_draw, -> { Float }, optional: true, nullable: false
+
+        field :window_amount, -> { Float }, optional: true, nullable: false
+
+        field :window_unit, -> { Schematic::Types::CreditSpendWindowUnit }, optional: true, nullable: false
       end
     end
   end

data/lib/schematic/dataexports/client.rb

@@ -56,7 +56,15 @@ module Schematic
           (response.body.to_s.empty? ? nil : Schematic::Dataexports::Types::ListDataExportsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
-          raise error_class.new(response.body, code: code)
+          error_types = {
+            400 => Schematic::Types::ApiError,
+            401 => Schematic::Types::ApiError,
+            403 => Schematic::Types::ApiError,
+            404 => Schematic::Types::ApiError,
+            500 => Schematic::Types::ApiError
+          }
+          error_body = Schematic::Errors::ResponseError.load_error_body(code, response.body, error_types)
+          raise error_class.new(response.body, code: code, body: error_body)
         end
       end
 
@@ -94,7 +102,15 @@ module Schematic
           (response.body.to_s.empty? ? nil : Schematic::Dataexports::Types::CreateDataExportResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
-          raise error_class.new(response.body, code: code)
+          error_types = {
+            400 => Schematic::Types::ApiError,
+            401 => Schematic::Types::ApiError,
+            403 => Schematic::Types::ApiError,
+            404 => Schematic::Types::ApiError,
+            500 => Schematic::Types::ApiError
+          }
+          error_body = Schematic::Errors::ResponseError.load_error_body(code, response.body, error_types)
+          raise error_class.new(response.body, code: code, body: error_body)
         end
       end
 
@@ -129,7 +145,14 @@ module Schematic
           (response.body.to_s.empty? ? nil : Schematic::Dataexports::Types::GetDataExportResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
-          raise error_class.new(response.body, code: code)
+          error_types = {
+            401 => Schematic::Types::ApiError,
+            403 => Schematic::Types::ApiError,
+            404 => Schematic::Types::ApiError,
+            500 => Schematic::Types::ApiError
+          }
+          error_body = Schematic::Errors::ResponseError.load_error_body(code, response.body, error_types)
+          raise error_class.new(response.body, code: code, body: error_body)
         end
       end
 
@@ -145,7 +168,7 @@ module Schematic
       # @example
       #   client.dataexports.get_data_export_artifact(data_export_id: "data_export_id")
       #
-      # @return [untyped]
+      # @return [String]
       def get_data_export_artifact(request_options: {}, **params)
         params = Schematic::Internal::Types::Utils.normalize_keys(params)
         request = Schematic::Internal::JSON::Request.new(
@@ -160,10 +183,17 @@ module Schematic
           raise Schematic::Errors::TimeoutError
         end
         code = response.code.to_i
-        return if code.between?(200, 299)
+        return response.body if code.between?(200, 299)
 
         error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
-        raise error_class.new(response.body, code: code)
+        error_types = {
+          401 => Schematic::Types::ApiError,
+          403 => Schematic::Types::ApiError,
+          404 => Schematic::Types::ApiError,
+          500 => Schematic::Types::ApiError
+        }
+        error_body = Schematic::Errors::ResponseError.load_error_body(code, response.body, error_types)
+        raise error_class.new(response.body, code: code, body: error_body)
       end
     end
   end

data/lib/schematic/datastream/client.rb

@@ -146,7 +146,11 @@ module Schematic
       # Raises DataStream::EvaluationError when the flag cannot be evaluated
       # locally (e.g. flag not in cache, rules engine unavailable), so the caller
       # can fall back to the API path.
-      def check_flag(eval_ctx, flag_key)
+      # options carries the caller's preflight (a simulated usage, an
+      # event-scoped usage, or a per-credit cost) through to the engine, so a
+      # client-side evaluation gates on the post-call balance. Nil evaluates the
+      # flag as it stands.
+      def check_flag(eval_ctx, flag_key, options = nil)
         flag = get_flag(flag_key)
         raise EvaluationError, "flag '#{flag_key}' not found in cache" unless flag
 
@@ -172,12 +176,22 @@ module Schematic
         end
 
         @logger.debug("Evaluating flag with rules engine: flag=#{flag_key}, company=#{company&.dig(:id)}, user=#{user&.dig(:id)}")
-        result = @rules_engine.check_flag(flag, company, user)
+        result = @rules_engine.check_flag_with_options(flag, company, user, options)
         @logger.debug("Rules engine evaluation result: value=#{result[:value]}, reason=#{result[:reason]}")
         result[:flag_key] = flag_key
         result
       end
 
+      # Evaluate a flag against an explicit company and user with preflight
+      # options, skipping the cache lookups check_flag does. The credit-lease
+      # path has already resolved both entities and needs to gate against a
+      # substituted credit balance, which only it can build.
+      def check_flag_with_options(flag, company, user, options)
+        raise EvaluationError, "rules engine not available" unless @rules_engine&.initialized?
+
+        @rules_engine.check_flag_with_options(flag, company, user, options)
+      end
+
       def update_company_metrics(company_keys, event_name, quantity)
         company = @company_cache.get_by_keys(company_keys)
         return unless company
@@ -231,6 +245,10 @@ module Schematic
         return cached if cached
 
         return nil if @replicator_mode
+        # A request sent over a closed socket is dropped without an error, so
+        # waiting on it just burns the resource timeout. Answer now and let the
+        # caller fall back.
+        return nil unless connected?
 
         request_entity(ENTITY_TYPE_COMPANY, keys, @pending_companies) do |_data|
           @company_cache.get_by_keys(keys)
@@ -242,6 +260,7 @@ module Schematic
         return cached if cached
 
         return nil if @replicator_mode
+        return nil unless connected?
 
         request_entity(ENTITY_TYPE_USER, keys, @pending_users) do |_data|
           @user_cache.get_by_keys(keys)
@@ -513,6 +532,17 @@ module Schematic
         @logger.info("Replicator is no longer ready") if was_ready
         @logger.warn("Replicator health check error: #{e.message}")
       end
+
+      # The cached company for these keys, without the fetch get_company falls
+      # back to. prewarm polls this to resolve a company id from secondary keys.
+      def get_cached_company(keys)
+        @company_cache.get_by_keys(keys)
+      end
+
+      # Cache-first flag and entity lookups, declared public after their
+      # definitions. The credit-lease check path resolves the flag, the company,
+      # and the user itself before evaluating, so it needs all three.
+      public :get_flag, :get_company, :get_user, :get_cached_company
     end
   end
 end