snapback 0.0.1 → 0.0.3

data/lib/snapback/app/install.rb

@@ -1,156 +0,0 @@
-require "lvm"
-require "singleton"
-
-module Snapback
-  module App
-    class Install
-      include Singleton
-
-      def go
-        puts ""
-        puts "Hi!"
-        puts ""
-        puts "I'm going to guide you through setting up Snapback."
-        puts "This script will check to see if your environment can run "
-        puts "Snapback and will setup configuration files for you."
-        puts ""
-
-        begin
-          run "Checking LVM is installed",
-            "lvm version"
-        rescue
-          raise "You do not have LVM installed on this computer. You cannot use snapback."
-        end
-
-        # Get information from LVM
-        lvm = LVM::LVM.new({:command => "sudo lvm", :version => "2.02.30"})
-        volume_groups = []
-        volume_group = nil
-
-        lvm.volume_groups.each do |vg|
-          volume_groups.push(vg)
-        end
-        puts ""
-
-        # Decide which Logical volume use to use
-        if volume_groups.size == 0 then
-          raise "You do not have any volume groups in LVM. Please setup LVM before using Snapback"
-        elsif volume_groups.size == 1 then
-          volume_group = volume_groups[0];
-          puts "You have one volume group named #{volume_group.name.green}."
-          puts "Snapback will use this logical volume."
-        else
-          puts "Here is a list of volume groups on your system:"
-
-          volume_groups.each do |vg|
-            puts "#{volume_groups.size}.\t#{vg.name}"
-          end
-
-          volume_group_number = ask_int "Which volume group would you like Snapback use", volume_groups.size
-          volume_group = volume_groups[volume_group_number - 1];
-        end
-
-        puts ""
-
-        # Check for mount directory
-        mysql_mount_dir = ask_string "Where do you want to mount your logical volumes [/mnt/mysql]"
-        puts ""
-
-        if mysql_mount_dir.empty? then
-          mysql_mount_dir = "/mnt/mysql"
-        end
-
-        if !check "Checking for directory #{mysql_mount_dir}", lambda {
-          File.directory? mysql_mount_dir
-        } then
-          run "Creating directory #{mysql_mount_dir}",
-            "mkdir -p #{mysql_mount_dir}"
-
-          on_rollback lambda {
-            run "Removing #{mysql_mount_dir} directory",
-              "rm -rf #{mysql_mount_dir}"
-          }
-        end
-
-        # MySQL connection
-
-        while true
-          puts ""
-          puts "Enter the crudentials to connect to MySQL"
-
-          $database = Snapback::Database.instance
-
-          $database.hostname = ask_string "MySQL hostname [localhost]"
-          if $database.hostname.empty? then
-            $database.hostname = "localhost"
-          end
-
-          $database.username = ask_string "MySQL username [root]"
-          if $database.username.empty? then
-            $database.username = "root"
-          end
-
-          $database.password = ask_string "MySQL password"
-
-          puts ""
-
-          begin
-            check "Connecting to MySQL database", lambda {
-              $database.connect
-              true
-            }
-          rescue
-            show_failed
-            next
-          end
-
-          break
-        end
-
-        if !check "Checking #{"innodb_file_per_table"} is #{"ON"}", lambda {
-          $database.innodb_file_per_table
-        } then
-          debug "Setting #{"innodb_file_per_table"} to #{"ON"}"
-          $database.set_innodb_file_per_table true
-
-          on_rollback lambda {
-            debug "Setting #{"innodb_file_per_table"} to #{"OFF"}"
-            $database.set_innodb_file_per_table false
-          }
-        end
-
-        $config = {
-          'lvm' => {
-              'volume_group'    => volume_group.name.to_s,
-              'prefix_database' => 'snapback-active',
-              'prefix_backup'   => 'snapback-backup'
-          },
-
-          'mysql' => {
-              'hostname'        => $database.hostname,
-              'username'        => $database.username,
-              'password'        => $database.password
-          },
-
-          'filesystem' => {
-              'mount'           => '/mnt/mysql'
-          }
-        }
-
-        File.open($options[:config], 'w+') { |f|
-          f.write($config.to_yaml)
-        }
-
-        on_rollback lambda {
-          File.unlink $options[:config]
-        }
-
-        puts ""
-        puts "Snapback is now installed and configured."
-        puts "To start using Snapback, run the following command: "
-        puts ""
-        puts "sudo snapback --help".yellow
-      end
-    end
-  end
-end

data/lib/snapback/app/mount.rb

