package main

import (
	"fmt"
	"html/template"
	"io/ioutil"
	"log"
	"os"
	"os/signal"
	"syscall"
	"net/http"
	"path/filepath"
	"sort"
	"strconv"
	"strings"
	"sync"
	"time"
	"tinybg/conf"
	"tinybg/utils"

	"github.com/gin-gonic/gin"
	"github.com/radovskyb/watcher"
)

type Post struct {
	ID      string //文章唯一ID 取 filename的md5
	Title   string
	Date    string
	Summary string
	Body    string
	File    string
	ImgFile string // 文章前的图片展示
	Item    string //分类
	Author  string //作者

	Cmts    []conf.Comment //评论
	CmtCnt  int            //评论数量
	VistCnt int            //浏览量
}

// Notice 跳转提示
type Notice struct {
	Mess    string
	IsSucc  bool
	TimeOut int
	Href    string
}

// ArticleRoute 文章的分类和路径信息
type ArticleRoute struct {
	Item string
	Name string
}

var artRouteMap map[string]ArticleRoute /*创建集合 */
var mapLock sync.RWMutex                //map是线程不安全的，写时使用线程锁防止崩溃

// NewArticles 最新文章按日期排序
type NewArticles []conf.Articles

// HotArticles 热门文章按访问量排序
type HotArticles []conf.Articles

//  NewArts  最新文章
var NewArts NewArticles

// HotArts 热门文章
var HotArts HotArticles

// NewPosts 总的文章，按时间排过序的
var NewPosts NewArticles

// NewCmts 最新评论
var NewCmts []conf.Comment

//文章排序的实现
//文章排序
//Len()
func (a NewArticles) Len() int {
	return len(a)
}

//Less():将有低到高排序
func (a NewArticles) Less(i, j int) bool {
	//fmt.Println(a[i].Date)
	//fmt.Println(a[j].Date)
	return a[i].Date > a[j].Date
}

//Swap()
func (a NewArticles) Swap(i, j int) {
	a[i], a[j] = a[j], a[i]
}

//Len()
func (a HotArticles) Len() int {
	return len(a)
}

//Less():将有低到高排序
func (a HotArticles) Less(i, j int) bool {
	//fmt.Println(a[i].Date)
	//fmt.Println(a[j].Date)
	return a[i].VistCnt > a[j].VistCnt
}

//Swap()
func (a HotArticles) Swap(i, j int) {
	a[i], a[j] = a[j], a[i]
}

// RefreshData 后台异步更新数据
func RefreshData() {
	//全局的最新评论，只显示最新的三条
	curcount := len(conf.Cmt.Comts) //总的评论数量
	NewCmts = conf.Cmt.Comts
	if curcount >= 3 {
		NewCmts = conf.Cmt.Comts[(curcount - 3):curcount] //最新的3条评论
	}
	//post := conf.Art.ArticlesMap[conf.Item.Items[id]]
	NewPosts = NewArticles{}
	HotArts = HotArticles{}
	for key, value := range conf.Art.ArticlesMap {
		fmt.Println(key)
		for _, value1 := range value {
			NewPosts = append(NewPosts, *value1)
			HotArts = append(HotArts, *value1)
		}
	}
	sort.Sort(NewPosts) //日期排序
	sort.Sort(HotArts)  //访问量排序
	//fmt.Println("IS Sorted?", sort.IsSorted(post))
	NewArts = NewPosts
	num := len(NewPosts)
	if num > 9 {
		NewArts = NewPosts[0:9]
		HotArts = HotArts[0:9]
	}
	//conf.Art.Save()
	//conf.Stat.Save()
}

func handleIndex(c *gin.Context) {
	//fmt.Printf("%#v\n", NewPosts)
	spage := c.DefaultQuery("page", "1")
	fmt.Printf("cur page:%s\n", spage)
	page, _ := strconv.Atoi(spage)
	nums := len(NewPosts)
	allpage := nums / 5
	if nums%5 != 0 {
		allpage = nums/5 + 1
	}
	fmt.Printf("all page num:%d\n", allpage)
	posts := NewPosts
	if (page * 5) < nums {
		posts = NewPosts[(page-1)*5 : page*5]
	} else {
		posts = NewPosts[(page-1)*5 : nums]
	}
	tabs := make([]int, allpage+2) //分页表
	if (page - 1) == 0 {
		tabs[0] = 1
	} else {
		tabs[0] = page - 1
	}
	for i := 1; i <= allpage; i++ {
		tabs[i] = i
	}
	if page+1 <= allpage {
		tabs[allpage+1] = page + 1
	} else {
		tabs[allpage+1] = 1
	}
	fmt.Printf("tabs:%#v\n", tabs)
	conf.Stat.ToCnt++
	ip := c.ClientIP()
	fmt.Printf("client ip:%s\n", ip)
	//conf.Stat.IPs = append(conf.Stat.IPs, ip)//不统计IP了。除了耗空间，没多大意义
	c.HTML(http.StatusOK, "index.html", gin.H{"post": posts, "items": conf.Item.Items, "about": conf.Abt, "newcmts": NewCmts, "newart": NewArts, "hotart": HotArts, "vistcnt": conf.Stat.ToCnt, "curpage": page, "tabs": tabs})
}

