tinybird 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 20843d4d73262c83e29f389e11e530c6c26128f93c1e41b12822147497714217
-  data.tar.gz: 6e9fa9bec789401fde7704554ace18a3a69304118e315e563e2bc56a1e40cba7
+  metadata.gz: '08ce180c2ced2aeb456764d97c50976aa56d5e534e9499fd3aba3b06937373f2'
+  data.tar.gz: 72bd467c7d10f9f2bf6d6aeacf52e27603a9c9a4b465274e4e5b0d982df29668
 SHA512:
-  metadata.gz: 77b30d000e0c7cba474b0e33e8f315eb7941749e6c23ed836ff8466a697db2b389e60e2213ce49232ea2d453f3df7a196ec6a203bd2bb052fed074d825364ef2
-  data.tar.gz: 1366183deb36160c210af308570df410813b81b602aaa7dc92e7ac28a56c896fe645f4b7a096631854b27dc1aeb1d06f0d65fa6b27c22ea49531f7ffefd43b2e
+  metadata.gz: e82be52459f3d7ff31e368f5cff3a0cf311850428271b921bc57584e640208fccfc65728b578ee9fd3edc181fcab7a2e4c764f5e722906a407ef9b81bd7941f8
+  data.tar.gz: 3c9e5bfdb537b5588af1ac8c08e907fcfef941b9564f4a0b8ed522ea65fdd5faaf4c9939f3bc6c117ec30ed2a8aebed18ceba64efe4adf9c3c2a15a89ca266fd

data/lib/tinybird/callable.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Tinybird
+  module Callable
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    module ClassMethods
+      def call(*args, **)
+        if args.empty?
+          new(**).call
+        else
+          new(*args, **).call
+        end
+      end
+    end
+  end
+end

data/lib/tinybird/configuration.rb

@@ -0,0 +1,26 @@
+# lib/tinybird/configuration.rb
+
+module Tinybird
+  class Configuration
+    attr_accessor :auth_token, :base_url
+
+    def initialize
+      @auth_token = ENV.fetch("TINYBIRD_AUTH_TOKEN", nil)
+      @base_url = ENV.fetch("TINYBIRD_BASE_URL", "https://api.tinybird.co/v0")
+    end
+  end
+
+  class << self
+    attr_writer :configuration
+
+    def configuration
+      @configuration ||= Configuration.new
+    end
+
+    alias_method :config, :configuration
+
+    def configure
+      yield configuration
+    end
+  end
+end

data/lib/tinybird/event/create.rb

@@ -0,0 +1,27 @@
+# lib/tinybird/event/create.rb
+
+module Tinybird
+  module Event
+    class Create
+      include Tinybird::Callable
+      include Tinybird::Requestable
+
+      requestable method: :post
+
+      attr_reader :body, :event_name, :wait, :headers
+
+      def initialize(body, event_name:, wait: false, headers: {})
+        @body = body
+        @event_name = event_name
+        @wait = wait
+        @headers = headers
+      end
+
+      private
+
+      def path_segment
+        "v0/events?name=#{event_name}&wait=#{wait}"
+      end
+    end
+  end
+end

data/lib/tinybird/event.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require "tinybird/event/create"
+
+module Tinybird
+  module Event
+    class << self
+      def create(body, event_name:, wait: false, headers: {})
+        Create.call(body, event_name: event_name, wait: wait, headers: headers)
+      end
+    end
+
+    private_constant :Create
+  end
+end

data/lib/tinybird/http_client.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Tinybird
+  class HttpClient
+    attr_reader :uri
+
+    def initialize(uri)
+      @uri = uri
+    end
+
+    def get(headers = {})
+      request = Net::HTTP::Get.new(uri, headers)
+      make_request(request)
+    end
+
+    def post(body, headers = {})
+      request = Net::HTTP::Post.new(uri, headers)
+      request.body = body.to_json
+      make_request(request)
+    end
+
+    private
+
+    def make_request(request)
+      request["Authorization"] = "Bearer #{Tinybird.config.auth_token}"
+      request["User-Agent"] = "tinybird-ruby/#{Tinybird::VERSION}"
+
+      Net::HTTP.start(uri.hostname, uri.port, use_ssl: uri.scheme == "https") do |http|
+        http.request(request).then do |response|
+          Response.new(response)
+        end
+      end
+    end
+  end
+end

