fog-google 0.2.0 → 0.3.0

data/lib/fog/google/models/pubsub/subscriptions.rb

@@ -0,0 +1,32 @@
+require "fog/core/collection"
+require "fog/google/models/pubsub/subscription"
+
+module Fog
+  module Google
+    class Pubsub
+      class Subscriptions < Fog::Collection
+        model Fog::Google::Pubsub::Subscription
+
+        # Lists all subscriptions that exist on the project.
+        #
+        # @return [Array<Fog::Google::Pubsub::Subscription>] list of
+        #   subscriptions
+        def all
+          data = service.list_subscriptions.body["subscriptions"] || []
+          load(data)
+        end
+
+        # Retrieves a subscription by name
+        #
+        # @param subscription_name [String] name of subscription to retrieve
+        # @return [Fog::Google::Pubsub::Topic] topic found, or nil if not found
+        def get(subscription_name)
+          subscription = service.get_subscription(subscription_name).body
+          new(subscription)
+        rescue Fog::Errors::NotFound
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/fog/google/models/pubsub/topic.rb

@@ -0,0 +1,72 @@
+require "fog/core/model"
+
+module Fog
+  module Google
+    class Pubsub
+      # Represents a Pubsub topic resource
+      #
+      # @see https://cloud.google.com/pubsub/reference/rest/v1/projects.topics#Topic
+      class Topic < Fog::Model
+        identity :name
+
+        # Creates this topic resource on the service.
+        #
+        # @return [Fog::Google::Pubsub::Topic] this instance
+        def create
+          requires :name
+
+          service.create_topic(name)
+          self
+        end
+
+        # Deletes this topic resource on the service.
+        #
+        # @return [Fog::Google::Pubsub::Topic] this instance
+        def destroy
+          requires :name
+
+          service.delete_topic(name)
+          self
+        end
+
+        # Publish a message to this topic. This method will automatically
+        # encode the message via base64 encoding.
+        #
+        # @param messages [Array<Hash{String => String}, #to_s>] list of messages
+        #   to send; if it's a hash, then the value of key "data" will be sent
+        #   as the message. Additionally, if the hash contains a value for key
+        #   "attributes", then they will be sent as well as attributes on the
+        #   message.
+        # @return [Array<String>] list of message ids
+        def publish(messages)
+          requires :name
+
+          # Ensure our messages are base64 encoded
+          encoded_messages = []
+
+          messages.each do |message|
+            encoded_message = {}
+            if message.is_a?(Hash)
+              encoded_message["attributes"] = message["attributes"]
+              if message.key?("data")
+                encoded_message["data"] = Base64.strict_encode64(message["data"])
+              end
+            else
+              encoded_message["data"] = Base64.strict_encode64(message.to_s)
+            end
+
+            encoded_messages << encoded_message
+          end
+
+          service.publish_topic(name, encoded_messages).body["messageIds"]
+        end
+
+        # Save the instance (does the same thing as #create)
+        # @return [Fog::Google::Pubsub::Topic] this instance
+        def save
+          create
+        end
+      end
+    end
+  end
+end

data/lib/fog/google/models/pubsub/topics.rb

@@ -0,0 +1,31 @@
+require "fog/core/collection"
+require "fog/google/models/pubsub/topic"
+
+module Fog
+  module Google
+    class Pubsub
+      class Topics < Fog::Collection
+        model Fog::Google::Pubsub::Topic
+
+        # Lists all topics that exist on the project.
+        #
+        # @return [Array<Fog::Google::Pubsub::Topic>] list of topics
+        def all
+          data = service.list_topics.body["topics"] || []
+          load(data)
+        end
+
+        # Retrieves a topic by name
+        #
+        # @param topic_name [String] name of topic to retrieve
+        # @return [Fog::Google::Pubsub::Topic] topic found, or nil if not found
+        def get(topic_name)
+          topic = service.get_topic(topic_name).body
+          new(topic)
+        rescue Fog::Errors::NotFound
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/fog/google/monitoring.rb

@@ -1,6 +1,9 @@
 module Fog
   module Google
     class Monitoring < Fog::Service
+      autoload :Mock, File.expand_path("../monitoring/mock", __FILE__)
+      autoload :Real, File.expand_path("../monitoring/real", __FILE__)
+
       requires :google_project
       recognizes :google_client_email, :google_key_location, :google_key_string, :google_client,
                  :app_name, :app_version, :google_json_key_location, :google_json_key_string
@@ -37,51 +40,6 @@ module Fog
 
       # MetricDescriptors
       request :list_metric_descriptors
