// 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 database

import (
	"context"
	"time"

	"go.mongodb.org/mongo-driver/bson/primitive"

	"git.imall.cloud/openim/chat/pkg/common/db/cache"
	"git.imall.cloud/openim/protocol/constant"
	"github.com/openimsdk/tools/db/mongoutil"
	"github.com/openimsdk/tools/db/pagination"
	"github.com/openimsdk/tools/db/tx"
	"github.com/redis/go-redis/v9"

	"git.imall.cloud/openim/chat/pkg/common/db/model/admin"
	admindb "git.imall.cloud/openim/chat/pkg/common/db/table/admin"
)

type AdminDatabaseInterface interface {
	GetAdmin(ctx context.Context, account string) (*admindb.Admin, error)
	GetAdminUserID(ctx context.Context, userID string) (*admindb.Admin, error)
	UpdateAdmin(ctx context.Context, userID string, update map[string]any) error
	ChangePassword(ctx context.Context, userID string, newPassword string) error
	ChangeOperationPassword(ctx context.Context, userID string, newPassword string) error
	ClearGoogleAuthKey(ctx context.Context, userID string) error
	AddAdminAccount(ctx context.Context, admin []*admindb.Admin) error
	DelAdminAccount(ctx context.Context, userIDs []string) error
	SearchAdminAccount(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.Admin, error)
	CreateApplet(ctx context.Context, applets []*admindb.Applet) error
	DelApplet(ctx context.Context, appletIDs []string) error
	GetApplet(ctx context.Context, appletID string) (*admindb.Applet, error)
	FindApplet(ctx context.Context, appletIDs []string) ([]*admindb.Applet, error)
	SearchApplet(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.Applet, error)
	FindOnShelf(ctx context.Context) ([]*admindb.Applet, error)
	UpdateApplet(ctx context.Context, appletID string, update map[string]any) error
	GetConfig(ctx context.Context) (map[string]string, error)
	SetConfig(ctx context.Context, cs map[string]string) error
	DelConfig(ctx context.Context, keys []string) error
	FindInvitationRegister(ctx context.Context, codes []string) ([]*admindb.InvitationRegister, error)
	DelInvitationRegister(ctx context.Context, codes []string) error
	UpdateInvitationRegister(ctx context.Context, code string, fields map[string]any) error
	CreatInvitationRegister(ctx context.Context, invitationRegisters []*admindb.InvitationRegister) error
	SearchInvitationRegister(ctx context.Context, keyword string, state int32, userIDs []string, codes []string, pagination pagination.Pagination) (int64, []*admindb.InvitationRegister, error)
	SearchIPForbidden(ctx context.Context, keyword string, state int32, pagination pagination.Pagination) (int64, []*admindb.IPForbidden, error)
	AddIPForbidden(ctx context.Context, ms []*admindb.IPForbidden) error
	FindIPForbidden(ctx context.Context, ms []string) ([]*admindb.IPForbidden, error)
	DelIPForbidden(ctx context.Context, ips []string) error
	FindDefaultFriend(ctx context.Context, userIDs []string) ([]string, error)
	AddDefaultFriend(ctx context.Context, ms []*admindb.RegisterAddFriend) error
	DelDefaultFriend(ctx context.Context, userIDs []string) error
	SearchDefaultFriend(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.RegisterAddFriend, error)
	FindDefaultGroup(ctx context.Context, groupIDs []string) ([]string, error)
	AddDefaultGroup(ctx context.Context, ms []*admindb.RegisterAddGroup) error
	DelDefaultGroup(ctx context.Context, groupIDs []string) error
	SearchDefaultGroup(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.RegisterAddGroup, error)
	CountTotalGroups(ctx context.Context) (int64, error)    // 统计群组总数
	CountTodayNewGroups(ctx context.Context) (int64, error) // 统计今天新建的群组数
	CountTotalFriends(ctx context.Context) (int64, error)   // 统计好友总数
	FindBlockInfo(ctx context.Context, userIDs []string) ([]*admindb.ForbiddenAccount, error)
	GetBlockInfo(ctx context.Context, userID string) (*admindb.ForbiddenAccount, error)
	BlockUser(ctx context.Context, f []*admindb.ForbiddenAccount) error
	DelBlockUser(ctx context.Context, userID []string) error
	SearchBlockUser(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.ForbiddenAccount, error)
	FindBlockUser(ctx context.Context, userIDs []string) ([]*admindb.ForbiddenAccount, error)
	SearchUserLimitLogin(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.LimitUserLoginIP, error)
	AddUserLimitLogin(ctx context.Context, ms []*admindb.LimitUserLoginIP) error
	DelUserLimitLogin(ctx context.Context, ms []*admindb.LimitUserLoginIP) error
	CountLimitUserLoginIP(ctx context.Context, userID string) (uint32, error)
	GetLimitUserLoginIP(ctx context.Context, userID string, ip string) (*admindb.LimitUserLoginIP, error)
	CacheToken(ctx context.Context, userID string, token string, expire time.Duration) error
	GetTokens(ctx context.Context, userID string) (map[string]int32, error)
	DeleteToken(ctx context.Context, userID string) error
	LatestVersion(ctx context.Context, platform string) (*admindb.Application, error)
	AddVersion(ctx context.Context, val *admindb.Application) error
	UpdateVersion(ctx context.Context, id primitive.ObjectID, update map[string]any) error
	DeleteVersion(ctx context.Context, id []primitive.ObjectID) error
	PageVersion(ctx context.Context, platforms []string, page pagination.Pagination) (int64, []*admindb.Application, error)
}

