/******************************************************************************\

             CAMotics is an Open-Source simulation and CAM software.
     Copyright (C) 2011-2021 Joseph Coffland <joseph@cauldrondevelopment.com>

       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 2 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 "PyLogger.h"
#include "Throw.h"

#include <cbang/log/Logger.h>

using namespace std;


PyLogger::PyLogger(PyObject *cb) : cb(cb) {
  Py_INCREF(cb);
  if (!PyCallable_Check(cb)) THROW("logger object not callable");
}


PyLogger::PyLogger(const PyLogger &o) : cb(o.cb) {Py_INCREF(cb);}


PyLogger::~PyLogger() {Py_DECREF(cb);}


PyLogger &PyLogger::operator=(const PyLogger &o) {
  cb = o.cb;
  Py_INCREF(cb);
  return *this;
}


streamsize PyLogger::write(const char *s, streamsize n) {
  // Args
  PyObject *args = PyTuple_New(1);
  if (!args) THROW("Failed to allocate tuple");

  PyTuple_SetItem(args, 0, PyUnicode_FromStringAndSize(s, n));

  // Call
  PyObject *result = PyObject_Call(cb, args, 0);
  Py_DECREF(args);

  if (!result) THROW("Logger callback failed");
  Py_DECREF(result);

  PyThrowIfError("Logger callback failed: ");

  return n;
}