# This script was made to improve upon keepnick.py

use strict;
use warnings;

no strict 'subs';

my $SCRIPT_NAME = 'keepnick';
my $SCRIPT_AUTHOR = 'The Krusty Krab <wowaname@volatile.ch>';
my $SCRIPT_VERSION = '1.0';
my $SCRIPT_LICENCE = 'Public domain';
my $SCRIPT_DESC = 'Keep your primary nickname';

our (%waiting, %connecting, %noison);

my %OPTIONS = (
	default_enable => ['Whether to enable keepnick on servers by default', '0'],
	check_time => ['Time between ISON checks (reload script to take effect)', '60'],
);


sub arrindex
{
	1 while $_[0] ne pop;
	return @_ - 1;
}

sub info
{
	weechat::print(weechat::buffer_search('irc', 'server.'.shift),
		weechat::prefix('server').shift);
}

sub my_nick
{	return lc weechat::info_get('irc_nick', shift); }

sub is_preferable
{
	my ($server, $target) = @_;
	my ($targidx, $myidx) = (
		arrindex($target, @{$waiting{$server}}),
		arrindex(my_nick($server), @{$waiting{$server}}));
	return ($myidx == -1 or $targidx < $myidx);
}

sub is_waiting
{
	my $server = lc shift;
	return exists $waiting{$server} ? @{ $waiting{$server} } : ();
}


sub disable
{	delete $waiting{lc shift}; }

sub enable
{
	my $server = shift;
	my $force = shift // 0;
	my @target = @_ ? split ',', shift : ();
	unless (@target) {
		my ($conf_nicks, $confd_nicks) = (
			weechat::config_get("irc.server.$server.nicks"),
			weechat::config_get("irc.server_default.nicks"));
		@target = split ',', weechat::config_string(
			weechat::config_option_is_null($conf_nicks)
			? $confd_nicks : $conf_nicks);
	}
	$server = lc $server;
	my ($defaultconf, $serverconf) = (
		weechat::config_get_plugin('default_enable'),
		weechat::config_get_plugin("server.$server"));
	unless ($force) {
		return () if $serverconf ne '' and !$serverconf;
		return () if $serverconf eq '' and !$defaultconf;
	}
	return () if lc $target[0] eq my_nick($server);

	$waiting{$server} = \@target;
	return @target;
}


sub do_nick
{
	my ($server, $nick) = @_;
	weechat::hook_signal_send('irc_input_send',
		weechat::WEECHAT_HOOK_SIGNAL_STRING,
		"$server;;;;/nick $nick");
	if (lc $waiting{$server}->[0] eq lc $nick) {
		# we're done if we have our primary nick choice, otherwise keep trying
		disable($server);
		info($server, "got your primary nick!");
	} else {
		info($server, "got one of your nicks; still trying for your primary nick");
	}
}

sub try_nick
{
	my ($server, $oldnick) = @_;
	($server, $oldnick) = (lc $server, lc $oldnick);
	my ($mynick, @confnick) = (my_nick($server), is_waiting($server));
	return unless @confnick;
	return unless is_preferable($server, $oldnick);

	if ($mynick eq $oldnick) { disable($server); }
	else { lc $_ eq $oldnick and do_nick($server, $_) for @confnick; }
}

sub ison_check
{
	my $iptr = weechat::infolist_get('irc_server', '', '');

	while (weechat::infolist_next($iptr)) {
		next unless weechat::infolist_integer($iptr, 'is_connected');
		my $server = lc weechat::infolist_string($iptr, 'name');
		next unless is_waiting($server);
		next unless exists $noison{$server};
		weechat::hook_hsignal_send('irc_redirect_command', {
			server => $server,
			pattern => 'ison',
			signal => $SCRIPT_NAME,
			timeout => weechat::config_get_plugin('check_time') - 1,
			});
		weechat::hook_signal_send('irc_input_send',
			weechat::WEECHAT_HOOK_SIGNAL_STRING,
			"$server;;;;/ison ".(join ' ', @{ $waiting{$server} }));
	}

	weechat::infolist_free($iptr);
	return weechat::WEECHAT_RC_OK;
}

