dreamy 0.5.1

data/lib/dreamy/commands/users.rb

@@ -0,0 +1,32 @@
+module Dreamy::Command
+  class Users < Base
+    
+    def index
+      users = @account.users
+      if users.empty?
+        display "No users on this account"
+      else
+        user_table = table do |t|
+          t.headings = 'Name', 'Account Type', 'Server', 'Disk used (MB)', 'Quota (MB)'
+          users.each { |u| t << [u.gecos, u.type, u.home, u.disk_used_mb, u.quota_mb]  }
+        end
+        display user_table
+      end
+    end
+    
+    def pw
+      users = @account.users(passwords=true)
+      if users.empty?
+        display "No users on this account"
+      else
+        user_table = table do |t|
+          t.headings = 'UserName', 'Password'
+          users.each { |u| t << [u.username, u.password]  }
+        end
+        display user_table
+        
+      end
+    end
+    
+  end
+end

data/lib/dreamy/core_extensions.rb

@@ -0,0 +1,9 @@
+class Hash
+  def to_param_array
+    p = Array.new
+    self.each do |k,v|
+      p << "#{k}=#{v}"
+    end
+    p
+  end
+end

data/lib/dreamy/dns.rb

@@ -0,0 +1,19 @@
+module Dreamy
+  class Dns
+    include EasyClassMaker
+
+    attributes  :account_id, :comment, :editable, :record, :type, :value, :zone
+
+    def self.new_from_xml(xml)
+      d = new
+      d.account_id  = (xml).at('account_id').innerHTML.to_i
+      d.comment     = (xml).at('comment').innerHTML
+      d.editable    = (xml).at('editable').innerHTML.to_i
+      d.record      = (xml).at('record').innerHTML
+      d.type        = (xml).at('type').innerHTML
+      d.value       = (xml).at('value').innerHTML
+      d.zone        = (xml).at('zone').innerHTML
+      d
+    end
+  end
+end

data/lib/dreamy/domain.rb

@@ -0,0 +1,38 @@
+module Dreamy
+  class Domain
+    include EasyClassMaker
+
+    attributes  :account_id, :domain, :fastcgi, :home, :hosting_type, :outside_url, :passenger,
+                :path, :php, :php_fcgid, :security, :type, :unique_ip, :user, :www_or_not, :xcache
+
+    def self.new_from_xml(xml)
+      d = new
+      
+      d.account_id    = (xml).at('account_id').innerHTML.to_i
+      d.domain        = (xml).at('domain').innerHTML
+      d.home          = (xml).at('home').innerHTML
+      d.type          = (xml).at('type').innerHTML
+      d.unique_ip     = (xml).at('unique_ip').innerHTML
+      
+      if d.type == 'http'
+        d.fastcgi       = (xml).at('fastcgi').innerHTML.to_i
+        d.hosting_type  = (xml).at('hosting_type').innerHTML
+        d.outside_url   = (xml).at('outside_url').innerHTML
+        d.passenger     = (xml).at('passenger').innerHTML.to_i
+        d.path          = (xml).at('path').innerHTML
+        d.php           = (xml).at('php').innerHTML
+        d.php_fcgid     = (xml).at('php_fcgid').innerHTML.to_i
+        d.security      = (xml).at('security').innerHTML.to_i
+        d.user          = (xml).at('user').innerHTML
+        d.www_or_not    = (xml).at('www_or_not').innerHTML
+        d.xcache        = (xml).at('xcache').innerHTML.to_i
+      end
+      
+      d
+    end
+    
+    def short_home
+      home.gsub(".Dreamy.com","")
+    end
+  end
+end

data/lib/dreamy/easy_class_maker.rb

@@ -0,0 +1,43 @@
+# This is pretty much just a macro for creating a class that allows
+# using a block to initialize stuff and to define getters and setters
+# really quickly.
+module Dreamy
+  module EasyClassMaker
+    
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+    
+    module ClassMethods
+      # creates the attributes class variable and creates each attribute's accessor methods
+      def attributes(*attrs)
+        @@attributes = attrs
+        @@attributes.each { |a| attr_accessor a }
+      end
+      
+      # read method for attributes class variable
+      def self.attributes; @@attributes end
+    end
+    
+    # allows for any class that includes this to use a block to initialize
+    # variables instead of assigning each one seperately
+    # 
+    # Example: 
+    # 
+    # instead of...
+    # 
+    # s = Status.new
+    # s.foo = 'thing'
+    # s.bar = 'another thing'
+    #
+    # you can ...
+    # 
+    # Status.new do |s|
+    #   s.foo = 'thing'
+    #   s.bar = 'another thing'
+    # end
+    def initialize
+      yield self if block_given?
+    end
+  end
+end

