Add a module that extend prompt command with server feature (connect, join, ..)
This commit is contained in:
parent
e4d4e68c45
commit
fe1fefecc0
163
modules/cmd_server.py
Normal file
163
modules/cmd_server.py
Normal file
@ -0,0 +1,163 @@
|
|||||||
|
# -*- 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/>.
|
||||||
|
|
||||||
|
nemubotversion = 3.2
|
||||||
|
NODATA = True
|
||||||
|
|
||||||
|
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)
|
||||||
|
else:
|
||||||
|
print ("connect: server `%s' not found." % s)
|
||||||
|
|
||||||
|
elif prompt.selectedServer is not None:
|
||||||
|
prompt.selectedServer.launch(context)
|
||||||
|
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 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.")
|
12
modules/cmd_server.xml
Normal file
12
modules/cmd_server.xml
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
<?xml version="1.0" ?>
|
||||||
|
<nemubotmodule name="cmd_server">
|
||||||
|
<command name="close" call="close" />
|
||||||
|
<command name="connect" call="connect" />
|
||||||
|
<command name="disconnect" call="disconnect" />
|
||||||
|
<command name="hotswap" call="hotswap" />
|
||||||
|
<command name="join" call="join" />
|
||||||
|
<command name="leave" call="join" />
|
||||||
|
<command name="part" call="join" />
|
||||||
|
<command name="send" call="send" />
|
||||||
|
<command name="zap" call="zap" />
|
||||||
|
</nemubotmodule>
|
Loading…
Reference in New Issue
Block a user