livenote: add option channel

c9s · Nov 11, 2024 · 9b1060e · 9b1060e
1 parent 0b2fdd4
commit 9b1060e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/pkg/livenote/options.go b/pkg/livenote/options.go
@@ -4,6 +4,16 @@ import "time"
 
 type Option interface{}
 
+type OptionChannel struct {
+	Channel string
+}
+
+func Channel(channel string) *OptionChannel {
+	return &OptionChannel{
+		Channel: channel,
+	}
+}
+
 type OptionCompare struct {
 	Value bool
 }

diff --git a/pkg/notifier/slacknotifier/slack.go b/pkg/notifier/slacknotifier/slack.go
@@ -241,6 +241,9 @@ func (n *Notifier) PostLiveNote(obj livenote.Object, opts ...livenote.Option) er
 	var shouldPin bool
 	var ttl time.Duration = 0
 
+	// load the default channel
+	channel := n.channel
+
 	for _, opt := range opts {
 		switch val := opt.(type) {
 		case *livenote.OptionOneTimeMention:
@@ -254,6 +257,10 @@ func (n *Notifier) PostLiveNote(obj livenote.Object, opts ...livenote.Option) er
 			shouldPin = val.Value
 		case *livenote.OptionTimeToLive:
 			ttl = val.Duration
+		case *livenote.OptionChannel:
+			if val.Channel != "" {
+				channel = val.Channel
+			}
 		}
 	}
 
@@ -265,7 +272,6 @@ func (n *Notifier) PostLiveNote(obj livenote.Object, opts ...livenote.Option) er
 		}
 	}
 
-	channel := n.channel
 	note := n.liveNotePool.Update(obj)
 	curObj = note.Object