// strTrip 去除字符串的空格和换行
func strTrip(src string) string {
	str := strings.Replace(src, " ", "", -1)
	str = strings.Replace(str, "\r", "", -1)
	str = strings.Replace(str, "\n", "", -1)
	return str
}

func getPosts() []Post {
	a := []Post{}
	conf.Item.Items = conf.Item.Items[0:0] //先清空
	files, _ := filepath.Glob("posts/*")
	for i, f := range files {
		fmt.Println(i)
		fmt.Println(f)

		file := f
		file = filepath.ToSlash(f) //转换路径分隔符，注意windows和linux下路径分隔符不同,所以需要转换支持不同平台
		fmt.Println(file)
		file = strings.Replace(file, "posts/", "", -1)
		//fmt.Println(file)
		file = strings.Replace(file, ".md", "", -1)
		fileread, _ := ioutil.ReadFile(f)
		lines := strings.Split(string(fileread), "\n")
		title := string(lines[0])
		date := strTrip(string(lines[1]))
		summary := string(lines[2])
		imgfile := string(lines[3])
		item := strTrip(string(lines[4]))
		author := string(lines[5])

		imgfile = strTrip(imgfile)
		body := strings.Join(lines[6:len(lines)], "\n")
		//body := strings.Join(lines[4:len(lines)], "\n")
		//fmt.Println(imgfile)
		id := utils.MD5Str(file)
		ar := ArticleRoute{item, f}
		mapLock.Lock()
		artRouteMap[id] = ar
		mapLock.Unlock()
		itemcount := len(conf.Item.Items)
		//查找分类看是否已存在
		if itemcount == 0 {
			conf.Item.Items = append(conf.Item.Items, item)
		} else {
			k := 0
			for k = 0; k < itemcount; k++ {
				if conf.Item.Items[k] == item {
					break
				}
			}
			if k >= itemcount {
				//分类之前未存在，添加分类
				conf.Item.Items = append(conf.Item.Items, item)
				itemcount = len(conf.Item.Items)
			}
		}

		fmt.Printf("item is:" + item + "\n")
		fmt.Printf("file name:%s,id=%s\n", file, id)
		art := conf.Articles{id, item, title, date, summary, body, imgfile, author, 0, 0}
		mapLock.Lock()
		if conf.Art.ArticlesMap[item] == nil {
			conf.Art.ArticlesMap[item] = make(map[string]*conf.Articles)
		}
		_, exists := conf.Art.ArticlesMap[item][id]
		if !exists {
			conf.Art.ArticlesMap[item][id] = &art
		} else {
			art = conf.Articles{id, item, title, date, summary, body, imgfile, author, conf.Art.ArticlesMap[item][id].CmtCnt, conf.Art.ArticlesMap[item][id].VistCnt}
			conf.Art.ArticlesMap[item][id] = &art
		}
		mapLock.Unlock()

		a = append(a, Post{id, title, date, summary, body, file, imgfile, item, author, nil, conf.Art.ArticlesMap[item][id].CmtCnt, conf.Art.ArticlesMap[item][id].VistCnt})
	}
	conf.Art.Save()
	conf.Item.Save()
	//fmt.Printf("%#v\n", conf.Art.ArticlesMap)
	//fmt.Printf("%#v\n", articlesMap)
	return a
}