data/lib/dreamy/mysql/db.rb

@@ -0,0 +1,17 @@
+module Dreamy
+  class MysqlDb
+    include EasyClassMaker
+
+    attributes  :account_id, :name, :description, :home, :disk_usage_mb
+
+    def self.new_from_xml(xml)
+      db = new
+      db.account_id     = (xml).at('account_id').innerHTML.to_i
+      db.name           = (xml).at('db').innerHTML
+      db.description    = (xml).at('description').innerHTML
+      db.home           = (xml).at('home').innerHTML
+      db.disk_usage_mb  = (xml).at('disk_usage_mb').innerHTML.to_f
+      db
+    end
+  end
+end

data/lib/dreamy/mysql/host.rb

@@ -0,0 +1,15 @@
+module Dreamy
+  class MysqlHost
+    include EasyClassMaker
+
+    attributes  :account_id, :domain, :home
+
+    def self.new_from_xml(xml)
+      host = new
+      host.account_id = (xml).at('account_id').innerHTML.to_i
+      host.domain     = (xml).at('domain').innerHTML
+      host.home       = (xml).at('home').innerHTML
+      host
+    end
+  end
+end

data/lib/dreamy/mysql/user.rb

@@ -0,0 +1,26 @@
+module Dreamy
+  class MysqlUser
+    include EasyClassMaker
+
+    attributes  :account_id, :db, :home, :username, :host, :select, :insert,
+                :update, :delete, :create, :drop, :index, :alter
+
+    def self.new_from_xml(xml)
+      u = new
+      u.account_id = (xml).at('account_id').innerHTML.to_i
+      u.db         = (xml).at('db').innerHTML
+      u.home       = (xml).at('home').innerHTML
+      u.username   = (xml).at('username').innerHTML
+      u.host       = (xml).at('host').innerHTML
+      u.select     = (xml).at('select_priv').innerHTML
+      u.insert     = (xml).at('insert_priv').innerHTML
+      u.update     = (xml).at('update_priv').innerHTML
+      u.delete     = (xml).at('delete_priv').innerHTML
+      u.create     = (xml).at('create_priv').innerHTML
+      u.drop       = (xml).at('drop_priv').innerHTML
+      u.index      = (xml).at('index_priv').innerHTML
+      u.alter      = (xml).at('alter_priv').innerHTML
+      u
+    end
+  end
+end

data/lib/dreamy/private_server.rb

@@ -0,0 +1,53 @@
+module Dreamy
+  class PrivateServer
+    include EasyClassMaker
+
+    attributes  :account_id, :name, :type, :memory, :start_date
+
+    def self.new_from_xml(xml)
+      ps = new
+      ps.account_id   = (xml).at('account_id').innerHTML.to_i
+      ps.name         = (xml).at('ps').innerHTML
+      ps.type         = (xml).at('type').innerHTML
+      ps.memory       = (xml).at('memory_mb').innerHTML.to_i
+      ps.start_date   = (xml).at('start_date').innerHTML
+      ps
+    end
+
+    def self.settings_from_xml(xml)
+      settings = {}
+      (xml/:data).each do |d|
+        settings[(d).at('setting').innerHTML] = (d).at('value').innerHTML
+      end
+      settings
+    end
+    
+    def self.size_from_xml(xml)
+      size = {}
+      size["memory_mb"]       = (xml).at('memory_mb').innerHTML.to_i
+      size["monthly_cost"]    = (xml).at('monthly_cost').innerHTML.to_f
+      size["period_cost"]     = (xml).at('period_cost').innerHTML.to_f
+      size["period_seconds"]  = (xml).at('period_seconds').innerHTML.to_i
+      size["stamp"]           = (xml).at('stamp').innerHTML
+      size
+    end
+    
+    def self.usage_from_xml(xml)
+      usage = {}
+      usage["load"]       = (xml).at('load').innerHTML.to_f
+      usage["memory_mb"]  = (xml).at('memory_mb').innerHTML.to_i
+      usage["stamp"]      = (xml).at('stamp').innerHTML
+      usage
+    end
+    
+    def self.pending_from_xml(xml)
+      pending = {}
+      pending["account_id"] = (xml).at('account_id').innerHTML
+      pending["ip"]         = (xml).at('ip').innerHTML
+      pending["type"]       = (xml).at('type').innerHTML
+      pending["stamp"]      = (xml).at('stamp').innerHTML
+      pending
+    end
+
+  end
+end

data/lib/dreamy/subscriber.rb

