snapback 0.0.1

data/lib/snapback/app/install.rb

@@ -0,0 +1,156 @@
+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

@@ -0,0 +1,62 @@
+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

@@ -0,0 +1,63 @@
+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

@@ -0,0 +1,74 @@
+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

@@ -0,0 +1,39 @@
+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

@@ -0,0 +1,102 @@
+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