mtgsy 0.0.1 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6eb2a4220e200d384fb285d46b9f7699fd7136ec
-  data.tar.gz: 98b95a7837173072dfd5ebdd89de11e4218e245d
+  metadata.gz: 5eb81b3ca3aa7b5260a4ac6ffdaa495443a50c07
+  data.tar.gz: 3c3bc8c0a0606530490ba26b4957f685d3edf2d6
 SHA512:
-  metadata.gz: c1a4f3bcc32b1af611d6b0abda76b08f6d78a6c6aa13aeb2b4ff1637fde0f858a1219a716f61d2c4d7bc3c1268def824058de6f236eb0f00b8c342ec8cfe7270
-  data.tar.gz: 08b33dd5265662da725e49b6e7fc4cc0c94173289140a5a4f37de7a1ed0008b873c7e4218fd0a959c537396e367813f933ecff3ba531f6273923a9ccd9db1a91
+  metadata.gz: ceec4d977f42565e3623d94045ee0414ea21240d15fbde873aadd9fd673785af1cb817436d8526f9ddf6e1243a9083bec77fcf2292309e131611a00048141381
+  data.tar.gz: 592b20fd9d189d2356419eb84b467154d691c2aacf9e9c5c8198a59043a24b02c99a3ec34916f70b0924e6b55b4b41d6a55544db8392e637826e1027c507bc87

data/lib/mtgsy/api_client.rb

@@ -6,15 +6,18 @@ module Mtgsy
     def initialize(options = {})
       @agent        = Mechanize.new
       @api_endpoint = "https://www.mtgsy.net/dns/api.php"
+      @record_types = [ 'A', 'AAAA', 'CNAME', 'HINFO', 'MX', 'NS', 'PTR', 'RP', 'SRV', 'TXT' ]
+
+      @debug        = options[:debug] ? options[:debug] : false
 
       # attr_accessor
       @domainname = options[:domainname] ? options[:domainname] : nil
       @username   = options[:username]   ? options[:username]   : nil
+      @ttl        = options[:ttl]        ? options[:ttl]        : 900
       @aux = 0
-      @ttl = 1800
 
       # attr_writer
-      @password   = options[:password]   ? options[:password]   : nil
+      @password     = options[:password]   ? options[:password]   : nil
     end
 
     attr_accessor :domainname, :username, :aux, :ttl
@@ -28,33 +31,109 @@ module Mtgsy
       aux     = options[:aux ] ? options[:aux].to_s  : @aux.to_s
       ttl     = options[:ttl ] ? options[:ttl].to_s  : @ttl.to_s
 
-      @agent.post(@api_endpoint, [
-        [ 'command'    ,command     ],
-        [ 'username'   ,@username   ],
-        [ 'password'   ,@password   ],
-        [ 'domainname' ,@domainname ],
-        [ 'name'       ,name        ],
-        [ 'type'       ,type        ],
-        [ 'data'       ,data        ],
-        [ 'aux'        ,aux         ],
-        [ 'ttl'        ,ttl         ]
-      ])
+      post_to_mtgsy([ command, name, type, data, aux, ttl ])
 
+      what_happened?(@agent, command)
     end
 
     def delete_record(options={})
       command = "deleterecord"
-      puts "TODO: Delete record"
+      name    = options[:name] ? options[:name].to_s : (raise "delete_record requires :name")
+      type    = options[:type] ? options[:type].to_s : nil
+      data    = options[:data] ? options[:data].to_s : nil
+      aux     = options[:aux ] ? options[:aux].to_s  : nil
+      ttl     = options[:ttl ] ? options[:ttl].to_s  : nil
+
+      post_to_mtgsy([ command, name, type, data, aux, ttl ])
+
+      what_happened?(@agent, command)
     end
 
     def update_record(options={})
       command = "updaterecord"
