libpurple/purple-url-handler

   1 #!/usr/bin/env python
   2
   3 import dbus
   4 import re
   5 import sys
   6 import time
   7 import urllib
   8
   9 bus = dbus.SessionBus()
  10 obj = None
  11 try:
  12     obj = bus.get_object("im.pidgin.purple.PurpleService",
  13                          "/im/pidgin/purple/PurpleObject")
  14 except dbus.DBusException, e:
  15     if e._dbus_error_name == "org.freedesktop.DBus.Error.ServiceUnknown":
  16         print "Error: no libpurple-powered client is running. Try starting Pidgin or Finch."
  17         sys.exit(1)
  18 purple = dbus.Interface(obj, "im.pidgin.purple.PurpleInterface")
  19
  20 class CheckedObject:
  21     def __init__(self, obj):
  22         self.obj = obj
  23
  24     def __getattr__(self, attr):
  25         return CheckedAttribute(self, attr)
  26
  27 class CheckedAttribute:
  28     def __init__(self, cobj, attr):
  29         self.cobj = cobj
  30         self.attr = attr
  31
  32     def __call__(self, *args):
  33         # Redirect stderr to suppress the printing of an " Introspect error"
  34         # message if nothing is listening on the bus.  We print a friendly
  35         # error message ourselves.
  36         real_stderr = sys.stderr
  37         sys.stderr = None
  38         result = self.cobj.obj.__getattr__(self.attr)(*args)
  39         sys.stderr = real_stderr
  40
  41 # This can be useful for debugging.
  42 #        if (result == 0):
  43 #            print "Error: " + self.attr + " " + str(args) + " returned " + str(result)
  44
  45         return result
  46
  47 cpurple = CheckedObject(purple)
  48
  49 def extendlist(list, length, fill):
  50     if len(list) < length:
  51         return list + [fill] * (length - len(list))
  52     else:
  53         return list
  54
  55 def convert(value):
  56     try:
  57         return int(value)
  58     except:
  59         return value
  60
  61 def account_not_found():
  62     print "No matching account found."
  63     sys.exit(1)
  64
  65 def bring_account_online(account):
  66     if not cpurple.PurpleAccountIsConnected(account):
  67         # The last argument is meant to be a GList * but the D-Bus binding
  68         # generator thing just wants a UInt32, which is pretty failing.
  69         # Happily, passing a 0 to mean an empty list turns out to work anyway.
  70         purple.PurpleAccountSetStatusList(account, "online", 1, 0)
  71         purple.PurpleAccountConnect(account)
  72
  73 def findaccount(protocolname, accountname="", matcher=None):
  74     if matcher:
  75         for account in cpurple.PurpleAccountsGetAll():
  76             if accountname != "" and accountname != cpurple.PurpleAccountGetUsername(a):
  77                 continue
  78             if matcher(account):
  79                 bring_account_online(account)
  80                 return account
  81         account_not_found()
  82
  83     # prefer connected accounts
  84     account = cpurple.PurpleAccountsFindConnected(accountname, protocolname)
  85     if (account != 0):
  86         return account
  87
  88     # try to get any account and connect it
  89     account = cpurple.PurpleAccountsFindAny(accountname, protocolname)
  90     if (account == 0):
  91         account_not_found()
  92
  93     bring_account_online(account)
  94     return account
  95
  96 def goim(account, screenname, message=None):
  97     # XXX: 1 == PURPLE_CONV_TYPE_IM
  98     conversation = cpurple.PurpleConversationNew(1, account, screenname)
  99     if message:
 100         purple.PurpleConvSendConfirm(conversation, message)
 101
 102 def gochat(account, params, message=None):
 103     connection = cpurple.PurpleAccountGetConnection(account)
 104     purple.ServJoinChat(connection, params)
 105
 106     if message != None:
 107         for i in range(20):
 108             # XXX: 2 == PURPLE_CONV_TYPE_CHAT
 109             conversation = purple.PurpleFindConversationWithAccount(2, params.get("channel", params.get("room")), account)
 110             if conversation:
 111                 purple.PurpleConvSendConfirm(conversation, message)
 112                 break
 113             else:
 114                 time.sleep(0.5)
 115
 116 def addbuddy(account, screenname, group="", alias=""):
 117     cpurple.PurpleBlistRequestAddBuddy(account, screenname, group, alias)
 118
 119
 120 def aim(uri):
 121     protocol = "prpl-aim"
 122     match = re.match(r"^aim:([^?]*)(\?(.*))", uri)
 123     if not match:
 124         print "Invalid aim URI: %s" % uri
 125         return
 126
 127     command = urllib.unquote_plus(match.group(1))
 128     paramstring = match.group(3)
 129     params = {}
 130     if paramstring:
 131         for param in paramstring.split("&"):
 132             key, value = extendlist(param.split("=", 1), 2, "")
 133             params[key] = urllib.unquote_plus(value)
 134     accountname = params.get("account", "")
 135     screenname = params.get("screenname", "")
 136
 137     account = findaccount(protocol, accountname)
 138
 139     if command.lower() == "goim":
 140         goim(account, screenname, params.get("message"))
 141     elif command.lower() == "gochat":
 142         gochat(account, params)
 143     elif command.lower() == "addbuddy":
 144         addbuddy(account, screenname, params.get("group", ""))
 145
 146 def gg(uri):
 147     protocol = "prpl-gg"
 148     match = re.match(r"^gg:(.*)", uri)
 149     if not match:
 150         print "Invalid gg URI: %s" % uri
 151         return
 152
 153     screenname = urllib.unquote_plus(match.group(1))
 154     account = findaccount(protocol)
 155     goim(account, screenname)
 156
 157 def icq(uri):
 158     protocol = "prpl-icq"
 159     match = re.match(r"^icq:([^?]*)(\?(.*))", uri)
 160     if not match:
 161         print "Invalid icq URI: %s" % uri
 162         return
 163
 164     command = urllib.unquote_plus(match.group(1))
 165     paramstring = match.group(3)
 166     params = {}
 167     if paramstring:
 168         for param in paramstring.split("&"):
 169             key, value = extendlist(param.split("=", 1), 2, "")
 170             params[key] = urllib.unquote_plus(value)
 171     accountname = params.get("account", "")
 172     screenname = params.get("screenname", "")
 173
 174     account = findaccount(protocol, accountname)
 175
 176     if command.lower() == "goim":
 177         goim(account, screenname, params.get("message"))
 178     elif command.lower() == "gochat":
 179         gochat(account, params)
 180     elif command.lower() == "addbuddy":
 181         addbuddy(account, screenname, params.get("group", ""))
 182
 183 def irc(uri):
 184     protocol = "prpl-irc"
 185     match = re.match(r"^irc:(//([^/]*)/)?([^?]*)(\?(.*))?", uri)
 186     if not match:
 187         print "Invalid irc URI: %s" % uri
 188         return
 189
 190     server = urllib.unquote_plus(match.group(2)) or ""
 191     target = match.group(3) or ""
 192     query = match.group(5) or ""
 193
 194     modifiers = {}
 195     if target:
 196         for modifier in target.split(",")[1:]:
 197             modifiers[modifier] = True
 198
 199     isnick = modifiers.has_key("isnick")
 200
 201     paramstring = match.group(5)
 202     params = {}
 203     if paramstring:
 204         for param in paramstring.split("&"):
 205             key, value = extendlist(param.split("=", 1), 2, "")
 206             params[key] = urllib.unquote_plus(value)
 207
 208     def correct_server(account):
 209         username = cpurple.PurpleAccountGetUsername(account)
 210         return ("@" in username) and (server == (username.split("@"))[1])
 211
 212     account = findaccount(protocol, matcher=correct_server)
 213
 214     if (target != ""):
 215         if (isnick):
 216             goim(account, urllib.unquote_plus(target.split(",")[0]), params.get("msg"))
 217         else:
 218             channel = urllib.unquote_plus(target.split(",")[0])
 219             if channel[0] != "#":
 220                 channel = "#" + channel
 221             gochat(account, {"server": server, "channel": channel, "password": params.get("key", "")}, params.get("msg"))
 222
 223 def msnim(uri):
 224     protocol = "prpl-msn"
 225     match = re.match(r"^msnim:([^?]*)(\?(.*))", uri)
 226     if not match:
 227         print "Invalid msnim URI: %s" % uri
 228         return
 229
 230     command = urllib.unquote_plus(match.group(1))
 231     paramstring = match.group(3)
 232     params = {}
 233     if paramstring:
 234         for param in paramstring.split("&"):
 235             key, value = extendlist(param.split("=", 1), 2, "")
 236             params[key] = urllib.unquote_plus(value)
 237     screenname = params.get("contact", "")
 238
 239     account = findaccount(protocol)
 240
 241     if command.lower() == "chat":
 242         goim(account, screenname)
 243     elif command.lower() == "add":
 244         addbuddy(account, screenname)
 245
 246 def myim(uri):
 247         protocol = "prpl-myspace"
 248         print "TODO: send uri: ", uri
 249         assert False, "Not implemented"
 250
 251 def sip(uri):
 252     protocol = "prpl-simple"
 253     match = re.match(r"^sip:(.*)", uri)
 254     if not match:
 255         print "Invalid sip URI: %s" % uri
 256         return
 257
 258     screenname = urllib.unquote_plus(match.group(1))
 259     account = findaccount(protocol)
 260     goim(account, screenname)
 261
 262 def xmpp(uri):
 263     protocol = "prpl-jabber"
 264     match = re.match(r"^xmpp:(//([^/?#]*)/?)?([^?#]*)(\?([^;#]*)(;([^#]*))?)?(#(.*))?", uri)
 265     if not match:
 266         print "Invalid xmpp URI: %s" % uri
 267         return
 268
 269     tmp = match.group(2)
 270     if (tmp):
 271         accountname = urllib.unquote_plus(tmp)
 272     else:
 273         accountname = ""
 274
 275     screenname = urllib.unquote_plus(match.group(3))
 276
 277     tmp = match.group(5)
 278     if (tmp):
 279         command = urllib.unquote_plus(tmp)
 280     else:
 281         command = ""
 282
 283     paramstring = match.group(7)
 284     params = {}
 285     if paramstring:
 286         for param in paramstring.split(";"):
 287             key, value = extendlist(param.split("=", 1), 2, "")
 288             params[key] = urllib.unquote_plus(value)
 289
 290     account = findaccount(protocol, accountname)
 291
 292     if command.lower() == "message":
 293         goim(account, screenname, params.get("body"))
 294     elif command.lower() == "join":
 295         room, server = screenname.split("@")
 296         gochat(account, {"room": room, "server": server})
 297     elif command.lower() == "roster":
 298         addbuddy(account, screenname, params.get("group", ""), params.get("name", ""))
 299     else:
 300         goim(account, screenname)
 301
 302 def gtalk(uri):
 303     protocol = "prpl-jabber"
 304     match = re.match(r"^gtalk:([^?]*)(\?(.*))", uri)
 305     if not match:
 306         print "Invalid gtalk URI: %s" % uri
 307         return
 308
 309     command = urllib.unquote_plus(match.group(1))
 310     paramstring = match.group(3)
 311     params = {}
 312     if paramstring:
 313         for param in paramstring.split("&"):
 314             key, value = extendlist(param.split("=", 1), 2, "")
 315             params[key] = urllib.unquote_plus(value)
 316     accountname = params.get("from_jid", "")
 317     jid = params.get("jid", "")
 318
 319     account = findaccount(protocol, accountname)
 320
 321     if command.lower() == "chat":
 322         goim(account, jid)
 323     elif command.lower() == "call":
 324         # XXX V&V prompt to establish call
 325         goim(account, jid)
 326
 327 def ymsgr(uri):
 328     protocol = "prpl-yahoo"
 329     match = re.match(r"^ymsgr:([^?]*)(\?([^&]*)(&(.*))?)", uri)
 330     if not match:
 331         print "Invalid ymsgr URI: %s" % uri
 332         return
 333
 334     command = urllib.unquote_plus(match.group(1))
 335     screenname = urllib.unquote_plus(match.group(3))
 336     paramstring = match.group(5)
 337     params = {}
 338     if paramstring:
 339         for param in paramstring.split("&"):
 340             key, value = extendlist(param.split("=", 1), 2, "")
 341             params[key] = urllib.unquote_plus(value)
 342
 343     account = findaccount(protocol)
 344
 345     if command.lower() == "sendim":
 346         goim(account, screenname, params.get("m"))
 347     elif command.lower() == "chat":
 348         gochat(account, {"room": screenname})
 349     elif command.lower() == "addfriend":
 350         addbuddy(account, screenname)
 351
 352
 353 def main(argv=sys.argv):
 354     if len(argv) != 2 or argv[1] == "--help" or argv[1] == "-h":
 355         print "Usage: %s URI" % argv[0]
 356         print "Example: %s \"xmpp:romeo@montague.net?message\"" % argv[0]
 357
 358         if len(argv) != 2:
 359             sys.exit(1)
 360         else:
 361             return 0
 362
 363     uri = argv[1]
 364     type = uri.split(":")[0]
 365
 366     try:
 367         if type == "aim":
 368             aim(uri)
 369         elif type == "gg":
 370             gg(uri)
 371         elif type == "icq":
 372             icq(uri)
 373         elif type == "irc":
 374             irc(uri)
 375         elif type == "msnim":
 376             msnim(uri)
 377         elif type == "myim":
 378             myim(uri)
 379         elif type == "sip":
 380             sip(uri)
 381         elif type == "xmpp":
 382             xmpp(uri)
 383         elif type == "gtalk":
 384             gtalk(uri)
 385         elif type == "ymsgr":
 386             ymsgr(uri)
 387         else:
 388             print "Unknown protocol: %s" % type
 389     except dbus.DBusException, e:
 390         print "Error: %s" % (e.message)
 391         sys.exit(1)
 392
 393 if __name__ == "__main__":
 394     main()