LocalAGI/services/connectors/slack.go

package connectors

import (
	"fmt"
	"log"
	"os"
	"strings"

	"github.com/mudler/LocalAgent/pkg/xlog"
	"github.com/sashabaranov/go-openai"

	"github.com/mudler/LocalAgent/core/agent"

	"github.com/slack-go/slack/socketmode"

	"github.com/slack-go/slack"
	"github.com/slack-go/slack/slackevents"
)

type Slack struct {
	appToken    string
	botToken    string
	channelID   string
	alwaysReply bool
}

func NewSlack(config map[string]string) *Slack {
	return &Slack{
		appToken:    config["appToken"],
		botToken:    config["botToken"],
		channelID:   config["channelID"],
		alwaysReply: config["alwaysReply"] == "true",
	}
}

func (t *Slack) AgentResultCallback() func(state agent.ActionState) {
	return func(state agent.ActionState) {
		// Send the result to the bot
	}
}

func (t *Slack) AgentReasoningCallback() func(state agent.ActionCurrentState) bool {
	return func(state agent.ActionCurrentState) bool {
		// Send the reasoning to the bot
		return true
	}
}

func cleanUpUsernameFromMessage(message string, b *slack.AuthTestResponse) string {
	cleaned := strings.ReplaceAll(message, "<@"+b.UserID+">", "")
	cleaned = strings.ReplaceAll(cleaned, "<@"+b.BotID+">", "")
	cleaned = strings.TrimSpace(cleaned)
	return cleaned
}

func (t *Slack) Start(a *agent.Agent) {
	api := slack.New(
		t.botToken,
		//	slack.OptionDebug(true),
		slack.OptionLog(log.New(os.Stdout, "api: ", log.Lshortfile|log.LstdFlags)),
		slack.OptionAppLevelToken(t.appToken),
	)

	client := socketmode.New(
		api,
		//socketmode.OptionDebug(true),
		//socketmode.OptionLog(log.New(os.Stdout, "socketmode: ", log.Lshortfile|log.LstdFlags)),
	)
	go func() {
		for evt := range client.Events {
			switch evt.Type {
			case socketmode.EventTypeConnecting:
				xlog.Info("Connecting to Slack with Socket Mode...")
			case socketmode.EventTypeConnectionError:
				xlog.Info("Connection failed. Retrying later...")
			case socketmode.EventTypeConnected:
				xlog.Info("Connected to Slack with Socket Mode.")
			case socketmode.EventTypeEventsAPI:
				eventsAPIEvent, ok := evt.Data.(slackevents.EventsAPIEvent)
				if !ok {
					xlog.Debug(fmt.Sprintf("Ignored %+v\n", evt))

					continue
				}

				client.Ack(*evt.Request)

				switch eventsAPIEvent.Type {
				case slackevents.CallbackEvent:
					innerEvent := eventsAPIEvent.InnerEvent

					b, err := api.AuthTest()
					if err != nil {
						fmt.Printf("Error getting auth test: %v", err)
					}

					switch ev := innerEvent.Data.(type) {
					case *slackevents.MessageEvent:
						if t.channelID == "" && !t.alwaysReply || // If we have set alwaysReply and no channelID
							t.channelID != ev.Channel { // If we have a channelID and it's not the same as the event channel
							// Skip messages from other channels
							xlog.Info("Skipping reply to channel", ev.Channel, t.channelID)
							continue
						}

						if b.UserID == ev.User {
							// Skip messages from ourselves
							continue
						}

						message := cleanUpUsernameFromMessage(ev.Text, b)
						go func() {

							ts := ev.ThreadTimeStamp

							res := a.Ask(
								agent.WithText(message),
							)

							_, _, err = api.PostMessage(ev.Channel,
								slack.MsgOptionText(res.Response, false),
								slack.MsgOptionPostMessageParameters(slack.PostMessageParameters{LinkNames: 1}),
								slack.MsgOptionTS(ts))
							if err != nil {
								xlog.Error(fmt.Sprintf("Error posting message: %v", err))
							}
						}()
					case *slackevents.AppMentionEvent:

						if b.UserID == ev.User {
							// Skip messages from ourselves
							continue
						}
						message := cleanUpUsernameFromMessage(ev.Text, b)

						// strip our id from the message
						xlog.Info("Message", message)

						go func() {
							ts := ev.ThreadTimeStamp

							var threadMessages []openai.ChatCompletionMessage

							if ts != "" {
								// Fetch the thread messages
								messages, _, _, err := api.GetConversationReplies(&slack.GetConversationRepliesParameters{
									ChannelID: ev.Channel,
									Timestamp: ts,
								})
								if err != nil {
									xlog.Error(fmt.Sprintf("Error fetching thread messages: %v", err))
								} else {
									for _, msg := range messages {
										role := "assistant"
										if msg.User != b.UserID {
											role = "user"
										}
										threadMessages = append(threadMessages,
											openai.ChatCompletionMessage{
												Role:    role,
												Content: cleanUpUsernameFromMessage(msg.Text, b),
											},
										)

									}
								}
							} else {
								threadMessages = append(threadMessages, openai.ChatCompletionMessage{
									Role:    "user",
									Content: cleanUpUsernameFromMessage(message, b),
								})
							}

							res := a.Ask(
								//	agent.WithText(message),
								agent.WithConversationHistory(threadMessages),
							)

							if ts != "" {
								_, _, err = api.PostMessage(ev.Channel,
									slack.MsgOptionText(res.Response, false),
									slack.MsgOptionPostMessageParameters(slack.PostMessageParameters{LinkNames: 1}),
									slack.MsgOptionTS(ts))
							} else {
								_, _, err = api.PostMessage(ev.Channel,
									slack.MsgOptionText(res.Response, false),
									slack.MsgOptionPostMessageParameters(slack.PostMessageParameters{LinkNames: 1}),
									slack.MsgOptionTS(ev.TimeStamp))
							}
							if err != nil {
								xlog.Error(fmt.Sprintf("Error posting message: %v", err))
							}
						}()
					case *slackevents.MemberJoinedChannelEvent:
						xlog.Error(fmt.Sprintf("user %q joined to channel %q", ev.User, ev.Channel))
					}
				default:
					client.Debugf("unsupported Events API event received")
				}
			default:
				xlog.Error(fmt.Sprintf("Unexpected event type received: %s", evt.Type))
			}
		}
	}()

	client.RunContext(a.Context())
}