func NewAdminDatabase(cli *mongoutil.Client, rdb redis.UniversalClient) (AdminDatabaseInterface, error) {
	a, err := admin.NewAdmin(cli.GetDB())
	if err != nil {
		return nil, err
	}
	forbidden, err := admin.NewIPForbidden(cli.GetDB())
	if err != nil {
		return nil, err
	}
	forbiddenAccount, err := admin.NewForbiddenAccount(cli.GetDB())
	if err != nil {
		return nil, err
	}
	limitUserLoginIP, err := admin.NewLimitUserLoginIP(cli.GetDB())
	if err != nil {
		return nil, err
	}
	invitationRegister, err := admin.NewInvitationRegister(cli.GetDB())
	if err != nil {
		return nil, err
	}
	registerAddFriend, err := admin.NewRegisterAddFriend(cli.GetDB())
	if err != nil {
		return nil, err
	}
	registerAddGroup, err := admin.NewRegisterAddGroup(cli.GetDB())
	if err != nil {
		return nil, err
	}
	applet, err := admin.NewApplet(cli.GetDB())
	if err != nil {
		return nil, err
	}
	clientConfig, err := admin.NewClientConfig(cli.GetDB())
	if err != nil {
		return nil, err
	}
	application, err := admin.NewApplication(cli.GetDB())
	if err != nil {
		return nil, err
	}
	return &AdminDatabase{
		tx:                 cli.GetTx(),
		admin:              a,
		ipForbidden:        forbidden,
		forbiddenAccount:   forbiddenAccount,
		limitUserLoginIP:   limitUserLoginIP,
		invitationRegister: invitationRegister,
		registerAddFriend:  registerAddFriend,
		registerAddGroup:   registerAddGroup,
		applet:             applet,
		clientConfig:       clientConfig,
		application:        application,
		cache:              cache.NewTokenInterface(rdb),
	}, nil
}

type AdminDatabase struct {
	tx                 tx.Tx
	admin              admindb.AdminInterface
	ipForbidden        admindb.IPForbiddenInterface
	forbiddenAccount   admindb.ForbiddenAccountInterface
	limitUserLoginIP   admindb.LimitUserLoginIPInterface
	invitationRegister admindb.InvitationRegisterInterface
	registerAddFriend  admindb.RegisterAddFriendInterface
	registerAddGroup   admindb.RegisterAddGroupInterface
	applet             admindb.AppletInterface
	clientConfig       admindb.ClientConfigInterface
	application        admindb.ApplicationInterface
	cache              cache.TokenInterface
}

func (o *AdminDatabase) GetAdmin(ctx context.Context, account string) (*admindb.Admin, error) {
	return o.admin.Take(ctx, account)
}

func (o *AdminDatabase) GetAdminUserID(ctx context.Context, userID string) (*admindb.Admin, error) {
	return o.admin.TakeUserID(ctx, userID)
}

func (o *AdminDatabase) UpdateAdmin(ctx context.Context, userID string, update map[string]any) error {
	return o.admin.Update(ctx, userID, update)
}

func (o *AdminDatabase) ChangePassword(ctx context.Context, userID string, newPassword string) error {
	return o.admin.ChangePassword(ctx, userID, newPassword)
}

func (o *AdminDatabase) ChangeOperationPassword(ctx context.Context, userID string, newPassword string) error {
	return o.admin.ChangeOperationPassword(ctx, userID, newPassword)
}

func (o *AdminDatabase) ClearGoogleAuthKey(ctx context.Context, userID string) error {
	return o.admin.ClearGoogleAuthKey(ctx, userID)
}

func (o *AdminDatabase) AddAdminAccount(ctx context.Context, admins []*admindb.Admin) error {
	return o.admin.Create(ctx, admins)
}

func (o *AdminDatabase) DelAdminAccount(ctx context.Context, userIDs []string) error {
	return o.admin.Delete(ctx, userIDs)
}

