spackle-ruby 0.0.11 → 0.0.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d370af3cb13da25635dff5eaee17972e6f540b3ab3ad949097abc0279ca64f88
-  data.tar.gz: 13fd6753a5dc9f8672cc5705cb13f8ebb8796b56ab685bed48921acc170f119d
+  metadata.gz: 18048e2400b535e381a4cf3130d671c5489df5453834133fb2a577ad0a175cb1
+  data.tar.gz: 7c98ce5a2d61102e20d079e000002b3f2faa4704efdb39638dedf6f61d491d48
 SHA512:
-  metadata.gz: c1ddcc3b1d6b16e37d4a8374d057ef6c312b2a9cac1934740bc8d237f038b1b5613a637d290ceadcc057a4f6176c07e152810a7577a0996c788d7c1c04e1b7d9
-  data.tar.gz: 1f4a29bddbb7c8d3ba73ebd5a760d424afa308b6f1c19819ef159d6a23dd32f4af329b8256561f067d0ee08df451762164dec8c5dc162aacbd30a6dbb1d7acbe
+  metadata.gz: fa08056244492c4697b5f36f43d2acd2c64c943a7597a36d2406115f6e0426ee9ebd3f823ddc75c4050f9bf671b56b4d8f17726fc0d9a1d60668ce20918ba2ba
+  data.tar.gz: 779445743a20139ac77fcc9e1b8022bd4ac2bf75350cb5feb7ada99c6d5d278459e507a9c27f51110b700e8d4b8844773ad9213fe770b20b6f1aa6a373d64216

data/lib/spackle/stores/dynamodb.rb

@@ -6,30 +6,48 @@ require 'logger'
 module Spackle
   class DynamoDBStore < BaseStore
     @client = nil
-    @store_config = {}
+    @credentials = nil
 
-    def initialize(client = nil, store_config = nil)
-      @store_config = store_config || {}
-      @client = client || bootstrap_client
+    def initialize()
+      @client = bootstrap_client
     end
 
     def get_customer_data(id)
       result = get_item(id)
 
       if result.nil? or result.item.nil?
-        raise SpackleError.new "Customer #{id} not found"
+        return fetch_state_from_api(id)
       end
 
       JSON.parse(result.item['State'])
     end
 
+    def fetch_state_from_api(id)
+      Util.log_warn("Customer #{id} not found in DynamoDB. Fetching from API...")
+      uri = URI(Spackle.api_base + '/customers/' + id + '/state')
+      https = Net::HTTP.new(uri.host, uri.port)
+      https.use_ssl = true
+
+      request = Net::HTTP::Get.new(uri.path)
+      request['Authorization'] = 'Bearer ' + Spackle.api_key
+
+      response = https.request(request)
+      if response.code != '200'
+        puts response.code
+        puts response.body
+        raise SpackleError.new "Customer #{id} not found"
+      end
+
+      JSON.parse(response.body)
+    end
+
     private
 
     def get_item(id)
       return @client.get_item({
-        table_name: @store_config['table_name'],
+        table_name: @credentials.session['adapter']['table_name'],
         key: {
-          AccountId: @store_config['identity_id'],
+          AccountId: @credentials.session['adapter']['identity_id'],
           CustomerId: "#{id}:#{Spackle.version}"
         }
       })
@@ -44,58 +62,56 @@ module Spackle
         raise SpackleError.new 'API key not set'
       end
 
-      uri = URI(Spackle.api_base + '/sessions')
-      https = Net::HTTP.new(uri.host, uri.port)
-      https.use_ssl = true
-
-      request = Net::HTTP::Post.new(uri.path)
-      request['Authorization'] = 'Bearer ' + Spackle.api_key
-
-      response = https.request(request)
-      data = JSON.parse(response.body)
-      Util.log_debug("Created session: #{data}")
-
-      @store_config = data['adapter']
-      credentials = SpackleCredentials.new(
-        @store_config['role_arn'],
-        @store_config['token'],
-        @store_config['region'],
-      )
-
+      @credentials = SpackleAWSCredentials.new
       Aws::DynamoDB::Client.new(
-        region: @store_config['region'],
-        credentials: credentials,
+        region: @credentials.session['adapter']['region'],
+        credentials: @credentials,
       )
     end
   end
 
-  class SpackleCredentials
+  class SpackleAWSCredentials
     include Aws::CredentialProvider
     include Aws::RefreshingCredentials
 
-    @region = nil
-    @role_arn = nil
-    @token = nil
+    @client = nil
+    @session = {}
 
-    def initialize(role_arn, token, region)
-      @region = region
-      @role_arn = role_arn
-      @token = token
+    def initialize()
+      @session = create_session
       @client = Aws::STS::Client.new(
-        region: @region,
+        region: @session['adapter']['region'],
         credentials: false,
       )
       super()
     end
 
+    def session
+      @session
+    end
+
     private
 
+    def create_session
+      uri = URI(Spackle.api_base + '/sessions')
+      https = Net::HTTP.new(uri.host, uri.port)
+      https.use_ssl = true
+
+      request = Net::HTTP::Post.new(uri.path)
+      request['Authorization'] = 'Bearer ' + Spackle.api_key
+
+      response = https.request(request)
+      data = JSON.parse(response.body)
+      Util.log_debug("Created session: #{data}")
+      data
+    end
+
     def refresh
       Util.log_debug('Refreshing DynamoDB credentials...')
       c = @client.assume_role_with_web_identity({
-        role_arn: @role_arn,
+        role_arn: @session['adapter']['role_arn'],
         role_session_name: Base64.strict_encode64(SecureRandom.uuid),
-        web_identity_token: @token
+        web_identity_token: @session['adapter']['token'],
       }).credentials
       @credentials = Aws::Credentials.new(
         c.access_key_id,

data/spackle.gemspec

@@ -2,7 +2,7 @@ $LOAD_PATH.unshift(::File.join(::File.dirname(__FILE__), "lib"))
 
 Gem::Specification.new do |s|
   s.name          = "spackle-ruby"
-  s.version       = "0.0.11"
+  s.version       = "0.0.12"
   s.summary       = "Spackle Ruby gem"
   s.description   = "Spackle is the easiest way to integrate your Ruby app with Stripe Billing. " \
                     "See https://www.spackle.so for details."

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: spackle-ruby
 version: !ruby/object:Gem::Version
-  version: 0.0.11
+  version: 0.0.12
 platform: ruby
 authors:
 - Spackle
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-04-02 00:00:00.000000000 Z
+date: 2023-04-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
@@ -112,7 +112,7 @@ metadata:
   github_repo: ssh://github.com/spackleso/spackle-ruby
   homepage_uri: https://docs.spackle.so/ruby
   source_code_uri: https://github.com/spackleso/spackle-ruby
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -127,8 +127,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3.1
-signing_key: 
+rubygems_version: 3.4.10
+signing_key:
 specification_version: 4
 summary: Spackle Ruby gem
 test_files: []