snapback 0.0.1 → 0.0.3

data/lib/snapback/dsl.rb

@@ -1,208 +0,0 @@
-require "colorize"
-require "open4"
-
-LOCKED_FILES = ['.', '..', 'lost+found']
-
-# DSL: on_rollback
-# Add a callback if a method (somewhere) fails
-def on_rollback method
-  Snapback::Transaction.instance.add_rollback_method method
-end
-
-def run description, command
-  puts "#{command}".yellow if $options[:verbose]
-  debug "#{description}".ljust(72)
-
-  begin
-    err = ""
-    status = Open4::popen4(command) do |pid, stdin, stdout, stderr|
-      err = stderr.read
-    end
-
-    if status != 0 then
-      raise err
-    end
-  rescue
-    show_failed
-    raise $!
-  end
-
-  show_ok
-end
-
-def check description, command
-  print description.ljust(72)
-  result = command.call
-  if result == true then
-    show_ok
-  else
-    show_no
-  end
-
-  result
-end
-
-def debug *args
-  print "#{args}"
-end
-
-def show_ok
-  puts "[#{"  OK  ".green}]"
-end
-
-def show_no
-  puts "[#{"  NO  ".red}]"
-end
-
-def show_failed
-  puts "[#{"FAILED".red}]"
-end
-
-def ask_int question, max
-  number = nil
-
-  while true
-    print "#{question}: "
-
-    number = $stdin.gets.chomp
-
-    if !Integer(number) then
-      puts "The value you entered is not a number."
-      puts ""
-      next
-    end
-
-    number = number.to_i
-
-    if number < 1 || number > max then
-      puts "Please enter a number between 1 and #{max}."
-      puts ""
-      next
-    end
-
-    break
-  end
-
-  number
-end
-
-def ask_string question
-  print "#{question}: "
-  $stdin.gets.chomp
-end
-
-
-# Move MySQL Files
-def move_mysql_files from_dir, to_dir
-  Dir.foreach(from_dir) do |filename|
-    if !LOCKED_FILES.include?(filename) then
-      run "Moving #{from_dir}/#{filename} to #{to_dir}",
-        "mv #{from_dir}/#{filename} #{to_dir}"
-    end
-  end
-end
-
-def get_mount_dir database
-  "#{$config['filesystem']['mount']}/#{$config['lvm']['prefix_database']}-#{database}"
-end
-
-# Flush table
-def exec_flush
-  check "Flushing database to file system with read lock", lambda {
-    $database.flush_lock
-  }
-
-  on_rollback lambda {
-    check "Removing read lock", lambda { $database.unlock }
-  }
-end
-
-# Mount the logical volume
-def exec_mount device, directory
-  # If it's already mounted, don't worry
-  mount_status = Open4::popen4("mountpoint -q #{directory}") do |pid, stdin, stdout, stderr|
-  end
-
-  if mount_status == 0 then
-    return
-  end
-
-  # Run command and set rollback
-  run "Mounting logical volume", 
-    "mount #{device} #{directory}"
-
-  on_rollback lambda {
-    run "Unmounting logical volume",
-      "umount #{directory}"
-  }
-end
-
-# Unmount
-def exec_unmount device, directory
-  # If it's not already mounted, don't worry
-  mount_status = Open4::popen4("mountpoint -q #{directory}") do |pid, stdin, stdout, stderr|
-  end
-
-  if mount_status != 0 then
-    return
-  end
-
-  # Run command and set rollback
-  run "Unmounting logical volume",
-    "umount #{directory}"
-
-  on_rollback lambda {
-    run "Mounting logical volume", 
-      "mount #{device} #{directory}"
-  }
-end
-
-# Symbolic-link
-def exec_link entry, exit
-  run "Linking symbolic directory",
-    "ln -s #{exit} #{entry}"
-
-  on_rollback lambda {
-    run "Unlinking symbolic directory",
-      "unlink #{entry}"
-  }
-end
-
-# Remove symbolic link
-def exec_unlink entry, exit
-  run "Unlinking symbolic directory",
-    "unlink #{entry}"
-
-  on_rollback lambda {
-    run "Linking symbolic directory",
-      "ln -s #{exit} #{entry}"
-  }
-end
-
-# Activate logical volume
-def exec_activate device
-  run "Activating logical volume",
-    "lvchange -ay #{device}"
-
-  on_rollback lambda {
-    run "Deactivate logical volume",
-      "lvchange -an #{device}"
-  }
-end
-
-# Activate logical volume
-def exec_deactivate device
-  run "Deactivate logical volume",
-    "lvchange -an #{device}"
-
-  on_rollback lambda {
-    run "Activating logical volume",
-      "lvchange -ay #{device}"
-  }
-end
-
-# Take ownership
-def exec_chown directory
-  run "Changing ownership",
-    "chown -R mysql:mysql #{directory}"
-end

data/lib/snapback/options.rb

@@ -1,85 +0,0 @@
-require 'optparse'
-
-module Snapback
-  class Options
-    def self.parse(args)
-      options = {}
-      
-      options[:command]  = nil
-      options[:database] = nil
-
-      options[:config]  = "#{File.expand_path('~')}/.snapback.yml"
-      options[:verbose] = false
-      options[:size]    = nil
-
-      opts = OptionParser.new do |opts|
-        opts.banner = "Usage: sudo #{$0} [command] [options]"
-
-        opts.separator "Create database snapshots using logical volume management (LVM)"
-        opts.separator ""
-        opts.separator "Commands:"
-        opts.separator "    install".ljust(37)                 + "Check the environment is sane and setup"
-        opts.separator "    create DBNAME -s SIZE".ljust(37)   + "Create a new snapback compliant database"
-        opts.separator "    snapshot DBNAME -s SIZE".ljust(37) + "Take a snapshot of the current position"
-        opts.separator "    commit DBNAME".ljust(37)           + "Commit the working copy"
-        opts.separator "    rollback DBNAME".ljust(37)         + "Rollback the working copy to the snapshot"
-        opts.separator "    drop DBNAME".ljust(37)             + "Drop an existing database"
-        opts.separator "    mount DBNAME".ljust(37)            + "Mount an existing database"
-        opts.separator "    unmount DBNAME".ljust(37)          + "Unmount an existing database"
-
-        opts.separator ""
-        opts.separator "Options:"
-
-        opts.on("-s", "--size SIZE",
-                "Allocate a size parameter (e.g.: 1M, 1G, 1T)") do |q|
-          options[:size] = q
-        end
-
-        opts.on("-v", "--verbose",
-                "Show detailed transaction information") do
-          options[:verbose] = true
-        end
-
-        opts.on("-h", "--help",
-                "Show this help screen") do
-          puts opts
-          exit
-        end
-
-        opts.on("-c", "--config",
-                "Configuration file") do |q|
-          options[:config] = q
-        end
-
-        opts.separator ""
-      end
-
-      opts.parse!(args)
-
-      case args[0]
-      when "install", "create", "snapshot", "commit", "rollback", "drop", "mount", "unmount"
-        options[:command] = args[0]
-      when nil
-        puts opts
-        exit
-      else
-        $stderr.puts "command '#{args[0]}' not recognised"
-        puts opts
-        exit
-      end
-
-      case options[:command]
-      when "create", "snapshot", "commit", "rollback", "drop", "mount", "unmount"
-        if args[1].nil? then
-          $stderr.puts "you must specify a database with command '#{options[:command]}'"
-          puts opts
-          exit
-        end
-
-        options[:database] = args[1]
-      end
-
-      options
-    end
-  end
-end