meepo 1.5.2

data/lib/invoker/power/setup/osx_setup.rb

@@ -0,0 +1,137 @@
+module Invoker
+  module Power
+    class OsxSetup < Setup
+      FIREWALL_PLIST_FILE = "/Library/LaunchDaemons/com.codemancers.invoker.firewall.plist"
+      RESOLVER_DIR = "/etc/resolver"
+
+      def resolver_file
+        File.join(RESOLVER_DIR, tld)
+      end
+
+      def setup_invoker
+        if setup_resolver_file
+          find_open_ports
+          install_resolver(port_finder.dns_port)
+          install_firewall(port_finder.http_port, port_finder.https_port)
+          # Before writing the config file, drop down to a normal user
+          drop_to_normal_user
+          create_config_file
+        else
+          Invoker::Logger.puts("Invoker is not configured to serve from subdomains".color(:red))
+        end
+        self
+      end
+
+      def uninstall_invoker
+        uninstall_invoker_flag = Invoker::CLI::Question.agree("Are you sure you want to uninstall firewall rules created by setup (y/n) : ")
+
+        if uninstall_invoker_flag
+          remove_resolver_file
+          unload_firewall_rule(true)
+          Invoker::Power::Config.delete
+          Invoker::Logger.puts("Firewall rules were removed")
+        end
+      end
+
+      def build_power_config
+        config = super
+        config[:dns_port] = port_finder.dns_port
+        config
+      end
+
+      def install_resolver(dns_port)
+        open_resolver_for_write { |fl| fl.write(resolve_string(dns_port)) }
+      rescue Errno::EACCES
+        Invoker::Logger.puts("Running setup requires root access, please rerun it with sudo".color(:red))
+        raise
+      end
+
+      def install_firewall(http_port, https_port)
+        File.open(FIREWALL_PLIST_FILE, "w") { |fl|
+          fl.write(plist_string(http_port, https_port))
+        }
+        unload_firewall_rule
+        load_firewall_rule
+      end
+
+      def load_firewall_rule
+        system("launchctl load -Fw #{FIREWALL_PLIST_FILE} 2>/dev/null")
+      end
+
+      def unload_firewall_rule(remove = false)
+        system("pfctl -a com.apple/250.InvokerFirewall -F nat 2>/dev/null")
+        system("launchctl unload -w #{FIREWALL_PLIST_FILE} 2>/dev/null")
+        system("rm -rf #{FIREWALL_PLIST_FILE}") if remove
+      end
+
+      # Ripped from POW code
+      def plist_string(http_port, https_port)
+        plist =<<-EOD
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+<key>Label</key>
+<string>com.codemancers.invoker</string>
+<key>ProgramArguments</key>
+<array>
+<string>sh</string>
+<string>-c</string>
+<string>#{firewall_command(http_port, https_port)}</string>
+</array>
+<key>RunAtLoad</key>
+<true/>
+<key>UserName</key>
+<string>root</string>
+</dict>
+</plist>
+        EOD
+        plist
+      end
+
+      def resolve_string(dns_port)
+        string =<<-EOD
+nameserver 127.0.0.1
+port #{dns_port}
+        EOD
+        string
+      end
+
+      # Ripped from Pow code
+      def firewall_command(http_port, https_port)
+        rules = [
+          "rdr pass on lo0 inet proto tcp from any to any port 80 -> 127.0.0.1 port #{http_port}",
+          "rdr pass on lo0 inet proto tcp from any to any port 443 -> 127.0.0.1 port #{https_port}"
+        ].join("\n")
+        "echo \"#{rules}\" | pfctl -a 'com.apple/250.InvokerFirewall' -f - -E"
+      end
+
+      def setup_resolver_file
+        return true unless File.exist?(resolver_file)
+
+        Invoker::Logger.puts "Invoker has detected an existing Pow installation. We recommend "\
+          "that you uninstall pow and rerun this setup.".color(:red)
+        Invoker::Logger.puts "If you have already uninstalled Pow, proceed with installation"\
+          " by pressing y/n."
+        replace_resolver_flag = Invoker::CLI::Question.agree("Replace Pow configuration (y/n) : ")
+
+        if replace_resolver_flag
+          Invoker::Logger.puts "Invoker has overwritten one or more files created by Pow. "\
+          "If .#{tld} domains still don't resolve locally, try turning off the wi-fi"\
+          " and turning it on. It'll force OS X to reload network configuration".color(:green)
+        end
+        replace_resolver_flag
+      end
+
+      private
+
+      def open_resolver_for_write
+        FileUtils.mkdir(RESOLVER_DIR) unless Dir.exist?(RESOLVER_DIR)
+        fl = File.open(resolver_file, "w")
+        yield fl
+      ensure
+        fl && fl.close
+      end
+    end
+  end
+end

data/lib/invoker/power/templates/400.html

