main.go

package main

import (
	"fmt"
	"github.com/shurcooL/githubv4"
	"lizi/core"
	"log"
	"os"
	"os/exec"
	"strconv"
	"text/template"
	"time"
)

var githubUserName = os.Getenv("LIZI_GITHUB_USER_NAME")
var githubRepo = os.Getenv("LIZI_GITHUB_DISCUSSION_REPO")
var githubAccessToken = os.Getenv("LIZI_GITHUB_ACCESS_TOKEN")

var githubPageEnabled, _ = strconv.ParseBool(getEnvWithDefault("LIZI_GITHUB_PAGE_ENABLED", "false"))

// var githubCommentRepo = "zuston/zuston.github.io"
var githubCommentRepo = os.Getenv("LIZI_GITHUB_COMMENT_REPO")

// var githubPageRepo = "github.com/zuston/zuston.github.io.git"
var githubPageRepo = os.Getenv("LIZI_GITHUB_PAGE_REPO")

// var githubPageAuthor = "Junfan Zhang"
var githubPageAuthor = os.Getenv("LIZI_GITHUB_PAGE_AUTHOR")

// var githubPageEmail = "zuston@apache.org"
var githubPageEmail = os.Getenv("LIZI_GITHUB_PAGE_EMAIL")

var api = core.NewApi(githubUserName, githubRepo, githubAccessToken)

func getEnvWithDefault(key string, defVal string) string {
	val, exist := os.LookupEnv(key)
	if !exist {
		return defVal
	}
	return val
}

var funcMap = template.FuncMap{
	"formatDate": func(unformated githubv4.DateTime) string {
		return unformated.Time.Format("2006-01-02")
	},
	"previewContent": func(fullContent githubv4.String) string {
		if len(fullContent) >= 250 {
			return string(fullContent)[0:250]
		}
		return string(fullContent)
	},
}

type IndexPageInfo struct {
	Discussions core.Discussions
	User        string
}

type PostPageInfo struct {
	Post        core.Node
	CommentRepo string
}

func Render() {
	// Remove the original output dir
	os.RemoveAll("./output")
	err := os.Mkdir("output", os.ModePerm)
	if err != nil {
		log.Fatal("Creating output directory failed.")
		return
	}
	cmd := exec.Command("cp", "-r", "./templates/js", "./output")
	cmd.Run()
	cmd = exec.Command("cp", "-r", "./templates/css", "./output")
	cmd.Run()
	log.Println("Successfully copy js/css resources to output directory.")

	discussions, err := api.FetchPosts("", "")
	if err != nil {
		log.Fatal("Errors on fetching all discussions records.")
		return
	}
	indexTemplate, err := template.New("index.html").Funcs(funcMap).ParseFiles(
		"templates/base/navbar.html",
		"templates/base/footer.html",
		"templates/index.html",
	)
	if err != nil {
		log.Fatal("Errors on rendering index.html")
		return
	}

	indexFile, err := os.Create("./output/index.html")
	indexTemplate.Execute(indexFile, IndexPageInfo{Discussions: discussions, User: githubUserName})

	// create the article dir
	err = os.Mkdir("./output/article", os.ModePerm)
	for _, metaInfo := range discussions.Nodes {
		number := int(metaInfo.Number)
		page, err := api.FetchPost(number)
		if err != nil {
			return
		}
		postTemplate, err := template.New("post.html").Funcs(funcMap).ParseFiles(
			"templates/base/navbar.html",
			"templates/base/footer.html",
			"templates/post.html")
		postFile, err := os.Create(fmt.Sprintf("./output/article/%d.html", number))
		postTemplate.Execute(postFile, PostPageInfo{
			page,
			githubCommentRepo,
		})
		log.Printf("Successfully rendered the post of [%d]", number)
	}
}

func execCommand(cmdDir string, cmdName string, cmdArgs ...string) {
	cmd := exec.Command(cmdName, cmdArgs...)
	cmd.Dir = cmdDir
	output, err := cmd.Output()
	log.Printf("%s - %v", string(output), err)
}

func Push2Github() {
	// Remove the original folder
	os.RemoveAll("/tmp/output")

	// cp the output to /tmp folder
	cmd := exec.Command("cp", "-r", "./output", "/tmp")
	err := cmd.Run()
	if err != nil {
		log.Fatal("Errors on copying output to tmp folder.")
	}
	// Initialize git repo
	execCmdWithDir := func(name string, args ...string) {
		execCommand("/tmp/output", name, args...)
	}
	execCmdWithDir("git", "init")
	execCmdWithDir("git", "remote", "add", "origin", "https://"+githubPageRepo)
	execCmdWithDir("git", "add", ".")
	execCmdWithDir("git", "config", "user.email", githubPageEmail)
	execCmdWithDir("git", "config", "user.name", githubPageAuthor)
	execCmdWithDir("git", "commit", "-m", "Publish latest post in "+time.ANSIC)
	execCmdWithDir("git", "push", "-f", fmt.Sprintf("https://%s@%s", githubAccessToken, githubPageRepo))
}

func main() {
	Render()
	log.Printf("Finished rendering the html.")

	// push to the Github page
	if githubPageEnabled {
		Push2Github()
		log.Printf("Finished pushing latest blog content to github page.")
	}
}