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