runit-man 1.2

data/bin/runit-man.rb

@@ -0,0 +1,19 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'optparse'
+
+require 'runit-man/app'
+
+RunitMan.set :active_services_directory, '/etc/service'
+RunitMan.set :all_services_directory,    '/etc/sv'
+
+OptionParser.new { |op|
+  op.on('-s server') { |val| RunitMan.set :server, val }
+  op.on('-p port')   { |val| RunitMan.set :port, val.to_i }
+  op.on('-b addr')   { |val| RunitMan.set :bind, val }
+  op.on('-a active_services_directory (/etc/service by default)') { |val| RunitMan.set :active_services_directory, val }
+  op.on('-f all_services_directory (/etc/sv by default)')         { |val| RunitMan.set :all_services_directory, val }
+}.parse!(ARGV.dup)
+
+RunitMan.run!

data/i18n/en.yml

@@ -0,0 +1,35 @@
+runit:
+  label: runit web management tool
+  title: "%1 - %2"
+  header: "%1: %2"
+  error: Error encountered while accessing resource
+  loading: "Please wait while loading…"
+  footer: "State of services updated automatically every %1 seconds."
+  services:
+    table:
+      caption: Services
+      headers:
+        pid: PID
+        name: Name
+        stat: Status
+        actions: Actions
+        log_file: Log file
+      values:
+        log_hint: Open tail 100 lines of %1 service log in new window
+        log_absent: Absent
+      footer: Updated at %1
+      subst:
+        inactive: <span class="inactive">inactive</span>
+        down: <span class="down">down<span>
+        run: <span class="run">run<span>
+      actions:
+        start: Start
+        stop: Stop
+        restart: Restart
+        switch_down: Deactivate
+        switch_up: Activate
+    log:
+      title: "Tail 100 lines of service %1 log - %2"
+      header: "Tail 100 lines of service <strong>%1</strong> log: %2"
+      updated: "Page updated at %1"
+      reload: Reload

data/i18n/ru.yml

@@ -0,0 +1,37 @@
+runit:
+  label: runit - управляющий web-интерфейс
+  title: "%1 - %2"
+  header: "%1: %2"
+  error: Ошибка обращения к ресурсу
+  loading: "Идёт загрузка…"
+  footer: "Состояние сервисов обновляется автоматически каждые %1 секунд."
+  services:
+    table:
+      caption: Сервисы
+      headers:
+        pid: PID
+        name: Название
+        stat: Состояние
+        actions: Действия
+        log_file: Лог
+      values:
+        log_hint: Отрыть последние 100 строк лога сервиса %1 в новом окне
+        log_absent: Отсутствует
+      footer: Обновлено %1
+      subst:
+        inactive: <span class="inactive">неактивен</span>
+        down: <span class="down">остановлен</span>
+        run: <span class="run">работает</span>
+        got: получен
+        want: ожидается
+      actions:
+        start: Запустить
+        stop: Остановить
+        restart: Перезапустить
+        switch_down: Выключить
+        switch_up: Включить
+    log:
+      title: "Последние 100 строк лога сервиса %1 - %2"
+      header: "Последние 100 строк лога сервиса <strong>%1</strong>: %2"
+      updated: "Страница обновлена %1"
+      reload: Обновить страницу

data/lib/runit-man/app.rb

@@ -0,0 +1,77 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+
+require 'sinatra/base'
+require 'sinatra/r18n'
+require 'runit-man/erb-to-erubis'
+require 'runit-man/helpers'
+
+R18n::Filters.on :variables
+
+CONTENT_TYPES = {
+  :html => 'text/html',
+  :css  => 'text/css',
+  :js   => 'application/x-javascript',
+  :json => 'application/json'
+}.freeze
+
+
+class RunitMan < Sinatra::Base
+  set :environment,  :production
+  set :static,       true
+  set :logging,      true
+  set :dump_errors,  true
+  set :raise_errors, false
+  set :root,         File.expand_path(File.join('..', '..'), File.dirname(__FILE__))
+
+  register Sinatra::R18n
+
+  helpers do
+    include Helpers
+  end
+
+  before do
+    base_content_type = case request.env['REQUEST_URI']
+      when /\.css$/  then :css
+      when /\.js$/   then :js
+      when /\.json$/ then :json
+      else                :html
+    end
+    content_type CONTENT_TYPES[base_content_type], :charset => 'utf-8'
+  end
+
+  get '/' do
+    @scripts = [ 'jquery-1.4.1.min' ]
+    @title = host_name
+    erb :index
+  end
+
+  get '/services' do
+    partial :services
+  end
+
+  get '/:name/log' do |name|
+    srv = ServiceInfo[name]
+    return not_found if srv.nil? || !srv.logged?
+    @scripts = []
+    @title = t.runit.services.log.title(h(name), h(host_name))
+    erb :log, :locals => {
+      :name => name,
+      :text => `tail -n 100 #{srv.log_file_location}`
+    }
+  end
+
+  def log_action(name, text)
+    addr = request.env.include?('X_REAL_IP') ? request.env['X_REAL_IP'] : request.env['REMOTE_ADDR']
+    puts "#{addr} - - [#{Time.now}] \"Do #{text} on #{name}\""
+  end
+
+  post '/:name/:action' do |name, action|
+    srv = ServiceInfo[name]
+    action = "#{action}!".to_sym
+    return not_found if srv.nil? || !srv.respond_to?(action)
+    srv.send(action)
+    log_action(name, action)
+    ''
+  end
+end

