amqp-client 0.1.0 → 0.2.0

data/lib/amqp/client/message.rb

@@ -1,5 +1,15 @@
 # frozen_string_literal: true
 
 module AMQP
-  Message = Struct.new(:exchange_name, :routing_key, :properties, :body, :redelivered)
+  Message = Struct.new(:channel, :delivery_tag, :exchange_name, :routing_key, :properties, :body, :redelivered, :consumer_tag) do
+    def ack
+      channel.basic_ack(delivery_tag)
+    end
+
+    def reject(requeue: false)
+      channel.basic_reject(delivery_tag, requeue)
+    end
+  end
+
+  ReturnMessage = Struct.new(:reply_code, :reply_text, :exchange, :routing_key, :properties, :body)
 end

data/lib/amqp/client/properties.rb

@@ -0,0 +1,201 @@
+# frozen_string_literal: true
+
+require_relative "./table"
+
+module AMQP
+  # Encode/decode AMQP Properties
+  Properties = Struct.new(:content_type, :content_encoding, :headers, :delivery_mode, :priority, :correlation_id,
+                          :reply_to, :expiration, :message_id, :timestamp, :type, :user_id, :app_id,
+                          keyword_init: true) do
+    def encode
+      flags = 0
+      arr = [flags]
+      fmt = String.new("S>")
+
+      if content_type
+        content_type.is_a?(String) || raise(ArgumentError, "content_type must be a string")
+
+        flags |= (1 << 15)
+        arr << content_type.bytesize << content_type
+        fmt << "Ca*"
+      end
+
+      if content_encoding
+        content_encoding.is_a?(String) || raise(ArgumentError, "content_encoding must be a string")
+
+        flags |= (1 << 14)
+        arr << content_encoding.bytesize << content_encoding
+        fmt << "Ca*"
+      end
+
+      if headers
+        headers.is_a?(Hash) || raise(ArgumentError, "headers must be a hash")
+
+        flags |= (1 << 13)
+        tbl = Table.encode(headers)
+        arr << tbl.bytesize << tbl
+        fmt << "L>a*"
+      end
+
+      if delivery_mode
+        headers.is_a?(Integer) || raise(ArgumentError, "delivery_mode must be an int")
+
+        flags |= (1 << 12)
+        arr << delivery_mode
+        fmt << "C"
+      end
+
+      if priority
+        priority.is_a?(Integer) || raise(ArgumentError, "priority must be an int")
+        flags |= (1 << 11)
+        arr << priority
+        fmt << "C"
+      end
+
+      if correlation_id
+        priority.is_a?(String) || raise(ArgumentError, "correlation_id must be a string")
+
+        flags |= (1 << 10)
+        arr << correlation_id.bytesize << correlation_id
+        fmt << "Ca*"
+      end
+
+      if reply_to
+        reply_to.is_a?(String) || raise(ArgumentError, "reply_to must be a string")
+
+        flags |= (1 << 9)
+        arr << reply_to.bytesize << reply_to
+        fmt << "Ca*"
+      end
+
+      if expiration
+        expiration.is_a?(String) || raise(ArgumentError, "expiration must be a string")
+
+        flags |= (1 << 8)
+        arr << expiration.bytesize << expiration
+        fmt << "Ca*"
+      end
+
+      if message_id
+        message_id.is_a?(String) || raise(ArgumentError, "message_id must be a string")
+
+        flags |= (1 << 7)
+        arr << message_id.bytesize << message_id
+        fmt << "Ca*"
+      end
+
+      if timestamp
+        timestamp.is_a?(Time) || raise(ArgumentError, "timestamp must be a time")
+
+        flags |= (1 << 6)
+        arr << timestamp.to_i
+        fmt << "Q>"
+      end
+
+      if type
+        type.is_a?(String) || raise(ArgumentError, "type must be a string")
+
+        flags |= (1 << 5)
+        arr << type.bytesize << type
+        fmt << "Ca*"
+      end
+
+      if user_id
+        user_id.is_a?(String) || raise(ArgumentError, "user_id must be a string")
+
+        flags |= (1 << 4)
+        arr << user_id.bytesize << user_id
+        fmt << "Ca*"
+      end
+
+      if app_id
+        app_id.is_a?(String) || raise(ArgumentError, "app_id must be a string")
+
+        flags |= (1 << 3)
+        arr << app_id.bytesize << app_id
+        fmt << "Ca*"
+      end
+
+      arr[0] = flags
+      arr.pack(fmt)
+    end
+
+    def self.decode(bytes)
+      h = new
+      flags = bytes.unpack1("S>")
+      pos = 2
+      if (flags & 0x8000).positive?
+        len = bytes[pos].ord
+        pos += 1
+        h[:content_type] = bytes.byteslice(pos, len).force_encoding("utf-8")
+        pos += len
+      end
+      if (flags & 0x4000).positive?
+        len = bytes[pos].ord
+        pos += 1
+        h[:content_encoding] = bytes.byteslice(pos, len).force_encoding("utf-8")
+        pos += len
+      end
+      if (flags & 0x2000).positive?
+        len = bytes.byteslice(pos, 4).unpack1("L>")
+        pos += 4
+        h[:headers] = Table.decode(bytes.byteslice(pos, len))
+        pos += len
+      end
+      if (flags & 0x1000).positive?
+        h[:delivery_mode] = bytes[pos].ord
+        pos += 1
+      end
+      if (flags & 0x0800).positive?
+        h[:priority] = bytes[pos].ord
+        pos += 1
+      end
+      if (flags & 0x0400).positive?
+        len = bytes[pos].ord
+        pos += 1
+        h[:correlation_id] = bytes.byteslice(pos, len).force_encoding("utf-8")
+        pos += len
+      end
+      if (flags & 0x0200).positive?
+        len = bytes[pos].ord
+        pos += 1
+        h[:reply_to] = bytes.byteslice(pos, len).force_encoding("utf-8")
+        pos += len
+      end
+      if (flags & 0x0100).positive?
+        len = bytes[pos].ord
+        pos += 1
+        h[:expiration] = bytes.byteslice(pos, len).force_encoding("utf-8")
+        pos += len
+      end
+      if (flags & 0x0080).positive?
+        len = bytes[pos].ord
+        pos += 1
+        h[:message_id] = bytes.byteslice(pos, len).force_encoding("utf-8")
+        pos += len
+      end
+      if (flags & 0x0040).positive?
+        h[:timestamp] = Time.at(bytes.byteslice(pos, 8).unpack1("Q>"))
+        pos += 8
+      end
+      if (flags & 0x0020).positive?
+        len = bytes[pos].ord
+        pos += 1
+        h[:type] = bytes.byteslice(pos, len).force_encoding("utf-8")
+        pos += len
+      end
+      if (flags & 0x0010).positive?
+        len = bytes[pos].ord
+        pos += 1
+        h[:user_id] = bytes.byteslice(pos, len).force_encoding("utf-8")
+        pos += len
+      end
+      if (flags & 0x0008).positive?
+        len = bytes[pos].ord
+        pos += 1
+        h[:app_id] = bytes.byteslice(pos, len).force_encoding("utf-8")
+      end
+      h
+    end
+  end
+end

data/lib/amqp/client/table.rb

@@ -0,0 +1,123 @@
+# frozen_string_literal: true
+
+module AMQP
+  # Encode/decode AMQP Tables
+  module Table
+    module_function
+
+    def encode(hash)
+      tbl = ""
+      hash.each do |k, v|
+        key = k.to_s
+        tbl += [key.bytesize, key, encode_field(v)].pack("C a* a*")
+      end
+      tbl
+    end
+
+    def decode(bytes)
+      h = {}
+      pos = 0
+
+      while pos < bytes.bytesize
+        key_len = bytes[pos].ord
+        pos += 1
+        key = bytes.byteslice(pos, key_len).force_encoding("utf-8")
+        pos += key_len
+        rest = bytes.byteslice(pos, bytes.bytesize - pos)
+        len, value = decode_field(rest)
+        pos += len + 1
+        h[key] = value
+      end
+      h
+    end
+
+    def encode_field(value)
+      case value
+      when Integer
+        if value > 2**31
+          ["l", value].pack("a q>")
+        else
+          ["I", value].pack("a l>")
+        end
+      when Float
+        ["d", value].pack("a G")
+      when String
+        ["S", value.bytesize, value].pack("a L> a*")
+      when Time
+        ["T", value.to_i].pack("a Q>")
+      when Array
+        bytes = value.map { |e| encode_field(e) }.join
+        ["A", bytes.bytesize, bytes].pack("a L> a*")
+      when Hash
+        bytes = Table.encode(value)
+        ["F", bytes.bytesize, bytes].pack("a L> a*")
+      when true
+        ["t", 1].pack("a C")
+      when false
+        ["t", 0].pack("a C")
+      when nil
+        ["V"].pack("a")
+      else raise "unsupported table field type: #{value.class}"
+      end
+    end
+
+    # returns [length of field including type, value of field]
+    def decode_field(bytes)
+      type = bytes[0]
+      pos = 1
+      case type
+      when "S"
+        len = bytes.byteslice(pos, 4).unpack1("L>")
+        pos += 4
+        [4 + len, bytes.byteslice(pos, len).force_encoding("utf-8")]
+      when "F"
+        len = bytes.byteslice(pos, 4).unpack1("L>")
+        pos += 4
+        [4 + len, decode(bytes.byteslice(pos, len))]
+      when "A"
+        len = bytes.byteslice(pos, 4).unpack1("L>")
+        a = []
+        while pos < len
+          length, value = decode_field(bytes.byteslice(pos, -1))
+          pos += length + 1
+          a << value
+        end
+        [4 + len, a]
+      when "t"
+        [1, bytes[pos].ord == 1]
+      when "b"
+        [1, bytes.byteslice(pos, 1).unpack1("c")]
+      when "B"
+        [1, bytes.byteslice(pos, 1).unpack1("C")]
+      when "s"
+        [2, bytes.byteslice(pos, 2).unpack1("s")]
+      when "u"
+        [2, bytes.byteslice(pos, 2).unpack1("S")]
+      when "I"
+        [4, bytes.byteslice(pos, 4).unpack1("l>")]
+      when "i"
+        [4, bytes.byteslice(pos, 4).unpack1("L>")]
+      when "l"
+        [8, bytes.byteslice(pos, 8).unpack1("q>")]
+      when "f"
+        [4, bytes.byteslice(pos, 4).unpack1("g")]
+      when "d"
+        [8, bytes.byteslice(pos, 8).unpack1("G")]
+      when "D"
+        scale = bytes[pos].ord
+        pos += 1
+        value = bytes.byteslice(pos, 4).unpack1("L>")
+        d = value / 10**scale
+        [5, d]
+      when "x"
+        len = bytes.byteslice(pos, 4).unpack1("L>")
+        [4 + len, bytes.byteslice(pos, len)]
+      when "T"
+        [8, Time.at(bytes.byteslice(pos, 8).unpack1("Q>"))]
+      when "V"
+        [0, nil]
+      else raise "unsupported table field type: #{type}"
+      end
+    end
+  end
+end

