plugins/log_plugin.py
author Mikael Berthe <mikael@lilotux.net>
Mon, 07 May 2007 11:59:17 +0200
changeset 35 8daaa0b29b71
parent 17 plugins/disabled/log_plugin.py@069f7fd5545d
permissions -rw-r--r--
Enable log_plugin

#$ neutron_plugin 01

import re

LOG_CACHE_FILE = 'dynamic/logcache.txt'

initialize_file(LOG_CACHE_FILE, '{}')
LOG_FILENAME_CACHE = eval(read_file(LOG_CACHE_FILE))

def log_write_header(fp, source, (year, month, day, hour, minute, second, weekday, yearday, daylightsavings)):
    fp.write("""<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en" lang="en">
<head>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
<style type="text/css">
<!--
.timestamp {color: #AAAAAA;}
.system {color: #009900; font-weight: bold;}
.emote {color: #AA0099;}
.self {color: #CC0000;}
.normal {color: #0000AA;}
h1 { color: #336699; font-family: sans-serif; border-bottom: #224466 solid 3pt; letter-spacing: 3px; margin-left: 20pt; }
h2 { color: #663399; font-family: sans-serif; letter-spacing: 2px; text-align: center }
//-->
</style>
</head>
<body>
<div style="color: #AAAAAA; text-align: right; font-family: monospace; letter-spacing: 3px">neutron log</div>
<h1>""" + source + """</h1>
<h2>""" + time.strftime('%A, %B %d, %Y', (year, month, day, hour, minute, second, weekday, yearday, daylightsavings)) + """</h2>
<br />
<tt>

""")

def log_write_footer(fp):
    fp.write('\n</tt>\n</body>\n</html>')

def log_get_fp(type, source, (year, month, day, hour, minute, second, weekday, yearday, daylightsavings)):
    if type == 'public':
        logdir = PUBLIC_LOG_DIR
    else:
        logdir = PRIVATE_LOG_DIR
    if logdir[-1] == '/':
        logdir = LOGDIR[:-1]
    str_year = str(year)
    str_month = str(month)
    str_day = str(day)
    filename = logdir + '/' + source + '/' + str_year + '/' + str_month + '/' + str_day + '.html'
    alt_filename = logdir + '/' + source + '/' + str_year + '/' + str_month + '/' + str_day + '_alt.html'
    if not os.path.exists(logdir):
        os.mkdir(logdir)
    if not os.path.exists(logdir + '/' + source):
        os.mkdir(logdir + '/' + source)
    if not os.path.exists(logdir + '/' + source + '/' + str_year):
        os.mkdir(logdir + '/' + source + '/' + str_year)
    if not os.path.exists(logdir + '/' + source + '/' + str_year + '/' + str_month):
        os.mkdir(logdir + '/' + source + '/' + str_year + '/' + str_month)
    if LOG_FILENAME_CACHE.has_key(source):
        if LOG_FILENAME_CACHE[source] != filename:
            fp_old = file(LOG_FILENAME_CACHE[source], 'a')
            log_write_footer(fp_old)
            fp_old.close()
        if os.path.exists(filename):
            fp = file(filename, 'a')
            return fp
        else:
            LOG_FILENAME_CACHE[source] = filename
            write_file(LOG_CACHE_FILE, str(LOG_FILENAME_CACHE))
            fp = file(filename, 'w')
            log_write_header(fp, source, (year, month, day, hour, minute, second, weekday, yearday, daylightsavings))
            return fp
    else:
        if os.path.exists(filename):
            LOG_FILENAME_CACHE[source] = filename
            write_file(LOG_CACHE_FILE, str(LOG_FILENAME_CACHE))
            fp = file(alt_filename, 'a')
            # log_write_header(fp, source, (year, month, day, hour, minute, second, weekday, yearday, daylightsavings))
            return fp
        else:
            LOG_FILENAME_CACHE[source] = filename
            write_file(LOG_CACHE_FILE, str(LOG_FILENAME_CACHE))
            fp = file(filename, 'w')
            log_write_header(fp, source, (year, month, day, hour, minute, second, weekday, yearday, daylightsavings))
            return fp

def log_get_timestamp(hour, minute):
    timestamp = '['
    if hour < 10:
        timestamp += '0'
    timestamp += str(hour)
    timestamp += ':'
    if minute < 10:
        timestamp += '0'
    timestamp += str(minute)
    timestamp += ']'
    return timestamp

