main.go

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"strconv"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/imshuai/serverchan"
)

const numPerPage = 25

var wnotice = serverchan.NewServerChan(func() string {
	byts, err := ioutil.ReadFile("serverchan.token")
	if err != nil {
		fatal("read serverchan.token fail with error:", err)
		os.Exit(1)
	}
	return string(byts)
}())

var newData = false

func check() {
	t := time.NewTicker(time.Hour * 24)
	for {
		select {
		case <-t.C:
			if newData {
				conn := rdb.Get()
				conn.Do("ZINTERSTORE", setTemp1, 2, listChecked, listUnchecked)
				conn.Do("ZUNIONSTORE", setTemp2, 2, listUnchecked, setTemp1, "WEIGHTS", 1, 0, "AGGREGATE", "MIN")
				conn.Do("RENAME", setTemp2, listUnchecked)
				conn.Close()

				count := getPicNum(listUnchecked)
				if err != nil {
					errorlog("check count of", listUnchecked, "fail with error:", err)
					continue
				}
				wnotice.Send("GirlPic有新图片需要审核了", fmt.Sprintf("\n\n接收到%d张新图片，尽快审核放出", count))
				newData = false
			}
		}
	}
}

func main() {
	go check()

	e := gin.Default()

	e.Static("/static", "./static")
	e.StaticFile("/favicon.ico", "./static/favicon.ico")
	e.LoadHTMLGlob("./tmpls/*")

	e.GET("/", func(c *gin.Context) {
		c.HTML(http.StatusOK, "index.html", nil)
	})

	e.POST("/data/pics", func(c *gin.Context) {
		pics := make([]string, 0)
		err := c.Bind(&pics)
		if err != nil {
			errorlog("bind post pics to []string fail with error:", err)
			c.String(200, "fail with error:%v", err)
			return
		}
		conn := rdb.Get()
		defer conn.Close()
		for _, v := range pics {
			conn.Send("ZADD", listUnchecked, 0, v)
		}
		_, err = conn.Do("")
		if err != nil {
			errorlog("store post pics fail with error:", err)
			c.String(200, "%v", "fail")
			return
		}
		//debug("store post pics success, return data:", r)
		newData = true
		c.String(200, "%v", "ok")
	})

	e.GET("/page/:i", func(c *gin.Context) {
		page, err := strconv.Atoi(c.Param("i"))
		if err != nil {
			info("pass wrong page param to route")
			c.AbortWithStatus(http.StatusBadRequest)
			return
		}
		pics := getPics(page)
		c.JSON(http.StatusOK, pics)
	})
	manage := e.Group("/review", gin.BasicAuth(gin.Accounts{
		"admin": "shuai6563",
	}))
	manage.GET("/", func(c *gin.Context) {
		c.HTML(http.StatusOK, "review.html", nil)
	})
	manage.GET("/next", func(c *gin.Context) {
		if c.Request.Header.Get("X-Requested-With") == "XMLHttpRequest" {
			pics := getPicWaitReview()
			c.JSON(http.StatusOK, pics)
		} else {
			c.AbortWithStatus(http.StatusMethodNotAllowed)
		}
	})
	manage.GET("/delete/:id", func(c *gin.Context) {
		if c.Request.Header.Get("X-Requested-With") == "XMLHttpRequest" {
			id, err := strconv.Atoi(c.Param("id"))
			if err != nil {
				info("pass wrong id param to route")
				c.AbortWithStatus(http.StatusBadRequest)
				return
			}
			conn := rdb.Get()
			defer conn.Close()
			conn.Do("LSET", listTemp, id, "deleted")
			c.JSON(http.StatusOK, gin.H{
				"effect": 1,
			})
		} else {
			c.AbortWithStatus(http.StatusMethodNotAllowed)
		}
	})
	e.Run(":34533")
}