func handleArticles(c *gin.Context) {
	id := c.Param("id")
	//mapLock.Lock()
	file := artRouteMap[id].Name
	item := artRouteMap[id].Item
	fmt.Println(id)
	fmt.Println(file)
	art := conf.Art.ArticlesMap[item][id] 
	//fmt.Println(art)
	//mapLock.Unlock()
	title := art.Title
	date := art.Date
	summary := art.Summary
	imgfile := art.ImgFile
	author := art.Author
	imgfile = strTrip(imgfile)
	body := art.File
	//fmt.Println(body)
	//fmt.Printf("%#v\n", conf.Cmt.Comts)
	cmts := []conf.Comment{}
	count := 0 //评论数量
	for i, v := range conf.Cmt.Comts {
		//fmt.Printf("%#v\n", c)
		if v.ID == id {
			cmts = append(cmts, conf.Cmt.Comts[i])
			count++
		}
	}
	//文章浏览量++
	//mapLock.Lock()
	//art := conf.Art.ArticlesMap[item][id]
	//art.VistCnt++ //浏览数量加一
	conf.Art.ArticlesMap[item][id].VistCnt++
	//mapLock.Unlock()

	go RefreshData()

	p := Post{id, title, date, summary, body, file, imgfile, item, author, cmts, conf.Art.ArticlesMap[item][id].CmtCnt, conf.Art.ArticlesMap[item][id].VistCnt}
	c.HTML(http.StatusOK, "article.html", gin.H{"post": p, "items": conf.Item.Items, "cmtcounts": count, "newcmts": NewCmts, "newart": NewArts, "hotart": HotArts, "vistcnt": conf.Stat.ToCnt})
}

func handleItems(c *gin.Context) {
	sid := c.Param("id")
	id, _ := strconv.Atoi(sid)
	//mapLock.Lock()
	all := conf.Art.ArticlesMap[conf.Item.Items[id]]
	//mapLock.Unlock()
	fmt.Printf("%#v\n", all)
	posts := NewArticles{}
	// 遍历 hash
	for _, value := range all {
		posts = append(posts, *value)
	}
	//日期排序
	sort.Sort(posts)
	//支持分页
	spage := c.DefaultQuery("page", "1")
	fmt.Printf("cur page:%s\n", spage)
	page, _ := strconv.Atoi(spage)
	nums := len(posts)
	allpage := nums / 5
	if nums%5 != 0 {
		allpage = nums/5 + 1
	}
	fmt.Printf("all page num:%d\n", allpage)
	curposts := posts
	if (page * 5) < nums {
		curposts = posts[(page-1)*5 : page*5]
	} else {
		curposts = posts[(page-1)*5 : nums]
	}
	tabs := make([]int, allpage+2) //分页表
	if (page - 1) == 0 {
		tabs[0] = 1
	} else {
		tabs[0] = page - 1
	}
	for i := 1; i <= allpage; i++ {
		tabs[i] = i
	}
	if page+1 <= allpage {
		tabs[allpage+1] = page + 1
	} else {
		tabs[allpage+1] = 1
	}
	fmt.Printf("tabs:%#v\n", tabs)
	c.HTML(http.StatusOK, "items.html", gin.H{"post": curposts, "items": conf.Item.Items, "newcmts": NewCmts, "newart": NewArts, "hotart": HotArts, "vistcnt": conf.Stat.ToCnt, "curitem": id, "curpage": page, "tabs": tabs})
}

func handlePostComment(c *gin.Context) {

	sid := c.PostForm("id")    //文章ID
	item := c.PostForm("item") //文章分类
	item = strTrip(item)
	title := c.PostForm("title")   //文章标题
	text := c.PostForm("text")     //评论内容
	author := c.PostForm("author") //评论者
	email := c.PostForm("email")   //邮箱
	url := c.PostForm("url")       //评论者网址
	time := time.Now().Format("2006-01-02 15:04:05")

	fmt.Println(sid, item, text, author, email, url, time)

	//关于我 页面的留言功能
	if sid == "about" {
		href := "/about"
		notice := Notice{"留言成功", true, 3, href}
		cmt := conf.Comment{sid, item, title, author, email, text, time, url}
		fmt.Printf("%#v\n", cmt)
		conf.Cmt.Comts = append(conf.Cmt.Comts, cmt)
		conf.Cmt.Save()
		c.HTML(http.StatusOK, "success.html", gin.H{"notice": notice})
		return
	}
	href := "/article/" + sid
	notice := Notice{"提交成功", true, 3, href}
	conf.Art.ArticlesMap[item][sid].CmtCnt++ //评论数量加一
	//fmt.Printf("%#v\n", conf.Art.ArticlesMap[item][sid])
	fmt.Println(conf.Art.ArticlesMap[item][sid].CmtCnt)
	//fmt.Printf("%#v\n", conf.Art.ArticlesMap[item])
	cmt := conf.Comment{sid, item, title, author, email, text, time, url}
	fmt.Printf("%#v\n", cmt)
	conf.Cmt.Comts = append(conf.Cmt.Comts, cmt)
	conf.Cmt.Save()
	//conf.Art.Save()

	go RefreshData()

	c.HTML(http.StatusOK, "success.html", gin.H{"notice": notice})
}

