多租户更改
This commit is contained in:
richjohnrambo
@@ -0,0 +1,752 @@
|
||||
// External services contacted through RPC
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/tinode/chat/pbx"
|
||||
"github.com/tinode/chat/server/logs"
|
||||
"github.com/tinode/chat/server/store/types"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/metadata"
|
||||
)
|
||||
|
||||
const (
|
||||
plgHi = 1 << iota
|
||||
plgAcc
|
||||
plgLogin
|
||||
plgSub
|
||||
plgLeave
|
||||
plgPub
|
||||
plgGet
|
||||
plgSet
|
||||
plgDel
|
||||
plgNote
|
||||
plgData
|
||||
plgMeta
|
||||
plgPres
|
||||
plgInfo
|
||||
|
||||
plgClientMask = plgHi | plgAcc | plgLogin | plgSub | plgLeave | plgPub | plgGet | plgSet | plgDel | plgNote
|
||||
plgServerMask = plgData | plgMeta | plgPres | plgInfo
|
||||
)
|
||||
|
||||
const (
|
||||
plgActCreate = 1 << iota
|
||||
plgActUpd
|
||||
plgActDel
|
||||
|
||||
plgActMask = plgActCreate | plgActUpd | plgActDel
|
||||
)
|
||||
|
||||
const (
|
||||
plgTopicMe = 1 << iota
|
||||
plgTopicFnd
|
||||
plgTopicP2P
|
||||
plgTopicGrp
|
||||
plgTopicSys
|
||||
plgTopicSlf
|
||||
plgTopicNew
|
||||
plgTopicNch
|
||||
|
||||
plgTopicCatMask = plgTopicMe | plgTopicFnd | plgTopicP2P | plgTopicGrp | plgTopicSys | plgTopicSlf
|
||||
)
|
||||
|
||||
const (
|
||||
plgFilterByTopicType = 1 << iota
|
||||
plgFilterByPacket
|
||||
plgFilterByAction
|
||||
)
|
||||
|
||||
var (
|
||||
plgPacketNames = []string{
|
||||
"hi", "acc", "login", "sub", "leave", "pub", "get", "set", "del", "note",
|
||||
"data", "meta", "pres", "info",
|
||||
}
|
||||
|
||||
plgTopicCatNames = []string{"me", "fnd", "p2p", "grp", "sys", "slf", "new", "nch"}
|
||||
)
|
||||
|
||||
// PluginFilter is a enum which defines filtering types.
|
||||
type PluginFilter struct {
|
||||
byPacket int
|
||||
byTopicType int
|
||||
byAction int
|
||||
}
|
||||
|
||||
// ParsePluginFilter parses filter config string.
|
||||
func ParsePluginFilter(s *string, filterBy int) (*PluginFilter, error) {
|
||||
if s == nil {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
parseByName := func(parts []string, options []string, def int) (int, error) {
|
||||
var result int
|
||||
|
||||
// Iterate over filter parts
|
||||
for _, inp := range parts {
|
||||
if inp != "" {
|
||||
inp = strings.ToLower(inp)
|
||||
// Split string like "hi,login,pres" or "me,p2p,fnd"
|
||||
values := strings.Split(inp, ",")
|
||||
// For each value in the input string, try to find it in the options set
|
||||
for _, val := range values {
|
||||
i := 0
|
||||
// Iterate over the options, i.e find "hi" in the slice of packet names
|
||||
for i = range options {
|
||||
if options[i] == val {
|
||||
result |= 1 << uint(i)
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if result != 0 && i == len(options) {
|
||||
// Mix of known and unknown options in the input
|
||||
return 0, errors.New("plugin: unknown value in filter " + val)
|
||||
}
|
||||
}
|
||||
|
||||
if result != 0 {
|
||||
// Found and parsed the right part
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// If the filter value is not defined, use default.
|
||||
if result == 0 {
|
||||
result = def
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
parseAction := func(parts []string) int {
|
||||
var result int
|
||||
for _, inp := range parts {
|
||||
Loop:
|
||||
for _, char := range inp {
|
||||
switch char {
|
||||
case 'c', 'C':
|
||||
result |= plgActCreate
|
||||
case 'u', 'U':
|
||||
result |= plgActUpd
|
||||
case 'd', 'D':
|
||||
result |= plgActDel
|
||||
default:
|
||||
// Unknown symbol means this is not an action string.
|
||||
result = 0
|
||||
break Loop
|
||||
}
|
||||
}
|
||||
|
||||
if result != 0 {
|
||||
// Found and parsed actions.
|
||||
break
|
||||
}
|
||||
}
|
||||
if result == 0 {
|
||||
result = plgActMask
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
filter := PluginFilter{}
|
||||
parts := strings.Split(*s, ";")
|
||||
var err error
|
||||
|
||||
if filterBy&plgFilterByPacket != 0 {
|
||||
if filter.byPacket, err = parseByName(parts, plgPacketNames, plgClientMask); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
if filterBy&plgFilterByTopicType != 0 {
|
||||
if filter.byTopicType, err = parseByName(parts, plgTopicCatNames, plgTopicCatMask); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
if filterBy&plgFilterByAction != 0 {
|
||||
filter.byAction = parseAction(parts)
|
||||
}
|
||||
|
||||
return &filter, nil
|
||||
}
|
||||
|
||||
// PluginRPCFilterConfig filters for an individual RPC call. Filter strings are formatted as follows:
|
||||
// <comma separated list of packet names> ; <comma separated list of topics or topic types> ; <actions (combination of C U D)>
|
||||
// For instance:
|
||||
// "acc,login;;CU" - grab packets {acc} or {login}; no filtering by topic, Create or Update action
|
||||
// "pub,pres;me,p2p;"
|
||||
type pluginRPCFilterConfig struct {
|
||||
// Filter by packet name, topic type [or exact name - not supported yet]. 2D: "pub,pres;p2p,me"
|
||||
FireHose *string `json:"fire_hose"`
|
||||
|
||||
// Filter by CUD, [exact user name - not supported yet]. 1D: "C"
|
||||
Account *string `json:"account"`
|
||||
// Filter by CUD, topic type[, exact name]: "p2p;CU"
|
||||
Topic *string `json:"topic"`
|
||||
// Filter by CUD, topic type[, exact topic name, exact user name]: "CU"
|
||||
Subscription *string `json:"subscription"`
|
||||
// Filter by C.D, topic type[, exact topic name, exact user name]: "grp;CD"
|
||||
Message *string `json:"message"`
|
||||
|
||||
// Call Find service, true or false
|
||||
Find bool
|
||||
}
|
||||
|
||||
type pluginConfig struct {
|
||||
Enabled bool `json:"enabled"`
|
||||
// Unique service name
|
||||
Name string `json:"name"`
|
||||
// Microseconds to wait before timeout
|
||||
Timeout int64 `json:"timeout"`
|
||||
// Filters for RPC calls: when to call vs when to skip the call
|
||||
Filters pluginRPCFilterConfig `json:"filters"`
|
||||
// What should the server do if plugin failed: HTTP error code
|
||||
FailureCode int `json:"failure_code"`
|
||||
// HTTP Error message to go with the code
|
||||
FailureMessage string `json:"failure_text"`
|
||||
// Address of plugin server of the form "tcp://localhost:123" or "unix://path_to_socket_file"
|
||||
ServiceAddr string `json:"service_addr"`
|
||||
}
|
||||
|
||||
// Plugin defines client-side parameters of a gRPC plugin.
|
||||
type Plugin struct {
|
||||
name string
|
||||
timeout time.Duration
|
||||
// Filters for individual methods
|
||||
filterFireHose *PluginFilter
|
||||
filterAccount *PluginFilter
|
||||
filterTopic *PluginFilter
|
||||
filterSubscription *PluginFilter
|
||||
filterMessage *PluginFilter
|
||||
filterFind bool
|
||||
failureCode int
|
||||
failureText string
|
||||
network string
|
||||
addr string
|
||||
|
||||
conn *grpc.ClientConn
|
||||
client pbx.PluginClient
|
||||
}
|
||||
|
||||
func pluginsInit(configString json.RawMessage) {
|
||||
// Check if any plugins are defined
|
||||
if len(configString) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
var config []pluginConfig
|
||||
if err := json.Unmarshal(configString, &config); err != nil {
|
||||
logs.Err.Fatal(err)
|
||||
}
|
||||
|
||||
nameIndex := make(map[string]bool)
|
||||
globals.plugins = make([]Plugin, len(config))
|
||||
count := 0
|
||||
for i := range config {
|
||||
conf := &config[i]
|
||||
if !conf.Enabled {
|
||||
continue
|
||||
}
|
||||
|
||||
if nameIndex[conf.Name] {
|
||||
logs.Err.Fatalf("plugins: duplicate name '%s'", conf.Name)
|
||||
}
|
||||
|
||||
globals.plugins[count] = Plugin{
|
||||
name: conf.Name,
|
||||
timeout: time.Duration(conf.Timeout) * time.Microsecond,
|
||||
failureCode: conf.FailureCode,
|
||||
failureText: conf.FailureMessage,
|
||||
}
|
||||
var err error
|
||||
if globals.plugins[count].filterFireHose, err =
|
||||
ParsePluginFilter(conf.Filters.FireHose, plgFilterByTopicType|plgFilterByPacket); err != nil {
|
||||
logs.Err.Fatal("plugins: bad FireHose filter", err)
|
||||
}
|
||||
if globals.plugins[count].filterAccount, err =
|
||||
ParsePluginFilter(conf.Filters.Account, plgFilterByAction); err != nil {
|
||||
logs.Err.Fatal("plugins: bad Account filter", err)
|
||||
}
|
||||
if globals.plugins[count].filterTopic, err =
|
||||
ParsePluginFilter(conf.Filters.Topic, plgFilterByTopicType|plgFilterByAction); err != nil {
|
||||
logs.Err.Fatal("plugins: bad Topic filter", err)
|
||||
}
|
||||
if globals.plugins[count].filterSubscription, err =
|
||||
ParsePluginFilter(conf.Filters.Subscription, plgFilterByTopicType|plgFilterByAction); err != nil {
|
||||
logs.Err.Fatal("plugins: bad Subscription filter", err)
|
||||
}
|
||||
if globals.plugins[count].filterMessage, err =
|
||||
ParsePluginFilter(conf.Filters.Message, plgFilterByTopicType|plgFilterByAction); err != nil {
|
||||
logs.Err.Fatal("plugins: bad Message filter", err)
|
||||
}
|
||||
|
||||
globals.plugins[count].filterFind = conf.Filters.Find
|
||||
|
||||
if parts := strings.SplitN(conf.ServiceAddr, "://", 2); len(parts) < 2 {
|
||||
logs.Err.Fatal("plugins: invalid server address format", conf.ServiceAddr)
|
||||
} else {
|
||||
globals.plugins[count].network = parts[0]
|
||||
globals.plugins[count].addr = parts[1]
|
||||
}
|
||||
|
||||
globals.plugins[count].conn, err = grpc.Dial(globals.plugins[count].addr, grpc.WithInsecure())
|
||||
if err != nil {
|
||||
logs.Err.Fatalf("plugins: connection failure %v", err)
|
||||
}
|
||||
|
||||
globals.plugins[count].client = pbx.NewPluginClient(globals.plugins[count].conn)
|
||||
|
||||
nameIndex[conf.Name] = true
|
||||
count++
|
||||
}
|
||||
|
||||
globals.plugins = globals.plugins[:count]
|
||||
if len(globals.plugins) == 0 {
|
||||
logs.Info.Println("plugins: no active plugins found")
|
||||
globals.plugins = nil
|
||||
} else {
|
||||
var names []string
|
||||
for i := range globals.plugins {
|
||||
names = append(names, globals.plugins[i].name+"("+globals.plugins[i].addr+")")
|
||||
}
|
||||
|
||||
logs.Info.Println("plugins: active", "'"+strings.Join(names, "', '")+"'")
|
||||
}
|
||||
}
|
||||
|
||||
func pluginsShutdown() {
|
||||
if globals.plugins == nil {
|
||||
return
|
||||
}
|
||||
|
||||
for i := range globals.plugins {
|
||||
globals.plugins[i].conn.Close()
|
||||
}
|
||||
}
|
||||
|
||||
func pluginGenerateClientReq(sess *Session, msg *ClientComMessage) *pbx.ClientReq {
|
||||
cmsg := pbCliSerialize(msg)
|
||||
if cmsg == nil {
|
||||
return nil
|
||||
}
|
||||
return &pbx.ClientReq{
|
||||
Msg: cmsg,
|
||||
Sess: &pbx.Session{
|
||||
SessionId: sess.sid,
|
||||
UserId: sess.uid.UserId(),
|
||||
AuthLevel: pbx.AuthLevel(sess.authLvl),
|
||||
UserAgent: sess.userAgent,
|
||||
RemoteAddr: sess.remoteAddr,
|
||||
DeviceId: sess.deviceID,
|
||||
Language: sess.lang,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func pluginContext(base context.Context, p *Plugin, tenantID types.TenantID) (context.Context, context.CancelFunc) {
|
||||
var ctx context.Context
|
||||
var cancel context.CancelFunc
|
||||
if p.timeout > 0 {
|
||||
ctx, cancel = context.WithTimeout(base, p.timeout)
|
||||
} else {
|
||||
ctx = base
|
||||
}
|
||||
if tenantID.IsValid() {
|
||||
ctx = metadata.AppendToOutgoingContext(ctx, "tenant-id", strconv.FormatInt(int64(tenantID), 10))
|
||||
}
|
||||
return ctx, cancel
|
||||
}
|
||||
|
||||
func pluginFireHose(sess *Session, msg *ClientComMessage) (*ClientComMessage, *ServerComMessage) {
|
||||
if globals.plugins == nil {
|
||||
// Return the original message to continue processing without changes
|
||||
return msg, nil
|
||||
}
|
||||
|
||||
var req *pbx.ClientReq
|
||||
|
||||
id, topic := pluginIDAndTopic(msg)
|
||||
ts := time.Now().UTC().Round(time.Millisecond)
|
||||
for i := range globals.plugins {
|
||||
p := &globals.plugins[i]
|
||||
if !pluginDoFiltering(p.filterFireHose, msg) {
|
||||
// Plugin is not interested in FireHose
|
||||
continue
|
||||
}
|
||||
|
||||
if req == nil {
|
||||
// Generate request only if needed
|
||||
req = pluginGenerateClientReq(sess, msg)
|
||||
if req == nil {
|
||||
// Failed to serialize message. Most likely the message is invalid.
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
ctx, cancel := pluginContext(context.Background(), p, msg.TenantID)
|
||||
if cancel != nil {
|
||||
defer cancel()
|
||||
}
|
||||
if resp, err := p.client.FireHose(ctx, req); err == nil {
|
||||
respStatus := resp.GetStatus()
|
||||
// CONTINUE means default processing
|
||||
if respStatus == pbx.RespCode_CONTINUE {
|
||||
continue
|
||||
}
|
||||
// DROP means stop processing of the message
|
||||
if respStatus == pbx.RespCode_DROP {
|
||||
return nil, nil
|
||||
}
|
||||
// REPLACE: ClientMsg was updated by the plugin. Use the new one for further processing.
|
||||
if respStatus == pbx.RespCode_REPLACE {
|
||||
repl := pbCliDeserialize(resp.GetClmsg())
|
||||
if repl != nil {
|
||||
repl.TenantID = msg.TenantID
|
||||
}
|
||||
return repl, nil
|
||||
}
|
||||
|
||||
// RESPOND: Plugin provided an alternative response message. Use it
|
||||
srv := pbServDeserialize(resp.GetSrvmsg())
|
||||
if srv != nil {
|
||||
srv.TenantID = msg.TenantID
|
||||
}
|
||||
return nil, srv
|
||||
|
||||
} else if p.failureCode != 0 {
|
||||
// Plugin failed and it's configured to stop further processing.
|
||||
logs.Err.Println("plugin: failed,", p.name, err)
|
||||
return nil, &ServerComMessage{
|
||||
TenantID: msg.TenantID,
|
||||
Ctrl: &MsgServerCtrl{
|
||||
Id: id,
|
||||
Code: p.failureCode,
|
||||
Text: p.failureText,
|
||||
Topic: topic,
|
||||
Timestamp: ts,
|
||||
},
|
||||
}
|
||||
} else {
|
||||
// Plugin failed but configured to ignore failure.
|
||||
logs.Warn.Println("plugin: failure ignored,", p.name, err)
|
||||
}
|
||||
}
|
||||
|
||||
return msg, nil
|
||||
}
|
||||
|
||||
// Ask plugin to perform search.
|
||||
func pluginFind(tenantID types.TenantID, user types.Uid, query string) (string, []types.Subscription, error) {
|
||||
if globals.plugins == nil {
|
||||
return query, nil, nil
|
||||
}
|
||||
|
||||
find := &pbx.SearchQuery{
|
||||
UserId: user.UserId(),
|
||||
Query: query,
|
||||
}
|
||||
for i := range globals.plugins {
|
||||
p := &globals.plugins[i]
|
||||
if !p.filterFind {
|
||||
// Plugin cannot service Find requests
|
||||
continue
|
||||
}
|
||||
|
||||
ctx, cancel := pluginContext(context.Background(), p, tenantID)
|
||||
if cancel != nil {
|
||||
defer cancel()
|
||||
}
|
||||
resp, err := p.client.Find(ctx, find)
|
||||
if err != nil {
|
||||
logs.Warn.Println("plugins: Find call failed", p.name, err)
|
||||
return "", nil, err
|
||||
}
|
||||
respStatus := resp.GetStatus()
|
||||
// CONTINUE means default processing
|
||||
if respStatus == pbx.RespCode_CONTINUE {
|
||||
continue
|
||||
}
|
||||
// DROP means stop processing the request
|
||||
if respStatus == pbx.RespCode_DROP {
|
||||
return "", nil, nil
|
||||
}
|
||||
// REPLACE: query string was changed. Use the new one for further processing.
|
||||
if respStatus == pbx.RespCode_REPLACE {
|
||||
return resp.GetQuery(), nil, nil
|
||||
}
|
||||
// RESPOND: Plugin provided a specific response. Use it
|
||||
return "", pbSubSliceDeserialize(resp.GetResult()), nil
|
||||
}
|
||||
|
||||
return query, nil, nil
|
||||
}
|
||||
|
||||
func pluginAccount(user *types.User, action int) {
|
||||
if globals.plugins == nil {
|
||||
return
|
||||
}
|
||||
|
||||
var event *pbx.AccountEvent
|
||||
for i := range globals.plugins {
|
||||
p := &globals.plugins[i]
|
||||
if p.filterAccount == nil || p.filterAccount.byAction&action == 0 {
|
||||
// Plugin is not interested in Account actions
|
||||
continue
|
||||
}
|
||||
|
||||
if event == nil {
|
||||
event = &pbx.AccountEvent{
|
||||
Action: pluginActionToCrud(action),
|
||||
UserId: user.Uid().UserId(),
|
||||
DefaultAcs: pbDefaultAcsSerialize(&MsgDefaultAcsMode{
|
||||
Auth: user.Access.Auth.String(),
|
||||
Anon: user.Access.Anon.String(),
|
||||
}),
|
||||
Public: interfaceToBytes(user.Public),
|
||||
Tags: user.Tags,
|
||||
}
|
||||
}
|
||||
|
||||
ctx, cancel := pluginContext(context.Background(), p, user.TenantID)
|
||||
if cancel != nil {
|
||||
defer cancel()
|
||||
}
|
||||
if _, err := p.client.Account(ctx, event); err != nil {
|
||||
logs.Warn.Println("plugins: Account call failed", p.name, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func pluginTopic(topic *Topic, action int) {
|
||||
if globals.plugins == nil {
|
||||
return
|
||||
}
|
||||
|
||||
var event *pbx.TopicEvent
|
||||
for i := range globals.plugins {
|
||||
p := &globals.plugins[i]
|
||||
if p.filterTopic == nil || p.filterTopic.byAction&action == 0 {
|
||||
// Plugin is not interested in Message actions
|
||||
continue
|
||||
}
|
||||
|
||||
if event == nil {
|
||||
event = &pbx.TopicEvent{
|
||||
Action: pluginActionToCrud(action),
|
||||
Name: topic.name,
|
||||
Desc: pbTopicSerializeToDesc(topic),
|
||||
}
|
||||
}
|
||||
|
||||
ctx, cancel := pluginContext(context.Background(), p, topic.tenantID)
|
||||
if cancel != nil {
|
||||
defer cancel()
|
||||
}
|
||||
if _, err := p.client.Topic(ctx, event); err != nil {
|
||||
logs.Warn.Println("plugins: Topic call failed", p.name, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func pluginSubscription(sub *types.Subscription, action int) {
|
||||
if globals.plugins == nil {
|
||||
return
|
||||
}
|
||||
|
||||
var event *pbx.SubscriptionEvent
|
||||
for i := range globals.plugins {
|
||||
p := &globals.plugins[i]
|
||||
if p.filterSubscription == nil || p.filterSubscription.byAction&action == 0 {
|
||||
// Plugin is not interested in Message actions
|
||||
continue
|
||||
}
|
||||
|
||||
if event == nil {
|
||||
event = &pbx.SubscriptionEvent{
|
||||
Action: pluginActionToCrud(action),
|
||||
Topic: sub.Topic,
|
||||
UserId: sub.User,
|
||||
|
||||
DelId: int32(sub.DelId),
|
||||
ReadId: int32(sub.ReadSeqId),
|
||||
RecvId: int32(sub.RecvSeqId),
|
||||
|
||||
Mode: &pbx.AccessMode{
|
||||
Want: sub.ModeWant.String(),
|
||||
Given: sub.ModeGiven.String(),
|
||||
},
|
||||
|
||||
Private: interfaceToBytes(sub.Private),
|
||||
}
|
||||
}
|
||||
|
||||
ctx, cancel := pluginContext(context.Background(), p, sub.TenantID)
|
||||
if cancel != nil {
|
||||
defer cancel()
|
||||
}
|
||||
if _, err := p.client.Subscription(ctx, event); err != nil {
|
||||
logs.Warn.Println("plugins: Subscription call failed", p.name, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Message accepted for delivery
|
||||
func pluginMessage(tenantID types.TenantID, data *MsgServerData, action int) {
|
||||
if globals.plugins == nil || action != plgActCreate {
|
||||
return
|
||||
}
|
||||
|
||||
var event *pbx.MessageEvent
|
||||
for i := range globals.plugins {
|
||||
p := &globals.plugins[i]
|
||||
if p.filterMessage == nil || p.filterMessage.byAction&action == 0 {
|
||||
// Plugin is not interested in Message actions
|
||||
continue
|
||||
}
|
||||
|
||||
if event == nil {
|
||||
event = &pbx.MessageEvent{
|
||||
Action: pluginActionToCrud(action),
|
||||
Msg: pbServDataSerialize(data).Data,
|
||||
}
|
||||
}
|
||||
|
||||
ctx, cancel := pluginContext(context.Background(), p, tenantID)
|
||||
if cancel != nil {
|
||||
defer cancel()
|
||||
}
|
||||
if _, err := p.client.Message(ctx, event); err != nil {
|
||||
logs.Warn.Println("plugins: Message call failed", p.name, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Returns false to skip, true to process
|
||||
func pluginDoFiltering(filter *PluginFilter, msg *ClientComMessage) bool {
|
||||
filterByTopic := func(topic string, flt int) bool {
|
||||
if topic == "" || flt == plgTopicCatMask {
|
||||
return true
|
||||
}
|
||||
|
||||
tt := topic
|
||||
if len(tt) > 3 {
|
||||
tt = topic[:3]
|
||||
}
|
||||
switch tt {
|
||||
case "me":
|
||||
return flt&plgTopicMe != 0
|
||||
case "fnd":
|
||||
return flt&plgTopicFnd != 0
|
||||
case "usr":
|
||||
return flt&plgTopicP2P != 0
|
||||
case "grp":
|
||||
return flt&plgTopicGrp != 0
|
||||
case "sys":
|
||||
return flt&plgTopicSys != 0
|
||||
case "slf":
|
||||
return flt&plgTopicSlf != 0
|
||||
case "new":
|
||||
return flt&plgTopicNew != 0
|
||||
case "nch":
|
||||
return flt&plgTopicNch != 0
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// Check if plugin has any filters for this call
|
||||
if filter == nil || filter.byPacket == 0 {
|
||||
return false
|
||||
}
|
||||
// Check if plugin wants all the messages
|
||||
if filter.byPacket == plgClientMask && filter.byTopicType == plgTopicCatMask {
|
||||
return true
|
||||
}
|
||||
// Check individual bits
|
||||
if msg.Hi != nil {
|
||||
return filter.byPacket&plgHi != 0
|
||||
}
|
||||
if msg.Acc != nil {
|
||||
return filter.byPacket&plgAcc != 0
|
||||
}
|
||||
if msg.Login != nil {
|
||||
return filter.byPacket&plgLogin != 0
|
||||
}
|
||||
if msg.Sub != nil {
|
||||
return filter.byPacket&plgSub != 0 && filterByTopic(msg.Sub.Topic, filter.byTopicType)
|
||||
}
|
||||
if msg.Leave != nil {
|
||||
return filter.byPacket&plgLeave != 0 && filterByTopic(msg.Leave.Topic, filter.byTopicType)
|
||||
}
|
||||
if msg.Pub != nil {
|
||||
return filter.byPacket&plgPub != 0 && filterByTopic(msg.Pub.Topic, filter.byTopicType)
|
||||
}
|
||||
if msg.Get != nil {
|
||||
return filter.byPacket&plgGet != 0 && filterByTopic(msg.Get.Topic, filter.byTopicType)
|
||||
}
|
||||
if msg.Set != nil {
|
||||
return filter.byPacket&plgSet != 0 && filterByTopic(msg.Set.Topic, filter.byTopicType)
|
||||
}
|
||||
if msg.Del != nil {
|
||||
return filter.byPacket&plgDel != 0 && filterByTopic(msg.Del.Topic, filter.byTopicType)
|
||||
}
|
||||
if msg.Note != nil {
|
||||
return filter.byPacket&plgNote != 0 && filterByTopic(msg.Note.Topic, filter.byTopicType)
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func pluginActionToCrud(action int) pbx.Crud {
|
||||
switch action {
|
||||
case plgActCreate:
|
||||
return pbx.Crud_CREATE
|
||||
case plgActUpd:
|
||||
return pbx.Crud_UPDATE
|
||||
case plgActDel:
|
||||
return pbx.Crud_DELETE
|
||||
}
|
||||
panic("plugin: unknown action")
|
||||
}
|
||||
|
||||
// pluginIDAndTopic extracts message ID and topic name.
|
||||
func pluginIDAndTopic(msg *ClientComMessage) (string, string) {
|
||||
if msg.Hi != nil {
|
||||
return msg.Hi.Id, ""
|
||||
}
|
||||
if msg.Acc != nil {
|
||||
return msg.Acc.Id, ""
|
||||
}
|
||||
if msg.Login != nil {
|
||||
return msg.Login.Id, ""
|
||||
}
|
||||
if msg.Sub != nil {
|
||||
return msg.Sub.Id, msg.Sub.Topic
|
||||
}
|
||||
if msg.Leave != nil {
|
||||
return msg.Leave.Id, msg.Leave.Topic
|
||||
}
|
||||
if msg.Pub != nil {
|
||||
return msg.Pub.Id, msg.Pub.Topic
|
||||
}
|
||||
if msg.Get != nil {
|
||||
return msg.Get.Id, msg.Get.Topic
|
||||
}
|
||||
if msg.Set != nil {
|
||||
return msg.Set.Id, msg.Set.Topic
|
||||
}
|
||||
if msg.Del != nil {
|
||||
return msg.Del.Id, msg.Del.Topic
|
||||
}
|
||||
if msg.Note != nil {
|
||||
return "", msg.Note.Topic
|
||||
}
|
||||
return "", ""
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user