rctl 0.0.1

data/bin/rctl

@@ -0,0 +1,76 @@
+#!/usr/bin/ruby
+#charset: utf-8
+
+#
+# Copyright (c) 2012 Olivier Amblet<olivier@amblet.net>
+# 
+# Permission is hereby granted, free of charge, to any person
+# obtaining a copy of this software and associated documentation
+# files (the "Software"), to deal in the Software without
+# restriction, including without limitation the rights to use,
+# copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the
+# Software is furnished to do so, subject to the following
+# conditions:
+# 
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+# OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+# OTHER DEALINGS IN THE SOFTWARE.
+#
+
+require 'rctl'
+require 'rctl/version'
+require 'optparse'
+
+ctl = Rctl::Ctl.new
+server_switch = []
+usage = false
+options = {}
+
+
+parser = OptionParser.new do |opt|
+  opt.banner = "Usage: rctl COMMAND [OPTIONS]"
+  opt.separator ""
+  opt.separator "Commands"
+  opt.separator "    start   start server(s)"
+  opt.separator "    stop    stop server(s)"
+  opt.separator ""
+  opt.separator "Options"
+
+  opt.on("-a", "--all", "do COMMAND on all servers (by default)") { server_switch = ctl.commands.keys }
+
+  ctl.commands.each do |name, args|
+    opt.on("-#{args['short']}", "--#{name.to_s}", "do COMMAND on #{name.to_s} server") do
+      server_switch.push name
+    end
+  end
+
+  opt.on "-c CONFIG_FILE", "--config", "Read an alternate YAML configuration file (default is created in ~/.rctl.rc.yaml)." do |f|
+    options[:config] = File.expand_path(f)
+  end
+  
+  opt.on("-v", "--version", "display version information") do
+    puts "Rctl CTL version #{Rctl::CtlVersion::VERSION}"
+    exit
+  end
+  
+  opt.on("-h", "--help", "display this help message") do
+    usage = true
+  end
+end
+parser.parse!
+
+if usage || !ARGV[0] || ARGV[0] == 'help'
+  puts parser
+else
+  server_switch = ctl.commands.keys if server_switch.empty?
+  exec ctl.generate(ARGV[0], server_switch)
+end

data/lib/rctl.rb

@@ -0,0 +1,93 @@
+require 'yaml'
+require 'pathname'
+
+module Rctl
+  class Ctl
+    attr_reader :commands
+    
+    def initialize(options = {})
+      if(options[:config])
+        @commands = YAML.load_file(options[:config])
+      else
+        if(!File.exists? "~/.rctl.rc.yml")
+          @commands = YAML.load_file(Pathname(__FILE__) + '../rctl/default_config.yml')
+          File.open(File.expand_path("~/.rctl.rc.yml"), 'w') do |file|
+            YAML.dump(@commands, file)
+          end
+        else
+          @commands = YAML.load_file("~/.rctl.rc.yml")
+        end
+      end
+      @start_order = define_start_order
+    end
+    
+    # launch command on selected services.
+    # return 0 if everything went ok.
+    def generate(command = :start, services = {})
+      generate_command(command, services)
+    end
+    
+  private
+    def define_start_order(remaining_server = nil)
+      if(remaining_server.nil?)
+        remaining_server = @commands.keys
+      end
+      ordered = []
+      to_order = []
+      remaining_server.each do |name|
+        if (@commands[name]['dependencies'].nil? || @commands[name]['dependencies'].empty?) ||
+           (@commands[name]['dependencies'].is_a?(Array) && (@commands[name]['dependencies'].select {|d| remaining_server.include?(d)}).count == 0) ||
+           (! remaining_server.include?(@commands[name]['dependencies'].to_s))
+          ordered.push name
+        else
+          to_order.push name
+        end
+      end
+      
+      if remaining_server.count == 0
+        return ordered
+      elsif remaining_server.count == to_order.count
+          throw Exception.new("Circular dependencies in remaining server #{to_order}")
+      else
+        ordered.concat define_start_order(to_order)
+      end
+    end
+    
+    def help
+      return "echo '#{@parser.to_s}'"
+    end
+    
+    def generate_command(command, services)
+      sw = @start_order.select {|s| services.include?(s)}
+      sw.reverse! if command == 'stop'
+      return(sw.map do |name|
+        "echo '---> #{command} #{name.to_s}' && #{@commands[name][command]}"
+      end.join(" && "))
+    end
+    
+    # Build the parser given @@server_desc data
+    def build_parser
+      @parser = OptionParser.new do |opt|
+        opt.banner = "Usage: rctl COMMAND [OPTIONS]"
+        opt.separator ""
+        opt.separator "Commands"
+        opt.separator "    start   start server(s)"
+        opt.separator "    stop    stop server(s)"
+        opt.separator ""
+        opt.separator "Options"
+
+        opt.on("-a", "--all", "do COMMAND on all servers (by default)") { all }
+
+        @commands.each do |name, args|
+          opt.on("-#{args[:short]}", "--#{name.to_s}", "do COMMAND on #{name.to_s} server") do
+            @server_switch.push name
+          end
+        end
+
+        opt.on("-h", "--help", "help") do
+          @usage = true
+        end
+      end
+    end    
+  end
+end

data/lib/rctl/default_config.yml

@@ -0,0 +1,28 @@
+mysql:
+    short: m
+    start: 'mysql.server start'
+    stop: 'mysql.server stop'
+mongodb:
+    short: o
+    start: 'mongod run --config /usr/local/etc/mongod.conf --fork --logpath /usr/local/var/log/mongodb/mongod.log'
+    stop: 'killall -2 mongod'
+solr:
+    dependencies:
+        - mongodb
+        - mysql
+    short: s
+    start: 'bundle exec rake sunspot:solr:start'
+    stop: "ps aux | grep 'solr' | grep -v 'grep' | awk '{print $2}' | xargs kill -2"
+unicorn:
+    dependencies:
+        - solr
+        - mongodb
+        - mysql
+    short: u
+    start: 'bundle exec unicorn -c ./config/unicorn.conf.rb -D'
+    stop: "ps aux | grep 'unicorn master' | grep -v 'grep' | awk '{print $2}' | xargs kill -QUIT"
+nginx:
+    dependencies: unicorn
+    short: n
+    start: 'sudo /usr/local/sbin/nginx'
+    stop: "ps aux | grep 'nginx' | grep -v 'grep' | awk '{print $2}' | sudo xargs kill -QUIT"

data/lib/rctl/version.rb

@@ -0,0 +1,5 @@
+module Rctl
+  module CtlVersion
+    VERSION = "0.0.1"
+  end
+end

metadata

@@ -0,0 +1,113 @@
+--- !ruby/object:Gem::Specification
+name: rctl
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Olivier Amblet
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-06-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: guard
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: guard-minitest
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: growl
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email: olivier@amblet.net
+executables:
+- rctl
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/rctl
+- lib/rctl/default_config.yml
+- lib/rctl/version.rb
+- lib/rctl.rb
+homepage: http://rctl.com
+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.24
+signing_key: 
+specification_version: 3
+summary: rctl let you start and stop your service stack in one command.
+test_files: []