OpenKF/server/internal/router/router.go

// Copyright © 2023 OpenIM open source community. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package router

import (
	swaggerFiles "github.com/swaggo/files"
	ginSwagger "github.com/swaggo/gin-swagger"

	_ "github.com/OpenIMSDK/OpenKF/server/docs"

	"github.com/gin-gonic/gin"

	"github.com/OpenIMSDK/OpenKF/server/internal/api"
	"github.com/OpenIMSDK/OpenKF/server/internal/config"
	"github.com/OpenIMSDK/OpenKF/server/internal/middleware"
	urltrie "github.com/OpenIMSDK/OpenKF/server/internal/middleware/hooks/url_trie"
)

// InitRouter init router.
func InitRouter() *gin.Engine {
	if config.GetString("app.debug") == "true" {
		gin.SetMode(gin.DebugMode)
	} else {
		gin.SetMode(gin.ReleaseMode)
	}

	r := gin.Default()
	r.Use(urltrie.RunHook(), middleware.EnableCROS())

	// swagger
	r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))

	apiv1 := r.Group("/api/v1")
	{
		register := apiv1.Group("/register")
		{
			register.POST("/email/code", api.SendCode)
			// register.POST("/github", api.GithubRegister)
		}

		// OpenIM callback api
		command := apiv1.Group("/openim/callback")
		{
			command.POST("/", api.OpenIMCallback)
			command.POST("/callbackBeforeSendSingleMsgCommand", api.BeforeSendSingleMsg)
			command.POST("/callbackAfterSendSingleMsgCommand", api.AfterSendSingleMsg)
			command.POST("/callbackMsgModifyCommand", api.MsgModify)
			command.POST("/callbackUserOnlineCommand", api.UserOnline)
			command.POST("/callbackUserOfflineCommand", api.UserOffline)
			command.POST("/callbackOfflinePushCommand", api.OfflinePush)
			command.POST("/callbackOnlinePushCommand", api.OnlinePush)
		}
	}

	return r
}