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
|
#include <yql/essentials/sql/v1/format/sql_format.h>
#include <library/cpp/getopt/last_getopt.h>
#include <google/protobuf/arena.h>
#include <util/stream/file.h>
int RunFormat(int argc, char* argv[]) {
NLastGetopt::TOpts opts = NLastGetopt::TOpts::Default();
TString outFileName;
TString inFileName;
TString queryString;
opts.AddLongOption('o', "output", "save output to file").RequiredArgument("file").StoreResult(&outFileName);
opts.AddLongOption('i', "input", "input file").RequiredArgument("input").StoreResult(&inFileName);
opts.AddLongOption('p', "print-query", "print given query before parsing").NoArgument();
opts.AddLongOption('f', "obfuscate", "obfuscate query").NoArgument();
opts.AddLongOption("ansi-lexer", "use ansi lexer").NoArgument();
opts.AddHelpOption();
NLastGetopt::TOptsParseResult res(&opts, argc, argv);
THolder<TFixedBufferFileOutput> outFile;
if (!outFileName.empty()) {
outFile.Reset(new TFixedBufferFileOutput(outFileName));
}
IOutputStream& out = outFile ? *outFile.Get() : Cout;
THolder<TUnbufferedFileInput> inFile;
if (!inFileName.empty()) {
inFile.Reset(new TUnbufferedFileInput(inFileName));
}
IInputStream& in = inFile ? *inFile.Get() : Cin;
queryString = in.ReadAll();
int errors = 0;
TString queryFile("query");
if (res.Has("print-query")) {
out << queryString << Endl;
}
google::protobuf::Arena arena;
NSQLTranslation::TTranslationSettings settings;
settings.Arena = &arena;
settings.AnsiLexer = res.Has("ansi-lexer");
auto formatter = NSQLFormat::MakeSqlFormatter(settings);
TString frm_query;
TString error;
NYql::TIssues issues;
if (!formatter->Format(queryString, frm_query, issues, res.Has("obfuscate") ?
NSQLFormat::EFormatMode::Obfuscate : NSQLFormat::EFormatMode::Pretty)) {
++errors;
Cerr << "Error formatting query: " << issues.ToString() << Endl;
} else {
out << frm_query << Endl;
}
return errors;
}
int main(int argc, char* argv[]) {
try {
return RunFormat(argc, argv);
} catch (const yexception& e) {
Cerr << "Caught exception:" << e.what() << Endl;
return 1;
} catch (...) {
Cerr << CurrentExceptionMessage() << Endl;
return 1;
}
return 0;
}
|
