/*
 * lftp - file transfer program
 *
 * Copyright (c) 1996-2017 by Alexander V. Lukyanov (lav@yars.free.net)
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <config.h>
#include <errno.h>
#include "trio.h"
#include "mmvJob.h"
#include "misc.h"
#include "plural.h"
#include "FileGlob.h"

mmvJob::mmvJob(FileAccess *session,const ArgV *args,const char *t,FA::open_mode m1)
 : SessionJob(session), dst_dir(t), m(m1),
   remove_target(false), moved_count(0), error_count(0), done(false)
{
   op.set(args->a0());
   for(int i=args->getindex(); i<args->count(); i++)
      wcd.push(xstrdup(args->getarg(i)));
}

void mmvJob::doOpen() const
{
   if(remove_target && session->OpenMode()!=FA::REMOVE)
      session->Open(curr_dst,FA::REMOVE);
   else
      session->Open2(curr_src,curr_dst,m);
}

int mmvJob::Do()
{
   int m=STALL;

   if(Done())
      return STALL;

   if(glob) {
      // handle ongoing globbing
      if(glob->Error())
      {
	 fprintf(stderr,"%s: %s: %s\n",cmd(),glob->GetPattern(),glob->ErrorText());
	 error_count++;
	 glob=0;
	 return MOVED;
      }
      if(!glob->Done())
	 return m;
      // have globbed file set now
      FileSet *files=glob->GetResult();
      files->rewind();
      for(FileInfo *fi=files->curr(); fi; fi=files->next())
	 src.push(fi->name.borrow());
      glob=0;
   }

   if(!curr_src) {
      // pick next file/wildcard to work on
      if(src.count()) {
	 curr_src.set(src.next());
	 curr_dst.set(dir_file(dst_dir,basename_ptr(curr_src)));
      } else if(wcd.count()) {
	 glob=session->MakeGlob(wcd.next());
	 glob->Roll();
	 return MOVED;
      } else {
	 // no more files to move
	 done=true;
	 return MOVED;
      }
   }

   if(!session->IsOpen())
      doOpen();

   int res=session->Done();
   if(res==FA::IN_PROGRESS || res==FA::DO_AGAIN)
      return m;
   if(res!=FA::OK && !isRemoving()) {
      fprintf(stderr,"%s: %s\n",cmd(),session->StrError(res));
      error_count++;
      session->Close();
      curr_src.unset();
      return MOVED;
   }
   if(isRemoving()) {
      doOpen(); // do the real move now.
      return MOVED;
   }
   session->Close();
   moved_count++;
   curr_src.unset();
   return MOVED;
}

xstring& mmvJob::FormatStatus(xstring& s,int v,const char *prefix)
{
   SessionJob::FormatStatus(s,v,prefix);
   if(Done())
      return s;
   if(glob)
      s.appendf("%sglob %s [%s]\n",prefix,glob->GetPattern(),glob->Status());
   else if(isRemoving())
      s.appendf("%srm %s [%s]\n",prefix,curr_dst.get(),session->CurrentStatus());
   else
      s.appendf("%s%s %s=>%s [%s]\n",prefix,cmd(),curr_src.get(),curr_dst.get(),session->CurrentStatus());
   return s;
}

void  mmvJob::ShowRunStatus(const SMTaskRef<StatusLine>& s)
{
   if(Done())
      return;
   if(glob)
      s->Show("glob %s [%s]",glob->GetPattern(),glob->Status());
   else if(isRemoving())
      s->Show("rm %s [%s]",curr_dst.get(),session->CurrentStatus());
   else
      s->Show("%s %s=>%s [%s]",cmd(),curr_src.get(),curr_dst.get(),session->CurrentStatus());
}

void  mmvJob::SayFinal()
{
   if(error_count>0)
      printf(plural("%s: %d error$|s$ detected\n",error_count),cmd(),error_count);
   if(m==FA::RENAME)
      printf(plural("%s: %d file$|s$ moved\n",moved_count),cmd(),moved_count);
   else
      printf(plural("%s: %d file$|s$ linked\n",moved_count),cmd(),moved_count);
}