githubmirror/LeetCode-Go

ctl/template_render.go

36 строк611 B

halfrost

Change ctl module level

11 сен 2022, 02:17

11 сен 2022, 02:17f0d16fe

package main

import (
	"bytes"
	"fmt"
	"html/template"
	"io/ioutil"
	"os"
)

func makeReadmeFile(mdrows Mdrows) {
	file := "./README.md"
	os.Remove(file)
	var b bytes.Buffer
	tmpl := template.Must(template.New("readme").Parse(readTMPL("template.markdown")))
	err := tmpl.Execute(&b, mdrows)
	if err != nil {
		fmt.Println(err)
	}
	// 保存 README.md 文件
	WriteFile(file, b.Bytes())
}

func readTMPL(path string) string {
	file, err := os.Open(path)
	if err != nil {
		fmt.Println(err)
	}
	defer file.Close()

	data, err := ioutil.ReadAll(file)
	if err != nil {
		fmt.Println(err)
	}
	return string(data)
}