mirror of
https://github.com/jlu5/PyLink.git
synced 2024-11-30 14:49:28 +01:00
296 lines
10 KiB
Python
296 lines
10 KiB
Python
# commands.py: base PyLink commands
|
|
import time
|
|
|
|
from pylinkirc import utils, __version__, world, real_version
|
|
from pylinkirc.log import log
|
|
from pylinkirc.coremods import permissions
|
|
|
|
from pylinkirc.coremods.login import pwd_context
|
|
|
|
default_permissions = {"*!*@*": ['commands.status', 'commands.showuser', 'commands.showchan']}
|
|
|
|
def main(irc=None):
|
|
"""Commands plugin main function, called on plugin load."""
|
|
# Register our permissions.
|
|
permissions.add_default_permissions(default_permissions)
|
|
|
|
def die(irc=None):
|
|
"""Commands plugin die function, called on plugin unload."""
|
|
permissions.remove_default_permissions(default_permissions)
|
|
|
|
@utils.add_cmd
|
|
def status(irc, source, args):
|
|
"""takes no arguments.
|
|
|
|
Returns your current PyLink login status."""
|
|
permissions.check_permissions(irc, source, ['commands.status'])
|
|
identified = irc.users[source].account
|
|
if identified:
|
|
irc.reply('You are identified as \x02%s\x02.' % identified)
|
|
else:
|
|
irc.reply('You are not identified as anyone.')
|
|
irc.reply('Operator access: \x02%s\x02' % bool(irc.is_oper(source)))
|
|
|
|
_none = '\x1D(none)\x1D'
|
|
def _do_showuser(irc, source, u):
|
|
"""Helper function for showuser."""
|
|
# Some protocol modules store UIDs as ints; make sure we check for that.
|
|
try:
|
|
int_u = int(u)
|
|
except ValueError:
|
|
pass
|
|
else:
|
|
if int_u in irc.users:
|
|
u = int_u
|
|
|
|
# Only show private info if the person is calling 'showuser' on themselves,
|
|
# or is an oper.
|
|
verbose = irc.is_oper(source) or u == source
|
|
|
|
if u not in irc.users:
|
|
irc.error('Unknown user %r.' % u)
|
|
return
|
|
|
|
f = lambda s: irc.reply(s, private=True)
|
|
|
|
userobj = irc.users[u]
|
|
f('Showing information on user \x02%s\x02 (%s@%s): %s' % (userobj.nick, userobj.ident,
|
|
userobj.host, userobj.realname))
|
|
|
|
sid = irc.get_server(u)
|
|
serverobj = irc.servers[sid]
|
|
ts = userobj.ts
|
|
|
|
# Show connected server & nick TS if available
|
|
serverinfo = '%s[%s]' % (serverobj.name, sid) \
|
|
if irc.has_cap('can-track-servers') else 'N/A'
|
|
tsinfo = '%s [UTC] (%s)' % (time.asctime(time.gmtime(int(ts))), ts) \
|
|
if irc.has_cap('has-ts') else 'N/A'
|
|
f('\x02Home server\x02: %s; \x02Nick TS:\x02 %s' % (serverinfo, tsinfo))
|
|
|
|
if verbose: # Oper/self only data: user modes, channels in, account info, etc.
|
|
f('\x02User modes\x02: %s' % irc.join_modes(userobj.modes, sort=True))
|
|
f('\x02Protocol UID\x02: %s; \x02Real host\x02: %s; \x02IP\x02: %s' % \
|
|
(u, userobj.realhost, userobj.ip))
|
|
channels = sorted(userobj.channels)
|
|
f('\x02Channels\x02: %s' % (' '.join(map(str, channels)) or _none))
|
|
f('\x02PyLink identification\x02: %s; \x02Services account\x02: %s; \x02Away status\x02: %s' % \
|
|
((userobj.account or _none), (userobj.services_account or _none), userobj.away or _none))
|
|
|
|
# Show relay user data if available
|
|
relay = world.plugins.get('relay')
|
|
if relay:
|
|
try:
|
|
userpair = relay.get_orig_user(irc, u) or (irc.name, u)
|
|
remoteusers = relay.relayusers[userpair].items()
|
|
except KeyError:
|
|
pass
|
|
else:
|
|
nicks = []
|
|
if remoteusers:
|
|
# Display all of the user's relay subclients, if there are any
|
|
nicks.append('%s:\x02%s\x02' % (userpair[0],
|
|
world.networkobjects[userpair[0]].users[userpair[1]].nick))
|
|
for r in remoteusers:
|
|
remotenet, remoteuser = r
|
|
remoteirc = world.networkobjects[remotenet]
|
|
nicks.append('%s:\x02%s\x02' % (remotenet, remoteirc.users[remoteuser].nick))
|
|
irc.reply("\x02Relay nicks\x02: %s" % ', '.join(nicks), private=True)
|
|
if verbose:
|
|
# Show the relay channels the user is in, if applicable
|
|
relaychannels = []
|
|
for ch in irc.users[u].channels:
|
|
relayentry = relay.get_relay(irc, ch)
|
|
if relayentry:
|
|
relaychannels.append(''.join(relayentry))
|
|
if relaychannels and verbose:
|
|
irc.reply("\x02Relay channels\x02: %s" % ' '.join(relaychannels), private=True)
|
|
|
|
@utils.add_cmd
|
|
def showuser(irc, source, args):
|
|
"""<user>
|
|
|
|
Shows information about <user>."""
|
|
permissions.check_permissions(irc, source, ['commands.showuser'])
|
|
try:
|
|
target = args[0]
|
|
except IndexError:
|
|
irc.error("Not enough arguments. Needs 1: nick.")
|
|
return
|
|
|
|
users = irc.nick_to_uid(target, multi=True) or [target]
|
|
|
|
for user in users:
|
|
_do_showuser(irc, source, user)
|
|
|
|
@utils.add_cmd
|
|
def showchan(irc, source, args):
|
|
"""<channel>
|
|
|
|
Shows information about <channel>."""
|
|
permissions.check_permissions(irc, source, ['commands.showchan'])
|
|
try:
|
|
channel = args[0]
|
|
except IndexError:
|
|
irc.error("Not enough arguments. Needs 1: channel.")
|
|
return
|
|
if channel not in irc.channels:
|
|
irc.error('Unknown channel %r.' % channel)
|
|
return
|
|
|
|
f = lambda s: irc.reply(s, private=True)
|
|
|
|
c = irc.channels[channel]
|
|
# Only show verbose info if caller is oper or is in the target channel.
|
|
verbose = source in c.users or irc.is_oper(source)
|
|
secret = ('s', None) in c.modes
|
|
if secret and not verbose:
|
|
# Hide secret channels from normal users.
|
|
irc.error('Unknown channel %r.' % channel)
|
|
return
|
|
|
|
nicks = [irc.users[u].nick for u in c.users]
|
|
|
|
f('Information on channel \x02%s\x02:' % channel)
|
|
if c.topic:
|
|
f('\x02Channel topic\x02: %s' % c.topic)
|
|
|
|
# Mark TS values as untrusted on Clientbot and others (where TS is read-only or not trackable)
|
|
f('\x02Channel creation time\x02: %s (%s) [UTC]%s' %
|
|
(time.asctime(time.gmtime(int(c.ts))), c.ts,
|
|
' [UNTRUSTED]' if not irc.has_cap('has-ts') else ''))
|
|
|
|
# Show only modes that aren't list-style modes.
|
|
modes = irc.join_modes([m for m in c.modes if m[0] not in irc.cmodes['*A']], sort=True)
|
|
f('\x02Channel modes\x02: %s' % modes)
|
|
if verbose:
|
|
nicklist = []
|
|
# Iterate over the user list, sorted by nick.
|
|
for user, nick in sorted(zip(c.users, nicks),
|
|
key=lambda userpair: userpair[1].lower()):
|
|
# Note: reversed() is used here because we're adding prefixes onto the nick in reverse
|
|
for pmode in reversed(c.get_prefix_modes(user)):
|
|
# Show prefix modes in order from highest to lowest.
|
|
nick = irc.prefixmodes.get(irc.cmodes.get(pmode, ''), '') + nick
|
|
nicklist.append(nick)
|
|
|
|
f('\x02User list\x02: %s' % ' '.join(nicklist))
|
|
|
|
# Show relay info, if applicable
|
|
relay = world.plugins.get('relay')
|
|
if relay:
|
|
relayentry = relay.get_relay(irc, channel)
|
|
if relayentry:
|
|
relays = ['\x02%s\x02' % ''.join(relayentry)]
|
|
relays += [''.join(link) for link in relay.db[relayentry]['links']]
|
|
f('\x02Relayed channels:\x02 %s' % (' '.join(relays)))
|
|
|
|
@utils.add_cmd
|
|
def version(irc, source, args):
|
|
"""takes no arguments.
|
|
|
|
Returns the version of the currently running PyLink instance."""
|
|
irc.reply("PyLink version \x02%s\x02 (in VCS: %s), released under the Mozilla Public License version 2.0." % (__version__, real_version))
|
|
irc.reply("The source of this program is available at \x02%s\x02." % world.source)
|
|
|
|
@utils.add_cmd
|
|
def echo(irc, source, args):
|
|
"""<text>
|
|
|
|
Echoes the text given."""
|
|
permissions.check_permissions(irc, source, ['commands.echo'])
|
|
if not args:
|
|
irc.error('No text to send!')
|
|
return
|
|
irc.reply(' '.join(args))
|
|
|
|
def _check_logout_access(irc, source, target, perms):
|
|
"""
|
|
Checks whether the source UID has access to log out the target UID.
|
|
This returns True if the source user has a permission specified,
|
|
or if the source and target are both logged in and have the same account.
|
|
"""
|
|
assert source in irc.users, "Unknown source user"
|
|
assert target in irc.users, "Unknown target user"
|
|
try:
|
|
permissions.check_permissions(irc, source, perms)
|
|
except utils.NotAuthorizedError:
|
|
if irc.users[source].account and (irc.users[source].account == irc.users[target].account):
|
|
return True
|
|
else:
|
|
raise
|
|
else:
|
|
return True
|
|
|
|
@utils.add_cmd
|
|
def logout(irc, source, args):
|
|
"""[<other nick/UID>]
|
|
|
|
Logs your account out of PyLink. If you have the 'commands.logout.force' permission, or are
|
|
attempting to log out yourself, you can also specify a nick to force a logout for."""
|
|
|
|
try:
|
|
othernick = args[0]
|
|
except IndexError: # No user specified
|
|
if irc.users[source].account:
|
|
irc.users[source].account = ''
|
|
else:
|
|
irc.error("You are not logged in!")
|
|
return
|
|
else:
|
|
otheruid = irc.nick_to_uid(othernick)
|
|
if not otheruid:
|
|
irc.error("Unknown user %s." % othernick)
|
|
return
|
|
else:
|
|
_check_logout_access(irc, source, otheruid, ['commands.logout.force'])
|
|
if irc.users[otheruid].account:
|
|
irc.users[otheruid].account = ''
|
|
else:
|
|
irc.error("%s is not logged in." % othernick)
|
|
return
|
|
|
|
irc.reply("Done.")
|
|
|
|
loglevels = {'DEBUG': 10, 'INFO': 20, 'WARNING': 30, 'ERROR': 40, 'CRITICAL': 50}
|
|
@utils.add_cmd
|
|
def loglevel(irc, source, args):
|
|
"""<level>
|
|
|
|
Sets the log level to the given <level>. <level> must be either DEBUG, INFO, WARNING, ERROR, or CRITICAL.
|
|
If no log level is given, shows the current one."""
|
|
permissions.check_permissions(irc, source, ['commands.loglevel'])
|
|
try:
|
|
level = args[0].upper()
|
|
try:
|
|
loglevel = loglevels[level]
|
|
except KeyError:
|
|
irc.error('Unknown log level "%s".' % level)
|
|
return
|
|
else:
|
|
world.console_handler.setLevel(loglevel)
|
|
irc.reply("Done.")
|
|
except IndexError:
|
|
irc.reply(world.console_handler.level)
|
|
|
|
@utils.add_cmd
|
|
def mkpasswd(irc, source, args):
|
|
"""<password>
|
|
Hashes a password for use in the configuration file."""
|
|
# TODO: restrict to only certain users?
|
|
try:
|
|
password = args[0]
|
|
except IndexError:
|
|
irc.error("Not enough arguments. (Needs 1, password)")
|
|
return
|
|
if not password:
|
|
irc.error("Password cannot be empty.")
|
|
return
|
|
|
|
if not pwd_context:
|
|
irc.error("Password encryption is not available (missing passlib).")
|
|
return
|
|
|
|
hashed_pass = pwd_context.encrypt(password)
|
|
irc.reply(hashed_pass, private=True)
|