Add Url Shortener Plugin

1 files changed, 177 insertions, 0 deletions

diff --git a/url_shortener/url_shortener.py b/url_shortener/url_shortener.py
new file mode 100644
index 0000000..d97bec7
--- /dev/null
+++ b/url_shortener/url_shortener.py
@@ -0,0 +1,177 @@
+# -*- 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()