package controllers

import (
	"car-service/extend/code"
	"car-service/extend/util"
	"car-service/models"
	"encoding/json"
	"fmt"
	"github.com/astaxie/beego"
	"net/http"
	"github.com/robfig/cron"
	"strconv"
	"strings"
	"time"
)

type RestrictionController struct {
	beego.Controller
}

func InitRestriction() {
	date := time.Now().Format("2006-01-02")
	var r models.Restriction
	err := r.SelectByDate(date)
	if err == nil {
		//今日已获取过限行信息
		todayRes = &RestrictionResult{
			Date: date,
			Week: r.Week,
			CityName: r.CityName,
			IsXianXing: r.IsXianXing,
		}
		arr := strings.Split(r.WeiHao, ",")
		if len(arr) >0 {
			for _,s := range arr {
				n, e := strconv.Atoi(s)
				if e ==nil {
					todayRes.XxWeiHao = append(todayRes.XxWeiHao,n)
				}
			}
		}
	} else {
		getRestrictionInfo()
	}
}

// @Title 查询限行尾号
// @Description 查询限行尾号
// @Success 200 {object} controllers.RestrictionResult
// @Failure 403 {string} json ""
// @router /restriction [get]
func (c *RestrictionController) Restriction() {
	resp := code.Code{}
	if todayRes !=nil {
		resp.Success= true
		resp.Status= http.StatusOK
		resp.Data= *todayRes
	} else {
		resp.Success= false
		resp.Status= http.StatusBadRequest
		resp.Data= "数据请求失败，请稍后重试"
	}
	c.Data["json"] = resp
	c.ServeJSON()
}

var todayRes *RestrictionResult

func Schedule() {
	c := cron.New()
	//每天凌晨5点执行一次，获取当日数据
	c.AddFunc("0 0 1 * * ?", func() {
		todayRes = nil
		getRestrictionInfo()
	})
	c.Start()
}

type RestrictionResult struct {
	Date 		string 		`json:"date"`
	Week 		string 		`json:"week"`
	CityName 	string		`json:"cityName"`
	IsXianXing 	int 		`json:"isxianxing"`
	XxWeiHao 	[]int		`json:"xxWeiHao"`
}

type JuHeResult struct {
	Reason 		string 					`json:"reason"`
	Result 		CityRestriction  		`json:"result"`
	ErrorCode 	int 					`json:"error_code"`
}

type CityRestriction struct {
	Date 		string 			`json:"date"`
	Week 		string 			`json:"week"`
	City 		string 			`json:"city"`
	CityName 	string 			`json:"cityname"`
	Des 		[]interface{}	`json:"des"`
	Fine 		string 			`json:"fine"`
	Remarks 	string 			`json:"remarks"`
	IsXianXing 	int 			`json:"isxianxing"`
	XXWeiHao 	[]int 			`json:"xxweihao"`
	Holiday 	string 			`json:"holiday"`
}

func getRestrictionInfo() {
	juHeAppKey := beego.AppConfig.String("juheweihaokey")
	url := "http://v.juhe.cn/xianxing/index?key="+ juHeAppKey +"&city=beijing&type=1"
	b, err := util.DoPostRequest(url, util.CONTENT_TYPE_JSON, nil, nil, nil)
	if err != nil {
		fmt.Println("err:", err)
		return
	}
	var result JuHeResult
	err = json.Unmarshal(b, &result)
	if err != nil {
		fmt.Println("unmarshal err:", err)
		return
	}
	if result.ErrorCode == 0 {
		todayRes = &RestrictionResult{
			Date: result.Result.Date,
			Week: result.Result.Week,
			CityName: result.Result.CityName,
			IsXianXing: result.Result.IsXianXing,
			XxWeiHao: result.Result.XXWeiHao,
		}
		rc := models.Restriction{
			Date: result.Result.Date,
			Week: result.Result.Week,
			CityName: result.Result.CityName,
			IsXianXing: result.Result.IsXianXing,
		}
		var arr []string
		for _,n := range result.Result.XXWeiHao {
			arr=append(arr, strconv.Itoa(n))
		}
		rc.WeiHao = strings.Join(arr, ",")
		rc.Insert()
	}
}