fcmpush 1.1.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 512cdee1b75c4be432fd3f1be390063aca41cb1e4483a25c43b1a09f4e81cd2e
-  data.tar.gz: '09fb06b6bcf93ccf00654e638a8b5c5f6473b1c4fac28a12b052ec735a219cf1'
+  metadata.gz: 1daa7e3e80decc9d67986e0ee52eee2863e267dada73aab4ded259f4499577f0
+  data.tar.gz: c3f1ae5e9a147f98e32166ccabdc0d8f8a9ad37cf5157827c1e9d3eec67f6e15
 SHA512:
-  metadata.gz: e58895de6a8f39f07a6cdbc795d499310d575a8c76b9d9522535fed3e6017accbe8b4ef89372d666a8f9ff0a8478b2595501aff29f843e29ba679cb2f4e0e1e9
-  data.tar.gz: e6bd28c37659008f8b312206700e6ebfa9eb4304b41934e7899a57943810b600b8bf3982f4b0e2b3faf438bc2340c3e282907551602c1daa170f050873e40cc4
+  metadata.gz: 44d3ca77d9d4ce80b28f4cae8a33b8a6c7ce424b796d1c7509faf353053d64f44eed78b30bf683e92c8d7da0186b19d49ee026c1297ca986dd5e8bf344789282
+  data.tar.gz: 5e6cc5dbff0abaa26db3896e0662a06ea64cab2f2e5bd655710a3571eae9b407d17e0396dcce113199a8e245f105a2145a9e194dc95d84ff57e1ebedabab9e5b

data/lib/fcmpush/batch.rb

@@ -0,0 +1,37 @@
+module Fcmpush
+  module Batch
+    PART_BOUNDRY = '__END_OF_PART__'.freeze
+
+    def make_batch_payload(messages, headers)
+      uri = URI.join(domain, path)
+      subrequests = messages.map do |payload|
+        req = Net::HTTP::Post.new(uri, headers)
+        req.body = payload
+        req
+      end
+      subrequests.map.with_index { |req, idx| create_part(req, PART_BOUNDRY, idx) }.join + "--#{PART_BOUNDRY}\r\n"
+    end
+
+    def create_part(request, part_boundry, idx)
+      serialized_request = serialize_sub_request(request)
+      "--#{part_boundry}\r\n" \
+        "Content-Length: #{serialized_request.length}\r\n" \
+        "Content-Type: application/http\r\n" \
+        "Content-Id: #{idx + 1}\r\n" \
+        "Content-Transfer-Encoding: binary\r\n" \
+        "\r\n" \
+        "#{serialized_request}\r\n"
+    end
+
+    def serialize_sub_request(request)
+      body_str = request.body.is_a?(String) ? request.body : request.body.to_json
+      subreqest = "POST #{request.path} HTTP/1.1\r\n" \
+                  "Content-Length: #{body_str.length}\r\n"
+      request.to_hash.each do |k, v|
+        subreqest += "#{k}: #{v.join(';')}\r\n"
+      end
+      subreqest += "\r\n" \
+                   "#{body_str}\r\n"
+    end
+  end
+end

data/lib/fcmpush/batch_response.rb

@@ -0,0 +1,39 @@
+require 'delegate' if RUBY_VERSION >= '2.7'
+
+module Fcmpush
+  class BatchResponse < DelegateClass(Net::HTTPResponse)
+    alias response __getobj__
+    alias headers to_hash
+    HAS_SYMBOL_GC = RUBY_VERSION > '2.2.0'
+
+    def json
+      parsable? ? @parsed ||= parse_body(body) : nil
+    end
+
+    def inspect
+      "#<BatchResponse response: #{response.inspect}, json: #{json}>"
+    end
+    alias to_s inspect
+
+    def parsable?
+      !body.nil? && !body.empty?
+    end
+
+    def success_count
+      @success_count ||= json.length - failure_count
+    end
+
+    def failure_count
+      @failure_count ||= json.select { |i| i[:error] }.size
+    end
+
+    private
+
+      def parse_body(raw_body)
+        devider = raw_body.match(/(\r\n--batch_.*)\r\n/)[1]
+        raw_body.split(devider)[1..-2].map do |response|
+          JSON.parse(response.split("\r\n\r\n")[2], symbolize_names: HAS_SYMBOL_GC)
+        end
+      end
+  end
+end

