#!/usr/bin/perl
use warnings;
use strict;
use utf8;
use AnyEvent;
use AnyEvent::XMPP::Client;
use AnyEvent::XMPP::Ext::Disco;
use AnyEvent::XMPP::Ext::Version;
use AnyEvent::XMPP::Namespaces qw/xmpp_ns/;
use AnyEvent::XMPP::Ext::MUC;
use AnyEvent::XMPP::Util qw/node_jid res_jid/;
use Data::Dump qw(dump);
use Encode;
use Carp qw(confess);

use lib 'lib';
use MQR::Redis;

binmode STDOUT, ":utf8";

my $jid = $ENV{XMPP_JID} || die "XMPP_JID";
my $pw  = $ENV{XMPP_PASSWD} || die "XMPP_PASSWD";
my $room = $ENV{XMPP_ROOM} || warn "no XMPP_ROOM - not joining any muc rooms!\n";
my $subscribe  = $ENV{XMPP_SUBSCRIBE} || warn "no XMPP_SUBSCRIBE - pushing ALL MSG\n";

my $j       = AnyEvent->condvar;
my $cl      = AnyEvent::XMPP::Client->new (debug => 1);
my $disco   = AnyEvent::XMPP::Ext::Disco->new;
my $version = AnyEvent::XMPP::Ext::Version->new;
my $muc     = AnyEvent::XMPP::Ext::MUC->new (disco => $disco);

$cl->add_extension ($disco);
$cl->add_extension ($version);
$cl->add_extension ($muc);

$cl->set_presence(undef, 'I\'m a talking bot.', 1);

$cl->add_account ($jid, $pw);
warn "connecting to $jid...\n";

warn "# jid:$jid root:$room subscribe:$subscribe\n";

our $contacts;
our $muc_rooms;

sub publish;

MQR::Redis->redis->psubscribe( "MSG $subscribe", sub {
	my ( $body, $channel ) = @_;
	warn "<<<< ", dump( $channel, $body );
	Encode::_utf8_on($body);

	my ( undef, $gw, $room, $user, $from ) = split(/\s/,$channel);

	return if $from eq $jid; # FIXME skip own

	foreach my $to ( keys %$contacts, keys %$muc_rooms ) {
		next if $from =~ m/\Q$to\E/; # FIXME
		my $type = defined $muc_rooms->{$from} ? 'groupchat' : 'chat';
		publish "SEND $jid $type $user $to" => $body;
	}
}) if $subscribe;

MQR::Redis->redis->psubscribe( "SEND $jid *", sub {
	my ( $body, $channel ) = @_;
	warn "<<<< ",dump( $channel, $body );
	Encode::_utf8_on($body);

	my ( undef, $gw, $type, $user, $to ) = split(/\s/, $channel);
	warn "SEND $jid $type $to <$user> $body\n";
	$cl->send_message( "<$user> $body", $to => $jid, $type );
});

#redis->psubscribe( '*' => sub { warn @_, $/ };

sub publish {
	my ( $channel, $body ) = @_;
	Encode::_utf8_off($body);
	MQR::Redis->publish( $channel, $body );
}

$cl->reg_cb (
   session_ready => sub {
      my ($cl, $acc) = @_;
      warn "connected!\n";
      $muc->join_room ($acc->connection, $room, node_jid ($acc->jid));
      $muc->reg_cb (
         message => sub {
            my ($cl, $room, $msg, $is_echo) = @_;
		$muc_rooms->{ $room->nick_jid }++;
warn "# MUC message ",dump( $room->nick_jid, $msg->any_body, $is_echo );

            return if $is_echo;
            return if $msg->is_delayed;

			my $from = $msg->from;
			my $user = $from;
			$user =~ s{^.+/}{};
			publish "MSG $jid groupchat $user $from" => $msg->any_body;

            my $mynick = res_jid ($room->nick_jid);
            if ($msg->any_body =~ /^\s*\Q$mynick\E:\s+(.*?)\s*$/) {
               my $ans = answer_to ($1);
               my $repl = $msg->make_reply;
               $repl->add_body ($ans);
               $repl->send;
            }
         }
      );
   },
   message => sub {
		my ($cl, $acc, $msg) = @_;

		my $from = $msg->from;
		my $body = $msg->any_body;

#		$contacts->{ $from }++; # don't push to anyone who sent message

		my $user = $from;
		$user =~ s{\@.+$}{};

		publish "MSG $jid chat $user $from" => $body;

#		my $repl = $msg->make_reply;
#		$repl->add_body( $response );
#		$repl->send;
#		warn "#>>> $response\n";

   },
   contact_request_subscribe => sub {
      my ($cl, $acc, $roster, $contact) = @_;
      $contact->send_subscribed;
      warn "Subscribed to ".$contact->jid."\n";
   },
   error => sub {
      my ($cl, $acc, $error) = @_;
      warn "Error encountered: ".$error->string."\n";
      $j->broadcast;
   },
   disconnect => sub {
      warn "Got disconnected: [@_]\n";
      $j->broadcast;
   },
	roster_update => sub {
		my ($con,$account,$roster) = @_;
warn "XXXXX", ref($account), " | ", ref($roster);
		foreach my $contact ( $roster->get_contacts ) {
			$contacts->{ $contact->{jid} }++;
			warn "# contacts ",dump($contacts);
		}
		warn "# contacts ",dump($contacts);
	},
);

$cl->start;

$j->wait;
