configEditor.go

Rama master · réplica de lectura del repositorio original.

package main

import (
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"sort"
	"sync"
	"time"

	"github.com/gorilla/websocket"
	"github.com/sirupsen/logrus"

	"github.com/Luzifer/twitch-bot/v3/pkg/twitch"
	"github.com/Luzifer/twitch-bot/v3/plugins"
)

const (
	moduleConfigEditor    = "config-editor"
	websocketPingInterval = 30 * time.Second
)

var (
	availableActorDocs     []plugins.ActionDocumentation
	availableActorDocsLock sync.RWMutex

	upgrader = websocket.Upgrader{}
)

func registerActorDocumentation(doc plugins.ActionDocumentation) {
	availableActorDocsLock.Lock()
	defer availableActorDocsLock.Unlock()

	availableActorDocs = append(availableActorDocs, doc)
	sort.Slice(availableActorDocs, func(i, j int) bool {
		return availableActorDocs[i].Name < availableActorDocs[j].Name
	})
}

func init() {
	registerEditorAutoMessageRoutes()
	registerEditorFrontend()
	registerEditorGeneralConfigRoutes()
	registerEditorGlobalMethods()
	registerEditorRulesRoutes()
}

func registerEditorFrontend() {
	router.HandleFunc("/", func(w http.ResponseWriter, _ *http.Request) {
		f, err := configEditorFrontend.Open("editor/index.html")
		if err != nil {
			http.Error(w, fmt.Errorf("opening index.html: %w", err).Error(), http.StatusNotFound)
			return
		}

		if _, err = io.Copy(w, f); err != nil {
			logrus.WithError(err).Debug("writing frontend index response")
		}
	})

	router.HandleFunc("/editor/vars.json", func(w http.ResponseWriter, _ *http.Request) {
		w.Header().Set("Content-Type", "application/json")
		if err := json.NewEncoder(w).Encode(struct {
			DefaultBotScopes  []string
			IRCBadges         []string
			KnownEvents       []*string
			TemplateFunctions []string
			TwitchClientID    string
			Version           string
		}{
			DefaultBotScopes:  botDefaultScopes,
			IRCBadges:         twitch.KnownBadges,
			KnownEvents:       knownEvents,
			TemplateFunctions: tplFuncs.GetFuncNames(),
			TwitchClientID:    cfg.TwitchClient,
			Version:           version,
		}); err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
		}
	})

	router.PathPrefix("/editor").Handler(http.FileServer(configEditorFrontend))
}