save servers to pocketbase by user id

This commit is contained in:
2023-10-06 23:02:05 +02:00
parent 34f3bcac3d
commit d971843fee
4 changed files with 47 additions and 0 deletions
+13
View File
@@ -44,6 +44,7 @@ func Run() {
var conversationHistoryMap = make(map[string][]openai.ChatCompletionMessage)
var sshConnections = make(map[string]*SSHConnection)
func hasAdminRole(roles []string) bool {
for _, role := range roles {
if role == AllowedUserID {
@@ -129,6 +130,18 @@ func newMessage(discord *discordgo.Session, message *discordgo.MessageCreate) {
// Store the SSH connection for later use
sshConnections[message.Author.ID] = sshConn
// Save server information to PocketBase
serverInfo := &pb.ServerInfo{
UserID: message.Author.ID,
ConnectionDetails: connectionDetails,
}
err = pb.CreateRecord("servers", serverInfo)
if err != nil {
log.Error(err)
discord.ChannelMessageSend(message.ChannelID, "Error saving server information.")
return
}
discord.ChannelMessageSend(message.ChannelID, "Connected to remote server!")
} else {
discord.ChannelMessageSend(message.ChannelID, "You are not authorized to use this command.")