-
-      class Mock
-        include Fog::Google::Shared
-
-        def initialize(options)
-          shared_initialize(options[:google_project], GOOGLE_MONITORING_API_VERSION, GOOGLE_MONITORING_BASE_URL)
-        end
-
-        def self.data
-          @data ||= Hash.new do |hash, key|
-            hash[key] = {
-              :timeseries => {},
-              :timeseries_descriptors => {},
-              :metric_descriptors => {}
-            }
-          end
-        end
-
-        def self.reset
-          @data = nil
-        end
-
-        def data
-          self.class.data[project]
-        end
-
-        def reset_data
-          self.class.data.delete(project)
-        end
-      end
-
-      class Real
-        include Fog::Google::Shared
-
-        attr_accessor :client
-        attr_reader :monitoring
-
-        def initialize(options)
-          shared_initialize(options[:google_project], GOOGLE_MONITORING_API_VERSION, GOOGLE_MONITORING_BASE_URL)
-          options.merge!(:google_api_scope_url => GOOGLE_MONITORING_API_SCOPE_URLS.join(" "))
-
-          @client = initialize_google_client(options)
-          @monitoring = @client.discovered_api("cloudmonitoring", api_version)
-        end
-      end
     end
   end
 end

data/lib/fog/google/monitoring/mock.rb

@@ -0,0 +1,35 @@
+module Fog
+  module Google
+    class Monitoring
+      class Mock
+        include Fog::Google::Shared
+
+        def initialize(options)
+          shared_initialize(options[:google_project], GOOGLE_MONITORING_API_VERSION, GOOGLE_MONITORING_BASE_URL)
+        end
+
+        def self.data
+          @data ||= Hash.new do |hash, key|
+            hash[key] = {
+              :timeseries => {},
+              :timeseries_descriptors => {},
+              :metric_descriptors => {}
+            }
+          end
+        end
+
+        def self.reset
+          @data = nil
+        end
+
+        def data
+          self.class.data[project]
+        end
+
+        def reset_data
+          self.class.data.delete(project)
+        end
+      end
+    end
+  end
+end

data/lib/fog/google/monitoring/real.rb

@@ -0,0 +1,20 @@
+module Fog
+  module Google
+    class Monitoring
+      class Real
+        include Fog::Google::Shared
+
+        attr_accessor :client
+        attr_reader :monitoring
+
+        def initialize(options)
+          shared_initialize(options[:google_project], GOOGLE_MONITORING_API_VERSION, GOOGLE_MONITORING_BASE_URL)
+          options[:google_api_scope_url] = GOOGLE_MONITORING_API_SCOPE_URLS.join(" ")
+
+          @client = initialize_google_client(options)
+          @monitoring = @client.discovered_api("cloudmonitoring", api_version)
+        end
+      end
+    end
+  end
+end

data/lib/fog/google/pubsub.rb

