embulk-input-stripe 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0aff836b6c0954949a87020f23110c9d3eeceb2f
-  data.tar.gz: c7526f57d26570220fed99a33e4ab9003600a7af
+  metadata.gz: 9dec6b8863fd51c494e86648eee325183181cbfe
+  data.tar.gz: 04053c6698600553235ceaed3fba204da86e455d
 SHA512:
-  metadata.gz: 1f4333f9ee53eb65c4ef0137711ae4e251da9f1d03846a729dfdd28e2a7a995517da8f5a76fd8dfbf11d830fc8a0708a358d6cd6c912bee0f13669cc82bc5250
-  data.tar.gz: 1326ef0c452bd0a2df0af8531612a3bb6ce353e412c5f5e5f2cd30573e95e18dcfe6011516aaf3fb97c40e0660a3818cd3026019fd77393be9c9f942a3daaceb
+  metadata.gz: c5e78a9a6dacc71c2906dfd75cbc6705637fd357bf8dca8787796959ff04c68fe4209f9bb96e03bf12dad9075884e3a609a69b2425ad6ec37cd9b021fb05bff8
+  data.tar.gz: 1a06f0b0ae36bc069cc635924a59f4c45b377487e691d9a7d3e13b2c1ebb189c15551191f0ec843f662fc0497135fb67a92ab6ec3398fb7a69a1cecd5898b669

data/lib/embulk/input/stripe.rb

@@ -4,6 +4,7 @@ require 'stripe'
 require_relative 'stripe/customers'
 require_relative 'stripe/invoices'
 require_relative 'stripe/subscriptions'
+require_relative 'stripe/charges'
 
 module Embulk
   module Input
@@ -22,7 +23,7 @@ module Embulk
         columns = fields.map.with_index do |field, index|
           field_name = field['name']
           field_type = field['type'].to_sym
-          Column.new(index, field_name.gsub('.', '_'), field_type)
+          Column.new(index, field_name.gsub(/[\.\[\]]+/, '_'), field_type)
         end
 
         resume(task, columns, 1, &control)
@@ -47,6 +48,8 @@ module Embulk
           @invoices = Invoices.new(fields)
         when 'subscriptions'
           @subscriptions = Subscriptions.new(fields)
+        when 'charges'
+          @charges = Charges.new(fields)
         else
           raise StandardError, "Resource type #{resource_type} is not supported."
         end
@@ -60,6 +63,8 @@ module Embulk
                          @invoices
                        when 'subscriptions'
                          @subscriptions
+                       when 'charges'
+                         @charges
                        end
 
         target_items.get.each do |item|

data/lib/embulk/input/stripe/charges.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require 'stripe'
+require_relative './stripe_items'
+
+class Charges < StripeItems
+  def items
+    Stripe::Charge.list(limit: 100)
+  end
+end

data/lib/embulk/input/stripe/stripe_items.rb

@@ -3,6 +3,9 @@
 require 'embulk/column'
 
 class StripeItems
+  BEFORE_BRACKET_REGEX = Regexp.compile(/^([^\[]+)/)
+  INDEX_REGEX = Regexp.compile(/\[([0-9]+)\]/)
+
   def initialize(fields)
     @fields = fields
   end
@@ -27,7 +30,19 @@ class StripeItems
       # Drill down into sub items (assumes fields are delimited with '.').
       sub_fields = field['name'].split('.')
       sub_fields.reduce(sub) do |memo, sub_field|
-        memo[sub_field]
+        name_match = BEFORE_BRACKET_REGEX.match(sub_field)
+        before_bracket = name_match[1]
+
+        child = memo[before_bracket]
+
+        index_match = INDEX_REGEX.match(sub_field)
+
+        if index_match
+          index = index_match[1]
+          child[index.to_i]
+        else
+          child
+        end
       end
     end
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: embulk-input-stripe
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Daisuke Shimamoto
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-04-26 00:00:00.000000000 Z
+date: 2021-05-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   requirement: !ruby/object:Gem::Requirement
@@ -46,6 +46,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/embulk/input/stripe.rb
+- lib/embulk/input/stripe/charges.rb
 - lib/embulk/input/stripe/customers.rb
 - lib/embulk/input/stripe/invoices.rb
 - lib/embulk/input/stripe/stripe_items.rb