invoker 1.0.4 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 36f535d2a652c9164fbe92e7df1e9aff140223d2
-  data.tar.gz: f84168b2628f0f40024de2b46c1de5017a9069d1
+  metadata.gz: 17e01fb0ade988c21a62f385d51e3b07e58f9016
+  data.tar.gz: 2e4e0c36f2df111493b7c05dea7e0ae99244de2c
 SHA512:
-  metadata.gz: b0fb344263af24e67273104073e872b070bfeb73f2a9e782953f29d355216ab5d2c18c254b5d175619ba0e6bd7bd4461e2956e01b5d022574005f54d38640854
-  data.tar.gz: 1e758cf427cd35b11773c06896d3b3fc2ebd9fab50ac8c60ff06379cab2369fc41eb5353f9a53e6e34c5a5b5e04664bb9570ffa001ed6dc7b9fd867aa305fa6b
+  metadata.gz: a7601b89cd0db7b1cc0e780885d14f27ec652739702b5c2a59cf961b677db347bfd9e764ada4a53057de80a418ca3d486949c1a6909ee414505a8b5323593708
+  data.tar.gz: 552b15ffb14c384d02f3ec972ccd8f3a2191afedab5bfa899f19c9fb1ecf0e4df86c65cdbfa475df8babfc746401bc6d914bcb26f0abda2f6fd04283dab092c3

data/.gitignore

@@ -8,3 +8,8 @@ multi.ini
 tags
 .rvmrc
 vendor/
+_site
+.bundle
+coverage
+invoker_profile/
+*.pid

data/.rubocop.yml

