poezio/plugins/irc.py
mathieui a4771623a5
Fix a bug in the IRC plugin
(forgot to add the arguments to one call)
2015-01-15 16:44:55 +01:00

376 lines
14 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

"""
Plugin destined to be used together with the Biboumi IRC gateway.
For more information about Biboumi, please see the `official website`_.
This plugin is here as a non-default extension of the poezio configuration
made to work with IRC rooms and logins. It also defines commands aimed at
reducing the amount of effort needed to navigate smoothly between IRC and
XMPP rooms.
Configuration
-------------
Global configuration
~~~~~~~~~~~~~~~~~~~~
.. glossary::
:sorted:
gateway
**Default:** ``irc.poez.io``
The JID of the IRC gateway to use. If empty, irc.poez.io will be
used. Please try to run your own, though, its painless to setup.
initial_connect
**Default:** ``true``
If you want to join all the rooms and try to authenticate with
nickserv when the plugin gets loaded. If ``false``, you will have
to use the :term:`/irc_login` command to authenticate, and the
:term:`/irc_join` command to join preconfigured rooms.
.. note:: There is no nickname option because the default from poezio will be used.
Server-specific configuration
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Write a configuration section for each server, with the server address as the
section name, and the following options:
.. glossary::
:sorted:
login_command
**Default:** ``[empty]``
The command used to identify with the services (e.g. ``IDENTIFY mypassword``).
login_nick
**Default:** ``[empty]``
The nickname to whom the auth command will be sent.
nickname
**Default:** ``[empty]``
Your nickname on this server. If empty, the default configuration will be used.
rooms [IRC plugin]
**Default:** ``[empty]``
The list of rooms to join on this server (e.g. ``#room1:#room2``).
.. note:: If no login_command or login_nick is set, the authentication phase
wont take place and you will join the rooms without authentication
with nickserv or whatever.
Commands
~~~~~~~~
.. glossary::
:sorted:
/irc_login
**Usage:** ``/irc_login [server1] [server2]…``
Authenticate with the specified servers if they are correctly
configured. If no servers are provided, the plugin will try
them all. (You need to set :term:`login_nick` and
:term:`login_command` as well)
/irc_join
**Usage:** ``/irc_join <room or server>``
Join the specified room on the same server as the current tab (can
be a private conversation or a chatroom). If a server that appears
in the conversation is specified instead of a room, the plugin
will try to join all the rooms configured with autojoin on that
server.
/irc_query
**Usage:** ``/irc_query <nickname> [message]``
Open a private conversation with the given nickname, on the same IRC
server as the current tab (can be a private conversation or a
chatroom). Doing `/irc_query foo "hello there"` when the current
tab is #foo%irc.example.com@biboumi.example.com is equivalent to
`/message foo!irc.example.com@biboumi.example.com "hello there"`
Example configuration
~~~~~~~~~~~~~~~~~~~~~
.. code-block:: ini
[irc]
gateway = irc.poez.io
[irc.freenode.net]
nickname = mynick
login_nick = nickserv
login_command = identify mypassword
rooms = #testroom1:#testroom2
[irc.geeknode.org]
nickname = anothernick
login_nick = C
login_command = nick identify mypassword
rooms = #testvroum
.. _official website: http://biboumi.louiz.org/
"""
from plugin import BasePlugin
from decorators import command_args_parser
import common
import tabs
class Plugin(BasePlugin):
def init(self):
if self.config.get('initial_connect', True):
self.initial_connect()
self.api.add_command('irc_login', self.command_irc_login,
usage='[server] [server]…',
help=('Connect to the specified servers if they '
'exist in the configuration and the login '
'options are set. If not is given, the '
'plugin will try all the sections in the '
'configuration.'),
short='Login to irc servers with nickserv',
completion=self.completion_irc_login)
self.api.add_command('irc_join', self.command_irc_join,
usage='<room or server>',
help=('Join <room> in the same server as the '
'current tab (if it is an IRC tab). Or '
'join all the preconfigured rooms in '
'<server> '),
short='Join irc rooms more easily',
completion=self.completion_irc_join)
self.api.add_command('irc_query', self.command_irc_query,
usage='<nickname> [message]',
help=('Open a private conversation with the '
'given <nickname>, on the current IRC '
'server. Optionally immediately send '
'the given message. For example, if the '
'current tab is #foo%irc.example.com@'
'biboumi.example.com, doing `/irc_query '
'nick "hi there"` is equivalent to '
'`/message nick!irc.example.com@biboumi.'
'example.com "hi there"`'),
short='Open a private conversation with an IRC user')
def join(self, gateway, server):
"Join irc rooms on a server"
nick = self.config.get_by_tabname('nickname', server, default='') or self.core.own_nick
rooms = self.config.get_by_tabname('rooms', server, default='').split(':')
for room in rooms:
room = '{}%{}@{}/{}'.format(room, server, gateway, nick)
self.core.command_join(room)
def initial_connect(self):
gateway = self.config.get('gateway', 'irc.poez.io')
sections = self.config.sections()
for section in (s for s in sections if s != 'irc'):
room_suffix = '%{}@{}'.format(section, gateway)
already_opened = False
for tab in self.core.tabs:
if tab.name.endswith(room_suffix) and tab.joined:
already_opened = True
break
login_command = self.config.get_by_tabname('login_command', section, default='')
login_nick = self.config.get_by_tabname('login_nick', section, default='')
nick = self.config.get_by_tabname('nickname', section, default='') or self.core.own_nick
if login_command and login_nick:
def login(gw, sect, log_nick, log_cmd, room_suff):
dest = '{}!{}'.format(log_nick, room_suff)
self.core.xmpp.send_message(mto=dest, mbody=log_cmd, mtype='chat')
delayed = self.api.create_delayed_event(5, self.join, gw, sect)
self.api.add_timed_event(delayed)
if not already_opened:
self.core.command_join(room_suffix + '/' + nick)
delayed = self.api.create_delayed_event(5, login, gateway, section,
login_nick, login_command,
room_suffix[1:])
self.api.add_timed_event(delayed)
else:
login(gateway, section, login_nick, login_command, room_suffix[1:])
elif not already_opened:
self.join(gateway, section)
@command_args_parser.quoted(0, -1)
def command_irc_login(self, args):
"""
/irc_login [server] [server]…
"""
gateway = self.config.get('gateway', 'irc.poez.io')
if args:
not_present = []
sections = self.config.sections()
for section in args:
if section not in sections:
not_present.append(section)
continue
login_command = self.config.get_by_tabname('login_command', section, default='')
login_nick = self.config.get_by_tabname('login_nick', section, default='')
if not login_command and not login_nick:
not_present.append(section)
continue
room_suffix = '%{}@{}'.format(section, gateway)
dest = '{}!{}'.format(login_nick, room_suffix[1:])
self.core.xmpp.send_message(mto=dest, mbody=login_command, mtype='chat')
if len(not_present) == 1:
self.api.information('Section %s does not exist or is not configured' % not_present[0], 'Warning')
elif len(not_present) > 1:
self.api.information('Sections %s do not exist or are not configured' % ', '.join(not_present), 'Warning')
else:
sections = self.config.sections()
for section in (s for s in sections if s != 'irc'):
login_command = self.config.get_by_tabname('login_command', section, default='')
login_nick = self.config.get_by_tabname('login_nick', section, default='')
if not login_nick and not login_command:
continue
room_suffix = '%{}@{}'.format(section, gateway)
dest = '{}!{}'.format(login_nick, room_suffix[1:])
self.core.xmpp.send_message(mto=dest, mbody=login_command, mtype='chat')
def completion_irc_login(self, the_input):
"""
completion for /irc_login
"""
args = the_input.text.split()
if '' in args:
args.remove('')
pos = the_input.get_argument_position()
sections = self.config.sections()
if 'irc' in sections:
sections.remove('irc')
for section in args:
try:
sections.remove(section)
except:
pass
return the_input.new_completion(sections, pos)
@command_args_parser.quoted(1, 1)
def command_irc_join(self, args):
"""
/irc_join <room or server>
"""
if not args:
return self.core.command_help('irc_join')
sections = self.config.sections()
if 'irc' in sections:
sections.remove('irc')
if args[0] in sections and self.config.get_by_tabname('rooms', args[0]):
self.join_server_rooms(args[0])
else:
self.join_room(args[0])
@command_args_parser.quoted(1, 1)
def command_irc_query(self, args):
"""
Open a private conversation with the given nickname, on the current IRC
server.
"""
if args is None:
return self.core.command_help('irc_query')
current_tab_info = self.get_current_tab_irc_info()
if not current_tab_info:
return
server, gateway = current_tab_info
nickname = args[0]
message = None
if len(args) == 2:
message = args[1]
jid = '{}!{}@{}'.format(nickname, server, gateway)
if message:
self.core.command_message('{} "{}"'.format(jid, message))
else:
self.core.command_message('{}'.format(jid))
def join_server_rooms(self, section):
"""
Join all the rooms configured for a section
(section = irc server)
"""
gateway = self.config.get('gateway', 'irc.poez.io')
rooms = self.config.get_by_tabname('rooms', section).split(':')
nick = self.config.get_by_tabname('nickname', section)
if nick:
nick = '/' + nick
else:
nick = ''
suffix = '%{}@{}{}'.format(section, gateway, nick)
for room in rooms:
self.core.command_join(room + suffix)
def join_room(self, name):
"""
Join a room with only its name and the current tab
"""
current_tab_info = self.get_current_tab_irc_info()
if not current_tab_info:
return
server, gateway = current_tab_info
room = '{}%{}@{}'.format(name, server, gateway)
if self.config.get_by_tabname('nickname', server):
room += '/' + self.config.get_by_tabname('nickname', server)
self.core.command_join(room)
def get_current_tab_irc_info(self):
"""
Return a tuple with the irc server and the gateway hostnames of the
current tab. If the current tab is not an IRC channel or private
conversation, a warning is displayed and None is returned
"""
gateway = self.config.get('gateway', 'irc.poez.io')
current = self.core.current_tab()
current_jid = common.safeJID(current.name)
if not current_jid.server == gateway:
self.api.information('The current tab does not appear to be an IRC one', 'Warning')
return None
if isinstance(current, tabs.OneToOneTab):
if not '!' in current_jid.node:
server = current_jid.node
else:
ignored, server = current_jid.node.rsplit('!', 1)
elif isinstance(current, tabs.MucTab):
if not '%' in current_jid.node:
server = current_jid.node
else:
ignored, server = current_jid.node.rsplit('%', 1)
else:
self.api.information('The current tab does not appear to be an IRC one', 'Warning')
return None
return server, gateway
def completion_irc_join(self, the_input):
"""
completion for /irc_join
"""
sections = self.config.sections()
if 'irc' in sections:
sections.remove('irc')
return the_input.new_completion(sections, 1)