add transform for StripePaymentMethod

Author: zcesur

scripts/database_migration.exs

@@ -24,6 +24,7 @@ defmodule DatabaseMigration do
   alias Algora.Bounties.Claim
   alias Algora.Payments.Account
   alias Algora.Payments.Customer
+  alias Algora.Payments.PaymentMethod
   alias Algora.Payments.Transaction
   alias Algora.Workspace.Installation
   alias Algora.Workspace.Ticket
@@ -44,7 +45,8 @@ defmodule DatabaseMigration do
     "BountyTransfer" => "transactions",
     "GithubInstallation" => "installations",
     "StripeAccount" => "accounts",
-    "StripeCustomer" => "customers"
+    "StripeCustomer" => "customers",
+    "StripePaymentMethod" => "payment_methods"
   }
 
   @schema_mappings %{
@@ -61,11 +63,13 @@ defmodule DatabaseMigration do
     "BountyTransfer" => Transaction,
     "GithubInstallation" => Installation,
     "StripeAccount" => Account,
-    "StripeCustomer" => Customer
+    "StripeCustomer" => Customer,
+    "StripePaymentMethod" => PaymentMethod
   }
 
   @backfilled_tables [
     "accounts",
+    "payment_methods",
     "customers",
     "installations",
     "repositories",
@@ -492,6 +496,26 @@ defmodule DatabaseMigration do
     }
   end
 
+  defp transform("StripePaymentMethod", row, db) do
+    customer = db |> Map.get("StripeCustomer", []) |> Enum.find(&(&1["id"] == row["customer_id"]))
+
+    if !customer do
+      raise "StripeCustomer not found: #{inspect(row)}"
+    end
+
+    %{
+      "id" => row["id"],
+      "provider" => "stripe",
+      "provider_id" => row["stripe_id"],
+      "provider_meta" => nil,
+      "provider_customer_id" => customer["stripe_id"],
+      "is_default" => row["is_default"],
+      "customer_id" => row["customer_id"],
+      "inserted_at" => row["created_at"],
+      "updated_at" => row["updated_at"]
+    }
+  end
+
   defp transform(_, _row, _db), do: nil
 
   def process_dump(input_file, output_file) do

scripts/v1-progress.yaml

@@ -471,12 +471,12 @@
     - name: 1
     - region: -1
 - "StripePaymentMethod":
-    - id: 0
-    - created_at: 0
-    - updated_at: 0
-    - stripe_id: 0
-    - org_id: 0
-    - is_default: 0
+    - id: 1
+    - created_at: 1
+    - updated_at: 1
+    - stripe_id: 1
+    - org_id: -1
+    - is_default: 1
 - "StripeSubscription":
     - id: -1
     - created_at: -1

scripts/v2-progress.yaml

@@ -168,15 +168,15 @@
     - CONSTRAINT: 0
     - CONSTRAINT: 0
 - payment_methods:
-    - id: 0
-    - provider: 0
-    - provider_id: 0
-    - provider_meta: 0
-    - provider_customer_id: 0
-    - is_default: 0
-    - customer_id: 0
-    - inserted_at: 0
-    - updated_at: 0
+    - id: 1
+    - provider: 1
+    - provider_id: 1
+    - provider_meta: -1
+    - provider_customer_id: 1
+    - is_default: 1
+    - customer_id: 1
+    - inserted_at: 1
+    - updated_at: 1
 - repositories:
     - id: 0
     - provider: 0