@@ -0,0 +1,17 @@
+module Dreamy
+  class Subscriber
+    include EasyClassMaker
+
+    attributes  :email, :confirmed, :subscribe_date, :name, :num_bounces
+
+    def self.new_from_xml(xml)
+      s = new
+      s.email           = (xml).at('email').innerHTML
+      s.confirmed       = (xml).at('confirmed').innerHTML.to_i
+      s.subscribe_date  = (xml).at('subscribe_date').innerHTML
+      s.name            = (xml).at('name').innerHTML
+      s.num_bounces     = (xml).at('num_bounces').innerHTML.to_i
+      s
+    end
+  end
+end

data/lib/dreamy/user.rb

@@ -0,0 +1,21 @@
+module Dreamy
+  class User
+    include EasyClassMaker
+
+    attributes  :account_id, :disk_used_mb, :gecos, :home, :password, :quota_mb, :shell, :type, :username
+
+    def self.new_from_xml(xml)
+      u = new
+      u.account_id    = (xml).at('account_id').innerHTML.to_i
+      u.disk_used_mb  = (xml).at('disk_used_mb').innerHTML.to_f
+      u.gecos         = (xml).at('gecos').innerHTML
+      u.home          = (xml).at('home').innerHTML
+      u.password      = (xml).at('password').innerHTML
+      u.quota_mb      = (xml).at('quota_mb').innerHTML.to_i
+      u.shell         = (xml).at('shell').innerHTML
+      u.type          = (xml).at('type').innerHTML
+      u.username      = (xml).at('username').innerHTML
+      u
+    end
+  end
+end

data/test/announce_test.rb

@@ -0,0 +1,32 @@
+require File.dirname(__FILE__) + "/test_helper.rb"
+
+class DreamyAnnounceListTest < Test::Unit::TestCase
+
+  context "Creation" do
+    setup do
+      @xml = <<EOF
+<data>
+  <name>Super Announce</name>
+  <account_id>8675309</account_id>
+  <domain>anessalee.net</domain>
+  <listname>testlist</listname>
+  <max_bounces>5</max_bounces>
+  <num_subscribers>22</num_subscribers>
+  <start_date>2009-04-20</start_date>
+</data>
+EOF
+    end
+
+    should "create a new list of announce lists from xml" do
+      l = Dreamy::AnnounceList.new_from_xml(Hpricot.XML(@xml))
+      assert_equal "Super Announce", l.name
+      assert_equal 8675309, l.account_id
+      assert_equal "anessalee.net", l.domain
+      assert_equal "testlist", l.short_name
+      assert_equal 5, l.max_bounces
+      assert_equal 22, l.subscribers
+      assert_equal "2009-04-20", l.start_date
+    end
+  end
+  
+end

data/test/base_test.rb