sub irc_ison
{
	my %hashtable = %{ pop() };
	unless ($hashtable{output}) {
		$noison{lc $hashtable{server}} = 1;
		return weechat::WEECHAT_RC_ERROR;
	}
	my %nicks = map { lc $_ => 1 }
		split / +/, ($hashtable{output} =~ s/^:[^ ]* 303 [^ ]+ :?//r);

	for my $confnick (is_waiting($hashtable{server})) {
		next if exists $nicks{lc $confnick}; # still taken
		do_nick($hashtable{server}, $confnick);
		last;
	}
	return weechat::WEECHAT_RC_OK;
}

sub irc_nick
{
	my (undef, $server, $oldnick, $newnick) = (shift,
		shift =~ /(.+),irc_raw_in_nick/i,
		shift =~ /:([^! ]+)[^ ]* nick :?(.*)/i);
	return weechat::WEECHAT_RC_OK if lc $oldnick eq lc $newnick;
	try_nick($server, $oldnick);
	return weechat::WEECHAT_RC_OK;
}

sub irc_quit
{
	my (undef, $server, $nick) = (shift,
		shift =~ /(.+),irc_raw_in_quit/i,
		shift =~ /:([^! ]+)[^ ]* quit /i);
	try_nick($server, $nick);
	return weechat::WEECHAT_RC_OK;
}

sub irc_433
{
	my ($server) = $_[1] =~ /(.+),irc_raw_in2_433/i;
	$server = lc $server;
	# nick is taken when we connect? enable
	if (exists $connecting{$server}) {
		enable($server) and info($server, 'keepnick enabled');
	}
	return weechat::WEECHAT_RC_OK;
}

sub irc_connecting
{
	$connecting{lc pop} = 1;
	return weechat::WEECHAT_RC_OK;
}

sub irc_connected
{
	delete $connecting{lc pop};
	return weechat::WEECHAT_RC_OK;
}

sub irc_disconnect
{
	my $server = lc pop;
	delete $connecting{$server};
	disable($server);

	return weechat::WEECHAT_RC_OK;
}

sub cmd_keepnick
{
	my (undef, $buffer, $command) = @_;
	my $server = lc weechat::buffer_get_string($buffer, 'localvar_server');
	for ($command) {
		chomp;
		if (/^$/) {
			info($server, is_waiting($server)
				? "trying to get '".(join "', '", is_waiting($server))."'"
				: "not keeping your nick");
		}
		elsif (/^-enable$/) {
			my @waiting = enable($server, 1);
			if (@waiting) { info($server, "enabled, now waiting for '".(join "', '", @waiting)."'"); }
			else { info($server, "you already have your primary nick! doing nothing"); }
		}
		elsif (/^-disable$/) {
			disable($server);
			info($server, "no longer trying to keep your nick");
		}
		elsif (/^-autoenable$/) {
			weechat::config_get_plugin('default_enable')
				? weechat::config_unset_plugin("server.$server")
				: weechat::config_set_plugin("server.$server", '1');
			enable($server);
			info($server, "keepnick enabled on this server");
		}
		elsif (/^-autodisable$/) {
			weechat::config_get_plugin('default_enable')
				? weechat::config_set_plugin("server.$server", '0')
				: weechat::config_unset_plugin("server.$server");
			disable($server);
			info($server, "keepnick disabled on this server");
		}
		else {
			my @waiting = enable($server, 1, $command =~ s/ /,/gr);
			if (@waiting) { info($server, "now waiting for '".(join "', '", @waiting)."'"); }
			else { info($server, "you already have your primary nick! doing nothing"); }
		}
	}

	return weechat::WEECHAT_RC_OK;
}

if (weechat::register($SCRIPT_NAME, $SCRIPT_AUTHOR, $SCRIPT_VERSION,
 $SCRIPT_LICENCE, $SCRIPT_DESC, '', '')) {
	for my $option (keys %OPTIONS) {
		weechat::config_set_plugin($option, $OPTIONS{$option}[1])
			unless weechat::config_is_set_plugin($option);
		weechat::config_set_desc_plugin($option, $OPTIONS{$option}[0]);
	}

	weechat::hook_command('keepnick', $SCRIPT_DESC,
		"[nick]\n-enable|-disable\n-autoenable|-autodisable",
		"This command allows you to check the current server keepnick\n".
		"status or to manually specify nicks (separated by comma) to keep.\n".
		"You may also manually -enable or -disable keepnick on the current\n".
		"server. Both -autoenable and -autodisable choices are saved in\n".
		"config under 'plugins.var.perl.$SCRIPT_NAME.server.xxx' where 'xxx'\n".
		"is the server name.",
		'-enable|-disable|-autoenable|-autodisable %-', 'cmd_keepnick', '');
	weechat::hook_signal('*,irc_raw_in_nick', 'irc_nick', '');
	weechat::hook_signal('*,irc_raw_in_quit', 'irc_quit', '');
	weechat::hook_signal('*,irc_raw_in2_433', 'irc_433', '');
	weechat::hook_signal('irc_server_connecting', 'irc_connecting', '');
	weechat::hook_signal('irc_server_connected', 'irc_connected', '');
	weechat::hook_signal('irc_server_disconnected', 'irc_disconnect', '');
	weechat::hook_timer(weechat::config_get_plugin('check_time') * 1000, 0, 0,
		'ison_check', '');
	weechat::hook_hsignal("irc_redirection_${SCRIPT_NAME}_ison", 'irc_ison', '');

	my $iptr = weechat::infolist_get('irc_server', '', '');
	enable(weechat::infolist_string($iptr, 'name'))
		while (weechat::infolist_next($iptr));
	weechat::infolist_free($iptr);
}