#
# Copyright (c) 2010 by Nils Görs <weechatter@arcor.de>
#
# just a simple buddylist using the nicklist from channels you are in.
# it also works if nicklist is hidden.
#
# 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/>.
#
# 0.9: servername without nicks will not be displayed in buddylist
#	servername will be displayed in different colour or will be hidden if not connected (option "color.server.offline")
#	buddylist bar will be hidden if you are not connected to a server (option "hide.bar")
# 0.8: - major changes - (buddylist file changed!!!)
#	buddylist uses now server / nick structure
#       change entries in buddylist to : servername,nickname (e.g.: freenode,nils_2)
#	or add your buddies again with the add function.
# 0.7: nick change will be recognize
# 0.6: nick wasn't set to offline, if buddy leave channel
# 0.5: server information will be used now instead of nicklist
#	reduction of cpu load (reported by ArcAngel and tigrmesh)
#	bar will be removed if you unload the script. (requested by bazerka)
#	help page will be displayed when you call buddylist without arguments
# 0.4: added option "sort"
# 0.3: remove spaces for indenting when bar position is top/bottom
#     : hook_config when settings changed.
# 0.2: work-around for crash when searching nick in buffer without nicklist (function nicklist_search_nick) removed 
# 0.1: initial release
#
# TODO: waiting for redirection ;-)

use strict;

my $prgname		= "buddylist";
my $version		= "0.9";
my $description		= "Simple buddylist that shows the status of your buddies.";

my $buffer		= "";
my $default_buddylist	= "buddylist.txt";
my %buddies		= ();					# to store the buddylist with status for each nick
my %buddylist_level = (0 => "online", 1 => "away", 2 => "offline");
my %default_color_buddylist = ("online" => "yellow",
                             "away"    => "cyan",
                             "offline"    => "blue");
my $color_default	= "lightcyan";
my $position		= "top";
my $hide_bar		= "on";					# hide buddylist bar when all servers are offline

my %nick_structure	= ();					# to store servername, nickname and status
#$VAR1 = {
#	'freenode' =>	{
#				'nils_2' => 'online',
#				'nick2' => 'offline',
#				'nick3' => 'online'
#			},                       
#	'fu-berlin' =>	{
#				'nils_2' => 'away'    
#			}
#	};

# first function called by a WeeChat-script.
weechat::register($prgname, "Nils Görs <weechatter\@arcor.de>", $version,
                  "GPL3", $description, "shutdown", "");

init();
buddylist_read();

weechat::bar_item_new($prgname, "build_buddylist", "");
weechat::bar_new($prgname, "0", "0", "root", "", "left", "horizontal",
					    "vertical", "0", "0", "default", "default", "default", "1",
					    $prgname);
weechat::command("", "/bar show " . $prgname);

weechat::hook_signal("*,irc_in2_352", "from_hook_who","");			# RFC command with use,channel,status etc..
weechat::hook_signal("*,irc_in_part", "remove_nick", "");
weechat::hook_signal("*,irc_in_quit", "remove_nick", "");
weechat::hook_signal("*,irc_in_nick", "nick_changed", "");
weechat::hook_config("*.$prgname.*", "config_signal", "");			# buddylist settings changed?
weechat::hook_signal("*,irc_in_part", "remove_nick", "");
weechat::hook_signal("irc_server_connected", "server_connected", "");
weechat::hook_signal("irc_server_disconnected", "server_disconnected", "");

