Open-IM-Server/internal/api/route.go

package api

import (
	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/mw"
	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/prome"
	"github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry"
	"github.com/gin-gonic/gin"
	"github.com/gin-gonic/gin/binding"
	"github.com/go-playground/validator/v10"
	"github.com/go-redis/redis/v8"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
)

func NewGinRouter(zk discoveryregistry.SvcDiscoveryRegistry, rdb redis.UniversalClient) *gin.Engine {
	zk.AddOption(mw.GrpcClient(), grpc.WithTransportCredentials(insecure.NewCredentials())) // 默认RPC中间件
	gin.SetMode(gin.ReleaseMode)
	//f, _ := os.Create("../logs/api.log")
	//gin.DefaultWriter = io.MultiWriter(f)
	//gin.SetMode(gin.DebugMode)
	r := gin.New()
	if v, ok := binding.Validator.Engine().(*validator.Validate); ok {
		_ = v.RegisterValidation("required_if", RequiredIf)
	}
	log.Info("load config: ", config.Config)
	r.Use(gin.Recovery(), mw.CorsHandler(), mw.GinParseOperationID())
	if config.Config.Prometheus.Enable {
		prome.NewApiRequestCounter()
		prome.NewApiRequestFailedCounter()
		prome.NewApiRequestSuccessCounter()
		r.Use(prome.PrometheusMiddleware)
		r.GET("/metrics", prome.PrometheusHandler())
	}
	userRouterGroup := r.Group("/user")
	{
		u := NewUser(zk)
		userRouterGroupChild := mw.NewRouterGroup(userRouterGroup, "")
		userRouterGroupChildToken := mw.NewRouterGroup(userRouterGroup, "", mw.WithGinParseToken(rdb))
		userRouterGroupChild.POST("/user_register", u.UserRegister)
		userRouterGroupChildToken.POST("/update_user_info", u.UpdateUserInfo) //1
		userRouterGroupChildToken.POST("/set_global_msg_recv_opt", u.SetGlobalRecvMessageOpt)
		userRouterGroupChildToken.POST("/get_users_info", u.GetUsersPublicInfo) //1
		userRouterGroupChildToken.POST("/get_all_users_uid", u.GetAllUsersID)   // todo
		userRouterGroupChildToken.POST("/account_check", u.AccountCheck)        // todo
		userRouterGroupChildToken.POST("/get_users", u.GetUsers)
	}
	//friend routing group
	friendRouterGroup := r.Group("/friend")
	{
		f := NewFriend(zk)
		friendRouterGroup.Use(mw.GinParseToken(rdb))
		friendRouterGroup.POST("/delete_friend", f.DeleteFriend)                  //1
		friendRouterGroup.POST("/get_friend_apply_list", f.GetFriendApplyList)    //1
		friendRouterGroup.POST("/get_self_friend_apply_list", f.GetSelfApplyList) //1
		friendRouterGroup.POST("/get_friend_list", f.GetFriendList)               //1
		friendRouterGroup.POST("/add_friend", f.ApplyToAddFriend)                 //1
		friendRouterGroup.POST("/add_friend_response", f.RespondFriendApply)      //1
		friendRouterGroup.POST("/set_friend_remark", f.SetFriendRemark)           //1
		friendRouterGroup.POST("/add_black", f.AddBlack)                          //1
		friendRouterGroup.POST("/get_black_list", f.GetPaginationBlacks)          //1
		friendRouterGroup.POST("/remove_black", f.RemoveBlack)                    //1
		friendRouterGroup.POST("/import_friend", f.ImportFriends)                 //1
		friendRouterGroup.POST("/is_friend", f.IsFriend)                          //1
	}
	groupRouterGroup := r.Group("/group")
	g := NewGroup(zk)
	{
		groupRouterGroup.Use(mw.GinParseToken(rdb))
		groupRouterGroup.POST("/create_group", g.NewCreateGroup)                                //1
		groupRouterGroup.POST("/set_group_info", g.NewSetGroupInfo)                             //1
		groupRouterGroup.POST("/join_group", g.JoinGroup)                                       //1
		groupRouterGroup.POST("/quit_group", g.QuitGroup)                                       //1
		groupRouterGroup.POST("/group_application_response", g.ApplicationGroupResponse)        //1
		groupRouterGroup.POST("/transfer_group", g.TransferGroupOwner)                          //1
		groupRouterGroup.POST("/get_recv_group_applicationList", g.GetRecvGroupApplicationList) //1
		groupRouterGroup.POST("/get_user_req_group_applicationList", g.GetUserReqGroupApplicationList)
		groupRouterGroup.POST("/get_groups_info", g.GetGroupsInfo) //1
		groupRouterGroup.POST("/kick_group", g.KickGroupMember)    //1
		//groupRouterGroup.POST("/get_group_all_member_list", g.GetGroupAllMemberList) //1
		groupRouterGroup.POST("/get_group_members_info", g.GetGroupMembersInfo) //1
		groupRouterGroup.POST("/invite_user_to_group", g.InviteUserToGroup)     //1
		groupRouterGroup.POST("/get_joined_group_list", g.GetJoinedGroupList)
		groupRouterGroup.POST("/dismiss_group", g.DismissGroup) //
		groupRouterGroup.POST("/mute_group_member", g.MuteGroupMember)
		groupRouterGroup.POST("/cancel_mute_group_member", g.CancelMuteGroupMember) //MuteGroup
		groupRouterGroup.POST("/mute_group", g.MuteGroup)
		groupRouterGroup.POST("/cancel_mute_group", g.CancelMuteGroup)
		//groupRouterGroup.POST("/set_group_member_nickname", g.SetGroupMemberNickname)
		groupRouterGroup.POST("/set_group_member_info", g.SetGroupMemberInfo)
		groupRouterGroup.POST("/get_group_abstract_info", g.GetGroupAbstractInfo)
	}
	superGroupRouterGroup := r.Group("/super_group")
	{
		superGroupRouterGroup.Use(mw.GinParseToken(rdb))
		superGroupRouterGroup.POST("/get_joined_group_list", g.GetJoinedSuperGroupList)
		superGroupRouterGroup.POST("/get_groups_info", g.GetSuperGroupsInfo)
	}
	////certificate
	authRouterGroup := r.Group("/auth")
	{
		a := NewAuth(zk)
		u := NewUser(zk)
		authRouterGroupChild := mw.NewRouterGroup(authRouterGroup, "")
		authRouterGroupChildToken := mw.NewRouterGroup(authRouterGroup, "", mw.WithGinParseToken(rdb))
		authRouterGroupChild.POST("/user_register", u.UserRegister)    //1
		authRouterGroupChild.POST("/user_token", a.UserToken)          //1
		authRouterGroupChildToken.POST("/parse_token", a.ParseToken)   //1
		authRouterGroupChildToken.POST("/force_logout", a.ForceLogout) //1
	}
	////Third service
	thirdGroup := r.Group("/third")
	{
		t := NewThird(zk)
		thirdGroup.Use(mw.GinParseToken(rdb))
		thirdGroup.POST("/get_rtc_invitation_info", t.GetSignalInvitationInfo)
		thirdGroup.POST("/get_rtc_invitation_start_app", t.GetSignalInvitationInfoStartApp)
		thirdGroup.POST("/fcm_update_token", t.FcmUpdateToken)
		thirdGroup.POST("/set_app_badge", t.SetAppBadge)

		thirdGroup.POST("/apply_put", t.ApplyPut)
		thirdGroup.POST("/get_put", t.GetPut)
		thirdGroup.POST("/confirm_put", t.ConfirmPut)
		thirdGroup.POST("/get_url", t.GetURL)
		thirdGroup.GET("/get_url", t.GetURL)
	}
	////Message
	msgGroup := r.Group("/msg")
	{
		m := NewMsg(zk)
		msgGroup.Use(mw.GinParseToken(rdb))
		msgGroup.POST("/newest_seq", m.GetSeq)
		msgGroup.POST("/send_msg", m.SendMessage)
		msgGroup.POST("/pull_msg_by_seq", m.PullMsgBySeqs)
		msgGroup.POST("/del_msg", m.DelMsg)
		msgGroup.POST("/del_super_group_msg", m.DelSuperGroupMsg)
		msgGroup.POST("/clear_msg", m.ClearMsg)

		msgGroup.POST("/batch_send_msg", m.ManagementBatchSendMsg)
		msgGroup.POST("/check_msg_is_send_success", m.CheckMsgIsSendSuccess)
		msgGroup.POST("/get_users_online_status", m.GetUsersOnlineStatus)
		msgGroup.POST("/account_check", m.AccountCheck)
		//msgGroup.POST("/set_message_reaction_extensions", msg.SetMessageReactionExtensions)
		//msgGroup.POST("/get_message_list_reaction_extensions", msg.GetMessageListReactionExtensions)
		//msgGroup.POST("/add_message_reaction_extensions", msg.AddMessageReactionExtensions)
		//msgGroup.POST("/delete_message_reaction_extensions", msg.DeleteMessageReactionExtensions)
	}
	////Conversation
	conversationGroup := r.Group("/conversation")
	{
		c := NewConversation(zk)
		conversationGroup.Use(mw.GinParseToken(rdb))
		conversationGroup.POST("/get_all_conversations", c.GetAllConversations)
		conversationGroup.POST("/get_conversation", c.GetConversation)
		conversationGroup.POST("/get_conversations", c.GetConversations)
		conversationGroup.POST("/set_conversation", c.SetConversation)
		conversationGroup.POST("/batch_set_conversation", c.BatchSetConversations)
		conversationGroup.POST("/set_recv_msg_opt", c.SetRecvMsgOpt)
		conversationGroup.POST("/modify_conversation_field", c.ModifyConversationField)
	}
	return r
}