benharri
/
dotfiles
mirror of https://tildegit.org/ben/dotfiles
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
dotfiles/weechat/.weechat/python/autojoinem.py

402 lines
20 KiB
Python
Raw Blame History

# -*- coding: utf-8 -*-
#
# Copyright (c) 2013-2020 by nils_2 <weechatter@arcor.de>
#
# add/del channel(s) to/from autojoin option
#
# 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/>.
#
# idea by azizLIGHTS
#
# 2020-10-20: nils_2, (freenode.#weechat)
# 0.7 : fix problem with empty fields (reported by MutantMummy: https://github.com/weechat/scripts/issues/438)
# 2017-01-06: nils_2, (freenode.#weechat)
# 0.6 : fix problem with non existing server (reported by Niols)
# 2016-12-19: nils_2, (freenode.#weechat)
# 0.5 : fix problem with empty autojoin (reported by Caelum)
# 2016-06-05: nils_2, (freenode.#weechat)
# 0.4 : make script python3 compatible
# 2015-11-14: nils_2, (freenode.#weechat)
# 0.3 : fix: problem with (undef) option
# 2014-01-19: nils_2, (freenode.#weechat)
# 0.2 : fix: adding keys to already existing keys failed
# 2013-12-22: nils_2, (freenode.#weechat)
# 0.1 : initial release
#
# requires: WeeChat version 0.3.x
#
# Development is currently hosted at
# https://github.com/weechatter/weechat-scripts
try:
import weechat,re
except Exception:
print("This script must be run under WeeChat.")
print("Get WeeChat now at: http://www.weechat.org/")
quit()
SCRIPT_NAME = "autojoinem"
SCRIPT_AUTHOR = "nils_2 <weechatter@arcor.de>"
SCRIPT_VERSION = "0.7"
SCRIPT_LICENSE = "GPL"
SCRIPT_DESC = "add/del channel(s) to/from autojoin option"
OPTIONS = { 'sorted' : ('off','channels will be sorted in autojoin-option. if autojoin-option contains channel-keys, this option will be ignored.'),
}
def add_autojoin_cmd_cb(data, buffer, args):
if args == "": # no args given. quit
return weechat.WEECHAT_RC_OK
argv = args.strip().split(' ')
# remove empty fields
argv2 = [feld for feld in argv if feld != '']
argv = argv2
# if (len(argv) <= 1):
# weechat.prnt(buffer,"%s%s: too few arguments." % (weechat.prefix('error'),SCRIPT_NAME))
# return weechat.WEECHAT_RC_OK
server = weechat.buffer_get_string(buffer, 'localvar_server') # current server
channel = weechat.buffer_get_string(buffer, 'localvar_channel') # current channel
buf_type = weechat.buffer_get_string(buffer, 'localvar_type')
# only "add <servername>" given by user
if (len(argv) == 2):
weechat.prnt(buffer,"%s%s: invalid number of arguments." % (weechat.prefix('error'),SCRIPT_NAME))
return weechat.WEECHAT_RC_OK
# '-key' keyword in command line?
if '-key' in argv:
found_key_word = argv.index('-key')
key_words = argv[int(found_key_word)+1:]
# don't use "-key" in argv
argv = argv[:int(found_key_word)]
# ADD argument
if (argv[0].lower() == 'add'):
# add current channel to autojoin. Only option "add" was given..
if (len(argv) == 1):
if server == "" or channel == "" or server == channel or buf_type == "" or buf_type != 'channel':
weechat.prnt(buffer,"%s%s: current buffer is not a channel buffer." % (weechat.prefix('error'),SCRIPT_NAME))
return weechat.WEECHAT_RC_OK
list_of_channels, list_of_current_keys = get_autojoin_list(buffer,server)
# no channels in option!
if list_of_channels == 1 and list_of_current_keys == 1:
ptr_config_autojoin = weechat.config_get('irc.server.%s.autojoin' % server)
rc = weechat.config_option_set(ptr_config_autojoin,channel,1)
return weechat.WEECHAT_RC_OK
if channel in list_of_channels:
weechat.prnt(buffer,"%s%s: channel '%s' already in autojoin for server '%s'" % (weechat.prefix("error"),SCRIPT_NAME,channel,server))
else:
# first char of channel '#' ?
if channel[0] == '#':
if '-key' in args and len(key_words) > 1:
weechat.prnt(buffer,"%s%s: too many key(s) for given channel(s) " % (weechat.prefix('error'),SCRIPT_NAME))
return weechat.WEECHAT_RC_OK
elif '-key' in args and len(key_words) == 1:
list_of_channels.insert(0,channel)
list_of_current_keys = ','.join(key_words)
# strip leading ','
if list_of_current_keys[0] == ',':
list_of_current_keys = list_of_current_keys.lstrip(',')
else:
list_of_channels.append(channel)
if not set_autojoin_list(server,list_of_channels, list_of_current_keys):
weechat.prnt(buffer,"%s%s: set new value for option failed..." % (weechat.prefix('error'),SCRIPT_NAME))
# server and channels given by user
elif (len(argv) >= 3):
server = argv[1]
list_of_channels = argv[2:]
if '-key' in args and len(list_of_channels) < len(key_words):
weechat.prnt(buffer,"%s%s: too many key(s) for given channel(s) " % (weechat.prefix('error'),SCRIPT_NAME))
return weechat.WEECHAT_RC_OK
list_of_current_channels,list_of_current_keys = get_autojoin_list(buffer,server)
# autojoin option is empty
if list_of_current_channels == 1:
# no channel -> no key!
list_of_current_keys = ""
if '-key' in args:
list_of_current_keys = ','.join(key_words)
# strip leading ','
if list_of_current_keys[0] == ',':
list_of_current_keys = list_of_current_keys.lstrip(',')
if not set_autojoin_list(server,list_of_channels, list_of_current_keys):
weechat.prnt(buffer,"%s%s: set new value for option failed..." % (weechat.prefix('error'),SCRIPT_NAME))
else:
if '-key' in args:
j = 0
new_keys = []
list_of_new_keys = []
for i in list_of_channels:
if i not in list_of_current_channels and j <= len(key_words):
# weechat.prnt(buffer,"channel: %s, channel key is: '%s'" % (i,key_words[j]))
list_of_current_channels.insert(j,i)
new_keys.insert(j,key_words[j])
j += 1
missing_channels = list_of_current_channels
list_of_new_keys = ','.join(new_keys)
if list_of_current_keys:
list_of_current_keys = list_of_new_keys + ',' + list_of_current_keys
else:
list_of_current_keys = list_of_new_keys
# strip leading ','
if list_of_current_keys[0] == ',':
list_of_current_keys = list_of_current_keys.lstrip(',')
else:
# check given channels with channels already set in option
missing_channels = get_difference(list_of_channels,list_of_current_channels)
missing_channels = list_of_current_channels + missing_channels
if not set_autojoin_list(server,missing_channels, list_of_current_keys):
weechat.prnt(buffer,"%s%s: set new value for option failed..." % (weechat.prefix('error'),SCRIPT_NAME))
return weechat.WEECHAT_RC_OK
# DEL argument
if (argv[0].lower() == 'del'):
# del current channel from autojoin. Only option "del" was given..
if (len(argv) == 1):
if server == "" or channel == "" or server == channel or buf_type == "" or buf_type != 'channel':
weechat.prnt(buffer,"%s%s: current buffer is not a channel buffer." % (weechat.prefix('error'),SCRIPT_NAME))
return weechat.WEECHAT_RC_OK
list_of_channels, list_of_keys = get_autojoin_list(buffer,server)
# no channels in option, nothing to delete
if list_of_channels == 1 and list_of_current_keys == 1:
return weechat.WEECHAT_RC_OK
if channel not in list_of_channels:
weechat.prnt(buffer,"%s%s: channel '%s' not found in autojoin for server '%s'" % (weechat.prefix("error"),SCRIPT_NAME,channel,server))
return weechat.WEECHAT_RC_OK
else:
# first char of channel '#' ?
if channel[0] == '#':
channel_key_index = list_of_channels.index(channel)
if not list_of_keys:
list_of_channels.remove(list_of_channels[channel_key_index])
list_of_current_keys = ''
else:
list_of_keys_tup = list_of_keys.split(",")
list_of_current_keys = list_of_keys
# channel does not have a key (position of channel > number of keys!)
if channel_key_index + 1 > len(list_of_keys_tup):
list_of_channels.remove(list_of_channels[channel_key_index])
# remove channel and key from autjoin option
else:
list_of_channels.remove(list_of_channels[channel_key_index])
list_of_keys_tup.remove(list_of_keys_tup[channel_key_index])
# does a key exists, after removing?
if len(list_of_keys_tup) > 0:
list_of_current_keys = ','.join(list_of_keys_tup)
# strip leading ','
if list_of_current_keys[0] == ',':
list_of_current_keys = list_of_current_keys.lstrip(',')
else: # all keys deleted
list_of_current_keys = ''
# unset option if everything is gone.
if not list_of_channels and not list_of_current_keys:
ptr_config_autojoin = weechat.config_get('irc.server.%s.autojoin' % server)
if ptr_config_autojoin:
rc = weechat.config_option_unset(ptr_config_autojoin)
return weechat.WEECHAT_RC_OK
if not set_autojoin_list(server,list_of_channels, list_of_current_keys):
weechat.prnt(buffer,"%s%s: set new value for option failed..." % (weechat.prefix('error'),SCRIPT_NAME))
# server and channels given by user
elif (len(argv) >= 3):
server = argv[1]
list_of_current_channels,list_of_current_keys = get_autojoin_list(buffer,server)
# autojoin option is empty
if list_of_current_channels == 1:
weechat.prnt(buffer,"%s%s: nothing to delete..." % (weechat.prefix('error'),SCRIPT_NAME))
return weechat.WEECHAT_RC_OK
else:
list_of_channels = args.split(" ")[2:]
if list_of_current_keys:
list_of_current_keys_tup = list_of_current_keys.split(",")
else:
list_of_current_keys_tup = ''
for i in list_of_channels:
# check if given channel is in list of options
if not i in list_of_current_channels:
continue
channel_key_index = list_of_current_channels.index(i)
# channel does not have a key (position of channel > number of keys!)
if channel_key_index + 1 > len(list_of_current_keys_tup):
list_of_current_channels.remove(i)
# if len(list_of_current_channels) <= 0:
# list_of_current_channels = ''
else: # remove channel and key from autjoin option
list_of_current_channels.remove(i)
list_of_current_keys_tup.remove(list_of_current_keys_tup[channel_key_index])
# does an key exists, after removing?
if len(list_of_current_keys_tup) > 0:
list_of_current_keys = ','.join(list_of_current_keys_tup)
# strip leading ','
if list_of_current_keys[0] == ',':
list_of_current_keys = list_of_current_keys.lstrip(',')
else: # all keys deleted
list_of_current_keys = ''
# for j in list_of_current_channels:
# weechat.prnt(buffer,"chan:%s" % j)
# for j in list_of_current_keys_tup:
# weechat.prnt(buffer,"key :%s" % j)
# unset option if everything is gone.
if not list_of_current_channels and not list_of_current_keys:
ptr_config_autojoin = weechat.config_get('irc.server.%s.autojoin' % server)
if ptr_config_autojoin:
rc = weechat.config_option_unset(ptr_config_autojoin)
return weechat.WEECHAT_RC_OK
if not set_autojoin_list(server,list_of_current_channels, list_of_current_keys):
weechat.prnt(buffer,"%s%s: set new value for option failed..." % (weechat.prefix('error'),SCRIPT_NAME))
return weechat.WEECHAT_RC_OK
def get_difference(list1, list2):
return list(set(list1).difference(set(list2)))
# returns a list of channels and a list of keys
# 1 = something failed, 0 = channel found
def get_autojoin_list(buffer,server):
ptr_config_autojoin = weechat.config_get('irc.server.%s.autojoin' % server)
# option not found! server does not exist
if not ptr_config_autojoin:
weechat.prnt("","%s%s: server '%s' does not exist." % (weechat.prefix('error'),SCRIPT_NAME,server))
return 1,1
# get value from autojoin option
channels = weechat.config_string(ptr_config_autojoin)
if not channels:
return 1,1
# check for keys
if len(re.findall(r" ", channels)) == 0:
list_of_channels = channels.split(",")
list_of_keys = []
elif len(re.findall(r" ", channels)) == 1:
list_of_channels2,list_of_keys = channels.split(" ")
list_of_channels = list_of_channels2.split(",")
else:
weechat.prnt("","%s%s: irc.server.%s.autojoin not valid..." % (weechat.prefix('error'),SCRIPT_NAME,server))
return 1,1
return list_of_channels, list_of_keys
def set_autojoin_list(server,list_of_channels, list_of_keys):
ptr_config_autojoin = weechat.config_get('irc.server.%s.autojoin' % server)
if not ptr_config_autojoin:
return 0
if OPTIONS['sorted'].lower() == 'on' and not list_of_keys:
# no keys, sort the channel-list
channels = '%s' % ','.join(sorted(list_of_channels))
else:
# don't sort channel-list with given key
channels = '%s' % ','.join(list_of_channels)
# strip leading ','
if channels[0] == ',':
channels = channels.lstrip(',')
# add keys to list of channels
if list_of_keys:
channels = '%s %s' % (channels,list_of_keys)
rc = weechat.config_option_set(ptr_config_autojoin,channels,1)
if not rc:
return 0
return 1
def autojoinem_completion_cb(data, completion_item, buffer, completion):
# server = weechat.buffer_get_string(buffer, 'localvar_server') # current buffer
input_line = weechat.buffer_get_string(buffer, 'input')
# get information out of the input_line
argv = input_line.strip().split(" ",3)
if (len(argv) >= 3 and argv[1] == 'del'):
server = argv[2]
list_of_channels,list_of_keys = get_autojoin_list(buffer,server)
if list_of_channels == 1:
return weechat.WEECHAT_RC_OK
if (len(argv) >= 4 and argv[1] == 'del'):
list_of_current_channels = argv[3].split(' ')
missing_channels = get_difference(list_of_channels,list_of_current_channels)
if not missing_channels:
return weechat.WEECHAT_RC_OK
list_of_channels = missing_channels
for i, elem in enumerate(list_of_channels):
weechat.hook_completion_list_add(completion, list_of_channels[i], 0, weechat.WEECHAT_LIST_POS_END)
return weechat.WEECHAT_RC_OK
# ================================[ weechat options & description ]===============================
def init_options():
for option,value in OPTIONS.items():
weechat.config_set_desc_plugin(option, '%s (default: "%s")' % (value[1], value[0]))
if not weechat.config_is_set_plugin(option):
weechat.config_set_plugin(option, value[0])
OPTIONS[option] = value[0]
else:
OPTIONS[option] = weechat.config_get_plugin(option)
def toggle_refresh(pointer, name, value):
global OPTIONS
option = name[len('plugins.var.python.' + SCRIPT_NAME + '.'):] # get optionname
OPTIONS[option] = value # save new value
return weechat.WEECHAT_RC_OK
# ================================[ main ]===============================
if __name__ == "__main__":
if weechat.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE, SCRIPT_DESC, '', ''):
version = weechat.info_get("version_number", "") or 0
weechat.hook_command(SCRIPT_NAME,SCRIPT_DESC,
'add <server> [<channel1>[ <channel2>...]] | [-key <channelkey> [<channelkey>...]] ||'
'del <server> [<channel1>[ <channel2>...]]',
'add <server> <channel>: add channel to irc.server.<servername>.autojoin\n'
' -key <channelkey>: name of channelkey\n'
'del <server> <channel>: del channel from irc.server.<servername>.autojoin\n'
'\n'
'Examples:\n'
' add current channel to corresponding server option:\n'
' /' + SCRIPT_NAME + ' add\n'
' add all channels from all server to corresponding server option:\n'
' /allchan /' + SCRIPT_NAME + ' add\n'
' add channel #weechat to autojoin option on server freenode:\n'
' /' + SCRIPT_NAME + ' add freenode #weechat\n'
' add channel #weechat and #weechat-de to autojoin option on server freenode, with channel key for channel #weechat:\n'
' /' + SCRIPT_NAME + ' add freenode #weechat #weechat-de -key my_channel_key\n'
' del channels #weechat and #weechat-de from autojoin option on server freenode:\n'
' /' + SCRIPT_NAME + ' del freenode #weechat #weechat-de',
'add %(irc_servers) %(irc_server_channels)|%*||'
'del %(irc_servers) %(plugin_autojoinem)|%*',
'add_autojoin_cmd_cb', '')
init_options()
weechat.hook_completion('plugin_autojoinem', 'autojoin_completion', 'autojoinem_completion_cb', '')
weechat.hook_config('plugins.var.python.' + SCRIPT_NAME + '.*', 'toggle_refresh', '')
# if int(version) >= 0x00030600:
# else:
# weechat.prnt("","%s%s %s" % (weechat.prefix("error"),SCRIPT_NAME,": needs version 0.3.6 or higher"))
# weechat.command("","/wait 1ms /python unload %s" % SCRIPT_NAME)
Reference in New Issue View Git Blame Copy Permalink