smart_intersectionV2.0/server/initialize/router.go

package initialize

import (
	"github.com/gin-contrib/cors"
	"github.com/gin-gonic/gin"
	swaggerFiles "github.com/swaggo/files"
	ginSwagger "github.com/swaggo/gin-swagger"
	"net/http"
	"os"
	"server/docs"
	"server/global"
	"server/middleware"
	"server/router"
	"time"
)

type justFilesFilesystem struct {
	fs http.FileSystem
}

func (fs justFilesFilesystem) Open(name string) (http.File, error) {
	f, err := fs.fs.Open(name)
	if err != nil {
		return nil, err
	}

	stat, err := f.Stat()
	if stat.IsDir() {
		return nil, os.ErrPermission
	}

	return f, nil
}

// 初始化总路由

func Routers() *gin.Engine {
	Router := gin.New()
	Router.Use(gin.Recovery())
	if gin.Mode() == gin.DebugMode {
		Router.Use(gin.Logger())
	}

	// 关键：全局配置 CORS（必须在所有路由注册前添加）
	Router.Use(cors.New(cors.Config{
		// 1. 允许的前端域名：必须包含你的前端地址（http://localhost:8080）
		AllowOrigins: []string{"http://localhost:8080"},
		// 2. 允许的请求方法：HLS 请求是 GET，需包含；其他接口（POST）也需包含
		AllowMethods: []string{"GET", "POST", "OPTIONS", "DELETE", "PUT"},
		// 3. 允许的请求头：前端可能携带的头（如 Content-Type，若有 token 也需包含）
		AllowHeaders: []string{"Origin", "Content-Type", "x-token", "x-user-id"},
		// 4. 是否允许携带 Cookie（若前端需要传 Cookie 则设为 true）
		AllowCredentials: true,
		// 5. 预检请求的缓存时间（减少浏览器频繁发送预检请求）
		MaxAge: 12 * time.Hour,
	}))

	InstallPlugin(Router) // 安装插件
	systemRouter := router.RouterGroupApp.System
	exampleRouter := router.RouterGroupApp.Example
	projectRouter := router.RouterGroupApp.Project
	devicesRouter := router.RouterGroupApp.Devices
	// 如果想要不使用nginx代理前端网页，可以修改 web/.env.production 下的
	// VUE_APP_BASE_API = /
	// VUE_APP_BASE_PATH = http://localhost
	// 然后执行打包命令 npm run build。在打开下面3行注释
	//Router.Static("/favicon.ico", "./dist/favicon.ico")
	//Router.Static("/assets", "./dist/assets")   // dist里面的静态资源
	//Router.StaticFile("/", "./dist/index.html") // 前端网页入口页面

	Router.StaticFS(global.GVA_CONFIG.Local.StorePath, justFilesFilesystem{http.Dir(global.GVA_CONFIG.Local.StorePath)}) // Router.Use(middleware.LoadTls())  // 如果需要使用https 请打开此中间件 然后前往 core/server.go 将启动模式 更变为 Router.RunTLS("端口","你的cre/pem文件","你的key文件")
	// 跨域，如需跨域可以打开下面的注释
	// Router.Use(middleware.Cors()) // 直接放行全部跨域请求
	// Router.Use(middleware.CorsByRules()) // 按照配置的规则放行跨域请求
	//global.GVA_LOG.Info("use middleware cors")
	docs.SwaggerInfo.BasePath = global.GVA_CONFIG.System.RouterPrefix
	Router.GET(global.GVA_CONFIG.System.RouterPrefix+"/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
	global.GVA_LOG.Info("register swagger handler")
	// 方便统一添加路由组前缀 多服务器上线使用

	PublicGroup := Router.Group(global.GVA_CONFIG.System.RouterPrefix)
	{
		// 健康监测
		PublicGroup.GET("/health", func(c *gin.Context) {
			c.JSON(http.StatusOK, "ok")
		})
	}
	{
		systemRouter.InitBaseRouter(PublicGroup) // 注册基础功能路由 不做鉴权
	}
	PrivateGroup := Router.Group(global.GVA_CONFIG.System.RouterPrefix)
	PrivateGroup.Use(middleware.JWTAuth()).Use(middleware.CasbinHandler())
	{
		systemRouter.InitApiRouter(PrivateGroup, PublicGroup)    // 注册功能api路由
		systemRouter.InitJwtRouter(PrivateGroup)                 // jwt相关路由
		systemRouter.InitUserRouter(PrivateGroup)                // 注册用户路由
		systemRouter.InitMenuRouter(PrivateGroup)                // 注册menu路由
		systemRouter.InitCasbinRouter(PrivateGroup)              // 权限相关路由
		systemRouter.InitAuthorityRouter(PrivateGroup)           // 注册角色路由
		systemRouter.InitSysDictionaryRouter(PrivateGroup)       // 字典管理
		systemRouter.InitSysOperationRecordRouter(PrivateGroup)  // 操作记录
		systemRouter.InitSysDictionaryDetailRouter(PrivateGroup) // 字典详情管理
		systemRouter.InitAuthorityBtnRouterRouter(PrivateGroup)  // 字典详情管理
		systemRouter.InitStreamRouter(PrivateGroup, PublicGroup)
		exampleRouter.InitFileUploadAndDownloadRouter(PrivateGroup)
		projectRouter.InitProjectRouter(PrivateGroup) //注册项目管理路由
		devicesRouter.InitScreensRouter(PrivateGroup) //注册设备管理路由
		devicesRouter.InitProgramRouter(PrivateGroup)
		devicesRouter.InitSoundPeriodRouter(PrivateGroup)
		devicesRouter.InitVoiceRouter(PrivateGroup)
		devicesRouter.InitCameraRouter(PrivateGroup, PublicGroup)
		devicesRouter.InitCaptureVehiclesRouter(PrivateGroup, PublicGroup)
	}

	global.GVA_LOG.Info("router register success")
	return Router
}