shnell 0.1.0 → 0.2.0

data/bin/shnell

@@ -2,7 +2,8 @@
 
 require 'rubygems'
 require 'commander/import'
-require File.join(File.dirname(__FILE__), '../lib', 'shnell')
+require 'filander'
+require File.join(File.dirname(__FILE__), '../lib/shnell')
 
 program :name, 'shnell'
 program :version, Shnell::VERSION
@@ -11,24 +12,66 @@ program :help, 'Author', 'Johannes Jörg Schmidt, TF <schmidt@netzmerk.com>'
 
 default_command :help
 
-SCRIPTS_ROOT = Pathname.new(Dir.pwd)
+SCRIPTS_ROOT    = Pathname.new(Dir.pwd)
+CONFIG_FILENAME = File.join(SCRIPTS_ROOT, "config.yml")
+
+def scripts(pattern = nil)
+  if pattern
+    exp = "**/*#{pattern}*.rb"
+  else
+    exp = '**/*.rb'
+  end
+  Dir[SCRIPTS_ROOT.join(exp)].select { |f| File.basename(f) != 'backup.rb' }.sort
+end
+
+def backup_scripts(pattern = nil)
+  if pattern
+    exp = "**/*#{pattern}*/backup.rb"
+  else
+    exp = '**/backup.rb'
+  end
+  Dir[SCRIPTS_ROOT.join(exp)].sort
+end
+
+def say_script(filename, message, quiet = false)
+  return if quiet
+  name = Pathname.new(filename).relative_path_from(SCRIPTS_ROOT)
+  message = "#{message} #{name.sub(/\.rb$/, '')}..."
+  message = $terminal.color(message, :bold) if defined? $terminal
+  say message
+end
 
 command :list do |c|
-  c.syntax = 'shnell list [pattern]'
-  c.description = 'List available scripts.'
+  c.syntax = 'shnell list [scrips|backups] [pattern]'
+  c.description = 'List available scripts and backups.'
+
+  c.example "List all available scripts and backups", "shnell list"
+  c.example "List all available backups", "shnell list backups"
+  c.example "List all available scripts", "shnell list scripts"
+  c.example 'List available scripts which include "ruby"', "shnell list scripts ruby"
 
   c.action do |args, options|
+    name = args.shift
     pattern = args.shift
-
-    if pattern
-      exp = "**/*#{pattern}*.rb"
+    if name == "scripts"
+      scripts(pattern).each do |f|
+        path = Pathname.new(f)
+        name = path.relative_path_from(SCRIPTS_ROOT)
+        puts name.sub(/\.rb$/, '')
+      end
+    elsif name == "backups"
+      backup_scripts(pattern).each do |f|
+        path = Pathname.new(f)
+        name = path.relative_path_from(SCRIPTS_ROOT)
+        puts name.sub(/\.rb$/, '')
+      end
     else
-      exp = '**/*.rb'
-    end
-    Dir[SCRIPTS_ROOT.join(exp)].sort.each do |f|
-      path = Pathname.new(f)
-      name = path.relative_path_from(SCRIPTS_ROOT)
-      puts name.sub(/\.rb$/, '')
+      all = scripts(pattern) + backup_scripts(pattern)
+      all.sort.each do |f|
+        path = Pathname.new(f)
+        name = path.relative_path_from(SCRIPTS_ROOT)
+        puts name.sub(/\.rb$/, '')
+      end
     end
   end
 end
@@ -42,12 +85,17 @@ command :run do |c|
   c.option '-q', '--quiet', 'Supress status output'
   c.option '-s', '--skip', 'Skip files that already exist'
 
+  c.example 'Run script "./basic/apache.rb"', "shnell run basic/apache"
+  c.example "Run all available scripts", "shnell run all"
+  c.example 'Run available scripts which include "ruby"', "shnell run all ruby"
+  c.example "Dry run all available scripts (make no changes)", "shnell run all --pretend"
+
   c.action do |args, options|
     $:.unshift SCRIPTS_ROOT
     script = args.shift
-    require 'filander'
     include Filander
 
+    Filander.quiet = options.quiet
     behavior = :ask
     behavior = :skip if options.skip
     behavior = :pretend if options.pretend
@@ -55,17 +103,46 @@ command :run do |c|
     Filander.behavior = behavior
 
     if script == "all"
-      Dir[File.join(SCRIPTS_ROOT, '**/*.rb')].sort.each do |file|
-        name = Pathname.new(file).relative_path_from(SCRIPTS_ROOT)
-        message = "Running #{name.sub(/\.rb$/, '')}..."
-        message = $terminal.color(message, :bold) if defined? $terminal
-        say message
+      pattern = args.shift
+
+      scripts(pattern).each do |file|
+        say_script file, "Running", options.quiet
         Filander.source_root = file.sub(/\.rb$/, '')
         require file
       end