@@ -0,0 +1,59 @@
+module Fog
+  module Google
+    class Pubsub < Fog::Service
+      autoload :Mock, File.expand_path("../pubsub/mock", __FILE__)
+      autoload :Real, File.expand_path("../pubsub/real", __FILE__)
+
+      requires :google_project
+      recognizes :google_client_email, :google_key_location, :google_key_string, :google_client,
+                 :app_name, :app_version, :google_json_key_location, :google_json_key_string
+
+      GOOGLE_PUBSUB_API_VERSION    = "v1".freeze
+      GOOGLE_PUBSUB_BASE_URL       = "https://www.googleapis.com/pubsub".freeze
+      GOOGLE_PUBSUB_API_SCOPE_URLS = %w(https://www.googleapis.com/auth/pubsub).freeze
+
+      ##
+      # MODELS
+      model_path "fog/google/models/pubsub"
+
+      # Topic
+      model :topic
+      collection :topics
+
+      # Subscription
+      model :subscription
+      collection :subscriptions
+
+      # ReceivedMessage
+      model :received_message
+
+      ##
+      # REQUESTS
+      request_path "fog/google/requests/pubsub"
+
+      # Topic
+      request :list_topics
+      request :get_topic
+      request :create_topic
+      request :delete_topic
+      request :publish_topic
+
+      # Subscription
+      request :list_subscriptions
+      request :get_subscription
+      request :create_subscription
+      request :delete_subscription
+      request :pull_subscription
+      request :acknowledge_subscription
+
+      # Helper class for getting a subscription name
+      #
+      # @param subscription [Subscription, #to_s] subscription instance or name
+      #   of subscription
+      # @return [String] name of subscription
+      def self.subscription_name(subscription)
+        subscription.is_a?(Subscription) ? subscription.name : subscription.to_s
+      end
+    end
+  end
+end

data/lib/fog/google/pubsub/mock.rb

@@ -0,0 +1,34 @@
+module Fog
+  module Google
+    class Pubsub
+      class Mock
+        include Fog::Google::Shared
+
+        def initialize(options)
+          shared_initialize(options[:google_project], GOOGLE_PUBSUB_API_VERSION, GOOGLE_PUBSUB_BASE_URL)
+        end
+
+        def self.data
+          @data ||= Hash.new do |hash, key|
+            hash[key] = {
+              :topics => {},
+              :subscriptions => {}
+            }
+          end
+        end
+
+        def self.reset
+          @data = nil
+        end
+
+        def data
+          self.class.data[project]
+        end
+
+        def reset_data
+          self.class.data.delete(project)
+        end
+      end
+    end
+  end
+end

data/lib/fog/google/pubsub/real.rb

@@ -0,0 +1,20 @@
+module Fog
+  module Google
+    class Pubsub
+      class Real
+        include Fog::Google::Shared
+
+        attr_accessor :client
+        attr_reader :pubsub
+
+        def initialize(options)
+          shared_initialize(options[:google_project], GOOGLE_PUBSUB_API_VERSION, GOOGLE_PUBSUB_BASE_URL)
+          options[:google_api_scope_url] = GOOGLE_PUBSUB_API_SCOPE_URLS.join(" ")
+
+          @client = initialize_google_client(options)
+          @pubsub = @client.discovered_api("pubsub", api_version)
+        end
+      end
+    end
+  end
+end

data/lib/fog/google/requests/pubsub/acknowledge_subscription.rb

@@ -0,0 +1,46 @@
+module Fog
+  module Google
+    class Pubsub
+      class Real
+        # Acknowledges a message received from a subscription.
+        #
+        # @see https://cloud.google.com/pubsub/reference/rest/v1/projects.subscriptions/acknowledge
+        def acknowledge_subscription(subscription, ack_ids)
+          api_method = @pubsub.projects.subscriptions.acknowledge
+
+          parameters = {
+            "subscription" => subscription.to_s
+          }
+
+          body = {
+            "ackIds" => ack_ids
+          }
+
+          request(api_method, parameters, body)
+        end
+      end
+
+      class Mock
+        def acknowledge_subscription(subscription, ack_ids)
+          unless data[:subscriptions].key?(subscription)
+            subscription_resource = subscription.to_s.split("/")[-1]
+            body = {
+              "error" => {
+                "code"    => 404,
+                "message" => "Resource not found (resource=#{subscription_resource}).",
+                "status"  => "NOT_FOUND"
+              }
+            }
+            status = 404
+            return build_excon_response(body, status)
+          end
+
+          sub = data[:subscriptions][subscription]
+          sub[:messages].delete_if { |msg| ack_ids.member?(msg["messageId"]) }
+
+          build_excon_response(nil, 200)
+        end
+      end
+    end
+  end
+end

data/lib/fog/google/requests/pubsub/create_subscription.rb

@@ -0,0 +1,57 @@
+module Fog
+  module Google
+    class Pubsub
+      class Real
+        # Create a subscription resource on a topic.
+        #
+        # @param subscription_name [#to_s] name of the subscription to create.
+        #   Note that it must follow the restrictions of subscription names;
+        #   specifically it must be named within a project (e.g.
+        #   "projects/my-project/subscriptions/my-subscripton")
+        # @param topic [Topic, #to_s] topic instance or name of topic to create
+        #   subscription on
+        # @param push_config [Hash] configuration for a push config (if empty
+        #   hash, then no push_config is created)
+        # @param ack_deadline_seconds [Number] how long the service waits for
+        #   an acknowledgement before redelivering the message; if nil then
+        #   service default of 10 is used
+        # @see https://cloud.google.com/pubsub/reference/rest/v1/projects.subscriptions/create
+        def create_subscription(subscription_name, topic, push_config = {}, ack_deadline_seconds = nil)
+          api_method = @pubsub.projects.subscriptions.create
+
+          parameters = {}
+          parameters["name"] = subscription_name.to_s unless subscription_name.nil?
+
+          body = {
+            "topic" => (topic.is_a?(Topic) ? topic.name : topic.to_s)
+          }
+
+          if !push_config.nil? && push_config.key?("push_endpoint")
+            body["pushConfig"] = push_config["push_endpoint"].clone
+            body["pushConfig"]["attributes"] = push_config["attributes"] if push_config.key?("attributes")
+          end
+
+          body["ackDeadlineSeconds"] = ack_deadline_seconds unless ack_deadline_seconds.nil?
+
+          request(api_method, parameters, body)
+        end
+      end
+
+      class Mock
+        def create_subscription(subscription_name, topic, push_config = {}, ack_deadline_seconds = nil)
+          subscription = {
+            "name"               => subscription_name,
+            "topic"              => topic,
+            "pushConfig"         => push_config,
+            "ackDeadlineSeconds" => ack_deadline_seconds
+          }
+
+          # We also track pending messages
+          data[:subscriptions][subscription_name] = subscription.merge(:messages => [])
+
+          build_excon_response(subscription, 200)
+        end
+      end
+    end
+  end
+end