-      puts "TODO: Update record"
+      name    = options[:name] ? options[:name].to_s : (raise "update_record requires :name")
+      type    = options[:type] ? options[:type].to_s : nil
+      data    = options[:data] ? options[:data].to_s : nil
+      aux     = options[:aux ] ? options[:aux].to_s  : nil
+      ttl     = options[:ttl ] ? options[:ttl].to_s  : nil
+
+      post_to_mtgsy([ command, name, type, data, aux, ttl ])
+
+      what_happened?(@agent, command)
+    end
+
+    def refresh!
+      command = "listrecords"
+      post_to_mtgsy([ command ])
+
+      @records_ALL = @agent.page.body.split('<br>')
+      @records_ALL.pop
+      @records_ALL.collect! { |i| i.split(",") }
+
+
+      @record_types.each do |x|
+        tmp_placeholder = []
+        @records_ALL.each do |record|
+          tmp_placeholder << record if record[RECORD_TYPE] == x
+        end
+        instance_variable_set("@records_#{x}", tmp_placeholder)
+      end
+    end
+
+    def records(type="ALL")
+      self.refresh! unless @records_ALL
+
+      eval("@records_#{type}")
+    end
+
+    def record_types?
+      @record_types
     end
 
     private
-      def what_happened?(mechanize_instance)
-        mechanize_instance.page.code
+      def post_to_mtgsy(params)
+        command = params[Mtgsy::POST_COMMAND]
+        name    = params[Mtgsy::POST_NAME]
+        type    = params[Mtgsy::POST_TYPE]
+        data    = params[Mtgsy::POST_DATA]
+        aux     = params[Mtgsy::POST_AUX]
+        data    = params[Mtgsy::POST_TTL]
+
+        post_data = []
+        post_data << [ 'command'    ,command ]     if command
+        post_data << [ 'username'   ,@username ]   if @username
+        post_data << [ 'password'   ,@password ]   if @password
+        post_data << [ 'domainname' ,@domainname ] if @domainname
+        post_data << [ 'name'       ,name ]        if name
+        post_data << [ 'type'       ,type ]        if type
+        post_data << [ 'data'       ,data ]        if data
+        post_data << [ 'aux'        ,aux ]         if aux
+        post_data << [ 'ttl'        ,ttl ]         if ttl
+
+        p post_data if @debug
+
+        @agent.post(@api_endpoint, post_data)
+      end
+
+      def what_happened?(mechanize_instance, command)
+        mtgsy_return_code = mechanize_instance.page.body.match(/[0-9]+/)[0].to_i
+
+        case mtgsy_return_code
+        when 800
+            puts "OK"
+        when 400
+          $stderr.puts "No domain name specified"
+        when 100
+          $stderr.puts "Balance insufficcient"
+        when 300
+          $stderr.puts "Invalid login information supplied"
+        when 305
+          $stderr.puts "Domain not found"
+        when 310
+          $stderr.puts "Record not found / problem adding record"
+        when 200
+          $stderr.puts "Insufficcient information supplied"
+        end
       end
   end
 end

data/lib/mtgsy/version.rb

@@ -1,3 +1,3 @@
 module Mtgsy
-  VERSION = "0.0.1"
+  VERSION = "0.1.0"
 end

data/lib/mtgsy.rb

@@ -2,4 +2,18 @@ require "mtgsy/version"
 require "mtgsy/api_client"
 
 module Mtgsy
+  # e.g. post_to_mtgsy([ command, name, type, data, aux, ttl ])
+  POST_COMMAND = 0
+  POST_NAME    = 1
+  POST_TYPE    = 2
+  POST_DATA    = 3
+  POST_AUX     = 4
+  POST_TTL     = 5
+
+  # For the record types, e.g. ["vc32staging", "A", "50.56.9.191", "0", "123"]
+  RECORD_NAME = 0
+  RECORD_TYPE = 1
+  RECORD_DATA = 2
+  RECORD_AUX  = 3
+  RECORD_TTL  = 4
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mtgsy
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.1.0
 platform: ruby
 authors:
 - Jason Barnett