// $Id$
//-----------------------------------------------------------------------
//       The GSI Online Offline Object Oriented (Go4) Project
//         Experiment Data Processing at EE department, GSI
//-----------------------------------------------------------------------
// Copyright (C) 2000- GSI Helmholtzzentrum fuer Schwerionenforschung GmbH
//                     Planckstr. 1, 64291 Darmstadt, Germany
// Contact:            http://go4.gsi.de
//-----------------------------------------------------------------------
// This software can be used under the license agreements as stated
// in Go4License.txt file which is part of the distribution.
//-----------------------------------------------------------------------

#include "TGo4ClientTask.h"

#include "TGo4Log.h"

#include "TGo4BufferQueue.h"
#include "TGo4ClientStatus.h"
#include "TGo4ThreadHandler.h"
#include "TGo4TaskHandler.h"

TGo4ClientTask::TGo4ClientTask(const char *name,
                               const char *serverhost,
                               UInt_t negotiationport,
                               Bool_t blockingmode,
                               Bool_t standalone,
                               Bool_t autostart,
                               Bool_t autocreate,
                               Bool_t ismaster,
                               Bool_t autoconnect) :
   TGo4Task(name,blockingmode,autostart,autocreate,ismaster),
   fxTaskHandler(nullptr),
   fbAutoConnect(autoconnect),
   fbServerConnected(kFALSE)
{
   fxServerHostname = serverhost;
   TString nomen("TaskHandler of ");
   nomen += name;
   fxTaskHandler = new TGo4TaskHandler(nomen.Data(), this, kTRUE, IsMaster(), negotiationport);
   fxCommandQ = dynamic_cast<TGo4BufferQueue *>(GetTaskHandler()->GetCommandQueue());
   fxStatusQ = dynamic_cast<TGo4BufferQueue *>(GetTaskHandler()->GetStatusQueue());
   fxDataQ = dynamic_cast<TGo4BufferQueue *>(GetTaskHandler()->GetDataQueue());
   if (standalone) {
      Launch(); // create threads, start application control timer
   } else {
      // subclass must call Launch at end of its ctor
   }
}

TGo4ClientTask::~TGo4ClientTask()
{
   DisconnectServer();
   delete fxTaskHandler;
}

TGo4TaskHandler *TGo4ClientTask::GetTaskHandler()
{
   return fxTaskHandler;
}

Int_t TGo4ClientTask::Initialization()
{
   // already initialized, return ok value
   if (fbInitDone)
      return 0;

   Int_t rev = TGo4Task::Initialization(); // this will launch threads, etc.
   if (rev == 0) {
      // success: then try to connect (all threads are up now)
      if (fbAutoConnect) {
         if (!ConnectServer(GetServerHostName(), 0))
            Terminate();
      }
   }
   return rev;
}

Bool_t TGo4ClientTask::ConnectServer(const char *node, UInt_t negport,
                            Go4CommandMode_t role,
                            const char *passwd)
{
   if (fbServerConnected) {
      TGo4Log::Warn(" ClientTask::ConnectServer ''%s'': ServerTask already connected", GetName());
      return kTRUE;
   }

   if (negport)
      fxTaskHandler->SetNegotiationPort(negport); // null negport will use the default port
   if (IsMaster())
      fxTaskHandler->ClearQueues(); // do not clear queues on slave side,
                                    // because analysis status might be already in it!

   TGo4Log::Info("TGo4ClientTask::ConnectServer with role %d and passwd %s", role, passwd);
   if (passwd) {
      fxTaskHandler->SetRole(role);
      if (role == kGo4ComModeObserver) {
         TGo4TaskHandler::SetObservAccount(nullptr, passwd);
      } else if (role == kGo4ComModeController) {
         TGo4TaskHandler::SetCtrlAccount(nullptr, passwd);
      } else if (role == kGo4ComModeAdministrator) {
         TGo4TaskHandler::SetAdminAccount(nullptr, passwd);
      }
   }
   if (fxTaskHandler->Connect(node, nullptr)) {
      //      TGo4Log::Info(" ClientTask ''%s'': connected successfully to ServerTask at node %s (port %d) ",
      //               GetName(),node, negport);
      fbServerConnected = kTRUE;
      fxWorkHandler->StartAll(); // all transports are there, then start waiting threads
      SendStatusMessage(1, kTRUE, "Client %s connected successfully to Server task at node %s", GetName(),
                        node); // note that negport does not contain actual portnumber here
      return kTRUE;
   } else {
      TGo4Log::Error(" ClientTask ''%s'': FAILED connection to ServerTask at node %s", GetName(), node);
      return kFALSE;
   }
}

Bool_t TGo4ClientTask::DisconnectServer(Bool_t isterminating)
{
   Bool_t rev = kTRUE;
   if (fbServerConnected) {
      StopWorkThreads();
      if (IsMaster()) {
         //         fxTaskHandler->ClearQueues(); // prevent monitor commands to interfere with disconnect
         TGo4Queue *cq = GetCommandQueue();
         if (cq)
            cq->Clear(); // only clear command queue on master side,
                         // otherwise we lose status messages from server
      }
      ////////UNDER CONSTRUCTION
      WakeCommandQueue(TGo4Task::Get_fgiTERMID()); // will stop local command thread, and remote
      SendStopBuffers(); // note: this should only be done after disconnect login was successful JA
      rev = fxTaskHandler->DisConnect();
      /////////
      if (rev)
         fbServerConnected = kFALSE;
      if (!isterminating)
         StartWorkThreads();
   }
   return rev;
}

void TGo4ClientTask::Quit()
{
   TGo4Log::Debug(" ClientTask''%s'' is quitting... ",GetName());
   SendStatusMessage(2,kTRUE,"ClientTask %s  is terminating...",GetName());
   TGo4Task::Quit();
   DisconnectServer(kTRUE);

   Terminate(!IsMaster()); // never terminate master process
}

TGo4TaskStatus *TGo4ClientTask::CreateStatus()
{
   TGo4ClientStatus* stat= new TGo4ClientStatus(GetName());
   UpdateStatus(stat); // set the internals
   return stat;
}

void TGo4ClientTask::UpdateStatus(TGo4TaskStatus *state)
{
   TGo4Task::UpdateStatus(state);
   TGo4ClientStatus* clstate=dynamic_cast<TGo4ClientStatus*>(state);
   if(clstate)
      clstate->SetNames(GetServerHostName());
}

void TGo4ClientTask::AddLocalCommand(TGo4Command *com)
{
   if (!com)
      return;
   if (fxCommandQ) {
      fxCommandQ->AddBufferFromObject(com);
   } else {
      TGo4Log::Debug(" !!! ClientTask - ERROR adding local command - no command queue !!! ");
   }
}

TGo4BufferQueue *TGo4ClientTask::GetCommandQueue(const char *)
{
   return fxCommandQ;
}

TGo4BufferQueue * TGo4ClientTask::GetStatusQueue(const char *)
{
   return fxStatusQ;
}
TGo4BufferQueue * TGo4ClientTask::GetDataQueue(const char *)
{
   return fxDataQ;
}