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

#pragma once

#include <wx/filename.h>
#include "common/StringHelpers.h"

#define g_MaxPath 255 // 255 is safer with antiquated Win32 ASCII APIs.

// --------------------------------------------------------------------------------------
//  wxDirName
// --------------------------------------------------------------------------------------
class wxDirName : protected wxFileName
{
public:
	explicit wxDirName(const wxFileName& src)
	{
		Assign(src.GetPath(), wxEmptyString);
	}

	wxDirName()
		: wxFileName()
	{
	}
	wxDirName(const wxDirName& src)
		: wxFileName(src)
	{
	}
	explicit wxDirName(const char* src) { Assign(fromUTF8(src)); }
	explicit wxDirName(const wxString& src) { Assign(src); }

	// ------------------------------------------------------------------------
	void Assign(const wxString& volume, const wxString& path)
	{
		wxFileName::Assign(volume, path, wxEmptyString);
	}

	void Assign(const wxString& path)
	{
		wxFileName::Assign(path, wxEmptyString);
	}

	void Assign(const wxDirName& path)
	{
		wxFileName::Assign(path);
	}

	void Clear() { wxFileName::Clear(); }

	wxCharBuffer ToUTF8() const { return GetPath().ToUTF8(); }
	wxCharBuffer ToAscii() const { return GetPath().ToAscii(); }
	wxString ToString() const { return GetPath(); }

	// ------------------------------------------------------------------------
	bool IsWritable() const { return IsDirWritable(); }
	bool IsReadable() const { return IsDirReadable(); }
	bool Exists() const { return DirExists(); }
	bool FileExists() const { return wxFileName::FileExists(); }
	bool IsOk() const { return wxFileName::IsOk(); }
	bool IsRelative() const { return wxFileName::IsRelative(); }
	bool IsAbsolute() const { return wxFileName::IsAbsolute(); }

	bool SameAs(const wxDirName& filepath) const
	{
		return wxFileName::SameAs(filepath);
	}

	//Returns true if the file is somewhere inside this directory (and both file and directory are not relative).
	bool IsContains(const wxFileName& file) const
	{
		if (this->IsRelative() || file.IsRelative())
			return false;

		wxFileName f(file);

		while (1)
		{
			if (this->SameAs(wxDirName(f.GetPath())))
				return true;

			if (f.GetDirCount() == 0)
				return false;

			f.RemoveLastDir();
		}

		return false;
	}

	bool IsContains(const wxDirName& dir) const
	{
		return IsContains((wxFileName)dir);
	}


	//Auto relative works as follows:
	// 1. if either base or subject are relative, return subject (should never be used with relative paths).
	// 2. else if subject is somewhere inside base folder, then result is subject relative to base.
	// 3. (windows only, implicitly) else if subject is on the same driveletter as base, result is absolute path of subject without the driveletter.
	// 4. else, result is absolute path of subject.
	//
	// returns ok if both this and base are absolute paths.
	static wxString MakeAutoRelativeTo(const wxFileName _subject, const wxString& pathbase)
	{
		wxFileName subject(_subject);
		wxDirName base(pathbase);
		if (base.IsRelative() || subject.IsRelative())
			return subject.GetFullPath();

		wxString bv(base.GetVolume());
		bv.MakeUpper();
		wxString sv(subject.GetVolume());
		sv.MakeUpper();

		if (base.IsContains(subject))
		{
			subject.MakeRelativeTo(base.GetFullPath());
		}
		else if (base.HasVolume() && subject.HasVolume() && bv == sv)
		{
			wxString unusedVolume;
			wxString pathSansVolume;
			subject.SplitVolume(subject.GetFullPath(), &unusedVolume, &pathSansVolume);
			subject = pathSansVolume;
		}
		//implicit else: this stays fully absolute

		return subject.GetFullPath();
	}

	static wxString MakeAutoRelativeTo(const wxDirName subject, const wxString& pathbase)
	{
		return MakeAutoRelativeTo(wxFileName(subject), pathbase);
	}

	// Returns the number of sub folders in this directory path
	size_t GetCount() const { return GetDirCount(); }

	// ------------------------------------------------------------------------
	wxFileName Combine(const wxFileName& right) const;
	wxDirName Combine(const wxDirName& right) const;

	// removes the lastmost directory from the path
	void RemoveLast() { wxFileName::RemoveDir(GetCount() - 1); }

	wxDirName& Normalize(int flags = wxPATH_NORM_ALL, const wxString& cwd = wxEmptyString);
	wxDirName& MakeRelativeTo(const wxString& pathBase = wxEmptyString);
	wxDirName& MakeAbsolute(const wxString& cwd = wxEmptyString);

	// ------------------------------------------------------------------------

	void AssignCwd(const wxString& volume = wxEmptyString) { wxFileName::AssignCwd(volume); }
	bool SetCwd() { return wxFileName::SetCwd(); }

	// wxWidgets is missing the const qualifier for this one!  Shame!
	void Rmdir() const;
	bool Mkdir() const;

	// ------------------------------------------------------------------------

	wxDirName& operator=(const wxDirName& dirname)
	{
		Assign(dirname);
		return *this;
	}
	wxDirName& operator=(const wxString& dirname)
	{
		Assign(dirname);
		return *this;
	}
	wxDirName& operator=(const char* dirname)
	{
		Assign(fromUTF8(dirname));
		return *this;
	}

	wxFileName operator+(const wxFileName& right) const { return Combine(right); }
	wxDirName operator+(const wxDirName& right) const { return Combine(right); }
	wxFileName operator+(const wxString& right) const { return Combine(wxFileName(right)); }
	wxFileName operator+(const char* right) const { return Combine(wxFileName(fromUTF8(right))); }

	bool operator==(const wxDirName& filename) const { return SameAs(filename); }
	bool operator!=(const wxDirName& filename) const { return !SameAs(filename); }

	bool operator==(const wxFileName& filename) const { return SameAs(wxDirName(filename)); }
	bool operator!=(const wxFileName& filename) const { return !SameAs(wxDirName(filename)); }

	// compare with a filename string interpreted as a native file name
	bool operator==(const wxString& filename) const { return SameAs(wxDirName(filename)); }
	bool operator!=(const wxString& filename) const { return !SameAs(wxDirName(filename)); }

	const wxFileName& GetFilename() const { return *this; }
	wxFileName& GetFilename() { return *this; }
};

// --------------------------------------------------------------------------------------
//  Path Namespace
// --------------------------------------------------------------------------------------
// Cross-platform utilities for manipulation of paths and filenames.  Mostly these fall
// back on wxWidgets APIs internally, but are still helpful because some of wx's file stuff
// has minor glitches, or requires sloppy wxFileName typecasting.
//
namespace Path
{
	extern bool IsRelative(const wxString& path);
	extern s64 GetFileSize(const wxString& path);

	extern wxString Normalize(const wxString& srcpath);
	extern wxString Normalize(const wxDirName& srcpath);
	extern wxString MakeAbsolute(const wxString& srcpath);

	extern wxString Combine(const wxString& srcPath, const wxString& srcFile);
	extern wxString Combine(const wxDirName& srcPath, const wxFileName& srcFile);
	extern wxString Combine(const wxString& srcPath, const wxDirName& srcFile);
	extern wxString ReplaceExtension(const wxString& src, const wxString& ext);
	extern wxString ReplaceFilename(const wxString& src, const wxString& newfilename);
	extern wxString GetFilename(const wxString& src);
	extern wxString GetDirectory(const wxString& src);
	extern wxString GetFilenameWithoutExt(const wxString& src);
	extern wxString GetRootDirectory(const wxString& src);
} // namespace Path