rrunit 0.0.3

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 rrunit.gemspec
+gemspec

data/Rakefile

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

data/bin/rrunit

@@ -0,0 +1,54 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+$LOAD_PATH.unshift(File.expand_path(File.dirname(__FILE__) + '/../lib'))
+require 'rrunit'
+require 'optparse'
+require 'json'
+
+options = {
+}
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: rrunit -c <service-command>"
+  
+  opts.on('-c', "--command <command>", "The service command") do |command|
+    options[:command] = command
+  end
+  
+  opts.on('-l', "--logdir <logdir>", "The service logdir") do |logdir|
+    options[:log_dir] = logdir
+  end
+  
+  opts.on('-o', "--outputdir <outputdir>", "The directory to output the service") do |output|
+    options[:output] = output
+  end
+  
+  opts.on('-e', "--env <json>", "The environment variables") do |env|
+    options[:environment] = JSON.parse(env)
+  end
+  
+  opts.on('-x', "--xenv <json>", "The exported environment variables") do |env|
+    options[:exported_environment] = JSON.parse(env)
+  end
+  
+  opts.on('-a', "--args <quoted-string>", "The arguments to the service command") do |args|
+    options[:args] = args.split(' ')
+  end
+  
+end.parse!
+
+unless options.key?(:command)
+  puts "You must at least specify a command"
+  exit(1)
+end
+
+output = options.delete(:output)
+output = '/tmp' unless output
+output = "#{output}/#{options[:name]}"
+
+r = Rrunit::Service.new(options.delete(:command))
+options.each do |key,value|
+  r.send(key, value)
+end
+
+r.write(output)

data/lib/rrunit/service.rb

@@ -0,0 +1,94 @@
+module Rrunit
+
+  class Service
+    
+    attr_reader :name
+    attr_reader :command
+    
+    def initialize(command)
+      @command = command
+      @name = File.basename(command)
+      @environment = Hash.new
+      @exported_environment = Hash.new
+      @args = Array.new
+    end
+    
+    def args(*args)
+      @args = args unless args.empty?
+      args.empty? ? @args : self
+    end
+    
+    def log_dir(log_dir=nil)
+      @log_dir = log_dir unless log_dir.nil?
+      log_dir.nil? ? @log_dir : self
+    end
+    
+    def environment(env=nil)
+      @environment=env if env.is_a?(Hash)
+      env.nil? ? @environment : self
+    end
+    
+    def exported_environment(env=nil)
+      @exported_environment=env if env.is_a?(Hash)
+      env.nil? ? @exported_environment : self
+    end
+    
+    def env_var(value)
+      case
+      when value.is_a?(String)
+        "\"#{value}\""
+      else
+        value.to_s
+      end
+    end
+    
+    def service
+      sv = Array.new
+      cmd = "exec #{command}"
+      cmd << " #{@args.join(' ')}" unless @args.empty?
+      sv << "#!/bin/bash"
+      @exported_environment.each do |key, value|
+        sv << "export #{key.to_s.upcase}=#{env_var(value)}"
+      end
+      @environment.each do |key, value|
+        sv << "#{key.to_s.upcase}=#{env_var(value)}"
+      end
+      sv << cmd
+      sv.join("\n")
+    end
+    
+    def log_service
+      if @log_dir
+        sv = Array.new
+        cmd = "exec svlogd -tt #{@log_dir}"
+        sv << "#!/bin/bash"
+        sv << cmd
+        sv.join("\n")
+      end
+    end
+    
+    def write(path)
+      @output_directory = path
+      FileUtils.mkdir_p @output_directory
+      File.open("#{@output_directory}/run", 'w') do |f|
+        f << service
+        f << "\n"
+      end
+      FileUtils.chmod 0755, "#{@output_directory}/run"
+      if @log_dir
+        FileUtils.mkdir_p "#{@output_directory}/log"
+        File.open("#{@output_directory}/log/run", 'w') do |f|
+          f << log_service
+          f << "\n"
+        end
+        FileUtils.chmod 0755, "#{@output_directory}/log/run"
+      end
+    end
+    
+    def rm
+      FileUtils.rm_r @output_directory
+    end
+    
+  end
+  
+end

data/lib/rrunit/version.rb

@@ -0,0 +1,3 @@
+module Rrunit
+  VERSION = "0.0.3"
+end

data/lib/rrunit.rb

@@ -0,0 +1,7 @@
+require 'fileutils'
+require "rrunit/version"
+require "rrunit/service"
+#module Rrunit
+#  # Your code goes here...
+#end
+#

data/rrunit.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "rrunit/version"
+
+Gem::Specification.new do |s|
+  s.name        = "rrunit"
+  s.version     = Rrunit::VERSION
+  s.authors     = ["John Axel Eriksson"]
+  s.email       = ["john@insane.se"]
+  s.homepage    = "https://github.com/johnae/rrunit"
+  s.summary     = %q{Creates and writes runit service files}
+  s.description = %q{Creates and writes runit service files}
+
+  s.rubyforge_project = "rrunit"
+
+  s.add_development_dependency "bundler", ">= 1.0.10"
+  s.add_development_dependency('rspec', '>= 2.6.0')
+
+  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

data/spec/spec_helper.rb

@@ -0,0 +1,10 @@
+rrunit_path = File.expand_path('./lib', File.dirname(__FILE__))
+$:.unshift(rrunit_path) if File.directory?(rrunit_path) && !$:.include?(rrunit_path)
+
+require 'rrunit'
+
+RSpec.configure do |config|
+  config.mock_with :rspec
+  config.color_enabled = true
+  config.formatter = 'd'
+end

