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
  10 sub home_dir {
  11         my $login = shift;
  12         my ( undef, undef, $uid, $gid, undef, undef, $email, $dir, $shell ) = getpwnam $login;
  13         return $dir;
  14 }
  15
  16 use Gearman::Worker;
  17 my $worker = Gearman::Worker->new;
  18 $worker->job_servers('127.0.0.1:4730');
  19
  20 my $tc = Transmission::Client->new( autodie => 1 );
  21
  22 my $j = JSON::XS->new;
  23 $j->allow_blessed(1);
  24 $j->convert_blessed(0);
  25
  26 use TokyoCabinet;
  27
  28 my $name = $ENV{ZSLICE} || die "need ZSLICE";
  29 my $torrent_dir = "/$name/torrent";
  30 $name =~ s/\W+/_/g;
  31
  32 $worker->register_function( $name . '_torrent_list' => sub {
  33         my ($job) = @_;
  34         my @cols = split(/\s+/,$job->arg);
  35         @cols = qw( id name seeders leechers ) unless @cols;
  36         my @list;
  37         for my $torrent ($tc->read_torrents) {
  38                 warn "## torrent = ",dump($torrent);
  39                 my $hash;
  40                 $hash->{$_} = $torrent->$_ foreach @cols;
  41                 push @list, $hash;
  42         }
  43         warn "# torrent_list = ",dump( \@list );
  44         return $j->encode( \@list );
  45 } );
  46
  47 $worker->register_function( $name . '_torrent_download' => sub {
  48         my $work = $_[0]->arg;
  49         chomp $work;
  50         warn "# torrent_download [$work]\n";
  51
  52         my $user = $1 if $work =~ s{~(\w+)/}{};
  53         my $path = $work;
  54
  55         my $dir = (getpwnam($user))[7] || die "no user $user";
  56
  57         my $to_dir = "$torrent_dir/download/$user";
  58         if ( ! -e $to_dir ) {
  59                 mkdir $to_dir;
  60                 # fix perm
  61                 my ($uid,$gid) = (stat($torrent_dir))[4,5];
  62                 chown $uid, $gid, $to_dir;
  63         }
  64
  65         $tc->add( filename => "$dir/$path", 'download-dir' => $to_dir );
  66
  67         return $to_dir;
  68 } );
  69
  70 $worker->register_function( $name . '_torrent_share' => sub {
  71         my $work = $_[0]->arg;
  72         chomp $work;
  73         warn "# torrent_share [$work]\n";
  74
  75         my $user = $1 if $work =~ s{~(\w+)/}{};
  76         my $path = $work;
  77
  78         my $dir = (getpwnam($user))[7] || die "no user $user";
  79         my $md5_path = "$dir/.md5";
  80         my %h;
  81         tie %h, "TokyoCabinet::HDB", $md5_path || die "$md5_path: $!";
  82         my $md5 = $h{$path} || die "can't find $path in $md5_path";
  83         untie %h;
  84
  85         warn "# $md5 $path";
  86
  87         if ( ! -e "$torrent_dir/download/$md5" ) {
  88                 symlink "/rsync1/s1/md5/$md5", "$torrent_dir/download/$md5"; # FIXME slice path
  89                 warn "# created $torrent_dir/download/$md5";
  90         }
  91
  92         if ( ! -e "$torrent_dir/md5/$md5.torrent" ) {
  93                 my $tracker = "http://213.186.16.27:6502/announce"; # FIXME
  94                 system "mktorrent --announce $tracker --no-date --verbose --output $torrent_dir/md5/$md5.torrent $torrent_dir/download/$md5";
  95
  96         }
  97
  98         symlink "$torrent_dir/md5/$md5.torrent", "$dir/$path.torrent" unless -e "$dir/$path.torrent";
  99
 100         $tc->add( filename => "$torrent_dir/md5/$md5.torrent" );
 101
 102         return "$dir/$path";
 103 } );
 104
 105 warn "$0 $name pid $$ waitng for jobs\n";
 106 $worker->work while 1;
 107