CWE-ou-minkata/Sources/Plasma/FeatureLib/pfPython/pyStreamGlue.cpp

/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

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/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
#include "pyStream.h"

#include <Python.h>

// glue functions
PYTHON_CLASS_DEFINITION(ptStream, pyStream);

PYTHON_DEFAULT_NEW_DEFINITION(ptStream, pyStream)
PYTHON_DEFAULT_DEALLOC_DEFINITION(ptStream)

PYTHON_INIT_DEFINITION(ptStream, args, keywords)
{
    PYTHON_RETURN_INIT_OK;
}

PYTHON_METHOD_DEFINITION(ptStream, open, args)
{
    PyObject* filenameObj;
    PyObject* flagsObj;
    if (!PyArg_ParseTuple(args, "OO", &filenameObj, &flagsObj))
    {
        PyErr_SetString(PyExc_TypeError, "open expects two strings");
        PYTHON_RETURN_ERROR;
    }

    std::wstring filename;
    if (PyUnicode_Check(filenameObj))
    {
        int strLen = PyUnicode_GetSize(filenameObj);
        wchar_t* text = TRACKED_NEW wchar_t[strLen + 1];
        PyUnicode_AsWideChar((PyUnicodeObject*)filenameObj, text, strLen);
        text[strLen] = L'\0';
        filename = text;
        delete [] text;
    }
    else if (PyString_Check(filenameObj))
    {
        // we'll allow this, just in case something goes weird
        char* text = PyString_AsString(filenameObj);
        wchar_t* wText = hsStringToWString(text);
        filename = wText;
        delete [] wText;
    }
    else
    {
        PyErr_SetString(PyExc_TypeError, "open expects two strings");
        PYTHON_RETURN_ERROR;
    }

    std::wstring flags;
    if (PyUnicode_Check(flagsObj))
    {
        int strLen = PyUnicode_GetSize(flagsObj);
        wchar_t* text = TRACKED_NEW wchar_t[strLen + 1];
        PyUnicode_AsWideChar((PyUnicodeObject*)flagsObj, text, strLen);
        text[strLen] = L'\0';
        flags = text;
        delete [] text;
    }
    else if (PyString_Check(flagsObj))
    {
        // we'll allow this, just in case something goes weird
        char* text = PyString_AsString(flagsObj);
        wchar_t* wText = hsStringToWString(text);
        flags = wText;
        delete [] wText;
    }
    else
    {
        PyErr_SetString(PyExc_TypeError, "open expects two strings");
        PYTHON_RETURN_ERROR;
    }

    PYTHON_RETURN_BOOL(self->fThis->Open(filename.c_str(), flags.c_str()));
}

PYTHON_METHOD_DEFINITION_NOARGS(ptStream, readlines)
{
    std::vector<std::string> lines = self->fThis->ReadLines();
    PyObject* retVal = PyList_New(lines.size());
    for (int i = 0; i < lines.size(); i++)
        PyList_SetItem(retVal, i, PyString_FromString(lines[i].c_str()));
    return retVal;
}

PYTHON_METHOD_DEFINITION(ptStream, writelines, args)
{
    PyObject* stringList = NULL;
    if (!PyArg_ParseTuple(args, "O", &stringList))
    {
        PyErr_SetString(PyExc_TypeError, "writelines expects a list of strings");
        PYTHON_RETURN_ERROR;
    }
    if (!PyList_Check(stringList))
    {
        PyErr_SetString(PyExc_TypeError, "writelines expects a list of strings");
        PYTHON_RETURN_ERROR;
    }
    std::vector<std::string> strings;
    int len = PyList_Size(stringList);
    for (int i = 0; i < len; i++)
    {
        PyObject* element = PyList_GetItem(stringList, i);
        if (!PyString_Check(element))
        {
            PyErr_SetString(PyExc_TypeError, "writelines expects a list of strings");
            PYTHON_RETURN_ERROR;
        }
        strings.push_back(PyString_AsString(element));
    }
    PYTHON_RETURN_BOOL(self->fThis->WriteLines(strings));
}

PYTHON_BASIC_METHOD_DEFINITION(ptStream, close, Close)

PYTHON_METHOD_DEFINITION_NOARGS(ptStream, isOpen)
{
    PYTHON_RETURN_BOOL(self->fThis->IsOpen());
}

PYTHON_START_METHODS_TABLE(ptStream)
    PYTHON_METHOD(ptStream, open, "Params: fileName,flags\nOpen a stream file for reading or writing"),
    PYTHON_METHOD_NOARGS(ptStream, readlines, "Reads a list of strings from the file"),
    PYTHON_METHOD(ptStream, writelines, "Params: lines\nWrite a list of strings to the file"),
    PYTHON_BASIC_METHOD(ptStream, close, "Close the status log file"),
    PYTHON_METHOD_NOARGS(ptStream, isOpen, "Returns whether the stream file is currently opened"),
PYTHON_END_METHODS_TABLE;

// Type structure definition
PLASMA_DEFAULT_TYPE(ptStream, "A basic stream class");

// required functions for PyObject interoperability
PYTHON_CLASS_NEW_IMPL(ptStream, pyStream)

PYTHON_CLASS_CHECK_IMPL(ptStream, pyStream)
PYTHON_CLASS_CONVERT_FROM_IMPL(ptStream, pyStream)

///////////////////////////////////////////////////////////////////////////
//
// AddPlasmaClasses - the python module definitions
//
void pyStream::AddPlasmaClasses(PyObject *m)
{
    PYTHON_CLASS_IMPORT_START(m);
    PYTHON_CLASS_IMPORT(m, ptStream);
    PYTHON_CLASS_IMPORT_END(m);
}