func (o *AdminDatabase) SearchAdminAccount(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.Admin, error) {
	return o.admin.Search(ctx, keyword, pagination)
}

func (o *AdminDatabase) CreateApplet(ctx context.Context, applets []*admindb.Applet) error {
	return o.applet.Create(ctx, applets)
}

func (o *AdminDatabase) DelApplet(ctx context.Context, appletIDs []string) error {
	return o.applet.Del(ctx, appletIDs)
}

func (o *AdminDatabase) GetApplet(ctx context.Context, appletID string) (*admindb.Applet, error) {
	return o.applet.Take(ctx, appletID)
}

func (o *AdminDatabase) FindApplet(ctx context.Context, appletIDs []string) ([]*admindb.Applet, error) {
	return o.applet.FindID(ctx, appletIDs)
}

func (o *AdminDatabase) SearchApplet(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.Applet, error) {
	return o.applet.Search(ctx, keyword, pagination)
}

func (o *AdminDatabase) FindOnShelf(ctx context.Context) ([]*admindb.Applet, error) {
	return o.applet.FindOnShelf(ctx)
}

func (o *AdminDatabase) UpdateApplet(ctx context.Context, appletID string, update map[string]any) error {
	return o.applet.Update(ctx, appletID, update)
}

func (o *AdminDatabase) GetConfig(ctx context.Context) (map[string]string, error) {
	return o.clientConfig.Get(ctx)
}

func (o *AdminDatabase) SetConfig(ctx context.Context, cs map[string]string) error {
	return o.clientConfig.Set(ctx, cs)
}

func (o *AdminDatabase) DelConfig(ctx context.Context, keys []string) error {
	return o.clientConfig.Del(ctx, keys)
}

func (o *AdminDatabase) FindInvitationRegister(ctx context.Context, codes []string) ([]*admindb.InvitationRegister, error) {
	return o.invitationRegister.Find(ctx, codes)
}

func (o *AdminDatabase) DelInvitationRegister(ctx context.Context, codes []string) error {
	return o.invitationRegister.Del(ctx, codes)
}

func (o *AdminDatabase) UpdateInvitationRegister(ctx context.Context, code string, fields map[string]any) error {
	return o.invitationRegister.Update(ctx, code, fields)
}

func (o *AdminDatabase) CreatInvitationRegister(ctx context.Context, invitationRegisters []*admindb.InvitationRegister) error {
	return o.invitationRegister.Create(ctx, invitationRegisters)
}

func (o *AdminDatabase) SearchInvitationRegister(ctx context.Context, keyword string, state int32, userIDs []string, codes []string, pagination pagination.Pagination) (int64, []*admindb.InvitationRegister, error) {
	return o.invitationRegister.Search(ctx, keyword, state, userIDs, codes, pagination)
}

func (o *AdminDatabase) SearchIPForbidden(ctx context.Context, keyword string, state int32, pagination pagination.Pagination) (int64, []*admindb.IPForbidden, error) {
	return o.ipForbidden.Search(ctx, keyword, state, pagination)
}

func (o *AdminDatabase) AddIPForbidden(ctx context.Context, ms []*admindb.IPForbidden) error {
	return o.ipForbidden.Create(ctx, ms)
}

func (o *AdminDatabase) FindIPForbidden(ctx context.Context, ms []string) ([]*admindb.IPForbidden, error) {
	return o.ipForbidden.Find(ctx, ms)
}

func (o *AdminDatabase) DelIPForbidden(ctx context.Context, ips []string) error {
	return o.ipForbidden.Delete(ctx, ips)
}

func (o *AdminDatabase) FindDefaultFriend(ctx context.Context, userIDs []string) ([]string, error) {
	return o.registerAddFriend.FindUserID(ctx, userIDs)
}

func (o *AdminDatabase) AddDefaultFriend(ctx context.Context, ms []*admindb.RegisterAddFriend) error {
	return o.registerAddFriend.Add(ctx, ms)
}

func (o *AdminDatabase) DelDefaultFriend(ctx context.Context, userIDs []string) error {
	return o.registerAddFriend.Del(ctx, userIDs)
}

func (o *AdminDatabase) SearchDefaultFriend(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.RegisterAddFriend, error) {
	return o.registerAddFriend.Search(ctx, keyword, pagination)
}

func (o *AdminDatabase) FindDefaultGroup(ctx context.Context, groupIDs []string) ([]string, error) {
	return o.registerAddGroup.FindGroupID(ctx, groupIDs)
}

func (o *AdminDatabase) AddDefaultGroup(ctx context.Context, ms []*admindb.RegisterAddGroup) error {
	return o.registerAddGroup.Add(ctx, ms)
}