@@ -1,62 +0,0 @@
-require "singleton"
-
-module Snapback
-  module App
-    class Mount
-      include Singleton
-
-      def go
-        volume_group_name = "#{$config['lvm']['volume_group']}"
-        logical_volume_name = "#{$config['lvm']['prefix_database']}-#{$options[:database]}"
-        mount_dir = get_mount_dir $options[:database]
-        mysql_data_dir = $database.get_data_dir
-
-        exec_flush
-
-        # Stop the MySQL Server
-        $database.server_stop
-
-        on_rollback lambda {
-          $database.server_start
-        }
-
-        # Create a new directory for where the MySQL database can be mounted
-        # mkdir /mnt/mysql/{dbName};
-
-        if !File.directory? mount_dir then
-          run "Make mount directory", 
-            "mkdir #{mount_dir}"
-
-          on_rollback lambda {
-            run "Removing mount directory",
-              "rmdir #{$mount_dir}"
-          }
-
-          run "Changing permissions of mount directory",
-            "chmod 0777 #{mount_dir}"
-        end
-        
-        # # Mount the new logical volume
-        # mount /dev/{vgName}/mysql-{dbName} /mnt/mysql/{dbName};
-        exec_mount "/dev/#{volume_group_name}/#{logical_volume_name}", mount_dir
-
-        # # Symbolic-link the MySQL data directory to the new logical volume
-        # ln -s /mnt/mysql/{dbName} {mysql-data-dir}/{dbName}/
-        exec_link "#{mysql_data_dir}/#{$options[:database]}", mount_dir
-
-        # # Change the permissions & ownership to MySQL 
-        # chown -R mysql:mysql {mysql-data-dir}/{dbName}/
-        # chown -R mysql:mysql /mnt/mysql/{dbName}/
-        exec_chown "#{mysql_data_dir}/#{$options[:database]}"
-        exec_chown mount_dir
-
-        # Stop the MySQL Server
-        $database.server_start
-
-        on_rollback lambda {
-          $database.server_stop
-        }
-      end
-    end
-  end
-end

data/lib/snapback/app/rollback.rb

@@ -1,63 +0,0 @@
-require "singleton"
-
-module Snapback
-  module App
-    class Rollback
-      include Singleton
-
-      def go
-        volume_group_name = "#{$config['lvm']['volume_group']}"
-        logical_volume_name = "#{$config['lvm']['prefix_database']}-#{$options[:database]}"
-        mount_dir = get_mount_dir $options[:database]
-        mysql_data_dir = $database.get_data_dir
-
-        # Flush the MySQL Logs
-        exec_flush
-
-        # Stop the MySQL Server
-        $database.server_stop
-
-        on_rollback lambda {
-          $database.server_start
-        }
-
-        # Remove the symbolic link
-        exec_unlink "#{mysql_data_dir}/#{$options[:database]}", mount_dir
-
-        # Unmount the logical volume
-        exec_unmount "/dev/#{volume_group_name}/#{logical_volume_name}", mount_dir
-
-        # Deactivate the logical volume
-        exec_deactivate "/dev/#{volume_group_name}/#{logical_volume_name}"
-
-        # Merge the old logical volume into the new one
-        # lvconvert --merge /dev/{vgName}/backup-{dbName}
-        run "Merging the snapshot into the live logical volume",
-          "lvconvert --merge /dev/#{volume_group_name}/#{$config['lvm']['prefix_backup']}-#{$options[:database]}"
-
-        # Active the master drive
-        # lvchange -ay /dev/{vgName}/mysql-{dbName}
-        exec_activate "/dev/#{volume_group_name}/#{logical_volume_name}"
-
-        # Mount the logical volume
-        exec_mount "/dev/#{volume_group_name}/#{logical_volume_name}", mount_dir
-
-        # Symbolic-link the MySQL data directory to the new logical volume
-        exec_link "#{mysql_data_dir}/#{$options[:database]}", mount_dir
-
-        # Change the permissions & ownership to MySQL 
-        # chown -R mysql:mysql {mysql-data-dir}/{dbName}/
-        # chown -R mysql:mysql /mnt/mysql/{dbName}/
-        exec_chown "#{mysql_data_dir}/#{$options[:database]}"
-        exec_chown "#{mount_dir}"
-
-        # Start the MySQL Server
-        $database.server_start
-
-        on_rollback lambda {
-          $database.server_stop
-        }
-      end
-    end
-  end
-end

data/lib/snapback/app/snapshot.rb

