Blame src/mkdirJob.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
Packit Service a2489d
#include <errno.h>
Packit Service a2489d
#include "mkdirJob.h"
Packit Service a2489d
#include "plural.h"
Packit Service a2489d
#include "url.h"
Packit Service a2489d
#include "misc.h"
Packit Service a2489d
Packit Service a2489d
#define super SessionJob
Packit Service a2489d
#define orig_session super::session
Packit Service a2489d
Packit Service a2489d
mkdirJob::mkdirJob(FileAccess *s,ArgV *a)
Packit Service a2489d
   : super(s), args(a), session(orig_session)
Packit Service a2489d
{
Packit Service a2489d
   quiet=false;
Packit Service a2489d
   failed=file_count=0;
Packit Service a2489d
Packit Service a2489d
   args->rewind();
Packit Service a2489d
   const char *op=args->getarg(0);
Packit Service a2489d
Packit Service a2489d
   curr=first=0;
Packit Service a2489d
   opt_p=false;
Packit Service a2489d
Packit Service a2489d
   int opt;
Packit Service a2489d
   while((opt=args->getopt("pf"))!=EOF)
Packit Service a2489d
   {
Packit Service a2489d
      switch(opt)
Packit Service a2489d
      {
Packit Service a2489d
      case 'p':
Packit Service a2489d
	 opt_p=true;
Packit Service a2489d
	 break;
Packit Service a2489d
      case 'f':
Packit Service a2489d
	 quiet=true;
Packit Service a2489d
	 break;
Packit Service a2489d
      default:
Packit Service a2489d
	 return;
Packit Service a2489d
      }
Packit Service a2489d
   }
Packit Service a2489d
   args->back();
Packit Service a2489d
   first=curr=args->getnext();
Packit Service a2489d
   if(curr==0)
Packit Service a2489d
   {
Packit Service a2489d
      fprintf(stderr,"Usage: %s [-p] [-f] paths...\n",op);
Packit Service a2489d
      return;
Packit Service a2489d
   }
Packit Service a2489d
}
Packit Service a2489d
Packit Service a2489d
int mkdirJob::Do()
Packit Service a2489d
{
Packit Service a2489d
   if(Done())
Packit Service a2489d
      return STALL;
Packit Service a2489d
   if(session->IsClosed())
Packit Service a2489d
   {
Packit Service a2489d
      ParsedURL u(curr,true);
Packit Service a2489d
      if(u.proto)
Packit Service a2489d
      {
Packit Service a2489d
	 session=my_session=FileAccess::New(&u);
Packit Service a2489d
	 session->SetPriority(fg?1:0);
Packit Service a2489d
	 session->Mkdir(u.path,opt_p);
Packit Service a2489d
      }
Packit Service a2489d
      else
Packit Service a2489d
      {
Packit Service a2489d
	 session=orig_session;
Packit Service a2489d
	 session->Mkdir(curr,opt_p);
Packit Service a2489d
      }
Packit Service a2489d
   }
Packit Service a2489d
Packit Service a2489d
   int res=session->Done();
Packit Service a2489d
   if(res==FA::DO_AGAIN || res==FA::IN_PROGRESS)
Packit Service a2489d
      return STALL;
Packit Service a2489d
   if(res<0)
Packit Service a2489d
   {
Packit Service a2489d
      failed++;
Packit Service a2489d
      if(!quiet)
Packit Service a2489d
	 fprintf(stderr,"%s: %s\n",args->getarg(0),session->StrError(res));
Packit Service a2489d
   }
Packit Service a2489d
   file_count++;
Packit Service a2489d
   session->Close();
Packit Service a2489d
   curr=args->getnext();
Packit Service a2489d
   return MOVED;
Packit Service a2489d
}
Packit Service a2489d
Packit Service a2489d
xstring& mkdirJob::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
Packit Service a2489d
   return s.appendf("%s`%s' [%s]\n",prefix,curr,session->CurrentStatus());
Packit Service a2489d
}
Packit Service a2489d
Packit Service a2489d
void  mkdirJob::ShowRunStatus(const SMTaskRef<StatusLine>& s)
Packit Service a2489d
{
Packit Service a2489d
   if(Done())
Packit Service a2489d
      return;
Packit Service a2489d
Packit Service a2489d
   s->Show("%s `%s' [%s]",args->getarg(0),
Packit Service a2489d
      squeeze_file_name(curr,s->GetWidthDelayed()-40),
Packit Service a2489d
      session->CurrentStatus());
Packit Service a2489d
}
Packit Service a2489d
Packit Service a2489d
void  mkdirJob::SayFinal()
Packit Service a2489d
{
Packit Service a2489d
   if(failed==file_count)
Packit Service a2489d
      return;
Packit Service a2489d
   const char *op=args->getarg(0);
Packit Service a2489d
   if(file_count==1)
Packit Service a2489d
      // A directory has just been created
Packit Service a2489d
      printf(_("%s ok, `%s' created\n"),op,first);
Packit Service a2489d
   else if(failed)
Packit Service a2489d
      printf(plural("%s failed for %d of %d director$y|ies$\n",file_count),
Packit Service a2489d
	    op,failed,file_count);
Packit Service a2489d
   else
Packit Service a2489d
      printf(plural("%s ok, %d director$y|ies$ created\n",file_count),
Packit Service a2489d
	    op,file_count);
Packit Service a2489d
}
Packit Service a2489d
Packit Service a2489d
void mkdirJob::Fg()
Packit Service a2489d
{
Packit Service a2489d
   super::Fg();
Packit Service a2489d
   if(orig_session!=session)
Packit Service a2489d
      orig_session->SetPriority(1);
Packit Service a2489d
}
Packit Service a2489d
void mkdirJob::Bg()
Packit Service a2489d
{
Packit Service a2489d
   if(orig_session!=session)
Packit Service a2489d
      orig_session->SetPriority(0);
Packit Service a2489d
   super::Bg();
Packit Service a2489d
}