func (o *AdminDatabase) DelDefaultGroup(ctx context.Context, groupIDs []string) error {
	return o.registerAddGroup.Del(ctx, groupIDs)
}

func (o *AdminDatabase) SearchDefaultGroup(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.RegisterAddGroup, error) {
	return o.registerAddGroup.Search(ctx, keyword, pagination)
}

func (o *AdminDatabase) CountTotalGroups(ctx context.Context) (int64, error) {
	return o.registerAddGroup.CountTotal(ctx)
}

func (o *AdminDatabase) CountTodayNewGroups(ctx context.Context) (int64, error) {
	return o.registerAddGroup.CountToday(ctx)
}

func (o *AdminDatabase) CountTotalFriends(ctx context.Context) (int64, error) {
	return o.registerAddFriend.CountTotal(ctx)
}

func (o *AdminDatabase) FindBlockInfo(ctx context.Context, userIDs []string) ([]*admindb.ForbiddenAccount, error) {
	return o.forbiddenAccount.Find(ctx, userIDs)
}

func (o *AdminDatabase) GetBlockInfo(ctx context.Context, userID string) (*admindb.ForbiddenAccount, error) {
	return o.forbiddenAccount.Take(ctx, userID)
}

func (o *AdminDatabase) BlockUser(ctx context.Context, f []*admindb.ForbiddenAccount) error {
	return o.forbiddenAccount.Create(ctx, f)
}

func (o *AdminDatabase) DelBlockUser(ctx context.Context, userID []string) error {
	return o.forbiddenAccount.Delete(ctx, userID)
}

func (o *AdminDatabase) SearchBlockUser(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.ForbiddenAccount, error) {
	return o.forbiddenAccount.Search(ctx, keyword, pagination)
}

func (o *AdminDatabase) FindBlockUser(ctx context.Context, userIDs []string) ([]*admindb.ForbiddenAccount, error) {
	return o.forbiddenAccount.Find(ctx, userIDs)
}

func (o *AdminDatabase) SearchUserLimitLogin(ctx context.Context, keyword string, pagination pagination.Pagination) (int64, []*admindb.LimitUserLoginIP, error) {
	return o.limitUserLoginIP.Search(ctx, keyword, pagination)
}

func (o *AdminDatabase) AddUserLimitLogin(ctx context.Context, ms []*admindb.LimitUserLoginIP) error {
	return o.limitUserLoginIP.Create(ctx, ms)
}

func (o *AdminDatabase) DelUserLimitLogin(ctx context.Context, ms []*admindb.LimitUserLoginIP) error {
	return o.limitUserLoginIP.Delete(ctx, ms)
}

func (o *AdminDatabase) CountLimitUserLoginIP(ctx context.Context, userID string) (uint32, error) {
	return o.limitUserLoginIP.Count(ctx, userID)
}

func (o *AdminDatabase) GetLimitUserLoginIP(ctx context.Context, userID string, ip string) (*admindb.LimitUserLoginIP, error) {
	return o.limitUserLoginIP.Take(ctx, userID, ip)
}

func (o *AdminDatabase) CacheToken(ctx context.Context, userID string, token string, expire time.Duration) error {
	isSet, err := o.cache.AddTokenFlagNXEx(ctx, userID, token, constant.NormalToken, expire)
	if err != nil {
		return err
	}
	if !isSet {
		// already exists, update
		if err = o.cache.AddTokenFlag(ctx, userID, token, constant.NormalToken); err != nil {
			return err
		}
	}
	return nil
}

func (o *AdminDatabase) GetTokens(ctx context.Context, userID string) (map[string]int32, error) {
	return o.cache.GetTokensWithoutError(ctx, userID)
}

func (o *AdminDatabase) DeleteToken(ctx context.Context, userID string) error {
	return o.cache.DeleteTokenByUid(ctx, userID)
}

func (o *AdminDatabase) LatestVersion(ctx context.Context, platform string) (*admindb.Application, error) {
	return o.application.LatestVersion(ctx, platform)
}

func (o *AdminDatabase) AddVersion(ctx context.Context, val *admindb.Application) error {
	return o.application.AddVersion(ctx, val)
}

func (o *AdminDatabase) UpdateVersion(ctx context.Context, id primitive.ObjectID, update map[string]any) error {
	return o.application.UpdateVersion(ctx, id, update)
}

func (o *AdminDatabase) DeleteVersion(ctx context.Context, id []primitive.ObjectID) error {
	return o.application.DeleteVersion(ctx, id)
}

func (o *AdminDatabase) PageVersion(ctx context.Context, platforms []string, page pagination.Pagination) (int64, []*admindb.Application, error) {
	return o.application.PageVersion(ctx, platforms, page)
}