nemubot/modules/cmd_server.py

# -*- coding: utf-8 -*-

# Nemubot is a modulable IRC bot, built around XML configuration files.
# Copyright (C) 2012  Mercier Pierre-Olivier
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from networkbot import NetworkBot

nemubotversion = 3.2
NODATA = True

def getserver(toks, context, prompt):
    """Choose the server in toks or prompt"""
    if len(toks) > 1 and toks[0] in context.servers:
        return (context.servers[toks[0]], toks[1:])
    elif prompt.selectedServer is not None:
        return (prompt.selectedServer, toks)
    else:
        return (None, toks)

def close(data, toks, context, prompt):
    """Disconnect and forget (remove from the servers list) the server"""
    if len(toks) > 1:
        for s in toks[1:]:
            if s in servers:
                context.servers[s].disconnect()
                del context.servers[s]
            else:
                print ("close: server `%s' not found." % s)
    elif prompt.selectedServer is not None:
        prompt.selectedServer.disconnect()
        del prompt.servers[selectedServer.id]
        prompt.selectedServer = None
    return

def connect(data, toks, context, prompt):
    """Make the connexion to a server"""
    if len(toks) > 1:
        for s in toks[1:]:
            if s in context.servers:
                context.servers[s].launch(context.receive_message)
            else:
                print ("connect: server `%s' not found." % s)

    elif prompt.selectedServer is not None:
        prompt.selectedServer.launch(context.receive_message)
    else:
        print ("  Please SELECT a server or give its name in argument.")

def disconnect(data, toks, context, prompt):
    """Close the connection to a server"""
    if len(toks) > 1:
        for s in toks[1:]:
            if s in context.servers:
                if not context.servers[s].disconnect():
                    print ("disconnect: server `%s' already disconnected." % s)
            else:
                print ("disconnect: server `%s' not found." % s)
    elif prompt.selectedServer is not None:
        if not prompt.selectedServer.disconnect():
            print ("disconnect: server `%s' already disconnected."
                   % prompt.selectedServer.id)
    else:
        print ("  Please SELECT a server or give its name in argument.")

def discover(data, toks, context, prompt):
    """Discover a new bot on a server"""
    (srv, toks) = getserver(toks, context, prompt)
    if srv is not None:
        for name in toks[1:]:
            if "!" in name:
                bot = context.add_networkbot(srv, name)
                bot.connect()
            else:
                print ("  %s is not a valid fullname, for example: nemubot!nemubotV3@bot.nemunai.re")
    else:
        print ("  Please SELECT a server or give its name in first argument.")

def hotswap(data, toks, context, prompt):
    """Reload a server class"""
    if len(toks) > 1:
        print ("hotswap: apply only on selected server")
    elif prompt.selectedServer is not None:
        del context.servers[prompt.selectedServer.id]
        srv = server.Server(selectedServer.node, selectedServer.nick,
                            selectedServer.owner, selectedServer.realname,
                            selectedServer.s)
        context.servers[srv.id] = srv
        prompt.selectedServer.kill()
        prompt.selectedServer = srv
        prompt.selectedServer.start()
    else:
        print ("  Please SELECT a server or give its name in argument.")

def join(data, toks, context, prompt):
    """Join or leave a channel"""
    rd = 1
    if len(toks) <= rd:
        print ("%s: not enough arguments." % toks[0])
        return

    if toks[rd] in context.servers:
        srv = context.servers[toks[rd]]
        rd += 1
    elif prompt.selectedServer is not None:
        srv = prompt.selectedServer
    else:
        print ("  Please SELECT a server or give its name in argument.")
        return

    if len(toks) <= rd:
        print ("%s: not enough arguments."  % toks[0])
        return

    if toks[0] == "join":
        if len(toks) > rd + 1:
            srv.join(toks[rd], toks[rd + 1])
        else:
            srv.join(toks[rd])
    elif toks[0] == "leave" or toks[0] == "part":
        srv.leave(toks[rd])
    return

def send(data, toks, context, prompt):
    """Send a message on a channel"""
    rd = 1
    if len(toks) <= rd:
        print ("send: not enough arguments.")
        return

    if toks[rd] in context.servers:
        srv = context.servers[toks[rd]]
        rd += 1
    elif prompt.selectedServer is not None:
        srv = prompt.selectedServer
    else:
        print ("  Please SELECT a server or give its name in argument.")
        return

    if len(toks) <= rd:
        print ("send: not enough arguments.")
        return

    #Check the server is connected
    if not srv.connected:
        print ("send: server `%s' not connected." % srv.id)
        return

    if toks[rd] in srv.channels:
        chan = toks[rd]
        rd += 1
    else:
        print ("send: channel `%s' not authorized in server `%s'."
               % (toks[rd], srv.id))
        return

    if len(toks) <= rd:
        print ("send: not enough arguments.")
        return

    srv.send_msg_final(chan, toks[rd])
    return "done"

def zap(data, toks, context, prompt):
    """Hard change connexion state"""
    if len(toks) > 1:
        for s in toks[1:]:
            if s in context.servers:
                context.servers[s].connected = not context.servers[s].connected
            else:
                print ("zap: server `%s' not found." % s)
    elif prompt.selectedServer is not None:
        prompt.selectedServer.connected = not prompt.selectedServer.connected
    else:
        print ("  Please SELECT a server or give its name in argument.")