data/lib/amqp/client/version.rb

@@ -2,6 +2,6 @@
 
 module AMQP
   class Client
-    VERSION = "0.1.0"
+    VERSION = "0.2.0"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: amqp-client
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Carl Hörberg
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2021-05-28 00:00:00.000000000 Z
+date: 2021-08-19 00:00:00.000000000 Z
 dependencies: []
 description: Work in progress
 email:
@@ -23,7 +23,6 @@ files:
 - ".rubocop_todo.yml"
 - CHANGELOG.md
 - Gemfile
-- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile
@@ -36,6 +35,8 @@ files:
 - lib/amqp/client/errors.rb
 - lib/amqp/client/frames.rb
 - lib/amqp/client/message.rb
+- lib/amqp/client/properties.rb
+- lib/amqp/client/table.rb
 - lib/amqp/client/version.rb
 homepage: https://github.com/cloudamqp/amqp-client.rb
 licenses:

data/Gemfile.lock

@@ -1,42 +0,0 @@
-PATH
-  remote: .
-  specs:
-    amqp-client (0.1.0)
-
-GEM
-  remote: https://rubygems.org/
-  specs:
-    ast (2.4.2)
-    minitest (5.14.2)
-    parallel (1.20.1)
-    parser (3.0.1.0)
-      ast (~> 2.4.1)
-    rainbow (3.0.0)
-    rake (13.0.3)
-    regexp_parser (2.1.1)
-    rexml (3.2.4)
-    rubocop (1.12.1)
-      parallel (~> 1.10)
-      parser (>= 3.0.0.0)
-      rainbow (>= 2.2.2, < 4.0)
-      regexp_parser (>= 1.8, < 3.0)
-      rexml
-      rubocop-ast (>= 1.2.0, < 2.0)
-      ruby-progressbar (~> 1.7)
-      unicode-display_width (>= 1.4.0, < 3.0)
-    rubocop-ast (1.4.1)
-      parser (>= 2.7.1.5)
-    ruby-progressbar (1.11.0)
-    unicode-display_width (2.0.0)
-
-PLATFORMS
-  x86_64-linux
-
-DEPENDENCIES
-  amqp-client!
-  minitest (~> 5.0)
-  rake (~> 13.0)
-  rubocop (~> 1.7)
-
-BUNDLED WITH
-   2.2.15