X-Git-Url: https://ruin.nu/git/?p=NDIRC.git;a=blobdiff_plain;f=Bot.pm;fp=Bot.pm;h=363531217864018f1faf26ab54ec00a37cbeae9d;hp=5693dd4d0b353c392d51a173736421262ba3f74d;hb=468a4e3e07c0d9e6ed66a75c73f0d7ff2921b90a;hpb=cd18307dc7e3886266937935feffbb41b1df422d

diff --git a/Bot.pm b/Bot.pm
index 5693dd4..3635312 100644
--- a/Bot.pm
+++ b/Bot.pm
@@ -26,14 +26,18 @@ use Moose;
 
 use POE::Component::IRC::Common qw/irc_to_utf8/;
 use POE::Session;
+use POE::Kernel;
 use POE::Component::IRC::Plugin::Logger;
 use POE::Component::IRC::Plugin::BotTraffic;
 use POE::Component::IRC::Plugin::Connector;
 use POE::Component::IRC::Plugin::AutoJoin;
 use POE::Component::IRC::Plugin::NickReclaim;
 
+use Mojo::Discord;
+use Mojo::IOLoop;
+
 use NDIRC::Dispatcher;
-use NDIRC::Context;
+use NDIRC::IrcContext;
 
 use IO::File;
 
@@ -43,6 +47,33 @@ has disp => (
 	lazy_build => 1
 );
 
