freyr 0.3.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.irbrc

@@ -0,0 +1,4 @@
+require File.dirname(__FILE__) + '/lib/freyr.rb'
+include Freyr
+
+Service.add_file("Freyrfile")

data/Freyrfile

@@ -0,0 +1,3 @@
+service :sleep do
+  start 'sudo sleep 20'
+end

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Tal Atlas
+
+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.

data/README.markdown

@@ -0,0 +1,73 @@
+# freyr
+
+Manage all your application's services.
+
+Freyr will start any services you want in the background and capture stdout into a log file for you to view whenever you want.
+Great for if you are working on an app which has peripheral services that need to be run.
+
+## Basic use
+
+    $ gem install freyr
+
+Install freyr it'll also give you a binary named freyr.
+
+To use it you can create a Freyrfile or .freyrrc. It will automatically look for those files in whatever your current
+directory is. It will also look for a .freyrrc in your home directory. You can also specify a file manually.
+
+The file has this basic structure:
+
+    service :memcache do
+      start 'memcached -vvv'
+    end
+
+That will allow you to start, stop, restart, and watch the STDOUT of a service.
+
+Basic usage can be done like
+
+    $ freyr start memcache
+
+Just calling `freyr` or `freyr list` will output all of the services being tracked and their status.
+
+## Some more options
+
+A few more of the options for defining a service are:
+
+* start/stop/restart - Commands to be run when you send any of those individual commands. If stop is not defined it will
+default to sending a KILL signal, if restart is not defined it will run stop then start.
+* dir - The directory to run the file in, defaults to '/' (considering defaulting to current directory)
+* proc_match - a string or regular expression to check for after starting. This is useful if freyr can't capture the pid
+from a normal launch, eg when launching though a shell script.
+* ping - A url to ping to alert you when the service is up and running, will let you know if there's an error.
+* group - assign this service to a group. Calling start/stop/restart on the group will run across all of the members
+
+See a complete list of definition options [here](https://github.com/Talby/freyr/wiki/Service-Definition-Options)
+
+These options also give you some more options with to use with the CLI
+
+* `freyr update_pid [service]` - Updates the pid stored by checking for the proc_match. Useful if you started the service
+by some method other than freyr or it restarted with a different pid.
+* `freyr ping` - Tries to ping the service once
+* `freyr -p` - Shows you all services and the status of pings for each.
+
+To see a complete list of options type in `freyr help`.
+
+## TODO
+
+* Growl notifications
+* Better error handling
+* Service definition validation
+* Plugin architecture
+
+## Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+## Copyright
+
+Copyright (c) 2010 Tal Atlas. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,51 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "freyr"
+    gem.summary = %Q{Service manager and runner}
+    gem.description = %Q{Define all services you need to run and this will launch,daemonize,and monitor them for you.}
+    gem.email = "me@tal.by"
+    gem.homepage = "http://github.com/Talby/freyr"
+    gem.authors = ["Tal Atlas"]
+    gem.add_development_dependency "rspec", ">= 1.2.9"
+    gem.add_development_dependency "yard", ">= 0"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+    gem.add_dependency 'thor', ">= 0.10"
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+begin
+  require 'spec/rake/spectask'
+  Spec::Rake::SpecTask.new(:spec) do |spec|
+    spec.libs << 'lib' << 'spec'
+    spec.spec_files = FileList['spec/**/*_spec.rb']
+  end
+  
+  
+  Spec::Rake::SpecTask.new(:rcov) do |spec|
+    spec.libs << 'lib' << 'spec'
+    spec.pattern = 'spec/**/*_spec.rb'
+    spec.rcov = true
+  end
+  
+  task :spec => :check_dependencies
+  
+  task :default => :spec
+rescue LoadError
+  puts "Expecting rspec to be installed. Please install with: gem install rspec"
+end
+
+begin
+  require 'yard'
+  YARD::Rake::YardocTask.new
+rescue LoadError
+  task :yardoc do
+    abort "YARD is not available. In order to run yardoc, you must: sudo gem install yard"
+  end
+end

data/VERSION

@@ -0,0 +1 @@
+0.3.0

data/bin/freyr

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+$LOAD_PATH.unshift File.dirname(__FILE__) + '/../lib'
+require 'freyr'
+require 'freyr/cli'
+
+Freyr::CLI.start

data/freyr.gemspec

@@ -0,0 +1,73 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{freyr}
+  s.version = "0.3.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Tal Atlas"]
+  s.date = %q{2011-01-24}
+  s.default_executable = %q{freyr}
+  s.description = %q{Define all services you need to run and this will launch,daemonize,and monitor them for you.}
+  s.email = %q{me@tal.by}
+  s.executables = ["freyr"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+    "README.markdown"
+  ]
+  s.files = [
+    ".document",
+    ".irbrc",
+    "Freyrfile",
+    "LICENSE",
+    "README.markdown",
+    "Rakefile",
+    "VERSION",
+    "bin/freyr",
+    "freyr.gemspec",
+    "lib/freyr.rb",
+    "lib/freyr/cli.rb",
+    "lib/freyr/cli/helpers.rb",
+    "lib/freyr/cli/launching.rb",
+    "lib/freyr/cli/management.rb",
+    "lib/freyr/cli/monitor.rb",
+    "lib/freyr/command.rb",
+    "lib/freyr/pinger.rb",
+    "lib/freyr/service.rb",
+    "lib/freyr/service_info.rb",
+    "spec/freyr_spec.rb",
+    "spec/spec.opts",
+    "spec/spec_helper.rb"
+  ]
+  s.homepage = %q{http://github.com/Talby/freyr}
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.7}
+  s.summary = %q{Service manager and runner}
+  s.test_files = [
+    "spec/freyr_spec.rb",
+    "spec/spec_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<rspec>, [">= 1.2.9"])
+      s.add_development_dependency(%q<yard>, [">= 0"])
+      s.add_runtime_dependency(%q<thor>, [">= 0.10"])
+    else
+      s.add_dependency(%q<rspec>, [">= 1.2.9"])
+      s.add_dependency(%q<yard>, [">= 0"])
+      s.add_dependency(%q<thor>, [">= 0.10"])
+    end
+  else
+    s.add_dependency(%q<rspec>, [">= 1.2.9"])
+    s.add_dependency(%q<yard>, [">= 0"])
+    s.add_dependency(%q<thor>, [">= 0.10"])
+  end
+end
+

data/lib/freyr/cli/helpers.rb

@@ -0,0 +1,113 @@
+module Freyr
+  class CLI < Thor
+    
+  private
+    # Ugh, this is ugly, colorizing stuff is rough
+    def list_all_services(args={})
+      args[:highlight_name] = (args[:highlight_name]||[]).collect {|n| n.to_s}
+      args[:highlight_state] = (args[:highlight_state]||[]).collect {|n| n.to_s}
+      
+      max_length = 0
+      
+      groups_join = '|'
+      
+      lengths = Service.s.collect do |s|
+        n = "   #{s.name}(#{s.groups.join(groups_join)})"
+        max_length = n.length if n.length > max_length
+        n.length
+      end
+      
+      max_length += 3 # min distance between name and group
+      
+      strs = []
+      Service.s.each_with_index do |s,i|
+        str = '  '
+        if s.sudo
+          str << set_color('*', :yellow)
+        else
+          str << ' '
+        end
+        
+        name = set_color(s.name, :blue)
+        if args[:highlight_name].include?(s.name.to_s)
+          name = set_color(name, :on_white)
+        end
+        str << name
+        str << " "*(max_length - lengths[i])
+        if s.groups.empty?
+          str << '  '
+        else
+          str <<  '('
+          s.groups.each do |g|
+            str << set_color(g,:red)
+            next if g == s.groups.last
+            str << groups_join
+          end
+          str << ')'
+        end
+        
+        str << ' - '
+        if s.alive?
+          state = set_color(' Alive ', :green, false)
+        else
+          state = set_color(' Dead ', :red, true)
+        end
+        
+        if args[:highlight_state].include?(s.name.to_s)
+          state = set_color(state, :on_white)
+        end
+        
+        str << state
+        
+        if args[:ping] && s.alive? && pinger = s.ping!
+          png = "(#{pinger.code})"
+          if pinger.success?
+            png = set_color(png, :green, false)
+          elsif pinger.server_error?
+            png = set_color(png, :yellow, true)
+          else
+            png = set_color(png, :red, true)
+          end
+          
+          str << png
+        elsif args[:ping]
+          str << ' '*'(123)'.size
+        end
+        
+        str << "\n" #if str =~ /\s$/ # Thor's display only adds a new line if the last char isn't a space
+        
+        strs << str
+      end
+      
+      strs
+    end
+    
+    def set_color *args
+      @shell.set_color(*args)
+    end
+    
+    def get_from_name name
+      name ||= Dir.pwd.match(/.*\/(.+)/)[1]
+      
+      if options.namespace && s = Service["#{options.namespace}:#{name}"].first
+        [s] # only pickng one because if it's namespaced it's not a group
+      else
+        Service[name]
+      end
+    end
+    
+    def get_services
+      if options['config-file'] && !options['config-file'].empty?
+        if File.exist?(options['config-file'])
+          Service.add_file(options['config-file'])
+        else
+          say("Can't find file #{options['config-file']}",:red) 
+        end
+      end
+      
+      ['Freyrfile','.freyrrc','~/.freyrrc'].each do |f|
+        Service.add_file(f)
+      end unless options['ignore-local']
+    end
+  end
+end

data/lib/freyr/cli/launching.rb

@@ -0,0 +1,74 @@
+module Freyr
+  class CLI < Thor
+    # TODO: DRY up start/restart/stop commands
+    
+    desc 'start [SERVICE=dirname]', 'Start particular service'
+    def start(name=nil)
+      services = get_from_name(name)
+      if !services.empty?
+        names = services.collect {|s| s.name}
+        say "Starting the " << set_color(names.join(', '), :blue) << ' services'
+        
+        changed_names = services.collect do |s|
+          begin
+            pid = s.start!
+            s.name if pid
+          rescue AdminRequired
+            say "Please run in sudo to launch #{s.name}.", :red
+            nil
+          end
+        end.compact
+        
+        list_all_services(:highlight_state => changed_names).each {|l| say(l)}
+      else
+        say "Can't find the #{name} service", :red
+      end
+    end
+    
+    desc 'stop [SERVICE=dirname]', 'Stop particular service'
+    def stop(name=nil)
+      services = get_from_name(name)
+      if !services.empty?
+        names = services.collect {|s| s.name}
+        say "Stopping the " << set_color(names.join(', '), :blue) << ' services'
+        changed_names = services.collect {|s| s.name if s.alive?}.compact
+        
+        services.each do |s|
+          begin
+            s.stop!
+          rescue AdminRequired
+            say "Please run in sudo to stop #{s.name}.", :red
+          end
+        end
+        
+        list_all_services(:highlight_state => changed_names).each {|l| say(l)}
+      else
+        say "Can't find the #{name} service", :red
+      end
+    end
+    
+    desc 'restart [SERVICE=dirname]', 'restart particular service'
+    def restart(name=nil)
+      services = get_from_name(name)
+      if !services.empty?
+        names = services.collect {|s| s.name}
+        say "Restarting the " << set_color(names.join(', '), :blue) << ' services'
+        
+        services.each do |s|
+          begin
+            s.restart!
+          rescue AdminRequired
+            say "Please run in sudo to launch #{s.name}.", :red
+            name.delete(s.name)
+            nil
+          end
+        end
+        
+        list_all_services(:highlight_state => names).each {|l| say(l)}
+      else
+        say "Can't find the #{name} service", :red
+      end
+    end
+    
+  end
+end

data/lib/freyr/cli/management.rb

@@ -0,0 +1,21 @@
+module Freyr
+  class CLI < Thor
+    
+    desc 'update_pid [SERVICE=dirname]', 'Update pid from proc_match (good to use if service already launched)'
+    def update_pid(name=nil)
+      services = get_from_name(name)
+      if s = services.first
+        if pid = s.command.update_pid
+          say "Updated pid for "<< set_color(s.name,:blue) << ' to ' << set_color(pid,:red)
+        elsif s.proc_match
+          say "Couldn't find pid for process matcher #{s.proc_match.inspect}", :red
+        else
+          say "Service #{s.name} doesn't have a value for proc_match set.", :red
+        end
+      else
+        say "Couldn't find service with name #{name}.", :red
+      end
+    end
+    
+  end
+end

data/lib/freyr/cli/monitor.rb

@@ -0,0 +1,58 @@
+module Freyr
+  class CLI < Thor
+    
+    desc 'list', 'lists all available services:'
+    method_option :ping, :type => :boolean, :default => false, :aliases => '-p', :desc => 'also ping each service which can ping'
+    def list
+      strs = list_all_services(:ping => options.ping?)
+      
+      if strs.empty?
+        say "No services available", :red
+      else
+        say "List of all available services (#{set_color('*', :yellow)} denotes root proc)"
+        strs.each_with_index do |s,i|
+          say s
+        end
+      end
+    end
+    
+    desc 'tail [SERVICE=dirname]', 'read stdout of the service'
+    method_option :lines, :type => :numeric, :default => 50, :desc => 'Number of lines to show on initial tail'
+    method_option :'no-follow', :type => :boolean, :default => false, :desc => 'Disable auto follow, just print tail and exit'
+    def tail(name=nil)
+      services = get_from_name(name)
+      if !services.empty?
+        
+        services.first.tail!(options.lines, !options['no-follow'])
+      else
+        say "Can't find the #{name} service", :red
+      end
+    end
+    
+    desc 'ping', 'see the response from pinging the url'
+    def ping(name=nil)
+      service = get_from_name(name).first
+      
+      if service
+        if service.ping
+          pinger = Pinger.new(service)
+          resp = pinger.ping
+          if pinger.success?
+            say "Up and running", :green
+          elsif pinger.server_error?
+            say "500 Error"
+          elsif resp
+            say "Returned #{resp.code} code", :red
+          else
+            say "Couldn't reach service", :red
+          end
+        else
+          say 'No url to ping for this service'
+        end
+      else
+        say "Can't find the #{name} service", :red
+      end
+    end
+    
+  end
+end

data/lib/freyr/cli.rb

@@ -0,0 +1,27 @@
+require 'thor'
+%w{launching management monitor helpers}.each do |f|
+  require File.expand_path(File.dirname(__FILE__)+"/cli/#{f}.rb")
+end
+
+module Freyr
+  class CLI < Thor
+    include Thor::Actions
+    
+    def initialize(*)
+      super
+      get_services
+    end
+    
+    default_task :list
+    class_option :'config-file', :desc => 'config file to use', :type => :string
+    class_option :'ignore-local', :desc => "don't use the local Freyrfile or .freyrrc", :type => :boolean, :default => false
+    class_option :namespace, :type => :string, :desc => 'namespace to look in'
+    map "-v" => :version
+    
+    desc 'version', 'displays current gem version'
+    def version
+      say Freyr::VERSION
+    end
+    
+  end
+end

data/lib/freyr/command.rb

@@ -0,0 +1,269 @@
+module Freyr
+  
+  class Command
+    extend Forwardable
+    
+    ROOT_PIDS = '/var/run/freyr'
+    FREYR_PIDS = ENV['USER'] == 'root' ? ROOT_PIDS : File.expand_path('.freyr', '~')
+    
+    if !File.exist?(FREYR_PIDS)
+      Dir.mkdir(FREYR_PIDS)
+    elsif !File.directory?(FREYR_PIDS)
+      File.delete(FREYR_PIDS)
+      Dir.mkdir(FREYR_PIDS)
+    end
+    
+    ServiceInfo::ATTRS.each do |meth|
+      define_method(meth) do |*args|
+        @service.__send__(meth,*args) if @service
+      end
+    end
+    
+    attr_reader :command, :name, :service
+    
+    def initialize(name, command=nil, args = {})
+      if name.is_a?(Service)
+        @service = name
+        @name = service.name
+        @command = service.start_command
+        @env = service.env
+      else
+        @name = name
+        @command = command
+        
+        @env = args[:env]
+      end
+    end
+    
+    def pid_file
+      File.join(file_dir,"#{@name}.pid")
+    end
+    
+    def file_dir
+      admin? ? ROOT_PIDS : FREYR_PIDS
+    end
+    
+    def read_pid
+      return unless File.exist?(pid_file)
+      p = File.open(pid_file).read
+      p ? p.to_i : nil
+    end
+    
+    def pid(force = false)
+      @pid = nil if force
+      @pid ||= read_pid
+    end
+    
+    def alive?
+      return unless pid
+      Process.getpgid(pid)
+      true
+    rescue Errno::ESRCH
+      retried = false unless defined?(retried)
+      
+      if !retried && @pid = pid_from_procname
+        save
+        retried = true
+        retry
+      end
+      
+      if pid(true)
+        File.delete(pid_file) unless admin? && !is_root?
+      end
+      
+      false
+    end
+    
+    def delete_if_dead
+      return if admin? && !is_root?
+      File.delete(pid_file) unless alive?
+    end
+    
+    def save
+      if File.exist?(pid_file)
+        old_pid = read_pid
+        begin
+          Process.kill('KILL', old_pid) if old_pid && old_pid.to_s != @pid.to_s
+        rescue Errno::ESRCH
+        end
+      end
+      
+      File.open(pid_file, 'w') {|f| f.write(@pid)}
+    end
+    
+    def run!
+      return unless command
+      kill! if alive?
+      
+      require_admin
+      
+      total_time = Time.now
+      
+      pid = spawn(command)
+        
+      @pid = pid
+        
+      Process.detach(@pid)
+      
+      if proc_match
+        puts "Waiting for pid from match of #{proc_match.inspect}"
+        
+        start = Time.now
+        until (pid = pid_from_procname) || (Time.now-start) > 40
+          print '.'; STDOUT.flush
+          sleep(0.2)
+        end
+        
+        raise "\nCouldnt find pid after 40 seconds" unless pid
+        
+        puts '*'
+        
+        @pid = pid
+      end
+      
+      puts "PID of new #{name} process is #{@pid}"
+      save
+      
+      if ping
+        pinger = Pinger.new(self)
+        
+        puts '',"Waiting for response from #{pinger.url}"
+        
+        # Move this pinger code somewhere else
+        start = Time.now
+        begin
+          print '.'; STDOUT.flush
+          pinger.ping
+          sleep(0.6)
+        end until pinger.server_probably_launched? || (Time.now-start) > 40 || !alive?
+        
+        if alive?
+          
+          if pinger.response
+            puts '*',"Last response recieved with code #{pinger.response.code}"
+          else
+            puts 'x',"Couldn't reach #{name} service"
+          end
+        else
+          puts 'x'
+        end
+      end
+      
+      if alive?
+        puts "Launch took about #{(Time.now-total_time).ceil} seconds"
+        
+        @pid
+      else
+        puts "#{name} service wasn't launched correctly. For details see: #{log}"
+        delete_if_dead
+      end
+    end
+    
+    def update_pid
+      if @pid = pid_from_procname
+        save
+        @pid
+      end
+    end
+    
+    def pid_from_procname
+      return unless proc_match
+      
+      pids = `ps -eo pid,command`.split("\n").inject({}) do |r, pid|
+        if m = pid.match(/^\s*(\d+)\s(.+)$/)
+          r[m[2]] = m[1].to_i
+        end
+        r
+      end
+      
+      if procline = pids.keys.find {|p| p.match(proc_match)}
+        pids[procline]
+      end
+    end
+    
+    def kill!(sig=nil)
+      require_admin
+      sig ||= stop_sig || 'KILL'
+      
+      if pid(true)
+        Process.kill(sig, pid)
+      end
+    end
+    
+    def restart!
+      require_admin
+      
+      if restart
+        chdir
+        system(restart)
+        update_pid
+      elsif restart_sig
+        kill!(restart_sig)
+      else
+        run!
+      end
+    end
+    
+    def admin?
+      sudo
+    end
+    
+    private
+    
+    def require_admin
+      raise AdminRequired if admin? && !is_root?
+    end
+    
+    def is_root?
+      ENV['USER'] == 'root'
+    end
+    
+    def chdir
+      Dir.chdir File.expand_path(dir||'/')
+    end
+    
+    def spawn(command)
+      fork do
+        # File.umask self.umask if self.umask
+        # uid_num = Etc.getpwnam(self.uid).uid if uid
+        # gid_num = Etc.getgrnam(self.gid).gid if gid
+        
+        # ::Dir.chroot(self.chroot) if self.chroot
+        ::Process.setsid
+        # ::Process.groups = [gid_num] if self.gid
+        # ::Process::Sys.setgid(gid_num) if self.gid
+        # ::Process::Sys.setuid(uid_num) if self.uid
+        chdir
+        $0 = "freyr - #{name} (#{command})"
+        STDIN.reopen "/dev/null"
+        if log_cmd
+          STDOUT.reopen IO.popen(log_cmd, "a")
+        elsif log
+          STDOUT.reopen log, "a"
+        end
+        if err_log_cmd
+          STDERR.reopen IO.popen(err_log_cmd, "a") 
+        elsif err_log && (log_cmd || err_log != log)
+          STDERR.reopen err_log, "a"
+        else
+          STDERR.reopen STDOUT
+        end
+        
+        # close any other file descriptors --- I think this is for anything after stderr
+        3.upto(256){|fd| IO::new(fd).close rescue nil}
+        
+        if @env && @env.is_a?(Hash)
+          @env.each do |(key, value)|
+            ENV[key] = value
+          end
+        end
+        
+        exec(command) unless command.empty?
+      end
+    end
+    
+  end
+  
+  class AdminRequired < Errno::EACCES; end
+  
+end

data/lib/freyr/pinger.rb

@@ -0,0 +1,52 @@
+require 'net/http'
+
+module Freyr
+  class Pinger
+    # URL to ping
+    attr_reader :url
+    # Time that the ping took
+    attr_reader :time
+    # Response object
+    attr_reader :response
+    
+    def initialize(command)
+      @command = command
+      @url = command.ping
+    end
+    
+    # The URI object for the given URL
+    def uri
+      @uri ||= URI.parse(url)
+    end
+    
+    # Send a ping to the url
+    def ping
+      t = Time.now
+      @response = Net::HTTP.get_response(uri)
+    rescue Errno::ECONNREFUSED
+    ensure
+      @time = Time.now-t
+      @response
+    end
+    
+    def code
+      response.code if response
+    end
+    
+    # Did the response recieve a success http code
+    def success?
+      response.is_a?(Net::HTTPSuccess)
+    end
+    
+    # Did the response recieve a 500 error
+    def server_error?
+      response.is_a?(Net::HTTPInternalServerError)
+    end
+    
+    # Did it recieve 2xx or 500
+    def server_probably_launched?
+      success? || server_error?
+    end
+    
+  end
+end

data/lib/freyr/service.rb

@@ -0,0 +1,111 @@
+module Freyr
+  class Service
+    extend Forwardable
+    
+    attr_reader :service_info, :command
+    
+    def initialize(s)
+      @service_info = s
+      @command = Command.new(self)
+    end
+    
+    def_delegators :@service_info, *ServiceInfo::ATTRS
+    def_delegator  :@service_info, :start, :start_command
+    
+    def env
+      service_info.env || {}
+    end
+    
+    def log
+      @service_info.log || File.join(command.file_dir,"#{name}.log")
+    end
+    
+    def start!
+      return unless start_command
+      command.run! unless alive?
+    end
+    
+    def stop!
+      command.kill! if start_command
+    end
+    
+    def restart!
+      command.restart!
+    end
+    
+    def is_group?(name)
+      groups.find {|g| g.to_s == name.to_s}
+    end
+    
+    def alive?
+      command.alive?
+    end
+    
+    def ping!
+      if ping
+        pinger = Pinger.new(self)
+        pinger.ping
+        pinger
+      end
+    end
+    
+    def tail!(size = 600, follow = true)
+      f = follow ? 'f' : ''
+      exec("tail -#{size}#{f} #{log}")
+    end
+    
+    def describe
+      %Q{#{name}(#{groups.join(',')}) - #{start_command}}
+    end
+    
+    def matches?(n)
+      n = n.to_s
+      return true if name.to_s == n
+      
+      also.find {|a| a.to_s == n}
+    end
+    
+    class << self
+      
+      def s
+        @all_services ||= []
+      end
+      
+      def names
+        @all_names ||= []
+      end
+      
+      def groups
+        @all_groups ||= []
+      end
+      
+      def selectors
+        names+groups
+      end
+      
+      def add_file f
+        s
+        
+        services = ServiceInfo.from_file(f).collect do |ser|
+          raise 'Cannot have two things of the same name' if selectors.include?(ser.name)
+          names << ser.name
+          @all_groups |= ser.groups
+          new(ser)
+        end
+        
+        @all_services += services
+      end
+      
+      def [](name)
+        if ser = s.find {|sr| sr.matches?(name)}
+          [ser]
+        else
+          s.select {|sr| sr.is_group?(name)}
+        end
+      end
+      
+    end
+    
+  end
+  
+end

data/lib/freyr/service_info.rb

@@ -0,0 +1,89 @@
+module Freyr
+  class ServiceInfo
+    attr_reader :groups
+    
+    ATTRS = [:name,:dir,:log_cmd,:log,:err_log_cmd,:err_log,:umask,
+              :uid,:gid,:chroot,:proc_match,:restart,:stop,:stop_sig,
+              :restart_sig,:sudo,:groups,:ping,:also]
+    
+    def initialize(name=nil, args={}, &block)
+      @groups = []
+      @also = []
+      if name.is_a?(Hash)
+        @name = name.keys.first
+        @groups << name[@name]
+      else
+        @name = name
+      end
+      
+      instance_eval(&block)
+    end
+    
+    def use_sudo
+      @sudo = true
+    end
+    
+    def group(*val)
+      @groups |= val
+    end
+    
+    def also_as(*val)
+      @also |= val
+    end
+    
+    MODIFIERS = {
+      :start => :_sudo_checker,
+      :restart => :_sudo_checker,
+      :stop => :_sudo_checker
+    }
+    
+    def method_missing key, val=nil
+      key = key.to_s.gsub(/\=$/,'').to_sym
+      
+      if val
+        val = send(MODIFIERS[key],val) if MODIFIERS[key]
+        instance_variable_set("@#{key}", val)
+      else
+        instance_variable_get("@#{key}")
+      end
+    end
+    
+    SUDO_MATCHER = /^sudo\s+/
+    
+  private
+    
+    # If someone doesn't explicitly not want the script running as admin
+    # and the val looks like it's supposed to be run in sudo remove sudo
+    # and force to be run as admin
+    def _sudo_checker(val)
+      if @admin != false && val =~ SUDO_MATCHER
+        val.gsub!(SUDO_MATCHER,'')
+        use_sudo
+      end
+      val.chomp
+    end
+    
+    class << self
+      
+      def from_file file
+        file = File.expand_path(file)
+        return [] unless File.exist?(file)
+        @added_services = []
+        instance_eval(File.open(file).read)
+        @added_services
+      end
+      
+    private
+      
+      def namespace name
+        @namespace = name
+      end
+      
+      def service name=nil, &blk
+        name = "#{@namespace}:#{name}" if @namespace
+        @added_services << new(name,&blk)
+      end
+    end
+    
+  end
+end

data/lib/freyr.rb

@@ -0,0 +1,9 @@
+require 'forwardable'
+
+module Freyr
+  VERSION = File.open(File.expand_path(File.dirname(__FILE__)+'/../VERSION')).read
+end
+
+%w{service_info service command pinger}.each do |f|
+  require File.expand_path(File.dirname(__FILE__)+"/freyr/#{f}.rb")
+end

data/spec/freyr_spec.rb

@@ -0,0 +1,7 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "Freyr" do
+  it "fails" do
+    fail "hey buddy, you should probably rename this file and start specing for real"
+  end
+end

data/spec/spec.opts

@@ -0,0 +1 @@
+--color

data/spec/spec_helper.rb

@@ -0,0 +1,9 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'freyr'
+require 'spec'
+require 'spec/autorun'
+
+Spec::Runner.configure do |config|
+  
+end

metadata

@@ -0,0 +1,134 @@
+--- !ruby/object:Gem::Specification 
+name: freyr
+version: !ruby/object:Gem::Version 
+  hash: 19
+  prerelease: false
+  segments: 
+  - 0
+  - 3
+  - 0
+  version: 0.3.0
+platform: ruby
+authors: 
+- Tal Atlas
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-01-24 00:00:00 -05:00
+default_executable: freyr
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 13
+        segments: 
+        - 1
+        - 2
+        - 9
+        version: 1.2.9
+  type: :development
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: yard
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: thor
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 31
+        segments: 
+        - 0
+        - 10
+        version: "0.10"
+  type: :runtime
+  version_requirements: *id003
+description: Define all services you need to run and this will launch,daemonize,and monitor them for you.
+email: me@tal.by
+executables: 
+- freyr
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.markdown
+files: 
+- .document
+- .irbrc
+- Freyrfile
+- LICENSE
+- README.markdown
+- Rakefile
+- VERSION
+- bin/freyr
+- freyr.gemspec
+- lib/freyr.rb
+- lib/freyr/cli.rb
+- lib/freyr/cli/helpers.rb
+- lib/freyr/cli/launching.rb
+- lib/freyr/cli/management.rb
+- lib/freyr/cli/monitor.rb
+- lib/freyr/command.rb
+- lib/freyr/pinger.rb
+- lib/freyr/service.rb
+- lib/freyr/service_info.rb
+- spec/freyr_spec.rb
+- spec/spec.opts
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/Talby/freyr
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Service manager and runner
+test_files: 
+- spec/freyr_spec.rb
+- spec/spec_helper.rb