332 lines
16 KiB
Python
332 lines
16 KiB
Python
## $Id: commands.py,v 1.11 2005/11/30 17:03:11 normanr Exp $
|
|
|
|
## Ad-Hoc Command manager
|
|
## Mike Albon (c) 5th January 2005
|
|
|
|
## This program is free software; you can redistribute it and/or modify
|
|
## it under the terms of the GNU General Public License as published by
|
|
## the Free Software Foundation; either version 2, 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 General Public License for more details.
|
|
|
|
|
|
"""This module is a ad-hoc command processor for xmpppy. It uses the plug-in mechanism like most of the core library. It depends on a DISCO browser manager.
|
|
|
|
There are 3 classes here, a command processor Commands like the Browser, and a command template plugin Command, and an example command.
|
|
|
|
To use this module:
|
|
|
|
Instansiate the module with the parent transport and disco browser manager as parameters.
|
|
'Plug in' commands using the command template.
|
|
The command feature must be added to existing disco replies where neccessary.
|
|
|
|
What it supplies:
|
|
|
|
Automatic command registration with the disco browser manager.
|
|
Automatic listing of commands in the public command list.
|
|
A means of handling requests, by redirection though the command manager.
|
|
"""
|
|
|
|
from protocol import *
|
|
from client import PlugIn
|
|
|
|
class Commands(PlugIn):
|
|
"""Commands is an ancestor of PlugIn and can be attached to any session.
|
|
|
|
The commands class provides a lookup and browse mechnism. It follows the same priciple of the Browser class, for Service Discovery to provide the list of commands, it adds the 'list' disco type to your existing disco handler function.
|
|
|
|
How it works:
|
|
The commands are added into the existing Browser on the correct nodes. When the command list is built the supplied discovery handler function needs to have a 'list' option in type. This then gets enumerated, all results returned as None are ignored.
|
|
The command executed is then called using it's Execute method. All session management is handled by the command itself.
|
|
"""
|
|
def __init__(self, browser):
|
|
"""Initialises class and sets up local variables"""
|
|
PlugIn.__init__(self)
|
|
DBG_LINE='commands'
|
|
self._exported_methods=[]
|
|
self._handlers={'':{}}
|
|
self._browser = browser
|
|
|
|
def plugin(self, owner):
|
|
"""Makes handlers within the session"""
|
|
# Plug into the session and the disco manager
|
|
# We only need get and set, results are not needed by a service provider, only a service user.
|
|
owner.RegisterHandler('iq',self._CommandHandler,typ='set',ns=NS_COMMANDS)
|
|
owner.RegisterHandler('iq',self._CommandHandler,typ='get',ns=NS_COMMANDS)
|
|
self._browser.setDiscoHandler(self._DiscoHandler,node=NS_COMMANDS,jid='')
|
|
|
|
def plugout(self):
|
|
"""Removes handlers from the session"""
|
|
# unPlug from the session and the disco manager
|
|
self._owner.UnregisterHandler('iq',self_CommandHandler,ns=NS_COMMANDS)
|
|
for jid in self._handlers:
|
|
self._browser.delDiscoHandler(self._DiscoHandler,node=NS_COMMANDS)
|
|
|
|
def _CommandHandler(self,conn,request):
|
|
"""The internal method to process the routing of command execution requests"""
|
|
# This is the command handler itself.
|
|
# We must:
|
|
# Pass on command execution to command handler
|
|
# (Do we need to keep session details here, or can that be done in the command?)
|
|
jid = str(request.getTo())
|
|
try:
|
|
node = request.getTagAttr('command','node')
|
|
except Exception:
|
|
conn.send(Error(request,ERR_BAD_REQUEST))
|
|
raise NodeProcessed
|
|
if jid in self._handlers:
|
|
if node in self._handlers[jid]:
|
|
self._handlers[jid][node]['execute'](conn,request)
|
|
else:
|
|
conn.send(Error(request,ERR_ITEM_NOT_FOUND))
|
|
raise NodeProcessed
|
|
elif node in self._handlers['']:
|
|
self._handlers[''][node]['execute'](conn,request)
|
|
else:
|
|
conn.send(Error(request,ERR_ITEM_NOT_FOUND))
|
|
raise NodeProcessed
|
|
|
|
def _DiscoHandler(self,conn,request,typ):
|
|
"""The internal method to process service discovery requests"""
|
|
# This is the disco manager handler.
|
|
if typ == 'items':
|
|
# We must:
|
|
# Generate a list of commands and return the list
|
|
# * This handler does not handle individual commands disco requests.
|
|
# Pseudo:
|
|
# Enumerate the 'item' disco of each command for the specified jid
|
|
# Build responce and send
|
|
# To make this code easy to write we add an 'list' disco type, it returns a tuple or 'none' if not advertised
|
|
list = []
|
|
items = []
|
|
jid = str(request.getTo())
|
|
# Get specific jid based results
|
|
if jid in self._handlers:
|
|
for each in self._handlers[jid].keys():
|
|
items.append((jid,each))
|
|
else:
|
|
# Get generic results
|
|
for each in self._handlers[''].keys():
|
|
items.append(('',each))
|
|
if items != []:
|
|
for each in items:
|
|
i = self._handlers[each[0]][each[1]]['disco'](conn,request,'list')
|
|
if i is not None:
|
|
list.append(Node(tag='item',attrs={'jid':i[0],'node':i[1],'name':i[2]}))
|
|
iq = request.buildReply('result')
|
|
if request.getQuerynode(): iq.setQuerynode(request.getQuerynode())
|
|
iq.setQueryPayload(list)
|
|
conn.send(iq)
|
|
else:
|
|
conn.send(Error(request,ERR_ITEM_NOT_FOUND))
|
|
raise NodeProcessed
|
|
elif typ == 'info':
|
|
return {'ids':[{'category':'automation','type':'command-list'}],'features':[]}
|
|
|
|
def addCommand(self,name,cmddisco,cmdexecute,jid=''):
|
|
"""The method to call if adding a new command to the session, the requred parameters of cmddisco and cmdexecute are the methods to enable that command to be executed"""
|
|
# This command takes a command object and the name of the command for registration
|
|
# We must:
|
|
# Add item into disco
|
|
# Add item into command list
|
|
if jid not in self._handlers:
|
|
self._handlers[jid]={}
|
|
self._browser.setDiscoHandler(self._DiscoHandler,node=NS_COMMANDS,jid=jid)
|
|
if name in self._handlers[jid]:
|
|
raise NameError,'Command Exists'
|
|
else:
|
|
self._handlers[jid][name]={'disco':cmddisco,'execute':cmdexecute}
|
|
# Need to add disco stuff here
|
|
self._browser.setDiscoHandler(cmddisco,node=name,jid=jid)
|
|
|
|
def delCommand(self,name,jid=''):
|
|
"""Removed command from the session"""
|
|
# This command takes a command object and the name used for registration
|
|
# We must:
|
|
# Remove item from disco
|
|
# Remove item from command list
|
|
if jid not in self._handlers:
|
|
raise NameError,'Jid not found'
|
|
if name not in self._handlers[jid]:
|
|
raise NameError, 'Command not found'
|
|
else:
|
|
#Do disco removal here
|
|
command = self.getCommand(name,jid)['disco']
|
|
del self._handlers[jid][name]
|
|
self._browser.delDiscoHandler(command,node=name,jid=jid)
|
|
|
|
def getCommand(self,name,jid=''):
|
|
"""Returns the command tuple"""
|
|
# This gets the command object with name
|
|
# We must:
|
|
# Return item that matches this name
|
|
if jid not in self._handlers:
|
|
raise NameError,'Jid not found'
|
|
elif name not in self._handlers[jid]:
|
|
raise NameError,'Command not found'
|
|
else:
|
|
return self._handlers[jid][name]
|
|
|
|
class Command_Handler_Prototype(PlugIn):
|
|
"""This is a prototype command handler, as each command uses a disco method
|
|
and execute method you can implement it any way you like, however this is
|
|
my first attempt at making a generic handler that you can hang process
|
|
stages on too. There is an example command below.
|
|
|
|
The parameters are as follows:
|
|
name : the name of the command within the jabber environment
|
|
description : the natural language description
|
|
discofeatures : the features supported by the command
|
|
initial : the initial command in the from of {'execute':commandname}
|
|
|
|
All stages set the 'actions' dictionary for each session to represent the possible options available.
|
|
"""
|
|
name = 'examplecommand'
|
|
count = 0
|
|
description = 'an example command'
|
|
discofeatures = [NS_COMMANDS,NS_DATA]
|
|
# This is the command template
|
|
def __init__(self,jid=''):
|
|
"""Set up the class"""
|
|
PlugIn.__init__(self)
|
|
DBG_LINE='command'
|
|
self.sessioncount = 0
|
|
self.sessions = {}
|
|
# Disco information for command list pre-formatted as a tuple
|
|
self.discoinfo = {'ids':[{'category':'automation','type':'command-node','name':self.description}],'features': self.discofeatures}
|
|
self._jid = jid
|
|
|
|
def plugin(self,owner):
|
|
"""Plug command into the commands class"""
|
|
# The owner in this instance is the Command Processor
|
|
self._commands = owner
|
|
self._owner = owner._owner
|
|
self._commands.addCommand(self.name,self._DiscoHandler,self.Execute,jid=self._jid)
|
|
|
|
def plugout(self):
|
|
"""Remove command from the commands class"""
|
|
self._commands.delCommand(name,self._jid)
|
|
|
|
def getSessionID(self):
|
|
"""Returns an id for the command session"""
|
|
self.count = self.count+1
|
|
return 'cmd-%s-%d'%(self.name,self.count)
|
|
|
|
def Execute(self,conn,request):
|
|
"""The method that handles all the commands, and routes them to the correct method for that stage."""
|
|
# New request or old?
|
|
try:
|
|
session = request.getTagAttr('command','sessionid')
|
|
except Exception:
|
|
session = None
|
|
try:
|
|
action = request.getTagAttr('command','action')
|
|
except Exception:
|
|
action = None
|
|
if action is None: action = 'execute'
|
|
# Check session is in session list
|
|
if session in self.sessions:
|
|
if self.sessions[session]['jid']==request.getFrom():
|
|
# Check action is vaild
|
|
if action in self.sessions[session]['actions']:
|
|
# Execute next action
|
|
self.sessions[session]['actions'][action](conn,request)
|
|
else:
|
|
# Stage not presented as an option
|
|
self._owner.send(Error(request,ERR_BAD_REQUEST))
|
|
raise NodeProcessed
|
|
else:
|
|
# Jid and session don't match. Go away imposter
|
|
self._owner.send(Error(request,ERR_BAD_REQUEST))
|
|
raise NodeProcessed
|
|
elif session is not None:
|
|
# Not on this sessionid you won't.
|
|
self._owner.send(Error(request,ERR_BAD_REQUEST))
|
|
raise NodeProcessed
|
|
else:
|
|
# New session
|
|
self.initial[action](conn,request)
|
|
|
|
def _DiscoHandler(self,conn,request,type_):
|
|
"""The handler for discovery events"""
|
|
if type_ == 'list':
|
|
return (request.getTo(),self.name,self.description)
|
|
elif type_ == 'items':
|
|
return []
|
|
elif type_ == 'info':
|
|
return self.discoinfo
|
|
|
|
class TestCommand(Command_Handler_Prototype):
|
|
""" Example class. You should read source if you wish to understate how it works.
|
|
Generally, it presents a "master" that giudes user through to calculate something.
|
|
"""
|
|
name = 'testcommand'
|
|
description = 'a noddy example command'
|
|
def __init__(self,jid=''):
|
|
""" Init internal constants. """
|
|
Command_Handler_Prototype.__init__(self,jid)
|
|
self.pi = 3.14
|
|
self.initial = {'execute':self.cmdFirstStage}
|
|
|
|
def cmdFirstStage(self,conn,request):
|
|
""" Determine """
|
|
# This is the only place this should be repeated as all other stages should have SessionIDs
|
|
try:
|
|
session = request.getTagAttr('command','sessionid')
|
|
except Exception:
|
|
session = None
|
|
if session is None:
|
|
session = self.getSessionID()
|
|
sessions[session]={'jid':request.getFrom(),'actions':{'cancel':self.cmdCancel,'next':self.cmdSecondStage},'data':{'type':None}}
|
|
# As this is the first stage we only send a form
|
|
reply = request.buildReply('result')
|
|
form = DataForm(title='Select type of operation',data=['Use the combobox to select the type of calculation you would like to do, then click Next',DataField(name='calctype',label='Calculation Type',value=sessions[session]['data']['type'],options=[['circlediameter','Calculate the Diameter of a circle'],['circlearea','Calculate the area of a circle']],typ='list-single',required=1)])
|
|
replypayload = [Node('actions',attrs={'execute':'next'},payload=[Node('next')]),form]
|
|
reply.addChild(name='command',attrs={'xmlns':NS_COMMAND,'node':request.getTagAttr('command','node'),'sessionid':session,'status':'executing'},payload=replypayload)
|
|
self._owner.send(reply)
|
|
raise NodeProcessed
|
|
|
|
def cmdSecondStage(self,conn,request):
|
|
form = DataForm(node = result.getTag(name='command').getTag(name='x',namespace=NS_DATA))
|
|
sessions[request.getTagAttr('command','sessionid')]['data']['type']=form.getField('calctype')
|
|
sessions[request.getTagAttr('command','sessionid')]['actions']={'cancel':self.cmdCancel,None:self.cmdThirdStage,'previous':cmdFirstStage}
|
|
# The form generation is split out to another method as it may be called by cmdThirdStage
|
|
self.cmdSecondStageReply(conn,request)
|
|
|
|
def cmdSecondStageReply(self,conn,request):
|
|
reply = request.buildReply('result')
|
|
form = DataForm(title = 'Enter the radius', data=['Enter the radius of the circle (numbers only)',DataField(label='Radius',name='radius',typ='text-single')])
|
|
replypayload = [Node('actions',attrs={'execute':'complete'},payload=[Node('complete'),Node('prev')]),form]
|
|
reply.addChild(name='command',attrs={'xmlns':NS_COMMAND,'node':request.getTagAttr('command','node'),'sessionid':request.getTagAttr('command','sessionid'),'status':'executing'},payload=replypayload)
|
|
self._owner.send(reply)
|
|
raise NodeProcessed
|
|
|
|
def cmdThirdStage(self,conn,request):
|
|
form = DataForm(node = result.getTag(name='command').getTag(name='x',namespace=NS_DATA))
|
|
try:
|
|
num = float(form.getField('radius'))
|
|
except Exception:
|
|
self.cmdSecondStageReply(conn,request)
|
|
if sessions[request.getTagAttr('command','sessionid')]['data']['type'] == 'circlearea':
|
|
result = num*(pi**2)
|
|
else:
|
|
result = num*2*pi
|
|
reply = result.buildReply(request)
|
|
form = DataForm(typ='result',data=[DataField(label='result',name='result',value=result)])
|
|
reply.addChild(name='command',attrs={'xmlns':NS_COMMAND,'node':request.getTagAttr('command','node'),'sessionid':request.getTagAttr('command','sessionid'),'status':'completed'},payload=form)
|
|
self._owner.send(reply)
|
|
raise NodeProcessed
|
|
|
|
def cmdCancel(self,conn,request):
|
|
reply = request.buildReply('result')
|
|
reply.addChild(name='command',attrs={'xmlns':NS_COMMAND,'node':request.getTagAttr('command','node'),'sessionid':request.getTagAttr('command','sessionid'),'status':'cancelled'})
|
|
self._owner.send(reply)
|
|
del sessions[request.getTagAttr('command','sessionid')]
|
|
|
|
|
|
|
|
# vim: se ts=3: |