+has discord => (
+	is => 'rw',
+	isa => 'Object'
+);
+
+has discord_name => (
+	is => 'rw',
+	isa => 'Str'
+);
+
+has discord_id => (
+	is => 'rw',
+	isa => 'Str'
+);
+
+has discord_channels => (
+	is => 'rw',
+	isa => 'HashRef',
+	default => sub { {} }
+);
+
+has targets => (
+	is => 'ro',
+	isa => 'HashRef[ArrayRef[Str]]',
+	default => sub{ {} },
+);
+
 # We registered for all events, this will produce some debug info.
 sub _default {
 	my ($event, $args) = @_[ARG0 .. $#_];
@@ -60,11 +91,16 @@ sub _default {
 	return 0;
 }
 
+my $irc;
+sub irc {
+	return $irc;
+}
+
 sub _start {
-	my ($kernel,$heap,$session) = @_[KERNEL,HEAP,SESSION];
+	my ($self,$kernel,$heap,$session) = @_[OBJECT,KERNEL,HEAP,SESSION];
 
 	# retrieve our component's object from the heap where we stashed it
-	my $irc = $heap->{irc};
+	$irc = $heap->{irc};
 	$kernel->sig( DIE => 'sig_DIE' );
 	$kernel->sig( USR1 => 'sig_usr1' );
 	$kernel->sig( USR2 => 'sig_usr2' );
@@ -94,6 +130,29 @@ sub _start {
 	$irc->yield( connect => { server => 'irc.netgamers.org' } );
 
 	$kernel->delay( refresh => 60 );
+
+	if (my $f =  new IO::File 'discord'){
+		my $user = <$f>;
+		chomp $user;
+		my $token = <$f>;
+		chomp $token;
+
+		$self->discord(Mojo::Discord->new(
+				'token'     => $token,
+				'name'      => $user,
+				'url'       => 'https://nd.ruin.nu',
+				'version'   => '1.0',
+				'callbacks' => {
+					'READY'          => sub { $self->discord_ready(@_) },
+					'MESSAGE_CREATE' => sub { $self->discord_message_create(@_) },
+					'GUILD_CREATE' => sub { $self->discord_guild_create(@_) },
+					'CHANNEL_CREATE' => sub { $self->discord_channel_create(@_) },
+				},
+				'reconnect' => 1,
+				'verbose'   => 1,
+			));
+		$self->discord->init();
+	}
 	return;
 }
 
@@ -188,13 +247,16 @@ sub _build_disp {
 		$disp->load(@commands);
 	}
 
+	%{$self->targets} = ();
 	my $channels = new IO::File 'channels' or die $!;;
 	while (<$channels>){
 		my ($chan, @types) = split /\s+/;
 		say "$chan - @types";
 		if ($chan =~ /^(.*):(.*)$/){
 			$chan = $1;
-			$disp->set_target($2,$chan);
+			$self->targets->{$2} = [] unless exists $self->targets->{$2};
+			push @{$self->targets->{$2}},$chan;
+			say "$2 - @{$self->targets->{$2}}";
 		}
 		$disp->add_channel($chan,\@types);
 	}
@@ -276,6 +338,58 @@ sub  refresh {
 sub irc_join {
 }
 
+sub discord_ready {
+	my $self = shift;
+	my $hash = shift;
+	$self->discord_id($hash->{user}{id});
+	$self->discord_name($hash->{user}{username});
+
+	say localtime(time) . " - Connected to Discord. $self->{discord_id}";
+}
+
+sub discord_message_create {
+}
+
+sub discord_guild_create {
+	my $self = shift;
+	my $hash = shift;
+
+	for my $chan (@{$hash->{channels}}) {
+		say localtime(time) . " - $chan->{id} - $chan->{name}";
+		$self->discord_channels->{$chan->{id}} = $chan;
+	}
+}
+
+sub discord_channel_create {
+	my $self = shift;
+	my $chan = shift;
+
+	for my $key (keys %{$chan}) {
+		say localtime(time) . " - $key - $chan->{$key}";
+	}
+	$self->discord_channels->{$chan->{id}} = $chan;
+}
+
+sub handleCommand {
+	my ($self, $c, $msg) = @_;
+
+	my ($p,$command,$args) = ($msg =~ /^([.])(\S+)(?: (.+))?/);
+
+	if ($msg =~ m{https?://[\w.]+/.+?scan(_id|_grp)?=(\w+)}){
+		if (!$command || $command =~ m{^https?://}){
+			($p,$command,$args) = ('.','addscan',$msg);
+		}elsif($command ne 'addscan'){
+			$self->handleCommand ($c, ".addscan $msg")
+		}
+	}
+
+	return 0 unless $self->disp->has_command($command,$c->channel);
+
+	say localtime(time) . " - $msg";
+
+	return $self->disp->run_command($c,$command,$args);
+}
+
 sub parseCommand {
 	my ($self, $msg, $server, $nick, $address, $channel, $model) = @_;
 
@@ -308,17 +422,58 @@ sub parseCommand {
 	}
 
 	$address =~ s/.*@(.*)/$1/;
-	my $c = NDIRC::Context->new({
+	my $c = NDIRC::IrcContext->new({
 			host => $address,
 			nick => $nick,
 			channel => $channel,
 			disp => $self->disp,
 			model => $model,
 			server => $server,
+			bot => $self,
 			reply_string => $reply_string,
 		});
 
 	return $self->disp->run_command($c,$command,$args);
 }
 
+sub toTarget {
+	my ($self, $target, $msg) = @_;
+
+	return unless exists $self->targets->{$target};
+
+	$self->message($msg, @{$self->targets->{$target}});
+
+}
+
+sub message {
+	my ($self, $msg, @targets) = @_;
+
+	for (@targets) {
+		when (/^D-(\d+)$/) {
+			$self->discordMessage($1, $msg);
+		}
+		default {
+			$self->ircMessage(privmsg => $_, $msg);
+		}
+	}
+}
+
+sub ircMessage {
+	my ($self, $command, $target, $msg) = @_;
+
+	$msg =~ s`<b>(.*?)</b>`${\(chr(2))}$1${\(chr(15))}`gi;
+	$msg =~ s`<c(\d+)>(.*?)</c>`${\(chr(3))}$1$2${\(chr(15))}`gi;
+
+	$self->irc->yield($command, $target, $msg);
+}
+
+sub discordMessage {
+	my ($self, $target, $msg) = @_;
+
+	$msg =~ s`<b>(.*?)</b>`**$1**`gi;
+	$msg =~ s`<c(\d+)>(.*?)</c>`*$2*`gi;
+
+	$self->discord->send_message($target, $msg );
+}
+
 1;