#!/usr/local/bin/perl -w
# Program to ask battle.net if a user is online
# Written by John D. Lewis <garlon@yahoo.com>
# Version: 1.0
#
# bnet-online.pl username [server] [username password]
#
# Tested on WinNT machine with ActiveState Perl 5.005_02

$|++;

use IO::Socket;

## ----------------------------------------------------------
## -- Edit these values to customize how this program runs --
## ----------------------------------------------------------
my $SLEEP_SECONDS = 8;
my $VERBOSE = 0; # Set to "1" if you want to see all of Battle.net's output
my $MAX_RECONNECT_TRIES = 5; # Don't try to reconnect more than this many times
my $DEFAULT_USERNAME_TO_QUERY = "someone"; # If not specified on the command-line, use this ID
my $LOGIN_USERNAME = "guest"; # better netizen logging in as guest
# But you may specify a registered user (and password)
#my $LOGIN_USERNAME = "reguser";
#my $LOGIN_PASSWORD = "somethingclever";

## ---------------------------------------------------
## ------- DON'T EDIT ANYTHING BELOW THIS LINE -------
## ---------------------------------------------------
my $DEFAULT_SERVER = "exodus.battle.net";
my $DEFAULT_PORT = 6112;
my $MAX_LOOP_COUNT = 20;
my $POST_LOGON_HEADER_LINE_COUNT = 5;
my $LOGON_HEADER_LINE_COUNT = 4;

# Person's name must be supplied on command line, or
# default will be used
my $who = shift || $DEFAULT_USERNAME_TO_QUERY;

my $server = shift || $DEFAULT_SERVER;
my $port = $DEFAULT_PORT;

my $socket = &login($server, $port);

my $x;

for ($x = 0; $x <= $LOGON_HEADER_LINE_COUNT; $x++) {
    $from_server = <$socket>;
    print "$from_server" if ($VERBOSE);
};

print "Logged on...\n" unless ($VERBOSE);

print $socket "\n";
print $socket "$LOGIN_USERNAME\n";

$from_server = <$socket>;
#print "$from_server\n";
if ($LOGIN_USERNAME !~ /guest/i) {
    print $socket "$LOGIN_USERNAME\n";
    $from_server = <$socket>; # Password: prompt
    $from_server = <$socket>; # empty line
    print $socket "\n";
} else {
    print $socket "\n";
    $from_server = <$socket>; # empty line
    $from_server = <$socket>; # empty line
};

&readUntilInfoDone();

my $join_command = "/join temp_channelXY123";
my $reply = 0;
print $socket "$join_command\n";
while ($from_server = <$socket>) {
    last if ($reply);
    $reply++ if ($from_server =~ /^$join_command/);
};
print "$from_server\n" if ($VERBOSE);

my $command = "/whois $who";
print "$command: reply = ";
print $socket "$command\n\n";
do {
    # ignore all the other junk
    $from_socket = <$socket>; # repeats command
} until ($from_socket =~ /^$command/);
$from_socket = <$socket>; # real response from server
print "$from_socket";

close($socket);

1;

sub readUntilInfoDone($?) {
    my $INFO_LINE_COUNT = shift || 7;
    my $line_count = 0;
    while (my $from_server = <$socket>) {
        next if ($from_server =~ /^100[123]/);
        my ($code_num, $code_txt, $text) = split /\s/, $from_server, 3;
        chomp($text);
        $text =~ s/^\s*\"(.+)\"\s*$/$1/;
        print "[$code_txt] $text\n" if ($VERBOSE);
        last if ++$line_count >= $INFO_LINE_COUNT;
    }
};

sub login($$$?) {
    my $remote_host = shift;
    my $remote_port = shift;

    my $conn_attempt = shift || 1;

    return() if ($conn_attempt >= $MAX_RECONNECT_TRIES);

    my $args = "c";
    my $from_server;

    $socket = IO::Socket::INET->new(PeerAddr => $remote_host,
                                    PeerPort => $remote_port,
                                    Proto    => "tcp",
                                    Type     => SOCK_STREAM);

    print $socket "$args\n";
    $from_server = <$socket>;

    if ($from_server =~ /refused/) {
        print STDERR "Connection refused (try #" . $conn_attempt . ")! Trying again in $SLEEP_SECONDS seconds...\n";
        sleep($SLEEP_SECONDS);
        undef $from_server;
        undef $socket;
        return(&login($remote_host, $remote_port, ++$conn_attempt));
    };
    return($socket);
};