cyla

#!/usr/bin/perl -w
#
# Configurable variables
#
#       src             "socket" when using scgi_socket, or anything else when using scgi_port
#       socket          rTorrent's rpc socket (scgi_local)      - using scgi_local      - needed, when "src" is set to "socket"
#       ip              rTorrent's ip address                   - using scgi_port       - needed, when "src" is NOT set to "socket"
#       port            rTorrent's scgi port (scgi_port)        - using scgi_port       - needed, when "src" is NOT set to "socket"
#       maxdls          maximum number of paralell downloads    -                       - required
#

use IO::Socket;

my $src         = "";
my $ip          = "127.0.0.1";
my $port        = "4912";
my $socket      = "";
my $maxdls      = 1;

my $pattern     = qr/<value><string>([A-Z0-9]+)<\/string><\/value>/;

my $line        = "<?xml version=\"1.0\" encoding=\"utf-8\"?><methodCall><methodName>d.multicall</methodName><params><param><value><string>incomplete</string></value></param><param><value><string>d.get_hash=</string></value></param></para
ms></methodCall>";
my $llen        = length $line;
my $header      = "CONTENT_LENGTH\000${llen}\000SCGI\001\000";
my $hlen        = length $header;
$line           = "${hlen}:${header},${line}";

if ( ( defined $src ) && ( $src eq "socket" ) ) {
        socket( SOCK, PF_UNIX, SOCK_STREAM, 0 ) or die;
        connect( SOCK, sockaddr_un( $socket ) ) or die $!;
} else {
        socket( SOCK, PF_INET, SOCK_STREAM, getprotobyname( "tcp" ) );
        connect( SOCK, sockaddr_in( $port, inet_aton( $ip ) ) );
}

print SOCK $line;
flush SOCK;

my @torrents = ();
while ( $line = <SOCK> ) {
        if ( $line =~ /$pattern/ ) {
                push( @torrents, $1 );
        }
}
close (SOCK);

my $count = 0;
while ( ( ( $count < $maxdls ) ) && ( scalar( @torrents ) > 0 ) ) {
        $torrent = $torrents[0];

        $line   = "<?xml version=\"1.0\" encoding=\"utf-8\"?><methodCall><methodName>f.multicall</methodName><params><param><value><string>$torrent</string></value></param><param><value><string>0</string></value></param><param><value><str
ing>f.set_priority=1</string></value></param></params></methodCall>";
        $llen   = length $line;
        $header = "CONTENT_LENGTH\000${llen}\000SCGI\001\000";
        $hlen   = length $header;
        $line   = "${hlen}:${header},${line}";

        if ( ( defined $src ) && ( $src eq "socket" ) ) {
                socket( SOCK, PF_UNIX, SOCK_STREAM, 0 ) or die;
                connect( SOCK, sockaddr_un( $socket ) ) or die $!;
        } else {
                socket( SOCK, PF_INET, SOCK_STREAM, getprotobyname( "tcp" ) );
                connect( SOCK, sockaddr_in( $port, inet_aton( $ip ) ) );
        }
        print SOCK $line;
        flush SOCK;

        $line   = "<?xml version=\"1.0\" encoding=\"utf-8\"?><methodCall><methodName>d.update_priorities</methodName><params><param><value><string>$torrent</string></value></param></params></methodCall>";
        $llen   = length $line;
        $header = "CONTENT_LENGTH\000${llen}\000SCGI\001\000";
        $hlen   = length $header;
        $line   = "${hlen}:${header},${line}";

        if ( ( defined $src ) && ( $src eq "socket" ) ) {
                socket( SOCK, PF_UNIX, SOCK_STREAM, 0 ) or die;
                connect( SOCK, sockaddr_un( $socket ) ) or die $!;
        } else {
                socket( SOCK, PF_INET, SOCK_STREAM, getprotobyname( "tcp" ) );
                connect( SOCK, sockaddr_in( $port, inet_aton( $ip ) ) );
        }
        print SOCK $line;
        flush SOCK;

        shift( @torrents );
        $count++;
}

foreach $torrent ( @torrents ) {
        $line   = "<?xml version=\"1.0\" encoding=\"utf-8\"?><methodCall><methodName>f.multicall</methodName><params><param><value><string>$torrent</string></value></param><param><value><string>0</string></value></param><param><value><str
ing>f.set_priority=0</string></value></param></params></methodCall>";
        $llen   = length $line;
        $header = "CONTENT_LENGTH\000${llen}\000SCGI\001\000";
        $hlen   = length $header;
        $line   = "${hlen}:${header},${line}";

        if ( ( defined $src ) && ( $src eq "socket" ) ) {
                socket( SOCK, PF_UNIX, SOCK_STREAM, 0 ) or die;
                connect( SOCK, sockaddr_un( $socket ) ) or die $!;
        } else {
                socket( SOCK, PF_INET, SOCK_STREAM, getprotobyname( "tcp" ) );
                connect( SOCK, sockaddr_in( $port, inet_aton( $ip ) ) );
        }
        print SOCK $line;
        flush SOCK;

        $line   = "<?xml version=\"1.0\" encoding=\"utf-8\"?><methodCall><methodName>d.update_priorities</methodName><params><param><value><string>$torrent</string></value></param></params></methodCall>";
        $llen   = length $line;
        $header = "CONTENT_LENGTH\000${llen}\000SCGI\001\000";
        $hlen   = length $header;
        $line   = "${hlen}:${header},${line}";

        if ( ( defined $src ) && ( $src eq "socket" ) ) {
                socket( SOCK, PF_UNIX, SOCK_STREAM, 0 ) or die;
                connect( SOCK, sockaddr_un( $socket ) ) or die $!;
        } else {
                socket( SOCK, PF_INET, SOCK_STREAM, getprotobyname( "tcp" ) );
                connect( SOCK, sockaddr_in( $port, inet_aton( $ip ) ) );
        }
        print SOCK $line;
        flush SOCK;
}

exit;