dyn-rb 1.0.2

data/lib/dyn/traffic/base.rb

@@ -0,0 +1,57 @@
+#
+# Author:: Sunny Gleason (<sunny@thesunnycloud.com>)
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2013 Dyn, Inc.
+# Copyright:: Copyright (c) 2010 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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 Dyn
+  module Traffic
+    class Base
+      attr_accessor :zone
+
+      # Publish any pending changes to the zone - required to make any alterations permanent.
+      #
+      # See: https://manage.dynect.net/help/docs/api2/rest/resources/Zone.html
+      #
+      # @param [String] The zone to publish - if one is provided when instantiated, we use that.
+      # @return [Hash] The dynect API response
+      def publish
+        @dyn.put("Zone/#{@zone}", { "publish" => true })
+      end
+
+      # Freeze the zone.
+      #
+      # See: https://manage.dynect.net/help/docs/api2/rest/resources/Zone.html
+      #
+      # @param [String] The zone to freeze - if one is provided when instantiated, we use that.
+      # @return [Hash] The dynect API response
+      def freeze
+        @dyn.put("Zone/#{@zone}", { "freeze" => true })
+      end
+
+      # Thaw the zone.
+      #
+      # See: https://manage.dynect.net/help/docs/api2/rest/resources/Zone.html
+      #
+      # @param [String] The zone to thaw - if one is provided when instantiated, we use that.
+      # @return [Hash] The dynect API response
+      def thaw
+        @dyn.put("Zone/#{@zone}", { "thaw" => true })
+      end
+    end
+  end
+end

data/lib/dyn/traffic/gslb.rb

@@ -0,0 +1,169 @@
+#
+# Author:: Sunny Gleason (<sunny@thesunnycloud.com>)
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Evan  (<evan@pagerduty.com>)
+# Copyright:: Copyright (c) 2013 Dyn, Inc.
+# Copyright:: Copyright (c) 2010 Opscode, Inc.
+# Copyright:: Copyright (c) 2013 PagerDuty, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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 Dyn
+  module Traffic
+    class GSLB < Base
+      def initialize(dyn, zone, options)
+        @dyn           = dyn
+        @zone          = zone
+        @resource_path = "GSLB/#{@zone}"
+
+        @fqdn         = options[:fqdn]
+        @ttl          = options[:ttl] || 30
+        @host_list    = options[:host_list] || {}
+        @contact_nick = options[:contact_nick] || 'owner'
+      
+        @region_code  = options[:region_code] || 'global'
+        @monitor      = options[:monitor] || {}
+        @serve_count  = options[:serve_count] || 1
+        @min_healthy  = options[:min_healthy] || 1
+      end
+
+      def [](host_list_key)
+        @host_list[host_list_key]
+      end
+
+      def fqdn(value=nil)
+        value ? (@fqdn = value; self) : @fqdn
+      end
+
+      def contact_nick(value=nil)
+        value ? (@contact_nick = value; self) : @contact_nick
+      end
+
+      def ttl(value=nil)
+        value ? (@ttl = value; self) : @ttl
+      end
+
+      def min_healthy(value=nil)
+        value ? (@min_healthy = value; self) : @min_healthy
+      end
+
+      def serve_count(value=nil)
+        value ? (@serve_count = value; self) : @serve_count
+      end
+
+      def region_code(value=nil)
+        # US West, US Central, US East, EU West, EU Central, EU East, Asia, global
+        value ? (@region_code = value; self) : @region_code
+      end
+
+      def host_list(value=nil)
+        value ? (@host_list = value; self) : @host_list
+      end
+
+      def monitor(value=nil)
+        # :protocol => 'HTTP', :interval => 1, :retries => 2, :timeout => 10, :port => 8000,
+        # :path => '/healthcheck', :host => 'example.com', :header => 'X-User-Agent: DynECT Health\n', :expected => 'passed'
+        if value
+          @monitor = {}
+          value.each do |k,v|
+            @monitor[k] = v
+          end
+        end
+        @monitor
+      end
+
+      def add_host(value)
+        # :address => 'x.x.x.x', :label => 'friendly-name', :weight => 10, :serve_mode => 'obey'
+        @host_list[value[:address]] = value
+        self
+      end
+
+      def resource_path
+        "GSLB/#{@zone}"
+      end
+
+      def get(fqdn=nil, region_code='global')
+        if fqdn
+          results = @dyn.get("#{@resource_path}/#{fqdn}")
+          region = {}
+          results["region"].each {|r| region = r if r["region_code"] == region_code}
+          raise Dyn::Exceptions::RequestFailed, "Cannot find #{region_code} GSLB pool for #{fqdn}" if region.empty?
+
+          # Default monitor timeout is 0, but specifying timeout 0 on a put or post results in an exception
+          results["monitor"].delete("timeout") if results["monitor"]["timeout"] == 0
+
+          host_list = {}
+          region["pool"].each do |h|
+            host_list[h["address"]] = {
+                                      :address => h["address"],
+                                      :label => h["label"],
+                                      :weight => h["weight"],
+                                      :serve_mode => h["serve_mode"]
+                                      }
+          end
+          Dyn::Traffic::GSLB.new(@dyn, results["zone"], {
+                                   :fqdn => results["fqdn"],
+                                   :ttl => results["ttl"],
+                                   :host_list => host_list,
+                                   :contact_nick => results["contact_nickname"],
+                                   :region_code => region["region_code"],
+                                   :monitor => results["monitor"],
+                                   :serve_count => region["serve_count"],
+                                   :min_healthy => region["min_healthy"]
+                                   })
+        else
+          @dyn.get(resource_path)
+        end
+      end
+
+      def find(fqdn, query_hash)
+        results = []
+        get(fqdn).each do |rr|
+          query_hash.each do |key, value|
+            results << rr if rr[key.to_s] == value
+          end
+        end
+        results
+      end
+
+      def save(replace=false)
+        if replace == true || replace == :replace
+          @dyn.put("#{@resource_path}/#{@fqdn}", self)
+        else
+          @dyn.post("#{@resource_path}/#{@fqdn}", self)
+        end
+        self
+      end
+
+      def delete
+        @dyn.delete("#{@resource_path}/#{fqdn}")
+      end
+
+      def to_json
+        {
+          "ttl"   => @ttl,
+          "monitor" => @monitor,
+          "region" => {
+            "region_code" => @region_code,
+            "serve_count" => @serve_count,
+            "min_healthy" => @min_healthy,
+            "pool" => @host_list.values
+          },
+          "contact_nickname" => @contact_nick
+        }.to_json
+      end
+    end
+  end
+end

