diff options
Diffstat (limited to 'url_shortener/url_shortener.py')
| -rw-r--r-- | url_shortener/url_shortener.py | 232 |
1 files changed, 0 insertions, 232 deletions
diff --git a/url_shortener/url_shortener.py b/url_shortener/url_shortener.py deleted file mode 100644 index bd47efe..0000000 --- a/url_shortener/url_shortener.py +++ /dev/null @@ -1,232 +0,0 @@ -# -*- coding: utf-8 -*- - -import gtk -import json -import urllib -import urllib2 -from common import app -from common import ged -from plugins import GajimPlugin -from plugins.helpers import log_calls -from plugins.gui import GajimPluginConfigDialog - -APIKEY = 'R_fcba926fc7978bd19acbca73ec82b2be' -USER = 'dicson' - -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' - 'https://trac.gajim.org/timeline\n' - 'Into this link:\n' - 'http://bit.ly/THy6ZK') - 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-...)')), - 'IN_MAX_CHARS': (50, ('MAX_CHARS(30-...)')), - 'SHORTEN_OUTGOING': (False, ''),} - self.events_handlers = {'message-outgoing': (ged.OUT_PRECORE, - self.handle_outgoing_msg), - 'gc-message-outgoing': (ged.OUT_PRECORE, - self.handle_outgoing_msg)} - self.chat_control = None - self.controls = [] - - def handle_outgoing_msg(self, event): - if not self.active: - return - if not event.message: - return - if not self.config['SHORTEN_OUTGOING']: - return - if hasattr(event, 'shortened'): - return - - iterator = app.interface.basic_pattern_re.finditer(event.message) - for match in iterator: - start, end = match.span() - link = event.message[start:end] - if len(link) < self.config['MAX_CHARS']: - continue - short_link = None - try: - params = urllib.urlencode({'longUrl': link, - 'login': USER, - 'apiKey': 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: - short_link = j['data']['url'] - except urllib2.HTTPError, e: - pass - if short_link: - event.message = event.message.replace(link, short_link) - event.callback_args[1] = event.message - event.shortened = True - - @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, - additional_data={}): - 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, additional_data={}): - 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.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): - if not self.plugin.active: - return - 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 = app.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['IN_MAX_CHARS']: - return - end_iter = buffer_.get_end_iter() - mark = buffer_.create_mark(None, end_iter, True) - app.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': USER, - 'apiKey': 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': USER, - 'apiKey': 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, ['vbox1']) - self.max_chars_spinbutton = self.xml.get_object('max_chars') - self.max_chars_spinbutton.get_adjustment().set_all(30, 30, 99999, 1, - 10, 0) - self.in_max_chars_spinbutton = self.xml.get_object('in_max_chars') - self.in_max_chars_spinbutton.get_adjustment().set_all(30, 30, 99999, 1, - 10, 0) - self.shorten_outgoing = self.xml.get_object('shorten_outgoing') - hbox = self.xml.get_object('vbox1') - 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']) - self.in_max_chars_spinbutton.set_value(self.plugin.config['IN_MAX_CHARS']) - self.shorten_outgoing.set_active(self.plugin.config['SHORTEN_OUTGOING']) - - def avatar_size_value_changed(self, spinbutton): - self.plugin.config['MAX_CHARS'] = spinbutton.get_value() - - def on_in_max_chars_value_changed(self, spinbutton): - self.plugin.config['IN_MAX_CHARS'] = spinbutton.get_value() - - def shorten_outgoing_toggled(self, checkbutton): - self.plugin.config['SHORTEN_OUTGOING'] = checkbutton.get_active() |