helper 整理了 *_test.go 文件模板。顺手整理了 helper 的代码。

Author: aQua

README_TAIL.md

@@ -1,5 +1,5 @@
 ## helper
-[helper](./helper.v3)会帮助处理大部分琐碎的工作。
+[helper](./helper.v4)会帮助处理大部分琐碎的工作。
 
 ## notes
 [notes](./notes)记录了我答题过程中，对知识点的总结。

helper.v4/data.go

@@ -71,6 +71,8 @@ func diff(new, old *leetcode) {
 	// 对比 已完成的问题
 	lenNew := len(new.Problems)
 	lenOld := len(old.Problems)
+	// TODO: 处理数据错位问题
+	//	isOrderChanged := false
 	isChanged := false
 
 	i := 0
@@ -95,7 +97,7 @@ func diff(new, old *leetcode) {
 	}
 
 	for i < lenNew {
-		log.Printf("新题 %d %s", new.Problems[i].ID, new.Problems[i].Title)
+		log.Printf("出现新题: %d %s", new.Problems[i].ID, new.Problems[i].Title)
 		i++
 	}
 }

helper.v4/leetcode.go

@@ -16,7 +16,6 @@ type leetcode struct {
 	Problems   problems
 }
 
-// TODO: 去除 username 和 ranking 的全局变量
 func newLeetCode() *leetcode {
 	return &leetcode{
 		Username: cfg.Login,

helper.v4/main.go

@@ -3,17 +3,15 @@ package main
 import (
 	"fmt"
 	"log"
-	"net/http"
 	"os"
 	"strconv"
 
 	"github.com/BurntSushi/toml"
-	"github.com/mozillazg/request"
 )
 
 // 程序辅助设置
 const (
-	VERSION = "4.0.0"
+	VERSION = "4.1.0"
 	USAGE   = `使用方法：
 	1. 运行 helper 会重新生成项目的README.md。
 	2. 运行 helper 123 会生成第123题的答题文件夹。`
@@ -31,8 +29,6 @@ func init() {
 	}
 	log.Printf("Hi, %s. \n", cfg.Login)
 
-	// 对 req 赋值
-	req = request.NewRequest(new(http.Client))
 }
 
 func main() {
@@ -52,7 +48,9 @@ func main() {
 		}
 	}
 
-	// 下载题目资料不需要登录 leetcode
+	// 由于网络原因，有时候 signin 比较慢
+	signin()
+
 	if problemNum > 0 {
 		lc, err := readLeetCodeRecord()
 		if err != nil {
@@ -62,8 +60,6 @@ func main() {
 		return
 	}
 
-	signin()
-
 	categories := []string{
 		"Algorithms",
 		// "database",

helper.v4/net.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"io/ioutil"
 	"log"
+	"net/http"
 	"regexp"
 	"strings"
 
@@ -21,7 +22,9 @@ var req *request.Request
 // 登录 leetcode
 func signin() {
 	log.Println("正在登录中...")
-
+	
+	// 对 req 赋值
+	req = request.NewRequest(new(http.Client))
 	// 配置request
 	req.Headers = map[string]string{
 		"Accept-Encoding": "",

helper.v4/problem.go

@@ -31,28 +31,34 @@ func makeProblemDir(ps problems, problemNum int) {
 		return
 	}
 
-	pb.build()
+	build(pb)
 }
 
-func (p problem) build() {
+func build(p problem) {
 	if p.IsAccepted && GoKit.Exist(p.Dir) {
 		log.Fatalf("第 %d 题已经accepted，请**删除**或**重命名**  %s 文件夹后，再尝试。", p.ID, p.Dir)
 	}
 
+	// 对于没有 accepted 的题目，直接删除重建
+	if err := os.RemoveAll(p.Dir); err != nil {
+		log.Fatalln("无法删除目录", p.Dir)
+	}
+
 	mask := syscall.Umask(0)
 	defer syscall.Umask(mask)
 
 	err := os.Mkdir(p.Dir, 0755)
 	if err != nil {
-		log.Fatalf("无法创建目录: %s", p.Dir)
+		log.Fatalf("无法创建目录，%s ：%s", p.Dir, err)
 	}
-	fc, fcHead, para, ans := getFunction(p.link())
 
-	log.Printf("开始创建 %d.%s 的文件夹...\n", p.ID, p.Title)
+	log.Printf("开始创建 %d %s 的文件夹...\n", p.ID, p.Title)
 
 	creatREADME(p)
+
+	fc, fcName, para, ans := getFunction(p.link())
 	creatGo(p, fc)
-	creatGoTest(p, fcHead, para, ans)
+	creatGoTest(p, fcName, para, ans)
 
 	log.Printf("%d.%s 的文件夹，创建完毕。\n", p.ID, p.Title)
 }
@@ -82,15 +88,18 @@ func creatREADME(p problem) {
 
 func getQuestionDescription(URL string) string {
 	doc, err := goquery.NewDocument(URL)
+
 	if err != nil {
 		log.Fatal(err)
 	}
+
 	return strings.TrimSpace(doc.Find("div.question-description").Text())
 }
 
-func getFunction(URL string) (fc, fcHead, p, a string) {
+func getFunction(URL string) (fc, fcName, para, ansType string) {
 	data := getRaw(URL)
 	str := string(data)
+
 	i := strings.Index(str, "codeDefinition:")
 	j := i + strings.Index(str[i:], "enableTestMode:")
 	str = str[i:j]
@@ -111,23 +120,24 @@ func getFunction(URL string) (fc, fcHead, p, a string) {
 	k := 0
 	i0 := strings.Index(fc, " ")
 	i = strings.Index(fc, "(")
-	fcHead = fc[i0+1 : i]
+	fcName = fc[i0+1 : i]
 
 	j = strings.Index(fc, ")")
 	k = strings.Index(fc, "{")
-	p = strings.Replace(fc[i+1:j], ",", "\n", -1)
-	a = fc[j+1 : k]
+	para = strings.Replace(fc[i+1:j], ",", "\n", -1)
+	ansType = fc[j+1 : k]
 
 	fc = fc[:k] + "{\n\n}"
+
 	return
 }
 
-func creatGo(p problem, fc string) {
+func creatGo(p problem, function string) {
 	fileFormat := `package %s
 
 %s
 `
-	content := fmt.Sprintf(fileFormat, p.packageName(), fc)
+	content := fmt.Sprintf(fileFormat, p.packageName(), function)
 	filename := fmt.Sprintf("%s/%s.go", p.Dir, p.TitleSlug)
 
 	err := ioutil.WriteFile(filename, []byte(content), 0755)
@@ -136,57 +146,41 @@ func creatGo(p problem, fc string) {
 	}
 }
 
-func creatGoTest(p problem, fcHead, para, ans string) {
+func creatGoTest(p problem, fcName, para, ansType string) {
 	fileFormat := `package %s
 
 import (
-	"testing"
 	"fmt"
+	"testing"
 
 	"github.com/stretchr/testify/assert"
 )
 
-type question struct {
-	para
-	ans
-}
-
-// para 是参数
-type para struct {
-	%s
-}
-
-// ans 是答案
-type ans struct {
-	one %s
-}
-
 func Test_%s(t *testing.T) {
 	ast := assert.New(t)
 
-	qs := []question{
+	// tcs is testcase slice
+	tcs := []struct {
+		%s
+		ans %s
+	}{
+
+		{
+			,
 
-		question{
-			para{
-					,
-			},
-			ans{
-					,
-			},
 		},
-	
-		// 如需多个测试，可以复制上方元素。
 	}
 
-	for _, q := range qs {
-		a, p := q.ans, q.para
-		fmt.Printf("~~%s~~\n", p)
+	for _, tc := range tcs {
+		fmt.Printf("~~%s~~\n", tc)
 
-		ast.Equal(a.one, %s(p.  ), "输入:%s", p)
+		ast.Equal(tc.ans, %s(%s), "输入:%s", tc)
 	}
 }
 `
-	content := fmt.Sprintf(fileFormat, p.packageName(), para, ans, p.packageName(), `%v`, fcHead, `%v`)
+	tcPara := getTcPara(para)
+
+	content := fmt.Sprintf(fileFormat, p.packageName(), p.packageName(), para, ansType, `%v`, fcName, tcPara, `%v`)
 	filename := fmt.Sprintf("%s/%s_test.go", p.Dir, p.TitleSlug)
 
 	err := ioutil.WriteFile(filename, []byte(content), 0755)
@@ -195,6 +189,26 @@ func Test_%s(t *testing.T) {
 	}
 }
 
+// 把 函数的参数 变成 tc 的参数
+func getTcPara(para string) string {
+	// 把 para 按行切分
+	paras := strings.Split(para, `\n`)
+
+	// 把单个参数按空格，切分成参数名和参数类型
+	temp := make([][]string, len(paras))
+	for i := range paras {
+		temp[i] = strings.Split(paras[i], ` `)
+	}
+
+	// 在参数名称前添加 "tc." 并组合在一起
+	res := "tc." + temp[0][0]
+	for i := 1; i < len(temp); i++ {
+		res += "," + " tc." + temp[i][0]
+	}
+
+	return res
+}
+
 func (p problem) packageName() string {
 	return fmt.Sprintf("Problem%04d", p.ID)
 }