render
/
spoc-bot-vr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

slack can parse optional channel wrapper for scrape

main
Bel LaPointe 2024-04-16 15:03:24 -06:00
parent d9d91193dd
commit 95b0394199
3 changed files with 36 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
main.go
View File

@ -159,7 +159,8 @@ func newHandlerPutAPIV1RPCScrapeSlack(cfg Config) http.HandlerFunc {
if cfg.Debug { if cfg.Debug {
log.Printf("rpc/scrapeslack => %s", messageJSON) log.Printf("rpc/scrapeslack => %s", messageJSON)
} }
if err := cfg.slackToModelPipeline.reader.Enqueue(r.Context(), messageJSON); err != nil { b, _ := json.Marshal(ChannelWrapper{Channel: channel, V: messageJSON})
if err := cfg.slackToModelPipeline.reader.Enqueue(r.Context(), b); err != nil {
errs = append(errs, err) errs = append(errs, err)
} else { } else {
n += 1 n += 1
@ -217,8 +218,8 @@ func handlerPostAPIV1EventsSlackInitialize(w http.ResponseWriter, r *http.Reques
func _newHandlerPostAPIV1EventsSlack(cfg Config) http.HandlerFunc { func _newHandlerPostAPIV1EventsSlack(cfg Config) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) { return func(w http.ResponseWriter, r *http.Request) {
b, _ := io.ReadAll(r.Body) body, _ := io.ReadAll(r.Body)
r.Body = io.NopCloser(bytes.NewReader(b)) r.Body = io.NopCloser(bytes.NewReader(body))
var allowList struct { var allowList struct {
Token string Token string
@ -226,7 +227,7 @@ func _newHandlerPostAPIV1EventsSlack(cfg Config) http.HandlerFunc {
Channel string Channel string
} }
} }
if err := json.Unmarshal(b, &allowList); err != nil { if err := json.Unmarshal(body, &allowList); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest) http.Error(w, err.Error(), http.StatusBadRequest)
return return
} else if allowList.Token != cfg.SlackToken { } else if allowList.Token != cfg.SlackToken {
@ -243,7 +244,7 @@ func _newHandlerPostAPIV1EventsSlack(cfg Config) http.HandlerFunc {
return return
} }
if err := cfg.slackToModelPipeline.reader.Enqueue(r.Context(), b); err != nil { if err := cfg.slackToModelPipeline.reader.Enqueue(r.Context(), body); err != nil {
log.Printf("failed to ingest: %v", err) log.Printf("failed to ingest: %v", err)
http.Error(w, err.Error(), http.StatusInternalServerError) http.Error(w, err.Error(), http.StatusInternalServerError)
return return

13
slack.go
View File

@ -227,6 +227,11 @@ func parseSlack(b []byte) (parsedSlackMessage, error) {
} }
func _parseSlack(b []byte) (slackMessage, error) { func _parseSlack(b []byte) (slackMessage, error) {
var wrapper ChannelWrapper
if err := json.Unmarshal(b, &wrapper); err == nil && len(wrapper.V) > 0 {
b = wrapper.V
}
var result slackMessage var result slackMessage
err := json.Unmarshal(b, &result) err := json.Unmarshal(b, &result)
switch result.Type { switch result.Type {
@ -247,6 +252,9 @@ func _parseSlack(b []byte) (slackMessage, error) {
} }
result.Event.PreviousMessage = nil result.Event.PreviousMessage = nil
} }
if wrapper.Channel != "" {
result.Event.Channel = wrapper.Channel
}
return result, err return result, err
} }
@ -257,3 +265,8 @@ func (this slackEvent) Empty() bool {
func (this parsedSlackMessage) Time() time.Time { func (this parsedSlackMessage) Time() time.Time {
return time.Unix(int64(this.TS), 0) return time.Unix(int64(this.TS), 0)
} }
type ChannelWrapper struct {
Channel string
V json.RawMessage
}

17
slack_test.go
View File

@ -249,3 +249,20 @@ func TestParseSlackTestdata(t *testing.T) {
}) })
} }
} }
func TestWrappedSlack(t *testing.T) {
b, _ := os.ReadFile("testdata/slack_events/human_thread_message_from_opsgenie_alert.json")
b2, _ := json.Marshal(ChannelWrapper{Channel: "X", V: json.RawMessage(b)})
if got, err := _parseSlack(b); err != nil {
t.Fatal(err)
} else if got2, err := _parseSlack(b2); err != nil {
t.Fatal(err)
} else if got2.Event.Channel != "X" {
t.Error(got2.Event.Channel)
} else if got2.Event.ParentID == "" {
t.Error(got2.Event)
} else if got.Event.ParentID != got2.Event.ParentID {
t.Error(got, got2)
}
}