Subversion Repositories spk

#pragma once

#include "VirtualFileSystem.h"

#include "lists.h"

using namespace SPK;

class CCatFile;

enum {DIFFTYPE_ADDITION, DIFFTYPE_REMOVAL, DIFFTYPE_CHANGE};

tstruct SDiffEntry {

	int			iType;

	int			iID;

} SDiffEntry;

tstruct SDiffEntryAddition : SDiffEntry {

	Utils::WString	sEntry;

	SDiffEntryAddition() { iType = DIFFTYPE_ADDITION; }

} SDiffEntryAddition;

tstruct SDiffEntryRemoval : SDiffEntry {

	SDiffEntryRemoval() { iType = DIFFTYPE_REMOVAL; }

} SDiffEntryRemoval;

tstruct SDiffEntryChange : SDiffEntry {

	SDiffEntryChange() : iPos(0) { iType = DIFFTYPE_CHANGE; }

	int			iPos;

	Utils::WString	sFrom;

	Utils::WString	sEntry;

} SDiffEntryChange;

tstruct SDiffFile {

	Utils::WString sFile;

	CLinkList<SDiffEntry> m_lEntries;

} SDiffFile;

enum {MDERR_NONE, MDERR_FILENOTFOUND, MDERR_CANTOPEN, MDERR_CANTOPENMOD};

tclass CModDiff

{

private:

	enum {

		MERGETYPE_NONE,

		MERGETYPE_TSHIPS,

	};

public:

	// static

	static bool CanBeDiffed(const Utils::WString &file);

	// constructors

	CModDiff(const Utils::WString &dir, const Utils::WString &sAddon, int maxPatch = 0);

	~CModDiff(void);

	// public functions

	void SetTempDirectory(const Utils::WString &temp) { m_sTempDir = temp; }

	bool LoadDirectory(const Utils::WString &dir);

	bool IsLoaded() { return m_bLoaded; }

	const Utils::WString &GetDirectory() const { return m_sCurrentDir; }

	bool CreateDiff(const Utils::WString &mod);

	bool startDiff(const Utils::WString &sModFile);

	bool doDiff(const Utils::WString &sModFile);

	SDiffFile *diffFile(const Utils::WString &baseFile, const  Utils::WString &modFile, const Utils::WString &fileType);

	void Clean();

	bool WriteDiff(const Utils::WString &file);

	bool ReadDiff(const Utils::WString &file);

	bool ApplyDiff(const Utils::WString &mod);

	bool ApplyMod(const Utils::WString &mod);

	void SetMaxPatch(int patch) { m_iMaxPatch = patch; }

	CLinkList<SDiffFile> &GetDiffFiles() { return m_lFiles; }

	void ClearError() { m_iError = MDERR_NONE; }

	int Error() { return m_iError; }

private:

	// private functions

	int _amountPosition(const Utils::WString &fileType);

	bool _isLineComplete(const Utils::WString &line, const Utils::WString &fileType, bool first);

	bool _readGameFile(const Utils::WString &file, Utils::WStringList &lines, int *id);

	void _compareLine(const Utils::WString &line1, const Utils::WString &line2, int type, int id, SDiffFile *diffFile);

	bool _validFile(const Utils::WString &file);

	void _adjustFile(const Utils::WString &sFile, SDiffFile *pDiff, bool bReverse);

	bool _adjustTShips(SDiffFile *pDiff, bool bReserve);

	int _specialType(const Utils::WString &sFile) const;

	Utils::WString _extractFile(const Utils::WString &sFile, const Utils::WString &sTo);

	// Variables

	Utils::WString m_sAddon;

	Utils::WString m_sCurrentDir;		// the current game directory (that the VFS is opened too)

	Utils::WString m_sTempDir;				// temporary dir (used to write temporary files to)

	CVirtualFileSystem m_fileSystem;	// the VFS of the game directory (for finding the files to use)

	CLinkList<SDiffFile> m_lFiles;		// list of files that have changed

	bool	m_bLoaded;					// if the directory is loaded and ready to be diffed

	int		m_iError;					// error id of process

	int		m_iMaxPatch;				// The max fake patch to check to

	CCatFile		*m_pCatFile;

};