ruby-magicwrite 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a7196a229f9665af5e45c3c039aff8aed41a4ae495a1ec90ef882f7f2f025858
-  data.tar.gz: bcbc8eac9a03a650b7ac39da3aaab4715a78b0d895cd583565bc703ee8262c57
+  metadata.gz: 78dec62f5b00bafdf4c419f17e5f908ce8431ed0d0ed1d5289a9b12556160990
+  data.tar.gz: d707b12d56bca20072ab023ba01f8e7f12b86ad16b675e151c6b9fc9dcd2ea51
 SHA512:
-  metadata.gz: 92e2cc734dfd82bbf196b8dff81873dc516706b2fb4e9889f3a7c8d5360153c21226c5f1f4dbcfd6d1616b7a7a5518886d3894f74ff574bc7ad78dca17db9e13
-  data.tar.gz: 3895dfe940935bd163ebb0e60b57209f074ebbb1c6bcd898aad98f1f7a38df32e7acfdd61c0233c4741e55b62fb653728ac196e0af6e30b7adcf356507564217
+  metadata.gz: 218a31aab9ea79dd52f51dba0622d137ada3ee533532f49bdb06d1127aacd577873d9ea24ddca9fe7c0a50e8dca9a8c5731e4ce2c0996e4211fc270764a607b4
+  data.tar.gz: 6dab53f804184b3e3179c1a51fb5cdf00493b2ec7781e59ed1e2982565797d42b8953473f3e0789b9ef590dc5e827fd344ad86f91fd8d6a2064250b8561c41ad

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ruby-magicwrite (0.1.3)
+    ruby-magicwrite (0.1.4)
       faraday (>= 1)
       faraday-multipart (>= 1)
 

data/README.md

@@ -12,13 +12,17 @@ gem "ruby-magicwrite"
 
 And then execute:
 
+```bash
 $ bundle install
+```
 
 ### Gem install
 
 Or install with:
 
-$ gem install ruby-magicwrite
+```bash
+$ gem install ruby-openai
+```
 
 and require with:
 

data/lib/magicwrite/client.rb

@@ -24,6 +24,10 @@ module MagicWrite
       @ingestions ||= MagicWrite::Ingestions.new
     end
 
+    def memberships
+      @memberships ||= MagicWrite::Memberships.new
+    end
+
     def session
       MagicWrite::Client.get(path: '/session')
     end

data/lib/magicwrite/error.rb

@@ -0,0 +1,20 @@
+module MagicWrite
+  class Error < StandardError
+    class BadRequest < Error; end
+    class Unauthorized < Error; end
+    class PaymentRequired < Error; end
+    class Forbidden < Error; end
+    class NotFound < Error; end
+    class MethodNotAllowed < Error; end
+    class RequestTimeout < Error; end
+    class Conflict < Error; end
+    class UnprocessableEntity < Error; end
+    class TooManyRequests < Error; end
+    class InternalServerError < Error; end
+    class NotImplemented < Error; end
+    class BadGateway < Error; end
+    class ServiceUnavailable < Error; end
+    class GatewayTimeout < Error; end
+    class UnknownError < Error; end
+  end
+end

data/lib/magicwrite/error_handler.rb

@@ -0,0 +1,38 @@
+module MagicWrite
+  class ErrorHandler
+    HTTP_CODE = {
+      400 => MagicWrite::Error::BadRequest,
+      401 => MagicWrite::Error::Unauthorized,
+      402 => MagicWrite::Error::PaymentRequired,
+      403 => MagicWrite::Error::Forbidden,
+      404 => MagicWrite::Error::NotFound,
+      405 => MagicWrite::Error::MethodNotAllowed,
+      408 => MagicWrite::Error::RequestTimeout,
+      409 => MagicWrite::Error::Conflict,
+      422 => MagicWrite::Error::UnprocessableEntity,
+      429 => MagicWrite::Error::TooManyRequests,
+      500 => MagicWrite::Error::InternalServerError,
+      501 => MagicWrite::Error::NotImplemented,
+      502 => MagicWrite::Error::BadGateway,
+      503 => MagicWrite::Error::ServiceUnavailable,
+      504 => MagicWrite::Error::GatewayTimeout
+    }.freeze
+
+    def initialize(response)
+      @response = response
+      @status = response.status
+    end
+
+    def raise_error
+      raise error_class, response.reason_phrase
+    end
+
+    private
+
+    attr_reader :response, :status
+
+    def error_class
+      HTTP_CODE[status] || MagicWrite::Error::UnknownError
+    end
+  end
+end

data/lib/magicwrite/http.rb

@@ -1,55 +1,67 @@
 module MagicWrite
   module HTTP
     def get(path:, parameters: {})
-      to_json(conn.get(uri(path: path)) do |req|
-        req.headers = headers
-        req.body = parameters.to_json
-      end&.body)
+      build_response(conn.get(uri(path: path)) do |request|
+        request.headers = headers
+        request.body = parameters.to_json
+      end)
     end
 
     def json_post(path:, parameters:)
