pushapp 0.0.1

data/lib/pushapp/hook.rb

@@ -0,0 +1,127 @@
+require 'erb'
+
+module Pushapp
+  class Hook
+
+    ANSI_COLORS = {
+      :reset          => 0,
+      :black          => 30,
+      :red            => 31,
+      :green          => 32,
+      :yellow         => 33,
+      :blue           => 34,
+      :magenta        => 35,
+      :cyan           => 36,
+      :white          => 37,
+      :bright_black   => 30,
+      :bright_red     => 31,
+      :bright_green   => 32,
+      :bright_yellow  => 33,
+      :bright_blue    => 34,
+      :bright_magenta => 35,
+      :bright_cyan    => 36,
+      :bright_white   => 37,
+    }
+
+    HOOK_COLORS = %w( cyan yellow green magenta blue bright_cyan bright_yellow
+                      bright_magenta bright_blue red bright_green bright_red )
+
+    attr_accessor :remote
+
+    def initialize(remote)
+      @remote  = remote
+      @config  = remote.config
+      @options = remote.options
+
+      @remote_index = @config.remotes.index(@remote)
+      @remote_color = ANSI_COLORS[HOOK_COLORS[@remote_index % HOOK_COLORS.length].to_sym]
+      @echo_color   = ANSI_COLORS[:bright_green]
+      @error_color  = ANSI_COLORS[:bright_red]
+    end
+
+    def setup
+      prepare_hook
+      deploy_hook
+    end
+
+    private
+
+    def padded_max_length
+      if @remote.group
+        @config.remotes_grouped_by(@remote.group).map {|r| r.full_name}.max.length
+      else
+        @remote.name.length
+      end
+    end
+
+    def padded_name
+      @padden_name ||= @remote.full_name.ljust([padded_max_length, "remote:".length].max)
+    end
+
+    def prefix
+      "\e[1G\e[#{@remote_color}m#{padded_name} |\e[0m "
+    end
+
+    def pre
+      "#\{ENV[\"PAP_PRE\"]\}"
+    end
+
+    def colorize
+      %{2>&1 | ruby -pe '$_="#{pre}#\{$_\}"'}
+    end
+
+    def info message
+      %{ruby -e 'puts "#{pre}\e[#{@echo_color}m\e[1m[pushapp]\e[0m\e[#{@echo_color}m INFO: #{message}\e[0m"'}
+    end
+
+    def load_template(template_name)
+      ::ERB.new(File.read(find_template(template_name))).result(binding)
+    end
+
+    def find_template(template_name)
+      "#{Pushapp::TEMPLATE_ROOT}/#{template_name}.erb"
+    end
+
+    def prepare_hook
+      # info "Generating and uploading post-receive hook for #{@remote.name}"
+      hook = generate_hook
+      write hook
+    end
+
+    def deploy_hook
+      # debug "Copying hook for #{@remote.name} to #{@remote.location}"
+      copy_hook
+      set_hook_permissions
+    end
+
+    def generate_hook
+      load_template 'hook/base'
+    end
+
+    def write(hook)
+      File.open(Pushapp::TMP_HOOK, "wb") do |f|
+        f.puts hook
+      end
+    end
+
+    def set_hook_permissions
+      @remote.run "#{make_hook_executable}"
+    end
+
+    def copy_hook
+      # debug "Making hook executable"
+      # TODO: handle missing user?
+      if @remote.host
+        Pushapp::Pipe.run "scp #{Pushapp::TMP_HOOK} #{@remote.user}@#{@remote.host}:#{@remote.path}/.git/hooks/post-receive"
+      else
+        Pushapp::Pipe.run "cp #{Pushapp::TMP_HOOK} #{@remote.path}/.git/hooks/post-receive"
+      end
+    end
+
+    def make_hook_executable
+      # debug "Making hook executable"
+      "chmod +x #{@remote.path}/.git/hooks/post-receive"
+    end
+  end
+end
+

data/lib/pushapp/logger.rb

