178 lines
7.4 KiB
Python
178 lines
7.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import gtk
|
|
import json
|
|
import urllib
|
|
import urllib2
|
|
from common import gajim
|
|
from plugins import GajimPlugin
|
|
from plugins.helpers import log_calls
|
|
from plugins.gui import GajimPluginConfigDialog
|
|
|
|
class UrlShortenerPlugin(GajimPlugin):
|
|
@log_calls('UrlShortenerPlugin')
|
|
def init(self):
|
|
self.description = _('Plugin that allows users to shorten a long URL '
|
|
'in received messages.\n'
|
|
'For example, you can turn this link:\n'
|
|
'http://maps.google.com/maps?f=d&saddr=New+York+Penn+Station&daddr='
|
|
'9th+Ave+%26+14th+St,+New+York,+NY&hl=en&geocode=&mra=ls&dirflg=r&date='
|
|
'11%2F12%2F08&time=4:13pm&ttype=dep&noexp=0&noal=0&sort=&sll='
|
|
'40.746175,-73.998395&sspn=0.014468,0.036392&ie=UTF8&z=14\n'
|
|
'Into this link:\n'
|
|
'http://bit.ly/CUjV')
|
|
self.config_dialog = UrlShortenerPluginConfigDialog(self)
|
|
self.gui_extension_points = {
|
|
'chat_control_base': (self.connect_with_chat_control,
|
|
self.disconnect_from_chat_control),
|
|
'print_special_text': (self.print_special_text,
|
|
self.print_special_text1),}
|
|
self.config_default_values = {
|
|
'MAX_CHARS': (50, _('MAX_CHARS(30-...)'))}
|
|
self.chat_control = None
|
|
self.controls = []
|
|
|
|
@log_calls('UrlShortenerPlugin')
|
|
def connect_with_chat_control(self, chat_control):
|
|
self.chat_control = chat_control
|
|
control = Base(self, self.chat_control)
|
|
self.controls.append(control)
|
|
|
|
@log_calls('UrlShortenerPlugin')
|
|
def disconnect_from_chat_control(self, chat_control):
|
|
for control in self.controls:
|
|
control.disconnect_from_chat_control()
|
|
self.controls = []
|
|
|
|
def print_special_text(self, tv, special_text, other_tags, graphics=True):
|
|
for control in self.controls:
|
|
if control.chat_control.conv_textview != tv:
|
|
continue
|
|
control.print_special_text(special_text, other_tags, graphics=True)
|
|
|
|
def print_special_text1(self, chat_control, special_text, other_tags=None,
|
|
graphics=True):
|
|
for control in self.controls:
|
|
if control.chat_control == chat_control:
|
|
control.disconnect_from_chat_control()
|
|
self.controls.remove(control)
|
|
|
|
class Base(object):
|
|
def __init__(self, plugin, chat_control):
|
|
self.user = 'dicson'
|
|
self.apikey = 'R_fcba926fc7978bd19acbca73ec82b2be'
|
|
self.plugin = plugin
|
|
self.chat_control = chat_control
|
|
self.textview = self.chat_control.conv_textview
|
|
|
|
self.id_ = self.textview.tv.connect('motion_notify_event',
|
|
self.on_textview_motion_notify_event)
|
|
self.chat_control.handlers[self.id_] = self.textview.tv
|
|
|
|
def print_special_text(self, special_text, other_tags, graphics=True):
|
|
is_xhtml_link = None
|
|
text_is_valid_uri = False
|
|
buffer_ = self.textview.tv.get_buffer()
|
|
|
|
# Detect XHTML-IM link
|
|
ttt = buffer_.get_tag_table()
|
|
tags_ = [(ttt.lookup(t) if isinstance(t, str) else t) for t in other_tags]
|
|
for t in tags_:
|
|
is_xhtml_link = getattr(t, 'href', None)
|
|
if is_xhtml_link:
|
|
break
|
|
# Check if we accept this as an uri
|
|
schemes = gajim.config.get('uri_schemes').split()
|
|
for scheme in schemes:
|
|
if special_text.startswith(scheme):
|
|
text_is_valid_uri = True
|
|
if special_text.startswith('www.') or special_text.startswith('ftp.') \
|
|
or text_is_valid_uri and not is_xhtml_link:
|
|
if len(special_text) < self.plugin.config['MAX_CHARS']:
|
|
return
|
|
end_iter = buffer_.get_end_iter()
|
|
mark = buffer_.create_mark(None, end_iter, True)
|
|
gajim.thread_interface(self.insert_hyperlink, [mark, special_text,
|
|
ttt])
|
|
self.textview.plugin_modified = True
|
|
|
|
def insert_hyperlink(self, mark, special_text, ttt):
|
|
try:
|
|
params = urllib.urlencode({'longUrl': special_text,
|
|
'login': self.user,
|
|
'apiKey': self.apikey,
|
|
'format': 'json'})
|
|
req = urllib2.Request('http://api.bit.ly/v3/shorten?%s' % params)
|
|
response = urllib2.urlopen(req)
|
|
j = json.load(response)
|
|
if j['status_code'] == 200:
|
|
special_text = j['data']['url']
|
|
except urllib2.HTTPError, e:
|
|
pass
|
|
|
|
buffer_ = mark.get_buffer()
|
|
end_iter = buffer_.get_iter_at_mark(mark)
|
|
buffer_.insert_with_tags(end_iter, special_text, ttt.lookup('url'))
|
|
|
|
def on_textview_motion_notify_event(self, widget, event):
|
|
pointer_x, pointer_y = self.textview.tv.window.get_pointer()[0:2]
|
|
x, y = self.textview.tv.window_to_buffer_coords(gtk.TEXT_WINDOW_TEXT,
|
|
pointer_x, pointer_y)
|
|
tags = self.textview.tv.get_iter_at_location(x, y).get_tags()
|
|
tag_table = self.textview.tv.get_buffer().get_tag_table()
|
|
buffer_ = self.textview.tv.get_buffer()
|
|
for tag in tags:
|
|
if tag != tag_table.lookup('url'):
|
|
continue
|
|
it = self.textview.tv.get_iter_at_location(x, y)
|
|
st = it.copy()
|
|
st.backward_to_tag_toggle(tag_table.lookup('url'))
|
|
it.forward_to_tag_toggle(tag_table.lookup('url'))
|
|
text = buffer_.get_text(st, it, include_hidden_chars=True)
|
|
if text.startswith('http://bit.ly/'):
|
|
try:
|
|
params = urllib.urlencode({'shortUrl': text,
|
|
'login': self.user,
|
|
'apiKey': self.apikey,
|
|
'format': 'json'})
|
|
req = urllib2.Request('http://api.bit.ly/v3/expand?%s' \
|
|
% params)
|
|
response = urllib2.urlopen(req)
|
|
j = json.load(response)
|
|
if j['status_code'] != 200:
|
|
raise Exception('%s'%j['status_txt'])
|
|
txt = j['data']['expand'][0]['long_url']
|
|
self.textview.tv.set_tooltip_text(txt)
|
|
self.textview.on_textview_motion_notify_event(widget, event)
|
|
return
|
|
except Exception, e:
|
|
break
|
|
|
|
self.textview.tv.set_tooltip_text('')
|
|
self.textview.on_textview_motion_notify_event(widget, event)
|
|
|
|
def disconnect_from_chat_control(self):
|
|
pass
|
|
|
|
|
|
class UrlShortenerPluginConfigDialog(GajimPluginConfigDialog):
|
|
def init(self):
|
|
self.GTK_BUILDER_FILE_PATH = self.plugin.local_file_path(
|
|
'config_dialog.ui')
|
|
self.xml = gtk.Builder()
|
|
self.xml.set_translation_domain('gajim_plugins')
|
|
self.xml.add_objects_from_file(self.GTK_BUILDER_FILE_PATH, ['hbox1'])
|
|
self.max_chars_spinbutton = self.xml.get_object('max_chars')
|
|
self.max_chars_spinbutton.get_adjustment().set_all(30, 30, 99999, 1,
|
|
10, 0)
|
|
hbox = self.xml.get_object('hbox1')
|
|
self.child.pack_start(hbox)
|
|
|
|
self.xml.connect_signals(self)
|
|
|
|
def on_run(self):
|
|
self.max_chars_spinbutton.set_value(self.plugin.config['MAX_CHARS'])
|
|
|
|
def avatar_size_value_changed(self, spinbutton):
|
|
self.plugin.config['MAX_CHARS'] = spinbutton.get_value()
|