feat: support Java (#3)

zcong1993 · Jan 18, 2022 · 6f44c33 · 6f44c33
1 parent f318eb1
commit 6f44c33
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -26,7 +26,7 @@ $ leetcode-tool help
 
 ```js
 {
-  "lang": "go", // 项目全局语言, 配置后 new 命令 --lang 参数可省略, 目前支持 go ts js py3
+  "lang": "go", // 项目全局语言, 配置后 new 命令 --lang 参数可省略, 目前支持 go ts js py3 java
   "cookie": "xxx"
 }
 ```
@@ -41,6 +41,7 @@ $ leetcode-tool help
 - Typescript ts
 - Javascript js
 - Python3 py3
+- Java java
 
 ## 主要功能
 

diff --git a/cmd/new/main.go b/cmd/new/main.go
@@ -49,6 +49,10 @@ var (
 			LeetcodeLang: "Python3",
 			TplFiles:     []TplFile{{"code", "solve_%s.py", codeStrPy3}, {"test", "test_%s.py", testCodeStrPy3}, {"__init__", "__init__.py", ""}},
 		},
+		"java": {
+			LeetcodeLang: "Java",
+			TplFiles:     []TplFile{{"code", "solve_%s.java", codeStrJava}, {"test", "test_%s.java", testCodeStrJava}},
+		},
 	}
 )
 
@@ -212,3 +216,29 @@ var (
 	pass
 `
 )
+
+var (
+	codeStrJava = `package {{ .Folder }};
+
+/**
+ * @index {{ .Index }}
+ * @title {{ .Title }}
+ * @difficulty {{ .Difficulty }}
+ * @tags {{ .TagStr }}
+ * @draft false
+ * @link {{ .Link }}
+ * @frontendId {{ .FrontendId }}
+*/
+{{ .Code }}
+`
+
+	testCodeStrJava = `package {{ .Folder }};
+
+public class test_{{ printf "%04s" .Index }} {
+	public static void main(String[] args) {
+		Solution solution = new Solution();
+		// do some test
+	}
+}
+`
+)