@@ -0,0 +1,235 @@
+require File.dirname(__FILE__) + "/test_helper.rb"
+
+class DreamyBaseTest < Test::Unit::TestCase
+  
+  @@base = Dreamy::Base.new(CREDS["user"],CREDS["key"])
+  
+  context "Initialization" do
+
+    should "require email and API key" do
+      assert_raise(ArgumentError) { Dreamy::Base.new }
+    end
+    
+  end
+  
+  context "Domains" do
+
+    should "return array of domain objects" do
+      domains = @@base.domains
+      assert_kind_of Array, domains
+      assert_kind_of Dreamy::Domain, domains.first unless domains.empty?
+    end
+    
+  end
+  
+  context "Users" do
+
+    should "return array of user objects" do
+      users = @@base.users
+      assert_kind_of Array, users
+      assert_kind_of Dreamy::User, users.first unless users.empty?
+    end
+    
+    should "not include user passwords by default" do
+      assert_equal "********", @@base.users.first.password
+    end
+  end
+  
+  context "DNS" do
+
+    should "return array of DNS records" do
+      records = @@base.dns
+      assert_kind_of Array, records
+      assert_kind_of Dreamy::Dns, records.first unless records.empty?
+    end
+    
+    context "adding a record" do
+      setup { @record_count = @@base.dns.size }
+      
+      should "require 3 arguments" do
+        assert_raise(ArgumentError) { @@base.dns_add("first","second") }
+      end
+      
+      should "add record and return true with valid data" do
+        assert @@base.dns_add("test." + CREDS["domain"],"A","208.97.188.35")
+        assert_equal @record_count + 1, @@base.dns.size
+      end
+      
+      should "not add record and raise error with invalid data" do
+        assert_raise(Dreamy::ApiError) { @@base.dns_add("test." + CREDS["domain"],"B","208.97.188.35") }
+        assert_equal @record_count, @@base.dns.size
+      end
+      
+    end
+    
+    context "removing a record" do
+      
+      should "require 3 arguments" do
+        assert_raise(ArgumentError) { @@base.dns_remove("first","second") }
+      end
+      
+      should "remove record and return true with valid data" do
+        record_count = @@base.dns.size
+        assert @@base.dns_remove("test." + CREDS["domain"],"A","208.97.188.35")
+        assert_equal record_count - 1, @@base.dns.size
+      end
+      
+      should "not remove record and raise error with invalid data" do
+        record_count = @@base.dns.size
+        assert_raise(Dreamy::ApiError) { @@base.dns_remove("test." + CREDS["domain"],"B","208.97.188.35") }
+        assert_equal record_count, @@base.dns.size
+      end
+      
+    end
+    
+  end
+  
+  context "Announcement lists" do
+    
+    context "listing lists" do
+      
+      should "return array of AnnounceList records" do
+        lists = @@base.announce_lists
+        assert_kind_of Array, lists
+        assert_kind_of Dreamy::AnnounceList, lists.first unless lists.empty?
+      end
+      
+    end
+    
+    context "subscribers" do
+      
+      should "require values for listname and domain" do
+        assert_raise(ArgumentError) { @@base.announce_list() }
+      end
+      
+      should "return array of Subscriber records" do
+        subscribers = @@base.announce_list(CREDS["listname"],CREDS["domain"])
+        assert_kind_of Array, subscribers
+        assert_kind_of Dreamy::Subscriber, subscribers.first unless subscribers.empty?
+      end
+      
+    end
+    
+    context "adding a subscriber" do
+      
+      should "require values for listname, domain, and email" do
+        assert_raise(ArgumentError) { @@base.announce_add() }
+      end
+      
+      # should "return true on success" do
+      #         assert @@base.announce_add(CREDS["listname"],CREDS["domain"],"new_guy@test.com")
+      #       end
+      
+    end
+    
+    context "removing a subscriber" do
+      
+      should "require values for listname, domain, and email" do
+        assert_raise(ArgumentError) { @@base.announce_remove() }
+      end
+      
+      # should "return true on success" do
+      #         assert @@base.announce_remove(CREDS["listname"],CREDS["domain"],"new_guy@test.com")
+      #       end
+      
+    end
+    
+  end # announcement lists
+  
+  context "MySQL" do
+    
+    should "return array of MysqlDb records" do
+      dbs = @@base.mysql_dbs
+      assert_kind_of Array, dbs
+      assert_kind_of Dreamy::MysqlDb, dbs.first unless dbs.empty?
+    end
+
+    should "return array of MysqlHost records" do
+      hosts = @@base.mysql_hosts
+      assert_kind_of Array, hosts
+      assert_kind_of Dreamy::MysqlDb, hosts.first unless hosts.empty?
+    end
+    
+    should "return array of MysqlUser records" do
+      users = @@base.mysql_users
+      assert_kind_of Array, users
+      assert_kind_of Dreamy::MysqlUser, users.first unless users.empty?
+    end
+    
+  end
+  
+  context "Private Servers" do
+    setup { @ps = CREDS["ps"] }
+    
+    should "return an array of PrivateServer objects" do
+      ps = @@base.ps
+      assert_kind_of Array, ps
+      assert_kind_of Dreamy::PrivateServer, ps.first unless ps.empty?
+    end
+    
+    context "settings" do
+      
+      should "return a hash of settings" do
+        settings = @@base.ps_settings(@ps)
+        assert_kind_of Hash, settings
+      end
+      
+      should "require values for ps, setting, and value when setting value" do
+        assert_raise(ArgumentError) { @@base.ps_set() }        
+      end
+      
+      should "set specified setting" do
+        @@base.ps_set(@ps,'comment','testps')
+        settings = @@base.ps_settings(@ps)
+        assert_equal 'testps', settings['comment']
+      end
+      
+    end
+    
+    context "Size" do
+      
+      should "return an array of size history hashes" do
+        sizes = @@base.ps_size_history(@ps)
+        assert_kind_of Array, sizes
+        assert_kind_of Hash, sizes.first
+      end
+      
+      should "set new size" do
+        @@base.ps_size_set(@ps,"200")
+        assert_equal 200, @@base.ps_size_history(@ps).last["memory_mb"]
+      end
+      
+    end
+    
+    context "Rebooting" do
+      
+      should "return an array of reboot time stamps" do
+        reboots = @@base.ps_reboot_history(@ps)
+        assert_kind_of Array, reboots
+      end
+      
+    end
+    
+    context "Usage" do
+      
+      should "return array of usage hashes" do
+        usage = @@base.ps_usage(@ps)
+        assert_kind_of Array, usage
+        assert_kind_of Hash, usage.first
+      end
+      
+    end
+    
+    context "Pending requests" do
+      
+      should "return array of pending hashes" do
+        pending = @@base.ps_pending
+        assert_kind_of Array, pending
+        assert_kind_of Hash, pending.first unless pending.empty?
+      end
+      
+    end
+    
+  end
+  
+end