mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-21 07:17:39 +00:00
43dc925f16
* fix: add iam roles to permissions * fix: show state initial on usersearch * fix: search project roles returns only roles of project * fix: add project member owner on project create * fix: create new object oon failed event * feat: parse error body on chat message * feat: remove comment
87 lines
2.2 KiB
Go
87 lines
2.2 KiB
Go
package chat
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"github.com/caos/logging"
|
|
caos_errs "github.com/caos/zitadel/internal/errors"
|
|
"github.com/caos/zitadel/internal/notification/providers"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"unicode/utf8"
|
|
)
|
|
|
|
type Chat struct {
|
|
URL *url.URL
|
|
SplitCount int
|
|
}
|
|
|
|
func InitChatProvider(config ChatConfig) (*Chat, error) {
|
|
url, err := url.Parse(config.Url)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Chat{
|
|
URL: url,
|
|
SplitCount: config.SplitCount,
|
|
}, nil
|
|
}
|
|
|
|
func (chat *Chat) CanHandleMessage(_ providers.Message) bool {
|
|
return true
|
|
}
|
|
|
|
func (chat *Chat) HandleMessage(message providers.Message) error {
|
|
contentText := message.GetContent()
|
|
for _, splittedMsg := range splitMessage(contentText, chat.SplitCount) {
|
|
chatMsg := &ChatMessage{Text: splittedMsg}
|
|
if err := chat.SendMessage(chatMsg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (chat *Chat) SendMessage(message providers.Message) error {
|
|
chatMsg, ok := message.(*ChatMessage)
|
|
if !ok {
|
|
return caos_errs.ThrowInternal(nil, "EMAIL-s8JLs", "message is not ChatMessage")
|
|
}
|
|
req, err := json.Marshal(chatMsg)
|
|
if err != nil {
|
|
return caos_errs.ThrowInternal(err, "PROVI-s8uie", "Could not unmarshal content")
|
|
}
|
|
|
|
response, err := http.Post(chat.URL.String(), "application/json; charset=UTF-8", bytes.NewReader(req))
|
|
if err != nil {
|
|
return caos_errs.ThrowInternal(err, "PROVI-si93s", "unable to send message")
|
|
}
|
|
if response.StatusCode != 200 {
|
|
defer response.Body.Close()
|
|
bodyBytes, err := ioutil.ReadAll(response.Body)
|
|
if err != nil {
|
|
return caos_errs.ThrowInternal(err, "PROVI-PSLd3", "unable to read response message")
|
|
}
|
|
logging.LogWithFields("PROVI-PS0kx", "Body", string(bodyBytes)).Warn("Chat Message post didnt get 200 OK")
|
|
return caos_errs.ThrowInternal(nil, "PROVI-LSopw", string(bodyBytes))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func splitMessage(message string, count int) []string {
|
|
if count == 0 {
|
|
return []string{message}
|
|
}
|
|
var splits []string
|
|
var l, r int
|
|
for l, r = 0, count; r < len(message); l, r = r, r+count {
|
|
for !utf8.RuneStart(message[r]) {
|
|
r--
|
|
}
|
|
splits = append(splits, message[l:r])
|
|
}
|
|
splits = append(splits, message[l:])
|
|
return splits
|
|
}
|