-    else
+    elsif script
       Filander.source_root = File.join(SCRIPTS_ROOT, script)
       require script
+    else
+      say "No script set."
+    end
+  end
+end
+
+command :backup do |c|
+  c.syntax = 'shnell backup script'
+  c.description = 'Run a backup script.'
+
+  c.option '-q', '--quiet', 'Supress status output'
+
+  c.action do |args, options|
+    script = args.shift
+
+    require File.join(File.dirname(__FILE__), '../lib/backup')
+    include Backup
+
+    if script == "all"
+      pattern = args.shift
+
+      backup_scripts(pattern).each do |file|
+        say_script file, "Backup", options.quiet
+        require file
+      end
+    elsif script
+      require script
+    else
+      say "No backup script set."
     end
   end
 end
+

data/lib/backup.rb

@@ -0,0 +1,149 @@
+require 'net/ftp'
+require 'yaml'
+
+module Backup
+  include Filander
+
+  class BackupScript
+    class << self
+      def backup(name, &block)
+        backup = self.new(name)
+        backup.prepare
+        backup.instance_eval(&block)
+        backup.backup!
+        backup.finish
+      end
+
+      def restore(name, &block)
+        backup = self.new(name)
+        backup.instance_eval(&block)
+        backup.restore!
+        backup.finish
+      end
+    end
+
+    def initialize(service)
+      @service_root = service
+      @service_name = File.basename(service)
+      @database     = @service_name.gsub(/[-\.]/, '_')
+      @tempdir      = File.join("/tmp", @service_name)
+      @tarball      = "#{@tempdir}.tar.bz2"
+      @directories  = []
+      @databases    = []
+    end
+
+    def prepare
+      backup_cmd "mkdir -p #{@tempdir}"
+    end
+
+    def directory(filename)
+      report :directory, filename
+      @directories << filename
+    end
+
+    def database(database = @database)
+      report :database, database
+      @databases << database
+    end
+
+    def backup!
+      @directories.each do |filename|
+        filename = File.join(@service_root, filename)
+        backup_cmd "cp -r #{filename} #{@tempdir}/"
+      end
+      @databases.each do |database|
+        backup_cmd "mysqldump -u#{db_user} #{"-p#{db_password}" if db_password != ''} #{database} > #{File.join(@tempdir, "#{database}.sql")}"
+      end
+      backup_cmd "tar -C /tmp -cjf #{@tarball} #{@service_name}"
+      Net::FTP.open(ftp_host) do |ftp|
+        ftp.login ftp_user, ftp_password
+        ftp.putbinaryfile @tarball
+      end
+    end
+
+    def restore!
+      Net::FTP.open(ftp_host) do |ftp|
+        ftp.login ftp_user, ftp_password
+        ftp.getbinaryfile @tarball
+      end
+      backup_cmd "tar -C /tmp -xf #{@tarball}"
+      @databases.each do |database|
+        backup_cmd "mysql -u#{db_user} #{"-p#{db_password}" if db_password != ''} #{database} < #{File.join(@tempdir, "#{database}.sql")}"
+      end
+      @directories.each do |filename|
+        source = File.join(@tempdir, filename)
+        dest   = File.join(@service_root, filename)
+        backup_cmd "cp -r #{source} #{dest}"
+      end
+    end
+
+    def finish
+      backup_cmd "rm -rf #{@tempdir}"
+      backup_cmd "rm #{@tarball}"
+    end
+
+    private
+
+    def backup_cmd(command)
+      system command
+    end
+
+    def db_user
+      read_config(:db, :user)
+    end
+
+    def db_password
+      read_config(:db, :password)
+    end
+
+    def ftp_host
+      read_config(:ftp, :host)
+    end
+
+    def ftp_user
+      read_config(:ftp, :user)
+    end
+
+    def ftp_password
+      read_config(:ftp, :password)
+    end
+
+    def read_config(scope, key)
+      hash = config[scope.to_s] || {}
+      value = hash[key.to_s]
+      if value.nil?
+        question = "Give me the #{scope} #{key}:"
+        value = if key.to_s =~ /password/
+                  ask(question) { |q| q.echo = "*" }
+                else
+                  ask(question)
+                end
+        write_config scope, key, value
+      end
+      value
+    end
+
+    def write_config(scope, key, value)
+      hash = config
+      hash[scope.to_s] ||= {}
+      hash[scope.to_s].update key.to_s => value
+      File.open(CONFIG_FILENAME, "w") do |file|
+        file << YAML.dump(hash)
+      end
+    end
+
+    def config
+      content = File.read(CONFIG_FILENAME) rescue nil
+      return {} unless content
+      YAML.load(content)
+    end
+  end
+
+  def backup(filename, &block)
+    BackupScript.backup filename, &block
+  end
+
+  def restore(filename, &block)
+    BackupScript.restore filename, &block
+  end
+end