@@ -0,0 +1,40 @@
+<!doctype html>
+<html>
+  <head>
+    <meta charset="utf-8">
+    <title>Invoker</title>
+    <style>
+    body {
+      margin: 0;
+      padding: 0;
+      background: #fff;
+      line-height: 18px;
+    }
+    div.page {
+      padding: 36px 90px;
+    }
+    h1, h2, p, li {
+      font-family: Helvetica, sans-serif;
+      font-size: 13px;
+    }
+    h1 {
+      line-height: 45px;
+      font-size: 36px;
+      margin: 0;
+    }
+    h2 {
+      line-height: 27px;
+      font-size: 18px;
+      font-weight: normal;
+      margin: 0;
+    }
+    </style>
+  </head>
+  <body class="">
+    <div class="page">
+      <h1>Bad request</h1>
+      <hr>
+      <h2>Invoker couldn't understand the request</h2>
+    </div>
+  </body>
+</html>

data/lib/invoker/power/templates/404.html

@@ -0,0 +1,40 @@
+<!doctype html>
+<html>
+<head>
+  <meta charset="utf-8">
+  <title>Invoker</title>
+  <style>
+    body {
+      margin: 0;
+      padding: 0;
+      background: #fff;
+      line-height: 18px;
+    }
+    div.page {
+      padding: 36px 90px;
+    }
+    h1, h2, p, li {
+      font-family: Helvetica, sans-serif;
+      font-size: 13px;
+    }
+    h1 {
+      line-height: 45px;
+      font-size: 36px;
+      margin: 0;
+    }
+    h2 {
+      line-height: 27px;
+      font-size: 18px;
+      font-weight: normal;
+      margin: 0;
+    }
+  </style>
+</head>
+<body class="">
+  <div class="page">
+    <h1>Application not found</h1>
+    <hr>
+    <h2>Invoker could not find the application. Please check the configuration file.<h2>
+  </div>
+</body>
+</html>

data/lib/invoker/power/templates/503.html

@@ -0,0 +1,40 @@
+<!doctype html>
+<html>
+<head>
+  <meta charset="utf-8">
+  <title>Invoker</title>
+  <style>
+    body {
+      margin: 0;
+      padding: 0;
+      background: #fff;
+      line-height: 18px;
+    }
+    div.page {
+      padding: 36px 90px;
+    }
+    h1, h2, p, li {
+      font-family: Helvetica, sans-serif;
+      font-size: 13px;
+    }
+    h1 {
+      line-height: 45px;
+      font-size: 36px;
+      margin: 0;
+    }
+    h2 {
+      line-height: 27px;
+      font-size: 18px;
+      font-weight: normal;
+      margin: 0;
+    }
+  </style>
+</head>
+<body class="">
+  <div class="page">
+    <h1>Application not running</h1>
+    <hr>
+    <h2>Invoker did not get any response. Please check if the application is running.<h2>
+  </div>
+</body>
+</html>

data/lib/invoker/power/url_rewriter.rb

