// Copyright © 2023 OpenIM. 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 config import ( "github.com/openimsdk/tools/db/mongoutil" "github.com/openimsdk/tools/db/redisutil" "github.com/openimsdk/tools/mq/kafka" "github.com/openimsdk/tools/s3/cos" "github.com/openimsdk/tools/s3/kodo" "github.com/openimsdk/tools/s3/minio" "github.com/openimsdk/tools/s3/oss" "strings" "time" ) type CacheConfig struct { Topic string `mapstructure:"topic"` SlotNum int `mapstructure:"slotNum"` SlotSize int `mapstructure:"slotSize"` SuccessExpire int `mapstructure:"successExpire"` FailedExpire int `mapstructure:"failedExpire"` } type LocalCache struct { User CacheConfig `mapstructure:"user"` Group CacheConfig `mapstructure:"group"` Friend CacheConfig `mapstructure:"friend"` Conversation CacheConfig `mapstructure:"conversation"` } type Log struct { StorageLocation string `mapstructure:"storageLocation"` RotationTime uint `mapstructure:"rotationTime"` RemainRotationCount uint `mapstructure:"remainRotationCount"` RemainLogLevel int `mapstructure:"remainLogLevel"` IsStdout bool `mapstructure:"isStdout"` IsJson bool `mapstructure:"isJson"` WithStack bool `mapstructure:"withStack"` } type Minio struct { Bucket string `mapstructure:"bucket"` AccessKeyID string `mapstructure:"accessKeyID"` SecretAccessKey string `mapstructure:"secretAccessKey"` SessionToken string `mapstructure:"sessionToken"` InternalAddress string `mapstructure:"internalAddress"` ExternalAddress string `mapstructure:"externalAddress"` PublicRead bool `mapstructure:"publicRead"` } type Mongo struct { URI string `mapstructure:"uri"` Address []string `mapstructure:"address"` Database string `mapstructure:"database"` Username string `mapstructure:"username"` Password string `mapstructure:"password"` MaxPoolSize int `mapstructure:"maxPoolSize"` MaxRetry int `mapstructure:"maxRetry"` } type Kafka struct { Username string `mapstructure:"username"` Password string `mapstructure:"password"` ProducerAck string `mapstructure:"producerAck"` CompressType string `mapstructure:"compressType"` Address []string `mapstructure:"address"` ToRedisTopic string `mapstructure:"toRedisTopic"` ToMongoTopic string `mapstructure:"toMongoTopic"` ToPushTopic string `mapstructure:"toPushTopic"` ToRedisGroupID string `mapstructure:"toRedisGroupID"` ToMongoGroupID string `mapstructure:"toMongoGroupID"` ToPushGroupID string `mapstructure:"toPushGroupID"` Tls TLSConfig `mapstructure:"tls"` } type TLSConfig struct { EnableTLS bool `mapstructure:"enableTLS"` CACrt string `mapstructure:"caCrt"` ClientCrt string `mapstructure:"clientCrt"` ClientKey string `mapstructure:"clientKey"` ClientKeyPwd string `mapstructure:"clientKeyPwd"` InsecureSkipVerify bool `mapstructure:"insecureSkipVerify"` } type API struct { Api struct { ListenIP string `mapstructure:"listenIP"` Ports []int `mapstructure:"ports"` } `mapstructure:"api"` Prometheus struct { Enable bool `mapstructure:"enable"` Ports []int `mapstructure:"ports"` GrafanaURL string `mapstructure:"grafanaURL"` } `mapstructure:"prometheus"` } type CronTask struct { CronExecuteTime string `mapstructure:"cronExecuteTime"` RetainChatRecords int `mapstructure:"retainChatRecords"` FileExpireTime int `mapstructure:"fileExpireTime"` } type OfflinePushConfig struct { Enable bool `mapstructure:"enable"` Title string `mapstructure:"title"` Desc string `mapstructure:"desc"` Ext string `mapstructure:"ext"` } type NotificationConfig struct { IsSendMsg bool `mapstructure:"isSendMsg"` ReliabilityLevel int `mapstructure:"reliabilityLevel"` UnreadCount bool `mapstructure:"unreadCount"` OfflinePush OfflinePushConfig `mapstructure:"offlinePush"` } type Notification struct { GroupCreated NotificationConfig `mapstructure:"groupCreated"` GroupInfoSet NotificationConfig `mapstructure:"groupInfoSet"` JoinGroupApplication NotificationConfig `mapstructure:"joinGroupApplication"` MemberQuit NotificationConfig `mapstructure:"memberQuit"` GroupApplicationAccepted NotificationConfig `mapstructure:"groupApplicationAccepted"` GroupApplicationRejected NotificationConfig `mapstructure:"groupApplicationRejected"` GroupOwnerTransferred NotificationConfig `mapstructure:"groupOwnerTransferred"` MemberKicked NotificationConfig `mapstructure:"memberKicked"` MemberInvited NotificationConfig `mapstructure:"memberInvited"` MemberEnter NotificationConfig `mapstructure:"memberEnter"` GroupDismissed NotificationConfig `mapstructure:"groupDismissed"` GroupMuted NotificationConfig `mapstructure:"groupMuted"` GroupCancelMuted NotificationConfig `mapstructure:"groupCancelMuted"` GroupMemberMuted NotificationConfig `mapstructure:"groupMemberMuted"` GroupMemberCancelMuted NotificationConfig `mapstructure:"groupMemberCancelMuted"` GroupMemberInfoSet NotificationConfig `mapstructure:"groupMemberInfoSet"` GroupMemberSetToAdmin NotificationConfig `yaml:"groupMemberSetToAdmin"` GroupMemberSetToOrdinary NotificationConfig `yaml:"groupMemberSetToOrdinaryUser"` GroupInfoSetAnnouncement NotificationConfig `mapstructure:"groupInfoSetAnnouncement"` GroupInfoSetName NotificationConfig `mapstructure:"groupInfoSetName"` FriendApplicationAdded NotificationConfig `mapstructure:"friendApplicationAdded"` FriendApplicationApproved NotificationConfig `mapstructure:"friendApplicationApproved"` FriendApplicationRejected NotificationConfig `mapstructure:"friendApplicationRejected"` FriendAdded NotificationConfig `mapstructure:"friendAdded"` FriendDeleted NotificationConfig `mapstructure:"friendDeleted"` FriendRemarkSet NotificationConfig `mapstructure:"friendRemarkSet"` BlackAdded NotificationConfig `mapstructure:"blackAdded"` BlackDeleted NotificationConfig `mapstructure:"blackDeleted"` FriendInfoUpdated NotificationConfig `mapstructure:"friendInfoUpdated"` UserInfoUpdated NotificationConfig `mapstructure:"userInfoUpdated"` UserStatusChanged NotificationConfig `mapstructure:"userStatusChanged"` ConversationChanged NotificationConfig `mapstructure:"conversationChanged"` ConversationSetPrivate NotificationConfig `mapstructure:"conversationSetPrivate"` } type Prometheus struct { Enable bool `mapstructure:"enable"` Ports []int `mapstructure:"ports"` } type MsgGateway struct { RPC struct { RegisterIP string `mapstructure:"registerIP"` Ports []int `mapstructure:"ports"` } `mapstructure:"rpc"` Prometheus Prometheus `mapstructure:"prometheus"` ListenIP string `mapstructure:"listenIP"` LongConnSvr struct { Ports []int `mapstructure:"ports"` WebsocketMaxConnNum int `mapstructure:"websocketMaxConnNum"` WebsocketMaxMsgLen int `mapstructure:"websocketMaxMsgLen"` WebsocketTimeout int `mapstructure:"websocketTimeout"` } `mapstructure:"longConnSvr"` MultiLoginPolicy int `mapstructure:"multiLoginPolicy"` } type MsgTransfer struct { Prometheus Prometheus `mapstructure:"prometheus"` } type Push struct { RPC struct { RegisterIP string `mapstructure:"registerIP"` ListenIP string `mapstructure:"listenIP"` Ports []int `mapstructure:"ports"` } `mapstructure:"rpc"` Prometheus Prometheus `mapstructure:"prometheus"` MaxConcurrentWorkers int `mapstructure:"maxConcurrentWorkers"` Enable string `mapstructure:"enable"` GeTui struct { PushUrl string `mapstructure:"pushUrl"` MasterSecret string `mapstructure:"masterSecret"` AppKey string `mapstructure:"appKey"` Intent string `mapstructure:"intent"` ChannelID string `mapstructure:"channelID"` ChannelName string `mapstructure:"channelName"` } `mapstructure:"geTui"` FCM struct { FilePath string `mapstructure:"filePath"` AuthURL string `mapstructure:"authURL"` } `mapstructure:"fcm"` JPNS struct { AppKey string `mapstructure:"appKey"` MasterSecret string `mapstructure:"masterSecret"` PushURL string `mapstructure:"pushURL"` PushIntent string `mapstructure:"pushIntent"` } `mapstructure:"jpns"` IOSPush struct { PushSound string `mapstructure:"pushSound"` BadgeCount bool `mapstructure:"badgeCount"` Production bool `mapstructure:"production"` } `mapstructure:"iosPush"` } type Auth struct { RPC struct { RegisterIP string `mapstructure:"registerIP"` ListenIP string `mapstructure:"listenIP"` Ports []int `mapstructure:"ports"` } `mapstructure:"rpc"` Prometheus Prometheus `mapstructure:"prometheus"` TokenPolicy struct { Expire int64 `mapstructure:"expire"` } `mapstructure:"tokenPolicy"` } type Conversation struct { RPC struct { RegisterIP string `mapstructure:"registerIP"` ListenIP string `mapstructure:"listenIP"` Ports []int `mapstructure:"ports"` } `mapstructure:"rpc"` Prometheus Prometheus `mapstructure:"prometheus"` } type Friend struct { RPC struct { RegisterIP string `mapstructure:"registerIP"` ListenIP string `mapstructure:"listenIP"` Ports []int `mapstructure:"ports"` } `mapstructure:"rpc"` Prometheus Prometheus `mapstructure:"prometheus"` } type Group struct { RPC struct { RegisterIP string `mapstructure:"registerIP"` ListenIP string `mapstructure:"listenIP"` Ports []int `mapstructure:"ports"` } `mapstructure:"rpc"` Prometheus Prometheus `mapstructure:"prometheus"` } type Msg struct { RPC struct { RegisterIP string `mapstructure:"registerIP"` ListenIP string `mapstructure:"listenIP"` Ports []int `mapstructure:"ports"` } `mapstructure:"rpc"` Prometheus Prometheus `mapstructure:"prometheus"` FriendVerify bool `mapstructure:"friendVerify"` } type Third struct { RPC struct { RegisterIP string `mapstructure:"registerIP"` ListenIP string `mapstructure:"listenIP"` Ports []int `mapstructure:"ports"` } `mapstructure:"rpc"` Prometheus Prometheus `mapstructure:"prometheus"` Object struct { Enable string `mapstructure:"enable"` Cos Cos `mapstructure:"cos"` Oss Oss `mapstructure:"oss"` Kodo Kodo `mapstructure:"kodo"` Aws struct { Endpoint string `mapstructure:"endpoint"` Region string `mapstructure:"region"` Bucket string `mapstructure:"bucket"` AccessKeyID string `mapstructure:"accessKeyID"` AccessKeySecret string `mapstructure:"accessKeySecret"` PublicRead bool `mapstructure:"publicRead"` } `mapstructure:"aws"` } `mapstructure:"object"` } type Cos struct { BucketURL string `mapstructure:"bucketURL"` SecretID string `mapstructure:"secretID"` SecretKey string `mapstructure:"secretKey"` SessionToken string `mapstructure:"sessionToken"` PublicRead bool `mapstructure:"publicRead"` } type Oss struct { Endpoint string `mapstructure:"endpoint"` Bucket string `mapstructure:"bucket"` BucketURL string `mapstructure:"bucketURL"` AccessKeyID string `mapstructure:"accessKeyID"` AccessKeySecret string `mapstructure:"accessKeySecret"` SessionToken string `mapstructure:"sessionToken"` PublicRead bool `mapstructure:"publicRead"` } type Kodo struct { Endpoint string `mapstructure:"endpoint"` Bucket string `mapstructure:"bucket"` BucketURL string `mapstructure:"bucketURL"` AccessKeyID string `mapstructure:"accessKeyID"` AccessKeySecret string `mapstructure:"accessKeySecret"` SessionToken string `mapstructure:"sessionToken"` PublicRead bool `mapstructure:"publicRead"` } type User struct { RPC struct { RegisterIP string `mapstructure:"registerIP"` ListenIP string `mapstructure:"listenIP"` Ports []int `mapstructure:"ports"` } `mapstructure:"rpc"` Prometheus Prometheus `mapstructure:"prometheus"` } type Redis struct { Address []string `mapstructure:"address"` Username string `mapstructure:"username"` Password string `mapstructure:"password"` ClusterMode bool `mapstructure:"clusterMode"` DB int `mapstructure:"storage"` MaxRetry int `mapstructure:"MaxRetry"` } type BeforeConfig struct { Enable bool `mapstructure:"enable"` Timeout int `mapstructure:"timeout"` FailedContinue bool `mapstructure:"failedContinue"` } type AfterConfig struct { Enable bool `mapstructure:"enable"` Timeout int `mapstructure:"timeout"` AttentionIds []string `mapstructure:"attentionIds"` } type Share struct { Secret string `mapstructure:"secret"` RpcRegisterName RpcRegisterName `mapstructure:"rpcRegisterName"` IMAdminUserID []string `mapstructure:"imAdminUserID"` } type RpcRegisterName struct { User string `mapstructure:"user"` Friend string `mapstructure:"friend"` Msg string `mapstructure:"msg"` Push string `mapstructure:"push"` MessageGateway string `mapstructure:"messageGateway"` Group string `mapstructure:"group"` Auth string `mapstructure:"auth"` Conversation string `mapstructure:"conversation"` Third string `mapstructure:"third"` } func (r *RpcRegisterName) GetServiceNames() []string { return []string{ r.User, r.Friend, r.Msg, r.Push, r.MessageGateway, r.Group, r.Auth, r.Conversation, r.Third, } } // FullConfig stores all configurations for before and after events type Webhooks struct { URL string `mapstructure:"url"` BeforeSendSingleMsg BeforeConfig `mapstructure:"beforeSendSingleMsg"` BeforeUpdateUserInfoEx BeforeConfig `mapstructure:"beforeUpdateUserInfoEx"` AfterUpdateUserInfoEx AfterConfig `mapstructure:"afterUpdateUserInfoEx"` AfterSendSingleMsg AfterConfig `mapstructure:"afterSendSingleMsg"` BeforeSendGroupMsg BeforeConfig `mapstructure:"beforeSendGroupMsg"` BeforeMsgModify BeforeConfig `mapstructure:"beforeMsgModify"` AfterSendGroupMsg AfterConfig `mapstructure:"afterSendGroupMsg"` AfterUserOnline AfterConfig `mapstructure:"afterUserOnline"` AfterUserOffline AfterConfig `mapstructure:"afterUserOffline"` AfterUserKickOff AfterConfig `mapstructure:"afterUserKickOff"` BeforeOfflinePush BeforeConfig `mapstructure:"beforeOfflinePush"` BeforeOnlinePush BeforeConfig `mapstructure:"beforeOnlinePush"` BeforeGroupOnlinePush BeforeConfig `mapstructure:"beforeGroupOnlinePush"` BeforeAddFriend BeforeConfig `mapstructure:"beforeAddFriend"` BeforeUpdateUserInfo BeforeConfig `mapstructure:"beforeUpdateUserInfo"` AfterUpdateUserInfo AfterConfig `mapstructure:"afterUpdateUserInfo"` BeforeCreateGroup BeforeConfig `mapstructure:"beforeCreateGroup"` AfterCreateGroup AfterConfig `mapstructure:"afterCreateGroup"` BeforeMemberJoinGroup BeforeConfig `mapstructure:"beforeMemberJoinGroup"` BeforeSetGroupMemberInfo BeforeConfig `mapstructure:"beforeSetGroupMemberInfo"` AfterSetGroupMemberInfo AfterConfig `mapstructure:"afterSetGroupMemberInfo"` AfterQuitGroup AfterConfig `mapstructure:"afterQuitGroup"` AfterKickGroupMember AfterConfig `mapstructure:"afterKickGroupMember"` AfterDismissGroup AfterConfig `mapstructure:"afterDismissGroup"` BeforeApplyJoinGroup BeforeConfig `mapstructure:"beforeApplyJoinGroup"` AfterGroupMsgRead AfterConfig `mapstructure:"afterGroupMsgRead"` AfterSingleMsgRead AfterConfig `mapstructure:"afterSingleMsgRead"` BeforeUserRegister BeforeConfig `mapstructure:"beforeUserRegister"` AfterUserRegister AfterConfig `mapstructure:"afterUserRegister"` AfterTransferGroupOwner AfterConfig `mapstructure:"afterTransferGroupOwner"` BeforeSetFriendRemark BeforeConfig `mapstructure:"beforeSetFriendRemark"` AfterSetFriendRemark AfterConfig `mapstructure:"afterSetFriendRemark"` AfterGroupMsgRevoke AfterConfig `mapstructure:"afterGroupMsgRevoke"` AfterJoinGroup AfterConfig `mapstructure:"afterJoinGroup"` BeforeInviteUserToGroup BeforeConfig `mapstructure:"beforeInviteUserToGroup"` AfterSetGroupInfo AfterConfig `mapstructure:"afterSetGroupInfo"` BeforeSetGroupInfo BeforeConfig `mapstructure:"beforeSetGroupInfo"` AfterRevokeMsg AfterConfig `mapstructure:"afterRevokeMsg"` BeforeAddBlack BeforeConfig `mapstructure:"beforeAddBlack"` AfterAddFriend AfterConfig `mapstructure:"afterAddFriend"` BeforeAddFriendAgree BeforeConfig `mapstructure:"beforeAddFriendAgree"` AfterDeleteFriend AfterConfig `mapstructure:"afterDeleteFriend"` BeforeImportFriends BeforeConfig `mapstructure:"beforeImportFriends"` AfterImportFriends AfterConfig `mapstructure:"afterImportFriends"` AfterRemoveBlack AfterConfig `mapstructure:"afterRemoveBlack"` } type ZooKeeper struct { Schema string `mapstructure:"schema"` Address []string `mapstructure:"address"` Username string `mapstructure:"username"` Password string `mapstructure:"password"` } type Discovery struct { Enable string `mapstructure:"enable"` Etcd Etcd `mapstructure:"etcd"` ZooKeeper ZooKeeper `mapstructure:"zooKeeper"` } type Etcd struct { RootDirectory string `mapstructure:"rootDirectory"` Address []string `mapstructure:"address"` Username string `mapstructure:"username"` Password string `mapstructure:"password"` } func (m *Mongo) Build() *mongoutil.Config { return &mongoutil.Config{ Uri: m.URI, Address: m.Address, Database: m.Database, Username: m.Username, Password: m.Password, MaxPoolSize: m.MaxPoolSize, MaxRetry: m.MaxRetry, } } func (r *Redis) Build() *redisutil.Config { return &redisutil.Config{ ClusterMode: r.ClusterMode, Address: r.Address, Username: r.Username, Password: r.Password, DB: r.DB, MaxRetry: r.MaxRetry, } } func (k *Kafka) Build() *kafka.Config { return &kafka.Config{ Username: k.Username, Password: k.Password, ProducerAck: k.ProducerAck, CompressType: k.CompressType, Addr: k.Address, TLS: kafka.TLSConfig{ EnableTLS: k.Tls.EnableTLS, CACrt: k.Tls.CACrt, ClientCrt: k.Tls.ClientCrt, ClientKey: k.Tls.ClientKey, ClientKeyPwd: k.Tls.ClientKeyPwd, InsecureSkipVerify: k.Tls.InsecureSkipVerify, }, } } func (m *Minio) Build() *minio.Config { formatEndpoint := func(address string) string { if strings.HasPrefix(address, "http://") || strings.HasPrefix(address, "https://") { return address } return "http://" + address } return &minio.Config{ Bucket: m.Bucket, AccessKeyID: m.AccessKeyID, SecretAccessKey: m.SecretAccessKey, SessionToken: m.SessionToken, PublicRead: m.PublicRead, Endpoint: formatEndpoint(m.InternalAddress), SignEndpoint: formatEndpoint(m.ExternalAddress), } } func (c *Cos) Build() *cos.Config { return &cos.Config{ BucketURL: c.BucketURL, SecretID: c.SecretID, SecretKey: c.SecretKey, SessionToken: c.SessionToken, PublicRead: c.PublicRead, } } func (o *Oss) Build() *oss.Config { return &oss.Config{ Endpoint: o.Endpoint, Bucket: o.Bucket, BucketURL: o.BucketURL, AccessKeyID: o.AccessKeyID, AccessKeySecret: o.AccessKeySecret, SessionToken: o.SessionToken, PublicRead: o.PublicRead, } } func (o *Kodo) Build() *kodo.Config { return &kodo.Config{ Endpoint: o.Endpoint, Bucket: o.Bucket, BucketURL: o.BucketURL, AccessKeyID: o.AccessKeyID, AccessKeySecret: o.AccessKeySecret, SessionToken: o.SessionToken, PublicRead: o.PublicRead, } } func (l *CacheConfig) Failed() time.Duration { return time.Second * time.Duration(l.FailedExpire) } func (l *CacheConfig) Success() time.Duration { return time.Second * time.Duration(l.SuccessExpire) } func (l *CacheConfig) Enable() bool { return l.Topic != "" && l.SlotNum > 0 && l.SlotSize > 0 }