/*  PCSX2 - PS2 Emulator for PCs
 *  Copyright (C) 2002-2010  PCSX2 Dev Team
 *
 *  PCSX2 is free software: you can redistribute it and/or modify it under the terms
 *  of the GNU Lesser General Public License as published by the Free Software Found-
 *  ation, either version 3 of the License, or (at your option) any later version.
 *
 *  PCSX2 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 PCSX2.
 *  If not, see <http://www.gnu.org/licenses/>.
 */

#include <errno.h>
#include <wx/stream.h>

#include "common/wxBaseTools.h"
#include "common/pxStreams.h"
#include "common/StringHelpers.h"

// --------------------------------------------------------------------------------------
//  pxStreamBase  (implementations)
// --------------------------------------------------------------------------------------
pxStreamBase::pxStreamBase(const wxString& filename)
	: m_filename(filename)
{
}

bool pxStreamBase::IsOk() const
{
	wxStreamBase* woot = GetWxStreamBase();
	return woot && woot->IsOk();
}

wxFileOffset pxStreamBase::Length() const
{
	if (!GetWxStreamBase())
		return 0;
	return GetWxStreamBase()->GetLength();
}

// --------------------------------------------------------------------------------------
//  pxInputStream  (implementations)
// --------------------------------------------------------------------------------------
// Interface for reading data from a gzip stream.
//

pxInputStream::pxInputStream(const wxString& filename, std::unique_ptr<wxInputStream>& input)
	: pxStreamBase(filename)
	, m_stream_in(std::move(input))
{
}

pxInputStream::pxInputStream(const wxString& filename, wxInputStream* input)
	: pxStreamBase(filename)
	, m_stream_in(input)
{
}

wxStreamBase* pxInputStream::GetWxStreamBase() const { return m_stream_in.get(); }

wxFileOffset pxInputStream::Tell() const
{
	return m_stream_in->TellI();
}

wxFileOffset pxInputStream::Seek(wxFileOffset ofs, wxSeekMode mode)
{
	return m_stream_in->SeekI(ofs, mode);
}

void pxInputStream::SetStream(const wxString& filename, std::unique_ptr<wxInputStream>& stream)
{
	m_filename = filename;
	m_stream_in = std::move(stream);
}

void pxInputStream::SetStream(const wxString& filename, wxInputStream* stream)
{
	m_filename = filename;
	m_stream_in = std::unique_ptr<wxInputStream>(stream);
}

void pxInputStream::Read(void* dest, size_t size)
{
	m_stream_in->Read(dest, size);
	if (m_stream_in->GetLastError() == wxSTREAM_READ_ERROR)
	{
		int err = errno;
		if (!err)
			throw Exception::BadStream(m_filename).SetDiagMsg(L"Cannot read from file (bad file handle?)");

		ScopedExcept ex(Exception::FromErrno(m_filename, err));
		ex->SetDiagMsg(L"cannot read from file: " + ex->DiagMsg());
		ex->Rethrow();
	}

	// IMPORTANT!  The underlying file/source Eof() stuff is not really reliable, so we
	// must always use the explicit check against the number of bytes read to determine
	// end-of-stream conditions.

	if ((size_t)m_stream_in->LastRead() < size)
		throw Exception::EndOfStream(m_filename);
}

// --------------------------------------------------------------------------------------
//  pxOutputStream
// --------------------------------------------------------------------------------------
pxOutputStream::pxOutputStream(const wxString& filename, std::unique_ptr<wxOutputStream>& output)
	: pxStreamBase(filename)
	, m_stream_out(std::move(output))
{
}

pxOutputStream::pxOutputStream(const wxString& filename, wxOutputStream* output)
	: pxStreamBase(filename)
	, m_stream_out(output)
{
}

wxStreamBase* pxOutputStream::GetWxStreamBase() const { return m_stream_out.get(); }

wxFileOffset pxOutputStream::Tell() const
{
	return m_stream_out->TellO();
}

wxFileOffset pxOutputStream::Seek(wxFileOffset ofs, wxSeekMode mode)
{
	return m_stream_out->SeekO(ofs, mode);
}

void pxOutputStream::SetStream(const wxString& filename, std::unique_ptr<wxOutputStream>& stream)
{
	m_filename = filename;
	m_stream_out = std::move(stream);
}

void pxOutputStream::SetStream(const wxString& filename, wxOutputStream* stream)
{
	m_filename = filename;
	m_stream_out = std::unique_ptr<wxOutputStream>(stream);
}


void pxOutputStream::Write(const void* src, size_t size)
{
	m_stream_out->Write(src, size);
	if (m_stream_out->GetLastError() == wxSTREAM_WRITE_ERROR)
	{
		int err = errno;
		if (!err)
			throw Exception::BadStream(m_filename).SetDiagMsg(L"Cannot write to file/stream.");

		ScopedExcept ex(Exception::FromErrno(m_filename, err));
		ex->SetDiagMsg(L"Cannot write to file: " + ex->DiagMsg());
		ex->Rethrow();
	}
}

// --------------------------------------------------------------------------------------
//  pxTextStream
// --------------------------------------------------------------------------------------

// Returns TRUE if the source is UTF8, or FALSE if it's just ASCII crap.
bool pxReadLine(wxInputStream& input, std::string& dest)
{
	dest.clear();
	bool isUTF8 = false;
	while (true)
	{
		char c;
		input.Read(&c, sizeof(c));
		if (c == 0)
			break;
		if (input.Eof())
			break;
		if (c == '\n')
			break; // eat on UNIX
		if (c == '\r')
		{
			input.Read(&c, sizeof(c));
			if (c == 0)
				break;
			if (input.Eof())
				break;
			if (c == '\n')
				break;

			input.Ungetch(c);
			break;
		}
		dest += c;
		if (c & 0x80)
			isUTF8 = true;
	}

	return isUTF8;
}

void pxReadLine(wxInputStream& input, wxString& dest, std::string& intermed)
{
	dest.clear();
	if (pxReadLine(input, intermed))
		dest = fromUTF8(intermed.c_str());
	else
	{
		// Optimized ToAscii conversion.
		// wx3.0 : NOT COMPATIBLE!!  (on linux anyway)
		const char* ascii = intermed.c_str();
		while (*ascii != 0)
			dest += (wchar_t)(unsigned char)*ascii++;
	}
}

void pxReadLine(wxInputStream& input, wxString& dest)
{
	std::string line;
	pxReadLine(input, dest, line);
}

wxString pxReadLine(wxInputStream& input)
{
	wxString result;
	pxReadLine(input, result);
	return result;
}

void pxWriteLine(wxOutputStream& output)
{
	output.Write("\n", 1);
}

void pxWriteLine(wxOutputStream& output, const wxString& text)
{
	if (!text.IsEmpty())
	{
		pxToUTF8 utf8(text);
		output.Write(utf8, utf8.Length());
	}
	pxWriteLine(output);
}

void pxWriteMultiline(wxOutputStream& output, const wxString& src)
{
	if (src.IsEmpty())
		return;

	wxString result(src);
	result.Replace(L"\r\n", L"\n");
	result.Replace(L"\r", L"\n");

	pxToUTF8 utf8(result);
	output.Write(utf8, utf8.Length());
}