weechat::hook_command($prgname, $description,

	"<add>[nick_1 [... nick_n]] | <del>[nick_1 [... nick_n]]", 

	"<add> [nick(s)] add nick(s) to the buddylist\n".
	"<del> [nick(s)] delete nick(s) from the buddylist\n".
	"\n".
	"Options:\n".
	"'plugins.var.perl.buddylist.buddylist'           : path/file-name to store your buddies.\n".
	"'plugins.var.perl.buddylist.color.away'          : colour for away buddies.\n".
	"'plugins.var.perl.buddylist.color.offline'       : colour for offline buddies.\n".
	"'plugins.var.perl.buddylist.color.online'        : colour for online buddies.\n".
	"'plugins.var.perl.buddylist.color.server'        : colour for servername.\n".
	"'plugins.var.perl.buddylist.color.server.offline : colour for disconnected server (default: hide).\n".
	"'plugins.var.perl.buddylist.hide.bar             : hide buddylist-bar when all servers with buddies are offline (default: on).\n".
	"'plugins.var.perl.buddylist.sort'                : sort method for buddylist.\n".
	"                                         default : $prgname will be sort by nickname\n".
	"                                         status  : $prgname will be sort by status (online, away, offline)\n\n".
	"If $prgname don't refresh buddylist, check the following WeeChat options:\n".
	"'irc.network.away_check'          : interval between two checks, in minutes. (has to be >= 1 (default:0)).\n".
	"'irc.network.away_check_max_nicks': channels with high number of nicks will not be checked (default: 25).\n".
	"\n".
	"buddyname has to be written 'case sensitive' (it's recommended to use nick-completion).\n".
	"\n".
	"Examples:\n".
	"Add buddy to buddylist:\n".
	"/$prgname add buddyname\n".
	"Delete buddy from buddylist:\n".
	"/$prgname del buddyname\n",
	"add|del", "settings", "");
server_check();

return weechat::WEECHAT_RC_OK;


sub server_connected{
  server_check();
  weechat::bar_item_update($prgname);
  return weechat::WEECHAT_RC_OK;
}

sub server_disconnected{
  server_check();
  weechat::bar_item_update($prgname);
  return weechat::WEECHAT_RC_OK;
}

sub config_signal{								# also update when server dis/connected
  server_check();
  weechat::bar_item_update($prgname);
  return weechat::WEECHAT_RC_OK;
}
my $str = "";
sub build_buddylist{
	$str = "";
	if ($hide_bar eq "on"){
	  return $str;								# and print it.
	}

# get bar position (left/right/top/bottom) and sort (default/status)
	my $option = weechat::config_get("weechat.bar.$prgname.position");
	if ($option ne ""){
		$position = weechat::config_string($option);
	}
	my $sort = weechat::config_get_plugin("sort");

	if ($sort eq "status"){							# use sort option "status"
	  foreach my $s ( sort keys %nick_structure ) {				# sortiert die Server alphabetisch
	    if (keys (%{$nick_structure{$s}}) eq "0"){				# check out if nicks exists for server
	      next;								# no nick for server. jump to next server
	    }

	    my $color = weechat::config_get_plugin("color.default");
	    my $visual = " ";							# placeholder after servername
	    my $cr = "\n";
	    $visual  = $cr if (($position eq "left") || ($position eq "right"));
	    $color = "default" if ($color eq "");
	    $color = weechat::config_get_plugin("color.server");

### get status from server
	      my $infolist_server = weechat::infolist_get("irc_server","",$s);	# get pointer for server %s
	      weechat::infolist_next($infolist_server);
	      my $is_connected = weechat::infolist_integer($infolist_server,"is_connected");	# get status of connection for server (1 = connected | 0 = disconnected)
	      weechat::infolist_free($infolist_server);				# don't forget to free infolist ;-)
	      if ($is_connected == 0){
		$color = weechat::config_get_plugin("color.server.offline");
		if ($color eq "hide"){						# hide offline servers?
		  next;								# yes!
		}
	      }


	    $str .= weechat::color($color) . $s . ":" . $visual;		# add servername ($s ;) to buddylist
# sorted by value first and nick case insensitiv as second
	  foreach my $n (sort { $nick_structure{$s}{$a} cmp $nick_structure{$s}{$b}} (sort {uc($a) cmp uc($b)} (sort keys(%{$nick_structure{$s}})))){
			createoutput($s,$n);
		}}
	  } else {								# use "default" for sort by name only
	  foreach my $s ( sort keys %nick_structure ) {				# sortiert die Server alphabetisch
	    my $color = weechat::config_get_plugin("color.default");
	    my $visual = " ";							# placeholder after servername
	    my $cr = "\n";
	    $visual  = $cr if (($position eq "left") || ($position eq "right"));
	    $color = "default" if ($color eq "");
	    $color = weechat::config_get_plugin("color.server");
	    $str .= weechat::color($color) . $s . ":" . $visual;		# add servername ($s ;) to buddylist

	      foreach my $n (sort {uc($a) cmp uc($b)} (sort keys(%{$nick_structure{$s}} ))){ # sort by name case insensitiv
			createoutput($s,$n);
	      }
		}
	  }
    if ($str eq ""){
	$str = "buddylist is empty for connected server or you are not connected to a server.";
    }
	  return $str;
}
sub createoutput{
	my ($server,$nick) = ($_[0],$_[1]);
	my $cr = "\n";
	my $color = weechat::config_get_plugin("color.default");
	$color = "default" if ($color eq "");
	my $status = $nick_structure{$server}{$nick};				# get buddy status
		$color = weechat::config_get_plugin("color.".$buddylist_level{$status});
### visual settings for left, right or top and bottom
	my $visual = " ";							# placeholder
	my $mover = "";								# move it to right
	if (($position eq "left") || ($position eq "right")){
	  $visual  = $cr;
	  $mover  = "  ";
	}
	$str .= weechat::color($color) . $mover . "$nick" . $visual;
 }