data/lib/runit-man/erb-to-erubis.rb

@@ -0,0 +1,17 @@
+module Sinatra::Erb
+  def erb(content, options={})
+    begin
+      require 'erubis'
+      @@erb_class = Erubis::Eruby
+    rescue LoadError
+      require "erb"
+      @@erb_class = ::ERB
+    end
+    render(:erb, content, options)
+  end
+
+private
+  def render_erb(content, options = {})
+    @@erb_class.new(content).result(binding)
+  end
+end

data/lib/runit-man/helpers.rb

@@ -0,0 +1,47 @@
+require 'socket'
+require 'runit-man/service_info'
+require 'runit-man/partials'
+require 'sinatra/content_for'
+
+module Helpers
+  include Rack::Utils
+  include Sinatra::Partials
+  include Sinatra::ContentFor
+  alias_method :h, :escape_html
+
+  attr_accessor :even_or_odd_state
+
+  def host_name
+    unless @host_name
+      @host_name = Socket.gethostbyname(Socket.gethostname).first
+    end
+    @host_name
+  end
+
+  def service_infos
+    ServiceInfo.all
+  end
+
+  def service_action(name, action, label)
+    partial :service_action, :locals => {
+      :name   => name,
+      :action => action,
+      :label  => label
+    }
+  end
+
+  def even_or_odd
+    self.even_or_odd_state = !even_or_odd_state
+    even_or_odd_state
+  end
+
+  def stat_subst(s)
+    s.split(/\s/).map do |s|
+      if s =~ /(\w+)/ && t.runit.services.table.subst[$1].translated?
+        s.sub(/\w+/, t.runit.services.table.subst[$1].to_s)
+      else
+        s
+      end
+    end.join(' ')
+  end
+end

data/lib/runit-man/log_location_cache.rb

@@ -0,0 +1,73 @@
+class LogLocationCache
+  TIME_LIMIT = 6000
+
+  def initialize
+    clear
+  end
+
+  def [](pid)
+    pid = pid.to_i
+    unless pids.include?(pid)
+      set_pid_log_location(pid, get_pid_location(pid))
+    end
+    pids[pid][:value]
+  end
+
+private
+  attr_accessor :query_counter
+  attr_accessor :pids
+
+  def clear
+    self.query_counter = 0
+    self.pids = {}
+    self
+  end
+
+  def remove_old_values
+    self.query_counter = query_counter + 1
+    if query_counter < 1000
+      return
+    end
+    self.query_counter = 0
+    limit = Time.now - TIME_LIMIT
+    pids.keys.each do |pid|
+      if pids[pid][:time] < limit
+        pids.remove(pid)
+      end
+    end
+    self
+  end
+
+  def get_pid_location(lpid)
+    folder = log_folder(lpid)
+    return nil if folder.nil?
+    File.join(folder, 'current')
+  end
+
+  def log_command(lpid)
+    return nil if lpid.nil?
+    ps_output = `ps -o args -p #{lpid} 2>&1`.split("\n")
+    ps_output.shift
+    cmd = ps_output.first
+    cmd = cmd.chomp unless cmd.nil?
+    cmd = nil if cmd == ''
+    cmd
+  end
+
+  def log_folder(lpid)
+    cmd = log_command(lpid)
+    return nil if cmd.nil?
+    args = cmd.split(/\s+/).select { |arg| arg !~ /^\-/ }
+    return nil if args.shift != 'svlogd'
+    args.shift
+  end
+
+  def set_pid_log_location(pid, log_location)
+    remove_old_values
+    pids[pid.to_i] = {
+      :value => log_location,
+      :time  => Time.now
+    }
+    self
+  end
+end

data/lib/runit-man/partials.rb

