#!/usr/bin/perl -w
# Fastcmd - Fast Copy/Move/Delete whole directory trees like mv -rd rm -rd
#moded by A George; use at your own risk!
#EXAMPLE call under linux or windows: perl /home/alistair/perl/fastcmd.pl del . (deletes everything in current directory)
#tested under Linux deleted 30,000 files in 4 seconds on ext4 hdd partition
#ui error (not bug) couldn't rmdir directory .: Invalid argument at /home/alistair/perl/fastremove.pl line 17.

use File::Find;
use File::Path qw(make_path);
use strict;
use warnings FATAL=>"all";
use File::Copy::Recursive qw(fcopy rcopy dircopy fmove rmove dirmove);
use File::Basename;

use 5.010;

$SIG{INT} = \&interrupt; #trap Ctrl-C 

my ($function, $sourcedir, $targetdir) = @ARGV;
die ": missing operand. Usage: $0 <OPTION>... <SOURCE_DIRECTORY>... [TARGET_DIRECTORY]...
OPTION all the FILE(s) and DIRECTORIES:
-c = copy or -co = copy with overwrite
-m = move or -mo = move with overwrite
-d = delete
\n" unless (defined $function && defined $sourcedir); #@ARGV;

#check validity input
opendir(my $source_handle, $sourcedir) or die "Can't opendir $sourcedir: $!\n";
if (!defined($targetdir)) {$targetdir = '.'};
my $overwrite = 0 + (length $function>2) && $function =~ m/..o/ ;
#strip last character (dont need the 'o' now)
$overwrite>0 && chop($function);
#print STDERR "$overwrite $function\n";
#$File::Copy::Recursive::CPRFComp = 1;
$File::Copy::Recursive::SkipFlop = 1;
#use below later for logfile
#$File::Copy::Recursive::RMTrgDir = $overwrite;
#$File::Copy::Recursive::RMTrgFil = $overwrite;

#Hook information subroutine
use Hook::WrapSub qw(wrap_subs);
use IO::Handle;
use Time::HiRes qw(gettimeofday);
my $t0 = gettimeofday( );
my $FDcount = 0;
my $lastitem = "";
my $num_of_files_and_dirs = 0;
use File::Spec::Functions;

sub after_dircopy {
#if ($lastitem eq $!) {print STDERR "\n$lastitem"; return;}
#$lastitem = $!;
my $t1 = gettimeofday( );
my $elapsed = $t1 - $t0;
printf("\rFile: Elapsed time since start: H%02d:M%02d:S%02d Approx number of files and directories processed: %02d Ctrl-C breaks",
($t1 - $t0) / (60*60),
($t1 - $t0) / (   60) % 60,
($t1 - $t0)           % 60,
++$FDcount); 
STDOUT->flush;
}


wrap_subs sub {}, 'File::Copy::Recursive::fcopy','File::Copy::Recursive::pathrm','File::Copy::Recursive::pathempty',\&after_dircopy;

#if user enters './' we replicate source directory, otherwise if '.' destination current, otherwise if '' destination still current '.'
if ($targetdir =~ /^[.\/]{2}$/) {$targetdir = './'.basename($sourcedir);}

#if ($function ne "-d")
#{
#print"\n$overwrite $function, $sourcedir, $targetdir \nConfirm you wish to proceed y/n?"; 
#chomp(my $input = <STDIN>);
#if ($input !~ /^[Y]?$/i) {exit();}
#}

if ($function eq "-d") 		#DELETE EVERYTHING
{
print "This script is DANGEROUS - Are you sure you want to delete EVERYTHING in directory ",$sourcedir," (y/n): "; #,@ARGV," (y/n): ";
chomp(my $input = <STDIN>);
if ($input !~ /^[Y]?$/i) {exit();}
File::Copy::Recursive::pathrm("$sourcedir/","1") or print"\nProcess completed\n";
exit;
} 


elsif ($function eq "-m")	#MOVE ROUTINE
{
exit;
length $targetdir>0 or die "No target directory given $targetdir/: $!\n";
make_path("$targetdir/") or die "Cannot create $targetdir/: $!\n";
$num_of_files_and_dirs = dirmove("$sourcedir/", "$targetdir/")
or die "Cannot move $sourcedir/: $!\n";
print"\nFiles & Dirs removed: $num_of_files_and_dirs\n";
exit;
}

else						#COPY ROUTINE
{
length $targetdir>0 or die "No target directory given $targetdir/: $!\n";
$num_of_files_and_dirs = dircopy("$sourcedir/", "$targetdir/") or die "\nCannot copy $!";
print "\nFiles & Dirs copied: $num_of_files_and_dirs\n";
exit;
}

sub interrupt {die "\nProcess interrupted with by Ctrl-C\n Number of Files: $num_of_files_and_dirs/ processed\n"; exit(0);}

#ENDS