@@ -0,0 +1,40 @@
+module Invoker
+  module Power
+    class UrlRewriter
+      def select_backend_config(complete_path)
+        possible_matches = extract_host_from_domain(complete_path)
+        exact_match = nil
+        possible_matches.each do |match|
+          if match
+            exact_match = dns_check(process_name: match)
+            break if exact_match.port
+          end
+        end
+        exact_match
+      end
+
+      def extract_host_from_domain(complete_path)
+        matching_strings = []
+        tld_match_regex.map do |regexp|
+          if (match_result = complete_path.match(regexp))
+            matching_strings << match_result[1]
+          end
+        end
+        matching_strings.uniq
+      end
+
+      private
+
+      def tld_match_regex
+        tld = Invoker.config.tld
+        [/([\w.-]+)\.#{tld}(\:\d+)?$/, /([\w-]+)\.#{tld}(\:\d+)?$/]
+      end
+
+      def dns_check(dns_args)
+        Invoker::IPC::UnixClient.send_command("dns_check", dns_args) do |dns_response|
+          dns_response
+        end
+      end
+    end
+  end
+end

data/lib/invoker/process_manager.rb

@@ -0,0 +1,198 @@
+module Invoker
+  # Class is responsible for managing all the processes Invoker is supposed
+  # to manage. Takes care of starting, stopping and restarting processes.
+  class ProcessManager
+    LABEL_COLORS = [:green, :yellow, :blue, :magenta, :cyan]
+    attr_accessor :open_pipes, :workers
+
+    def initialize
+      @open_pipes = {}
+      @workers = {}
+      @worker_mutex = Mutex.new
+      @thread_group = ThreadGroup.new
+    end
+
+    def start_process(process_info)
+      m, s = PTY.open
+      s.raw! # disable newline conversion.
+
+      pid = run_command(process_info, s)
+
+      s.close
+
+      worker = CommandWorker.new(process_info.label, m, pid, select_color)
+
+      add_worker(worker)
+      wait_on_pid(process_info.label, pid)
+    end
+
+    # Start a process given their name
+    # @param process_name [String] Command label of process specified in config file.
+    def start_process_by_name(process_name)
+      if process_running?(process_name)
+        Invoker::Logger.puts "\nProcess '#{process_name}' is already running".color(:red)
+        return false
+      end
+
+      process_info = Invoker.config.process(process_name)
+      start_process(process_info) if process_info
+    end
+
+    # Remove a process from list of processes managed by invoker supervisor.It also
+    # kills the process before removing it from the list.
+    #
+    # @param remove_message [Invoker::IPC::Message::Remove]
+    # @return [Boolean] if process existed and was removed else false
+    def stop_process(remove_message)
+      worker = workers[remove_message.process_name]
+      command_label = remove_message.process_name
+      return false unless worker
+      signal_to_use = remove_message.signal || 'INT'
+
+      Invoker::Logger.puts("Removing #{command_label} with signal #{signal_to_use}".color(:red))
+      kill_or_remove_process(worker.pid, signal_to_use, command_label)
+    end
+
+    # Receive a message from user to restart a Process
+    # @param [Invoker::IPC::Message::Reload]
+    def restart_process(reload_message)
+      command_label = reload_message.process_name
+      if stop_process(reload_message.remove_message)
+        Invoker.commander.schedule_event(command_label, :worker_removed) do
+          start_process_by_name(command_label)
+        end
+      else
+        start_process_by_name(command_label)
+      end
+    end
+
+    def run_power_server
+      return unless Invoker.can_run_balancer?(false)
+
+      powerup_id = Invoker::Power::Powerup.fork_and_start
+      wait_on_pid("powerup_manager", powerup_id)
+      at_exit do
+        begin
+          Process.kill("INT", powerup_id)
+        rescue Errno::ESRCH; end
+      end
+    end
+
+    # Given a file descriptor returns the worker object
+    #
+    # @param fd [IO] an IO object with valid file descriptor
+    # @return [Invoker::CommandWorker] The worker object which is associated with this fd
+    def get_worker_from_fd(fd)
+      open_pipes[fd.fileno]
+    end
+
+    def load_env(directory = nil)
+      directory ||= ENV['PWD']
+
+      if !directory || directory.empty? || !Dir.exist?(directory)
+        return {}
+      end
+
+      default_env = File.join(directory, '.env')
+      local_env = File.join(directory, '.env.local')
+      env = {}
+
+      if File.exist?(default_env)
+        env.merge!(Dotenv::Environment.new(default_env))
+      end
+
+      if File.exist?(local_env)
+        env.merge!(Dotenv::Environment.new(local_env))
+      end
+
+      env
+    end
+
+    def kill_workers
+      @workers.each do |key, worker|
+        kill_or_remove_process(worker.pid, "INT", worker.command_label)
+      end
+      @workers = {}
+    end
+
+    # List currently running commands
+    def process_list
+      Invoker::IPC::Message::ListResponse.from_workers(workers)
+    end
+
+    private
+
+    def wait_on_pid(command_label, pid)
+      raise Invoker::Errors::ToomanyOpenConnections if @thread_group.enclosed?
+
+      thread = Thread.new do
+        Process.wait(pid)
+        message = "Process with command #{command_label} exited with status #{$?.exitstatus}"
+        Invoker::Logger.puts("\n#{message}".color(:red))
+        Invoker.notify_user(message)
+        Invoker.commander.trigger(command_label, :exit)
+      end
+      @thread_group.add(thread)
+    end
+
+    def select_color
+      selected_color = LABEL_COLORS.shift
+      LABEL_COLORS.push(selected_color)
+      selected_color
+    end
+
+    def process_running?(command_label)
+      !!workers[command_label]
+    end
+
+    def kill_or_remove_process(pid, signal_to_use, command_label)
+      process_kill(pid, signal_to_use)
+      true
+    rescue Errno::ESRCH
+      Invoker::Logger.puts("Killing process with #{pid} and name #{command_label} failed".color(:red))
+      remove_worker(command_label, false)
+      false
+    end
+
+    def process_kill(pid, signal_to_use)
+      if signal_to_use.to_i == 0
+        Process.kill(signal_to_use, -Process.getpgid(pid))
+      else
+        Process.kill(signal_to_use.to_i, -Process.getpgid(pid))
+      end
+    end
+
+    # Remove worker from all collections
+    def remove_worker(command_label, trigger_event = true)
+      worker = @workers[command_label]
+      if worker
+        @open_pipes.delete(worker.pipe_end.fileno)
+        @workers.delete(command_label)
+      end
+      if trigger_event
+        Invoker.commander.trigger(command_label, :worker_removed)
+      end
+    end
+
+    # add worker to global collections
+    def add_worker(worker)
+      @open_pipes[worker.pipe_end.fileno] = worker
+      @workers[worker.command_label] = worker
+      Invoker.commander.watch_for_read(worker.pipe_end)
+    end
+
+    def run_command(process_info, write_pipe)
+      command_label = process_info.label
+
+      Invoker.commander.schedule_event(command_label, :exit) { remove_worker(command_label) }
+
+      env_options = load_env(process_info.dir)
+
+      spawn_options = {
+        :chdir => process_info.dir || ENV['PWD'], :out => write_pipe, :err => write_pipe,
+        :pgroup => true, :close_others => true, :in => :close
+      }
+      Invoker.run_without_bundler { spawn(env_options, process_info.cmd, spawn_options) }
+    end
+  end
+end