Better regex for automatic fill return value

tan-wei · tan-wei · commit b943755b93a2 · 2025-03-12T12:19:45.000+08:00
diff --git a/src/main.rs b/src/main.rs
@@ -203,7 +203,7 @@ fn parse_discuss_link(problem: &Problem) -> String {
 }
 
 fn insert_return_in_code(return_type: &str, code: &str) -> String {
-    let re = Regex::new(r"\{[ \n]+}").unwrap();
+    let re = Regex::new(r"\{[\s\n]+\}").unwrap();
     match return_type {
         "ListNode" => re
             .replace(&code, "{\n        Some(Box::new(ListNode::new(0)))\n    }")

Original file line number	Diff line number	Diff line change
`@@ -203,7 +203,7 @@ fn parse_discuss_link(problem: &Problem) -> String {`
`203`	`203`	`}`
`204`	`204`
`205`	`205`	`fn insert_return_in_code(return_type: &str, code: &str) -> String {`
`206`		`- let re = Regex::new(r"\{[ \n]+}").unwrap();`
	`206`	`+ let re = Regex::new(r"\{[\s\n]+\}").unwrap();`
`207`	`207`	`match return_type {`
`208`	`208`	`"ListNode" => re`
`209`	`209`	`.replace(&code, "{\n Some(Box::new(ListNode::new(0)))\n }")`