data/spec/unit/runit_service_spec.rb

@@ -0,0 +1,145 @@
+require 'spec_helper'
+
+module Rrunit
+  describe Service do
+    
+    context "default settings" do
+    
+      let(:runit_service) { Service.new('/usr/bin/service') }
+      
+      it "has a name" do
+        runit_service.name.should == 'service'
+      end
+      
+      it "has a command" do
+        runit_service.command.should == '/usr/bin/service'
+      end
+      
+      it "has a basic runit run string" do
+        runit_service.service.should == ["#!/bin/bash","exec #{runit_service.command}"].join("\n")
+      end
+      
+    end
+    
+    context "with arguments" do
+      
+      let(:runit_service) { Service.new('service').args('-arg1', 1, '-arg2', 2) }
+    
+      it "has arguments for command" do
+        runit_service.args.should == ['-arg1', 1, '-arg2', 2]
+      end
+      
+      it "has a runit run string with arguments" do
+        runit_service.service.should == ["#!/bin/bash","exec #{runit_service.command} -arg1 1 -arg2 2"].join("\n")
+      end
+      
+    end
+    
+    context "with environment" do
+      
+      let(:runit_service) { 
+        Service.new('service')
+          .environment(variable1: "127.0.0.1", variable2: "www.google.com", variable3: 100)
+          .exported_environment(evariable1: "exported")
+      }
+    
+      it "has a runit run string with environment variables set" do
+        runit_service.service.should == ["#!/bin/bash", "export EVARIABLE1=\"exported\"", "VARIABLE1=\"127.0.0.1\"", "VARIABLE2=\"www.google.com\"", "VARIABLE3=100", "exec #{runit_service.command}"].join("\n")
+      end
+      
+    end
+    
+    context "with log_dir" do
+      
+      let(:runit_service) {
+        Service.new('service')
+          .args('-arg1', 1, '-arg2', 2)
+          .log_dir('/tmp/somedir')
+      }
+      
+      it "has a log_dir" do
+        runit_service.log_dir.should == '/tmp/somedir'
+      end
+      
+      it "has a log_service" do
+        runit_service.log_service.should == ["#!/bin/bash","exec svlogd -tt #{runit_service.log_dir}"].join("\n")
+      end
+      
+    end
+    
+    context "when writing the files" do
+      
+      let(:runit_output_directory) { "/tmp/runit_generator_spec_output_directory" }
+      
+      before do
+        FileUtils.rm_rf(runit_output_directory)
+        runit_service.write(runit_output_directory)
+      end
+      
+      after(:all) do
+        FileUtils.rm_rf(runit_output_directory)
+      end
+      
+      context "without log dir" do
+        
+        let(:runit_service) { 
+          Service.new('service')
+            .args('-arg1', 1, '-arg2', 2)
+            .exported_environment(var1: 100, var2: "HELLO")
+        }
+        
+        it "writes the runit configuration files to specified location" do
+          File.directory?(runit_output_directory).should be true
+          File.exists?("#{runit_output_directory}/run").should be true
+        end
+        
+        it "the runit run file should be executable" do
+          File.stat("#{runit_output_directory}/run").executable?.should be true
+        end
+        
+        it "the program run file should contain the service" do
+          File.read("#{runit_output_directory}/run").should == runit_service.service+"\n"
+        end
+        
+      end
+      
+      context "with log dir" do
+        
+        let(:runit_service) { 
+          Service.new('service')
+            .args('-arg1', 1, '-arg2', 2)
+            .exported_environment(var1: 100, var2: "HELLO")
+            .log_dir('/tmp/somedir')
+        }
+        
+        it "writes the runit configuration files to specified location" do
+          File.directory?(runit_output_directory).should be true
+          File.exists?("#{runit_output_directory}/run").should be true
+        end
+        
+        it "the runit run file should be executable" do
+          File.stat("#{runit_output_directory}/run").executable?.should be true
+        end
+        
+        it "the runit run file should contain the service" do
+          File.read("#{runit_output_directory}/run").should == runit_service.service+"\n"
+        end
+        
+        it "writes the runit log service to specified location" do
+          File.directory?("#{runit_output_directory}/log").should be true
+          File.exists?("#{runit_output_directory}/log/run").should be true
+        end
+        
+        it "the runit log service run file should be executable" do
+          File.stat("#{runit_output_directory}/log/run").executable?.should be true
+        end
+        
+        it "the runit log service run file should contain the log service" do
+          File.read("#{runit_output_directory}/log/run").should == runit_service.log_service+"\n"
+        end
+        
+      end
+    end
+    
+  end
+end

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification
+name: rrunit
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+  prerelease: 
+platform: ruby
+authors:
+- John Axel Eriksson
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-06-23 00:00:00.000000000 +02:00
+default_executable: 
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: &2164580600 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.10
+  type: :development
+  prerelease: false
+  version_requirements: *2164580600
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &2164580100 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 2.6.0
+  type: :development
+  prerelease: false
+  version_requirements: *2164580100
+description: Creates and writes runit service files
+email:
+- john@insane.se
+executables:
+- rrunit
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- bin/rrunit
+- lib/rrunit.rb
+- lib/rrunit/service.rb
+- lib/rrunit/version.rb
+- rrunit.gemspec
+- spec/spec_helper.rb
+- spec/unit/runit_service_spec.rb
+has_rdoc: true
+homepage: https://github.com/johnae/rrunit
+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: rrunit
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: Creates and writes runit service files
+test_files:
+- spec/spec_helper.rb
+- spec/unit/runit_service_spec.rb