File Coverage

blib/lib/Net/uTorrent.pm

Criterion	Covered	Total	%
statement	27	234	11.5
branch	0	16	0.0
condition			n/a
subroutine	9	31	29.0
pod	20	22	90.9
total	56	303	18.4

package Net::uTorrent;

use URI;
use URI::QueryParam;
use HTTP::Request::Common;
use LWP::UserAgent;
use JSON::XS;
use HTML::TreeBuilder;

use 5.010000;
use strict;
use warnings;

require Exporter;

our @ISA = qw(Exporter);

our %EXPORT_TAGS = ( 'all' => [ qw() ] );

our @EXPORT_OK = ( @{ $EXPORT_TAGS{'all'} } );

our @EXPORT = qw();

our $VERSION = '0.02';


 our $token;
 our $api_url;
 our $http_request = HTTP::Request->new;
 our $ua = LWP::UserAgent->new;
 our $html_tree = HTML::TreeBuilder->new;
 our $json = JSON::XS->new;
 our $cid; #cache id
 our $error;



sub new {
		my ($class,%args) = @_;
		my $self = bless({}, $class);

		my $host = $args{'hostname'};
		my $port = $args{'port'};
		my  $user = $args{'user'};
		my $pass = $args{'pass'};

		$ua->credentials (
					"$host:$port",
					'uTorrent',
					$user => $pass
				 );
		$api_url = "http://$host:$port/gui/";
		my $http = $ua->get($api_url."token.html");
		my $parsed = $html_tree->parse($http->decoded_content); 		# Build HTML tree of elements
		$token = $parsed->{'_body'}->{'_content'}[0]->{'_content'}[0];	# Get authentication token

		return $self;
	  }


sub token {
		return $token;
	  }

sub login_success {

		 if ($token) { return 1; }
		 else { return 0; }

		  }

