0.9.0.beta.1

Signed-off-by: David Dollar <ddollar@gmail.com>

.gitignore

@@ -1,4 +1,7 @@
-/.bundle
-/coverage
-/man
-/pkg
+.bundle
+coverage
+example/log/*
+man/*.?
+man/*.html
+man/*.markdown
+pkg

Gemfile

@@ -0,0 +1,17 @@
+source "http://rubygems.org"
+
+group :development do
+  gem 'parka'
+  gem 'rake'
+  gem 'ronn'
+end
+
+group :test do
+  gem 'fakefs', '~> 0.2.1'
+  gem 'rcov',   '~> 0.9.8'
+  gem 'rr',     '~> 0.10.11'
+  gem 'rspec',  '~> 2.0.0.beta.19'
+end
+
+gem 'term-ansicolor', '~> 1.0.5'
+gem 'thor',           '~> 0.13.6'

Gemfile.lock

@@ -0,0 +1,45 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    diff-lcs (1.1.2)
+    fakefs (0.2.1)
+    hpricot (0.8.2)
+    mime-types (1.16)
+    mustache (0.11.2)
+    parka (0.3.1)
+      rest-client
+      thor
+    rake (0.8.7)
+    rcov (0.9.8)
+    rdiscount (1.6.5)
+    rest-client (1.6.0)
+      mime-types (>= 1.16)
+    ronn (0.7.3)
+      hpricot (>= 0.8.2)
+      mustache (>= 0.7.0)
+      rdiscount (>= 1.5.8)
+    rr (0.10.11)
+    rspec (2.0.0.beta.19)
+      rspec-core (= 2.0.0.beta.19)
+      rspec-expectations (= 2.0.0.beta.19)
+      rspec-mocks (= 2.0.0.beta.19)
+    rspec-core (2.0.0.beta.19)
+    rspec-expectations (2.0.0.beta.19)
+      diff-lcs (>= 1.1.2)
+    rspec-mocks (2.0.0.beta.19)
+    term-ansicolor (1.0.5)
+    thor (0.13.8)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  fakefs (~> 0.2.1)
+  parka
+  rake
+  rcov (~> 0.9.8)
+  ronn
+  rr (~> 0.10.11)
+  rspec (~> 2.0.0.beta.19)
+  term-ansicolor (~> 1.0.5)
+  thor (~> 0.13.6)

README.markdown

@@ -0,0 +1,4 @@
+Foreman
+=======
+
+See the [man page](http://ddollar.github.com/foreman) for usage.

Rakefile

@@ -0,0 +1,51 @@
+require "rubygems"
+require "bundler"
+Bundler.setup
+
+require "rake"
+require "rspec"
+require "rspec/core/rake_task"
+
+$:.unshift File.expand_path("../lib", __FILE__)
+require "foreman"
+
+task :default => :spec
+task :release => :man
+
+desc "Run all specs"
+Rspec::Core::RakeTask.new(:spec) do |t|
+  t.pattern = 'spec/**/*_spec.rb'
+end
+
+desc "Generate RCov code coverage report"
+task :rcov => "rcov:build" do
+  %x{ open coverage/index.html }
+end
+
+Rspec::Core::RakeTask.new("rcov:build") do |t|
+  t.pattern = 'spec/**/*_spec.rb'
+  t.rcov = true
+  t.rcov_opts = [ "--exclude", Gem.default_dir , "--exclude", "spec" ]
+end
+
+desc 'Build the manual'
+task :man do
+  ENV['RONN_MANUAL']  = "Foreman Manual"
+  ENV['RONN_ORGANIZATION'] = "Foreman #{Foreman::VERSION}"
+  sh "ronn -w -s toc -r5 --markdown man/*.ronn"
+  sh "git add README.markdown"
+  sh "git commit -m 'update readme' || echo 'nothing to commit'"
+end
+
+task :pages => :man do
+  sh %{
+    cp man/foreman.1.html /tmp/foreman.1.html
+    git checkout gh-pages
+    rm ./index.html
+    cp /tmp/foreman.1.html ./index.html
+    git add -u index.html
+    git commit -m "rebuilding man page"
+    git push origin -f gh-pages
+    git checkout master
+  }
+end

autotest/discover.rb

@@ -0,0 +1 @@
+Autotest.add_discovery { "rspec2" }

bin/foreman

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+$:.unshift File.expand_path("../../lib", __FILE__)
+
+require "foreman/cli"
+
+Foreman::CLI.start

data/example/Pstypes

@@ -0,0 +1,2 @@
+ticker ./ticker $PORT
+error  ./error

data/example/error

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+puts "will error in 10s"
+sleep 5
+raise "Dying"

data/example/log/neverdie.log