@@ -0,0 +1,20 @@
+# stolen from http://github.com/cschneid/irclogger/blob/master/lib/partials.rb
+#   and made a lot more robust by me
+# this implementation uses erb by default. if you want to use any other template mechanism
+#   then replace `erb` on line 13 and line 17 with `haml` or whatever
+module Sinatra::Partials
+  def partial(template, *args)
+    template_array = template.to_s.split('/')
+    template = template_array[0..-2].join('/') + "/_#{template_array[-1]}"
+    options = args.last.is_a?(Hash) ? args.pop : {}
+    options.merge!(:layout => false)
+    if collection = options.delete(:collection) then
+      collection.inject([]) do |buffer, member|
+        buffer << erb(:"#{template}", options.merge(:layout =>
+        false, :locals => {template_array[-1].to_sym => member}))
+      end.join("\n")
+    else
+      erb(:"#{template}", options)
+    end
+  end
+end

data/lib/runit-man/service_info.rb

@@ -0,0 +1,156 @@
+require 'runit-man/log_location_cache'
+
+class ServiceInfo
+  attr_reader :name
+
+  def initialize(a_name)
+    @name = a_name
+  end
+
+  def logged?
+    File.directory?(log_supervise_folder)
+  end
+
+  def stat
+    return 'inactive' unless supervise?
+    r = 'indeterminate'
+    File.open(File.join(supervise_folder, 'stat'), 'r') { |f| r = f.gets }
+    r
+  end
+
+  def active?
+    File.directory?(active_service_folder) || File.symlink?(active_service_folder)
+  end
+
+  def switchable?
+    File.symlink?(active_service_folder) || File.directory?(inactive_service_folder)
+  end
+
+  def run?
+    stat =~ /\brun\b/
+  end
+
+  def up!
+    send_signal! :u
+  end
+
+  def down!
+    send_signal! :d
+  end
+
+  def switch_down!
+    down!
+    File.unlink(active_service_folder)
+  end
+
+  def switch_up!
+    File.symlink(inactive_service_folder, active_service_folder)
+  end
+
+  def restart!
+    down!
+    up!
+  end
+
+  def pid
+    r = nil
+    if supervise?
+      File.open(File.join(supervise_folder, 'pid'), 'r') { |f| r = f.gets }
+    end
+    r = r.chomp unless r.nil?
+    r = nil if r == ''
+    r
+  end
+
+  def log_pid
+    r = nil
+    if logged?
+      File.open(File.join(log_supervise_folder, 'pid'), 'r') { |f| r = f.gets }
+    end
+    r = r.chomp unless r.nil?
+    r = nil if r == ''
+    r
+  end
+
+  def log_file_location
+    rel_path = self.class.log_location_cache[log_pid]
+    return nil if rel_path.nil?
+    File.expand_path(rel_path, log_run_folder)
+  end
+
+private
+  def inactive_service_folder
+    File.join(RunitMan.all_services_directory, name)
+  end
+
+  def active_service_folder
+    File.join(RunitMan.active_services_directory, name)
+  end
+
+  def supervise_folder
+    File.join(active_service_folder, 'supervise')
+  end
+
+  def log_run_folder
+    File.join(active_service_folder, 'log')
+  end
+
+  def log_supervise_folder
+    File.join(log_run_folder, 'supervise')
+  end
+
+  def supervise?
+    File.directory?(supervise_folder)
+  end
+
+  def send_signal!(signal)
+    return unless supervise?
+    File.open(File.join(supervise_folder, 'control'), 'w') { |f| f.print signal.to_s }
+  end
+
+  class << self
+    def all
+      all_service_names.sort.map do |name|
+        ServiceInfo.new(name)
+      end
+    end
+
+    def [](name)
+      all_service_names.include?(name) ? ServiceInfo.new(name) : nil
+    end
+
+    def log_location_cache
+      unless @log_location_cache
+        @log_location_cache = LogLocationCache.new
+      end
+      @log_location_cache
+    end
+
+  private
+    def itself_or_parent?(name)
+      name == '.' || name == '..'
+    end
+
+    def active_service_names
+      return [] unless File.directory?(RunitMan.active_services_directory)
+      Dir.entries(RunitMan.active_services_directory).reject do |name|
+        full_name = File.join(RunitMan.active_services_directory, name)
+        itself_or_parent?(name) || (!File.symlink?(full_name) && !File.directory?(full_name))
+      end
+    end
+
+    def inactive_service_names
+      return [] unless File.directory?(RunitMan.all_services_directory)
+      actives = active_service_names
+      Dir.entries(RunitMan.all_services_directory).reject do |name|
+        full_name = File.join(RunitMan.all_services_directory, name)
+        itself_or_parent?(name) || !File.directory?(full_name) || actives.include?(name)
+      end
+    end
+
+    def all_service_names
+      (active_service_names + inactive_service_names)
+    end
+
+  end
+end