data/lib/dyn/traffic/http_redirect.rb

@@ -0,0 +1,83 @@
+#
+# Author:: Sunny Gleason (<sunny@thesunnycloud.com>)
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2013 Dyn, Inc.
+# Copyright:: Copyright (c) 2010 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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 Dyn
+  module Traffic
+    class HTTPRedirect
+      attr_accessor :fqdn, :code, :keep_uri, :url
+
+      def initialize(dyn, zone, fqdn=nil)
+        @dyn = dyn
+        @zone = zone
+        @fqdn = fqdn
+        @code = nil
+        @keep_uri = nil
+        @url = nil
+      end
+
+      def fqdn(value=nil)
+        value ? (@fqdn = value; self) : @fqdn
+      end
+
+      def code(value=nil)
+        value ? (@code = value; self) : @code
+      end
+
+      def keep_uri(value=nil)
+        value ? (@keep_uri = value; self) : @keep_uri
+      end
+
+      def url(value=nil)
+        value ? (@url = value; self) : @url
+      end
+
+      def resource_path
+        "HTTPRedirect"
+      end
+
+      def get(fqdn = nil, record_id=nil)
+        @dyn.get("#{resource_path}/#{zone}/#{fqdn}/")
+      end
+
+      def save(replace=false)
+        if replace == true || replace == :replace
+          @dyn.put("#{resource_path}/#{@zone}/#{@fqdn}/", self)
+        else
+          @dyn.post("#{resource_path}/#{@zone}/#{@fqdn}/", self)
+        end
+        self
+      end
+
+      def delete
+        @dyn.delete("#{resource_path}/#{@zone}/#{@fqdn}/")
+      end
+
+      def to_json
+        {
+          "zone" => @zone,
+          "fqdn" => @fqdn,
+          "code" => @code,
+          "keep_uri" => @keep_uri,
+          "url" => @url
+        }.to_json
+      end
+    end
+  end
+end

data/lib/dyn/traffic/qps_report.rb

