/*
* lftp - file transfer program
*
* Copyright (c) 1996-2017 by Alexander V. Lukyanov (lav@yars.free.net)
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see .
*/
#include
#include
#include "trio.h"
#include "mmvJob.h"
#include "misc.h"
#include "plural.h"
#include "FileGlob.h"
mmvJob::mmvJob(FileAccess *session,const ArgV *args,const char *t,FA::open_mode m1)
: SessionJob(session), dst_dir(t), m(m1),
remove_target(false), moved_count(0), error_count(0), done(false)
{
op.set(args->a0());
for(int i=args->getindex(); icount(); i++)
wcd.push(xstrdup(args->getarg(i)));
}
void mmvJob::doOpen() const
{
if(remove_target && session->OpenMode()!=FA::REMOVE)
session->Open(curr_dst,FA::REMOVE);
else
session->Open2(curr_src,curr_dst,m);
}
int mmvJob::Do()
{
int m=STALL;
if(Done())
return STALL;
if(glob) {
// handle ongoing globbing
if(glob->Error())
{
fprintf(stderr,"%s: %s: %s\n",cmd(),glob->GetPattern(),glob->ErrorText());
error_count++;
glob=0;
return MOVED;
}
if(!glob->Done())
return m;
// have globbed file set now
FileSet *files=glob->GetResult();
files->rewind();
for(FileInfo *fi=files->curr(); fi; fi=files->next())
src.push(fi->name.borrow());
glob=0;
}
if(!curr_src) {
// pick next file/wildcard to work on
if(src.count()) {
curr_src.set(src.next());
curr_dst.set(dir_file(dst_dir,basename_ptr(curr_src)));
} else if(wcd.count()) {
glob=session->MakeGlob(wcd.next());
glob->Roll();
return MOVED;
} else {
// no more files to move
done=true;
return MOVED;
}
}
if(!session->IsOpen())
doOpen();
int res=session->Done();
if(res==FA::IN_PROGRESS || res==FA::DO_AGAIN)
return m;
if(res!=FA::OK && !isRemoving()) {
fprintf(stderr,"%s: %s\n",cmd(),session->StrError(res));
error_count++;
session->Close();
curr_src.unset();
return MOVED;
}
if(isRemoving()) {
doOpen(); // do the real move now.
return MOVED;
}
session->Close();
moved_count++;
curr_src.unset();
return MOVED;
}
xstring& mmvJob::FormatStatus(xstring& s,int v,const char *prefix)
{
SessionJob::FormatStatus(s,v,prefix);
if(Done())
return s;
if(glob)
s.appendf("%sglob %s [%s]\n",prefix,glob->GetPattern(),glob->Status());
else if(isRemoving())
s.appendf("%srm %s [%s]\n",prefix,curr_dst.get(),session->CurrentStatus());
else
s.appendf("%s%s %s=>%s [%s]\n",prefix,cmd(),curr_src.get(),curr_dst.get(),session->CurrentStatus());
return s;
}
void mmvJob::ShowRunStatus(const SMTaskRef& s)
{
if(Done())
return;
if(glob)
s->Show("glob %s [%s]",glob->GetPattern(),glob->Status());
else if(isRemoving())
s->Show("rm %s [%s]",curr_dst.get(),session->CurrentStatus());
else
s->Show("%s %s=>%s [%s]",cmd(),curr_src.get(),curr_dst.get(),session->CurrentStatus());
}
void mmvJob::SayFinal()
{
if(error_count>0)
printf(plural("%s: %d error$|s$ detected\n",error_count),cmd(),error_count);
if(m==FA::RENAME)
printf(plural("%s: %d file$|s$ moved\n",moved_count),cmd(),moved_count);
else
printf(plural("%s: %d file$|s$ linked\n",moved_count),cmd(),moved_count);
}