LeetCode-Go/ctl/util/util.go

package util

import (
	"bufio"
	"fmt"
	"io"
	"io/ioutil"
	"os"
	"path/filepath"
	"sort"
	"strconv"
	"strings"
)

// LoadSolutionsDir define
func LoadSolutionsDir() ([]int, []string, int) {
	solutionIds, soNames, total := loadFile("../leetcode/")
	fmt.Printf("读取了 %v 道题的题解，当前目录下有 %v 个文件(可能包含 .DS_Store)，目录中有 %v 道题在尝试中\n", len(solutionIds), total, total-len(solutionIds))
	return solutionIds, soNames, total - len(solutionIds)
}

func loadFile(path string) ([]int, []string, int) {
	files, err := ioutil.ReadDir(path)
	if err != nil {
		fmt.Println(err)
	}
	solutionIds, soNames, solutionsMap := []int{}, []string{}, map[int]string{}
	for _, f := range files {
		if f.Name()[4] == '.' {
			tmp, err := strconv.Atoi(f.Name()[:4])
			if err != nil {
				fmt.Println(err)
			}
			solutionIds = append(solutionIds, tmp)
			solutionsMap[tmp] = f.Name()
		}
	}
	sort.Ints(solutionIds)
	for _, v := range solutionIds {
		if name, ok := solutionsMap[v]; ok {
			soNames = append(soNames, name)
		}
	}
	return solutionIds, soNames, len(files)
}

// GetAllFile define
func GetAllFile(pathname string, fileList *[]string) ([]string, error) {
	rd, err := ioutil.ReadDir(pathname)
	for _, fi := range rd {
		if fi.IsDir() {
			//fmt.Printf("[%s]\n", pathname+"\\"+fi.Name())
			GetAllFile(pathname+fi.Name()+"/", fileList)
		} else {
			//fmt.Println(fi.Name())
			*fileList = append(*fileList, fi.Name())
		}
	}
	return *fileList, err
}

// LoadChapterFourDir define
func LoadChapterFourDir() ([]string, []int) {
	files, err := GetAllFile("../website/content/ChapterFour/", &[]string{})
	// files, err := ioutil.ReadDir("../website/content/ChapterFour/")
	if err != nil {
		fmt.Println(err)
	}
	solutions, solutionIds, solutionsMap := []string{}, []int{}, map[int]string{}
	for _, f := range files {
		if f[4] == '.' {
			tmp, err := strconv.Atoi(f[:4])
			if err != nil {
				fmt.Println(err)
			}
			solutionIds = append(solutionIds, tmp)
			// len(f.Name())-3 = 文件名去掉 .md 后缀
			solutionsMap[tmp] = f[:len(f)-3]
		}
	}
	sort.Ints(solutionIds)
	fmt.Printf("读取了第四章的 %v 道题的题解\n", len(solutionIds))
	for _, v := range solutionIds {
		if name, ok := solutionsMap[v]; ok {
			solutions = append(solutions, name)
		}
	}
	return solutions, solutionIds
}

// WriteFile define
func WriteFile(fileName string, content []byte) {
	file, err := os.OpenFile(fileName, os.O_RDWR|os.O_CREATE, 0777)
	if err != nil {
		fmt.Println(err)
	}
	defer file.Close()

	_, err = file.Write(content)
	if err != nil {
		fmt.Println(err)
	}
	//fmt.Println("write file successful")
}

// LoadFile define
func LoadFile(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
		}
		output = append(output, line...)
		output = append(output, []byte("\n")...)
	}
}

// DestoryDir define
func DestoryDir(path string) {
	filepath.Walk(path, func(path string, fi os.FileInfo, err error) error {
		if nil == fi {
			return err
		}
		if !fi.IsDir() {
			return nil
		}
		name := fi.Name()
		if strings.Contains(name, "temp") {
			fmt.Println("temp file name:", path)
			err := os.RemoveAll(path)
			if err != nil {
				fmt.Println("delet dir error:", err)
			}
		}
		return nil
	})
}

// CopyFile define
func CopyFile(dstName, srcName string) (written int64, err error) {
	src, err := os.Open(srcName)
	if err != nil {
		return
	}
	defer src.Close()
	dst, err := os.OpenFile(dstName, os.O_WRONLY|os.O_CREATE, 0644)
	if err != nil {
		return
	}
	defer dst.Close()
	return io.Copy(dst, src)
}

// BinarySearch define
func BinarySearch(nums []int, target int) int {
	low, high := 0, len(nums)-1
	for low <= high {
		mid := low + (high-low)>>1
		if nums[mid] == target {
			return mid
		} else if nums[mid] > target {
			high = mid - 1
		} else {
			low = mid + 1
		}
	}
	return -1
}

// GetChpaterFourFileNum define
func GetChpaterFourFileNum(num int) string {
	if num < 100 {
		return fmt.Sprintf("%04d~%04d", (num/100)*100+1, (num/100)*100+99)
	}
	return fmt.Sprintf("%04d~%04d", (num/100)*100, (num/100)*100+99)
}