Forum Debian Users Gang

#!/usr/bin/env ruby
require 'rubygems'
require 'jabber/bot'

# Tworzenie nowego publicznego Jabber::Bot
bot = Jabber::Bot.new(
  :jabber_id => 'jid@domena', 
  :password  => 'secret', 
  :master    => 'master1@domena',
  :master     => 'master2@domena',
  :master    => 'master3@domena',
  :is_public => true
)

#Dodanie kilku polecen
bot.add_command(
    :syntax            =>    'info',
    :description    =>    'Wyswietl informacje o pamieci oraz czasie zycia',
    :regex            =>    /^info$/
){info()}

#bot.add_command(
#    :syntax         =>  'pilnuj',
#    :description    =>  'Wyswietl informacje o pamieci oraz czasie zycia',
#    :regex          =>  /^pilnuj$/
#){pilnuj()}

bot.add_command(
 :syntax        => 'kill PID',
 :description   => 'Zabij proces o podanym PID',
 :regex         => /^kill\s+.+$/,
 :alias        => [ 
        :syntax => 'k PID', 
        :regex => /^k\s+.+$/ 
        ]
)do |sender, pid|
        #system("kill -9 '#{pid}'") 
        mms="Ale zabijanie nie dziala :P"
        "'#{mms}' Zabilem proces o ID: '#{pid}'"
end

bot.add_command(
    :syntax         =>  'topm <n>',
    :description    =>  'Pokaz n procesów zajmujących najwiecej pamięci',
    :regex          =>  /^topm\s+.+$/,
    :alias            =>    [
                    :syntax    =>    'm <n>',
                    :regex    =>    /^m\s+.+$/
                    ]
)do |sender, count|
    "'#{topm(count.to_i+1)}'"
end

bot.add_command(
    :syntax         =>  'topc <n>',
    :description    =>  'Pokaz n procesow najbardziej obciążających procesor',
    :regex          =>  /^topc\s+.+$/,
    :alias          =>  [
                    :syntax =>  'c <n>',
                    :regex  =>  /^c\s+.+$/
                    ]
)do |sender, count|
    "'#{topm(count.to_i+1)}'"
end

bot.add_command(
    :syntax         =>  'status',
    :description    =>  'Pokaż status najważniejszych usług',
    :regex          =>  /^status$/,
    :alias          =>  [
                    :syntax =>  's',
                    :regex  =>  /^s$/
                    ]
){status()}

bot.add_command(
    :syntax         =>  'who',
    :description    =>  'Pokaż zalogowanych uzytkownikow',
    :regex          =>  /^who$/,
    :alias          =>  [
                    :syntax =>  'w',
                    :regex  =>  /^w$/
                    ]
){who()}

bot.add_command(
    :syntax         =>  'apacz',
    :description    =>  'Restart Apache2',
    :regex          =>  /^apacz$/,
    :is_public        => true
){apacz()}

bot.add_command(
    :syntax         =>  'last LOGIN',
    :description    =>  'Pokaż ostatnie logowania podanego uzytkownia. Aby wyswietlić informacje o wszystkich uzytkownikach jako login należy podać ALL',
    :regex          =>  /^last\s+.+$/,
    :is_public        =>    true,
    :alias          =>  [
                    :syntax =>  'l LOGIN',
                    :regex  =>  /^l\s+.+$/
                    ]
)do |sender, user|
    "'#{rlast(user)}'"
end

bot.add_command(
    :syntax         =>  'restart USŁUGA',
    :description    =>  'Zrestartuj podaną uslugę. Dostępne usługi: httpd, mysql, pgslq, clam, antyspam, ssh, mail, jabber (UWAGA! Przy wyborze opcji jabber utracisz połączenie ze mną!)',
    :regex          =>  /^restart\s+.+$/,
    :alias          =>  [
                    :syntax =>  'r USŁUGA',
                    :regex  =>  /^r\s+.+$/
                    ]
)do |sender, user|
    ""
end

#### kilka przydatnych i uzywanych funkcji.

# N procesow najmocniej obciazajacych procesor
def topc(n)
    return `echo"";ps axo pid,user,size,time,state,cstime,pcpu,cmd --sort=-pcpu |head -n #{n}`
end

# N procesow zajmujacych najwiecej pamieci
def topm(n)
    return  `echo"";ps axo pid,user,size,time,state,cstime,pcpu,cmd --sort=-size |head -n #{n}`
end

# Status najwazniejszych uslug
def status()
    mysql_status        = `/etc/init.d/mysql status`
    pgsql_status        = `/etc/init.d/postgresql-8.3 status`
    
    if `ps -C apache2 -o pid= |wc -l`.to_i >=1 
        httpd_status = "OK"
    else
        httpd_status = "Nie dziła"
    end

    if `ps -C clamd -o pid=`.to_i > 0 and `ps -C freshclam -o pid=`.to_i > 0
        av_status = "OK"
    else
        av_status = "Nie dziła"
    end
    
    if `ps -C master -o pid=`.to_i > 0 and  `ps -C dovecot -o pid=`.to_i > 0
        mail_status = "OK"
    else
        mail_status = "Nie działa"
    end

    if `ps -C amavisd-new -o pid=`.split.length > 0 
        antyspam_status = "OK"
    else
        antyspam_status = "Nie działa"
    end

    if    pgsql_status.split[3] == "online" and !mysql_status.split.eql?("MySQLisstopped..")
        db_status = "OK"
    else
        db_status = "Nie działają"
    end

    return "HTTP - '#{httpd_status}'\nBazy danych - '#{db_status}'\nAntywirus - '#{av_status}'\nPoczta - '#{mail_status}'\nAntyspam - '#{antyspam_status}'"
end

def apacz()
    system("/etc/init.d/apache2 restart")
    return "Czerwonoskóre byle zrestartowane"
end

# Najważniejsze informacje o systemie - uptime i zasoby pamięci
def info()
    return `echo"";uptime;free -tom`
end

# Pokaż zalogowanych użytkowników
def who()
    return `echo"";w`
end

# ostatnie logowanie użytkownika.
def rlast(uzytkownik)
    if uzytkownik == "ALL"
        return `echo"";last`
    else
        return `echo"";last #{uzytkownik}`
    end
end

def restart(usluga)
    case '#{usluga}'
        when "ssh"        then
                            system("/etc/init.d/ssh restart")
        when "mysql"     then
                            system("/etc/init.d/mysql restart")
        when "pgsql"    then
                            system("/etc/init.d/postgresql-8.3 restart")
        when "httpd"    then
                            system("/etc/init.d/apache2 restart")
        when "clam"        then
                            system("/etc/init.d/clamav-daemon restart")
                            system("/etc/init.d/clamav-freshclam restart")
        when "antyspam"    then
                            system("/etc/init.d/amavis restart")
        when "mail"        then
                            system("/etc/init.d/postfix restart")
                            system("/etc/init.d/dovecot restart")
        when "jabber"    then
                            system("/etc/init.d/ejabberd restart");
    end
end

def pilnuj()
    for i in 0..2
        if i < 1 then
            if status().split.include?("Nie") then
                return "Sprawdź czy wysztkie usługi działają. Wydaje mi się że conajmniej jedna nie odpowiada"
                next
            end
            sleep(1)
            redo
        end    
    end
end
# Bring your new bot to life
bot.connect