#pragma once

#ifndef TFARMSERVER_H
#define TFARMSERVER_H

#include <memory>

#include "tcommon.h"

#include <QString>

#include "tfarmplatforms.h"

//------------------------------------------------------------------------------

#ifdef TFARMAPI
#undef TFARMAPI
#endif

#ifdef _WIN32
#ifdef TFARM_EXPORTS
#define TFARMAPI __declspec(dllexport)
#else
#define TFARMAPI __declspec(dllimport)
#endif
#else
#define TFARMAPI
#endif

//------------------------------------------------------------------------------

class TFARMAPI TFarmServer {
public:
  class HwInfo {
  public:
    HwInfo()
        : m_cpuCount(0)
        , m_totPhysMem(0)
        , m_availPhysMem(0)
        , m_totVirtMem(0)
        , m_availVirtMem(0)
        , m_type(NoPlatform) {}

    int m_cpuCount;
    unsigned int m_totPhysMem;
    unsigned int m_availPhysMem;
    unsigned int m_totVirtMem;
    unsigned int m_availVirtMem;
    TFarmPlatform m_type;
  };

  virtual ~TFarmServer() {}

  virtual int addTask(const QString &taskid, const QString &cmdline) = 0;
  virtual int terminateTask(const QString &taskid)  = 0;
  virtual int getTasks(std::vector<QString> &tasks) = 0;

  virtual void queryHwInfo(HwInfo &hwInfo) = 0;

  // used (by a controller) to notify a controller start
  virtual void attachController(const QString &name, const QString &addr,
                                int port) = 0;

  // used (by a controller) to notify a controller stop
  virtual void detachController(const QString &name, const QString &addr,
                                int port) = 0;
};

//------------------------------------------------------------------------------

class TFARMAPI TFarmServerFactory {
public:
  TFarmServerFactory();
  ~TFarmServerFactory();

  int create(const QString &hostName, const QString &addr, int port,
             TFarmServer **tfserver);
};

//------------------------------------------------------------------------------

class TFARMAPI TFarmServerStub {
public:
  TFarmServerStub(TFarmServer *farmServer, int port);
  ~TFarmServerStub();

  int run();
  int shutdown();

  int getPort() const;

private:
  class Imp;
  std::unique_ptr<Imp> m_imp;
};

#endif