contenttruck-rb 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 86c1b2b13c1dbfe1aef7cc2d1cd7c4dc6c7fa57b55b19b94d99d098555a6e473
-  data.tar.gz: 834aea2e6ca0d766fcdbe0f447694af07f757920391ac576cf87b1eb5b03b7ff
+  metadata.gz: 10f91fed1dcb74097e86b9ce6385ed9b1ff60ae8664aca186721a656f9fcdb0e
+  data.tar.gz: 53a2eca6ffadd08571dab8929577e1a7259594405b97da5df5517bf9b6bb0dfb
 SHA512:
-  metadata.gz: 17f2624823545a636be54aa9789cd4770d73c10ee59c788e4265ce456c3cdadb0012b840ad07c3a092e62dbc614ab8a66726ae4cd80553bbd7230c0c174d7755
-  data.tar.gz: a7bb450068fd878991099b8a40ae4c093a2d0fd712b20b070c01eed0bba6e80cbb8e9fa04acebb30b751c074b243dbbbdf496864c57be64e9f736bb26982b639
+  metadata.gz: a24a64be8a258857e60925639dd0dd9419498df75a240d54efa0c3f636e280a58986122ae38269e42ec0c13e79c746e2081f278918a65e269408b12a4350f455
+  data.tar.gz: abd7d4cc77ef0fc3eac98fbc81c6a3bb092344635f0ae789e7e7f380fa9551006c8e319b8e127c9ca1cbbf31b25ef77b4d64f77fa5f62279b8387d1c97045a8c

data/Gemfile.lock

@@ -0,0 +1,16 @@
+PATH
+  remote: .
+  specs:
+    contenttruck-rb (1.0.1)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    rake (13.0.6)
+
+DEPENDENCIES
+  contenttruck-rb!
+  rake (~> 13.0)
+
+BUNDLED WITH
+   2.4.8

data/lib/contenttruck/client.rb

