rig 0.4.3 → 0.4.4

data/.rvmrc

@@ -0,0 +1 @@
+rvm ruby-1.9.3-p125

data/Gemfile

@@ -6,3 +6,5 @@ gemspec
 gem 'fog'
 gem 'chef', '~> 0.10.8'
 gem 'net-ssh', '~> 2.1.4'
+gem 'mongodb'
+gem 'bson_ext'

data/lib/rig/command/database.rb

@@ -0,0 +1,43 @@
+require 'mongo'
+require 'awesome_print'
+
+module Rig
+  module Command
+    module Database
+      class List < Abstract
+        def execute
+          db = Rig::Connection.database
+
+          list = db.all.sort
+          print_table(%w{Database Size}, list)
+        end
+      end
+
+      class Create < Abstract
+        parameter "NAME", "name of the database"
+        parameter "[USER]", "user to add to database", :default => Rig::Connection.database.user
+        parameter "[PASS]", "password of user to add to database", :default => Rig::Connection.database.pass
+        def execute
+          db = Rig::Connection.database
+          db.create_database(name, user, pass) unless test?
+        end
+      end
+
+      class Destroy < Abstract
+        parameter "NAME", "name of the database"
+        def execute
+          db = Rig::Connection.database
+          db.destroy_database(name) unless test?
+        end
+      end
+
+      class Main < Abstract
+        subcommand "list", "list all databases", List
+        subcommand "create", "create a database", Create
+        subcommand "destroy", "destroy a database", Destroy
+        self.default_subcommand = "list"
+      end
+    end
+  end
+end
+Rig::Command::Main.subcommand "database", "manage mongodb or mysql databases", Rig::Command::Database::Main

data/lib/rig/command.rb

@@ -12,6 +12,7 @@ require 'rig/command/main'
 require 'rig/command/ssh'
 require 'rig/command/account'
 require 'rig/command/config'
+require 'rig/command/database'
 
 require 'rig/command/environment/main'
 require 'rig/command/environment/list'

data/lib/rig/model/connection.rb

@@ -20,6 +20,12 @@ module Rig
           Fog::DNS.new(Rig.account[:dns])
         end
       end
+
+      def database
+        @database ||= begin
+          Rig::Model::Database.new(Rig.account[:database])
+        end
+      end
     end
   end
 end

data/lib/rig/model/database/base.rb

@@ -0,0 +1,11 @@
+
+
+module Rig
+  module Model
+    module Database
+      class Base
+
+      end
+    end
+  end
+end

data/lib/rig/model/database/mongodb.rb

@@ -0,0 +1,70 @@
+require 'rubygems'
+require 'mongo'
+require 'bson'
+
+module Rig
+  module Model
+    module Database
+      class MongoDB
+
+        def initialize(opts)
+          @options = {
+              :host => "localhost",
+              :port => 27017,
+              :user => nil,
+              :pass => nil,
+              :admin_user => nil,
+              :admin_pass => nil
+          }.merge(opts)
+          connect
+        end
+
+        # accessors
+        def user; @options[:user]; end
+        def host; @options[:host]; end
+        def port; @options[:port]; end
+        def pass; @options[:pass]; end
+        def password; @options[:pass]; end
+        def admin_user; @options[:admin_user]; end
+        def admin_pass; @options[:admin_pass]; end
+        def admin_password; @options[:admin_pass]; end
+
+        def connect
+          @cxn ||= begin
+            Mongo::Connection.new(@options[:host], @options[:port])
+          end
+        end
+
+        def auth_admin
+          db = @cxn.db("admin")
+          db.authenticate(@options[:admin_user], @options[:admin_pass])
+        end
+
+        def all
+          begin
+            connect.database_info
+          rescue Mongo::OperationFailure => e
+            raise unless e.message =~ /need to login/
+            raise unless auth_admin
+            connect.database_info
+          rescue => e
+            puts "exception: <#{e.class}> #{e.message} at #{e.backtrace.first}"
+          end
+        end
+
+        def create_database(name, user, pass)
+          db = connect.db("admin")
+          raise "failed to authenticate" unless db.authenticate(@options[:admin_user], @options[:admin_pass])
+          newdb = connect.db(name)
+          newdb.add_user(user, pass)
+        end
+
+        def destroy_database(name)
+          db = connect.db("admin")
+          raise "failed to authenticate" unless db.authenticate(@options[:admin_user], @options[:admin_pass])
+          connect.drop_database(name)
+        end
+      end
+    end
+  end
+end

data/lib/rig/model/database/mysql.rb

@@ -0,0 +1,12 @@
+
+
+
+module Rig
+  module Model
+    module Database
+      class MySQL < Base
+
+      end
+    end
+  end
+end

data/lib/rig/model/database.rb

@@ -0,0 +1,23 @@
+require 'rig/model/database/base'
+require 'rig/model/database/mongodb'
+require 'rig/model/database/mysql'
+
+module Rig
+  module Model
+    module Database
+      class << self
+        def new(opts)
+          provider = opts.delete(:provider)
+          case provider.downcase
+            when 'mongodb'
+              Rig::Model::Database::MongoDB.new(opts)
+            when 'mysql'
+              Rig::Model::Database::MySQL.new(opts)
+            else
+              raise "unknown database #{provider.inspect}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rig/version.rb

@@ -3,7 +3,7 @@ unless defined?(Rig::Version)
     module Version
       MAJOR = 0
       MINOR = 4
-      TINY  = 3
+      TINY  = 4
       STRING = [MAJOR, MINOR, TINY].join('.')
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rig
 version: !ruby/object:Gem::Version
-  version: 0.4.3
+  version: 0.4.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-05-14 00:00:00.000000000 Z
+date: 2012-05-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: clamp
@@ -116,6 +116,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - .gitignore
+- .rvmrc
 - Gemfile
 - LICENSE
 - README.md
@@ -137,6 +138,7 @@ files:
 - lib/rig/command/balancer/main.rb
 - lib/rig/command/balancer/view.rb
 - lib/rig/command/config.rb
+- lib/rig/command/database.rb
 - lib/rig/command/dns/create.rb
 - lib/rig/command/dns/destroy.rb
 - lib/rig/command/dns/edit.rb
@@ -166,6 +168,10 @@ files:
 - lib/rig/model/account.rb
 - lib/rig/model/balancer.rb
 - lib/rig/model/connection.rb
+- lib/rig/model/database.rb
+- lib/rig/model/database/base.rb
+- lib/rig/model/database/mongodb.rb
+- lib/rig/model/database/mysql.rb
 - lib/rig/model/dns.rb
 - lib/rig/model/environment.rb
 - lib/rig/model/instance.rb