go-wxhelper/main.go

86 lines
1.9 KiB
Go
Raw Permalink Normal View History

2023-09-21 17:33:59 +08:00
package main
import (
"github.com/gin-gonic/gin"
2023-10-11 09:21:07 +08:00
"go-wechat/config"
"go-wechat/initialization"
2024-02-19 14:17:26 +08:00
"go-wechat/mq"
"go-wechat/router"
2023-09-21 17:33:59 +08:00
"go-wechat/tasks"
"go-wechat/tcpserver"
"go-wechat/utils"
2023-11-30 17:31:50 +08:00
"html/template"
2023-09-21 17:33:59 +08:00
"log"
2023-11-30 17:31:50 +08:00
"net/http"
"strings"
"time"
2023-09-21 17:33:59 +08:00
)
func init() {
initialization.InitConfig() // 初始化配置
initialization.InitWechatRobotInfo() // 初始化机器人信息
2023-12-11 10:44:23 +08:00
initialization.Plugin() // 注册插件
tasks.InitTasks() // 初始化定时任务
2024-02-19 14:17:26 +08:00
mq.Init() // 初始化MQ
2023-09-21 17:33:59 +08:00
}
func main() {
// 如果启用了自动配置回调,就设置一下
if config.Conf.Wechat.AutoSetCallback {
utils.ClearCallback()
time.Sleep(500 * time.Millisecond) // 休眠五百毫秒再设置
utils.SetCallback(config.Conf.Wechat.Callback)
}
// 启动TCP服务
go tcpserver.Start()
2023-09-21 17:33:59 +08:00
// 启动HTTP服务
app := gin.Default()
2023-11-30 17:31:50 +08:00
// 自定义模板引擎函数
app.SetFuncMap(template.FuncMap{
"codeToChinese": func(code string) string {
switch code {
case "friend":
return "好友列表"
case "group":
return "群组列表"
case "index":
return "首页"
2024-06-18 17:02:37 +08:00
case "assistant":
return "AI角色"
default:
return "其他页面"
}
},
"boolToChinese": func(flag bool) string {
2023-11-30 17:31:50 +08:00
if flag {
return "是"
2023-11-30 17:31:50 +08:00
}
return "否"
2023-11-30 17:31:50 +08:00
},
})
app.LoadHTMLGlob("views/*.html")
2023-11-30 14:20:48 +08:00
app.Static("/assets", "./views/static")
app.StaticFile("/favicon.ico", "./views/wechat.ico")
// 404返回数据
app.NoRoute(func(ctx *gin.Context) {
2023-11-30 17:31:50 +08:00
if strings.HasPrefix(ctx.Request.URL.Path, "/api") {
ctx.String(404, "接口不存在")
return
}
// 404直接跳转到首页
ctx.Redirect(302, "/404.html")
})
2023-11-30 17:31:50 +08:00
app.NoMethod(func(ctx *gin.Context) {
ctx.String(http.StatusMethodNotAllowed, "不支持的请求方式")
})
// 初始化路由
router.Init(app)
if err := app.Run(":8080"); err != nil {
log.Panicf("服务启动失败:%v", err)
2023-09-21 17:33:59 +08:00
}
}