sub torrents {
		my ($class,%args) = @_;

		my %params = (
			        list	=>	1,
			     );
		%params = (%params,%args); #Add optional cache id from argument.
		my @args = qw();
		push @args,\%params;
		my $json_text = api_query_result(@args);
		my $decoded_json = $json->decode($json_text);
		$cid = $decoded_json->{torrentc};

		my @torrent = qw();

		for (@{$decoded_json->{torrents}}) {
						     my @keys = qw (
						    		     infohash status name size percent downloaded
						   		     uploaded ratio upstream downstream eta label
						   		     peers_connected peers_in_swarm seeds_connected
						   		     seeds_in_swarm availability queue_order remaining
						   		   );
						    my @values = @$_;
						    my %torrent_details;
						    for (my $i = 0; $i <= $#keys; $i++) { $torrent_details{$keys[$i]} = $values[$i]; }
						    push @torrent,\%torrent_details;

						 }
		return \@torrent;
	     }

sub cache_id {
		return $cid;
}

sub get_settings {
		my ($class,$type) = @_;
		$type = 'array' unless $type;
		my @settings;
		my %settings;
		my @type = qw (integer boolean string);

		my @args = qw();
		push @args,{action => 'getsettings'};
		my $json_text = api_query_result(@args);
		my $decoded_json = $json->decode($json_text);

		for (@{$decoded_json->{settings}})  {
						      my %settings_info;
						      my @keys = qw (name type value);
						      my @values = @$_;
						      for (my $i = 0; $i <= $#keys; $i++) {  $settings_info{$keys[$i]} = $values[$i]; }
						      $settings_info{'type_human'} = $type[$settings_info{'type'}];
						      $settings{$settings_info{'name'}} = \%settings_info;
						      if ($type eq 'hash') { delete $settings_info{'name'}; }
						      push @settings,\%settings_info;
						    }
		if ($type eq 'array') { return \@settings; }
		elsif ($type eq 'hash') { return \%settings; }
}

sub set_settings {
		   my ($class,%settings) = @_;
		   my @args;
		   push @args, {action => 'setsetting'};
		   for my $key (keys %settings) {
		   push @args, { 's' => $key };
		   push @args, { 'v' => $settings{$key} };
		   }
		   api_query_result(@args);
		   return;
		 }
sub filelist {
		my ($class,@infohash) = @_;
		my @args;
		my @priority_human = qw(skip low normal high);
		my @keys = qw(filename size downloaded priority);

		push @args,{action => 'getfiles'};
		for my $hash (@infohash) { push @args,{hash => $hash}; }
		my $json_text = api_query_result(@args);
		my $i = 0; for ($json_text =~ /,"(files)":/g) { my $append = $i++; $json_text =~ s/,"(files)":/,"files$append":/; } # Ugly hack to avoid repeating keys due to bug in uTorrent's JSON generator
		my $decoded_json = $json->decode($json_text);
		my %torrent_hash;
		for my $key (keys %$decoded_json) {
						    if ($key ne 'build') {
						    			   my $torrent = $decoded_json->{$key};
						    			   my $infohash = $$torrent[0];
						    			   my $file_array = $$torrent[1];
						    			   my @files_in_torrent = qw();
						    			   for my $file (@$file_array) {
						    			   					my %fileinfo;
						    			   					for (my $i = 0; $i <= $#$file; $i++) { $fileinfo{$keys[$i]} = $$file[$i]; }
						    			   					$fileinfo{'priority_human'} = $priority_human[$fileinfo{'priority'}];
								    			   			push @files_in_torrent,\%fileinfo;

							    			   		        }
						    			   $torrent_hash{$infohash} = \@files_in_torrent;
						    			 }

						  }
		return \%torrent_hash;
}

sub get_properties {

  		my ($class,@infohash) = @_;
		my @args;
		push @args,{action => 'getprops'};
		for my $hash (@infohash) { push @args,{hash => $hash}; }
		my $json_text = api_query_result(@args);
		my $i = 0; for ($json_text =~ /,"(props)":/g) { my $append = $i++; $json_text =~ s/,"(props)":/,"props$append":/; } # Ugly hack to avoid repeating keys due to bug in uTorrent's JSON generator
		my $decoded_json = $json->decode($json_text);
		my %properties;
		for my $key (keys %$decoded_json) {
						    if ($key ne 'build') {
									    my $property = $decoded_json->{$key};
									    my @trackers = split(/\s+/,$$property[0]->{trackers});
									    $$property[0]->{trackers} = \@trackers;
									    $properties{$$property[0]->{hash}} = $$property[0];
									 }
						  }
		return \%properties;
}

sub set_properties {
		my ($class,@properties) = @_;
		my @args;
		push @args,{action => 'setprops'};
		for my $property (@properties) {
						 push @args, { hash => $$property{'hash'}};
						 delete $$property{'hash'};
						 for my $key (keys %$property) {
						 				 push (@args, { 's' => $key } );
						 				 push (@args, { 'v' => $$property{$key} } );
						 			       }
					       }
		api_query_result(@args);
		return;



}

sub start {
             my ($class,@infohash) = @_;
             my @args;
             push @args, {action => 'start'};
             for (@infohash) { push @args, { hash => $_ }; }
             api_query_result(@args);
}


sub stop {
             my ($class,@infohash) = @_;
             my @args;
             push @args, {action => 'stop'};
             for (@infohash) { push @args, { hash => $_ }; }
             api_query_result(@args);

}

sub pause {
             my ($class,@infohash) = @_;
             my @args;
             push @args, {action => 'pause'};
             for (@infohash) { push @args, { hash => $_ }; }
             api_query_result(@args);

}

sub resume {
             my ($class,@infohash) = @_;
             my @args;
             push @args, {action => 'unpause'};
             for (@infohash) { push @args, { hash => $_ }; }
             api_query_result(@args);

}

sub force_start {
             my ($class,@infohash) = @_;
             my @args;
             push @args, {action => 'forcestart'};
             for (@infohash) { push @args, { hash => $_ }; }
             api_query_result(@args);


}

sub recheck {
             my ($class,@infohash) = @_;
             my @args;
             push @args, {action => 'recheck'};
             for (@infohash) { push @args, { hash => $_ }; }
             api_query_result(@args);

}

sub remove_torrent {
             my ($class,@infohash) = @_;
             my @args;
             push @args, {action => 'remove'};
             for (@infohash) { push @args, { hash => $_ }; }
             api_query_result(@args);


}


sub remove_data {
             my ($class,@infohash) = @_;
             my @args;
             push @args, {action => 'removedata'};
             for (@infohash) { push @args, { hash => $_ }; }
             api_query_result(@args);


}


sub set_priority {
		my ($class,%args_hash) = @_;
		my @args;
		push @args, { action => 'setprio' };
		push @args, { hash   => $args_hash{'hash'} };
		push @args, { p => $args_hash{'priority'} };
		delete $args_hash{'hash'};
		for (@{$args_hash{'files'}}) { push @args, { f => $_}; }
		api_query_result(@args);
		return;
}


sub add_url {
             my ($class,$url) = @_;
             my @args;
             push @args, { action => 'add-url' };
             push @args, { 's' => $url };
             api_query_result(@args);
             return;
}


sub add_file {
	     my ($class,$file) = @_;
             my $api = URI->new($api_url);
             $api->query_param(action => 'add-file');
             $api->query_param(t => $token);

             # This is the only method that uses HTTP POST, so a separate subroutine is not necessary.

             my $request = $ua->request (
             					POST $api,
             					Content_Type	=>	'multipart/form-data',
             					Content		=>	[
          								  'torrent_file' => [$file]
             								]

             				);

             my $json_text = $request->content;
             my $decoded_json = $json->decode($json_text);
             my $status = $$decoded_json{error};
             $status = 'Success' unless $status;
             return { Status => $status};

}

############################### Module's own sub routines ###############################

sub api_query_result {
			my (@params) = @_;
			my $api = URI->new($api_url);
			for my $array_value (@params) {
							for my $key (keys %$array_value) {
												$api->query_param_append($key => $$array_value{$key});
											 }
						       }

			$api->query_param(token => $token);
			my $http = $ua->get($api);
			return $http->decoded_content;
		      }



1;

__END__