@@ -0,0 +1,30 @@
+module Dyn
+  module Traffic
+    class QPSReport
+      attr_accessor :csv
+
+      def initialize(dyn, csv)
+        @dyn = dyn
+        @csv = csv
+      end
+
+      def csv()
+        value ? (@csv = value; self) : @csv
+      end
+
+      def resource_path
+        "QPSReport"
+      end
+
+      def create(csv)
+        @dyn.create("#{resource_path}")
+      end
+
+      def to_json
+        {
+          "csv" => @csv,
+        }.to_json
+      end
+    end
+  end
+end

data/lib/dyn/traffic/resource.rb

@@ -0,0 +1,150 @@
+#
+# Author:: Sunny Gleason (<sunny@thesunnycloud.com>)
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2013 Dyn, Inc.
+# Copyright:: Copyright (c) 2010 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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 Dyn
+  module Traffic
+    class Resource
+      attr_accessor :fqdn, :record_type, :record_id, :ttl, :zone, :rdata
+
+      def initialize(dyn, zone, record_type, fqdn=nil, record_id=nil, ttl=nil, rdata={})
+        @dyn = dyn
+        @fqdn = fqdn
+        @record_type = record_type
+        @record_id = record_id
+        @ttl = ttl
+        @zone = zone
+        @rdata = rdata
+      end
+
+      def [](rdata_key)
+        @rdata[rdata_key]
+      end
+
+      def []=(rdata_key, rdata_value)
+        @rdata[rdata_key] = rdata_value
+      end
+
+      def fqdn(value=nil)
+        value ? (@fqdn = value; self) : @fqdn
+      end
+
+      def record_id(value=nil)
+        value ? (@record_id = value; self) : @record_id
+      end
+
+      def ttl(value=nil)
+        value ? (@ttl = value; self) : @ttl
+      end
+
+      def resource_path(full=false)
+        @record_type << "Record" unless @record_type[-6,6] == "Record"
+        if (full == true || full == :full)
+          "/REST/#{@record_type}/#{@zone}"
+        else
+          "#{@record_type}/#{@zone}"
+        end
+      end
+
+      def get(fqdn = nil, record_id=nil)
+        if record_id && fqdn
+          raw_rr = @dynect.get("#{resource_path}/#{fqdn}/#{record_id}")
+          Dyn::Traffic::Resource.new(@dyn,
+                                   raw_rr["record_type"],
+                                   raw_rr["zone"],
+                                   raw_rr["fqdn"],
+                                   raw_rr["record_id"],
+                                   raw_rr["ttl"],
+                                   raw_rr["rdata"])
+        elsif fqdn
+          results = @dyn.get("#{resource_path}/#{fqdn}")
+          raw_rr_list = results.map do |record|
+            if (record =~ /^#{resource_path(:full)}\/#{Regexp.escape(fqdn)}\/(\d+)$/)
+              self.get(fqdn, $1)
+            else
+              record
+            end
+          end
+          case raw_rr_list.length
+          when 0
+            raise Dyn::Exceptions::RequestFailed, "Cannot find #{record_type} record for #{fqdn}"
+          when 1
+            raw_rr_list[0]
+          else
+            raw_rr_list
+          end
+        else
+          @dyn.get(resource_path)
+        end
+      end
+
+      def find(fqdn, query_hash)
+        results = []
+        @dyn.get(fqdn).each do |rr|
+          query_hash.each do |key, value|
+            results << rr if rr[key.to_s] == value
+          end
+        end
+        results
+      end
+
+      def save(replace=false)
+        if record_id
+          @dyn.put("#{resource_path}/#{@fqdn}/#{record_id}", self)
+        else
+          if replace == true || replace == :replace
+            @dyn.put("#{resource_path}/#{@fqdn}", self)
+          else
+            @dyn.post("#{resource_path}/#{@fqdn}", self)
+          end
+        end
+        self
+      end
+
+      def delete
+        url = if record_id
+                "#{resource_path}/#{fqdn}/#{record_id}"
+              else
+                "#{resource_path}/#{fqdn}"
+              end
+        @dyn.delete(url)
+      end
+
+      def to_json
+        {
+          "rdata" => @rdata,
+          "ttl"   => @ttl
+        }.to_json
+      end
+
+      def method_missing(method_symbol, *args, &block)
+        method_string = method_symbol.to_s
+        if (args.length > 0 && method_string !~ /=$/)
+          @rdata[method_string] = args.length == 1 ? args[0] : args
+          self
+        elsif @rdata.has_key?(method_string)
+          @rdata[method_string]
+        else
+          raise NoMethodError, "undefined method `#{method_symbol.to_s}' for #{self.class.to_s}"
+        end
+      end
+
+    end
+  end
+end