admin.go

package admin

import (
	"context"
	"net/http"

	"github.com/jirwin/quadlek/quadlek"
	"github.com/slack-go/slack"
	"go.uber.org/zap"
)

func shutdown(ctx context.Context, cmdChannel <-chan *quadlek.CommandMsg) {
	for {
		select {
		case cmdMsg := <-cmdChannel:
			cmdMsg.Command.Reply() <- &quadlek.CommandResp{
				Text: "Shutting down...",
			}
			cmdMsg.Bot.Stop()

		case <-ctx.Done():
			zap.L().Info("Exiting quit command.")
			return
		}
	}
}

func restartInteraction(ctx context.Context, interactionChannel <-chan *quadlek.InteractionMsg) {
	for {
		select {
		case scMsg := <-interactionChannel:
			callbackID := ""

			switch scMsg.Interaction.Type {
			case "view_submission":
				callbackID = scMsg.Interaction.View.CallbackID
			default:
				callbackID = scMsg.Interaction.CallbackID
			}

			switch callbackID {
			case "restart":
				r := slack.ModalViewRequest{
					Type: "modal",
					Title: &slack.TextBlockObject{
						Type: "plain_text",
						Text: "Restart Quadlek",
					},
					Blocks: slack.Blocks{
						BlockSet: []slack.Block{
							&slack.SectionBlock{
								Type: "section",
								Text: &slack.TextBlockObject{
									Type: "plain_text",
									Text: "Are you sure you'd like to restart quadlek?",
								},
							},
						},
					},
					Submit: &slack.TextBlockObject{
						Type: "plain_text",
						Text: "Confirm",
					},
					CallbackID: "restart-confirm-modal",
				}
				_, err := scMsg.Bot.OpenView(scMsg.Interaction.TriggerID, r)
				if err != nil {
					zap.L().Error("error opening view", zap.Error(err))
					continue
				}

			case "restart-confirm-modal":
				zap.L().Info("shutting down...")
				scMsg.Bot.Stop()
			}

		case <-ctx.Done():
			zap.L().Info("Exiting quit command.")
			return
		}
	}
}

func healthCheck(ctx context.Context, whChan <-chan *quadlek.WebhookMsg) {
	for {
		select {
		case whMsg := <-whChan:
			// respond to webhook
			whMsg.ResponseWriter.WriteHeader(http.StatusOK)
			_, err := whMsg.ResponseWriter.Write([]byte{})
			if err != nil {
				continue
			}
			whMsg.Done <- true

		case <-ctx.Done():
			zap.L().Info("Exiting healthcheck ")
			return
		}
	}
}

func Register(adminChannel string) quadlek.Plugin {
	return quadlek.MakePlugin(
		"admin",
		[]quadlek.Command{
			quadlek.MakeCommand("shutdown", shutdown),
		},
		nil,
		nil,
		[]quadlek.Webhook{
			quadlek.MakeWebhook("healthCheck", healthCheck),
		},
		nil,
	)
}

func RegisterInteraction() quadlek.InteractionPlugin {
	return quadlek.MakeInteractionPlugin(
		"restart-quadlek",
		[]quadlek.Interaction{
			quadlek.MakeInteraction("restart", restartInteraction),
		},
	)
}