Allow main function with arguments

fornwall · Oct 27, 2021 · 5a0d918 · 5a0d918
1 parent 32ac67f
commit 5a0d918
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/manifest.rs b/src/manifest.rs
@@ -46,10 +46,10 @@ pub fn split_input(
 ) -> MainResult<(String, String)> {
     fn contains_main_method(line: &str) -> bool {
         let line = line.trim_start();
-        line.starts_with("fn main()")
-            || line.starts_with("pub fn main()")
-            || line.starts_with("async fn main()")
-            || line.starts_with("pub async fn main()")
+        line.starts_with("fn main(")
+            || line.starts_with("pub fn main(")
+            || line.starts_with("async fn main(")
+            || line.starts_with("pub async fn main(")
     }
 
     let template_buf;