Restructure project

stable/browser.py

@@ -0,0 +1,121 @@
+"""
+Open the last link received in a chat or room window using the systems default browser
+"""
+
+import prof
+import os
+import webbrowser
+import re
+
+_links = {}
+_lastlink = {}
+
+
+def _cmd_browser(url):
+    global _lastlink
+    link = None
+
+    # use arg if supplied
+    if (url is not None):
+        link = url
+    else:
+        jid = prof.get_current_recipient()
+        room = prof.get_current_muc()
+
+        # check if in chat window
+        if (jid is not None):
+
+            # check for link from recipient
+            if jid in _lastlink.keys():
+                link = _lastlink[jid]
+            else:
+                prof.cons_show("No links found from " + jid)
+
+        # check if in muc window
+        elif (room is not None):
+            if room in _lastlink.keys():
+                link = _lastlink[room]
+            else:
+                prof.cons_show("No links found from " + room)
+
+        # not in chat/muc window
+        else:
+            prof.cons_show("You must supply a URL to the /browser command")
+
+    # open the browser if link found
+    if (link is not None):
+        prof.cons_show("Opening " + link + " in browser")
+        _open_browser(link)
+
+
+def _open_browser(url):
+    savout = os.dup(1)
+    saverr = os.dup(2)
+    os.close(1)
+    os.close(2)
+    os.open(os.devnull, os.O_RDWR)
+    try:
+        webbrowser.open(url, new=2)
+    finally:
+        os.dup2(savout, 1)
+        os.dup2(saverr, 2)
+
+
+def prof_init(version, status, account_name, fulljid):
+    synopsis = [ 
+        "/browser",
+        "/browser <url>"
+    ]
+    description = "View a URL in the systems default browser. If no argument is supplied, the last URL in the current chat or room will be used. " \
+        "Tab autocompletion will go through all previous links in the current chat/room"
+    args = [
+        [ "<url>", "URL to open in the browser" ]
+    ]
+    examples = [
+        "/browser http://www.profanity.im"
+    ]
+
+    prof.register_command("/browser", 0, 1, synopsis, description, args, examples, _cmd_browser)
+
+
+def prof_on_chat_win_focus(barejid):
+    prof.completer_clear("/browser")
+    if barejid in _links:
+        prof.completer_add("/browser", _links[barejid])
+
+
+def prof_on_room_win_focus(barejid):
+    prof.completer_clear("/browser")
+    if barejid in _links:
+        prof.completer_add("/browser", _links[barejid])
+
+
+def _process_message(barejid, current_jid, message):
+    links = re.findall(r'(https?://\S+)', message)
+    if (len(links) > 0):
+        if barejid not in _links:
+            _links[barejid] = []
+        # add to list of links for barejid
+        for link in links:
+            if link not in _links[barejid]:
+                _links[barejid].append(link)
+        # add to autocompleter if message for current window
+        if current_jid == barejid:
+            prof.completer_add("/browser", _links[barejid])
+        # set last link for barejid
+        _lastlink[barejid] = links[len(links)-1]
+
+
+def prof_post_chat_message_display(barejid, resource, message):
+    current_jid = prof.get_current_recipient()
+    _process_message(barejid, current_jid, message)
+
+
+def prof_post_room_message_display(barejid, nick, message):
+    current_jid = prof.get_current_muc()
+    _process_message(barejid, current_jid, message)
+
+
+def prof_on_room_history_message(barejid, nick, message, timestamp):
+    current_jid = prof.get_current_muc()
+    _process_message(barejid, current_jid, message)