muzi/web/web.go

package web

// Main web UI controller

import (
	"context"
	"fmt"
	"html/template"
	"net/http"
	"net/url"
	"os"

	"muzi/config"
	"muzi/db"
	"muzi/scrobble"

	"github.com/go-chi/chi/v5"
	"github.com/go-chi/chi/v5/middleware"
)

// 50 MiB
const maxHeaderSize int64 = 50 * 1024 * 1024

func serverAddrStr() string {
	return config.Get().Server.Address
}

// Holds all the parsed HTML templates
var templates *template.Template

// Declares all functions for the HTML templates and parses them
func init() {
	funcMap := template.FuncMap{
		"sub":                 sub,
		"add":                 add,
		"formatInt":           formatInt,
		"formatTimestamp":     formatTimestamp,
		"formatTimestampFull": formatTimestampFull,
		"urlquery":            url.QueryEscape,
	}
	templates = template.Must(template.New("").Funcs(funcMap).ParseGlob("./templates/*.gohtml"))
}

// Returns T/F if a user is found in the users table
func hasUsers(ctx context.Context) bool {
	var count int
	err := db.Pool.QueryRow(ctx, "SELECT COUNT(*) FROM users;").Scan(&count)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Error checking for users: %v\n", err)
		return false
	}
	return count > 0
}

// Controls what is displayed at the root URL.
// If logged in: Logged in user's profile page.
// If logged out: Login page.
// If no users in DB yet: Create account page.
func rootHandler() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if !hasUsers(r.Context()) {
			http.Redirect(w, r, "/createaccount", http.StatusSeeOther)
			return
		}

		username := getLoggedInUsername(r)
		if username == "" {
			http.Redirect(w, r, "/login", http.StatusSeeOther)
			return
		}

		http.Redirect(w, r, "/profile/"+username, http.StatusSeeOther)
	}
}

// Serves all pages at the specified address.
func Start() {
	addr := config.Get().Server.Address
	r := chi.NewRouter()
	r.Use(middleware.Logger)
	r.Handle("/files/*", http.StripPrefix("/files", http.FileServer(http.Dir("./static"))))
	r.Get("/", rootHandler())
	r.Get("/login", loginPageHandler())
	r.Get("/createaccount", createAccountPageHandler())
	r.Get("/profile/{username}", profilePageHandler())
	r.Get("/profile/{username}/artist/{artist}", artistPageHandler())
	r.Get("/profile/{username}/song/{artist}/{song}", songPageHandler())
	r.Get("/profile/{username}/album/{artist}/{album}", albumPageHandler())
	r.Get("/profile/{username}/album/{album}", func(w http.ResponseWriter, r *http.Request) {
		username := chi.URLParam(r, "username")
		albumTitle, _ := url.QueryUnescape(chi.URLParam(r, "album"))
		userId, err := getUserIdByUsername(r.Context(), username)
		if err != nil {
			http.Error(w, "User not found", http.StatusNotFound)
			return
		}
		albums, _, _ := db.SearchAlbums(userId, albumTitle)
		if len(albums) > 0 {
			album := albums[0]
			artist, _ := db.GetArtistById(album.ArtistId)
			http.Redirect(w, r, "/profile/"+username+"/album/"+url.QueryEscape(artist.Name)+"/"+url.QueryEscape(album.Title), http.StatusSeeOther)
			return
		}
		http.Error(w, "Album not found", http.StatusNotFound)
	})
	r.Post("/profile/{username}/artist/{id}/edit", editArtistHandler())
	r.Post("/profile/{username}/song/{id}/edit", editSongHandler())
	r.Post("/profile/{username}/album/{id}/edit", editAlbumHandler())
	r.Patch("/api/artist/{id}/edit", artistInlineEditHandler())
	r.Patch("/api/song/{id}/edit", songInlineEditHandler())
	r.Patch("/api/album/{id}/edit", albumInlineEditHandler())
	r.Patch("/api/artist/{id}/batch", artistBatchEditHandler())
	r.Patch("/api/song/{id}/batch", songBatchEditHandler())
	r.Patch("/api/album/{id}/batch", albumBatchEditHandler())
	r.Post("/api/upload/image", imageUploadHandler())
	r.Get("/search", searchHandler())
	r.Get("/import", importPageHandler())
	r.Post("/loginsubmit", loginSubmit)
	r.Post("/createaccountsubmit", createAccount)
	r.Post("/import/lastfm", importLastFMHandler)
	r.Post("/import/spotify", importSpotifyHandler)
	r.Get("/import/lastfm/progress", importLastFMProgressHandler)
	r.Get("/import/spotify/progress", importSpotifyProgressHandler)

	r.Handle("/2.0", scrobble.NewLastFMHandler())
	r.Handle("/2.0/", scrobble.NewLastFMHandler())
	r.Post("/1/submit-listens", http.HandlerFunc(scrobble.NewListenbrainzHandler().ServeHTTP))
	r.Route("/scrobble/spotify", func(r chi.Router) {
		r.Get("/authorize", http.HandlerFunc(scrobble.NewSpotifyHandler().ServeHTTP))
		r.Get("/callback", http.HandlerFunc(scrobble.NewSpotifyHandler().ServeHTTP))
	})

	r.Get("/settings/spotify-connect", spotifyConnectHandler)
	r.Get("/settings", settingsPageHandler())
	r.Post("/settings/generate-apikey", generateAPIKeyHandler)
	r.Post("/settings/update-spotify", updateSpotifyCredentialsHandler)
	fmt.Printf("WebUI starting on %s\n", addr)
	prot := http.NewCrossOriginProtection()
	http.ListenAndServe(addr, prot.Handler(r))
}