wrapperr/main.go

package main

import (
	"aunefyren/wrapperr/files"
	"aunefyren/wrapperr/routes"
	"aunefyren/wrapperr/utilities"
	"flag"
	"fmt"
	"log"
	"net/http"
	"os"
	"strconv"
	"time"

	"github.com/gorilla/mux"

	_ "time/tzdata"
)

func main() {

	utilities.PrintASCII()

	// Create and define file for logging
	file, err := os.OpenFile("config/wrapperr.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0666)
	if err != nil {
		fmt.Println("Failed to load configuration file. Error: " + err.Error())

		os.Exit(1)
	}

	config, err := files.GetConfig()
	if err != nil {
		fmt.Println("Failed to load configuration file. Error: " + err.Error())

		os.Exit(1)
	}

	// Set time zone from config if it is not empty
	if config.Timezone != "" {
		loc, err := time.LoadLocation(config.Timezone)
		if err != nil {

			fmt.Println("Failed to set time zone from config. Error: " + err.Error())
			fmt.Println("Removing value...")

			config.Timezone = ""
			err = files.SaveConfig(config)
			if err != nil {
				fmt.Println("Failed to set new time zone in the config. Error: " + err.Error())
				fmt.Println("Exiting...")
				os.Exit(1)
			}

		} else {
			time.Local = loc
			fmt.Println("Timezone set to " + config.Timezone + ".")
		}
	}

	// Set log file is logging is enabled
	if config.UseLogs {
		log.SetOutput(file)

		fmt.Println("Log file enabled.")
		log.Println("Log file enabled.")
	}

	// Print current version
	fmt.Println("Running version " + config.WrapperrVersion + ".")
	if config.UseLogs {
		log.Println("Running version " + config.WrapperrVersion + ".")
	}

	// Define port variable with the port from the config file as default
	var port int
	flag.IntVar(&port, "port", config.WrapperrPort, "The port Wrapperr is listening on.")

	// Parse the flags from input
	flag.Parse()

	// Alert what port is in use
	fmt.Println("Starting Wrapperr on port " + strconv.Itoa(port) + ".")
	if config.UseLogs {
		log.Println("Starting Wrapperr on port " + strconv.Itoa(port) + ".")
	}

	// Assign routes
	router := mux.NewRouter().StrictSlash(true)

	var root string
	if config.WrapperrRoot != "" {
		root = "/" + config.WrapperrRoot
	} else {
		root = ""
	}

	// Admin auth routes
	router.HandleFunc(root+"/api/validate/admin", routes.ApiValidateAdmin)
	router.HandleFunc(root+"/api/get/config", routes.ApiGetConfig)
	router.HandleFunc(root+"/api/get/log", routes.ApiGetLog)
	router.HandleFunc(root+"/api/set/config", routes.ApiSetConfig)
	router.HandleFunc(root+"/api/update/admin", routes.ApiUpdateAdmin)

	// No-auth routes
	router.HandleFunc(root+"/api/get/config-state", routes.ApiWrapperrConfigured)
	router.HandleFunc(root+"/api/login/admin", routes.ApiLogInAdmin)
	router.HandleFunc(root+"/api/get/wrapperr-version", routes.ApiGetWrapperrVersion)
	router.HandleFunc(root+"/api/get/admin-state", routes.ApiGetAdminState)
	router.HandleFunc(root+"/api/get/functions", routes.ApiGetFunctions)
	router.HandleFunc(root+"/api/create/admin", routes.ApiCreateAdmin)
	router.HandleFunc(root+"/api/get/tautulli-connection", routes.ApiGetTautulliConncection)
	router.HandleFunc(root+"/api/get/share-link", routes.ApiGetShareLink)

	// User auth routes
	router.HandleFunc(root+"/api/get/login-url", routes.ApiGetLoginURL)
	router.HandleFunc(root+"/api/login/plex-auth", routes.ApiLoginPlexAuth)
	router.HandleFunc(root+"/api/validate/plex-auth", routes.ApiValidatePlexAuth)
	router.HandleFunc(root+"/api/create/share-link", routes.ApiCreateShareLink)
	router.HandleFunc(root+"/api/get/user-share-link", routes.ApiGetUserShareLink)
	router.HandleFunc(root+"/api/delete/user-share-link", routes.ApiDeleteUserShareLink)

	// Get stats route
	router.HandleFunc(root+"/api/get/statistics", routes.ApiWrapperGetStatistics)

	// Assets route
	assetsFileServer := http.FileServer(http.Dir("./web/assets/"))
	router.PathPrefix(root + "/assets").Handler(http.StripPrefix(root+"/assets", assetsFileServer))

	// JS route
	jsFileServer := http.FileServer(http.Dir("./web/js/"))
	router.PathPrefix(root + "/js").Handler(http.StripPrefix(root+"/js", jsFileServer))

	// HTML frontpage route
	router.HandleFunc(root+"/", func(w http.ResponseWriter, r *http.Request) {
		// Using the http.ServeFile function to serve the frontpage.html file
		http.ServeFile(w, r, "./web/html/frontpage.html")
	})

	// HTML admin route
	router.HandleFunc(root+"/admin", func(w http.ResponseWriter, r *http.Request) {
		// Using the http.ServeFile function to serve the admin.html file
		http.ServeFile(w, r, "./web/html/admin.html")
	})

	// TXT robots route
	router.HandleFunc(root+"/robots.txt", func(w http.ResponseWriter, r *http.Request) {
		// Using the http.ServeFile function to serve the robots.txt file
		http.ServeFile(w, r, "./web/txt/robots.txt")
	})

	// Start web-server
	log.Fatal(http.ListenAndServe(":"+strconv.Itoa(port), router))
}