1
0
mirror of https://github.com/profanity-im/profanity.git synced 2024-12-04 14:46:46 -05:00
profanity/plugins/www-view.py
James Booth ffb1359889 Added get_current_recipient to api
Browser plugin now uses last link received in a chat window
2013-08-11 01:35:11 +01:00

63 lines
1.5 KiB
Python

import prof
import os
import webbrowser
import re
lastlink = {}
# hooks
def prof_init(version, status):
prof.register_command("/browser", 0, 1,
"/browser [url]",
"View a URL in the browser.",
"View a URL in the browser, if no argument is supplied, " +
"the last received URL will be used.",
cmd_browser)
def prof_on_message(jid, message):
global lastlink
links = re.findall(r'(https?://\S+)', message)
if (len(links) > 0):
lastlink[jid] = links[len(links)-1]
# commands
def cmd_browser(url):
global lastlink
link = None
# use arg if supplied
if (url != None):
link = url
else:
jid = prof.get_current_recipient();
# check if in chat window
if (jid != None):
# check for link from recipient
if jid in lastlink.keys():
link = lastlink[jid]
else:
prof.cons_show("No links found from " + jid);
# not in chat window
else:
prof.cons_show("You must supply a URL to the /browser command")
# open the browser if link found
if (link != 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)