1 package CloudStore::Store;
7 use File::Path qw(make_path);
9 use Digest::MD5 qw(md5_base64);
10 use Data::Dump qw(dump);
22 die "no dir" unless $self->{dir};
23 $self->{md5pool} = $self->{dir} . '/md5';
25 warn "# new ",dump $self if $ENV{DEBUG};
31 my ( $self,$data ) = @_;
35 my ( $self,$data ) = @_;
40 $dir =~ s{/[^/]+$}{}; # strip filename
41 mkdir $dir unless -e $dir;
45 my ( $self,$data ) = @_;
47 if ( $data->{file} =~ m{^(.*/)?.sync/send/([^/]+)$} ) {
49 warn "SEND $2 from $from_dir\n";
51 open(my $send, '<', $self->blob_path($data) );
55 my ( $to, $file ) = split(/\s+/,$_,2);
56 my ( undef, undef, $uid, $gid, undef, undef, $email, $dir, $shell ) =
60 $from->{file} = $from_dir . $file;
61 my $from_path = $self->blob_path($from);
63 if ( ! -r $from_path ) {
64 warn "ERROR: $from_path: $!";
68 my $to_path = "$dir/received/$file";
71 warn "SEND $from_path -> $to_path\n";
72 unlink $to_path if -e $to_path; # FIXME why we need this?
73 $sent_files->{$to} += link $from_path, $to_path;
75 my $md5 = $self->md5sum($data)->get( $from_dir . $file ) || warn "no md5 for $from_dir$file";
76 $self->md5sum({login => $to})->put( "/received/$file" => $md5 );
79 warn "SENT ",dump $sent_files;
81 return 0; # skip dedup
82 } elsif ( $data->{file} =~ m{^(.*/)?.sync/pending/([^/]+)$} ) {
84 warn "PENDIG $2 from $from_dir";
85 open(my $pend, '<', $self->blob_path($data) );
89 if ( m/^DELETED\#(.+)$/ ) {
90 my $path = $from_dir . $1;
92 unlink $path && warn "unlink $path\n";
94 warn "MISSING $path to unlink\n";
97 } elsif ( ! /^(MOVED|RENAMED)\#/ ) {
102 my ( undef, $from, $to ) = split(/\#/,$_,3);
104 my ( $from_path, $to_path ) = map {
106 $tmp->{file} = $from_dir . $_;
107 $self->blob_path($tmp);
110 if ( ! -e $from_path ) {
111 warn "SKIPPED $from_path: $!";
115 warn "MV $from_path -> $to_path";
117 rename $from_path, $to_path;
119 my $md5 = $self->md5sum($data)->get( $from_dir . $from );
121 warn "ERROR: no md5sum $from_dir $from " unless $md5;
125 $self->md5sum($data)->out( $from_dir . $from );
126 $self->md5sum($data)->put( $from_dir . $to => $md5 );
128 warn "$md5 moved to $from_dir $to";
131 return 0; # skip dedup
134 #return $file->{size} > 4096 ? 1 : 0; # FIXME
138 # never called by rsync directly!
140 my ( $self,$data ) = @_;
141 # $self->file_set($data);
145 my ( $self, $data ) = @_;
147 my $md5 = $self->md5sum($data)->get( $data->{file} );
148 return unless $md5; # directories don't have md5sums
149 my $path = $self->{md5pool} . '/' . $md5;
150 my ($dev,$ino,$mode,$nlink,$uid,$gid,$rdev,$size,
151 $atime,$mtime,$ctime,$blksize,$blocks)
154 my $id = getpwnam 'md5';
155 chown $id,$gid, $path;
156 warn "# chown $id $gid $path";
159 $self->md5sum($data)->out( $data->{file} );
163 my ( $self, $data ) = @_;
168 my ( $self, $data ) = @_;
170 warn "# new_link ",dump $data;
172 if ( $data->{file} =~ m{^(.*/?)\.send/([^/]+)/(.+)$} ) {
173 my ( $dir, $to, $name ) = ( $1, $2, $3 );
174 my $path = $self->blob_path($data);
175 my $link_to = readlink $path;
177 if ( $link_to =~ s{^\Q/rsyncd-munged/\E}{/} ) {
180 $s =~ s{/[^/]+$}{}; # strip filename
181 while ( $link_to =~ s{/../}{/} ) {
182 $s =~ s{/[^/]+$}{} || die "can't strip $s";
183 warn "## simplify $s $link_to\n";
187 my $d = $self->blob_path({
192 # $name can contain directories so we must create them
194 $to_dir =~ s{/[^/]+$}{};
195 make_path $to_dir if ! -e $to_dir;
198 warn "ERROR: can't find source $s";
201 warn "link $s -> $d\n";
204 my ($l,$f) = ($1,$2) if $s =~ m{users/([^/]+)/blob/(.+)};
206 # my $origin = $self->file_get({
210 # $self->new_file($origin);
211 warn "INFO: sent file ",dump($l,$f);
212 my $md5 = $self->md5sum($data)->get($s);
213 $self->md5sum({ login => $to })->put($d => $md5 );
219 warn "ERROR: can't SEND To:$to Name:$name Link:$link_to";
225 my ( $self, $data ) = @_;
227 my $login = $data->{login} || confess "missing login in ",dump $data;
229 return $self->{md5sum}->{$login} if exists $self->{md5sum}->{$login};
231 my ( undef, undef, $uid, $gid, undef, undef, $email, $dir, $shell ) =
234 my $md5_path = "$dir/.md5";
236 my $db = TokyoCabinet::HDB->new();
237 $db->open($md5_path, $db->OWRITER | $db->OCREAT)
238 or die "can't open $md5_path: ",$db->errmsg( $db->ecode );
240 warn "open $md5_path";
242 $self->{md5sum}->{$login} = $db;
247 my ( $self, $data ) = @_;
248 if ( $data =~ m/\[(\d+)\] rsync \w+ (\S+) \w+ (\S+)/ ) {
249 my ( $pid, $module, $login ) = ( $1, $2, $3 );
252 my ( undef, undef, $uid, $gid, undef, undef, $email, $dir, $shell ) =
255 $self->{pid}->{$pid} = {
266 } elsif ( $data =~ m/\[(\d+)\] sent \S+ bytes\s+received \S+ bytes/ ) {
269 foreach my $login ( keys %{ $self->{md5sum} } ) {
270 $self->{md5sum}->{$login}->close;
271 warn "close md5sum $login";
273 delete $self->{md5sum};
275 delete $self->{pid}->{$pid};
280 # warn "## rsync_log $data";
285 my ( $self, $data ) = @_;
286 my $blob = $self->{pid}->{ $data->{pid} }->{dir} || die "no dir for $data->{pid} in ",dump( $self->{pid} );
287 $blob .= '/' . $data->{file};
293 my ( $self,$data ) = @_;
295 my $path = $self->blob_path($data);
297 if ( $data->{itemize} =~ m/^[c>]([fdL])/ ) { # received change/create
300 if ( $type eq 'f' ) {
301 $self->modify_file( $data ) && # selective dedup
302 $self->dedup( $data, $path );
303 } elsif ( $type eq 'd' ) {
304 $self->make_dir( $data );
305 } elsif ( $type eq 'L' ) {
306 $self->new_link( $data );
308 die "unknown type $type ", dump $data;
310 } elsif ( $data->{itemize} =~ m/\*deleting/ ) {
311 $self->remove_file($data);
313 warn "IGNORED ",dump($data) if $ENV{DEBUG};
319 my ( $self, $data ) = @_;
321 my $pool = $self->{md5pool} || die "no md5pool in ",dump $self;
322 mkdir $pool unless -e $pool;
324 my $md5 = $data->{md5} || die "no md5 in ",dump $data;
325 my $path = $self->blob_path($data);
327 if ( -e "$pool/$md5" ) {
328 warn "dedup hit $md5 $path\n";
329 my $dedup = $path . '.dedup';
330 rename $path, $dedup;
331 link "$pool/$md5", $path;
335 link $path, "$pool/$md5";
336 warn "dedup +++ $md5 $path";
339 $self->md5sum($data)->put( $data->{file} => $md5 );
342 my $empty_md5 = " " x 32;
345 my ( $self, $data, $path ) = @_;
347 if ( $data->{file} =~ /^(.+\/)?md5sum$/ ) {
350 warn "IMPORT ", $data->{file}, "\n";
351 open(my $md5sum, '<', $path);
354 my ( $md5, $file ) = split(/\s+/,$_,2);
355 if ( ! -e "$self->{md5path}/$md5" ) {
356 warn "MISSING $md5 $file\n";
364 my $new_path = $self->blob_path($new);
365 if ( ! -e $new_path ) {
366 # create path from md5sum file
367 my $only_dir = $1 if $new =~ m{^(.+)/[^/]+$};
368 make_path $only_dir unless -d $only_dir;
369 $imported += link "$self->{md5path}/$md5", $new_path;
370 $self->new_file($new);
371 warn "import from $path ",dump($new);
372 $self->md5pool( $new );
374 $self->md5pool( $new );
377 print "INFO imported $imported files from ",dump($data);
380 if ( $data->{md5} ne $empty_md5 ) {
381 $self->md5pool( $data );
383 warn "empty md5", dump $data;