// Package file provides ...
package file

import (
	"os"
	"path/filepath"
	"text/template"
	"time"

	"github.com/eiblog/eiblog/pkg/cache"
	"github.com/eiblog/eiblog/pkg/config"

	"github.com/eiblog/utils/tmpl"
	"github.com/sirupsen/logrus"
)

var xmlTmpl *template.Template

func init() {
	root := filepath.Join(config.WorkDir, "conf", "tpl", "*.xml")

	var err error
	xmlTmpl, err = template.New("").Funcs(template.FuncMap{
		"dateformat": tmpl.DateFormat,
	}).ParseGlob(root)
	if err != nil {
		panic(err)
	}
	go timerFeed()
	go timerSitemap()
}

// timerFeed 定时刷新feed
func timerFeed() {
	tpl := xmlTmpl.Lookup("feedTpl.xml")
	if tpl == nil {
		logrus.Info("file: not found: feedTpl.xml")
		return
	}

	t := time.NewTicker(time.Hour * 4)
	for now := range t.C {
		_, _, articles := cache.Ei.PageArticles(1, 20)
		params := map[string]interface{}{
			"Titile":    cache.Ei.Blogger.BTitle,
			"SubTitle":  cache.Ei.Blogger.SubTitle,
			"Host":      config.Conf.BlogApp.Host,
			"FeedrURL":  config.Conf.BlogApp.FeedRPC.FeedrURL,
			"BuildDate": now.Format(time.RFC1123Z),
			"Articles":  articles,
		}
		f, err := os.OpenFile("assets/feed.xml", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
		if err != nil {
			logrus.Error("file: timerFeed.OpenFile: ", err)
			continue
		}
		defer f.Close()
		err = tpl.Execute(f, params)
		if err != nil {
			logrus.Error("file: timerFeed.Execute: ", err)
			continue
		}
	}
}

// timerSitemap 定时刷新sitemap
func timerSitemap() {
	tpl := xmlTmpl.Lookup("sitemapTpl.xml")
	if tpl == nil {
		logrus.Info("file: not found: sitemapTpl.xml")
		return
	}

	t := time.NewTicker(time.Hour * 4)
	for range t.C {
		params := map[string]interface{}{
			"Articles": cache.Ei.Articles,
			"Host":     config.Conf.BlogApp.Host,
		}
		f, err := os.OpenFile("assets/sitemap.xml", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
		if err != nil {
			logrus.Error("file: timerSitemap.OpenFile: ", err)
			continue
		}
		defer f.Close()
		err = tpl.Execute(f, params)
		if err != nil {
			logrus.Error("file: timerSitemap.Execute: ", err)
			continue
		}
	}
}

// generateOpensearch 生成opensearch.xml
func generateOpensearch() {
	tpl := xmlTmpl.Lookup("opensearchTpl.xml")
	if tpl == nil {
		logrus.Info("file: not found: opensearchTpl.xml")
		return
	}
	params := map[string]string{
		"BTitle":   cache.Ei.Blogger.BTitle,
		"SubTitle": cache.Ei.Blogger.SubTitle,
		"Host":     config.Conf.BlogApp.Host,
	}
	f, err := os.OpenFile("static/opensearch.xml", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
	if err != nil {
		logrus.Error("file: generateOpensearch.OpenFile: ", err)
		return
	}
	defer f.Close()
	err = tpl.Execute(f, params)
	if err != nil {
		logrus.Error("file: generateOpensearch.Execute: ", err)
		return
	}
}

// generateRobots 生成robots.txt
func generateRobots() {
	tpl := xmlTmpl.Lookup("robotsTpl.xml")
	if tpl == nil {
		logrus.Info("file: not found: robotsTpl.xml")
		return
	}
	params := map[string]string{
		"Host": config.Conf.BlogApp.Host,
	}
	f, err := os.OpenFile("static/robots.txt", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
	if err != nil {
		logrus.Error("file: generateRobots.OpenFile: ", err)
		return
	}
	defer f.Close()
	err = tpl.Execute(f, params)
	if err != nil {
		logrus.Error("file: generateRobots.Execute: ", err)
		return
	}
}

// generateCrossdomain 生成crossdomain.xml
func generateCrossdomain() {
	tpl := xmlTmpl.Lookup("crossdomainTpl.xml")
	if tpl == nil {
		logrus.Info("file: not found: crossdomainTpl.xml")
		return
	}
	params := map[string]string{
		"Host": config.Conf.BlogApp.Host,
	}
	f, err := os.OpenFile("static/crossdomain.xml", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
	if err != nil {
		logrus.Error("file: generateCrossdomain.OpenFile: ", err)
		return
	}
	defer f.Close()
	err = tpl.Execute(f, params)
	if err != nil {
		logrus.Error("file: generateCrossdomain.Execute: ", err)
		return
	}
}