@@ -1,74 +0,0 @@
-require "singleton"
-
-module Snapback
-  module App
-    class Snapshot
-      include Singleton
-
-      def go
-        # Ensure we have a size parameter
-        if $options[:size].nil? then
-          raise "You must specify a size attribute. E.g.: -s 100M"
-        end
-
-        if !$database.db_exists?($options[:database]) then
-          raise "Database '#{$options[:database]}' does not exist"
-        end
-
-        volume_group_name = "#{$config['lvm']['volume_group']}"
-        logical_volume_name = "#{$config['lvm']['prefix_database']}-#{$options[:database]}"
-        mount_dir = get_mount_dir $options[:database]
-        mysql_data_dir = $database.get_data_dir
-
-        # Flush the MySQL Logs
-        exec_flush
-
-        # Stop the MySQL Server
-        $database.server_stop
-
-        on_rollback lambda {
-          $database.server_start
-        }
-
-        # Unlink
-        exec_unlink "#{mysql_data_dir}/#{$options[:database]}", mount_dir
-
-        # Unmount
-        exec_unmount "/dev/#{volume_group_name}/#{logical_volume_name}", mount_dir
-
-        # Deactivate
-        exec_deactivate "/dev/#{volume_group_name}/#{logical_volume_name}"
-        
-        # Branch the logical volume with a snapshot
-        run "Snapshot the logical volume",
-          "lvcreate -L #{$options[:size]} -s -n #{$config['lvm']['prefix_backup']}-#{$options[:database]} /dev/#{volume_group_name}/#{logical_volume_name}"
-
-        on_rollback lambda {
-          run "Remove the snapshot",
-            "lvremove /dev/#{volume_group_name}/#{$config['lvm']['prefix_backup']}-#{$options[:database]}"
-        }
-
-        # Active the master drive
-        exec_activate "/dev/#{volume_group_name}/#{logical_volume_name}"
-
-        # Mount the master drive
-        exec_mount "/dev/#{volume_group_name}/#{logical_volume_name}", mount_dir
-
-        # Symbolic-link the MySQL data directory to the new logical volume
-        # ln -s /mnt/mysql/{dbName} {mysql-data-dir}/{dbName}/
-        exec_link "#{mysql_data_dir}/#{$options[:database]}", mount_dir
-
-        # Change the permissions & ownership to MySQL
-        exec_chown "#{mysql_data_dir}/#{$options[:database]}"
-        exec_chown mount_dir
-
-        # Start the MySQL Server
-        $database.server_start
-
-        on_rollback lambda {
-          $database.server_stop
-        }
-      end
-    end
-  end
-end

data/lib/snapback/app/unmount.rb

@@ -1,39 +0,0 @@
-require "singleton"
-
-module Snapback
-  module App
-    class Unmount
-      include Singleton
-
-      def go
-        volume_group_name = "#{$config['lvm']['volume_group']}"
-        logical_volume_name = "#{$config['lvm']['prefix_database']}-#{$options[:database]}"
-        mount_dir = get_mount_dir $options[:database]
-        mysql_data_dir = $database.get_data_dir
-
-        # Flush
-        exec_flush
-
-        # Stop the MySQL Server
-        $database.server_stop
-
-        on_rollback lambda {
-          $database.server_start
-        }
-
-        # Unlink
-        exec_unlink "#{mysql_data_dir}/#{$options[:database]}", mount_dir
-
-        # Unmount
-        exec_unmount "/dev/#{volume_group_name}/#{logical_volume_name}", mount_dir
-
-        # Start the MySQL Server
-        $database.server_start
-
-        on_rollback lambda {
-          $database.server_stop
-        }
-      end
-    end
-  end
-end

data/lib/snapback/database.rb

@@ -1,102 +0,0 @@
-require 'singleton'
-require 'mysql'
-
-# ## Database functions ##
-
-module Snapback
-  class Database
-    include Singleton
-
-    attr_accessor :hostname, :username, :password
-
-    @connection = nil
-
-    @hostname = ""
-    @username = ""
-    @password = ""
-
-    def connect
-      @connection = Mysql.new @hostname, @username, @password
-    end
-
-    def flush_lock
-      begin
-        @connection.query "FLUSH TABLES WITH READ LOCK"
-      end
-
-      true
-    end
-
-    def unlock
-      begin
-        @connection.query "UNLOCK TABLES"
-      end
-
-      true
-    end
-
-    def get_data_dir
-      begin
-        rs = @connection.query "SHOW VARIABLES LIKE 'datadir'"
-        rs.each_hash do |row|
-          return row['Value'].gsub(/\/$/, '')
-        end
-      end
-    end
-
-    def innodb_file_per_table
-      begin
-        rs = @connection.query "SHOW VARIABLES LIKE 'innodb_file_per_table'"
-        rs.each_hash do |row|
-          return row['Value'] == "ON"
-        end
-      rescue
-        raise "MySQL Query failed"
-      end
-    end
-
-    def set_innodb_file_per_table on
-      begin
-        rs = @connection.prepare "SET GLOBAL innodb_file_per_table = ?"
-        rs.execute(on ? 'ON' : 'OFF')
-        return true
-      rescue
-        raise "MySQL Query failed"
-      end
-    end
-
-    def db_exists? database_name
-      sql = @connection.list_dbs database_name
-      sql.size == 1
-    end
-
-    def db_create database_name
-      @connection.query "CREATE DATABASE `#{Mysql.escape_string database_name}`"
-      true
-    end
-
-    def db_drop database_name
-      @connection.query "DROP DATABASE `#{Mysql.escape_string database_name}`"
-      true
-    end
-
-    def db_use database_name
-      @connection.select_db database_name
-      true
-    end
-
-    def server_stop
-      run "Stop MySQL server",
-        "service mysql stop"
-      true
-    end
-
-    def server_start
-      run "Start MySQL server and reconnect",
-        "service mysql start"
-
-      self.connect
-      true
-    end
-  end
-end