From ccd9454da870a262813588ea1b5b8e5b2274cf7c Mon Sep 17 00:00:00 2001
From: haroldus- <haroldus-@users.noreply.github.com>
Date: Sun, 23 May 2021 11:30:09 +1000
Subject: [PATCH] FIX: Allow more than 10 Stripe records (#62)

---
 lib/tasks/subscriptions.rake | 27 ++++++++++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/lib/tasks/subscriptions.rake b/lib/tasks/subscriptions.rake
index 2988ff0..8f47ed1 100644
--- a/lib/tasks/subscriptions.rake
+++ b/lib/tasks/subscriptions.rake
@@ -21,7 +21,28 @@ end
 
 def get_stripe_products
   puts 'Getting products from Stripe API'
-  Stripe::Product.list
+  starting_after = nil
+  products = []
+  loop do
+    results = Stripe::Product.list(limit: 100, starting_after: starting_after)
+    break if results.data.length == 0
+    products = products + results.data
+    starting_after = results.data.last.id
+  end
+  products
+end
+
+def get_stripe_subscriptions
+  puts 'Getting subscriptions from Stripe API'
+  starting_after = nil
+  subscriptions = []
+  loop do
+    results = Stripe::Subscription.list(limit: 100, starting_after: starting_after)
+    break if results.data.length == 0
+    subscriptions = subscriptions + results.data
+    starting_after = results.data.last.id
+  end
+  subscriptions
 end
 
 def import_products(products)
@@ -36,8 +57,8 @@ end
 def import_subscriptions
   puts 'Importing subscriptions'
   product_ids = DiscourseSubscriptions::Product.all.pluck(:external_id)
-  subscriptions = Stripe::Subscription.list
-  subscriptions_for_products = subscriptions[:data].select { |sub| product_ids.include?(sub[:items][:data][0][:plan][:product]) }
+  subscriptions = get_stripe_subscriptions
+  subscriptions_for_products = subscriptions.select { |sub| product_ids.include?(sub[:items][:data][0][:plan][:product]) }
 
   subscriptions_for_products.each do |subscription|
     product_id = subscription[:items][:data][0][:plan][:product]