summaryrefslogtreecommitdiff
path: root/replay/sv_fileservercleanup.h
blob: 65939e0bebf6aee15ab9f6861c4a60fe90c964c7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
//========= Copyright Valve Corporation, All rights reserved. ============//
//
//=======================================================================================//

#ifndef SV_FILESERVERCLEANUP_H
#define SV_FILESERVERCLEANUP_H
#ifdef _WIN32
#pragma once
#endif

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

#include "basethinker.h"
#include "spew.h"
#include "sv_basejob.h"

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

bool		SV_DoFileserverCleanup( bool bForceCleanAll, ISpewer *pSpewer/*=g_pDefaultSpewer*/ );
CBaseJob *SV_CreateDeleteFileJob();

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

class IFileserverCleanerJob
{
public:
	virtual ~IFileserverCleanerJob() {}

	virtual void		AddFileForDelete( const char *pFilename ) = 0;
	virtual int			GetNumFilesDeleted() const = 0;
};

IFileserverCleanerJob *SV_CastJobToIFileserverCleanerJob( CBaseJob *pJob );

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

class CFileserverCleaner : public CBaseThinker
{
public:
	CFileserverCleaner();

	void				MarkFileForDelete( const char *pFilename );

	int					GetNumFilesDeleted() const		{ return m_nNumFilesDeleted; }
	bool				HasFilesQueuedForDelete() const	{ return m_pCleanerJob != NULL; }

	void				BlockForCompletion();
	void				DoCleanAsynchronous( bool bPrintResult = false, ISpewer *pSpewer = g_pDefaultSpewer );

private:
	void				Clear();
	void				PrintResult();

	virtual void		Think();
	virtual float		GetNextThinkTime() const;

	CBaseJob			*m_pCleanerJob;
	bool				m_bRunning;
	bool				m_bPrintResult;
	int					m_nNumFilesDeleted;
	ISpewer				*m_pSpewer;
};

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

class CLocalFileDeleterJob : public CBaseJob,
							 public IFileserverCleanerJob
{
public:
	CLocalFileDeleterJob();

	virtual void		AddFileForDelete( const char *pFilename );
	virtual int			GetNumFilesDeleted() const	{ return m_nNumDeleted; }

	enum DeleteError_t
	{
		ERROR_FILE_DOES_NOT_EXIST,
	};

private:
	virtual JobStatus_t	DoExecute();

	CUtlStringList		m_vecFiles;
	int					m_nNumDeleted;
};

CLocalFileDeleterJob *SV_CreateLocalFileDeleterJob();

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

#endif	// SV_FILESERVERCLEANUP_H