Contents

require 'erubis'
require 'sys/proctable'
require 'sequel'

TEMPLATES = File.dirname(__FILE__) + '/../view/'
DEFAULT_BASE = TEMPLATES + 'base.eruby'
DEFAULT_TITLE = 'The administrator tool'
HOME = ENV["HOME"] 
RADMIN_HOME = HOME + "/.radministrator"
DBPATH = RADMIN_HOME + "/data.db"
DB = Sequel.connect("sqlite://#{DBPATH}")

def init_db
  init_procs
end

def init_procs
  return if DB.table_exists? :monitored_procs
  DB.create_table :monitored_procs do
    primary_key :id
    String :cmdline    
  end
end

def format_to_html(output)
  output.gsub("\n", "<br />")
end

def _render(template_name, options = {}, title = DEFAULT_TITLE,  base_name = DEFAULT_BASE)
  html = render_to_s(template_name, options)
  base_template = File.read(base_name)
  base = Erubis::Eruby.new(base_template)
  base.result(:title => title, :html => html)
end

def render_to_s(template_name, options = {})
  template = File.read(TEMPLATES + template_name.to_s + '.eruby')
  Erubis::Eruby.new(template).result(options)
end

def register_server
  Dir.mkdir RADMIN_HOME unless File.exists? RADMIN_HOME
  f = File.open(RADMIN_HOME + '/process.pid', "w")
  f.write(Process.pid)
end

def add_proc_to_monitor(procname)
  procs = DB[:monitored_procs]
  return if procs.where(:cmdline => procname).count > 0
  procs.insert(:cmdline => procname)  
end

def del_proc_from_monitor(procname)
  procs = DB[:monitored_procs]
  return if procs.where(:cmdline => procname).count == 0
  procs.where(:cmdline => procname).delete
end

def monitored_procs
  procnames = DB[:monitored_procs].all.map{|p| p[:cmdline]}.join("|")
  return [] if procnames == ""
  all_procs.select{|p| p.cmdline =~ /#{procnames}/i}
end

def all_procs
  Sys::ProcTable.ps
end

def kill9(pid)
  `kill -9 #{pid}`
end
init_db

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
admin-helper-0.0.1	lib/admin-helper/model.rb