-      to_json(conn.post(uri(path: path)) do |req|
+      build_response(conn.post(uri(path: path)) do |request|
         if parameters[:stream].respond_to?(:call)
-          req.options.on_data = to_json_stream(user_proc: parameters[:stream])
+          request.options.on_data = to_json_stream(user_proc: parameters[:stream])
           parameters[:stream] = true
         elsif parameters[:stream]
           raise ArgumentError, 'The stream parameter must be a Proc or have a #call method'
         end
 
-        req.headers = headers
-        req.body = parameters.to_json
-      end&.body)
+        request.headers = headers
+        request.body = parameters.to_json
+      end)
     end
 
     def json_put(path:, parameters:)
-      to_json(conn.put(uri(path: path)) do |req|
+      build_response(conn.put(uri(path: path)) do |request|
         if parameters[:stream].respond_to?(:call)
-          req.options.on_data = to_json_stream(user_proc: parameters[:stream])
+          request.options.on_data = to_json_stream(user_proc: parameters[:stream])
           parameters[:stream] = true
         elsif parameters[:stream]
           raise ArgumentError, 'The stream parameter must be a Proc or have a #call method'
         end
 
-        req.headers = headers
-        req.body = parameters.to_json
-      end&.body)
+        request.headers = headers
+        request.body = parameters.to_json
+      end)
     end
 
     def multipart_post(path:, parameters: nil)
-      to_json(conn(multipart: true).post(uri(path: path)) do |req|
-        req.headers = headers.merge({ 'Content-Type' => 'multipart/form-data' })
-        req.body = multipart_parameters(parameters)
-      end&.body)
+      build_response(conn(multipart: true).post(uri(path: path)) do |request|
+        request.headers = headers.merge({ 'Content-Type' => 'multipart/form-data' })
+        request.body = multipart_parameters(parameters)
+      end)
     end
 
     def delete(path:)
-      to_json(conn.delete(uri(path: path)) do |req|
-        req.headers = headers
-      end&.body)
+      build_response(conn.delete(uri(path: path)) do |request|
+        request.headers = headers
+      end)
     end
 
     private
 
+    def build_response(response)
+      return parsed_response(response) if response.status.between?(200, 299)
+
+      MagicWrite::ErrorHandler.new(response).raise_error
+    end
+
+    def parsed_response(response)
+      return unless response
+
+      to_json(response.body)
+    end
+
     def to_json(string)
       return unless string
 
@@ -69,9 +81,9 @@ module MagicWrite
     end
 
     def conn(multipart: false)
-      Faraday.new do |f|
-        f.options[:timeout] = MagicWrite.configuration.request_timeout
-        f.request(:multipart) if multipart
+      Faraday.new do |faraday|
+        faraday.options[:timeout] = MagicWrite.configuration.request_timeout
+        faraday.request(:multipart) if multipart
       end
     end
 

data/lib/magicwrite/memberships.rb

@@ -0,0 +1,15 @@
+module MagicWrite
+  class Memberships
+    def initialize(access_token: nil)
+      MagicWrite.configuration.access_token = access_token if access_token
+    end
+
+    def user
+      MagicWrite::Client.get(path: '/memberships/user')
+    end
+
+    def company
+      MagicWrite::Client.get(path: '/memberships/company')
+    end
+  end
+end

data/lib/magicwrite/version.rb

@@ -1,3 +1,3 @@
 module MagicWrite
-  VERSION = '0.1.3'.freeze
+  VERSION = '0.1.4'.freeze
 end

data/lib/magicwrite.rb

@@ -6,11 +6,13 @@ require_relative 'magicwrite/client'
 require_relative 'magicwrite/agents'
 require_relative 'magicwrite/companies'
 require_relative 'magicwrite/completions'
+require_relative 'magicwrite/error'
+require_relative 'magicwrite/error_handler'
 require_relative 'magicwrite/ingestions'
+require_relative 'magicwrite/memberships'
 require_relative 'magicwrite/version'
 
 module MagicWrite
-  class Error < StandardError; end
   class ConfigurationError < Error; end
 
   class Configuration

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby-magicwrite
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Rodrigo Campos
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-08-21 00:00:00.000000000 Z
+date: 2023-08-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -68,8 +68,11 @@ files:
 - lib/magicwrite/companies.rb
 - lib/magicwrite/compatibility.rb
 - lib/magicwrite/completions.rb
+- lib/magicwrite/error.rb
+- lib/magicwrite/error_handler.rb
 - lib/magicwrite/http.rb
 - lib/magicwrite/ingestions.rb
+- lib/magicwrite/memberships.rb
 - lib/magicwrite/version.rb
 - lib/ruby/magicwrite.rb
 - pull_request_template.md