def log_regex_url(matchobj):
    # 06.03.05(Sun) slipstream@yandex.ru urls parser
    return '<a href="' + matchobj.group(0) + '">' + matchobj.group(0) + '</a>'

def log_handler_message(type, source, body):
    if not body:
        return
    (year, month, day, hour, minute, second, weekday, yearday, daylightsavings) = time.gmtime()
    if type == 'public' and PUBLIC_LOG_DIR:
        groupchat = source[1]
        nick = source[2]
        # 06.03.05(Sun) slipstream@yandex.ru urls parser & line ends
        body = body.replace('&', '&amp;').replace('"', '&quot;').replace('<', '&lt;').replace('>', '&gt;')
        body = re.sub('(http|ftp)(\:\/\/[^\s<]+)', log_regex_url, body)
        body = body.replace('\n', '<br />')
        body = body.encode('utf-8');
        nick = nick.encode('utf-8');
        timestamp = log_get_timestamp(hour, minute)
        fp = log_get_fp('public', groupchat, (year, month, day, hour, minute, second, weekday, yearday, daylightsavings))
        fp.write('<font class="timestamp">' + timestamp + '</font> ')
        if not nick:
            fp.write('<font class="system">' + body + '</font><br />\n')
        elif body[:3].lower() == '/me':
            fp.write('<font class="emote">* ' + nick + body[3:] + '</font><br />\n')
        else:
            # 08.03.05(Tue) slipstream@yandex.ru encoding
            if nick == get_nick(groupchat).encode('utf-8'):
                fp.write('<font class="self">&lt;' + nick + '&gt;</font> ')
            else:
                fp.write('<font class="normal">&lt;' + nick + '&gt;</font> ')
            fp.write(body + '<br />\n')
        fp.close()
    elif type == 'private' and PRIVATE_LOG_DIR:
        jid = get_true_jid(source)
        nick = string.split(jid, '@')[0]
        # 06.03.05(Sun) slipstream@yandex.ru urls parser, line ends & encoding
        body = body.replace('&', '&amp;').replace('"', '&quot;').replace('<', '&lt;').replace('>', '&gt;')
        body = re.sub('(http|ftp)(\:\/\/[^\s<]+)', log_regex_url, body)
        body = body.replace('\n', '<br />')
        body = body.encode('utf-8');
        nick = nick.encode('utf-8');
        timestamp = log_get_timestamp(hour, minute)
        fp = log_get_fp('private', jid, (year, month, day, hour, minute, second, weekday, yearday, daylightsavings))
        fp.write('<font class="timestamp">' + timestamp + '</font> ')
        if body[:3].lower() == '/me':
            fp.write('<font class="emote">* ' + nick + body[3:] + '</font><br />\n')
        else:
            fp.write('<font class="normal">&lt;' + nick + '&gt;</font> ' + body + '<br />\n')
        fp.close()

def log_handler_outgoing_message(target, body):
    if GROUPCHATS.has_key(target) or not body:
        return
    (year, month, day, hour, minute, second, weekday, yearday, daylightsavings) = time.gmtime()
    jid = get_true_jid(target)
    nick = 'neutron'
    # 06.03.05(Sun) slipstream@yandex.ru urls parser, line ends & encoding
    body = body.replace('&', '&amp;').replace('"', '&quot;').replace('<', '&lt;').replace('>', '&gt;')
    body = re.sub('(http|ftp)(\:\/\/[^\s<]+)', log_regex_url, body)
    body = body.replace('\n', '<br />')
    body = body.encode('utf-8');
    nick = nick.encode('utf-8');
    timestamp = log_get_timestamp(hour, minute)
    fp = log_get_fp('private', jid, (year, month, day, hour, minute, second, weekday, yearday, daylightsavings))
    fp.write('<font class="timestamp">' + timestamp + '</font> ')
    if body[:3].lower() == '/me':
        fp.write('<font class="emote">* ' + nick + body[3:] + '</font><br />\n')
    else:
        fp.write('<font class="self">&lt;' + nick + '&gt;</font> ' + body + '<br />\n')
    fp.close()

if PUBLIC_LOG_DIR or PRIVATE_LOG_DIR:
    register_message_handler(log_handler_message)
if PRIVATE_LOG_DIR:
    register_outgoing_message_handler(log_handler_outgoing_message)