@@ -0,0 +1,34 @@
+require 'logger'
+require 'thor'
+
+module Pushapp
+  class Logger < ::Logger
+    attr_reader :shell
+
+    def initialize
+      super($stdout)
+      @shell = Thor::Shell::Color.new
+
+      self.progname = '[pushapp]'
+
+              #  DEBUG  INFO    WARN     ERROR FATAL  UNKNOWN
+      @colors = {
+        'DEBUG'  => :blue,
+        'INFO'   => :green,
+        'WARN'   => :magenta,
+        'ERROR'  => :red,
+        'FATAL'  => :red,
+        'UNKOWN' => :black
+      }
+
+      self.formatter = proc { |severity, datetime, progname, msg|
+        color    = @colors[severity]
+
+        progname = @shell.set_color progname, color, :bold
+        sev      = @shell.set_color "#{severity}:", color
+        msg      = @shell.set_color msg, color
+        "#{progname} #{sev} #{msg}\n"
+      }
+    end
+  end
+end

data/lib/pushapp/pipe.rb

@@ -0,0 +1,35 @@
+require 'open3'
+require 'stringio'
+
+module Pushapp
+  class Pipe
+
+    def self.run(command)
+      case command
+      when String
+        pipe(command)
+      when Pushapp::Tasks::Script
+        pipe(command.cmd)
+      when Pushapp::Tasks::Base
+        command.run
+      when Array
+        pipe(command)
+      else
+        raise "Unknown command format: '#{command.inspect}'"
+      end
+    end
+
+    def self.capture(cmd)
+      output, s = Open3.capture2e(cmd)
+      raise "Failed with status #{s.exitstatus}: #{cmd.inspect}" unless s.success?
+      output
+    end
+
+    private
+
+    def self.pipe cmd, stdin=$stdin, stdout=$stdout
+      s = Open3.pipeline(cmd, :in => stdin, :out => stdout).last
+      raise "Failed with status #{s.exitstatus}: #{cmd.inspect}" unless s.success?
+    end
+  end
+end

data/lib/pushapp/remote.rb

@@ -0,0 +1,141 @@
+require 'pushapp/tasks/rake'
+require 'shellwords'
+
+class Pushapp::Remote
+
+  attr_reader :config
+  attr_reader :tasks
+  attr_reader :name
+  attr_reader :location
+  attr_reader :options
+  attr_reader :group
+
+  def initialize(name, group, location, config, options = {})
+    @name     = name
+    @location = location
+    @config   = config
+    @options  = options
+    @group    = group
+    @tasks    = Hash.new { |hash, key| hash[key] = [] }
+  end
+
+  def full_name
+    [group, name].compact.join('-')
+  end
+
+  def rake(task_name, task_options={})
+    tasks[@event] << Pushapp::Tasks::Rake.new(task_name, merge_options(task_options))
+  end
+
+  def script(script_name, script_options={})
+    tasks[@event] << Pushapp::Tasks::Script.new(script_name, merge_options(script_options))
+  end
+
+  def task(task_name, task_options={})
+    tasks[@event] << config.known_task(task_name).new(merge_options(task_options).merge(task_name: task_name))
+  end
+
+  def on event, &block
+    @event = event.to_s
+    instance_eval(&block) if block_given?
+  end
+
+  def tasks_on event
+    tasks[event.to_s]
+  end
+
+  def path
+    if host
+      @location.match(/:(.*)$/)[1]
+    else
+      @location
+    end
+  end
+
+  def host
+    host = @location.match(/@(.*):/)
+    host[1] unless host.nil?
+  end
+
+  def user
+    user = @location.match(/(.*)@/)
+    user[1] unless user.nil?
+  end
+
+  def ssh!
+    exec "cd #{path} && #{shell_env} $SHELL -l"
+  end
+
+  #
+  # Set up Repositories and Hook
+  #
+  def setup!
+    run "#{init_repository} && #{setup_repository}"
+  end
+
+  #
+  # Update git remote and hook
+  #
+  def update!
+    Pushapp::Hook.new(self).setup
+  end
+
+  def exec cmd
+    if host
+      Kernel.exec "ssh -t #{user}@#{host} '#{cmd}'"
+    else
+      Kernel.exec cmd
+    end
+  end
+
+  def env_run cmd
+    if host
+      Pushapp::Pipe.run "ssh #{user}@#{host} 'cd #{path} && #{shell_env} $SHELL -l -c \"#{cmd}\"'"
+    else
+      Bundler.with_original_env do
+        Pushapp::Pipe.run "cd #{path} && #{shell_env} #{cmd}"
+      end
+    end
+  end
+
+  def run cmd
+    if host
+      Pushapp::Pipe.run "ssh #{user}@#{host} '#{cmd}'"
+    else
+      Bundler.with_original_env do
+        Pushapp::Pipe.run cmd
+      end
+    end
+  end
+
+  def env
+    (options[:env] || {})
+  end
+
+  private
+
+  #
+  # Initialize an empty repository
+  #
+  def init_repository
+    # Instead of git init with a path, so it does not fail on older
+    # git versions (https://github.com/effkay/blazing/issues/53)
+    "mkdir #{path}; cd #{path} && git init"
+  end
+
+  #
+  # Allow pushing to currently checked out branch
+  #
+  def setup_repository
+    "cd #{path} && git config receive.denyCurrentBranch ignore"
+  end
+
+  def shell_env
+    env.map {|k,v| "#{k}=\"#{Shellwords.escape(v)}\""}.join(" ")
+  end
+
+  def merge_options task_options={}
+    Pushapp.rmerge(task_options, options).merge(remote: name)
+  end
+
+end

