aboutsummaryrefslogtreecommitdiffstats
path: root/util/system/tempfile.h
blob: fb8ec62ec623c551b98f59531a9a73eacc261e3d (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
#pragma once

#include "fs.h"
#include "file.h"

#include <util/folder/path.h> 
#include <util/generic/string.h>

class TTempFile {
public:
    inline TTempFile(const TString& fname)
        : Name_(fname)
    {
    }

    inline ~TTempFile() {
        NFs::Remove(Name());
    }

    inline const TString& Name() const noexcept {
        return Name_;
    }

private:
    const TString Name_;
};

class TTempFileHandle: public TTempFile, public TFile {
public:
    TTempFileHandle();
    TTempFileHandle(const TString& fname);

    static TTempFileHandle InCurrentDir(const TString& filePrefix = "yandex", const TString& extension = "tmp"); 
    static TTempFileHandle InDir(const TFsPath& dirPath, const TString& filePrefix = "yandex", const TString& extension = "tmp"); 
 
private:
    TFile CreateFile() const;
};
 
/* 
 * Creates a unique temporary filename in specified directory. 
 * If specified directory is NULL or empty, then system temporary directory is used. 
 * 
 * Note, that the function is not race-free, the file is guaranteed to exist at the time the function returns, but not at the time the returned name is first used. 
 * Throws TSystemError on error. 
 *  
 * Returned filepath has such format: dir/prefixXXXXXX.extension or dir/prefixXXXXXX 
 * But win32: dir/preXXXX.tmp (prefix is up to 3 characters, extension is always tmp). 
 */ 
TString MakeTempName(const char* wrkDir = nullptr, const char* prefix = "yandex", const char* extension = "tmp");