# buddy changed his nick
sub nick_changed{
    my ($blank, $servername, $args) = @_;
	my ($server) = split(/,/, $servername);					# get name from server
	$args =~ /\:(.*)\!(.*)\:(.*)/;
	my $old_nickname = $1;
	my $new_nickname = $3;
	if (exists $nick_structure{$server}{$old_nickname}){			# nick in buddylist?
	  my $status = $nick_structure{$server}{$old_nickname};			# get old buddy status
	    $nick_structure{$server}{$new_nickname} = $status;			# add changed buddyname with old status
	    delete $nick_structure{$server}{$old_nickname};			# delete old buddyname
	  weechat::bar_item_update($prgname);
	}
}
# buddy leaves channel
sub remove_nick{
#($nick,$name,$ip,$action,$channel) = ($args =~ /\:(.*)\!n=(.*)@(.*?)\s(.*)\s(.*)/; # maybe for future use
    my ( $data, $servername, $args ) = @_;
	my ($server) = split(/,/, $servername);					# get name from server
	my ($nickname) = ($args =~ /\:(.*)\!/);
	if (exists $nick_structure{$server}{$nickname}){			# nick in buddylist?
		$nick_structure{$server}{$nickname} = 2;			# yes. but he went offline
		weechat::bar_item_update($prgname);
	}
}
# get information from who command
sub from_hook_who{
    my ( $data, $servername, $args ) = @_;

	my @words= split(" ",$args);						# [7] = nick
	($servername) = split(/,/, $servername);				# get name from server
	my $nickname = $words[7];
	if (exists $nick_structure{$servername}{$nickname}){			# nick in buddylist?
		my $status = 0;
		$status = 1 if (substr($words[8],0,1) eq "G");# buddy is away
		add_to_nicktable($servername, $nickname, $status);
		weechat::bar_item_update($prgname);
	}
}
# add buddy to my structure
sub add_to_nicktable{
    my ($servername, $nickname, $status) = @_;
	$nick_structure{$servername}{$nickname} = $status;			# create structure
}

sub settings{
my ($getargs) = ($_[2]);
	my $servername = current_buffer_test();
	if ($servername eq "0") {
	  weechat::print("",weechat::prefix("error")."$prgname: You can't add nor del buddies in core buffer.");
	return weechat::WEECHAT_RC_OK;
	}

	my ( $cmd, $args ) = ( $getargs =~ /(.*?)\s+(.*)/ );			# get parameters and cut cmd from nicks
		$cmd = $getargs unless $cmd;

	if (defined $args) {							# buddy choosed?
		foreach ( split( / +/, $args ) ) {
			if ($cmd eq "add"){
				$nick_structure{$servername}{$_} = 2;
				buddylist_save();
			}
			if ($cmd eq "del" and exists $nick_structure{$servername}{$_}){
				delete $nick_structure{$servername}{$_};
# delete servername from structure, if last nick from server was deleted
				delete $nick_structure{$servername} if (keys (%{$nick_structure{$servername}}) == 0);
				buddylist_save();
			}
		}
	}else{
		weechat::command("", "/help $prgname");				# no arguments given. Print help
	}
	weechat::bar_item_update($prgname);
	return weechat::WEECHAT_RC_OK;
}
# check for buffer. add/del function can only be used in channels, not in core buffer
sub current_buffer_test{
  my $buffer_name = weechat::buffer_get_string(weechat::current_buffer(),"name");	# get current buffer name
  if ($buffer_name =~ /\./){							# format?
      my ($servername, $channelname) = split (/\./,$buffer_name);		# split
      if ($servername eq "server"){						# user in server buffer?
	return $channelname;							# yes
	}
      return $servername;							# user in channel buffer!
      }
return 0;									# in core buffer
}