@@ -0,0 +1,155 @@
+# frozen_string_literal: true
+
+require 'uri'
+require 'net/http'
+require 'json'
+
+module Contenttruck
+  class UserError < StandardError
+    def initialize(code, message)
+      super(message)
+      @code = code
+    end
+
+    def code
+      @code
+    end
+  end
+
+  class Client
+    def initialize(base_url)
+      raise ArgumentError, 'base_url must be a String' unless base_url.is_a?(String)
+      @rpc_url = URI.parse(base_url)
+
+      # Set the path of the RPC URL to /_contenttruck.
+      @rpc_url.path = '/_contenttruck'
+    end
+
+    def upload(key, partition, relative_path, content_type, content_or_reader)
+      raise ArgumentError, 'key must be a String' unless key.is_a?(String)
+      raise ArgumentError, 'partition must be a String' unless partition.is_a?(String)
+      raise ArgumentError, 'relative_path must be a String' unless relative_path.is_a?(String)
+      raise ArgumentError, 'content_type must be a String' unless content_type.is_a?(String)
+      unless content_or_reader.is_a?(String) || content_or_reader.respond_to?(:read)
+        raise ArgumentError, 'content_or_reader must be a String or reader'
+      end
+
+      _do_rpc_request('Upload', {
+        'key' => key,
+        'partition' => partition,
+        'relative_path' => relative_path,
+      }, content_type, content_or_reader)
+    end
+
+    def delete(key, partition, relative_path)
+      raise ArgumentError, 'key must be a String' unless key.is_a?(String)
+      raise ArgumentError, 'partition must be a String' unless partition.is_a?(String)
+      raise ArgumentError, 'relative_path must be a String' unless relative_path.is_a?(String)
+
+      _do_rpc_request('Delete', {
+        'key' => key,
+        'partition' => partition,
+        'relative_path' => relative_path,
+      })
+    end
+
+    def create_key(sudo_key, partitions)
+      raise ArgumentError, 'sudo_key must be a String' unless sudo_key.is_a?(String)
+      raise ArgumentError, 'partitions must be an Array' unless partitions.is_a?(Array)
+      partitions.each do |partition|
+        raise ArgumentError, 'partitions must be an Array of Strings' unless partition.is_a?(String)
+      end
+
+      _do_rpc_request('CreateKey', {
+        'sudo_key' => sudo_key,
+        'partitions' => partitions,
+      })
+    end
+
+    def delete_key(sudo_key, key)
+      raise ArgumentError, 'sudo_key must be a String' unless sudo_key.is_a?(String)
+      raise ArgumentError, 'key must be a String' unless key.is_a?(String)
+
+      _do_rpc_request('DeleteKey', {
+        'sudo_key' => sudo_key,
+        'key' => key,
+      })
+    end
+
+    def create_partition(sudo_key, name, rule_set)
+      raise ArgumentError, 'sudo_key must be a String' unless sudo_key.is_a?(String)
+      raise ArgumentError, 'name must be a String' unless name.is_a?(String)
+      raise ArgumentError, 'rule_set must be a String' unless rule_set.is_a?(String)
+
+      _do_rpc_request('CreatePartition', {
+        'sudo_key' => sudo_key,
+        'name' => name,
+        'rule_set' => rule_set,
+      })
+    end
+
+    def delete_partition(sudo_key, name)
+      raise ArgumentError, 'sudo_key must be a String' unless sudo_key.is_a?(String)
+      raise ArgumentError, 'name must be a String' unless name.is_a?(String)
+
+      _do_rpc_request('DeletePartition', {
+        'sudo_key' => sudo_key,
+        'name' => name,
+      })
+    end
+
+    private
+
+    def _to_json(value)
+      # Basically works around the fact that some people might not have ActiveSupport loaded.
+      return value.to_json if value.respond_to?(:to_json)
+      JSON.generate(value)
+    end
+
+    def _get_user_error(response)
+      begin
+        content = JSON.parse(response.body)
+        return UserError.new(content['code'], content['message']) if content['code'] && content['message']
+      rescue; end
+    end
+
+    def _do_rpc_request(type, body, content_type = nil, content_reader = nil)
+      http = Net::HTTP.new(@rpc_url.host, @rpc_url.port)
+      http.use_ssl = true if @rpc_url.scheme == 'https'
+
+      request = Net::HTTP::Post.new(@rpc_url)
+      request['X-Type'] = type
+      if content_type
+        # Include the JSON in a header and send the body.
+        request['X-Json-Body'] = _to_json(body)
+        request['Content-Type'] = content_type
+        content_length = 0
+        if content_reader.instance_of?(String)
+          request.body = content_reader
+          content_length = content_reader.bytesize
+        else
+          request.body_stream = content_reader
+          content_length = content_reader.size
+        end
+        request['Content-Length'] = content_length.to_s
+      else
+        # Send the body as JSON.
+        request['Content-Type'] = 'application/json'
+        json = _to_json(body)
+        request['Content-Length'] = json.bytesize.to_s
+        request.body = json
+      end
+
+      response = http.request(request)
+
+      # Handle ok responses.
+      return nil if response.code == '204'
+      return JSON.parse(response.body) if response.code == '200'
+
+      # Handle errors.
+      x = _get_user_error(response)
+      raise x if x
+      raise StandardError, "Unexpected response code #{response.code} (body: #{response.body})"
+    end
+  end
+end

data/lib/contenttruck/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Contenttruck
-  VERSION = "1.0.0".freeze
+  VERSION = "1.0.1".freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: contenttruck-rb
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - Jake Gealer
@@ -18,11 +18,13 @@ extensions: []
 extra_rdoc_files: []
 files:
 - Gemfile
+- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile
 - contenttruck.gemspec
 - lib/contenttruck.rb
+- lib/contenttruck/client.rb
 - lib/contenttruck/version.rb
 - sig/contenttruck.rbs
 homepage: https://github.com/webscalesoftwareltd/contenttruck-rb