upstart-exporter 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in upstart-exporter.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/upstart-export

@@ -0,0 +1,29 @@
+#!/usr/bin/env ruby
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '../lib/'))
+
+require 'upstart-exporter'
+require 'optparse'
+
+USAGE = "Usage: upstart-export -p PROCFILE -n APPNAME"
+options = {}
+
+OptionParser.new do |opts|
+
+  opts.on("-p", "--procfile PROCFILE", "Procfile to export") do |p|
+    options[:procfile] = p
+  end
+
+  opts.on("-n", "--appname NAME", "Application name") do |n|
+    options[:app_name] = n
+  end
+
+end.parse!
+
+abort(USAGE) unless options[:procfile] && options[:app_name]
+
+
+begin
+  Upstart::Exporter.new(options).export
+rescue Upstart::ExportError => e
+  abort(e.to_s)
+end

data/lib/upstart-exporter/version.rb

@@ -0,0 +1,5 @@
+module Upstart
+  class Exporter
+    VERSION = "0.0.1"
+  end
+end

data/lib/upstart-exporter.rb

@@ -0,0 +1,180 @@
+require "upstart-exporter/version"
+
+module Upstart
+  class ExportError < RuntimeError; end
+
+  class Exporter
+    DEFAULTS = {
+      'helper_dir' => '/var/local/upstart_helpers/',
+      'upstart_dir' => '/etc/init/',
+      'run_user' => 'service'
+    }
+    CONF = '/etc/upstart-exporter.yaml'
+
+    attr_accessor :helper_dir, :upstart_dir, :run_user
+
+    def initialize(options)
+      process_opts(options)
+      read_global_opts
+      check_dir(@helper_dir)
+      check_dir(@upstart_dir)
+    end
+
+
+    def read_global_opts
+      config = if FileTest.file?(CONF)
+        YAML::load(File.read(CONF))
+      else
+        STDERR.puts "#{CONF} not found"
+        {}
+      end
+      %w{helper_dir upstart_dir run_user}.each do |param|
+        value = if config[param]
+          config[param]
+        else
+          STDERR.puts "Param #{param} is not set, taking default value #{DEFAULTS[param]}"
+          DEFAULTS[param]
+        end
+        self.send "#{param}=", value
+      end
+    end
+
+    def process_opts(options)
+      process_procfile(options[:procfile])
+      process_appname(options[:app_name])
+    end
+
+    def process_procfile(name)
+      error "#{name} is not a readable file" unless FileTest.file?(name)
+      @commands = {}
+      content = File.read(name)
+      content.lines.each do |line|
+        line.chomp!
+        if line =~ /^(\w+?):(.*)$/
+          label = $1
+          command = $2
+          @commands[label] = command
+        else
+          error "procfile lines should have the following format: 'some_label: command'"
+        end
+      end
+    end
+
+    def process_appname(app_name)
+      error "Application name should contain only letters (and underscore) and be nonempty, so #{app_name.inspect} is not suitable" unless app_name =~ /^\w+$/ 
+      @app_name = "fb-#{app_name}"
+    end
+
+    def check_dir(dir)
+      FileUtils.mkdir_p(dir) unless FileTest.directory?(dir)
+      error "Path #{dir} does not exist" unless FileTest.directory?(dir)
+      elements = [''] + dir.split('/').select{|el| !el.empty?}
+      path = ''
+      elements.each do |el|
+        path += (el + '/')
+        if FileTest.world_writable?(path)
+          error "Path #{path} (in #{dir}) is writable by others"
+        end
+      end
+    end
+
+    def export
+      clear
+      export_app
+      @commands.each do |cmd_name, cmd|
+        export_command(cmd_name, cmd)
+      end
+    end
+
+    HELPER_TPL = <<-HEREDOC
+#!/bin/sh
+%s
+HEREDOC
+
+    APP_TPL = <<-HEREDOC
+pre-start script
+
+bash << "EOF"
+  mkdir -p /var/log/%s
+  chown -R %s /var/log/%s
+EOF
+
+end script
+HEREDOC
+
+    COMMAND_TPL = <<-HEREDOC
+start on starting %s
+stop on stopping %s
+HEREDOC
+
+
+    COMMAND_REAL_TPL = <<-HEREDOC
+start on starting %s
+stop on stopping %s
+respawn
+
+exec su - %s -c '/bin/sh %s >> /var/log/%s/%s.log 2>&1'
+HEREDOC
+
+    def upstart_conf
+      File.join(@upstart_dir, "#{@app_name}.conf")
+    end
+
+    def app_cmd(cmd_name)
+      "#{@app_name}-#{cmd_name}"
+    end
+    
+    def upstart_cmd_conf(cmd_name)
+      File.join(@upstart_dir, "#{app_cmd(cmd_name)}.conf")
+    end
+    
+    def helper_cmd_conf(cmd_name)
+      File.join(@helper_dir, "#{app_cmd(cmd_name)}.sh")
+    end
+
+    def clear
+      FileUtils.rm(upstart_conf) if FileTest.file?(upstart_conf)
+      Dir[upstart_cmd_conf('*')].each do |f|
+        FileUtils.rm(f)
+      end
+      Dir[helper_cmd_conf('*')].each do |f|
+        FileUtils.rm(f)
+      end
+    end
+
+    def export_app
+      app_conf = APP_TPL % [@app_name, @run_user, @app_name]
+      File.open(upstart_conf, 'w') do |f|
+        f.write(app_conf)
+      end
+    end
+
+    def export_cmd_helper(cmd_name, cmd)
+      helper_script_cont = HELPER_TPL % [cmd]
+      File.open(helper_cmd_conf(cmd_name), 'w') do |f|
+        f.write(helper_script_cont)
+      end
+    end
+
+    def export_cmd_upstart_confs(cmd_name)
+      cmd_upstart_conf_content = COMMAND_TPL % [@app_name, @app_name]
+      File.open(upstart_cmd_conf(cmd_name), 'w') do |f|
+        f.write(cmd_upstart_conf_content)
+      end
+      
+      cmd_upstart_conf_content_real = COMMAND_REAL_TPL % [app_cmd(cmd_name), app_cmd(cmd_name), @run_user, helper_cmd_conf(cmd_name), @app_name, cmd_name]
+      File.open(upstart_cmd_conf(cmd_name + '-real'), 'w') do |f|
+        f.write(cmd_upstart_conf_content_real)
+      end
+    end
+
+    def export_command(cmd_name, cmd)
+      export_cmd_helper(cmd_name, cmd)
+      export_cmd_upstart_confs(cmd_name)
+    end
+
+    def error(msg)
+      raise Upstart::ExportError, msg 
+    end
+  end
+end

data/upstart-exporter.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "upstart-exporter/version"
+
+Gem::Specification.new do |s|
+  s.name        = "upstart-exporter"
+  s.version     = Upstart::Exporter::VERSION
+  s.authors     = ["Ilya Averyanov"]
+  s.email       = ["ilya@averyanov.org"]
+  s.homepage    = ""
+  s.summary     = %q{Gem for converting Procfile-like files to upstart scripts}
+  s.description = %q{Gem for converting Procfile-like files to upstart scripts}
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+end

metadata

@@ -0,0 +1,53 @@
+--- !ruby/object:Gem::Specification
+name: upstart-exporter
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Ilya Averyanov
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-12-08 00:00:00.000000000Z
+dependencies: []
+description: Gem for converting Procfile-like files to upstart scripts
+email:
+- ilya@averyanov.org
+executables:
+- upstart-export
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- bin/upstart-export
+- lib/upstart-exporter.rb
+- lib/upstart-exporter/version.rb
+- upstart-exporter.gemspec
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Gem for converting Procfile-like files to upstart scripts
+test_files: []