schematichq 1.5.2 → 1.5.3

data/lib/schematic/planmigrations/client.rb

@@ -159,6 +159,7 @@ module Schematic
       # @option request_options [Hash{String => Object}] :additional_body_parameters
       # @option request_options [Integer] :timeout_in_seconds
       # @option params [String, nil] :feature_id
+      # @option params [String, nil] :feature_plan_rollout_id
       # @option params [String, nil] :plan_version_id
       # @option params [Schematic::Types::PlanVersionMigrationStatus, nil] :status
       # @option params [Integer, nil] :limit
@@ -167,6 +168,7 @@ module Schematic
       # @example
       #   client.planmigrations.list_migrations(
       #     feature_id: "feature_id",
+      #     feature_plan_rollout_id: "feature_plan_rollout_id",
       #     plan_version_id: "plan_version_id",
       #     status: "cancelled",
       #     limit: 1000000,
@@ -178,6 +180,7 @@ module Schematic
         params = Schematic::Internal::Types::Utils.normalize_keys(params)
         query_params = {}
         query_params["feature_id"] = params[:feature_id] if params.key?(:feature_id)
+        query_params["feature_plan_rollout_id"] = params[:feature_plan_rollout_id] if params.key?(:feature_plan_rollout_id)
         query_params["plan_version_id"] = params[:plan_version_id] if params.key?(:plan_version_id)
         query_params["status"] = params[:status] if params.key?(:status)
         query_params["limit"] = params[:limit] if params.key?(:limit)
@@ -405,6 +408,7 @@ module Schematic
       # @option request_options [Hash{String => Object}] :additional_body_parameters
       # @option request_options [Integer] :timeout_in_seconds
       # @option params [String, nil] :feature_id
+      # @option params [String, nil] :feature_plan_rollout_id
       # @option params [String, nil] :plan_version_id
       # @option params [Schematic::Types::PlanVersionMigrationStatus, nil] :status
       # @option params [Integer, nil] :limit
@@ -413,6 +417,7 @@ module Schematic
       # @example
       #   client.planmigrations.count_migrations(
       #     feature_id: "feature_id",
+      #     feature_plan_rollout_id: "feature_plan_rollout_id",
       #     plan_version_id: "plan_version_id",
       #     status: "cancelled",
       #     limit: 1000000,
@@ -424,6 +429,7 @@ module Schematic
         params = Schematic::Internal::Types::Utils.normalize_keys(params)
         query_params = {}
         query_params["feature_id"] = params[:feature_id] if params.key?(:feature_id)
+        query_params["feature_plan_rollout_id"] = params[:feature_plan_rollout_id] if params.key?(:feature_plan_rollout_id)
         query_params["plan_version_id"] = params[:plan_version_id] if params.key?(:plan_version_id)
         query_params["status"] = params[:status] if params.key?(:status)
         query_params["limit"] = params[:limit] if params.key?(:limit)

data/lib/schematic/planmigrations/types/count_migrations_params.rb

@@ -7,6 +7,8 @@ module Schematic
       class CountMigrationsParams < Internal::Types::Model
         field :feature_id, -> { String }, optional: true, nullable: false
 
+        field :feature_plan_rollout_id, -> { String }, optional: true, nullable: false
+
         field :limit, -> { Integer }, optional: true, nullable: false
 
         field :offset, -> { Integer }, optional: true, nullable: false

data/lib/schematic/planmigrations/types/count_migrations_request.rb

@@ -6,6 +6,8 @@ module Schematic
       class CountMigrationsRequest < Internal::Types::Model
         field :feature_id, -> { String }, optional: true, nullable: false
 
+        field :feature_plan_rollout_id, -> { String }, optional: true, nullable: false
+
         field :plan_version_id, -> { String }, optional: true, nullable: false
 
         field :status, -> { Schematic::Types::PlanVersionMigrationStatus }, optional: true, nullable: false

data/lib/schematic/planmigrations/types/create_migration_input.rb

@@ -16,6 +16,8 @@ module Schematic
 
         field :proration_behavior, -> { Schematic::Types::MigrationProrationBehavior }, optional: true, nullable: false
 
+        field :scheduled_at, -> { String }, optional: true, nullable: false
+
         field :strategy, -> { Schematic::Types::PlanVersionMigrationStrategy }, optional: false, nullable: false
 
         field :target_plan_type, -> { Schematic::Types::PlanType }, optional: false, nullable: false

data/lib/schematic/planmigrations/types/list_migrations_params.rb

@@ -7,6 +7,8 @@ module Schematic
       class ListMigrationsParams < Internal::Types::Model
         field :feature_id, -> { String }, optional: true, nullable: false
 
