复制项目

pkg/msgprocessor/options.go

@@ -0,0 +1,173 @@
+// 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 msgprocessor
+
+import "git.imall.cloud/openim/protocol/constant"
+
+type (
+	Options    map[string]bool
+	OptionsOpt func(Options)
+)
+
+func NewOptions(opts ...OptionsOpt) Options {
+	options := make(map[string]bool, 11)
+	options[constant.IsNotNotification] = false
+	options[constant.IsSendMsg] = false
+	options[constant.IsHistory] = false
+	options[constant.IsPersistent] = false
+	options[constant.IsOfflinePush] = false
+	options[constant.IsUnreadCount] = false
+	options[constant.IsConversationUpdate] = false
+	options[constant.IsSenderSync] = true
+	options[constant.IsNotPrivate] = false
+	options[constant.IsSenderConversationUpdate] = false
+	options[constant.IsReactionFromCache] = false
+	for _, opt := range opts {
+		opt(options)
+	}
+
+	return options
+}
+
+func NewMsgOptions() Options {
+	options := make(map[string]bool, 11)
+	options[constant.IsOfflinePush] = false
+	return make(map[string]bool)
+}
+
+func WithOptions(options Options, opts ...OptionsOpt) Options {
+	for _, opt := range opts {
+		opt(options)
+	}
+	return options
+}
+
+func WithNotNotification(b bool) OptionsOpt {
+	return func(options Options) {
+		options[constant.IsNotNotification] = b
+	}
+}
+
+func WithSendMsg(b bool) OptionsOpt {
+	return func(options Options) {
+		options[constant.IsSendMsg] = b
+	}
+}
+
+func WithHistory(b bool) OptionsOpt {
+	return func(options Options) {
+		options[constant.IsHistory] = b
+	}
+}
+
+func WithPersistent() OptionsOpt {
+	return func(options Options) {
+		options[constant.IsPersistent] = true
+	}
+}
+
+func WithOfflinePush(b bool) OptionsOpt {
+	return func(options Options) {
+		options[constant.IsOfflinePush] = b
+	}
+}
+
+func WithUnreadCount(b bool) OptionsOpt {
+	return func(options Options) {
+		options[constant.IsUnreadCount] = b
+	}
+}
+
+func WithConversationUpdate() OptionsOpt {
+	return func(options Options) {
+		options[constant.IsConversationUpdate] = true
+	}
+}
+
+func WithSenderSync() OptionsOpt {
+	return func(options Options) {
+		options[constant.IsSenderSync] = true
+	}
+}
+
+func WithNotPrivate() OptionsOpt {
+	return func(options Options) {
+		options[constant.IsNotPrivate] = true
+	}
+}
+
+func WithSenderConversationUpdate() OptionsOpt {
+	return func(options Options) {
+		options[constant.IsSenderConversationUpdate] = true
+	}
+}
+
+func WithReactionFromCache() OptionsOpt {
+	return func(options Options) {
+		options[constant.IsReactionFromCache] = true
+	}
+}
+
+func (o Options) Is(notification string) bool {
+	v, ok := o[notification]
+	if !ok || v {
+		return true
+	}
+	return false
+}
+
+func (o Options) IsNotNotification() bool {
+	return o.Is(constant.IsNotNotification)
+}
+
+func (o Options) IsSendMsg() bool {
+	return o.Is(constant.IsSendMsg)
+}
+
+func (o Options) IsHistory() bool {
+	return o.Is(constant.IsHistory)
+}
+
+func (o Options) IsPersistent() bool {
+	return o.Is(constant.IsPersistent)
+}
+
+func (o Options) IsOfflinePush() bool {
+	return o.Is(constant.IsOfflinePush)
+}
+
+func (o Options) IsUnreadCount() bool {
+	return o.Is(constant.IsUnreadCount)
+}
+
+func (o Options) IsConversationUpdate() bool {
+	return o.Is(constant.IsConversationUpdate)
+}
+
+func (o Options) IsSenderSync() bool {
+	return o.Is(constant.IsSenderSync)
+}
+
+func (o Options) IsNotPrivate() bool {
+	return o.Is(constant.IsNotPrivate)
+}
+
+func (o Options) IsSenderConversationUpdate() bool {
+	return o.Is(constant.IsSenderConversationUpdate)
+}
+
+func (o Options) IsReactionFromCache() bool {
+	return o.Is(constant.IsReactionFromCache)
+}