azure-fix 0.7.0.pre3

data/lib/azure/queue/serialization.rb

@@ -0,0 +1,105 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+require 'azure/service/serialization'
+require 'azure/service/enumeration_results'
+
+require 'azure/queue/queue'
+require 'azure/queue/message'
+
+module Azure
+  module Queue
+    module Serialization
+      include Service::Serialization
+
+      def self.queue_messages_from_xml(xml, decode)
+        xml = slopify(xml)
+
+        expect_node("QueueMessagesList", xml)
+        results = []
+        return results unless (xml > "QueueMessage").any?
+  
+        if xml.QueueMessage.count == 0
+          results.push(queue_message_from_xml(xml.QueueMessage, decode))
+        else
+          xml.QueueMessage.each { |message_node|
+            results.push(queue_message_from_xml(message_node, decode))
+          }
+        end
+
+        results
+      end
+
+      def self.queue_message_from_xml(xml, decode)
+        xml = slopify(xml)
+        expect_node("QueueMessage", xml)
+
+        Message.new do |msg|
+          msg.id = xml.MessageId.text if (xml > "MessageId").any?
+          msg.insertion_time = xml.InsertionTime.text if (xml > "InsertionTime").any?
+          msg.expiration_time = xml.ExpirationTime.text if (xml > "ExpirationTime").any?
+          msg.dequeue_count = xml.DequeueCount.text.to_i if (xml > "DequeueCount").any?
+          msg.message_text = xml.MessageText.text if (xml > "MessageText").any?
+          msg.time_next_visible = xml.TimeNextVisible.text if (xml > "TimeNextVisible").any?
+          msg.pop_receipt = xml.PopReceipt.text if (xml > "PopReceipt").any?
+
+          msg.message_text = Base64.decode64(msg.message_text) if decode
+        end
+      end
+
+      def self.message_to_xml(message_text, encode)
+        if encode
+          builder = Nokogiri::XML::Builder.new do |xml|
+            xml.QueueMessage { xml.MessageText Base64.encode64(message_text) }
+          end
+        else
+          builder = Nokogiri::XML::Builder.new(:encoding => 'utf-8') do |xml|
+            xml.QueueMessage { xml.MessageText message_text.encode("utf-8") }
+          end
+        end
+        builder.to_xml
+      end
+
+      def self.queue_enumeration_results_from_xml(xml)
+        xml = slopify(xml)
+        expect_node("EnumerationResults", xml)
+
+        results = enumeration_results_from_xml(xml, Azure::Service::EnumerationResults.new)
+
+        return results unless (xml > "Queues").any? && ((xml > "Queues") > "Queue").any?
+
+        if xml.Queues.Queue.count == 0
+          results.push(queue_from_xml(xml.Queues.Queue))
+        else
+          xml.Queues.Queue.each { |queue_node|
+            results.push(queue_from_xml(queue_node))
+          }
+        end
+
+        results
+      end
+
+      def self.queue_from_xml(xml)
+        xml = slopify(xml)
+        expect_node("Queue", xml)
+
+        queue = Queue.new
+        queue.name = xml.Name.text if (xml > "Name").any?
+        queue.metadata = metadata_from_xml(xml.Metadata) if (xml > "Metadata").any?
+
+        queue
+      end
+    end
+  end
+end

data/lib/azure/service/access_policy.rb

@@ -0,0 +1,26 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+module Azure
+  module Service
+    class AccessPolicy
+      def initialize
+        yield self if block_given?
+      end
+      attr_accessor :start
+      attr_accessor :expiry
+      attr_accessor :permission
+    end
+  end
+end

data/lib/azure/service/cors.rb

@@ -0,0 +1,11 @@
+module Azure
+  module Service
+    class Cors 
+      def initialize
+        yield self if block_given?
+      end
+
+      attr_accessor :cors_rules
+    end
+  end
+end

data/lib/azure/service/cors_rule.rb

@@ -0,0 +1,15 @@
+module Azure
+  module Service
+    class CorsRule
+      def initialize
+        yield self if block_given?
+      end
+
+      attr_accessor :allowed_origins
+      attr_accessor :allowed_methods
+      attr_accessor :max_age_in_seconds
+      attr_accessor :exposed_headers
+      attr_accessor :allowed_headers
+    end
+  end
+end

