vzcdn 0.1.9 → 0.2.0

data/lib/zone_add.rb

@@ -1,36 +1,89 @@
 class ZoneAdd < Command
   def create_args
     # for level zone.Records.?
-    add_arg('name', 'Name of the node to which this record pertains', /[[:alnum:]]+/)
+    add_arg('name', 'Name of the node to which this record pertains', /[[:alnum:]]/)
+    add_arg('gname', 'Name of load balancing or failover group', /[[:alnum:]]+/)
     add_arg('rdata', 'Additional resource record data, such as IP Address', /.*/)
     add_arg('ttl', 'Time to live, in seconds', /\d+/)
+    add_arg('weight', 'Load balancing weight assigned to the record', /\d+/)
+    add_arg('is-primary', 'Indicates whether record is the primary server/domain to which traffic will be directed.', ["true", "false"])
   end
 
   def create_flows
-    add_flow_from_usage("<name> <rdata> <ttl>")
+    add_flow_from_usage("<gname> [<ttl>='300']", :add_group_cond)
+    add_flow_from_usage("<name> <rdata> <ttl>", :add_record_cond)
+    add_flow_from_usage("<weight> <rdata>", :add_group_record_cond, 'LoadBalancing')
+    add_flow_from_usage("<is-primary> <rdata>", :add_group_record_cond, 'Failover')
   end
 
-  def handle_level(level)
+  def add_group_record_cond(group_type='')
+    level = @input["level"]
+    levels = level.split('.')
+    if (levels.length == 4 && levels[1].start_with?(group_type) && levels[2] == 'Group')
+      nil
+    else
+      "invalid drill down level (#{level}) for adding record to a group"
+    end
+  end
+
+  def add_record_cond
+    level = @input["level"]
     levels = level.split('.')
-    if levels.length == 3 && levels[1] == 'Records'
-      rtype = levels[2]
-      struct = @input["struct"] 
-      struct << @record
-      StructurePrint.new("-").structure_print(struct, level)
+    if (levels[1] == 'Records' && levels.length == 3)
+      nil
     else
-      raise "Illegal or unimplemented add location"
+      "invalid drill down level (#{level}) for adding record"
     end
   end
 
+  def add_group_cond
+    level = @input["level"]
+    if (level == 'zone.LoadBalancingGroups' || level == 'zone.FailoverGroups')
+      nil
+    else
+      "invalid drill down level (#{level}) for adding load balancing group or failover group"
+    end
+  end
+
+  def add_at_level(struct, level, arghash)
+    if add_record_cond.nil?
+      element = { "Name" => arghash["name"], "Rdata" => arghash["rdata"], "TTL" => arghash["ttl"].to_i }
+    elsif  add_group_cond.nil?
+      element = { "Group" => { "A" => [ ], "AAAA" => [ ], "CNAME" => [ ] }, "GroupTypeId" => 3, "Name" => arghash["gname"], "TTL" => arghash["ttl"] }
+    elsif add_group_record_cond.nil?
+      puts "level = #{level}"
+      key = 
+        if arghash["weight"]
+          "LoadBalancingGroups"
+        elsif arghash["is-primary"]
+          "FailoverGroups"
+        end
+      groups = @zone_struct[key]
+      re = /(LoadBalancingGroups|FailoverGroups)\[([^\]]+)\]/
+      index = re.match(level)[2].to_i
+      index -= 1
+      element = {
+        "HealthCheck" => nil,
+      }
+      if arghash["weight"]
+        element["Weight"] = arghash["weight"].to_i
+      elsif arghash["is-primary"]
+        element["IsPrimary"] = (arghash["is-primary"] == "true")
+      end      
+      element["Rdata"] = arghash["rdata"]
+    else
+      raise "should not happen"
+    end
+    struct << element
+    StructurePrint.new("-").structure_print(struct, level)
+  end
+
   def execute(arghash, rest)
     struct = @input["struct"]
     level = @input["level"]
-    @zone_struct = @input["toplevel"]
     zone = @input["zone"]
