chalk_ruby 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 475cbc00a9b6c9a048e287fdca499ffd257a768c63471598c1db61d8a4757baf
-  data.tar.gz: bee580988929bc2219c4f6e6778e17eb8c2055ed80799be251d5af21295e32c8
+  metadata.gz: be35291268749664bfdee468de9a997bd421f3b16683ffb847e12f3089f22e46
+  data.tar.gz: 3670d6543e87f11cee499ed8477729364e57a2072437ed798f1a721d7823e93d
 SHA512:
-  metadata.gz: 64ea24c0bc3d3ea81f2d32c72a7220c70a5e1c7e88a0dcef1c5907dbccbc40d4f9a9d6d91e1917359ec516c7a864b9ba3a57c2848c98f5dba83a2cf61c8f1d7e
-  data.tar.gz: f8821a753ba007213b63977d6a0b465944320e20d96728db4edd7e57a4a2aeb9b5f79ef2dc18ce5cfa29e648151a354d90061f19f7dd43a77c85c7bf0e28dfa5
+  metadata.gz: 87b9e3a20075fda23a00ba5e38dc91827afcc18922b6b664e1fb31676623ce2d67627cd0d1a56831740fa03eeeaf99439841151073bfd1413b33e9811aa45281
+  data.tar.gz: 908b81be15e3c99571a7195f9464496b0f52bb5447f96728d82e2ed57365b5828dfef48c911a42d850ed9262a1bfb4c26b496cd7649e8a1512605c28a10a4c4f

data/README.md

@@ -2,6 +2,8 @@
 
 Ruby client for ChalkRuby
 
+https://rubygems.org/gems/chalk_ruby
+
 ## Getting Started
 
 First, install ChalkRuby Ruby API Client via the [RubyGems](https://rubygems.org/) package manager:
@@ -22,4 +24,4 @@ results = client.query(
 
 ## License
 
-The ChalkRuby Ruby API Client is an open-sourced software licensed under the [Apache 2 License](LICENSE.md).
+The Chalk Ruby API Client is an open-sourced software licensed under the [Apache 2 License](LICENSE.md).

data/lib/chalk_ruby/client.rb

@@ -26,6 +26,9 @@ module ChalkRuby
     # @param api_server [String]
     #   ChalkRuby API server
     #
+    # @param additional_headers [Hash[String, String]]
+    #   Additional headers to be sent with every request. Typically not required.
+    #
     # @return self
     #
     def self.create(
@@ -33,14 +36,16 @@ module ChalkRuby
       client_secret = nil,
       environment = nil,
       query_server = nil,
-      api_server = nil
+      api_server = nil,
+      additional_headers = {}
     )
       config = Config.new(
         client_id: client_id,
         client_secret: client_secret,
         environment: environment,
         query_server: query_server,
-        api_server: api_server
+        api_server: api_server,
+        additional_headers: additional_headers
       )
       create_with_config(config)
     end
@@ -153,7 +158,7 @@ module ChalkRuby
           include_meta: include_meta || false,
           store_plan_stages: store_plan_stages || false
         },
-        headers: get_authenticated_headers(branch: branch)
+        headers: get_authenticated_engine_headers(branch: branch)
       )
     end
 
@@ -221,15 +226,17 @@ module ChalkRuby
       )
     end
 
-    def get_authenticated_headers(branch:)
-      t = valid_token
-      {
+    def get_authenticated_engine_headers(branch:)
+      t      = valid_token
+      tokens = {
         'Content-Type': 'application/json',
         'Accept': 'application/json',
         'Authorization': 'Bearer ' + t.token,
         'X-Chalk-Env-Id': @config.environment || t.environment,
-        'X-Chalk-Branch-Id': branch
+        'X-Chalk-Branch-Id': branch,
+        'X-Chalk-Deployment-Type': branch.nil? ? 'engine' : 'branch'
       }
+      tokens.merge(@config.additional_headers)
     end
 
     def get_unauthenticated_headers
@@ -237,7 +244,7 @@ module ChalkRuby
         'Content-Type': 'application/json',
         'Accept': 'application/json',
         'X-Chalk-Env-Id': @config.environment
-      }
+      }.merge(@config.additional_headers)
     end
 
     def query_server_host

data/lib/chalk_ruby/config.rb

