jellyfixer/main.go

// nolint:gomnd,exhaustivestruct
package main

import (
	"context"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"time"

	"git.xsfx.dev/xsteadfastx/logginghandler"
	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	"github.com/spf13/viper"
)

// Logic Jellyfin URL.
type Logic struct {
	InternalURL string
	ExternalURL string
}

// PublicInfo Chromecast configuration Payload.
type PublicInfo struct {
	ID                     string
	LocalAddress           string
	OperatingSystem        string
	ProductName            string
	ServerName             string
	StartupWizardCompleted bool
	Version                string
}

func (l Logic) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	logger := logginghandler.Logger(r)

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) // nolint:gomnd
	defer cancel()

	req, err := http.NewRequestWithContext(ctx, http.MethodGet, fmt.Sprintf("%s/System/Info/Public", l.InternalURL), nil)
	if err != nil {
		logger.Error().Msg(err.Error())
		http.Error(w, "could not create request jellyfin", http.StatusInternalServerError)

		return
	}

	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		logger.Error().Msg(err.Error())
		http.Error(w, "could not access jellyfin", http.StatusInternalServerError)

		return
	}
	defer resp.Body.Close()

	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		logger.Error().Msg(err.Error())
		http.Error(w, "could not read body", http.StatusInternalServerError)

		return
	}

	var jd PublicInfo
	if err := json.Unmarshal(body, &jd); err != nil {
		logger.Error().Msg(err.Error())
		http.Error(w, "could not unmarshal body", http.StatusInternalServerError)

		return
	}

	jd.LocalAddress = l.ExternalURL

	nd, err := json.Marshal(jd)
	if err != nil {
		logger.Error().Msg(err.Error())
		http.Error(w, "could not marshal body", http.StatusInternalServerError)

		return
	}

	if _, err := w.Write(nd); err != nil {
		logger.Error().Msg(err.Error())
	}
}

func main() {
	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
	log.Logger = log.Logger.With().Caller().Logger()

	viper.SetEnvPrefix("jellyfixer")

	if err := viper.BindEnv("internal_url"); err != nil {
		log.Fatal().Msg(err.Error())
	}

	if err := viper.BindEnv("external_url"); err != nil {
		log.Fatal().Msg(err.Error())
	}

	if len(os.Args) == 2 {
		viper.SetConfigFile(os.Args[1])
	}

	viper.AutomaticEnv()

	if viper.GetString("internal_url") == "" {
		log.Fatal().Msg("needs key 'internal_url'")
	}

	if viper.GetString("external_url") == "" {
		log.Fatal().Msg("needs key 'external_url'")
	}

	l := Logic{
		InternalURL: viper.GetString("internal_url"),
		ExternalURL: viper.GetString("external_url"),
	}
	handler := logginghandler.Handler(l)

	http.Handle("/", handler)
	log.Info().Msg("starting server...")
	log.Fatal().Msg(http.ListenAndServe("0.0.0.0:8088", nil).Error())
}