wordmove 0.0.8 → 0.1.0.alpha

data/lib/wordmove/cli.rb

@@ -1,7 +1,6 @@
 require 'thor'
 require 'wordmove/generators/movefile'
-require 'wordmove/deployer'
-
+require 'wordmove/deployer/base'
 
 module Wordmove
   class CLI < Thor
@@ -12,27 +11,37 @@ module Wordmove
     end
 
     desc "pull", "Pulls WP data from remote host to the local machine"
-    method_option :skip_db,       :aliases => "-d", :type => :boolean
-    method_option :skip_uploads,  :aliases => "-u", :type => :boolean
-    method_option :skip_themes,   :aliases => "-t", :type => :boolean
-    method_option :skip_plugins,  :aliases => "-p", :type => :boolean
-    method_option :verbose,       :aliases => "-v", :type => :boolean
-    method_option :config,        :aliases => "-c"
+    method_option :db,       :aliases => "-d", :type => :boolean
+    method_option :uploads,  :aliases => "-u", :type => :boolean
+    method_option :themes,   :aliases => "-t", :type => :boolean
+    method_option :plugins,  :aliases => "-p", :type => :boolean
+    method_option :verbose,  :aliases => "-v", :type => :boolean
+    method_option :simulate, :aliases => "-s", :type => :boolean
+    method_option :config,   :aliases => "-c"
     def pull
-      deployer = Wordmove::Deployer.new(options)
-      deployer.pull
+      deployer = Wordmove::Deployer::Base.deployer_for(options)
+      %w(db uploads themes plugins).map(&:to_sym).each do |task|
+        if options[task]
+          deployer.send("pull_#{task}")
+        end
+      end
     end
 
     desc "push", "Pushes WP data from local machine to remote host"
-    method_option :skip_db,       :aliases => "-d", :type => :boolean
-    method_option :skip_uploads,  :aliases => "-u", :type => :boolean
-    method_option :skip_themes,   :aliases => "-t", :type => :boolean
-    method_option :skip_plugins,  :aliases => "-p", :type => :boolean
-    method_option :verbose,       :aliases => "-v", :type => :boolean
-    method_option :config,        :aliases => "-c"
+    method_option :db,       :aliases => "-d", :type => :boolean
+    method_option :uploads,  :aliases => "-u", :type => :boolean
+    method_option :themes,   :aliases => "-t", :type => :boolean
+    method_option :plugins,  :aliases => "-p", :type => :boolean
+    method_option :verbose,  :aliases => "-v", :type => :boolean
+    method_option :simulate, :aliases => "-s", :type => :boolean
+    method_option :config,   :aliases => "-c"
     def push
-      deployer = Wordmove::Deployer.new(options)
-      deployer.push
+      deployer = Wordmove::Deployer::Base.deployer_for(options)
+      %w(db uploads themes plugins).map(&:to_sym).each do |task|
+        if options[task]
+          deployer.send("push_#{task}")
+        end
+      end
     end
 
   end

data/lib/wordmove/deployer/base.rb

@@ -0,0 +1,135 @@
+require 'active_support/core_ext'
+require 'wordmove/logger'
+require 'escape'
+
+module Wordmove
+  module Deployer
+
+    class Base
+      attr_reader :options
+      attr_reader :logger
+
+      def self.deployer_for(options)
+        options = fetch_movefile(options[:config]).merge(options)
+        recursive_symbolize_keys!(options)
+        if options[:remote][:ftp]
+          require 'wordmove/deployer/ftp'
+          FTP.new(options)
+        elsif options[:remote][:ssh]
+          require 'wordmove/deployer/ssh'
+          SSH.new(options)
+        else
+          raise Thor::Error, "No valid adapter found."
+        end
+      end
+
+      def self.fetch_movefile(path)
+        path ||= "Movefile"
+        unless File.exists?(path)
+          raise Thor::Error, "Could not find a valid Movefile"
+        end
+        YAML::load(File.open(path))
+      end
+
+      def initialize(options = {})
+        @options = options
+        @logger = Logger.new(STDOUT)
+        @logger.level = Logger::DEBUG
+      end
+
+      def push_db;
+        logger.task "Pushing Database"
+      end
+
+      def pull_db
+        logger.task "Pulling Database"
+      end
+
+      def remote_get_directory(directory); end
+      def remote_put_directory(directory); end
+
+      %w(uploads themes plugins).each do |task|
+        define_method "push_#{task}" do
+          logger.task "Pushing #{task.titleize}"
+          remote_put_directory(local_wpcontent_path(task), remote_wpcontent_path(task))
+        end
+
+        define_method "pull_#{task}" do
+          logger.task "Pulling #{task.titleize}"
+          remote_get_directory(remote_wpcontent_path(task), local_wpcontent_path(task))
+        end
+      end
+
+      protected
+
+      def run(command)
+        logger.task_step true, command
+        unless simulate?
+          system(command)
+        end
+      end
+
+      def download(url, local_path)
+        logger.task_step true, "download #{url}"
+        unless simulate?
+          open(local_path, 'w') do |file|
+            file << open(url).read
+          end
+        end
+      end
+
+      def simulate?
+        options[:simulate]
+      end
+
+      def local_wpcontent_path(*args)
+        File.join(options[:local][:wordpress_path], "wp-content", *args)
+      end
+
+      def remote_wpcontent_path(*args)
+        File.join(options[:remote][:wordpress_path], "wp-content", *args)
+      end
+
+      def remote_wpcontent_url(*args)
+        options[:remote][:vhost] + File.join("/wp-content", *args)
+      end
+
+      def adapt_sql(save_to_path, local, remote)
+        logger.task_step true, "adapt dump"
+        unless simulate?
+          File.open(save_to_path, 'a') do |file|
+            file.write "UPDATE wp_options SET option_value=\"#{remote[:vhost]}\" WHERE option_name=\"siteurl\" OR option_name=\"home\";\n"
+          end
+        end
+      end
+
+      def mysql_dump_command(options, save_to_path)
+        arguments = [ "mysqldump" ]
+        arguments << "--host=#{options[:host]}" if options[:host].present?
+        arguments << "--user=#{options[:user]}" if options[:user].present?
+        arguments << "--password=#{options[:password]}" if options[:password].present?
+        arguments << "--default-character-set=#{options[:charset]}" if options[:charset].present?
+        arguments << options[:name]
+        Escape.shell_command(arguments) + " > #{save_to_path}"
+      end
+
+      def mysql_import_command(dump_path, options)
+        arguments = [ "mysql" ]
+        arguments << "--host=#{options[:host]}" if options[:host].present?
+        arguments << "--user=#{options[:user]}" if options[:user].present?
+        arguments << "--password=#{options[:password]}" if options[:password].present?
+        arguments << "--database=#{options[:name]}"
+        Escape.shell_command(arguments) + " < #{dump_path}"
+      end
+
+      private
+
+      def self.recursive_symbolize_keys! hash
+        hash.symbolize_keys!
+        hash.values.select{|v| v.is_a? Hash}.each{|h| recursive_symbolize_keys!(h)}
+      end
+
+    end
+
+  end
+end