+        field :feature_plan_rollout_id, -> { String }, optional: true, nullable: false
+
         field :limit, -> { Integer }, optional: true, nullable: false
 
         field :offset, -> { Integer }, optional: true, nullable: false

data/lib/schematic/planmigrations/types/list_migrations_request.rb

@@ -6,6 +6,8 @@ module Schematic
       class ListMigrationsRequest < Internal::Types::Model
         field :feature_id, -> { String }, optional: true, nullable: false
 
+        field :feature_plan_rollout_id, -> { String }, optional: true, nullable: false
+
         field :plan_version_id, -> { String }, optional: true, nullable: false
 
         field :status, -> { Schematic::Types::PlanVersionMigrationStatus }, optional: true, nullable: false

data/lib/schematic/plans/types/publish_plan_version_request_body.rb

@@ -12,6 +12,8 @@ module Schematic
 
         field :billing_cycle_anchor, -> { String }, optional: true, nullable: false
 
+        field :billing_start_date, -> { String }, optional: true, nullable: false
+
         field :coupon_external_id, -> { String }, optional: true, nullable: false
 
         field :custom_field_values, -> { Internal::Types::Array[Schematic::Types::CheckoutFieldValue] }, optional: true, nullable: false
@@ -32,6 +34,8 @@ module Schematic
 
         field :require_no_migration, -> { Internal::Types::Boolean }, optional: true, nullable: false
 
+        field :scheduled_at, -> { String }, optional: true, nullable: false
+
         field :send_invoice, -> { Internal::Types::Boolean }, optional: true, nullable: false
 
         field :tax_id, -> { Schematic::Types::TaxIdInput }, optional: true, nullable: false

data/lib/schematic/plans/types/retry_custom_plan_billing_request_body.rb

@@ -10,6 +10,8 @@ module Schematic
 
         field :billing_cycle_anchor, -> { String }, optional: true, nullable: false
 
+        field :billing_start_date, -> { String }, optional: true, nullable: false
+
         field :customer_email, -> { String }, optional: false, nullable: false
 
         field :days_until_due, -> { Integer }, optional: true, nullable: false

data/lib/schematic/rules_engine.rb

@@ -73,12 +73,26 @@ module Schematic
     end
 
     def check_flag(flag, company = nil, user = nil)
+      check_flag_with_options(flag, company, user, nil)
+    end
+
+    # Evaluate a flag with preflight options: a simulated usage, an event-scoped
+    # usage, or a pre-computed per-credit cost. The engine applies them to the
+    # conditions they match without mutating any state, so a caller can ask
+    # "would this call still be allowed after it lands".
+    #
+    # options is a hash with any of :credit_cost (credit id to cost),
+    # :usage, and :event_usage ({ event_subtype:, quantity: }). The engine reads
+    # them snake_case in both directions, so they go on the envelope as given.
+    def check_flag_with_options(flag, company = nil, user = nil, options = nil)
       raise "WASM rules engine not initialized" unless @initialized
 
       # Build combined JSON envelope (same format as Python/C#)
       envelope = { flag: strip_nulls(flag) }
       envelope[:company] = strip_nulls(company) if company
       envelope[:user] = strip_nulls(user) if user
+      # Omitted entirely when empty, so the engine uses its own defaults.
+      envelope[:options] = strip_nulls(engine_options(options)) if options && !options.empty?
 
       json_bytes = JSON.generate(envelope).encode("UTF-8")
 
@@ -123,6 +137,29 @@ module Schematic
 
     private
 
+    # The options as the engine takes them. usage and event_usage.quantity
+    # deserialize as i64 there, so a value with a decimal point fails the whole
+    # check. Rounded through the helper every other wire field uses, so a caller
+    # passing a preflight straight to check_flag_with_entitlement is asked the
+    # same question a lease check asks, and an infinite usage is handed over
+    # rather than raising out of ceil.
+    def engine_options(options)
+      out = options.dup
+      usage_key = out.key?(:usage) ? :usage : "usage"
+      out[usage_key] = Credits::Leases.wire_quantity(out[usage_key]) if out[usage_key].is_a?(Numeric)
+      event_key = out.key?(:event_usage) ? :event_usage : "event_usage"
+      event_usage = out[event_key]
+      return out unless event_usage.is_a?(Hash)
+
+      quantity_key = event_usage.key?(:quantity) ? :quantity : "quantity"
+      return out unless event_usage[quantity_key].is_a?(Numeric)
+
+      out[event_key] = event_usage.merge(
+        quantity_key => Credits::Leases.wire_quantity(event_usage[quantity_key])
+      )
+      out
+    end
+
     def export_func(name)
       exp = @instance.export(name)
       raise "WASM export '#{name}' not found" unless exp