root/dotfiles/screen/shyouhei/.screen/logmonitor.rb @ 36330

#! /bin/ruby -Ku

require 'backtick'
require 'thread'
require 'open-uri'
require 'json'
require 'time'
require 'cgi'

STDOUT.sync = true
Cons = Struct.new :car, :cdr
A = [Cons.new]
M = Mutex.new
A.last.cdr = A.first

def showline i, j, k
        t = j.strftime '%H:%M:%S'
        x = attribute ' dB', i
        d = /.{0,140}/.match k.chomp
        printf "[%s %s] %s\n", t, x, d[0]
end

def push who, what
        what.each do |(t, i)|
                showline who, t, i
        end
        c = Cons.new [who, what], nil
        A.last.cdr = c
        A.shift if A.size >= 8
        A.push c
        c.cdr = A.first
end

def plugin
        Thread.start do
                loop do
                        begin
                                yield
                        rescue Exception => e
                                b = e.backtrace.map {|i| [Time.now, i] }
                                b.reverse!
                                M.synchronize do
                                        push 'exception', [[Time.now, attribute('bB RI', e.message)]]
                                        push 'trace', b
                                end
                        end
                        sleep 30
                end
        end
end

def selective cmd
        r1 = /^/
        r2 = /[\r\n]/
        open "| sudo #{cmd}", 'rb' do |f|
                s = ''
                t = ''
                loop do
#                       sleep 2
                        IO.select [f]
                        f.read_nonblock 4096, s
                        t << s
                        a = t.split r1
                        t = r2.match(a.last) ? '' : a.pop
                        M.synchronize do
                                yield a
                        end
                end             
        end
end

# syslog thread
plugin do
        r = %r'
                \A(
         (?:Jan|Feb|Mar|Apr|May|Jun|Jul|Aug|Sep|Oct|Nov|Dec)\s
                   [\s123]\d\s\d\d:\d\d:\d\d
      )\s
      .+?:\s(.*)
                \z
   'xmn
        selective 'tail -f -n 300 /var/log/syslog' do |a|
                a.map! {|i| r.match i }
                a.compact!
                a.map! {|i| [Time.parse(i[1]), i[2]] }
                push 'syslog', a
        end
end

# tshark thread
plugin do
        r = /^\s+(?:[\d\.]+)\s(.+)$/
        selective "tshark -i any 'tcp and (((ip[2:2] - ((ip[0]&0xf)<<2)) - ((tcp[12]&0xf0)>>2)) != 0)'" do |a|
                t = Time.now
                a.map! {|i| r.match i }
                a.compact!
                a.map! {|i| i[1] }
                a.each do |i|
                        showline 'tshark', t, i
                end
        end
end

# twitter thread
plugin do
        # should be at most 170 [req./hour]
        opts = { :http_basic_authentication => %w'username password' }
        uri = 'http://api.twitter.com/1/statuses/home_timeline.json'
        since = ''
        loop do
                j = open "#{uri}#{since}", 'rb', opts do |f|
                        JSON.parse(f.read).sort_by {|i| i['id'] }
                end
                unless j.empty?
                        M.synchronize do
                                j.each do |i|
                                        t = Time.parse i['created_at']
                                        s = CGI.unescapeHTML i['text']
                                        w = attribute 'b dG', i['user']['screen_name']
                                        push 'twitter', [[t, "@#{w}> #{s}"]]
                                end
                                since = "?since_id=#{j.last['id']}"
                        end
                end
                sleep 127
        end
end

i = A.first
loop do
        sleep 2
        M.synchronize do
                j, i = i.car, i.cdr
                next unless j
                who, what = *j
                what.each do |(t, k)|
                        showline who, t, k
                        sleep 0.07
                end
        end
end

# Local Variables:
# mode: ruby
# coding: utf-8
# indent-tabs-mode: t
# tab-width: 3
# ruby-indent-level: 3
# fill-column: 79
# default-justification: full
# End:
# vi: ts=3 sw=3