data/lib/shnell.rb

@@ -1,3 +1,3 @@
 module Shnell
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

metadata

@@ -4,9 +4,9 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 0
-  - 1
+  - 2
   - 0
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Johannes J. Schmidt
@@ -40,9 +40,9 @@ dependencies:
       - !ruby/object:Gem::Version 
         segments: 
         - 0
-        - 4
+        - 5
         - 0
-        version: 0.4.0
+        version: 0.5.0
   type: :runtime
   version_requirements: *id002
 description: Ease Server Setup and Configuration
@@ -55,13 +55,11 @@ extra_rdoc_files:
 - LICENSE
 - README.rdoc
 files: 
-- .gitignore
-- LICENSE
-- README.rdoc
-- Rakefile
 - bin/shnell
+- lib/backup.rb
 - lib/shnell.rb
-- shnell.gemspec
+- LICENSE
+- README.rdoc
 has_rdoc: true
 homepage: http://github.com/jo/shnell
 licenses: []

data/.gitignore

@@ -1,21 +0,0 @@
-## MAC OS
-.DS_Store
-
-## TEXTMATE
-*.tmproj
-tmtags
-
-## EMACS
-*~
-\#*
-.\#*
-
-## VIM
-*.swp
-
-## PROJECT::GENERAL
-coverage
-rdoc
-pkg
-
-## PROJECT::SPECIFIC

data/Rakefile

@@ -1,33 +0,0 @@
-require 'rubygems'
-require 'rake'
-require File.join(File.dirname(__FILE__), 'lib', 'shnell')
-
-begin
-  require 'jeweler'
-  Jeweler::Tasks.new do |gem|
-    gem.name = "shnell"
-    gem.version = Shnell::VERSION
-    gem.summary = %Q{Server Setup and Configuration tools}
-    gem.description = %Q{Ease Server Setup and Configuration}
-    gem.email = "schmidt@netzmerk.com"
-    gem.homepage = "http://github.com/jo/shnell"
-    gem.authors = ["Johannes J. Schmidt"]
-    gem.add_dependency "commander", ">= 4.0.2"
-    gem.add_dependency "filander", ">= 0.4.0"
-  end
-  Jeweler::GemcutterTasks.new
-rescue LoadError
-  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
-end
-
-task :default => :test
-
-require 'rake/rdoctask'
-Rake::RDocTask.new do |rdoc|
-  version = File.exist?('VERSION') ? File.read('VERSION') : ""
-
-  rdoc.rdoc_dir = 'rdoc'
-  rdoc.title = "shnell #{version}"
-  rdoc.rdoc_files.include('README*')
-  rdoc.rdoc_files.include('lib/**/*.rb')
-end

data/shnell.gemspec

@@ -1,52 +0,0 @@
-# Generated by jeweler
-# DO NOT EDIT THIS FILE DIRECTLY
-# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
-# -*- encoding: utf-8 -*-
-
-Gem::Specification.new do |s|
-  s.name = %q{shnell}
-  s.version = "0.1.0"
-
-  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
-  s.authors = ["Johannes J. Schmidt"]
-  s.date = %q{2010-04-08}
-  s.default_executable = %q{shnell}
-  s.description = %q{Ease Server Setup and Configuration}
-  s.email = %q{schmidt@netzmerk.com}
-  s.executables = ["shnell"]
-  s.extra_rdoc_files = [
-    "LICENSE",
-     "README.rdoc"
-  ]
-  s.files = [
-    ".gitignore",
-     "LICENSE",
-     "README.rdoc",
-     "Rakefile",
-     "bin/shnell",
-     "lib/shnell.rb",
-     "shnell.gemspec"
-  ]
-  s.homepage = %q{http://github.com/jo/shnell}
-  s.rdoc_options = ["--charset=UTF-8"]
-  s.require_paths = ["lib"]
-  s.rubygems_version = %q{1.3.6}
-  s.summary = %q{Server Setup and Configuration tools}
-
-  if s.respond_to? :specification_version then
-    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
-    s.specification_version = 3
-
-    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
-      s.add_runtime_dependency(%q<commander>, [">= 4.0.2"])
-      s.add_runtime_dependency(%q<filander>, [">= 0.4.0"])
-    else
-      s.add_dependency(%q<commander>, [">= 4.0.2"])
-      s.add_dependency(%q<filander>, [">= 0.4.0"])
-    end
-  else
-    s.add_dependency(%q<commander>, [">= 4.0.2"])
-    s.add_dependency(%q<filander>, [">= 0.4.0"])
-  end
-end
-