-
-    @record = { "Name" => arghash["name"], "Rdata" => arghash["rdata"], "TTL" => arghash["ttl"].to_i }
-    handle_level(level)
-    zone.write_zone_file(@zone_struct)
-    
+    @zone_struct = @input["toplevel"]
+    add_at_level(struct, level, arghash)
+    zone.write_zone_file(@zone_struct, false)
   end
 end

data/lib/zone_delete.rb

@@ -0,0 +1,51 @@
+require_relative 'util'
+
+class ZoneDelete < Command
+  def create_args
+    add_arg('index','index of the item to be deleted.', /[[:digit:]]+/)
+  end
+
+  def create_flows
+    add_flow_from_usage("[<index>]", :verify_drill_down)
+  end
+
+  def verify_drill_down
+    @index = 0
+    @level = @input["level"]
+    if  @input["rest"].class == Array
+      @index = @input["rest"][0].to_i
+    end
+    dputs "in verify_drill_down, level=#{@level}, rest=#{@input["rest"]}"
+    if @level.nil? && @index == 0
+      return nil
+    elsif  @level.length > 0 && @index > 0
+      return nil
+    end
+    "invalid argument"
+  end
+
+  def execute(arghash, rest)
+    @index = arghash["index"].to_i
+    struct = @input["struct"]
+    if struct.nil?
+      zone = @input["zone"]
+      zone.delete
+    elsif struct.class == Array
+      if struct.length >= @index
+        struct.delete_at @index-1
+      else
+        raise "invalid index"
+      end
+      zone = @input["zone"]
+      zone.write_zone_file(@input["toplevel"], false)
+      StructurePrint.new("-").structure_print(struct, "")
+    else
+      raise "only elements from array can be deleted"
+    end
+  end
+
+
+
+
+  
+end

data/lib/zone_hc.rb

@@ -0,0 +1,78 @@
+class ZoneSetHealthCheck < Command
+  def create_args
+    add_arg('http-type', 'for HTTP or secure HTTP health checks', ['HTTP', 'HTTPs'])
+    add_arg('uri', 'URI to which health checks will be directed', /.*/)
+    add_arg('ip-version', 'IP version', "ip_versions")
+    add_arg('check-interval', 'number of seconds between health checks', /\d+/)
+    add_arg('fail-threshold', 'number of consecutive succeeds/fails for status change (1-10)', /([1-9]|10)/)
+    add_arg('email-address', 'email address to which notifications will be sent', /.*/)
+    add_arg('http-method', 'HTTP method for health check', 'http_methods')
+    add_arg('content-verification', 'text that will be used to verify health check', /.*/)
+    add_arg('reintegration-method', 'how and unhealthy server/domain will be integrated back', 'hc_rm')
+    add_arg('tcp-type', 'for TCP or TCP SSL health checks', ['TCP Open', 'TCP SSL'])
+    add_arg('ip-address', 'ip address to which TCP health checks will be directed (IPv4 or IPv6)', /[\d:\.]+/)
+    add_arg('port', 'ip port to which TCP health checks will be directed (1-65535)', /\d+/)
+  end
+
+  def create_flows
+    add_flow_from_usage("<http-type> <uri> <ip-version> <check-interval> <fail-threshold> <email-address> [<http-method>='GET'] [<content-verification>] [<reintegration-method>='Automatic']", :add_hc_cond)
+    add_flow_from_usage("<tcp-type> <ip-address> <port> <check-interval> <fail-threshold> <email-address> [<content-verification>] [<reintegration-method>='Automatic']", :add_hc_cond)
+  end
+
+  def add_hc_cond
+    struct = @input['struct']
+    if struct.has_key? 'HealthCheck'
+      nil
+    else
+      "invalid drill down level (#{level}) for setting health check"
+    end  
+  end
+
+  def execute(arghash, rest)
+    struct = @input['struct']
+    level = @input["level"]
+    zone = @input['zone']
+    @zone_struct = @input["toplevel"]
+    http_type = arghash['http-type']
+    if http_type
+      content_verification = arghash['content-verification']
+      if content_verification.nil?
+        content_verification = ''
+      end
+      hc = {
+        'CheckInterval' => arghash['check-interval'].to_i,
+        'CheckTypeId' => arghash['http-type'],
+        'ContentVerification' => content_verification,
+        'EmailNotificationAddress' => arghash['email-address'],
+        'FailedCheckThreshold' => arghash['fail-threshold'].to_i,
+        'HTTPMethodId' => arghash['http-method'],
+        'IPAddress' => '',
+        'IPVersion' => arghash['ip-version'],
+        'PortNumber' => nil,
+        'ReintegrationMethodId' => arghash['reintegration-method'],
+        'Status' => 0,
+        'StatusName' => '',
+        'Uri' => arghash['uri']
+      }
+    else
+      hc = {
+        'CheckInterval' => arghash['check-interval'].to_i,
+        'CheckTypeId' => arghash['tcp-type'],
+        'ContentVerification' => content_verification,
+        'EmailNotificationAddress' => arghash['email-address'],
+        'FailedCheckThreshold' => arghash['fail-threshold'].to_i,
+        'HTTPMethodId' => nil,
+        'IPAddress' => arghash['ip-address'],
+        'IPVersion' => nil,
+        'PortNumber' => arghash['port'].to_i,
+        'ReintegrationMethodId' => arghash['reintegration-method'],
+        'Status' => 0,
+        'StatusName' => '',
+        'Uri' => ''
+      }
+    end
+    struct['HealthCheck'] = hc
+    StructurePrint.new("-").structure_print(struct['HealthCheck'], level+'.HealthCheck')
+    zone.write_zone_file(@zone_struct)
+  end
+end

