CWE-ou-minkata/Sources/Plasma/FeatureLib/pfPython/pyStream.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==*/
//////////////////////////////////////////////////////////////////////
//
// pyStream   - a wrapper class to provide interface to the File stream stuff
//
//////////////////////////////////////////////////////////////////////

#include "pyStream.h"

#include "plFile/plEncryptedStream.h"

pyStream::pyStream()
: fStream( nil )
{
}

pyStream::~pyStream()
{
    Close();
}


hsBool pyStream::Open(const wchar* fileName, const wchar* flags)
{
    // make sure its closed first
    Close();

    if (fileName)
    {
        if (flags)
        {
            bool readflag = false;
            bool writeflag = false;
            bool encryptflag = false;
            int i;
            for (i=0 ; i < wcslen(flags) ; i++ )
            {
                if ( flags[i] == L'r' || flags[i] == L'R' )
                    readflag = true;
                if ( flags[i] == L'w' || flags[i] == L'W' )
                    writeflag = true;
                if ( flags[i] == L'e' || flags[i] == L'E' )
                    encryptflag = true;
            }
            // if there is a write flag, it takes priorty over read
            if (writeflag)
            {
                // force encryption?
                if (encryptflag)
                {
                    fStream = TRACKED_NEW plEncryptedStream;
                    fStream->Open(fileName, L"wb");
                }
                else
                    fStream = plEncryptedStream::OpenEncryptedFileWrite(fileName);
            }
            else
                fStream = plEncryptedStream::OpenEncryptedFile(fileName);
            return true;
        }
    }
    return false;
}

std::vector<std::string> pyStream::ReadLines()
{
    
    // read all the lines in the file and put in a python list object
    // create the list
    std::vector<std::string> pyPL;

    if (fStream)
    {
        char buf[4096];
        
        while (!fStream->AtEnd())
        {
            if (fStream->ReadLn(buf, sizeof(buf), 0, 0))
                pyPL.push_back(buf);
        }
    }

    return pyPL;
}

hsBool pyStream::WriteLines(const std::vector<std::string> & lines)
{
    if (fStream)
    {
        int i;
        for ( i=0 ; i<lines.size() ; i++ )
        {
            std::string element = lines[i];
            fStream->Write(element.length(),element.c_str());
        }
        return true;
    }

    return false;
}


void pyStream::Close()
{
    if (fStream)
    {
        fStream->Close();
        delete fStream;
    }
    fStream = nil;
}

hsBool pyStream::IsOpen()
{
    if (fStream)
        return true;
    return false;
}