Plugin API for Shared Channels: support auto invite (#25834)

* option for auto inviting plugin to all shared channels.

* auto-invite remotes to shared channels when flag set

* fix unit test

---------

Co-authored-by: Mattermost Build <build@mattermost.com>
Этот коммит содержится в:
Doug Lauder
2024-01-08 12:12:18 -05:00
коммит произвёл GitHub
родитель b05093d508
Коммит 000aa0d891
8 изменённых файлов: 69 добавлений и 6 удалений

Просмотреть файл

@@ -9,6 +9,7 @@ import (
"net/http"
"github.com/mattermost/mattermost/server/public/model"
"github.com/mattermost/mattermost/server/public/shared/mlog"
"github.com/mattermost/mattermost/server/public/shared/request"
"github.com/mattermost/mattermost/server/v8/channels/store"
)
@@ -325,11 +326,13 @@ func (a *App) OnSharedChannelsSyncMsg(msg *model.SyncMsg, rc *model.RemoteCluste
func (a *App) OnSharedChannelsPing(rc *model.RemoteCluster) bool {
pluginsEnvironment := a.GetPluginsEnvironment()
if pluginsEnvironment == nil {
a.Log().Error("Ping for shared channels cannot get plugins env")
return false
}
pluginHooks, err := pluginsEnvironment.HooksForPlugin(rc.PluginID)
if err != nil {
a.Log().Error("Ping for shared channels cannot get plugin hooks", mlog.String("plugin_id", rc.PluginID), mlog.Err(err))
return false
}

Просмотреть файл

@@ -191,6 +191,10 @@ func (s sqlRemoteClusterStore) GetAll(filter model.RemoteClusterQueryFilter) ([]
query = query.Where(sq.Eq{"rc.PluginID": filter.PluginID})
}
if filter.RequireOptions != 0 {
query = query.Where(sq.NotEq{fmt.Sprintf("(rc.Options & %d)", filter.RequireOptions): 0})
}
if filter.Topic != "" {
trimmed := strings.TrimSpace(filter.Topic)
if trimmed == "" || trimmed == "*" {

Просмотреть файл

@@ -45,7 +45,7 @@ func (rcs *Service) pingGenerator(pingChan chan *model.RemoteCluster, done <-cha
}
for _, rc := range remotes {
if rc.SiteURL != "" { // filter out unconfirmed invites
if rc.SiteURL != "" || rc.PluginID != "" { // filter out unconfirmed invites
pingChan <- rc
}
}

Просмотреть файл

@@ -248,6 +248,16 @@ func (scs *Service) processTask(task syncTask) error {
if err != nil {
return err
}
// add all remotes that have the autoinvited option.
filter = model.RemoteClusterQueryFilter{
RequireOptions: model.BitflagOptionAutoInvited,
}
remotesAutoInvited, err := scs.server.GetStore().RemoteCluster().GetAll(filter)
if err != nil {
return err
}
remotes = append(remotes, remotesAutoInvited...)
} else {
rc, err := scs.server.GetStore().RemoteCluster().Get(task.remoteID)
if err != nil {

Просмотреть файл

@@ -79,7 +79,22 @@ func (scs *Service) syncForRemote(task syncTask, rc *model.RemoteCluster) error
}
scr, err := scs.server.GetStore().SharedChannel().GetRemoteByIds(task.channelID, rc.RemoteId)
if err != nil {
if isNotFoundError(err) && rc.IsOptionFlagSet(model.BitflagOptionAutoInvited) {
// if SharedChannelRemote not found and remote has autoinvite flag, create a scr for it, thus inviting the remote.
scr = &model.SharedChannelRemote{
Id: model.NewId(),
ChannelId: task.channelID,
CreatorId: rc.CreatorId,
IsInviteAccepted: true,
IsInviteConfirmed: true,
RemoteId: rc.RemoteId,
LastPostCreateAt: model.GetMillis(),
LastPostUpdateAt: model.GetMillis(),
}
if scr, err = scs.server.GetStore().SharedChannel().SaveRemote(scr); err != nil {
return fmt.Errorf("cannot auto-create shared channel remote (channel_id=%s, remote_id=%s): %w", task.channelID, rc.RemoteId, err)
}
} else if err != nil {
return err
}

Просмотреть файл

@@ -105,6 +105,10 @@ func mungEmail(remotename string, maxLen int) string {
}
func isConflictError(err error) (string, bool) {
if err == nil {
return "", false
}
var errConflict *store.ErrConflict
if errors.As(err, &errConflict) {
return strings.ToLower(errConflict.Resource), true
@@ -117,3 +121,12 @@ func isConflictError(err error) (string, bool) {
}
return "", false
}
func isNotFoundError(err error) bool {
if err == nil {
return false
}
var errNotFound *store.ErrNotFound
return errors.As(err, &errNotFound)
}

Просмотреть файл

@@ -22,7 +22,8 @@ const (
RemoteNameMinLength = 1
RemoteNameMaxLength = 64
BitflagOptionAutoShareDMs Bitmask = 1 << iota
BitflagOptionAutoShareDMs Bitmask = 1 << iota // Any new DM/GM is automatically shared
BitflagOptionAutoInvited // Remote is automatically invited to all shared channels
)
var (
@@ -31,6 +32,18 @@ var (
type Bitmask uint32
func (bm *Bitmask) IsBitSet(flag Bitmask) bool {
return *bm != 0
}
func (bm *Bitmask) SetBit(flag Bitmask) {
*bm |= flag
}
func (bm *Bitmask) UnsetBit(flag Bitmask) {
*bm &= ^flag
}
type RemoteCluster struct {
RemoteId string `json:"remote_id"`
RemoteTeamId string `json:"remote_team_id"`
@@ -105,15 +118,15 @@ func (rc *RemoteCluster) IsValid() *AppError {
}
func (rc *RemoteCluster) IsOptionFlagSet(flag Bitmask) bool {
return rc.Options&flag != 0
return rc.Options.IsBitSet(flag)
}
func (rc *RemoteCluster) SetOptionFlag(flag Bitmask) {
rc.Options |= flag
rc.Options.SetBit(flag)
}
func (rc *RemoteCluster) UnsetOptionFlag(flag Bitmask) {
rc.Options &= ^flag
rc.Options.UnsetBit(flag)
}
func IsValidRemoteName(s string) bool {
@@ -340,4 +353,5 @@ type RemoteClusterQueryFilter struct {
CreatorId string
OnlyConfirmed bool
PluginID string
RequireOptions Bitmask
}

Просмотреть файл

@@ -305,6 +305,7 @@ type RegisterPluginOpts struct {
PluginID string // id of this plugin registering
CreatorID string // id of the user/bot registering
AutoShareDMs bool // when true, all DMs are automatically shared to this remote
AutoInvited bool // when true, the plugin is automatically invited and sync'd with all shared channels.
}
// GetOptionFlags returns a Bitmask of option flags as specified by the boolean options.
@@ -313,5 +314,8 @@ func (po RegisterPluginOpts) GetOptionFlags() Bitmask {
if po.AutoShareDMs {
flags |= BitflagOptionAutoShareDMs
}
if po.AutoInvited {
flags |= BitflagOptionAutoInvited
}
return flags
}