@@ -0,0 +1,4 @@
+tick
+tick
+./never_die:6:in `sleep': Interrupt
+	from ./never_die:6

data/example/ticker

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+while true
+  puts "tick: #{ARGV.inspect}"
+  sleep 1
+end

data/export/upstart/master.conf.erb

@@ -0,0 +1,8 @@
+pre-start script
+
+bash << "EOF"
+  mkdir -p <%= log_root %>
+  chown -R <%= user %> <%= log_root %>
+EOF
+
+end script

data/export/upstart/process.conf.erb

@@ -0,0 +1,6 @@
+start on starting <%= app %>-<%= process.name %>
+stop on stopping <%= app %>-<%= process.name %>
+respawn
+
+chdir <%= engine.directory %>
+exec su - <%= user %> -c 'export PORT=<%= port %>; <%= process.command %> >> <%= log_root %>/<%=process.name%>-<%=num%>.log 2>&1'

data/export/upstart/process_master.conf.erb

@@ -0,0 +1,2 @@
+start on starting <%= app %>
+stop on stopping <%= app %>

foreman.gemspec

@@ -0,0 +1,12 @@
+require "rubygems"
+require "parka/specification"
+
+Parka::Specification.new do |gem|
+  gem.name     = "foreman"
+  gem.version  = Foreman::VERSION
+  gem.summary  = "Process manager for applications with multiple components"
+  gem.homepage = "http://github.com/ddollar/foreman"
+
+  gem.executables  = "foreman"
+  gem.files       << "man/foreman.1"
+end

index.html

@@ -1,265 +0,0 @@
-<!DOCTYPE html>
-<html>
-<head>
-  <meta http-equiv='content-type' value='text/html;charset=utf8'>
-  <meta name='generator' value='Ronn/v0.7.3 (http://github.com/rtomayko/ronn/tree/0.7.3)'>
-  <title>foreman(1) - manage Procfile-based applications</title>
-  <style type='text/css' media='all'>
-  /* style: man */
-  body#manpage {margin:0}
-  .mp {max-width:100ex;padding:0 9ex 1ex 4ex}
-  .mp p,.mp pre,.mp ul,.mp ol,.mp dl {margin:0 0 20px 0}
-  .mp h2 {margin:10px 0 0 0}
-  .mp > p,.mp > pre,.mp > ul,.mp > ol,.mp > dl {margin-left:8ex}
-  .mp h3 {margin:0 0 0 4ex}
-  .mp dt {margin:0;clear:left}
-  .mp dt.flush {float:left;width:8ex}
-  .mp dd {margin:0 0 0 9ex}
-  .mp h1,.mp h2,.mp h3,.mp h4 {clear:left}
-  .mp pre {margin-bottom:20px}
-  .mp pre+h2,.mp pre+h3 {margin-top:22px}
-  .mp h2+pre,.mp h3+pre {margin-top:5px}
-  .mp img {display:block;margin:auto}
-  .mp h1.man-title {display:none}
-  .mp,.mp code,.mp pre,.mp tt,.mp kbd,.mp samp,.mp h3,.mp h4 {font-family:monospace;font-size:14px;line-height:1.42857142857143}
-  .mp h2 {font-size:16px;line-height:1.25}
-  .mp h1 {font-size:20px;line-height:2}
-  .mp {text-align:justify;background:#fff}
-  .mp,.mp code,.mp pre,.mp pre code,.mp tt,.mp kbd,.mp samp {color:#131211}
-  .mp h1,.mp h2,.mp h3,.mp h4 {color:#030201}
-  .mp u {text-decoration:underline}
-  .mp code,.mp strong,.mp b {font-weight:bold;color:#131211}
-  .mp em,.mp var {font-style:italic;color:#232221;text-decoration:none}
-  .mp a,.mp a:link,.mp a:hover,.mp a code,.mp a pre,.mp a tt,.mp a kbd,.mp a samp {color:#0000ff}
-  .mp b.man-ref {font-weight:normal;color:#434241}
-  .mp pre {padding:0 4ex}
-  .mp pre code {font-weight:normal;color:#434241}
-  .mp h2+pre,h3+pre {padding-left:0}
-  ol.man-decor,ol.man-decor li {margin:3px 0 10px 0;padding:0;float:left;width:33%;list-style-type:none;text-transform:uppercase;color:#999;letter-spacing:1px}
-  ol.man-decor {width:100%}
-  ol.man-decor li.tl {text-align:left}
-  ol.man-decor li.tc {text-align:center;letter-spacing:4px}
-  ol.man-decor li.tr {text-align:right;float:right}
-  </style>
-  <style type='text/css' media='all'>
-  /* style: toc */
-  .man-navigation {display:block !important;position:fixed;top:0;left:113ex;height:100%;width:100%;padding:48px 0 0 0;border-left:1px solid #dbdbdb;background:#eee}
-  .man-navigation a,.man-navigation a:hover,.man-navigation a:link,.man-navigation a:visited {display:block;margin:0;padding:5px 2px 5px 30px;color:#999;text-decoration:none}
-  .man-navigation a:hover {color:#111;text-decoration:underline}
-  </style>
-</head>
-<!--
-  The following styles are deprecated and will be removed at some point:
-  div#man, div#man ol.man, div#man ol.head, div#man ol.man.
-
-  The .man-page, .man-decor, .man-head, .man-foot, .man-title, and
-  .man-navigation should be used instead.
--->
-<body id='manpage'>
-  <div class='mp' id='man'>
-
-  <div class='man-navigation' style='display:none'>
-    <a href="#NAME">NAME</a>
-    <a href="#SYNOPSIS">SYNOPSIS</a>
-    <a href="#DESCRIPTION">DESCRIPTION</a>
-    <a href="#RUNNING">RUNNING</a>
-    <a href="#EXPORTING">EXPORTING</a>
-    <a href="#GLOBAL-OPTIONS">GLOBAL OPTIONS</a>
-    <a href="#EXPORT-FORMATS">EXPORT FORMATS</a>
-    <a href="#INITTAB-EXPORT">INITTAB EXPORT</a>
-    <a href="#UPSTART-EXPORT">UPSTART EXPORT</a>
-    <a href="#PROCFILE">PROCFILE</a>
-    <a href="#ENVIRONMENT">ENVIRONMENT</a>
-    <a href="#DEFAULT-OPTIONS">DEFAULT OPTIONS</a>
-    <a href="#EXAMPLES">EXAMPLES</a>
-    <a href="#COPYRIGHT">COPYRIGHT</a>
-    </div>
-
-  <ol class='man-decor man-head man head'>
-    <li class='tl'>foreman(1)</li>
-    <li class='tc'>Foreman Manual</li>
-    <li class='tr'>foreman(1)</li>
-  </ol>
-
-  <h2 id="NAME">NAME</h2>
-<p class="man-name">
-  <code>foreman</code> - <span class="man-whatis">manage Procfile-based applications</span>
-</p>
-
-<h2 id="SYNOPSIS">SYNOPSIS</h2>
-
-<p><code>foreman start [process]</code><br />
-<code>foreman run &lt;command></code><br />
-<code>foreman export &lt;format> [location]</code></p>
-
-<h2 id="DESCRIPTION">DESCRIPTION</h2>
-
-<p>Foreman is a manager for Procfile-based applications. Its aim is to
-abstract away the details of the Procfile format, and allow you to either run
-your application directly or export it to some other process management
-format.</p>
-
-<h2 id="RUNNING">RUNNING</h2>
-
-<p><code>foreman start</code> is used to run your application directly from the command line.</p>
-
-<p>If no additional parameters are passed, foreman will run one instance of each
-type of process defined in your Procfile.</p>
-
-<p>If a parameter is passed, foreman will run one instance of the specified
-application type.</p>
-
-<p>The following options control how the application is run:</p>
-
-<dl>
-<dt><code>-c</code>, <code>--concurrency</code></dt><dd><p>Specify the number of each process type to run. The value passed in
-should be in the format <code>process=num,process=num</code></p></dd>
-<dt><code>-e</code>, <code>--env</code></dt><dd><p>Specify one or more .env files to load</p></dd>
-<dt><code>-f</code>, <code>--procfile</code></dt><dd><p>Specify an alternate Procfile to load, implies <code>-d</code> at the Procfile root.</p></dd>
-<dt><code>-p</code>, <code>--port</code></dt><dd><p>Specify which port to use as the base for this application. Should be
-a multiple of 1000.</p></dd>
-</dl>
-
-
-<p><code>foreman run</code> is used to run one-off commands using the same environment
-as your defined processes.</p>
-
-<h2 id="EXPORTING">EXPORTING</h2>
-
-<p><code>foreman export</code> is used to export your application to another process
-management format.</p>
-
-<p>An location to export can be passed as an argument. This argument may be
-either required or optional depending on the export format.</p>
-
-<p>The following options control how the application is run:</p>
-
-<dl>
-<dt><code>-a</code>, <code>--app</code></dt><dd><p>Use this name rather than the application's root directory name as the
-name of the application when exporting.</p></dd>
-<dt><code>-c</code>, <code>--concurrency</code></dt><dd><p>Specify the number of each process type to run. The value passed in
-should be in the format <code>process=num,process=num</code></p></dd>
-<dt><code>-l</code>, <code>--log</code></dt><dd><p>Specify the directory to place process logs in.</p></dd>
-<dt><code>-p</code>, <code>--port</code></dt><dd><p>Specify which port to use as the base for this application. Should be
-a multiple of 1000.</p></dd>
-<dt><code>-t</code>, <code>--template</code></dt><dd><p>Specify an alternate template to use for creating export files.
-See <a href="https://github.com/ddollar/foreman/tree/master/data/export" data-bare-link="true">https://github.com/ddollar/foreman/tree/master/data/export</a> for examples.</p></dd>
-<dt><code>-u</code>, <code>--user</code></dt><dd><p>Specify the user the application should be run as. Defaults to the
-app name</p></dd>
-</dl>
-
-
-<h2 id="GLOBAL-OPTIONS">GLOBAL OPTIONS</h2>
-
-<p>These options control all modes of foreman's operation.</p>
-
-<dl>
-<dt><code>-d</code>, <code>--root</code></dt><dd><p>Specify an alternate application root. This defaults to the directory
-containing the Procfile.</p></dd>
-<dt><code>-e</code>, <code>--env</code></dt><dd><p>Specify an alternate environment file. You can specify more than one
-file by using: <code>--env file1,file2</code>.</p></dd>
-<dt><code>-f</code>, <code>--procfile</code></dt><dd><p>Specify an alternate location for the application's Procfile. This file's
-containing directory will be assumed to be the root directory of the
-application.</p></dd>
-</dl>
-
-
-<h2 id="EXPORT-FORMATS">EXPORT FORMATS</h2>
-
-<p>foreman currently supports the following output formats:</p>
-
-<ul>
-<li><p>bluepill</p></li>
-<li><p>inittab</p></li>
-<li><p>runit</p></li>
-<li><p>upstart</p></li>
-</ul>
-
-
-<h2 id="INITTAB-EXPORT">INITTAB EXPORT</h2>
-
-<p>Will export a chunk of inittab-compatible configuration:</p>
-
-<pre><code># ----- foreman example processes -----
-EX01:4:respawn:/bin/su - example -c 'PORT=5000 bundle exec thin start &gt;&gt; /var/log/web-1.log 2&gt;&amp;1'
-EX02:4:respawn:/bin/su - example -c 'PORT=5100 bundle exec rake jobs:work &gt;&gt; /var/log/job-1.log 2&gt;&amp;1'
-# ----- end foreman example processes -----
-</code></pre>
-
-<h2 id="UPSTART-EXPORT">UPSTART EXPORT</h2>
-
-<p>Will create a series of upstart scripts in the location you specify. Scripts
-will be structured to make the following commands valid:</p>
-
-<p>  <code>start appname</code></p>
-
-<p>  <code>stop appname-processname</code></p>
-
-<p>  <code>restart appname-processname-3</code></p>
-
-<h2 id="PROCFILE">PROCFILE</h2>
-
-<p>A Procfile should contain both a name for the process and the command used
-to run it.</p>
-
-<pre><code>web: bundle exec thin start
-job: bundle exec rake jobs:work
-</code></pre>
-
-<p>A process name may contain letters, numbers amd the underscore character.
-You can validate your Procfile format using the <code>check</code> command:</p>
-
-<pre><code>$ foreman check
-</code></pre>
-
-<h2 id="ENVIRONMENT">ENVIRONMENT</h2>
-
-<p>If a <code>.env</code> file exists in the current directory, the default environment will
-be read from it. This file should contain key/value pairs, separated by <code>=</code>, with
-one key/value pair per line.</p>
-
-<pre><code>FOO=bar
-BAZ=qux
-</code></pre>
-
-<h2 id="DEFAULT-OPTIONS">DEFAULT OPTIONS</h2>
-
-<p>If a <code>.foreman</code> file exists in the current directory, default options will
-be read from it. This file should be in YAML format with the long option
-name as keys. Example:</p>
-
-<pre><code>concurrency: alpha=0,bravo=1
-port: 15000
-</code></pre>
-
-<h2 id="EXAMPLES">EXAMPLES</h2>
-
-<p>Start one instance of each process type, interleave the output on stdout:</p>
-
-<pre><code>$ foreman start
-</code></pre>
-
-<p>Export the application in upstart format:</p>
-
-<pre><code>$ foreman export upstart /etc/init
-</code></pre>
-
-<p>Run one process type from the application defined in a specific Procfile:</p>
-
-<pre><code>$ foreman start alpha -p ~/myapp/Procfile
-</code></pre>
-
-<h2 id="COPYRIGHT">COPYRIGHT</h2>
-
-<p>Foreman is Copyright (C) 2010 David Dollar <a href="http://daviddollar.org" data-bare-link="true">http://daviddollar.org</a></p>
-
-
-  <ol class='man-decor man-foot man foot'>
-    <li class='tl'>Foreman 0.63.0</li>
-    <li class='tc'>April 2013</li>
-    <li class='tr'>foreman(1)</li>
-  </ol>
-
-  </div>
-</body>
-</html>

lib/foreman.rb

@@ -0,0 +1,8 @@
+module Foreman
+
+  VERSION = "0.9.0.beta.1"
+
+  class AppDoesNotExist < Exception; end
+
+end
+

lib/foreman/cli.rb

@@ -0,0 +1,75 @@
+require "foreman"
+require "foreman/engine"
+require "foreman/export"
+require "thor"
+
+class Foreman::CLI < Thor
+
+  class_option :pstypes, :type => :string, :aliases => "-f", :desc => "Default: Pstypes"
+
+  desc "start [PROCESS]", "Start the application, or a specific process"
+
+  method_option :port,        :type => :numeric, :aliases => "-p"
+  method_option :concurrency, :type => :string,  :aliases => "-c",
+    :banner => '"alpha=5,bar=3"'
+
+  def start(process=nil)
+    check_pstypes!
+
+    if process
+      engine.execute(process, options)
+    else
+      engine.start(options)
+    end
+  end
+
+  desc "export FORMAT LOCATION", "Export the application to another process management format"
+
+  method_option :app,         :type => :string,  :aliases => "-a"
+  method_option :log,         :type => :string,  :aliases => "-l"
+  method_option :port,        :type => :numeric, :aliases => "-p"
+  method_option :user,        :type => :string,  :aliases => "-u"
+  method_option :concurrency, :type => :string,  :aliases => "-c",
+    :banner => '"alpha=5,bar=3"'
+
+  def export(format, location=nil)
+    check_pstypes!
+
+    formatter = case format
+      when "upstart" then Foreman::Export::Upstart
+      when "inittab" then Foreman::Export::Inittab
+      else error "Unknown export format: #{format}."
+    end
+
+    formatter.new(engine).export(location, options)
+
+  rescue Foreman::Export::Exception => ex
+    error ex.message
+  end
+
+private ######################################################################
+
+  def check_pstypes!
+    error("#{pstypes} does not exist.") unless File.exist?(pstypes)
+  end
+
+  def engine
+    @engine ||= Foreman::Engine.new(pstypes)
+  end
+
+  def pstypes
+    options[:pstypes] || "Pstypes"
+  end
+
+private ######################################################################
+
+  def error(message)
+    puts "ERROR: #{message}"
+    exit 1
+  end
+
+  def pstypes_exists?(pstypes)
+    File.exist?(pstypes)
+  end
+
+end

lib/foreman/engine.rb

@@ -0,0 +1,169 @@
+require "foreman"
+require "foreman/process"
+require "foreman/utils"
+require "pty"
+require "tempfile"
+require "term/ansicolor"
+require "fileutils"
+
+class Foreman::Engine
+
+  attr_reader :pstypes
+  attr_reader :directory
+
+  extend Term::ANSIColor
+
+  COLORS = [ cyan, yellow, green, magenta, red ]
+
+  def initialize(pstypes)
+    @pstypes   = read_pstypes(pstypes)
+    @directory = File.expand_path(File.dirname(pstypes))
+  end
+
+  def processes
+    @processes ||= begin
+      pstypes.split("\n").inject({}) do |hash, line|
+        next if line.strip == ""
+        name, command = line.split(" ", 2)
+        process = Foreman::Process.new(name, command)
+        process.color = next_color
+        hash.update(process.name => process)
+      end
+    end
+  end
+
+  def start(options={})
+    proctitle "ruby: foreman master"
+
+    processes.each do |name, process|
+      fork process, options
+    end
+
+    trap("TERM") { puts "SIGTERM received"; kill_all("TERM") }
+    trap("INT")  { puts "SIGINT received";  kill_all("INT")  }
+
+    watch_for_termination
+  end
+
+  def execute(name, options={})
+    fork processes[name], options
+
+    trap("TERM") { puts "SIGTERM received"; kill_all("TERM") }
+    trap("INT")  { puts "SIGINT received";  kill_all("INT")  }
+
+    watch_for_termination
+  end
+
+  def port_for(process, num, base_port=nil)
+    base_port ||= 5000
+    offset = processes.keys.sort.index(process.name) * 100
+    base_port.to_i + offset + num - 1
+  end
+
+private ######################################################################
+
+  def fork(process, options={})
+    concurrency = Foreman::Utils.parse_concurrency(options[:concurrency])
+
+    1.upto(concurrency[process.name]) do |num|
+      fork_individual(process, num, port_for(process, num, options[:port]))
+    end
+  end
+
+  def fork_individual(process, num, port)
+    ENV["PORT"] = port.to_s
+    ENV["PS"]   = "#{process.name}.#{num}"
+
+    pid = Process.fork do
+      run(process)
+    end
+
+    info "started with pid #{pid}, PORT=#{port}", process
+    running_processes[pid] = process
+  end
+
+  def run(process, log_to_file=true)
+    proctitle "ruby: foreman #{process.name}"
+
+    begin
+      Dir.chdir directory do
+        FileUtils.mkdir_p "log"
+        command = process.command
+
+        PTY.spawn("#{process.command} 2>&1") do |stdin, stdout, pid|
+          until stdin.eof?
+            info stdin.gets, process
+          end
+        end
+      end
+    rescue PTY::ChildExited, Interrupt
+      begin
+        info "process exiting", process
+      rescue Interrupt
+      end
+    end
+  end
+
+  def kill_all(signal="TERM")
+    info "terminating"
+    running_processes.each do |pid, process|
+      info "killing #{process.name} in pid #{pid}"
+      Process.kill(signal, pid)
+    end
+  end
+
+  def info(message, process=nil)
+    print process.color if process
+    print "#{Time.now.strftime("%H:%M:%S")} #{pad_process_name(process)} | "
+    print Term::ANSIColor.reset
+    print message.chomp
+    puts
+  end
+
+  def longest_process_name
+    @longest_process_name ||= begin
+      longest = processes.keys.map { |name| name.length }.sort.last
+      longest = 6 if longest < 6 # system
+      longest
+    end
+  end
+
+  def pad_process_name(process)
+    name = process ? "#{ENV["PS"]}" : "system"
+    name.ljust(longest_process_name + 3) # add 3 for process number padding
+  end
+
+  def print_info
+    info "currently running processes:"
+    running_processes.each do |pid, process|
+      info "pid #{pid}", process
+    end
+  end
+
+  def proctitle(title)
+    $0 = title
+  end
+
+  def read_pstypes(pstypes)
+    File.read(pstypes)
+  end
+
+  def watch_for_termination
+    pid, status = Process.wait2
+    process = running_processes.delete(pid)
+    info "process terminated", process
+    kill_all
+    Process.waitall
+  end
+
+  def running_processes
+    @running_processes ||= {}
+  end
+
+  def next_color
+    @current_color ||= -1
+    @current_color  +=  1
+    @current_color >= COLORS.length ? "" : COLORS[@current_color]
+  end
+
+end

lib/foreman/export.rb

@@ -0,0 +1,8 @@
+require "foreman"
+
+module Foreman::Export
+  class Exception < ::Exception; end
+end
+
+require "foreman/export/upstart"
+require "foreman/export/inittab"

lib/foreman/export/base.rb

@@ -0,0 +1,38 @@
+require "foreman/export"
+require "foreman/utils"
+
+class Foreman::Export::Base
+
+  attr_reader :engine
+
+  def initialize(engine)
+    @engine = engine
+  end
+
+  def export
+    raise "export method must be overridden"
+  end
+
+private ######################################################################
+
+  def error(message)
+    raise Foreman::Export::Exception.new(message)
+  end
+
+  def say(message)
+    puts "[foreman export] %s" % message
+  end
+
+  def export_template(name)
+    File.read(File.expand_path("../../../../data/export/#{name}", __FILE__))
+  end
+
+  def write_file(filename, contents)
+    say "writing: #{filename}"
+
+    File.open(filename, "w") do |file|
+      file.puts contents
+    end
+  end
+
+end

lib/foreman/export/inittab.rb

@@ -0,0 +1,38 @@
+require "foreman/export/base"
+
+class Foreman::Export::Inittab < Foreman::Export::Base
+
+  def export(fname=nil, options={})
+    app = options[:app] || File.basename(engine.directory)
+    user = options[:user] || app
+    log_root = options[:log] || "/var/log/#{app}"
+
+    concurrency = Foreman::Utils.parse_concurrency(options[:concurrency])
+
+    inittab = []
+    inittab << "# ----- foreman #{app} processes -----"
+
+    engine.processes.values.inject(1) do |index, process|
+      1.upto(concurrency[process.name]) do |num|
+        id = app.slice(0, 2).upcase + sprintf("%02d", index)
+        port = engine.port_for(process, num, options[:port])
+        inittab << "#{id}:4:respawn:/bin/su - #{user} -c 'PORT=#{port} #{process.command} >> #{log_root}/#{process.name}-#{num}.log 2>&1'"
+        index += 1
+      end
+      index
+    end
+
+    inittab << "# ----- end foreman #{app} processes -----"
+
+    inittab = inittab.join("\n") + "\n"
+
+    if fname
+      FileUtils.mkdir_p(log_root) rescue error "could not create #{log_root}"
+      FileUtils.chown(user, nil, log_root) rescue error "could not chown #{log_root} to #{user}"
+      write_file(fname, inittab)
+    else
+      puts inittab
+    end
+  end
+
+end

lib/foreman/export/upstart.rb

@@ -0,0 +1,41 @@
+require "erb"
+require "foreman/export/base"
+
+class Foreman::Export::Upstart < Foreman::Export::Base
+
+  def export(location, options={})
+    error("Must specify a location") unless location
+
+    FileUtils.mkdir_p location
+
+    app = options[:app] || File.basename(engine.directory)
+    user = options[:user] || app
+    log_root = options[:log] || "/var/log/#{app}"
+
+    Dir["#{location}/#{app}*.conf"].each do |file|
+      say "cleaning up: #{file}"
+      FileUtils.rm(file)
+    end
+
+    concurrency = Foreman::Utils.parse_concurrency(options[:concurrency])
+
+    master_template = export_template("upstart/master.conf.erb")
+    master_config   = ERB.new(master_template).result(binding)
+    write_file "#{location}/#{app}.conf", master_config
+
+    process_template = export_template("upstart/process.conf.erb")
+
+    engine.processes.values.each do |process|
+      process_master_template = export_template("upstart/process_master.conf.erb")
+      process_master_config   = ERB.new(process_master_template).result(binding)
+      write_file "#{location}/#{app}-#{process.name}.conf", process_master_config
+
+      1.upto(concurrency[process.name]) do |num|
+        port = engine.port_for(process, num, options[:port])
+        process_config = ERB.new(process_template).result(binding)
+        write_file "#{location}/#{app}-#{process.name}-#{num}.conf", process_config
+      end
+    end
+  end
+
+end

lib/foreman/process.rb

@@ -0,0 +1,14 @@
+require "foreman"
+
+class Foreman::Process
+
+  attr_reader :name
+  attr_reader :command
+  attr_accessor :color
+
+  def initialize(name, command)
+    @name    = name
+    @command = command
+  end
+
+end

lib/foreman/utils.rb

@@ -0,0 +1,15 @@
+require "foreman"
+
+class Foreman::Utils
+
+  def self.parse_concurrency(concurrency)
+    @concurrency ||= begin
+      pairs = concurrency.to_s.gsub(/\s/, "").split(",")
+      pairs.inject(Hash.new(1)) do |hash, pair|
+        process, amount = pair.split("=")
+        hash.update(process => amount.to_i)        
+      end
+    end
+  end
+
+end

man/foreman.1.ronn

@@ -0,0 +1,126 @@
+foreman(1) -- manage Pstypes-based applications
+================================================
+
+## SYNOPSIS
+
+`foreman start [process]`<br>
+`foreman export <format> [location]`
+
+## DESCRIPTION
+
+**Foreman** is a manager for Pstypes-based applications. Its aim is to
+abstract away the details of the Pstypes format, and allow you to either run
+your application directly or export it to some other process management
+format.
+
+## RUNNING
+
+`foreman start` is used to run your application directly from the command line.
+
+If no additional parameters are passed, foreman will run one instance of each
+type of process defined in your Pstypes.
+
+If a parameter is passed, foreman will run one instance of the specified
+application type.
+
+The following options control how the application is run:
+
+  * `-c`, `--concurrency`:
+    Specify the number of each process type to run. The value passed in
+    should be in the format `process=num,process=num`
+
+  * `-p`, `--port`:
+    Specify which port to use as the base for this application. Should be
+    a multiple of 1000.
+
+## EXPORTING
+
+`foreman export` is used to export your application to another process
+management format.
+
+An location to export can be passed as an argument. This argument may be
+either required or optional depending on the export format.
+
+The following options control how the application is run:
+
+  * `-a`, `--app`:
+    Use this name rather than the application's root directory name as the
+    name of the application when exporting.
+
+  * `-c`, `--concurrency`:
+    Specify the number of each process type to run. The value passed in
+    should be in the format `process=num,process=num`
+
+  * `-l`, `--log`:
+    Specify the directory to place process logs in.
+
+  * `-p`, `--port`:
+    Specify which port to use as the base for this application. Should be
+    a multiple of 1000.
+
+  * `-u`, `--user`:
+    Specify the user the application should be run as. Defaults to the
+    app name
+
+## OPTIONS
+
+These options control all modes of foreman's operation.
+
+  * `-f`, `--pstypes`:
+    Specify an alternate location for the application's Pstypes. This file's
+    containing directory will be assumed to be the root directory of the
+    application.
+
+## EXPORT FORMATS
+
+foreman currently supports the following output formats:
+
+  * inittab
+
+  * upstart
+
+## INITTAB EXPORT
+
+Will export a chunk of inittab-compatible configuration:
+
+    # ----- foreman example processes -----
+    EX01:4:respawn:/bin/su - example -c 'PORT=5000 bundle exec thin start >> /var/log/web-1.log 2>&1'
+    EX02:4:respawn:/bin/su - example -c 'PORT=5100 bundle exec rake jobs:work >> /var/log/job-1.log 2>&1'
+    # ----- end foreman example processes -----
+
+## UPSTART EXPORT
+
+Will create a series of upstart scripts in the location you specify. Scripts
+will be structured to make the following commands valid:
+
+  `start appname`
+
+  `stop appname-processname`
+
+  `restart appname-processname-3`
+
+## PSTYPES
+
+A Pstyes file should contain both a name for the process and the command used
+to run it.
+
+    web  bundle exec thin start
+    job  bundle exec rake jobs:work
+
+## EXAMPLES
+
+Start one instance of each process type, interleave the output on stdout:
+
+    $ foreman start
+
+Export the application in upstart format:
+
+    $ foreman export upstart /etc/init
+
+Run one process type from the application defined in a specific Pstypes:
+
+    $ foreman start alpha -p ~/app/Pstypes
+
+## COPYRIGHT
+
+Foreman is Copyright (C) 2010 David Dollar <http://daviddollar.org>

spec/foreman/cli_spec.rb

@@ -0,0 +1,61 @@
+require "spec_helper"
+require "foreman/cli"
+
+describe "Foreman::CLI" do
+  subject { Foreman::CLI.new }
+
+  describe "start" do
+    describe "with a non-existent Pstypes" do
+      it "prints an error" do
+        mock_error(subject, "Pstypes does not exist.") do
+          dont_allow.instance_of(Foreman::Engine).start
+          subject.start
+        end
+      end
+    end
+
+    describe "with a Pstypes" do
+      before(:each) { write_pstypes }
+
+      it "runs successfully" do
+        dont_allow(subject).error
+        mock.instance_of(Foreman::Engine).start({})
+        subject.start
+      end
+    end
+  end
+
+  describe "export" do
+    describe "with a non-existent Pstypes" do
+      it "prints an error" do
+        mock_error(subject, "Pstypes does not exist.") do
+          dont_allow.instance_of(Foreman::Engine).export
+          subject.export("testapp")
+        end
+      end
+    end
+
+    describe "with a Pstypes" do
+      before(:each) { write_pstypes }
+
+      describe "with an invalid formatter" do
+        it "prints an error" do
+          mock_error(subject, "Unknown export format: invalidformatter.") do
+            subject.export("invalidformatter")
+          end
+        end
+      end
+
+      describe "with a valid config" do
+        before(:each) { write_foreman_config("testapp") }
+
+        it "runs successfully" do
+          dont_allow(subject).error
+          mock.instance_of(Foreman::Export::Upstart).export("/tmp/foo", {})
+          subject.export("upstart", "/tmp/foo")
+        end
+      end
+    end
+  end
+
+end

spec/foreman/engine_spec.rb

@@ -0,0 +1,50 @@
+require "spec_helper"
+require "foreman/engine"
+
+describe "Foreman::Engine" do
+  subject { Foreman::Engine.new("Pstypes") }
+
+  describe "initialize" do
+    describe "without an existing Pstypes" do
+      it "raises an error" do
+        lambda { subject }.should raise_error
+      end
+    end
+
+    describe "with a Pstypes" do
+      it "reads the processes" do
+        write_pstypes
+        subject.processes["alpha"].command.should == "./alpha"
+        subject.processes["bravo"].command.should == "./bravo"
+      end
+    end
+  end
+
+  describe "start" do
+    it "forks the processes" do
+      write_pstypes
+      mock(subject).fork(subject.processes["alpha"], {})
+      mock(subject).fork(subject.processes["bravo"], {})
+      mock(subject).watch_for_termination
+      subject.start
+    end
+
+    it "handles concurrency" do
+      write_pstypes
+      mock(subject).fork_individual(subject.processes["alpha"], 5000)
+      mock(subject).fork_individual(subject.processes["alpha"], 5001)
+      mock(subject).fork_individual(subject.processes["bravo"], 5100)
+      mock(subject).watch_for_termination
+      subject.start(:concurrency => "alpha=2")
+    end
+  end
+
+  describe "execute" do
+    it "runs the processes" do
+      write_pstypes
+      mock(subject).fork(subject.processes["alpha"], {})
+      mock(subject).watch_for_termination
+      subject.execute("alpha")
+    end
+  end
+end

spec/foreman/export/upstart_spec.rb

@@ -0,0 +1,2 @@
+require "spec_helper"
+require "foreman/export/upstart"

spec/foreman/export_spec.rb

@@ -0,0 +1,2 @@
+require "spec_helper"
+require "foreman/export"

spec/foreman/process_spec.rb

@@ -0,0 +1,2 @@
+require "spec_helper"
+require "foreman/process"

spec/foreman_spec.rb

@@ -0,0 +1,11 @@
+require "spec_helper"
+require "foreman"
+
+describe Foreman do
+
+  describe "VERSION" do
+    subject { Foreman::VERSION }
+    it { should be_a String }
+  end
+
+end

spec/spec_helper.rb

@@ -0,0 +1,38 @@
+require "rubygems"
+require "rspec"
+require "fakefs/safe"
+require "fakefs/spec_helpers"
+
+$:.unshift "lib"
+
+def mock_error(subject, message)
+  mock_exit do
+    mock(subject).puts("ERROR: #{message}")
+    yield
+  end
+end
+
+def mock_exit(&block)
+  block.should raise_error(SystemExit)
+end
+
+def write_foreman_config(app)
+  File.open("/etc/foreman/#{app}.conf", "w") do |file|
+    file.puts %{#{app}_processes="alpha bravo"}
+    file.puts %{#{app}_alpha="1"}
+    file.puts %{#{app}_bravo="2"}
+  end
+end
+
+def write_pstypes(pstypes="Pstypes")
+  File.open(pstypes, "w") do |file|
+    file.puts "alpha ./alpha"
+    file.puts "bravo ./bravo"
+  end
+end
+
+Rspec.configure do |config|
+  config.color_enabled = true
+  config.include FakeFS::SpecHelpers
+  config.mock_with :rr
+end