106 lines
3.4 KiB
Python
106 lines
3.4 KiB
Python
# -*- 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/>.
|
|
|
|
import imp
|
|
import os
|
|
import shlex
|
|
import sys
|
|
import traceback
|
|
|
|
from . import builtins
|
|
|
|
class Prompt:
|
|
def __init__(self, hc=dict(), hl=dict()):
|
|
self.selectedServer = None
|
|
|
|
self.HOOKS_CAPS = hc
|
|
self.HOOKS_LIST = hl
|
|
|
|
def add_cap_hook(self, name, call, data=None):
|
|
self.HOOKS_CAPS[name] = (lambda d, t, c, p: call(d, t, c, p), data)
|
|
|
|
|
|
def lex_cmd(self, line):
|
|
"""Return an array of tokens"""
|
|
ret = list()
|
|
try:
|
|
cmds = shlex.split(line)
|
|
bgn = 0
|
|
for i in range(0, len(cmds)):
|
|
if cmds[i] == ';':
|
|
if i != bgn:
|
|
cmds[bgn] = cmds[bgn].lower()
|
|
ret.append(cmds[bgn:i])
|
|
bgn = i + 1
|
|
|
|
if bgn != len(cmds):
|
|
cmds[bgn] = cmds[bgn].lower()
|
|
ret.append(cmds[bgn:len(cmds)])
|
|
|
|
return ret
|
|
except:
|
|
exc_type, exc_value, exc_traceback = sys.exc_info()
|
|
sys.stderr.write (traceback.format_exception_only(
|
|
exc_type, exc_value)[0])
|
|
return ret
|
|
|
|
def exec_cmd(self, toks, context):
|
|
"""Execute the command"""
|
|
if toks[0] in builtins.CAPS:
|
|
return builtins.CAPS[toks[0]](toks, context, self)
|
|
elif toks[0] in self.HOOKS_CAPS:
|
|
(f,d) = self.HOOKS_CAPS[toks[0]]
|
|
return f(d, toks, context, self)
|
|
else:
|
|
print("Unknown command: `%s'" % toks[0])
|
|
return ""
|
|
|
|
def getPS1(self):
|
|
"""Get the PS1 associated to the selected server"""
|
|
if self.selectedServer is None:
|
|
return "nemubot"
|
|
else:
|
|
return self.selectedServer.id
|
|
|
|
def run(self, context):
|
|
"""Launch the prompt"""
|
|
ret = ""
|
|
while ret != "quit" and ret != "reset" and ret != "refresh":
|
|
sys.stdout.write("\033[0;33m%s§\033[0m " % self.getPS1())
|
|
sys.stdout.flush()
|
|
|
|
try:
|
|
line = sys.stdin.readline()
|
|
if len(line) <= 0:
|
|
line = "quit"
|
|
print("quit")
|
|
cmds = self.lex_cmd(line.strip())
|
|
for toks in cmds:
|
|
try:
|
|
ret = self.exec_cmd(toks, context)
|
|
except:
|
|
exc_type, exc_value, exc_traceback = sys.exc_info()
|
|
traceback.print_exception(exc_type, exc_value, exc_traceback)
|
|
except KeyboardInterrupt:
|
|
print("")
|
|
return ret != "quit"
|
|
|
|
|
|
def hotswap(prompt):
|
|
return Prompt(prompt.HOOKS_CAPS, prompt.HOOKS_LIST)
|