subscription migrations

Author: joelhawksley

engine/db/migrate/019_create_inner_performance_events.rb

@@ -1,7 +1,14 @@
 # frozen_string_literal: true
 
+# Creates the legacy inner_performance_events table. Guarded with
+# `unless table_exists?` so it runs cleanly whether or not the table is already
+# present (the engine and its migrations are shared by multiple apps, and these
+# transient telemetry tables are later dropped by migrations 023/024 and so are
+# absent from the final schema).
 class CreateInnerPerformanceEvents < ActiveRecord::Migration[8.1]
-  def change
+  def up
+    return if table_exists?(:inner_performance_events)
+
     create_table :inner_performance_events, force: :cascade do |t|
       t.string :event
       t.string :name
@@ -13,4 +20,8 @@ def change
       t.string :type
     end
   end
+
+  def down
+    drop_table :inner_performance_events, if_exists: true
+  end
 end

engine/db/migrate/020_create_inner_performance_traces.rb

@@ -1,7 +1,18 @@
 # frozen_string_literal: true
 
+# Creates the legacy inner_performance_traces table (foreign key to
+# inner_performance_events). Guarded so it runs cleanly on databases where the
+# traces table already exists or the events table is absent — these transient
+# telemetry tables are dropped by migrations 023/024 and so never appear in the
+# final schema. Mirrors the defensive style of those drop migrations.
 class CreateInnerPerformanceTraces < ActiveRecord::Migration[8.1]
-  def change
+  def up
+    return if table_exists?(:inner_performance_traces)
+    # Can't add the foreign key if the referenced table was never created (e.g.
+    # a database that skipped or never ran migration 019). The table is dropped
+    # downstream anyway, so safely no-op.
+    return unless table_exists?(:inner_performance_events)
+
     create_table :inner_performance_traces do |t|
       t.references :event, type: :bigint, null: false, foreign_key: {to_table: :inner_performance_events}
       t.string :name
@@ -12,4 +23,8 @@ def change
       t.timestamps
     end
   end
+
+  def down
+    drop_table :inner_performance_traces, if_exists: true
+  end
 end

engine/db/migrate/026_add_billing_to_accounts.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+class AddBillingToAccounts < ActiveRecord::Migration[8.1]
+  def up
+    add_column :accounts, :trial_ends_on, :date
+    add_column :accounts, :stripe_customer_id, :string
+    add_column :accounts, :stripe_subscription_id, :string
+    add_column :accounts, :subscription_status, :string
+    add_column :accounts, :subscription_current_period_end, :datetime
+    add_column :accounts, :stripe_base_item_id, :string
+    add_column :accounts, :stripe_additional_item_id, :string
+
+    add_index :accounts, :stripe_customer_id, unique: true
+    add_index :accounts, :stripe_subscription_id, unique: true
+
+    # Backfill trial_ends_on from each account's owner user's trial_started_on + 60 days.
+    execute(<<~SQL.squish)
+      UPDATE accounts
+      SET trial_ends_on = users.trial_started_on + INTERVAL '60 days'
+      FROM account_users
+      JOIN users ON users.id = account_users.user_id
+      WHERE account_users.account_id = accounts.id
+        AND account_users.role = 'owner'
+        AND users.trial_started_on IS NOT NULL
+    SQL
+
+    # Any account still without an end date gets a 60-day trial from today.
+    execute(<<~SQL.squish)
+      UPDATE accounts
+      SET trial_ends_on = CURRENT_DATE + INTERVAL '60 days'
+      WHERE trial_ends_on IS NULL
+    SQL
+
+    remove_column :users, :trial_started_on
+  end
+
+  def down
+    add_column :users, :trial_started_on, :date
+
+    remove_index :accounts, :stripe_subscription_id
+    remove_index :accounts, :stripe_customer_id
+
+    remove_column :accounts, :stripe_additional_item_id
+    remove_column :accounts, :stripe_base_item_id
+    remove_column :accounts, :subscription_current_period_end
+    remove_column :accounts, :subscription_status
+    remove_column :accounts, :stripe_subscription_id
+    remove_column :accounts, :stripe_customer_id
+    remove_column :accounts, :trial_ends_on
+  end
+end

engine/db/migrate/027_add_subscription_grace_to_accounts.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+class AddSubscriptionGraceToAccounts < ActiveRecord::Migration[8.1]
+  def change
+    add_column :accounts, :subscription_grace_until, :datetime
+  end
+end

engine/db/migrate/028_collapse_subscription_item_columns.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+# Switches billing from two separate prices (base + additional line items) to a
+# single tiered price with one subscription line item whose quantity is the
+# account's device count. Collapses the two item-id columns into one.
+class CollapseSubscriptionItemColumns < ActiveRecord::Migration[8.1]
+  def up
+    rename_column :accounts, :stripe_base_item_id, :stripe_subscription_item_id
+    remove_column :accounts, :stripe_additional_item_id
+  end
+
+  def down
+    add_column :accounts, :stripe_additional_item_id, :string
+    rename_column :accounts, :stripe_subscription_item_id, :stripe_base_item_id
+  end
+end

engine/db/schema.rb

@@ -10,7 +10,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema[8.1].define(version: 25) do
+ActiveRecord::Schema[8.1].define(version: 28) do
   # These are extensions that must be enabled in order to support this database
   enable_extension "pg_catalog.plpgsql"
   enable_extension "pgcrypto"
@@ -32,9 +32,18 @@
     t.text "name", null: false
     t.string "precipitation_unit", default: "in", null: false
     t.string "speed_unit", default: "mph", null: false
+    t.string "stripe_customer_id"
+    t.string "stripe_subscription_id"
+    t.string "stripe_subscription_item_id"
+    t.datetime "subscription_current_period_end"
+    t.datetime "subscription_grace_until"
+    t.string "subscription_status"
     t.datetime "support_access_at"
     t.string "temperature_unit", default: "F", null: false
+    t.date "trial_ends_on"
     t.datetime "updated_at", null: false
+    t.index ["stripe_customer_id"], name: "index_accounts_on_stripe_customer_id", unique: true
+    t.index ["stripe_subscription_id"], name: "index_accounts_on_stripe_subscription_id", unique: true
   end
 
   create_table "apple_accounts", force: :cascade do |t|
@@ -308,7 +317,6 @@
     t.datetime "login_code_sent_at"
     t.datetime "remember_created_at"
     t.text "remember_token"
-    t.date "trial_started_on"
     t.datetime "updated_at", null: false
     t.index ["email"], name: "index_users_on_email", unique: true
   end