sr-scripts 0.1.13 → 0.1.14

data/bin/sr-update-dns

@@ -5,17 +5,9 @@ $:.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
 require 'fog'
 require 'sr-scripts'
 
-sdb = SrScripts::SimpleDB.get
 
-domain = "dns_info"
 dns_alias = ARGV[0] 
 ip_address = ARGV[1]
 
+SrScripts::HostsManager.new.update_dns(dns_alias, ip_address)
 
-
-if /^(\d{1,3}\.){3}\d{1,3}$/.match(ip_address) == nil
-	p "Usage: sr-update-dns hostname ip_address"
-	exit
-end
-
-sdb.put_attributes(domain, dns_alias, { "ip_address" => ip_address }, { :replace => "ip_address" })

data/bin/sr-update-hosts-file

@@ -5,16 +5,5 @@ $:.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
 require 'fog'
 require 'sr-scripts'
 
-sdb = SrScripts::SimpleDB.get
-
-rows = sdb.select("SELECT * FROM dns_info").body["Items"]
-
-if rows.length > 0
-	File.open('/etc/hosts', 'w') { |f|
-		f.puts "# THIS FILE MANAGED BY sr-update-hosts SCRIPT"
-		f.puts "127.0.0.1 localhost localhost.localdomain"
-		rows.each_pair { |item_name, values|
-			f.puts "#{values['ip_address']} #{item_name}"
-		}
-	}
-end
+manager = SrScripts::HostsManager.new
+manager.generate_file "/etc/hosts"

data/lib/sr-scripts/HostsManager.rb

@@ -0,0 +1,32 @@
+module SrScripts
+  class HostsManager
+    def generate_file(file_name)
+      sdb = SrScripts::SimpleDB.get
+
+      rows = sdb.select("SELECT * FROM dns_info").body["Items"]
+
+      if rows.length > 0
+        File.open(file_name, 'w') { |f|
+          f.puts "# THIS FILE MANAGED BY sr-update-hosts SCRIPT"
+          f.puts "127.0.0.1 localhost localhost.localdomain"
+          rows.each_pair { |item_name, values|
+            f.puts "#{values['ip_address']} #{item_name}"
+          }
+        }
+      end
+    end
+
+    def update_dns(host_name, ip_address)
+      sdb = SrScripts::SimpleDB.get
+      domain = "dns_info"
+
+      if /^(\d{1,3}\.){3}\d{1,3}$/.match(ip_address) == nil
+        p "Usage: sr-update-dns hostname ip_address"
+        exit
+      end
+
+      sdb.put_attributes(domain, host_name, { "ip_address" => ip_address }, { :replace => "ip_address" })
+
+    end
+  end
+end

data/lib/sr-scripts/version.rb

@@ -1,5 +1,5 @@
 module Sr
   module Scripts
-    VERSION = "0.1.13"
+    VERSION = "0.1.14"
   end
 end

data/lib/sr-scripts.rb

@@ -1,5 +1,8 @@
 require 'yaml'
 require 'logger'
+require 'fog'
+
+require 'sr-scripts/HostsManager'
 
 module SrScripts
   class ConfigFile

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 1
-  - 13
-  version: 0.1.13
+  - 14
+  version: 0.1.14
 platform: ruby
 authors: 
 - Davy Campano
@@ -62,6 +62,7 @@ files:
 - bin/sr-update-dns
 - bin/sr-update-hosts-file
 - lib/sr-scripts.rb
+- lib/sr-scripts/HostsManager.rb
 - lib/sr-scripts/version.rb
 - sr-scripts.gemspec
 has_rdoc: true