func handleAbout(c *gin.Context) {
	cmts := []conf.Comment{}
	count := 0 //评论数量
	for i, v := range conf.Cmt.Comts {
		//fmt.Printf("%#v\n", c)
		if v.ID == "about" {
			cmts = append(cmts, conf.Cmt.Comts[i])
			count++
		}
	}
	c.HTML(http.StatusOK, "about.html", gin.H{"items": conf.Item.Items, "cmtcounts": count, "cmts": cmts, "newcmts": NewCmts, "newart": NewArts, "hotart": HotArts, "vistcnt": conf.Stat.ToCnt})
}

func handleLiuYan(c *gin.Context) {
	cmts := []conf.Comment{}
	count := 0 //评论数量
	for i, v := range conf.Cmt.Comts {
		//fmt.Printf("%#v\n", c)
		if v.ID == "about" {
			cmts = append(cmts, conf.Cmt.Comts[i])
			count++
		}
	}
	c.HTML(http.StatusOK, "liuyan.html", gin.H{"items": conf.Item.Items, "cmtcounts": count, "cmts": cmts, "newcmts": NewCmts, "newart": NewArts, "hotart": HotArts, "vistcnt": conf.Stat.ToCnt})
}

// Add ...模板里使用加
func Add(a, b int) int {
	return a + b
}

// Dec ...模板里使用减
func Dec(a, b int) int {
	return a - b
}

func main() {

	pid := syscall.Getpid()
	log.Println("pid : ", pid)
	//创建监听退出chan
	c := make(chan os.Signal, 1)
	defer close(c)
	//监听指定信号 ctrl+c kill
	signal.Notify(c, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
	go NotifyAndExit(c)

	router := gin.Default()
	//增加几个模板自定义函数Add Dec
	router.SetFuncMap(template.FuncMap{
		"add": Add,
		"dec": Dec,
	})

	//关于
	conf.Abt.Name = "一米阳光"
	conf.Abt.Jobs = "嵌入式 linux | Android | go web"
	conf.Abt.WX = "yongzhen1111"
	conf.Abt.QQ = "534117529"
	conf.Abt.Email = "534117529@qq.com"
	conf.Abt.Save()

	//加载评论
	conf.Cmt.Load()
	//加载文章
	conf.Art.Load()
	//加载统计信息
	conf.Stat.Load()

	//文件目录监控:增加，修改，删除
	w := watcher.New()
	w.FilterOps(watcher.Create, watcher.Write, watcher.Remove)
	go func() {
		for {
			select {
			case event := <-w.Event:
				//fmt.Println(event) // Print the event's info.
				if !event.IsDir() {
					fmt.Println(event) // Print the event's info.
					//fmt.Println(event.Op)
					switch event.Op {
					case watcher.Create:
						getPosts()
						RefreshData()
						break
					case watcher.Write:
						getPosts()
						RefreshData()
						break
					case watcher.Remove:
						name := strings.Replace(event.Name(), ".md", "", -1)
						//fmt.Println(name)
						id := utils.MD5Str(name)
						mapLock.Lock()
						item := artRouteMap[id].Item
						//删除文件(文章)信息
						delete(artRouteMap, id)
						delete(conf.Art.ArticlesMap[item], id)
						mapLock.Unlock()
						getPosts()
						RefreshData()
						break

					}
				}
			case err := <-w.Error:
				log.Fatalln(err)
			case <-w.Closed:
				return
			}
		}
	}()
	// Watch this folder for changes.
	if err := w.Add("./posts"); err != nil {
		log.Fatalln(err)
	}
	// Trigger 3 events after watcher started.
	go func() {
		w.Wait()
		w.TriggerEvent(watcher.Create, nil)
		w.TriggerEvent(watcher.Write, nil)
		w.TriggerEvent(watcher.Remove, nil)
	}()
	go func() {
		// Start the watching process - it'll check for changes every 1000ms.
		if err := w.Start(time.Millisecond * 1000); err != nil {
			log.Fatalln(err)
		}
	}()

	artRouteMap = make(map[string]ArticleRoute)

	getPosts()
	//更新文章排序和浏览量
	RefreshData()
	//静态文件
	router.Static("/assets", "./static")
	//渲染html页面
	router.LoadHTMLGlob("views/*")
	router.GET("/", handleIndex)
	router.GET("/about", handleAbout)
	router.GET("/liuyan", handleLiuYan)
	router.GET("/article/:id", handleArticles)
	router.GET("/items/:id", handleItems)
	router.POST("/comment", handlePostComment)
	//运行的端口
	router.Run(":8000")
}

func NotifyAndExit(c <-chan os.Signal) {
	for s := range c {
	 switch s {
	 	case syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT: // ctrl + c
		 log.Println("程序退出", s)
		 conf.Art.Save()
		 os.Exit(0)
		default:
	 }
	}
}