configuration/weechat/python/shortenurl.py
2014-12-19 00:19:32 +01:00

146 lines
4.5 KiB
Python

# Copyright (c) 2010, 2011, 2012, 2013 by John Anderson <sontek@gmail.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
# History
# 2014-5-3, John Anderson <sontek@gmail.com>
# version 0.5.3: Fixed short_own bug introduced in 0.5, notify the short url
# instead of appending to the message (returning to behavior
# from 0.4)
# 2013-12-25, John Anderson <sontek@gmail.com>
# version 0.5: Added support for latest weechat (0.4+)
# 2011-10-24, Dmitry Geurkov <dmitry_627@mail.ru>
# version 0.4.1: added: option "ignore_list" for a blacklist of shorten urls.
# 2011-01-17, nils_2 <weechatter@arcor.de>
# version 0.4: URI will be shorten in /query, too.
# : added: option "short_own".
# 2010-11-08, John Anderson <sontek@gmail.com>:
# version 0.3: Get python 2.x binary for hook_process (fixes problem
# when python 3.x is default python version, requires
# WeeChat >= 0.3.4)
import re
import weechat
from urllib import urlencode
from urllib2 import urlopen
SCRIPT_NAME = "shortenurl"
SCRIPT_AUTHOR = "John Anderson <sontek@gmail.com>"
SCRIPT_VERSION = "0.5.3"
SCRIPT_LICENSE = "GPL3"
SCRIPT_DESC = "Shorten long incoming and outgoing URLs"
ISGD = 'http://is.gd/api.php?%s'
TINYURL = 'http://tinyurl.com/api-create.php?%s'
# script options
# shortener options:
# - isgd
# - tinyurl
settings = {
"color": "red",
"urllength": "30",
"shortener": "isgd",
"short_own": "off",
"ignore_list": "http://is.gd,http://tinyurl.com",
}
octet = r'(?:2(?:[0-4]\d|5[0-5])|1\d\d|\d{1,2})'
ipAddr = r'%s(?:\.%s){3}' % (octet, octet)
# Base domain regex off RFC 1034 and 1738
label = r'[0-9a-z][-0-9a-z]*[0-9a-z]?'
domain = r'%s(?:\.%s)*\.[a-z][-0-9a-z]*[a-z]?' % (label, label)
urlRe = re.compile(
r'(\w+://(?:%s|%s)(?::\d+)?(?:/[^\])>\s]*)?)' % (domain, ipAddr),
re.I
)
if weechat.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE,
SCRIPT_DESC, "", ""):
for option, default_value in settings.iteritems():
if weechat.config_get_plugin(option) == "":
weechat.config_set_plugin(option, default_value)
weechat.hook_print('', 'irc_privmsg', '', 1, 'notify', '')
weechat.hook_modifier('irc_out_privmsg', 'outgoing_hook', '')
def notify(data, buf, date, tags, displayed, hilight, prefix, msg):
color = weechat.color(weechat.config_get_plugin('color'))
reset = weechat.color('reset')
my_nick = weechat.buffer_get_string(buf, 'localvar_nick')
if prefix != my_nick:
urls = find_and_process_urls(msg)
for url, short_url in urls:
weechat.prnt(buf, '%(color)s[ %(url)s ]%(reset)s' % dict(
color=color,
url=short_url,
reset=reset))
return weechat.WEECHAT_RC_OK
def outgoing_hook(data, modifier, modifier_data, msg):
short_own = weechat.config_get_plugin('short_own')
if short_own == 'off':
return msg
urls = find_and_process_urls(msg)
for url, short_url in urls:
msg = msg.replace(url, '%(short_url)s [ %(url)s ]' % dict(
url=url,
short_url=short_url))
return msg
def find_and_process_urls(new_message):
urls = []
for url in urlRe.findall(new_message):
max_url_length = int(weechat.config_get_plugin('urllength'))
if len(url) > max_url_length and not should_ignore_url(url):
short_url = get_shortened_url(url)
urls.append((url, short_url))
return urls
def get_shortened_url(url):
shortener = weechat.config_get_plugin('shortener')
if shortener == 'isgd':
url = ISGD % urlencode({'longurl': url})
if shortener == 'tinyurl':
url = TINYURL % urlencode({'url': url})
try:
return urlopen(url).read()
except:
return url
def should_ignore_url(url):
ignorelist = weechat.config_get_plugin('ignore_list').split(',')
for ignore in ignorelist:
if len(ignore) > 0 and ignore in url:
return True
return False