auto import tree utils in template

aylei · aylei · commit 216effdf7367 · 2019-02-10T22:26:41.000+08:00
diff --git a/src/main.rs b/src/main.rs
@@ -64,6 +64,9 @@ fn parse_extra_use(code: &str) -> String {
     if code.contains("pub struct ListNode") {
         extra_use_line.push_str("\nuse super::util::linked_list::{ListNode, to_list};")
     }
+    if code.contains("pub struct TreeNode") {
+        extra_use_line.push_str("\nuse super::util::tree::{TreeNode, to_tree};")
+    }
     extra_use_line
 }
 

Original file line number	Diff line number	Diff line change
`@@ -64,6 +64,9 @@ fn parse_extra_use(code: &str) -> String {`
`64`	`64`	`if code.contains("pub struct ListNode") {`
`65`	`65`	`extra_use_line.push_str("\nuse super::util::linked_list::{ListNode, to_list};")`
`66`	`66`	`}`
	`67`	`+ if code.contains("pub struct TreeNode") {`
	`68`	`+ extra_use_line.push_str("\nuse super::util::tree::{TreeNode, to_tree};")`
	`69`	`+ }`
`67`	`70`	`extra_use_line`
`68`	`71`	`}`
`69`	`72`