tools/rorescompiler/main.cpp


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

#include <library/cpp/resource/registry.h>
 
#include <util/digest/city.h> 
#include <util/stream/output.h> 
#include <util/stream/file.h> 
#include <util/string/escape.h>
#include <util/string/vector.h> 
#include <util/string/split.h>
 
using namespace NResource; 
 
void GenOne(const TString& raw, const TString& key, IOutputStream& out) { 
    TString size = raw + "Size";
    TString name = ToString(CityHash64(key.data(), key.size()));
    out << "extern \"C\" const char " << raw << "[];\n"
        << "extern \"C\" const unsigned int " << size << ";\n"
        << "static const NResource::TRegHelper REG_name" << name
        << "(\"" << EscapeC(key) << "\", TStringBuf(" << raw << ", " << size << "));\n"
        << "\n";
}; 
 
int main(int argc, char** argv) { 
    if (argc < 3) { 
        Cerr << "usage: " << argv[0] << " outfile [key=value]+" << Endl; 
 
        return 1; 
    } 
 
    TFixedBufferFileOutput out(argv[1]);
 
    argv = argv + 2; 
    out << "#include <library/cpp/resource/registry.h>\n\n";
 
    while (*argv) { 
        TVector<TString> items = StringSplitter(TString(*(argv))).Split('=').Limit(2).ToList<TString>();
        GenOne(items[0], items[1], out); 
        argv++; 
    } 
}