# check server status
sub server_check{
  if (weechat::config_get_plugin("hide.bar") eq "off"){				# get user settings
  $hide_bar = "off";								# don't hide bar
  weechat::command("", "/bar show " . $prgname);
  return;
  }
 foreach my $s ( sort keys %nick_structure ) {					# sortiert die Server alphabetisch
    my $infolist_server = weechat::infolist_get("irc_server","",$s);		# get pointer for server %s
    weechat::infolist_next($infolist_server);
    my $is_connected = weechat::infolist_integer($infolist_server,"is_connected");	# get status of connection for server (1 = connected | 0 = disconnected)
    weechat::infolist_free($infolist_server);					# don't forget to free infolist ;-)
      if ($is_connected == 1){							# one server is at least online!
	$hide_bar = "off";							# show bar
	weechat::command("", "/bar show " . $prgname);
	last;
      } else {
	$hide_bar = "on";							# hide bar
	weechat::command("", "/bar hide " . $prgname);
      }
  }
}

### read the buddylist
sub buddylist_read {
	my $buddylist = weechat::config_get_plugin("buddylist");
	return unless -e $buddylist;
	open (WL, "<", $buddylist) || DEBUG("$buddylist: $!");
	while (<WL>) {
		chomp;								# kill LF
        my ( $servername, $nickname ) = split /,/;				# servername,nickname
	  if (not defined $nickname){
	    close WL;
	  weechat::print("",weechat::prefix("error")."$prgname: $buddylist is not valid.");
	    return;
	  }
		$nick_structure{$servername}{$nickname} = 2  if length $_;	# offline
	}
	close WL;
}
sub buddylist_save {
	my $buddylist = weechat::config_get_plugin( "buddylist" );
	open (WL, ">", $buddylist) || DEBUG("write buddylist: $!");
	  foreach my $s ( sort keys %nick_structure ) {				# sortiert die Server alphabetisch
	      foreach my $n ( sort keys %{$nick_structure{$s}} ) {		# sortiert die Nicks alphabetisch
		print WL "$s,$n\n";
	      }
	  }
	close WL;
}
# init the settings
sub init{
	if ( weechat::config_get_plugin("buddylist") eq '' ) {
		my $wd = weechat::info_get( "weechat_dir", "" );
		$wd =~ s/\/$//;
		weechat::config_set_plugin("buddylist", $wd . "/" . $default_buddylist );
	}
	if (weechat::config_get_plugin("color.default") eq "")
	{
		weechat::config_set_plugin("color.default", "default");
	}
	if (weechat::config_get_plugin("color.server") eq "")
	{
		weechat::config_set_plugin("color.server", "default");			# default = "real white"
	}
	if (weechat::config_get_plugin("color.server.offline") eq "")
	{
		weechat::config_set_plugin("color.server.offline", "hide");		# default = "hide offline server"
	}
	if (weechat::config_get_plugin("hide.bar") eq "")
	{
		weechat::config_set_plugin("hide.bar", "on");
	}
	if (weechat::config_get_plugin("sort") eq "")
	{
		weechat::config_set_plugin("sort", "default");
	}
# get color settings.
	foreach my $level (values %buddylist_level){
		if (weechat::config_get_plugin("color.".$level) eq ""){
			weechat::config_set_plugin("color.".$level,
					$default_color_buddylist{$level});
		}
	}
}

# hide bar when buddylist was closed
sub shutdown{
	weechat::command("", "/bar hide " . $prgname);
}
sub DEBUG {weechat::print('', "***\t" . $_[0]);}