embulk-input-stripe 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 0aff836b6c0954949a87020f23110c9d3eeceb2f
+  data.tar.gz: c7526f57d26570220fed99a33e4ab9003600a7af
+SHA512:
+  metadata.gz: 1f4333f9ee53eb65c4ef0137711ae4e251da9f1d03846a729dfdd28e2a7a995517da8f5a76fd8dfbf11d830fc8a0708a358d6cd6c912bee0f13669cc82bc5250
+  data.tar.gz: 1326ef0c452bd0a2df0af8531612a3bb6ce353e412c5f5e5f2cd30573e95e18dcfe6011516aaf3fb97c40e0660a3818cd3026019fd77393be9c9f942a3daaceb

data/lib/embulk/input/stripe.rb

@@ -0,0 +1,75 @@
+# frozen_string_literal: true
+
+require 'stripe'
+require_relative 'stripe/customers'
+require_relative 'stripe/invoices'
+require_relative 'stripe/subscriptions'
+
+module Embulk
+  module Input
+    class Stripe < InputPlugin
+      Plugin.register_input('stripe', self)
+
+      def self.transaction(config, &control)
+        ::Stripe.api_key = config.param('api_key', :string)
+        resource_type = config.param('resource_type', :string)
+        fields = config.param('fields', :array)
+
+        task = {
+          'resource_type' => resource_type,
+          'fields' => fields
+        }
+        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)
+        end
+
+        resume(task, columns, 1, &control)
+      end
+
+      def self.resume(task, columns, count, &control)
+        puts "Input started."
+        task_reports = yield(task, columns, count)
+        puts "Input finished. Commit reports = #{task_reports.to_json}"
+
+        {} # next_config_diff
+      end
+
+      def init
+        resource_type = task['resource_type']
+        fields = task['fields']
+
+        case resource_type
+        when 'customers'
+          @customers = Customers.new(fields)
+        when 'invoices'
+          @invoices = Invoices.new(fields)
+        when 'subscriptions'
+          @subscriptions = Subscriptions.new(fields)
+        else
+          raise StandardError, "Resource type #{resource_type} is not supported."
+        end
+      end
+
+      def run
+        target_items = case task['resource_type']
+                       when 'customers'
+                         @customers
+                       when 'invoices'
+                         @invoices
+                       when 'subscriptions'
+                         @subscriptions
+                       end
+
+        target_items.get.each do |item|
+          page_builder.add(item)
+        end
+
+        page_builder.finish
+
+        {} # task_report
+      end
+    end
+  end
+end

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

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require 'stripe'
+require_relative './stripe_items'
+
+class Customers < StripeItems
+  def items
+    Stripe::Customer.list
+  end
+end

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

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require 'stripe'
+require_relative './stripe_items'
+
+class Invoices < StripeItems
+  def items
+    Stripe::Invoice.list
+  end
+end

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

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'embulk/column'
+
+class StripeItems
+  def initialize(fields)
+    @fields = fields
+  end
+
+  def get
+    result = []
+
+    items.auto_paging_each do |sub|
+      result.push(to_row(sub))
+    end
+    result
+  end
+
+  private
+
+  def items
+    raise 'Please override this method and return the items.'
+  end
+
+  def to_row(sub)
+    @fields.map do |field|
+      # 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]
+      end
+    end
+  end
+end

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

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require 'stripe'
+require_relative './stripe_items'
+
+class Subscriptions < StripeItems
+  def items
+    Stripe::Subscription.list(status: 'all')
+  end
+end

metadata

@@ -0,0 +1,77 @@
+--- !ruby/object:Gem::Specification
+name: embulk-input-stripe
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Daisuke Shimamoto
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2021-04-26 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.32.1
+  name: stripe
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.32.1
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 13.0.3
+  name: rake
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 13.0.3
+description: Loads data from Stripe.
+email:
+- diskshima@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/embulk/input/stripe.rb
+- lib/embulk/input/stripe/customers.rb
+- lib/embulk/input/stripe/invoices.rb
+- lib/embulk/input/stripe/stripe_items.rb
+- lib/embulk/input/stripe/subscriptions.rb
+homepage: https://github.com/diskshima/embulk-input-stripe
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project:
+rubygems_version: 2.6.14
+signing_key:
+specification_version: 4
+summary: Embulk input plugin for Stripe
+test_files: []