data/lib/pushapp/tasks/base.rb

@@ -0,0 +1,27 @@
+require 'pushapp/config'
+require 'pushapp/logger'
+
+module Pushapp
+  module Tasks
+    class Base
+      attr_reader :options
+      attr_reader :logger
+
+      def initialize options={}
+        @options = options
+        @logger = Pushapp::Logger.new
+      end
+
+      def run
+      end
+
+      def self.register_as name
+        Pushapp::Config.register_task name, self
+      end
+
+      def inspect
+        options[:task_name]
+      end
+    end
+  end
+end

data/lib/pushapp/tasks/foreman_export.rb

@@ -0,0 +1,14 @@
+require 'pushapp/tasks/base'
+
+module Pushapp
+  module Tasks
+    class ForemanExport < Base
+
+      def run
+        logger.info 'Forman exporting...'
+      end
+
+      register_as :foreman_export
+    end
+  end
+end

data/lib/pushapp/tasks/rake.rb

@@ -0,0 +1,15 @@
+require 'pushapp/tasks/script'
+
+module Pushapp
+  module Tasks
+    class Rake < Script
+      def cmd
+        [env, "rake #{script}"]
+      end
+
+      def inspect
+        "rake: #{script}"
+      end
+    end
+  end
+end

data/lib/pushapp/tasks/script.rb

@@ -0,0 +1,31 @@
+require 'pushapp/tasks/base'
+
+module Pushapp
+  module Tasks
+    class Script < Base
+      attr_reader :script
+
+      def initialize script, options={}
+        super(options)
+
+        @script = script
+      end
+
+      def cmd
+        [env, script]
+      end
+
+      def env
+        Hash[env_options.map {|k, v| [k.to_s, v.to_s] }]
+      end
+
+      def env_options
+        options[:env] || {}
+      end
+
+      def inspect
+        "script: #{script}"
+      end
+    end
+  end
+end

data/lib/pushapp/version.rb

@@ -0,0 +1,3 @@
+module Pushapp
+  VERSION = '0.0.1'
+end

data/pushapp.gemspec

@@ -0,0 +1,26 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'pushapp/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'pushapp'
+  spec.version       = Pushapp::VERSION
+  spec.authors       = ["Yury Korolev"]
+  spec.email         = ["yurykorolev@me.com"]
+  spec.description   = %q{Push your App}
+  spec.summary       = %q{Push your App}
+  spec.homepage      = 'https://github.com/anjlab/pushapp'
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'thor'
+
+  spec.add_development_dependency 'bundler', '~> 1.3'
+  spec.add_development_dependency 'rake'
+  spec.add_development_dependency 'minitest'
+end