gearman/transmission.pl

   1 #!/usr/bin/perl
   2 use warnings;
   3 use strict;
   4
   5 use Transmission::Client;
   6 use JSON::XS;
   7 use Data::Dump qw(dump);
   8 use autodie;
   9 use File::Path qw(make_path remove_tree);
  10
  11 sub user_info {
  12         my $login = shift;
  13         my ( undef, undef, $uid, $gid, undef, undef, $email, $dir, $shell ) = getpwnam $login;
  14         return ( $uid, $gid, $dir );
  15 }
  16
  17 use Gearman::Worker;
  18 my $worker = Gearman::Worker->new;
  19 $worker->job_servers('127.0.0.1:4730');
  20
  21 my $tc = Transmission::Client->new( autodie => 1 );
  22
  23 my $j = JSON::XS->new;
  24 $j->allow_blessed(1);
  25 $j->convert_blessed(0);
  26
  27 use TokyoCabinet;
  28
  29 my $name = $ENV{ZSLICE} || die "need ZSLICE";
  30 my $torrent_dir = "/$name/torrent";
  31 $name =~ s/\W+/_/g;
  32
  33 $worker->register_function( $name . '_torrent_list' => sub {
  34         my ($job) = @_;
  35         my @cols = split(/\s+/,$job->arg);
  36         @cols = qw( id name seeders leechers ) unless @cols;
  37         my @list;
  38         for my $torrent ($tc->read_torrents) {
  39                 warn "## torrent = ",dump($torrent);
  40                 my $hash;
  41                 $hash->{$_} = $torrent->$_ foreach @cols;
  42                 push @list, $hash;
  43         }
  44         warn "# torrent_list = ",dump( \@list );
  45         return $j->encode( \@list );
  46 } );
  47
  48 $worker->register_function( $name . '_torrent_download' => sub {
  49         my $work = $_[0]->arg;
  50         chomp $work;
  51         warn "# torrent_download [$work]\n";
  52
  53         my $user = $1 if $work =~ s{~(\w+)/}{};
  54         my $path = $work;
  55
  56         my $dir = (getpwnam($user))[7] || die "no user $user";
  57
  58         my $torrent_file = "$dir/$path";
  59
  60         $path =~ s/\.torrent$//;
  61         my $to_dir = "$torrent_dir/download/$user/$path";
  62         my ($uid,$gid) = (stat($torrent_dir))[4,5];
  63         make_path $to_dir, { uid => $uid, gid => $gid };
  64
  65         my $t = $tc->add( filename => $torrent_file, 'download-dir' => $to_dir );
  66         warn "# added $to_dir = ",dump($t);
  67
  68         return $to_dir;
  69 } );
  70
  71 $worker->register_function( $name . '_torrent_download_done' => sub {
  72         my $work = $_[0]->arg;
  73         chomp $work;
  74         warn "# torrent_download_done [$work]\n";
  75
  76         my ( $id, $dir, $name, $hash ) = split(/\s+#\s+/,$work);
  77
  78         my $t = $tc->remove( $id );
  79         warn "# remove $id = ",dump($t);
  80
  81         if ( $dir =~ m{/download/(\w+)/(.+)$} ) {
  82                 my $user = $1;
  83                 my $path = $2;
  84
  85                 warn "# check $name md5sum";
  86                 my $md5sum = `md5sum "$dir/$name"`;
  87                 $md5sum =~ s/\s+.*$//s;
  88                 if ( $name ne $md5sum ) {
  89                         warn "ERROR: invalid md5sum filename ",dump( $name, $md5sum );
  90                 }
  91
  92                 warn "# move $name to $user $path\n";
  93                 my ( $uid, $gid, $home_dir ) = user_info $user;
  94                 link "$dir/$name", "$home_dir/$path";
  95                 chown $uid, $gid,  "$home_dir/$path";
  96                 warn "FIXME: add to md5 pool?";
  97
  98         } else {
  99                 warn "ERROR: can't find user in $dir for $name\n";
 100         }
 101
 102         warn "# $id removed $dir $name $hash\n";
 103
 104         remove_tree $dir;
 105
 106 } );
 107
 108 $worker->register_function( $name . '_torrent_share' => sub {
 109         my $work = $_[0]->arg;
 110         chomp $work;
 111         warn "# torrent_share [$work]\n";
 112
 113         my $user = $1 if $work =~ s{~(\w+)/}{};
 114         my $path = $work;
 115
 116         my $dir = (getpwnam($user))[7] || die "no user $user";
 117         my $md5_path = "$dir/.md5";
 118         my %h;
 119         tie %h, "TokyoCabinet::HDB", $md5_path || die "$md5_path: $!";
 120         my $md5 = $h{$path} || die "can't find $path in $md5_path";
 121         untie %h;
 122
 123         warn "# $md5 $path";
 124
 125         if ( ! -e "$torrent_dir/download/$md5" ) {
 126                 symlink "/rsync1/s1/md5/$md5", "$torrent_dir/download/$md5"; # FIXME slice path
 127                 warn "# created $torrent_dir/download/$md5";
 128         }
 129
 130         if ( ! -e "$torrent_dir/md5/$md5.torrent" ) {
 131                 my $tracker = "http://213.186.16.27:6502/announce"; # FIXME
 132                 system "mktorrent --announce $tracker --no-date --verbose --output $torrent_dir/md5/$md5.torrent $torrent_dir/download/$md5";
 133
 134         }
 135
 136         symlink "$torrent_dir/md5/$md5.torrent", "$dir/$path.torrent" unless -e "$dir/$path.torrent";
 137
 138         $tc->add( filename => "$torrent_dir/md5/$md5.torrent" );
 139
 140         return "$dir/$path";
 141 } );
 142
 143 warn "$0 $name pid $$ waitng for jobs\n";
 144 $worker->work while 1;
 145