data/lib/tinybird/query/get.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module Tinybird
+  module Query
+    class Get
+      include Tinybird::Callable
+      include Tinybird::Requestable
+
+      requestable method: :get
+
+      ALLOWED_FORMATS = %w[CSV CSVWithoutHeader CSVWithNames JSON TSV TSVWithoutHeader TSVWithNames PrettyCompact JSONEachRow Parquet].freeze
+
+      attr_reader :sql, :format, :headers
+
+      def initialize(sql: "", format: "JSON", headers: {})
+        @sql = sql
+        @format = format
+        @headers = headers
+        validate_format!
+      end
+
+      private
+
+      def path_segment
+        "v0/sql?q=#{formatted_sql}"
+      end
+
+      def body
+        nil
+      end
+
+      def formatted_sql
+        "#{sql} FORMAT #{format}"
+      end
+
+      def validate_format!
+        raise InvalidFormatError, "Invalid format: #{format}. Supported formats are: #{ALLOWED_FORMATS.join(", ")}" unless ALLOWED_FORMATS.include?(format)
+      end
+    end
+  end
+end

data/lib/tinybird/query.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require "tinybird/query/get"
+
+module Tinybird
+  module Query
+    class << self
+      def get(sql: "", format: "JSON", headers: {})
+        Get.call(sql: sql, format: format, headers: headers)
+      end
+    end
+
+    private_constant :Get
+  end
+end

data/lib/tinybird/requestable.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+module Tinybird
+  module Requestable
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    module ClassMethods
+      DEFAULT_REQUEST_METHODS = %i[get post delete].freeze
+
+      private
+
+      def requestable(**options)
+        method = options.fetch(:method, DEFAULT_REQUEST_METHODS)
+        define_http_method(method)
+      end
+
+      def define_http_method(method)
+        define_method(:http_method) do
+          method
+        end
+      end
+    end
+
+    def call(**args)
+      uri = URI(endpoint)
+      client = Tinybird::HttpClient.new(uri)
+      if http_method == :get
+        client.send(http_method, headers)
+      else
+        client.send(http_method, body, headers)
+      end
+    end
+
+    private
+
+    def base_url
+      Tinybird.config.base_url.sub(/\/$/, "")
+    end
+
+    def endpoint
+      [
+        base_url,
+        path_segment
+      ].join("/")
+    end
+
+    def path_segment
+      raise NotImplementedError, "#{self.class} must implement #path_segment"
+    end
+
+    def body
+      raise NotImplementedError, "#{self.class} must implement #body"
+    end
+
+    def headers
+      raise NotImplementedError, "#{self.class} must implement #headers"
+    end
+  end
+end

data/lib/tinybird/response.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module Tinybird
+  class Response
+    attr_reader :status, :headers, :body, :message
+
+    def initialize(response)
+      @status = response.code.to_i
+      @headers = response.to_hash
+      @message = response.message
+      @body = begin
+        JSON.parse(response.body)
+      rescue JSON::ParserError
+        response.body
+      end
+    end
+
+    def inspect
+      "#<#{self.class}:0x#{object_id.to_s(16)} #{status} #{message} readbody=true>"
+    end
+  end
+end

data/lib/tinybird/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Tinybird
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/tinybird.rb

@@ -1,7 +1,22 @@
 # frozen_string_literal: true
 
-require_relative "tinybird/version"
+require "json"
+require "net/http"
+require "uri"
+
+require "tinybird/version"
+require "tinybird/configuration"
+require "tinybird/http_client"
+require "tinybird/response"
+
+require "tinybird/callable"
+require "tinybird/requestable"
+
+require "tinybird/query"
+require "tinybird/event"
 
 module Tinybird
   class Error < StandardError; end
+
+  class InvalidFormatError < Error; end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tinybird
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Drew Monroe
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-08-03 00:00:00.000000000 Z
+date: 2024-08-06 00:00:00.000000000 Z
 dependencies: []
 description: A Ruby client for the Tinybird API
 email:
@@ -32,6 +32,15 @@ files:
 - bin/standardrb
 - docker-compose.yml
 - lib/tinybird.rb
+- lib/tinybird/callable.rb
+- lib/tinybird/configuration.rb
+- lib/tinybird/event.rb
+- lib/tinybird/event/create.rb
+- lib/tinybird/http_client.rb
+- lib/tinybird/query.rb
+- lib/tinybird/query/get.rb
+- lib/tinybird/requestable.rb
+- lib/tinybird/response.rb
 - lib/tinybird/version.rb
 - tinybird.gemspec
 homepage: https://github.com/dvmonroe/tinybird-ruby