@@ -0,0 +1,30 @@
+AllCops:
+  RunRailsCops: false
+  Excludes:
+    - db/**
+
+HashSyntax:
+  Description: 'Use either hash rocket or 1.9 styled hashes.'
+  Enabled: false
+
+LineLength:
+  Max: 100
+
+# Disable Certain Tests
+
+Documentation:
+  Description: 'Document classes and non-namespace modules.'
+  Enabled: false
+
+StringLiterals:
+  Description: 'Checks if uses of quotes match the configured preference.'
+  Enabled: false
+
+Encoding:
+  Description: 'Use UTF-8 as the source file encoding.'
+  Enabled: false
+
+SignalException:
+  Description: 'Do not enforce use of fail when raising exceptions.'
+  # Valid values are: semantic, only_raise and only_fail
+  EnforcedStyle: only_raise

data/.travis.yml

@@ -2,6 +2,7 @@ language: ruby
 rvm:
   - 1.9.3
   - 2.0.0
+  - 2.1.0
 
 script: bundle exec rake spec
 

data/Gemfile

@@ -5,3 +5,4 @@ gemspec
 gem 'pry'
 
 gem 'coveralls', require: false
+gem "simplecov", require: false

data/bin/invoker

@@ -1,11 +1,7 @@
 #!/usr/bin/env ruby
 
-begin
-  require "invoker"
-rescue LoadError
-  invoker_lib_path = File.expand_path('../../lib', __FILE__)
-  $:.unshift(invoker_lib_path)
-  require "invoker"
-end
+invoker_lib_path = File.expand_path('../../lib', __FILE__)
+$:.unshift(invoker_lib_path)
+require "invoker"
 
-Invoker::Runner.run(ARGV)
+Invoker::CLI.start(ARGV)

data/invoker.gemspec

@@ -25,17 +25,16 @@ Gem::Specification.new do |s|
   s.licenses = ["MIT"]
   s.require_paths = ["lib"]
   s.summary = %q{Something small for Process management}
-  s.add_dependency("slop", "~> 3.4.6")
-  s.add_dependency("rainbow", "~> 1.1.4")
-  s.add_dependency("iniparse", "~> 1.1.6")
-  s.add_dependency("formatador", "~> 0.2.4")
-  s.add_dependency("eventmachine", "~> 1.0.3")
-  s.add_dependency("em-proxy", "~> 0.1.8")
-  s.add_dependency("rubydns", "~> 0.6.5")
-  s.add_dependency("uuid", "~> 2.3.7")
-  s.add_dependency("highline", "~> 1.6.19")
-  s.add_dependency("http-parser-lite", "~> 0.6.0")
-  s.add_dependency("dotenv", "~> 0.9.0")
+  s.add_dependency("thor", "~> 0.19")
+  s.add_dependency("rainbow", "~> 2.0")
+  s.add_dependency("iniparse", "~> 1.1")
+  s.add_dependency("formatador", "~> 0.2")
+  s.add_dependency("eventmachine", "~> 1.0")
+  s.add_dependency("em-proxy", "~> 0.1")
+  s.add_dependency("rubydns", "~> 0.7")
+  s.add_dependency("uuid", "~> 2.3")
+  s.add_dependency("http-parser-lite", "~> 0.6")
+  s.add_dependency("dotenv", "~> 0.9")
   s.add_development_dependency("rspec")
   s.add_development_dependency("mocha")
   s.add_development_dependency("rake")

data/lib/invoker.rb

@@ -2,44 +2,118 @@ $: << File.dirname(__FILE__) unless $:.include?(File.expand_path(File.dirname(__
 
 require "fileutils"
 require "formatador"
-require 'rubydns'
-require 'em-proxy'
-require 'http-parser'
+
 require "ostruct"
 require "uuid"
-require "highline"
+require "json"
+require "rainbow"
+require "rainbow/ext/string"
+
 require "invoker/version"
 require "invoker/logger"
-require "invoker/runner"
-require "invoker/command_listener/server"
-require "invoker/command_listener/client"
-require "invoker/power"
+require "invoker/daemon"
+require "invoker/cli"
+require "invoker/dns_cache"
+require "invoker/ipc"
+require "invoker/power/config"
+require "invoker/power/port_finder"
+require "invoker/power/setup"
+require "invoker/power/powerup"
 require "invoker/errors"
 require "invoker/parsers/procfile"
 require "invoker/parsers/config"
-require "invoker/parsers/option_parser"
 require "invoker/commander"
+require "invoker/process_manager"
 require "invoker/command_worker"
 require "invoker/reactor"
 require "invoker/event/manager"
 require "invoker/process_printer"
 
 module Invoker
-  def self.darwin?
-    ruby_platform.downcase.include?("darwin")
-  end
+  class << self
+    attr_accessor :config, :tail_watchers, :commander
+    attr_accessor :dns_cache, :daemonize
 
-  def self.ruby_platform
-    RUBY_PLATFORM
-  end
+    alias_method :daemonize?, :daemonize
+
+    def darwin?
+      ruby_platform.downcase.include?("darwin")
+    end
+
+    def ruby_platform
+      RUBY_PLATFORM
+    end
+
+    def load_invoker_config(file, port)
+      @config = Invoker::Parsers::Config.new(file, port)
+      @dns_cache = Invoker::DNSCache.new(@invoker_config)
+      @tail_watchers = Invoker::CLI::TailWatcher.new
+      @commander = Invoker::Commander.new
+    end
+
+    def close_socket(socket)
+      socket.close
+    rescue StandardError => error
+      Invoker::Logger.puts "Error removing socket #{error}"
+    end
+
+    def daemon
+      @daemon ||= Invoker::Daemon.new
+    end
 
-  def self.can_run_balancer?(throw_warning = true)
-    return false unless darwin?
-    return true if File.exists?(Invoker::Power::Config::CONFIG_LOCATION)
+    def can_run_balancer?(throw_warning = true)
+      return false unless darwin?
+      return true if File.exist?(Invoker::Power::Config::CONFIG_LOCATION)
+
+      if throw_warning
+        Invoker::Logger.puts("Invoker has detected setup has not been run. Domain feature will not work without running setup command.".color(:red))
+      end
+      false
+    end
+
+    def setup_config_location
+      config_location = File.join(Dir.home, '.invoker')
+      return config_location if Dir.exist?(config_location)
+
+      if File.exist?(config_location)
+        old_config = File.read(config_location)
+        FileUtils.rm_f(config_location)
+      end
+
+      FileUtils.mkdir(config_location)
+
+      migrate_old_config(old_config, config_location) if old_config
+      config_location
+    end
+
+    def run_without_bundler
+      if defined?(Bundler)
+        Bundler.with_clean_env do
+          yield
+        end
+      else
+        yield
+      end
+    end
+
+    def notify_user(message)
+      run_without_bundler { check_and_notify_with_terminal_notifier(message) }
+    end
+
+    def check_and_notify_with_terminal_notifier(message)
+      return unless Invoker.darwin?
+
+      command_path = `which terminal-notifier`
+      if command_path && !command_path.empty?
+        system("terminal-notifier -message '#{message}' -title Invoker")
+      end
+    end
 
-    if throw_warning
-      Invoker::Logger.puts("Invoker has detected setup has not been run. Domain feature will not work without running setup command.".color(:red))
+    def migrate_old_config(old_config, config_location)
+      new_config = File.join(config_location, 'config')
+      File.open(new_config, 'w') do |file|
+        file.write(old_config)
+      end
     end
-    false
   end
 end

data/lib/invoker/cli.rb

@@ -0,0 +1,126 @@
+require "socket"
+require "thor"
+
+module Invoker
+  class CLI < Thor
+    def self.start(*args)
+      cli_args = args.flatten
+      # If it is not a valid task, it is probably file argument
+      if default_start_command?(cli_args)
+        args = [cli_args.unshift("start")]
+      end
+      super(*args)
+    end
+
+    desc "setup", "Run Invoker setup"
+    def setup
+      Invoker::Power::Setup.install
+    end
+
+    desc "version", "Print Invoker version"
+    def version
+      Invoker::Logger.puts Invoker::VERSION
+    end
+    map %w(-v --version) => :version
+
+    desc "uninstall", "Uninstall Invoker and all installed files"
+    def uninstall
+      Invoker::Power::Setup.uninstall
+    end
+
+    desc "start CONFIG_FILE", "Start Invoker Server"
+    option :port, type: :numeric, banner: "Port series to be used for starting rack servers"
+    option :daemon,
+      type: :boolean,
+      banner: "Daemonize the server into the background",
+      aliases: [:d]
+    def start(file)
+      Invoker.setup_config_location
+      port = options[:port] || 9000
+      Invoker.daemonize = options[:daemon]
+      Invoker.load_invoker_config(file, port)
+      warn_about_terminal_notifier
+      pinger = Invoker::CLI::Pinger.new(unix_socket)
+      abort("Invoker is already running".color(:red)) if pinger.invoker_running?
+      Invoker.commander.start_manager
+    end
+
+    desc "add process", "Add a program to Invoker server"
+    def add(name)
+      unix_socket.send_command('add', process_name: name)
+    end
+
+    desc "add_http process_name port", "Add an external http process to Invoker DNS server"
+    def add_http(name, port)
+      unix_socket.send_command('add_http', process_name: name, port: port)
+    end
+
+    desc "tail process1 process2", "Tail a particular process"
+    def tail(*names)
+      tailer = Invoker::CLI::Tail.new(names)
+      tailer.run
+    end
+
+    desc "reload process", "Reload a process managed by Invoker"
+    option :signal,
+      banner: "Signal to send for killing the process, default is SIGINT",
+      aliases: [:s]
+    def reload(name)
+      signal = options[:signal] || 'INT'
+      unix_socket.send_command('reload', process_name: name, signal: signal)
+    end
+
+    desc "list", "List all running processes"
+    def list
+      unix_socket.send_command('list') do |response_object|
+        Invoker::ProcessPrinter.new(response_object).tap { |printer| printer.print_table }
+      end
+    end
+
+    desc "remove process", "Stop a process managed by Invoker"
+    option :signal,
+      banner: "Signal to send for killing the process, default is SIGINT",
+      aliases: [:s]
+    def remove(name)
+      signal = options[:signal] || 'INT'
+      unix_socket.send_command('remove', process_name: name, signal: signal)
+    end
+
+    desc "stop", "Stop Invoker daemon"
+    def stop
+      Invoker.daemon.stop
+    end
+
+    private
+
+    def self.default_start_command?(args)
+      command_name = args.first
+      command_name &&
+        !command_name.match(/^-/) &&
+        !valid_tasks.include?(command_name)
+    end
+
+    def self.valid_tasks
+      tasks.keys + ["help"]
+    end
+
+    def unix_socket
+      Invoker::IPC::UnixClient.new
+    end
+
+    def warn_about_terminal_notifier
+      if RUBY_PLATFORM.downcase.include?("darwin")
+        command_path = `which terminal-notifier`
+        if !command_path || command_path.empty?
+          Invoker::Logger.puts "You can enable OSX notification for processes "\
+            "by installing terminal-notifier gem".color(:red)
+        end
+      end
+    end
+  end
+end
+
+require "invoker/cli/question"
+require "invoker/cli/tail_watcher"
+require "invoker/cli/tail"
+require "invoker/cli/pinger"

data/lib/invoker/cli/pinger.rb

@@ -0,0 +1,23 @@
+require "timeout"
+
+module Invoker
+  class CLI::Pinger
+    attr_accessor :unix_client
+    def initialize(unix_client)
+      @unix_client = unix_client
+    end
+
+    def invoker_running?
+      response = send_ping_and_read_response
+      response && response.status == 'pong'
+    end
+
+    private
+
+    def send_ping_and_read_response
+      Timeout.timeout(2) { unix_client.send_and_receive('ping') }
+    rescue Timeout::Error
+      nil
+    end
+  end
+end

data/lib/invoker/cli/question.rb

@@ -0,0 +1,15 @@
+module Invoker
+  class CLI::Question
+    def self.agree(question_text)
+      $stdout.print(question_text)
+      answer = $stdin.gets
+      answer.strip!
+      if answer =~ /\Ay(?:es)?|no?\Z/i
+        answer =~ /\Ay(?:es)?\Z/i
+      else
+        $stdout.puts "Please enter 'yes' or 'no'."
+        agree(question_text)
+      end
+    end
+  end
+end

data/lib/invoker/cli/tail.rb

@@ -0,0 +1,34 @@
+module Invoker
+  class CLI::Tail
+    attr_accessor :process_names
+    def initialize(process_names)
+      verify_process_name(process_names)
+      @process_names = process_names
+      @unix_socket = Invoker::IPC::UnixClient.new
+    end
+
+    def run
+      socket = @unix_socket.send_and_wait('tail', process_names: process_names)
+      trap('INT') { socket.close }
+      loop do
+        message = read_next_line(socket)
+        break unless message
+        puts message.tail_line
+      end
+    end
+
+    private
+
+    def verify_process_name(process_names)
+      if process_names.empty?
+        abort("Tail command requires one or more process name")
+      end
+    end
+
+    def read_next_line(socket)
+      Invoker::IPC.message_from_io(socket)
+    rescue
+      nil
+    end
+  end
+end