package main

import (
	"WechatGateWay/global"
	"WechatGateWay/handle"
	"WechatGateWay/third_part"
	"fmt"
	"github.com/robfig/cron"
	"github.com/spf13/viper"
	"log"
	"math/rand"
	"net/http"
	"os"
	"time"
)

func init() {
	rand.Seed(time.Now().UnixNano())
	// 读取配置文件
	workDir, _ := os.Getwd()
	viper.SetConfigName("application")
	viper.SetConfigType("yml")
	viper.AddConfigPath(workDir)
	err := viper.ReadInConfig()
	if err != nil {
		fmt.Println("config file not found")
		os.Exit(1)
	}
	global.WechatCorpId = viper.GetString("wechat.CorpId")
	global.WechatToken = viper.GetString("wechat.AppToken")
	global.WechatEncodingAesKey = viper.GetString("wechat.EncodingAesKey")
	global.WechatSendSecret = viper.GetString("wechat.SendSecret")
	global.WechatSendAid = viper.GetString("wechat.SendAid")
	global.WechatImagePath = viper.GetString("wechat.ImagePath")
	// receive_id 企业应用的回调，表示corpid
	global.WxCrypt = global.NewWXBizMsgCrypt(global.WechatToken, global.WechatEncodingAesKey, global.WechatCorpId, global.XmlType)
	third_part.GetRemoteToken()
	//初始化数据库
	global.DB = global.InitDB()
	defer global.DB.Close()
	log.Println("数据库初始化成功")

	log.Println("server init success")
}
func main() {
	c := cron.New()
	//每30分钟执行一次
	c.AddFunc("0 */30 * * * *", func() { go third_part.GetRemoteToken() })
	c.Start()
	log.Println("任务注册成功")

	// 开启一个http服务器，接收来自企业微信的消息
	http.HandleFunc("/", handle.HandleTencent)
	port := viper.GetString("server.port")
	if port == "" {
		port = "8080"
	}
	log.Println("server start at port:", port)
	log.Fatalln(http.ListenAndServe(":"+port, nil))
}