data/lib/zone_update.rb

@@ -0,0 +1,60 @@
+require_relative 'util'
+
+class ZoneUpdate < Command
+  def create_args
+    add_arg('name-value','of the form name:new-value, where name is the (abbreviated) name of field', /[[:alpha:]]+:.*/)
+  end
+
+  def create_flows
+    add_flow_from_usage("<name-value>...")
+  end
+
+  def boolean_value_of(s)
+    if s == 'true'
+      true
+    elsif s == 'false'
+      false
+    else
+      raise "illegal value of boolean:#{s}"
+    end
+  end
+
+  def update_struct(struct, key, value)
+    name = Util.match_key(key, struct)
+    if name.nil?
+      raise "no (unique) key matches" + key 
+    end
+    target_type = struct[name].class
+    case [target_type]
+    when [Fixnum]
+      value = value.to_i
+    when [TrueClass]
+      value = boolean_value_of(value)
+    when [FalseClass]
+      value = boolean_value_of(value)
+    when [String]
+    else
+      raise "cannot change item of type #{target_type}"
+    end
+    struct[name] = value
+    zone = @input["zone"]
+    zone.write_zone_file(@input["toplevel"], false)
+    StructurePrint.new("-").structure_print(struct, "")
+  end
+
+  def execute(arghash, rest)
+    nvarray = arghash["name-value"]
+    struct = @input["struct"]
+    if struct.class == Hash
+      nvarray.each { | nv |
+        name, value = Util.parse_line(nv,':')
+        update_struct(struct, name, value)
+      }
+    elsif struct.class == Array
+      raise "can only update single item" if struct.length > 1
+      update_struct(struct[0], name, value)
+    else
+      raise "can't update at this level"
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vzcdn
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.2.0
 platform: ruby
 authors:
 - Steve Preston
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-04-22 00:00:00.000000000 Z
+date: 2014-05-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rest-client
@@ -79,6 +79,7 @@ files:
 - lib/common.rb
 - lib/config_handler.rb
 - lib/config_reader.rb
+- lib/diff.rb
 - lib/mediaManagment.rb
 - lib/method.rb
 - lib/mkbind.sh
@@ -96,6 +97,9 @@ files:
 - lib/vzcdn/version.rb
 - lib/zone.rb
 - lib/zone_add.rb
+- lib/zone_delete.rb
+- lib/zone_hc.rb
+- lib/zone_update.rb
 - pkg/vzcdn-0.0.1.gem
 - test/test_all.rb
 - test/test_clui.rb