/////////////////////////////////////////////////////////////////////
// Logger.cpp - log text messages to std::ostream                  //
// ver 1.2                                                         //
//-----------------------------------------------------------------//
// Jim Fawcett (c) copyright 2015                                  //
// All rights granted provided this copyright notice is retained   //
//-----------------------------------------------------------------//
// Language:    C++, Visual Studio 2015                            //
// Application: Several Projects, CSE687 - Object Oriented Design  //
// Author:      Jim Fawcett, Syracuse University, CST 4-187        //
//              jfawcett@twcny.rr.com                              //
/////////////////////////////////////////////////////////////////////

#include <functional>
#include <fstream>
#include <windows.h>
#include "Logger.h"
#include "../Utilities/Utilities.h"

using namespace Logging;

//----< send text message to std::ostream >--------------------------

void Logger::write(const std::string& msg)
{
  if(_ThreadRunning)
    _queue.enQ(msg);
}
void Logger::title(const std::string& msg, char underline)
{
  std::string temp = "\n  " + msg + "\n " + std::string(msg.size() + 2, underline);
  write(temp);
}
//----< attach logger to existing std::ostream >---------------------

void Logger::attach(std::ostream* pOut) 
{ 
  streams_.push_back(pOut);
  //_pOut = pOut; 
}
//----< start logging >----------------------------------------------
/*
 *  log to all the attached streams
 */
void Logger::start()
{
  if (_ThreadRunning)
    return;
  _ThreadRunning = true;
  std::function<void()> tp = [=]() {
    while (true)
    {
      std::string msg = _queue.deQ();
      if (msg == "quit")
      {
        _ThreadRunning = false;
        break;
      }
      for (auto pStrm : streams_)
      {
        *pStrm << msg;
      }
    }
  };
  _pThr = new std::thread (tp);
  //thr.detach();
}
//----< has logger been started? >-----------------------------------

bool Logger::running()
{
  return _ThreadRunning;
}
//----< suspend logger >---------------------------------------------

void Logger::pause(bool doPause)
{
  if (doPause)
  {
    _Paused = true;
    ::SuspendThread(_pThr->native_handle());
  }
  else
  {
    _Paused = false;
    ::ResumeThread(_pThr->native_handle());
  }
}
//----< is logger currently paused? >--------------------------------

bool Logger::paused()
{
  return _Paused;
}
//----< wait for logging to empty input queue >----------------------

void Logger::flush()
{
  if (_ThreadRunning && !_Paused)
  {
    while (_queue.size() > 0)
      ;
    for (auto pStream : streams_)
      pStream->flush();
  }
}
//----< stop logging >-----------------------------------------------

void Logger::stop(const std::string& msg)
{
  if (_ThreadRunning)
  {
    if(msg != "")
      write(msg);
    write("quit");    // request thread to stop
    if (_pThr->joinable())
      _pThr->join();  // wait for queue to empty

    _ThreadRunning = false;
  }
}
//----< wait for logger to finish >----------------------------------

void Logger::wait()
{
  if (_ThreadRunning && _pThr->joinable())
    _pThr->join();
}
//----< stop logging thread >----------------------------------------

Logger::~Logger()
{
  stop();
}

struct Cosmetic
{
  ~Cosmetic() { std::cout << "\n\n"; }
} cosmetic;

#ifdef TEST_LOGGER

using Util = Utilities::StringHelper;

int main()
{
  //Util::Title("Testing Logger Class");
  Logger log;
  log.attach(&std::cout);
  std::ofstream out("logFile.txt");
  if (out.good())
    log.attach(&out);
  else
    std::cout << "\n  couldn't open logFile for writing";
  log.write("\n  won't get logged - not started yet");
  log.start();
  log.title("Testing Logger Class", '=');
  log.write("\n  one");
  log.write("\n  two");
  log.write("\n  fini");
  log.stop();
  log.write("\n  won't get logged - stopped");
  log.start();
  log.write("\n  starting again");
  log.write("\n  and stopping again");
  log.stop("\n  log terminating now");
  log.wait();

  StaticLogger<1>::attach(&std::cout);
  StaticLogger<1>::attach(&out);
  StaticLogger<1>::start();
  StaticLogger<1>::write("\n");
  StaticLogger<1>::title("Testing StaticLogger class");
  StaticLogger<1>::write("\n  static logger at work");
  Logger& logger = StaticLogger<1>::instance();
  logger.write("\n  static logger still at work");

  for(size_t i=0; i<5; ++i)
    logger.write("\n  a log msg");
  logger.write("\n  suspending logger");
  logger.pause(true);
  for (size_t i = 0; i<5; ++i)
    logger.write("\n  a log msg written while log suspended");

  logger.pause(false);
  logger.write("\n  a log msg written after log resumed");
  logger.stop("\n  stopping static logger");
  logger.wait();
  out.close();
}

#endif