data/lib/azure/service/enumeration_results.rb

@@ -0,0 +1,21 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+module Azure
+  module Service
+    class EnumerationResults < Array
+      attr_accessor :continuation_token
+    end
+  end
+end

data/lib/azure/service/logging.rb

@@ -0,0 +1,32 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+require 'azure/service/retention_policy'
+
+module Azure
+  module Service
+    class Logging 
+      def initialize
+        @retention_policy = RetentionPolicy.new
+        yield self if block_given?
+      end
+
+      attr_accessor :version
+      attr_accessor :delete
+      attr_accessor :read
+      attr_accessor :write
+      attr_accessor :retention_policy
+    end
+  end
+end

data/lib/azure/service/metrics.rb

@@ -0,0 +1,31 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+require 'azure/service/retention_policy'
+
+module Azure
+  module Service
+    class Metrics
+      def initialize
+        @retention_policy = RetentionPolicy.new
+        yield self if block_given?
+      end
+      
+      attr_accessor :version
+      attr_accessor :enabled
+      attr_accessor :include_apis
+      attr_accessor :retention_policy
+    end
+  end
+end

data/lib/azure/service/retention_policy.rb

@@ -0,0 +1,25 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+module Azure
+  module Service
+    class RetentionPolicy
+      def initialize
+        yield self if block_given?
+      end
+      attr_accessor :enabled
+      attr_accessor :days
+    end
+  end
+end

data/lib/azure/service/serialization.rb