data/lib/wordmove/deployer/ftp.rb

@@ -0,0 +1,117 @@
+require 'wordmove/deployer/base'
+require 'photocopier/ftp'
+require 'erb'
+require 'open-uri'
+
+module Wordmove
+  module Deployer
+    class FTP < Base
+
+      def initialize(options)
+        super
+        ftp_options = options[:remote][:ftp]
+        @copier = Photocopier::FTP.new(ftp_options.merge(logger: logger))
+      end
+
+      def push_db
+        super
+
+        remote_import_script_path = remote_wpcontent_path("import.php")
+        local_dump_path = local_wpcontent_path("dump.sql")
+        remote_dump_path = remote_wpcontent_path("dump.sql")
+
+        # dump local mysql into file
+        run mysql_dump_command(options[:local][:database], local_dump_path)
+        # gsub sql
+        adapt_sql(local_dump_path, options[:local], options[:remote])
+        # upload it
+        remote_put(local_dump_path, remote_dump_path)
+
+        # generate a secure one-time password
+        one_time_password = SecureRandom.hex(40)
+        # generate import script
+        import_script = generate_import_script(options[:remote][:database], one_time_password)
+        # upload import script
+        remote_put(import_script, remote_import_script_path)
+        # run import script
+        import_url = "#{remote_wpcontent_url("import.php")}?shared_key=#{one_time_password}&start=1&foffset=0&totalqueries=0&fn=dump.sql"
+        download(import_url, local_dump_path + "_")
+
+        # remove script remotely
+        remote_delete(remote_import_script_path)
+        # remove dump remotely
+        remote_delete(remote_dump_path)
+        # and locally
+        run "rm #{local_dump_path}"
+      end
+
+      def pull_db
+        super
+
+        remote_dump_script = remote_wpcontent_path("dump.php")
+        local_dump_path = local_wpcontent_path("dump.sql")
+
+        # generate a secure one-time password
+        one_time_password = SecureRandom.hex(40)
+
+        # generate dump script
+        dump_script = generate_dump_script(options[:remote][:database], one_time_password)
+        # upload the dump script
+        remote_put(dump_script, remote_dump_script)
+        # download the resulting dump (using the password)
+        dump_url = "#{remote_wpcontent_url("dump.php")}?shared_key=#{one_time_password}"
+        download(dump_url, local_dump_path)
+
+        # gsub sql
+        adapt_sql(local_dump_path, options[:remote], options[:local])
+        # import locally
+        run mysql_import_command(local_dump_path, options[:local][:database])
+
+        # remove it remotely
+        remote_delete(remote_dump_script)
+        # and locally
+        run "rm #{local_dump_path}"
+      end
+
+      private
+
+      %w(get get_directory put_directory delete).each do |command|
+        define_method "remote_#{command}" do |*args|
+          logger.task_step false, "#{command}: #{args.join(" ")}"
+          unless simulate?
+            @copier.send(command, *args)
+          end
+        end
+      end
+
+      def remote_put(thing, path)
+        if File.exists?(thing)
+          logger.task_step false, "copying #{thing} to #{path}"
+        else
+          logger.task_step false, "write #{path}"
+        end
+        unless simulate?
+          @copier.put(thing, path)
+        end
+      end
+
+      def escape_php(string)
+        # replaces \ with \\
+        # replaces ' with \'
+        string.gsub('\\','\\\\\\').gsub(/[']/, '\\\\\'')
+      end
+
+      def generate_dump_script(db, password)
+        template = ERB.new File.read(File.join(File.dirname(__FILE__), "../assets/dump.php.erb"))
+        template.result(binding)
+      end
+
+      def generate_import_script(db, password)
+        template = ERB.new File.read(File.join(File.dirname(__FILE__), "../assets/import.php.erb"))
+        template.result(binding)
+      end
+
+    end
+  end
+end
+

data/lib/wordmove/deployer/ssh.rb

@@ -0,0 +1,73 @@
+require 'wordmove/deployer/base'
+require 'photocopier/ssh'
+
+module Wordmove
+  module Deployer
+    class SSH < Base
+      def initialize(options)
+        super
+        ssh_options = options[:remote][:ssh]
+        @copier = Photocopier::SSH.new(ssh_options.merge(logger: logger))
+      end
+
+      def push_db
+        super
+
+        local_dump_path = local_wpcontent_path("dump.sql")
+        remote_dump_path = remote_wpcontent_path("dump.sql")
+
+        # dump local mysql into file
+        run mysql_dump_command(options[:local][:database], local_dump_path)
+        # gsub sql
+        adapt_sql(local_dump_path, options[:local], options[:remote])
+        # upload it
+        remote_put(local_dump_path, remote_dump_path)
+        # import it remotely
+        remote_run mysql_import_command(remote_dump_path, options[:remote][:database])
+        # remove it remotely
+        remote_delete(remote_dump_path)
+        # and locally
+        run "rm #{local_dump_path}"
+      end
+
+      def pull_db
+        super
+
+        local_dump_path = local_wpcontent_path("dump.sql")
+        remote_dump_path = remote_wpcontent_path("dump.sql")
+
+        # dump remote db into file
+        remote_run mysql_dump_command(options[:remote][:database], remote_dump_path)
+        # download remote dump
+        remote_get(remote_dump_path, local_dump_path)
+        # gsub sql
+        adapt_sql(local_dump_path, options[:remote], options[:local])
+        # import locally
+        run mysql_import_command(local_dump_path, options[:local][:database])
+        # remove it remotely
+        remote_delete(remote_dump_path)
+        # and locally
+        run "rm #{local_dump_path}"
+      end
+
+      private
+
+      %w(get put get_directory put_directory delete).each do |command|
+        define_method "remote_#{command}" do |*args|
+          logger.task_step false, "#{command}: #{args.join(" ")}"
+          unless simulate?
+            @copier.send(command, *args)
+          end
+        end
+      end
+
+      def remote_run(command)
+        logger.task_step false, command
+        unless simulate?
+          @copier.session.exec! command
+        end
+      end
+
+    end
+  end
+end

data/lib/wordmove/logger.rb

@@ -1,31 +1,23 @@
+# encoding: utf-8
+
 require 'colored'
+require 'logger'
 
 module Wordmove
-  class Logger
-
-    ERROR = 0
-    INFO = 1
-    VERBOSE = 2
-
-    attr_accessor :level
+  class Logger < ::Logger
 
-    def log(l, message)
-      colors = [ :red, :green, :cyan ]
-      if l <= level
-        puts "  " * [l-1, 0].max + message.send(colors[l])
-      end
+    def task(title)
+      puts ""
+      title = " ✓ #{title} "
+      puts "▬" * 2 + title.green + "▬" * (70 - title.length)
     end
 
-    def info(message)
-      log INFO, message
-    end
-
-    def verbose(message)
-      log VERBOSE, message
-    end
-
-    def error(message)
-      log ERROR, message
+    def task_step(local_step, title)
+      if local_step
+        puts "    local".cyan + " | ".black + title
+      else
+        puts "   remote".yellow + " | ".black + title
+      end
     end
 
   end

data/lib/wordmove/version.rb

@@ -1,3 +1,3 @@
 module Wordmove
-  VERSION = "0.0.8"
+  VERSION = "0.1.0.alpha"
 end

data/wordmove.gemspec

@@ -15,15 +15,12 @@ Gem::Specification.new do |gem|
   gem.require_paths = ["lib"]
   gem.version       = Wordmove::VERSION
 
-  gem.add_dependency 'colored'
-  gem.add_dependency 'escape'
-  gem.add_dependency 'rake'
-  gem.add_dependency 'net-ssh'
-  gem.add_dependency 'net-scp'
+  gem.add_dependency "colored"
+  gem.add_dependency "rake"
   gem.add_dependency "thor"
-  gem.add_dependency "activesupport", "~> 3.0.0"
+  gem.add_dependency "activesupport"
   gem.add_dependency "i18n"
-  gem.add_dependency "hashie"
+  gem.add_dependency "photocopier"
 
   gem.add_development_dependency "rspec"
   gem.add_development_dependency "cucumber"