mirror of
https://github.com/halfrost/LeetCode-Go.git
synced 2025-07-04 16:12:47 +08:00
349 lines
13 KiB
Go
349 lines
13 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/halfrost/LeetCode-Go/ctl/util"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
chapterOneFileOrder = []string{"_index", "Data_Structure", "Algorithm", "Time_Complexity"}
|
|
chapterOneMenuOrder = []string{"_index", "#关于作者", "Data_Structure", "Algorithm", "Time_Complexity"}
|
|
chapterTwoFileOrder = []string{"_index", "Array", "String", "Two_Pointers", "Linked_List", "Stack", "Tree", "Dynamic_Programming", "Backtracking", "Depth_First_Search", "Breadth_First_Search",
|
|
"Binary_Search", "Math", "Hash_Table", "Sorting", "Bit_Manipulation", "Union_Find", "Sliding_Window", "Segment_Tree", "Binary_Indexed_Tree"}
|
|
chapterThreeFileOrder = []string{"_index", "Segment_Tree", "UnionFind", "LRUCache", "LFUCache"}
|
|
preNextHeader = "----------------------------------------------\n<div style=\"display: flex;justify-content: space-between;align-items: center;\">\n"
|
|
preNextFotter = "</div>"
|
|
delLine = "----------------------------------------------\n"
|
|
delHeader = "<div style=\"display: flex;justify-content: space-between;align-items: center;\">"
|
|
delLabel = "<[a-zA-Z]+.*?>([\\s\\S]*?)</[a-zA-Z]*?>"
|
|
delFooter = "</div>"
|
|
|
|
//ErrNoFilename is thrown when the path the the file to tail was not given
|
|
ErrNoFilename = errors.New("You must provide the path to a file in the \"-file\" flag.")
|
|
|
|
//ErrInvalidLineCount is thrown when the user provided 0 (zero) as the value for number of lines to tail
|
|
ErrInvalidLineCount = errors.New("You cannot tail zero lines.")
|
|
|
|
chapterMap = map[string]map[string]string{
|
|
"ChapterOne": {
|
|
"_index": "第一章 序章",
|
|
"Data_Structure": "1.1 数据结构知识",
|
|
"Algorithm": "1.2 算法知识",
|
|
"Time_Complexity": "1.3 时间复杂度",
|
|
},
|
|
"ChapterTwo": {
|
|
"_index": "第二章 算法专题",
|
|
"Array": "2.01 Array",
|
|
"String": "2.02 String",
|
|
"Two_Pointers": "2.03 ✅ Two Pointers",
|
|
"Linked_List": "2.04 ✅ Linked List",
|
|
"Stack": "2.05 ✅ Stack",
|
|
"Tree": "2.06 Tree",
|
|
"Dynamic_Programming": "2.07 Dynamic Programming",
|
|
"Backtracking": "2.08 ✅ Backtracking",
|
|
"Depth_First_Search": "2.09 Depth First Search",
|
|
"Breadth_First_Search": "2.10 Breadth First Search",
|
|
"Binary_Search": "2.11 Binary Search",
|
|
"Math": "2.12 Math",
|
|
"Hash_Table": "2.13 Hash Table",
|
|
"Sorting": "2.14 ✅ Sorting",
|
|
"Bit_Manipulation": "2.15 ✅ Bit Manipulation",
|
|
"Union_Find": "2.16 ✅ Union Find",
|
|
"Sliding_Window": "2.17 ✅ Sliding Window",
|
|
"Segment_Tree": "2.18 ✅ Segment Tree",
|
|
"Binary_Indexed_Tree": "2.19 ✅ Binary Indexed Tree",
|
|
},
|
|
"ChapterThree": {
|
|
"_index": "第三章 一些模板",
|
|
"Segment_Tree": "3.1 Segment Tree",
|
|
"UnionFind": "3.2 UnionFind",
|
|
"LRUCache": "3.3 LRUCache",
|
|
"LFUCache": "3.4 LFUCache",
|
|
},
|
|
"ChapterFour": {
|
|
"_index": "第四章 Leetcode 题解",
|
|
},
|
|
}
|
|
)
|
|
|
|
func getChapterFourFileOrder() ([]string, []int) {
|
|
solutions, solutionIds := util.LoadChapterFourDir()
|
|
chapterFourFileOrder := []string{"_index"}
|
|
chapterFourFileOrder = append(chapterFourFileOrder, solutions...)
|
|
fmt.Printf("ChapterFour 中包括 _index 有 %v 个文件, len(id) = %v\n", len(chapterFourFileOrder), len(solutionIds))
|
|
return chapterFourFileOrder, solutionIds
|
|
}
|
|
|
|
func newLabelCommand() *cobra.Command {
|
|
mc := &cobra.Command{
|
|
Use: "label <subcommand>",
|
|
Short: "Label related commands",
|
|
}
|
|
//mc.PersistentFlags().StringVar(&logicEndpoint, "endpoint", "localhost:5880", "endpoint of logic service")
|
|
mc.AddCommand(
|
|
newAddPreNext(),
|
|
newDeletePreNext(),
|
|
)
|
|
return mc
|
|
}
|
|
|
|
func newAddPreNext() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "add-pre-next",
|
|
Short: "Add pre-next label",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
addPreNext()
|
|
},
|
|
}
|
|
// cmd.Flags().StringVar(&alias, "alias", "", "alias")
|
|
// cmd.Flags().StringVar(&appId, "appid", "", "appid")
|
|
return cmd
|
|
}
|
|
|
|
func newDeletePreNext() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "del-pre-next",
|
|
Short: "Delete pre-next label",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
delPreNext()
|
|
},
|
|
}
|
|
// cmd.Flags().StringVar(&alias, "alias", "", "alias")
|
|
// cmd.Flags().StringVar(&appId, "appid", "", "appid")
|
|
return cmd
|
|
}
|
|
|
|
func addPreNext() {
|
|
// Chpater one add pre-next
|
|
addPreNextLabel(chapterOneFileOrder, []string{}, []int{}, "", "ChapterOne", "ChapterTwo")
|
|
// Chpater two add pre-next
|
|
addPreNextLabel(chapterTwoFileOrder, chapterOneFileOrder, []int{}, "ChapterOne", "ChapterTwo", "ChapterThree")
|
|
// Chpater three add pre-next
|
|
addPreNextLabel(chapterThreeFileOrder, chapterTwoFileOrder, []int{}, "ChapterTwo", "ChapterThree", "ChapterFour")
|
|
// Chpater four add pre-next
|
|
//fmt.Printf("%v\n", getChapterFourFileOrder())
|
|
chapterFourFileOrder, solutionIds := getChapterFourFileOrder()
|
|
addPreNextLabel(chapterFourFileOrder, chapterThreeFileOrder, solutionIds, "ChapterThree", "ChapterFour", "")
|
|
}
|
|
|
|
func addPreNextLabel(order, preOrder []string, chapterFourIds []int, preChapter, chapter, nextChapter string) {
|
|
var (
|
|
exist bool
|
|
err error
|
|
res []byte
|
|
count int
|
|
)
|
|
for index, path := range order {
|
|
tmp := ""
|
|
if index == 0 {
|
|
if chapter == "ChapterOne" {
|
|
// 第一页不需要“上一章”
|
|
tmp = "\n\n" + delLine + fmt.Sprintf("<p align = \"right\"><a href=\"https://books.halfrost.com/leetcode/%v/%v/\">下一页➡️</a></p>\n", chapter, order[index+1])
|
|
} else {
|
|
if chapter == "ChapterFour" {
|
|
tmp = "\n\n" + preNextHeader + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/\">⬅️上一章</a></p>\n", preChapter, preOrder[len(preOrder)-1]) + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/%v/\">下一页➡️</a></p>\n", chapter, util.GetChpaterFourFileNum(chapterFourIds[(index-1)+1]), order[index+1]) + preNextFotter
|
|
} else {
|
|
tmp = "\n\n" + preNextHeader + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/\">⬅️上一章</a></p>\n", preChapter, preOrder[len(preOrder)-1]) + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/\">下一页➡️</a></p>\n", chapter, order[index+1]) + preNextFotter
|
|
}
|
|
}
|
|
} else if index == len(order)-1 {
|
|
if chapter == "ChapterFour" {
|
|
// 最后一页不需要“下一页”
|
|
tmp = "\n\n" + delLine + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/%v/\">⬅️上一页</a></p>\n", chapter, util.GetChpaterFourFileNum(chapterFourIds[(index-1)-1]), order[index-1])
|
|
} else {
|
|
tmp = "\n\n" + preNextHeader + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/\">⬅️上一页</a></p>\n", chapter, order[index-1]) + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/\">下一章➡️</a></p>\n", nextChapter) + preNextFotter
|
|
}
|
|
} else if index == 1 {
|
|
if chapter == "ChapterFour" {
|
|
tmp = "\n\n" + preNextHeader + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/\">⬅️上一页</a></p>\n", chapter) + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/%v/\">下一页➡️</a></p>\n", chapter, util.GetChpaterFourFileNum(chapterFourIds[(index-1)+1]), order[index+1]) + preNextFotter
|
|
} else {
|
|
tmp = "\n\n" + preNextHeader + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/\">⬅️上一页</a></p>\n", chapter) + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/\">下一页➡️</a></p>\n", chapter, order[index+1]) + preNextFotter
|
|
}
|
|
} else {
|
|
if chapter == "ChapterFour" {
|
|
tmp = "\n\n" + preNextHeader + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/%v/\">⬅️上一页</a></p>\n", chapter, util.GetChpaterFourFileNum(chapterFourIds[(index-1)-1]), order[index-1]) + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/%v/\">下一页➡️</a></p>\n", chapter, util.GetChpaterFourFileNum(chapterFourIds[(index-1)+1]), order[index+1]) + preNextFotter
|
|
} else {
|
|
tmp = "\n\n" + preNextHeader + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/\">⬅️上一页</a></p>\n", chapter, order[index-1]) + fmt.Sprintf("<p><a href=\"https://books.halfrost.com/leetcode/%v/%v/\">下一页➡️</a></p>\n", chapter, order[index+1]) + preNextFotter
|
|
}
|
|
}
|
|
|
|
if chapter == "ChapterFour" && index > 0 {
|
|
path = fmt.Sprintf("%v/%v", util.GetChpaterFourFileNum(chapterFourIds[(index-1)]), path)
|
|
}
|
|
|
|
exist, err = needAdd(fmt.Sprintf("../website/content/%v/%v.md", chapter, path))
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
// 当前没有上一页和下一页，才添加
|
|
if !exist && err == nil {
|
|
res, err = eofAdd(fmt.Sprintf("../website/content/%v/%v.md", chapter, path), tmp)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
util.WriteFile(fmt.Sprintf("../website/content/%v/%v.md", chapter, path), res)
|
|
count++
|
|
}
|
|
}
|
|
fmt.Printf("添加了 %v 个文件的 pre-next\n", count)
|
|
}
|
|
|
|
func eofAdd(filePath string, labelString string) ([]byte, error) {
|
|
f, err := os.OpenFile(filePath, os.O_RDONLY, 0644)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
reader, output := bufio.NewReader(f), []byte{}
|
|
|
|
for {
|
|
line, _, err := reader.ReadLine()
|
|
if err != nil {
|
|
if err == io.EOF {
|
|
output = append(output, []byte(labelString)...)
|
|
output = append(output, []byte("\n")...)
|
|
return output, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
output = append(output, line...)
|
|
output = append(output, []byte("\n")...)
|
|
}
|
|
}
|
|
|
|
func delPreNext() {
|
|
// Chpater one del pre-next
|
|
delPreNextLabel(chapterOneFileOrder, []int{}, "ChapterOne")
|
|
// Chpater two del pre-next
|
|
delPreNextLabel(chapterTwoFileOrder, []int{}, "ChapterTwo")
|
|
// Chpater three del pre-next
|
|
delPreNextLabel(chapterThreeFileOrder, []int{}, "ChapterThree")
|
|
// Chpater four del pre-next
|
|
chapterFourFileOrder, solutionIds := getChapterFourFileOrder()
|
|
delPreNextLabel(chapterFourFileOrder, solutionIds, "ChapterFour")
|
|
}
|
|
|
|
func delPreNextLabel(order []string, chapterFourIds []int, chapter string) {
|
|
count := 0
|
|
for index, path := range order {
|
|
lineNum := 5
|
|
if index == 0 && chapter == "ChapterOne" || index == len(order)-1 && chapter == "ChapterFour" {
|
|
lineNum = 3
|
|
}
|
|
if chapter == "ChapterFour" && index > 0 {
|
|
path = fmt.Sprintf("%v/%v", util.GetChpaterFourFileNum(chapterFourIds[(index-1)]), path)
|
|
}
|
|
|
|
exist, err := needAdd(fmt.Sprintf("../website/content/%v/%v.md", chapter, path))
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
// 存在才删除
|
|
if exist && err == nil {
|
|
removeLine(fmt.Sprintf("../website/content/%v/%v.md", chapter, path), lineNum+1)
|
|
count++
|
|
}
|
|
}
|
|
fmt.Printf("删除了 %v 个文件的 pre-next\n", count)
|
|
// 另外一种删除方法
|
|
// res, err := eofDel(fmt.Sprintf("../website/content/ChapterOne/%v.md", v))
|
|
// if err != nil {
|
|
// fmt.Println(err)
|
|
// return
|
|
// }
|
|
// util.WriteFile(fmt.Sprintf("../website/content/ChapterOne/%v.md", v), res)
|
|
}
|
|
|
|
func needAdd(filePath string) (bool, error) {
|
|
f, err := os.OpenFile(filePath, os.O_RDONLY, 0644)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer f.Close()
|
|
reader, output := bufio.NewReader(f), []byte{}
|
|
for {
|
|
line, _, err := reader.ReadLine()
|
|
if err != nil {
|
|
if err == io.EOF {
|
|
return false, nil
|
|
}
|
|
return false, err
|
|
}
|
|
if ok, _ := regexp.Match(delHeader, line); ok {
|
|
return true, nil
|
|
} else if ok, _ := regexp.Match(delLabel, line); ok {
|
|
return true, nil
|
|
} else {
|
|
output = append(output, line...)
|
|
output = append(output, []byte("\n")...)
|
|
}
|
|
}
|
|
}
|
|
|
|
func eofDel(filePath string) ([]byte, error) {
|
|
f, err := os.OpenFile(filePath, os.O_RDONLY, 0644)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
reader, output := bufio.NewReader(f), []byte{}
|
|
for {
|
|
line, _, err := reader.ReadLine()
|
|
if err != nil {
|
|
if err == io.EOF {
|
|
return output, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
if ok, _ := regexp.Match(delLine, line); ok {
|
|
reg := regexp.MustCompile(delLine)
|
|
newByte := reg.ReplaceAll(line, []byte(""))
|
|
output = append(output, newByte...)
|
|
output = append(output, []byte("\n")...)
|
|
} else if ok, _ := regexp.Match(delHeader, line); ok {
|
|
reg := regexp.MustCompile(delHeader)
|
|
newByte := reg.ReplaceAll(line, []byte(""))
|
|
output = append(output, newByte...)
|
|
output = append(output, []byte("\n")...)
|
|
} else if ok, _ := regexp.Match(delLabel, line); ok {
|
|
reg := regexp.MustCompile(delLabel)
|
|
newByte := reg.ReplaceAll(line, []byte(""))
|
|
output = append(output, newByte...)
|
|
output = append(output, []byte("\n")...)
|
|
} else if ok, _ := regexp.Match(delFooter, line); ok {
|
|
reg := regexp.MustCompile(delFooter)
|
|
newByte := reg.ReplaceAll(line, []byte(""))
|
|
output = append(output, newByte...)
|
|
output = append(output, []byte("\n")...)
|
|
} else {
|
|
output = append(output, line...)
|
|
output = append(output, []byte("\n")...)
|
|
}
|
|
}
|
|
}
|
|
|
|
func removeLine(path string, lineNumber int) {
|
|
file, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
info, _ := os.Stat(path)
|
|
mode := info.Mode()
|
|
array := strings.Split(string(file), "\n")
|
|
array = array[:len(array)-lineNumber-1]
|
|
ioutil.WriteFile(path, []byte(strings.Join(array, "\n")), mode)
|
|
//fmt.Println("remove line successful")
|
|
}
|