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