Support C# Statement queries using await when targeting executable

Closes #10
atifaziz · Mar 18, 2017 · 46b0feb · 46b0feb
1 parent b62ccb8
commit 46b0feb
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/Program.cs b/src/Program.cs
@@ -600,6 +600,20 @@ static void GenerateExecutable(string cscPath, string queryFilePath,
                         "    }",
                             source,
                         "}")
+                : CSharpSyntaxTree.ParseText("void Main() {" + source + "}")
+                                  .GetRoot()
+                                  .DescendantNodes()
+                                  .OfType<AwaitExpressionSyntax>()
+                                  .Any()
+                ? Seq.Return(
+                        "class UserQuery {",
+                        "    static int Main(string[] args) {",
+                        "        new UserQuery().Main().Wait(); return 0;",
+                        "    }",
+                        "    async Task Main() {",
+                                source,
+                        "    }",
+                        "}")
                 : Seq.Return(
                         "class UserQuery {",
                         "    static int Main(string[] args) {",