Limnoria/src/template.py

#!/usr/bin/env python

###
# Copyright (c) 2002, Jeremiah Fincher
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
#   * Redistributions of source code must retain the above copyright notice,
#     this list of conditions, and the following disclaimer.
#   * Redistributions in binary form must reproduce the above copyright notice,
#     this list of conditions, and the following disclaimer in the
#     documentation and/or other materials provided with the distribution.
#   * Neither the name of the author of this software nor the name of
#     contributors to this software may be used to endorse or promote products
#     derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
###

"""
This is the template for bots.  scripts/setup.py uses this file to make
customized startup files for bots.
"""

import re
import sys
import time
import optparse

started = time.time()

if 'src' not in sys.path:
    sys.path.insert(0, 'src')

import conf

conf.commandsOnStart = %%onStart%%

afterConnect = %%afterConnect%%

configVariables = %%configVariables%%

for (name, value) in configVariables.iteritems():
    setattr(conf, name, value)


def main():
    import debug
    import world
    import drivers
    import schedule
    schedule.addPeriodicEvent(world.upkeep, 300)
    world.startedAt = started
    try:
        while world.ircs:
            drivers.run()
    except:
        try:
            debug.recoverableException()
        except: # It must've been deadly for a reason :)
            sys.exit(0)

if __name__ == '__main__':
    ###
    # Options:
    # -p (profiling)
    # -O (optimizing)
    # -n, --nick (nick)
    # -s, --server (server)
    # --startup (commands to run onStart)
    # --connect (commands to run afterConnect)
    # --config (configuration values)
    parser = optparse.OptionParser(usage='Usage: %prog [options]',
                                   version='supybot %s' % conf.version)
    parser.add_option('-P', '--profile', action='store_true', dest='profile',
                      help='enables profiling')
    parser.add_option('-O', action='count', dest='optimize',
                      help='-O optimizes asserts out of the code; ' \
                           '-OO optimizes asserts and uses psyco.')
    parser.add_option('-n', '--nick', action='store',
                      dest='nick', default=%%nick%%,
                      help='nick the bot should use') # FIXME (quotes)
    parser.add_option('-s', '--server', action='store',
                      dest='server', default=%%server%%,
                      help='server to connect to') # FIXME (quotes)
    parser.add_option('-u', '--user', action='store',
                      dest='user', default=%%user%%,
                      help='full username the bot should use')
    parser.add_option('-i', '--ident', action='store',
                      dest='ident', default=%%ident%%,
                      help='ident the bot should use')
    parser.add_option('-p', '--password', action='store',
                      dest='password', default=%%password%%,
                      help='server password the bot should use')
    parser.add_option('--startup', action='append', dest='onStart',
                      help='file of additional commands to run at startup.')
    parser.add_option('--connect', action='append', dest='afterConnect',
                      help='file of additional commands to run after connect')
    parser.add_option('--config', action='append', dest='conf',
                      help='file of configuration variables to set')

    (options, args) = parser.parse_args()

    if options.optimize:
        __builtins__.__debug__ = False
        if options.optimize > 1:
            import psyco
            psyco.full()

    if options.onStart:
        for filename in options.onStart:
            fd = file(filename)
            for line in fd:
                conf.commandsOnStart.append(line.rstrip())
            fd.close()

    if options.afterConnect:
        for filename in options.afterConnect:
            fd = file(filename)
            for line in fd:
                afterConnect.append(line.rstrip())
            fd.close()

    assignmentRe = re.compile('\s*[:=]\s*')
    if options.conf:
        for filename in options.conf:
            fd = file(filename)
            for line in fd:
                (name, valueString) = assignmentRe.split(line.rstrip(), 1)
                try:
                    value = eval(valueString)
                except Exception, e:
                    sys.stderr.write('Invalid configuration value: %r' % \
                                     valueString)
                    sys.exit(-1)

    sys.path.append(conf.pluginDir)

    nick = options.nick
    user = options.user
    ident = options.ident
    password = options.password

    if ':' in options.server:
        serverAndPort = options.server.split(':', 1)
        serverAndPort[1] = int(serverAndPort[1])
        server = tuple(serverAndPort)
    else:
        server = (options.server, 6667)

    import irclib
    import ircmsgs
    import drivers
    import callbacks
    import OwnerCommands

    class ConfigAfter376(irclib.IrcCallback):
        public = False
        def __init__(self, commands):
            self.commands = commands
        def do376(self, irc, msg):
            for command in self.commands:
                msg = ircmsgs.privmsg(irc.nick, command, prefix=irc.prefix)
                irc.queueMsg(msg)
        do377 = do376

    # We pre-tokenize the commands just to save on significant amounts of work.
    conf.commandsOnStart = map(callbacks.tokenize, conf.commandsOnStart)

    irc = irclib.Irc(nick, user, ident, password)
    callback = OwnerCommands.Class()
    callback.configure(irc)
    irc.addCallback(callback)
    irc.addCallback(ConfigAfter376(afterConnect))
    driver = drivers.newDriver(server, irc)
    
    if options.profile:
        import profile
        profile.run('main()')
    else:
        main()


    
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: