lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-26 00:35:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

slacknotifier: implement and improve translateHandle

Browse Source
This commit is contained in:
c9s 2024-11-08 12:40:02 +08:00
parent d7eef21a00
commit 4b24e9a480
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
1 changed files with 34 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
pkg/notifier/slacknotifier/slack.go
View File

@ -32,7 +32,7 @@ type Notifier struct {
liveNotePool *livenote.Pool liveNotePool *livenote.Pool
userIdCache map[string]string userIdCache map[string]*slack.User
groupIdCache map[string]slack.UserGroup groupIdCache map[string]slack.UserGroup
} }
@ -44,7 +44,7 @@ func New(client *slack.Client, channel string, options ...NotifyOption) *Notifie
client: client, client: client,
taskC: make(chan notifyTask, 100), taskC: make(chan notifyTask, 100),
liveNotePool: livenote.NewPool(100), liveNotePool: livenote.NewPool(100),
userIdCache: make(map[string]string, 30), userIdCache: make(map[string]*slack.User, 30),
groupIdCache: make(map[string]slack.UserGroup, 50), groupIdCache: make(map[string]slack.UserGroup, 50),
} }
@ -108,26 +108,48 @@ var userIdRegExp = regexp.MustCompile(`^<@(.+?)>$`)
// groupIdRegExp matches strings like <!subteam^ID> // groupIdRegExp matches strings like <!subteam^ID>
var groupIdRegExp = regexp.MustCompile(`^<!subteam\^(.+?)>$`) var groupIdRegExp = regexp.MustCompile(`^<!subteam\^(.+?)>$`)
func (n *Notifier) lookupUserID(ctx context.Context, username string) (string, error) { func (n *Notifier) translateHandle(ctx context.Context, handle string) (string, error) {
if username == "" { if handle == "" {
return "", errors.New("username is empty") return "", errors.New("handle is empty")
} }
// if the given username is already in slack user id format, we don't need to look up // trim the prefix '@' if we get a string like '@username'
if userIdRegExp.MatchString(username) { if handle[0] == '@' {
return username, nil handle = handle[1:]
} }
if id, exists := n.userIdCache[username]; exists { // if the given handle is already in slack user id format, we don't need to look up
return id, nil if userIdRegExp.MatchString(handle) || groupIdRegExp.MatchString(handle) {
return handle, nil
} }
slackUser, err := n.client.GetUserInfoContext(ctx, username) if user, exists := n.userIdCache[handle]; exists {
return toUserHandle(user.ID), nil
}
if group, exists := n.groupIdCache[handle]; exists {
return toSubteamHandle(group.ID), nil
}
slackUser, err := n.client.GetUserInfoContext(ctx, handle)
if err != nil { if err != nil {
return "", err return "", err
} }
return slackUser.ID, nil if slackUser != nil {
n.userIdCache[handle] = slackUser
return toUserHandle(slackUser.ID), nil
}
return "", fmt.Errorf("handle %s not found", handle)
}
func toUserHandle(id string) string {
return fmt.Sprintf("<@%s>", id)
}
func toSubteamHandle(id string) string {
return fmt.Sprintf("<!subteam^%s>", id)
} }
func (n *Notifier) PostLiveNote(obj livenote.Object, opts ...livenote.Option) error { func (n *Notifier) PostLiveNote(obj livenote.Object, opts ...livenote.Option) error {