Blame src/mgetJob.cc

Packit Service a2489d
/*
Packit Service a2489d
 * lftp - file transfer program
Packit Service a2489d
 *
Packit Service a2489d
 * Copyright (c) 1996-2012 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 <sys/types.h>
Packit Service a2489d
#include <sys/stat.h>
Packit Service a2489d
#include "mgetJob.h"
Packit Service a2489d
#include "misc.h"
Packit Service a2489d
#include "ArgV.h"
Packit Service a2489d
#include "url.h"
Packit Service a2489d
Packit Service a2489d
void mgetJob::ShowRunStatus(const SMTaskRef<StatusLine>& s)
Packit Service a2489d
{
Packit Service a2489d
   if(glob)
Packit Service a2489d
   {
Packit Service a2489d
      s->Show("%s",glob->Status());
Packit Service a2489d
      return;
Packit Service a2489d
   }
Packit Service a2489d
   GetJob::ShowRunStatus(s);
Packit Service a2489d
}
Packit Service a2489d
xstring& mgetJob::FormatStatus(xstring& buf,int v,const char *prefix)
Packit Service a2489d
{
Packit Service a2489d
   if(glob)
Packit Service a2489d
   {
Packit Service a2489d
      SessionJob::FormatStatus(buf,v,prefix);
Packit Service a2489d
      const char *s=glob->Status();
Packit Service a2489d
      if(!s || !s[0])
Packit Service a2489d
	 return buf;
Packit Service a2489d
      return buf.appendf("\t%s\n",s);
Packit Service a2489d
   }
Packit Service a2489d
   return GetJob::FormatStatus(buf,v,prefix);
Packit Service a2489d
}
Packit Service a2489d
Packit Service a2489d
mgetJob::mgetJob(FileAccess *session,const ArgV *a,bool c,bool md)
Packit Service a2489d
   : GetJob(session,new ArgV(a->a0()),c),
Packit Service a2489d
   local_session(FileAccess::New("file"))
Packit Service a2489d
{
Packit Service a2489d
   make_dirs=md;
Packit Service a2489d
   for(int i=a->getindex(); i<a->count(); i++)
Packit Service a2489d
      wcd.push(xstrdup(a->getarg(i)));
Packit Service a2489d
}
Packit Service a2489d
Packit Service a2489d
int mgetJob::Do()
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",op,glob->GetPattern(),glob->ErrorText());
Packit Service a2489d
      glob_error:
Packit Service a2489d
	 count++;
Packit Service a2489d
	 errors++;
Packit Service a2489d
	 glob=0;
Packit Service a2489d
	 return MOVED;
Packit Service a2489d
      }
Packit Service a2489d
      if(!glob->Done())
Packit Service a2489d
	 return STALL;
Packit Service a2489d
      // have globbed file set now
Packit Service a2489d
      FileSet *files=glob->GetResult();
Packit Service a2489d
      if(files->get_fnum()==0)
Packit Service a2489d
      {
Packit Service a2489d
	 fprintf(stderr,_("%s: %s: no files found\n"),op,glob->GetPattern());
Packit Service a2489d
	 goto glob_error;
Packit Service a2489d
      }
Packit Service a2489d
      files->rewind();
Packit Service a2489d
      for(FileInfo *fi=files->curr(); fi; fi=files->next()) {
Packit Service a2489d
	 xstring& src=fi->name;
Packit Service a2489d
	 args->Append(src);
Packit Service a2489d
	 args->Append(output_file_name(src,0,!reverse,output_dir,make_dirs));
Packit Service a2489d
      }
Packit Service a2489d
      glob=0;
Packit Service a2489d
   }
Packit Service a2489d
Packit Service a2489d
   if(wcd.count()>0) {
Packit Service a2489d
      glob=new GlobURL(reverse?local_session:session,
Packit Service a2489d
			wcd.next(),GlobURL::FILES_ONLY);
Packit Service a2489d
      return MOVED;
Packit Service a2489d
   }
Packit Service a2489d
Packit Service a2489d
   return GetJob::Do();
Packit Service a2489d
}
Packit Service a2489d
Packit Service a2489d
mgetJob::~mgetJob()
Packit Service a2489d
{
Packit Service a2489d
}