gcomp/Storage.h - master - MadDAD/TFS4OpenConf

gcomp/Storage.h
132 строки4 KB
DmitryDreytser
Выгрузил исходники
01 фев 2017, 17:19
01 фев 2017, 17:19b9402b1
#ifndef STORAGE_H
#define STORAGE_H


#include "Declarations.h"
#include "StreamInfo.h"

#import "zlibeng.dll" no_namespace

void SetIOPre(LPCTSTR MsgPrefix, LPCTSTR FileName);

class CStorage
{
private:
	static IzlibEnginePtr iLibEngine;
public:
	static void InitZLibEngine();
	static void ReleaseZLibEngine();

protected:
	CList<IStorage*, IStorage*&> Storages;
	bool IsRootStorage;
	void Push(IStorage* pStorage);
	IStorage* Pop();

	//���-������� ������ � ��������, ����������� � ������� ������. 
	//����� ��� ����������� ������� �������� �� ������ ������
	CMapStringToPtr* ExtractedFiles;
	//���. ���-������� ��������. � �������� ������������ ����� �������� ��������� �������� ��������
	CMapPtrToPtr*    ExtractedObjects;

public:
	IStream* pStream;
	GCString StreamName;
	CStreamInfo StreamInfo;

private:
	void SetParentStorage(CStorage& Storage);

public:
	CStorage();
	CStorage(CStorage& Storage);
	//� �������� ���� ����������� ���������� � ������� �������, ������� ��� SubStorage
	//�� ����� ������ ��������� ��� - ����� ����������� ������ ��� ����������
	CStorage(CStorage& Storage, GCString SubStorage);
	~CStorage();

public:
	GCString FullPath();

	bool OpenCompound(GCString& CompoundFileName);
	bool Open(CStringArray* StorageName);
	bool Open(GCString& StorageName);
	bool CreateCompound(GCString& CompoundFileName);
	bool Create(CStringArray* StorageName);
	bool Create(GCString& StorageName);
	bool Delete(GCString& StorageName);
	void Close(bool CloseAll = false);

	bool OpenStream(GCString& StreamName);
	bool CreateStream(GCString& StreamName);
	void CloseStream();

	bool StorageExist(GCString& StorageName);
	bool StreamExist(GCString& StreamName);
	long GetStreamSize(GCString& StreamName);

	void Rewind();

	void GetCleanStream();
	bool CopyToFile(FILE* File, unsigned long bytes_to_copy);
	bool CopyToFile(GCString& FileName, bool OnlyRestOfSream = false);
	bool CopyToFile(GCString& StreamName, GCString& FileName);
	bool Extract(GCString& DestDir, bool WithContainerContents = true);
	bool CopyStringToFile(GCString& String, GCString& FileName);

	bool StreamFromFile(GCString& StreamName, GCString& FileName);
	bool StreamFromString(GCString StreamName, GCString& String);

	unsigned long AppendFile(IStream* pStream, FILE* File);
	unsigned long AppendFile(IStream* pStream, GCString& FileName);
	void AddZeroByte();

	bool FromFile(GCString& FileName);
	bool FromFile(FILE* File, LPCSTR FileName = NULL);
	bool StorageFromDir(GCString& StorageName, GCString& SrcDir);

	//���������� ������� ����� � ��������� ������. ��������� ���������� � 
	//������� ������� (��� � ������, ��� � � �����). ���� ������ bytes_to_compare > 0, 
	//�� ���������� ������ ��������� ���������� ����. ���� �� ������ - ���������� �� 
	//����� ������ ��� �����.
	int Compare(FILE* File, unsigned long bytes_to_compare);
	int Compare(GCString& FileName, unsigned long bytes_to_compare);
	int CompareWithString(GCString& str2);
	//��������� ��������� ������ ������� (����� ���������� ���������� ��������������������)
	int FullCompare(GCString& FileName);

	IStorage* GetStorage();

	void ParseStream(CMMSObject* Object, GCString& StreamName, bool ParseWitnNoIDs);

	void AddToExtractedList(GCString& FullFName, void* object);
	bool ObjectExtracted(void* object);
	int PruneDirectory(GCString& Path, bool NoEmptyFolders);

private:
	void ReleaseMaps();
	void WriteSizeToStream(IStream* pStream);
	bool ExtractPictureGallery(GCString& FileName);
	
	bool GalleryImported; //���� - ����� �� ������������� Gallery 3 ����
	bool ImportPictureGallery(GCString& StreamName, GCString& FileName);

	void RemoveDirectory(GCString& Path);
};



LPCSTR GetStgErrorString(HRESULT err);

ULARGE_INTEGER GetStreamSize(IStream* pStream);

IStorage* OpenStorage(IStorage* pParentStorage, wchar_t* wStorName);
IStorage* OpenStorage(IStorage* pParentStorage, GCString& StorName);

IStream* OpenStream(IStorage* pParentStorage, wchar_t* wStreamName);
IStream* OpenStream(IStorage* pParentStorage, GCString& StreamName);

bool StringDifferFromFile(GCString& String, GCString& FileName);
bool CopyStringToFile(GCString& String, GCString& FileName);

#endif