Blame src/mgetJob.cc

Packit 8f70b4
/*
Packit 8f70b4
 * lftp - file transfer program
Packit 8f70b4
 *
Packit 8f70b4
 * Copyright (c) 1996-2012 by Alexander V. Lukyanov (lav@yars.free.net)
Packit 8f70b4
 *
Packit 8f70b4
 * This program is free software; you can redistribute it and/or modify
Packit 8f70b4
 * it under the terms of the GNU General Public License as published by
Packit 8f70b4
 * the Free Software Foundation; either version 3 of the License, or
Packit 8f70b4
 * (at your option) any later version.
Packit 8f70b4
 *
Packit 8f70b4
 * This program is distributed in the hope that it will be useful,
Packit 8f70b4
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
Packit 8f70b4
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Packit 8f70b4
 * GNU General Public License for more details.
Packit 8f70b4
 *
Packit 8f70b4
 * You should have received a copy of the GNU General Public License
Packit 8f70b4
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
Packit 8f70b4
 */
Packit 8f70b4
Packit 8f70b4
#include <config.h>
Packit 8f70b4
#include <sys/types.h>
Packit 8f70b4
#include <sys/stat.h>
Packit 8f70b4
#include "mgetJob.h"
Packit 8f70b4
#include "misc.h"
Packit 8f70b4
#include "ArgV.h"
Packit 8f70b4
#include "url.h"
Packit 8f70b4
Packit 8f70b4
void mgetJob::ShowRunStatus(const SMTaskRef<StatusLine>& s)
Packit 8f70b4
{
Packit 8f70b4
   if(glob)
Packit 8f70b4
   {
Packit 8f70b4
      s->Show("%s",glob->Status());
Packit 8f70b4
      return;
Packit 8f70b4
   }
Packit 8f70b4
   GetJob::ShowRunStatus(s);
Packit 8f70b4
}
Packit 8f70b4
xstring& mgetJob::FormatStatus(xstring& buf,int v,const char *prefix)
Packit 8f70b4
{
Packit 8f70b4
   if(glob)
Packit 8f70b4
   {
Packit 8f70b4
      SessionJob::FormatStatus(buf,v,prefix);
Packit 8f70b4
      const char *s=glob->Status();
Packit 8f70b4
      if(!s || !s[0])
Packit 8f70b4
	 return buf;
Packit 8f70b4
      return buf.appendf("\t%s\n",s);
Packit 8f70b4
   }
Packit 8f70b4
   return GetJob::FormatStatus(buf,v,prefix);
Packit 8f70b4
}
Packit 8f70b4
Packit 8f70b4
mgetJob::mgetJob(FileAccess *session,const ArgV *a,bool c,bool md)
Packit 8f70b4
   : GetJob(session,new ArgV(a->a0()),c),
Packit 8f70b4
   local_session(FileAccess::New("file"))
Packit 8f70b4
{
Packit 8f70b4
   make_dirs=md;
Packit 8f70b4
   for(int i=a->getindex(); i<a->count(); i++)
Packit 8f70b4
      wcd.push(xstrdup(a->getarg(i)));
Packit 8f70b4
}
Packit 8f70b4
Packit 8f70b4
int mgetJob::Do()
Packit 8f70b4
{
Packit 8f70b4
   if(glob) {
Packit 8f70b4
      // handle ongoing globbing
Packit 8f70b4
      if(glob->Error())
Packit 8f70b4
      {
Packit 8f70b4
	 fprintf(stderr,"%s: %s: %s\n",op,glob->GetPattern(),glob->ErrorText());
Packit 8f70b4
      glob_error:
Packit 8f70b4
	 count++;
Packit 8f70b4
	 errors++;
Packit 8f70b4
	 glob=0;
Packit 8f70b4
	 return MOVED;
Packit 8f70b4
      }
Packit 8f70b4
      if(!glob->Done())
Packit 8f70b4
	 return STALL;
Packit 8f70b4
      // have globbed file set now
Packit 8f70b4
      FileSet *files=glob->GetResult();
Packit 8f70b4
      if(files->get_fnum()==0)
Packit 8f70b4
      {
Packit 8f70b4
	 fprintf(stderr,_("%s: %s: no files found\n"),op,glob->GetPattern());
Packit 8f70b4
	 goto glob_error;
Packit 8f70b4
      }
Packit 8f70b4
      files->rewind();
Packit 8f70b4
      for(FileInfo *fi=files->curr(); fi; fi=files->next()) {
Packit 8f70b4
	 xstring& src=fi->name;
Packit 8f70b4
	 args->Append(src);
Packit 8f70b4
	 args->Append(output_file_name(src,0,!reverse,output_dir,make_dirs));
Packit 8f70b4
      }
Packit 8f70b4
      glob=0;
Packit 8f70b4
   }
Packit 8f70b4
Packit 8f70b4
   if(wcd.count()>0) {
Packit 8f70b4
      glob=new GlobURL(reverse?local_session:session,
Packit 8f70b4
			wcd.next(),GlobURL::FILES_ONLY);
Packit 8f70b4
      return MOVED;
Packit 8f70b4
   }
Packit 8f70b4
Packit 8f70b4
   return GetJob::Do();
Packit 8f70b4
}
Packit 8f70b4
Packit 8f70b4
mgetJob::~mgetJob()
Packit 8f70b4
{
Packit 8f70b4
}