data/lib/fcmpush/client.rb

@@ -1,13 +1,17 @@
 require 'fcmpush/exceptions'
+require 'fcmpush/batch'
 require 'fcmpush/json_response'
+require 'fcmpush/batch_response'
 
 module Fcmpush
   V1_ENDPOINT_PREFIX = '/v1/projects/'.freeze
   V1_ENDPOINT_SUFFIX = '/messages:send'.freeze
   TOPIC_DOMAIN = 'https://iid.googleapis.com'.freeze
   TOPIC_ENDPOINT_PREFIX = '/iid/v1'.freeze
+  BATCH_ENDPOINT = '/batch'.freeze
 
   class Client
+    include Batch
     attr_reader :domain, :path, :connection, :configuration, :server_key, :access_token, :access_token_expiry
 
     def initialize(domain, project_id, configuration, **options)
@@ -65,6 +69,14 @@ module Fcmpush
       raise NetworkError, "A network error occurred: #{e.class} (#{e.message})"
     end
 
+    def batch_push(messages, query: {}, headers: {})
+      uri, request = make_batch_request(messages, query, headers)
+      response = exception_handler(connection.request(uri, request))
+      BatchResponse.new(response)
+    rescue Timeout::Error, Errno::EINVAL, Errno::ECONNRESET, EOFError, Net::HTTPBadResponse, Net::HTTPHeaderSyntaxError, Net::ProtocolError => e
+      raise NetworkError, "A network error occurred: #{e.class} (#{e.message})"
+    end
+
     private
 
       def make_push_request(body, query, headers)
@@ -126,5 +138,18 @@ module Fcmpush
           registration_tokens: instance_ids
         }.to_json
       end
+
+      def make_batch_request(messages, query, headers)
+        uri = URI.join(domain, BATCH_ENDPOINT)
+        uri.query = URI.encode_www_form(query) unless query.empty?
+
+        access_token_refresh
+        headers = v1_authorized_header(headers)
+        post = Net::HTTP::Post.new(uri, headers)
+        post['Content-Type'] = "multipart/mixed; boundary=#{::Fcmpush::Batch::PART_BOUNDRY}"
+        post.body = make_batch_payload(messages, headers)
+
+        [uri, post]
+      end
   end
 end

data/lib/fcmpush/version.rb

@@ -1,3 +1,3 @@
 module Fcmpush
-  VERSION = '1.1.0'.freeze
+  VERSION = '1.2.0'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: fcmpush
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors:
 - miyataka
-autorequire: 
+autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-03-17 00:00:00.000000000 Z
+date: 2020-07-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: googleauth
@@ -80,7 +80,7 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.0'
-description: 
+description:
 email:
 - voyager.3taka28@gmail.com
 executables: []
@@ -89,6 +89,8 @@ extra_rdoc_files: []
 files:
 - fcmpush.gemspec
 - lib/fcmpush.rb
+- lib/fcmpush/batch.rb
+- lib/fcmpush/batch_response.rb
 - lib/fcmpush/client.rb
 - lib/fcmpush/configuration.rb
 - lib/fcmpush/exceptions.rb
@@ -100,7 +102,7 @@ licenses:
 metadata:
   homepage_uri: https://github.com/miyataka/fcmpush
   source_code_uri: https://github.com/miyataka/fcmpush
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -119,7 +121,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubygems_version: 3.0.3
-signing_key: 
+signing_key:
 specification_version: 4
 summary: Firebase Cloud Messaging API wrapper for ruby, supports HTTP v1. And including
   access_token Auto Refresh feature!