private static void Generate(Express.ExpressListener listener, string outDir,
ILanguageGenerator generator, IFunctionsGenerator functionsGenerator)
{
var names = new List <string>();
var sd = listener.TypeData.Where(kvp => kvp.Value is SelectType).
Select(v => new { v.Key, v.Value }).
ToDictionary(t => t.Key, t => (SelectType)t.Value);
generator.SelectData = sd;
var ed = listener.TypeData.Where(kvp => kvp.Value is EnumType).
Select(v => new { v.Key, v.Value }).
ToDictionary(t => t.Key, t => (EnumType)t.Value);
generator.EnumData = ed;
foreach (var kvp in listener.TypeData)
{
var td = kvp.Value;
File.WriteAllText(Path.Combine(outDir, $"{td.Name}.{generator.FileExtension}"), td.ToString());
names.Add(td.Name);
}
generator.GenerateManifest(outDir, names);
if (functionsGenerator != null)
{
functionsGenerator.SelectData = sd;
var functionsPath = Path.Combine(outDir, functionsGenerator.FileName);
File.WriteAllText(functionsPath, functionsGenerator.Generate(listener.FunctionData.Values));
}
}
static void Main(string[] args)
{
ParseOptions(args);
if (showHelp)
{
return;
}
var generators = new List <Tuple <ILanguageGenerator, IFunctionsGenerator> >();
if (language == "csharp")
{
generators.Add(new Tuple <ILanguageGenerator, IFunctionsGenerator>(
new CsharpLanguageGenerator(), new CsharpFunctionsGenerator()));
}
else if (language == "proto")
{
generators.Add(new Tuple <ILanguageGenerator, IFunctionsGenerator>(new ProtobufGenerator(), null));
}
else if (language == "ts")
{
generators.Add(new Tuple <ILanguageGenerator, IFunctionsGenerator>(new TypescriptGenerator(),
new TypescriptFunctionsGenerator()));
}
using (FileStream fs = new FileStream(expressPath, FileMode.Open))
{
var input = new AntlrInputStream(fs);
var lexer = new Express.ExpressLexer(input);
var tokens = new CommonTokenStream(lexer);
var parser = new Express.ExpressParser(tokens);
parser.BuildParseTree = true;
var tree = parser.schemaDecl();
var walker = new ParseTreeWalker();
var testSb = new StringBuilder();
foreach (var generator in generators)
{
var listener = new Express.ExpressListener(generator.Item1);
walker.Walk(listener, tree);
Generate(listener, outDir, generator.Item1, generator.Item2);
}
if (!outputTokens)
{
return;
}
var tokenStr = new StringBuilder();
foreach (var t in tokens.GetTokens())
{
tokenStr.AppendLine(t.ToString());
}
Console.WriteLine(tokenStr);
}
}