@@ -10,7 +10,8 @@ module ChalkRuby
                   :environment,
                   :query_timeout,
                   :api_timeout,
-                  :connect_timeout
+                  :connect_timeout,
+                  :additional_headers
 
     # Creates a new ChalkRuby::Config object for use with ChalkRuby::Client.
     #
@@ -39,15 +40,19 @@ module ChalkRuby
     # @option options [Integer?] :connect_timeout
     #   The timeout for connect operations (in seconds).
     #
+    # @option options [Hash<String, String>?] :additional_headers
+    #  Additional headers to be sent with every request. Typically not required.
+    #
     def initialize(opts = {})
-      @client_id       = opts[:client_id] || ENV['CHALK_CLIENT_ID']
-      @client_secret   = opts[:client_secret] || ENV['CHALK_CLIENT_SECRET']
-      @environment     = opts[:environment] || ENV['CHALK_ACTIVE_ENVIRONMENT']
-      @query_server    = opts[:query_server] || ENV['CHALK_QUERY_SERVER'] || Defaults::QUERY_SERVER
-      @api_server      = opts[:api_server] || ENV['CHALK_API_SERVER'] || Defaults::API_SERVER
-      @query_timeout   = opts[:query_timeout] || Defaults::API_TIMEOUT
-      @api_timeout     = opts[:api_timeout] || Defaults::QUERY_TIMEOUT
-      @connect_timeout = opts[:connect_timeout] || Defaults::CONNECT_TIMEOUT
+      @client_id          = opts[:client_id] || ENV['CHALK_CLIENT_ID']
+      @client_secret      = opts[:client_secret] || ENV['CHALK_CLIENT_SECRET']
+      @environment        = opts[:environment] || ENV['CHALK_ACTIVE_ENVIRONMENT']
+      @query_server       = opts[:query_server] || ENV['CHALK_QUERY_SERVER'] || Defaults::QUERY_SERVER
+      @api_server         = opts[:api_server] || ENV['CHALK_API_SERVER'] || Defaults::API_SERVER
+      @query_timeout      = opts[:query_timeout] || Defaults::API_TIMEOUT
+      @api_timeout        = opts[:api_timeout] || Defaults::QUERY_TIMEOUT
+      @connect_timeout    = opts[:connect_timeout] || Defaults::CONNECT_TIMEOUT
+      @additional_headers = opts[:additional_headers] || {}
 
       raise ChalkError, 'No Client ID provided, please set :client_id' if @client_id.nil?
       raise ChalkError, 'No Client Secret provided, please set :client_secret' if @client_secret.nil?

data/lib/chalk_ruby/version.rb

@@ -1,3 +1,3 @@
 module ChalkRuby
-  VERSION = '0.1.1'.freeze
+  VERSION = '0.1.2'.freeze
 end

data/sig/chalk_ruby/client.rbs

@@ -4,7 +4,14 @@ module ChalkRuby
     @token: Token?
     @transporter: Http::HttpRequesterChalk
 
-    def self.create: -> Client
+    def self.create: (
+        String? client_id,
+        String? client_secret,
+        String? environment,
+        String? query_server,
+        String? api_server,
+        Hash[String, String]? additional_headers
+      ) -> Client
 
     def self.create_with_config: -> Client
 
@@ -27,7 +34,7 @@ module ChalkRuby
 
     private
 
-    def get_authenticated_headers: -> Hash[Symbol, String | nil]
+    def get_authenticated_engine_headers: (String? branch) -> Hash[Symbol, String | nil]
 
     def get_unauthenticated_headers: -> Hash[Symbol, String | nil]
 

data/sig/chalk_ruby/config.rbs

@@ -1,5 +1,6 @@
 module ChalkRuby
   class Config
+    attr_accessor additional_headers: Hash[String, String]?
     attr_accessor client_id: String
     attr_accessor client_secret: String
 
@@ -22,7 +23,8 @@ module ChalkRuby
         query_timeout: Integer?,
         api_timeout: Integer?,
         connect_timeout: Integer?,
-        compression_type: String?
+        compression_type: String?,
+        additional_headers: Hash[String, String]?
       }
 
     def initialize: (options opts) -> void

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chalk_ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Chalk AI, Inc.
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-01-08 00:00:00.000000000 Z
+date: 2024-01-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler