weechat-opal/scripts/embellish.py

# -*- coding: utf-8 -*-
#
# Released into the Public Domain

"""embellish: make your chats noticable"""

# v1.3 - wowaname
#  changed hook so script doesn't modify the input buffer
# v1.2 - wowaname
#  whitelist_buffers added
#  message won't be decorated if blank
#  changed default ignore_buffers
# v1.1 - wowaname
#  ignore_buffers has regex support
#  additional profile support
#  colours restored on ^C or ^O in message
# v1.0 - wowaname
#  Initial release

import random
import re
import weechat

SCRIPT_NAME    = "embellish"
SCRIPT_AUTHOR  = "The Krusty Krab <wowaname@volatile.ch>"
SCRIPT_VERSION = "1.3"
SCRIPT_LICENSE = "Public domain"
SCRIPT_DESC    = "Makes your chats noticable"

# script options
settings = {
	"pre": (
		"\x0313♥",
		"Text to add to the beginning of messages.",
		),
	"suf": (
		"\x0313♥",
		"Text to add to the end of messages.",
		),
	"fgs": (
		"04,05,06,13",
		"Colour codes to cycle for the foreground. "
		"Leave blank for no foreground colours.",
		),
	"bgs": (
		"",
		"Colour codes to cycle for the background. "
		"Leave blank for no background colours.",
		),
	"ignore_buffers": (
		"bitlbee.*,scripts",
		"List of buffers to ignore. Glob matches unless "
		"you prefix the name with 're:'.",
		),
	"whitelist_buffers": (
		"",
		"List of buffers to whitelist. Glob match unless "
		"you prefix the name with 're:'. Useful with "
		"ignore_buffers = \"*\"",
		),
	"whitelist_cmds": (
		"me,amsg,say",
		"Commands to embellish.",
		),
	"profiles": (
		"> greentext,! alert",
		"List of prefix/profile pairs. If you type one of "
		"these prefixes at the beginning of your message, "
		"the options will switch to (profile)_pre, "
		"(profile)_suf, (profile)_fgs, and (profile)_bgs. ",
		),
	"greentext_pre": ">",
	"greentext_suf": "",
	"greentext_fgs": "03",
	"greentext_bgs": "",
}



if weechat.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE,
 SCRIPT_DESC, "", ""):
	for opt, val in settings.iteritems():
		setting, desc = val if type(val) == tuple else (val, "")
		if desc: weechat.config_set_desc_plugin(opt, desc)
		if weechat.config_is_set_plugin(opt): continue
		weechat.config_set_plugin(opt, setting)

	weechat.hook_modifier("input_text_for_buffer", "cb_embellish", "")

def glob_match (haystack, needle):
	return re.search("^%s$" %
	 re.escape(haystack).replace(r"\?", ".").replace(r"\*", ".*?"),
	 needle)

def cb_embellish(data, mod, buf, input):
	buffer_name = weechat.buffer_get_string(buf, "name").lower()
	output = ""
	profile = ""

	for pattern in weechat.config_get_plugin("whitelist_buffers").lower().split(","):
		if (pattern.startswith("re:") and
		 re.search(pattern[3:], buffer_name)) or glob_match(pattern, buffer_name):
			break
	else:
		for pattern in weechat.config_get_plugin("ignore_buffers").lower().split(","):
			if (pattern.startswith("re:") and
			 re.search(pattern[3:], buffer_name)) or glob_match(pattern, buffer_name):
				return input

	if not input:
		return input

	if (input.startswith("/") and not input.startswith("/ ") and input != "/"
	 and input.split(" ", 1)[0].split("\n", 1)[0].find("/", 1) < 0):
		for cmd in weechat.config_get_plugin("whitelist_cmds").lower().split(","):
			if not input.startswith("/%s " % cmd): continue
			output = "/%s " % cmd
			input = input.split(" ", 1)[1] if " " in input else ""
			break
		else:
			return input

	if input.startswith("//"): input = input[1:]

	for profile_pairs in weechat.config_get_plugin("profiles").split(","):
		prefix, name = profile_pairs.split()
		if not input.startswith("%s " % prefix): continue
		profile = "%s_" % name
		input = input.split(" ",1)[1] if " " in input else ""
		for opt in ("pre", "suf", "fgs", "bgs"):
			if weechat.config_is_set_plugin(profile + opt): continue
			weechat.config_set_plugin(profile + opt, "")
		break

	fgs = weechat.config_get_plugin("%sfgs" % profile).split(",")
	bgs = weechat.config_get_plugin("%sbgs" % profile).split(",")
	base = "\x0f\x03%s%s%s" % (
		random.choice(fgs),
		"," if bgs != [""] else "",
		random.choice(bgs),
		) if fgs != [""] or bgs != [""] else ""
	if base:
		input = re.sub(
			"\x03([^0-9])",
			"\x03%s\\1" % base,
			input.replace("\x0f","\x0f%s" % base))

	#l = (409 - len(base) * 3 -
	# weechat.info_get("irc_server_isupport_value", "*,NICKLEN")
	l = 400 - len(base) * 3
	o = []
	for line in input.replace("\r", "").split("\n"):
		for i in xrange(0, len(line), l):
			o.append("%s%s%s %s%s %s" % (
				base,
				weechat.config_get_plugin("%spre" % profile),
				base,
				line[i:i+l].rstrip(),
				base,
				weechat.config_get_plugin("%ssuf" % profile),
				))

	return output + "\n".join(o)