@@ -0,0 +1,299 @@
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+require 'nokogiri'
+
+require 'azure/service/enumeration_results'
+require 'azure/service/signed_identifier'
+require 'azure/service/access_policy'
+
+require 'azure/service/storage_service_properties'
+require 'azure/service/logging'
+require 'azure/service/metrics'
+require 'azure/service/retention_policy'
+require 'azure/service/cors'
+require 'azure/service/cors_rule'
+
+module Azure
+  module Service
+    module Serialization
+      module ClassMethods
+
+        def signed_identifiers_from_xml(xml)
+          xml = slopify(xml)
+          expect_node("SignedIdentifiers", xml)
+
+          identifiers = []
+          return identifiers unless (xml > "SignedIdentifier").any?
+
+          if xml.SignedIdentifier.count == 0
+            identifiers.push(signed_identifier_from_xml(xml.SignedIdentifier))
+          else
+            xml.SignedIdentifier.each { |identifier_node| 
+              identifiers.push(signed_identifier_from_xml(identifier_node))
+            }
+          end
+
+          identifiers
+        end
+
+        def signed_identifiers_to_xml(signed_identifiers)
+          builder = Nokogiri::XML::Builder.new(:encoding=>"utf-8") do |xml|
+            xml.SignedIdentifiers {
+              signed_identifiers.each do |identifier|
+                xml.SignedIdentifier {
+                  xml.Id identifier.id
+                  xml.AccessPolicy {
+                    xml.Start identifier.access_policy.start
+                    xml.Expiry identifier.access_policy.expiry
+                    xml.Permission identifier.access_policy.permission
+                  }
+                }
+              end
+            }
+          end
+          builder.to_xml
+        end
+
+        def signed_identifier_from_xml(xml)
+          xml = slopify(xml)
+          expect_node("SignedIdentifier", xml)
+
+          SignedIdentifier.new do |identifier|
+            identifier.id = xml.Id.text if (xml > "Id").any?
+            identifier.access_policy = access_policy_from_xml(xml.AccessPolicy) if (xml > "AccessPolicy").any?
+          end
+        end
+
+        def access_policy_from_xml(xml)
+          xml = slopify(xml)
+          expect_node("AccessPolicy", xml)
+
+          AccessPolicy.new do |policy|
+            policy.start = xml.Start.text if (xml > "Start").any?
+            policy.expiry = xml.Expiry.text if (xml > "Expiry").any?
+            policy.permission = xml.Permission.text if (xml > "Permission").any?
+          end
+        end
+
+        def enumeration_results_from_xml(xml, results)
+          xml = slopify(xml)
+          expect_node("EnumerationResults", xml)
+
+          results = results || EnumerationResults.new; 
+
+          results.continuation_token = xml.NextMarker.text if (xml > "NextMarker").any?
+          results
+        end
+
+        def metadata_from_xml(xml)
+          xml = slopify(xml)
+          expect_node("Metadata", xml)
+
+          metadata = {}
+
+          xml.children.each { |meta_node|
+
+            key = meta_node.name.downcase
+            if metadata.has_key? key 
+              metadata[key] = [metadata[key]] unless metadata[key].respond_to? :push
+              metadata[key].push(meta_node.text)
+            else
+              metadata[key] = meta_node.text
+            end
+          }
+          metadata
+        end
+
+        def metadata_from_headers(headers)
+          metadata = {}
+
+          headers.each { |k, v| 
+            if key = k[/^x-ms-meta-(.*)/, 1]
+              if metadata.has_key? key 
+                metadata[key] = [metadata[key]] unless metadata[key].respond_to? :push
+                metadata[key].push(v)
+              else
+                metadata[key] = v
+              end
+            end
+          }
+
+          metadata
+        end
+
+        def retention_policy_to_xml(retention_policy, xml)
+          xml.RetentionPolicy {
+              xml.Enabled retention_policy.enabled unless retention_policy.enabled == nil
+              xml.Days retention_policy.days if retention_policy.days
+            }
+        end
+
+        def retention_policy_from_xml(xml)
+          xml = slopify(xml)
+          expect_node("RetentionPolicy", xml)
+
+          RetentionPolicy.new do |policy|
+            policy.enabled = to_bool(xml.Enabled.text) if (xml > "Enabled").any?
+            policy.days = xml.Days.text.to_i if (xml > "Days").any?
+          end
+        end
+
+        def metrics_to_xml_children(metrics, xml)
+          xml.Version metrics.version if metrics.version
+          xml.Enabled metrics.enabled unless metrics.enabled == nil
+          xml.IncludeAPIs metrics.include_apis unless metrics.include_apis == nil
+          retention_policy_to_xml(metrics.retention_policy, xml) if metrics.retention_policy
+        end
+
+        def hour_metrics_to_xml(metrics, xml)
+          xml.HourMetrics { 
+            metrics_to_xml_children(metrics, xml)
+          }
+        end
+
+        def minute_metrics_to_xml(metrics, xml)
+          xml.MinuteMetrics { 
+            metrics_to_xml_children(metrics, xml)
+          }
+        end
+
+        def metrics_from_xml(xml)
+          xml = slopify(xml)
+
+          Metrics.new do |metrics|
+            metrics.version = xml.Version.text if (xml > "Version").any?
+            metrics.enabled = to_bool(xml.Enabled.text) if (xml > "Enabled").any?
+            metrics.include_apis = to_bool(xml.IncludeAPIs.text) if (xml > "IncludeAPIs").any?
+            metrics.retention_policy = retention_policy_from_xml(xml.RetentionPolicy)
+          end
+        end
+
+        def logging_to_xml(logging, xml)
+          xml.Logging { 
+            xml.Version logging.version if logging.version
+            xml.Delete logging.delete unless logging.delete == nil
+            xml.Read logging.read unless logging.read == nil
+            xml.Write logging.write unless logging.write == nil
+            retention_policy_to_xml(logging.retention_policy, xml) if logging.retention_policy
+          }
+        end
+
+        def logging_from_xml(xml)
+          xml = slopify(xml)
+          expect_node("Logging", xml)
+
+          Logging.new do |logging|
+            logging.version = xml.Version.text if (xml > "Version").any?
+            logging.delete = to_bool(xml.Delete.text) if (xml > "Delete").any?
+            logging.read = to_bool(xml.Read.text) if (xml > "Read").any?
+            logging.write = to_bool(xml.Write.text) if (xml > "Write").any?
+            logging.retention_policy = retention_policy_from_xml(xml.RetentionPolicy)
+          end
+        end
+
+        def cors_to_xml(cors, xml)
+          xml.Cors {
+            cors.cors_rules.to_a.each do |cors_rule|
+              cors_rule_to_xml(cors_rule, xml)
+            end
+          }
+        end
+
+        def cors_rule_to_xml(cors_rule, xml)
+          xml.CorsRule {
+            xml.AllowedOrigins cors_rule.allowed_origins.join(",") if cors_rule.allowed_origins
+            xml.AllowedMethods cors_rule.allowed_methods.join(",") if cors_rule.allowed_methods
+            xml.MaxAgeInSeconds cors_rule.max_age_in_seconds if cors_rule.max_age_in_seconds
+            xml.ExposedHeaders cors_rule.exposed_headers.join(",") if cors_rule.exposed_headers
+            xml.AllowedHeaders cors_rule.allowed_headers.join(",") if cors_rule.allowed_headers
+          }
+        end
+
+        def cors_from_xml(xml)
+          xml = slopify(xml)
+          expect_node("Cors", xml)
+
+          Cors.new do |cors|
+            cors.cors_rules = xml.children.to_a.map {|child| cors_rule_from_xml(child)}
+          end
+        end
+
+        def cors_rule_from_xml(xml)
+          xml = slopify(xml)
+          expect_node("CorsRule", xml)
+
+          CorsRule.new do |cors_rule|
+            cors_rule.allowed_origins =  ary_from_node(xml.AllowedOrigins) if (xml > "AllowedOrigins").any?
+            cors_rule.allowed_methods =  ary_from_node(xml.AllowedMethods) if (xml > "AllowedMethods").any?
+            cors_rule.max_age_in_seconds = xml.MaxAgeInSeconds.text.to_i if (xml > "MaxAgeInSeconds").any?
+            cors_rule.exposed_headers = ary_from_node(xml.ExposedHeaders) if (xml > "ExposedHeaders").any?
+            cors_rule.allowed_headers = ary_from_node(xml.AllowedHeaders) if (xml > "AllowedHeaders").any?
+          end
+        end
+
+        def ary_from_node(node)
+          node.text.split(",").map {|s| s.strip}
+        end
+        
+        def service_properties_to_xml(properties)
+          builder = Nokogiri::XML::Builder.new(:encoding => 'utf-8') do |xml|
+            xml.StorageServiceProperties {
+              xml.DefaultServiceVersion(properties.default_service_version) if properties.default_service_version
+              logging_to_xml(properties.logging, xml) if properties.logging
+              hour_metrics_to_xml(properties.hour_metrics, xml) if properties.hour_metrics
+              minute_metrics_to_xml(properties.minute_metrics, xml) if properties.minute_metrics
+              cors_to_xml(properties.cors, xml) if properties.cors
+            }
+          end
+          builder.to_xml
+        end
+
+        def service_properties_from_xml(xml)
+          xml = slopify(xml)
+          expect_node("StorageServiceProperties", xml)
+
+          StorageServiceProperties.new do |props|
+            props.default_service_version = xml.DefaultServiceVersion.text if (xml > "DefaultServiceVersion").any?
+            props.logging = logging_from_xml(xml.Logging)
+            props.hour_metrics = metrics_from_xml(xml.HourMetrics)
+            props.minute_metrics = metrics_from_xml(xml.MinuteMetrics)
+            props.cors = cors_from_xml(xml.Cors)
+          end
+        end
+
+        def to_bool(s)
+          (s || "").downcase == 'true'
+        end
+
+        def slopify(xml)
+          node = (xml.is_a? String) ? Nokogiri.Slop(xml).root : xml
+          node.slop! if node.is_a? Nokogiri::XML::Document unless node.respond_to? :method_missing
+          node = node.root if node.is_a? Nokogiri::XML::Document
+          node
+        end
+
+        def expect_node(node_name, xml)
+          raise "Xml is not a #{node_name} node." unless xml.name == node_name
+        end
+      end
+
+      extend ClassMethods
+      
+      def self.included( other )
+        other.extend( ClassMethods )
+      end
+    end
+  end
+end