plaintext-encyclopedia/main.go

package main

// Copyright (c) 2024 Julian Müller (ChaoticByte)

import (
	"html/template"
	"log"
	"net/http"
	"os"
	"path"
	"strings"
)

var logger *log.Logger
var appTemplate *template.Template = template.New("app")
var db Database

type TemplateData struct {
	Title string
	SiteDescription string
	TOC map[string]string
	EntryTitle string
	Entry string
	Footer []template.HTML
}

func loadTemplate() {
	data, err := os.ReadFile(TemplateFile)
	if err != nil { logger.Panicln(err) }
	appTemplate.Parse(string(data))
}

func handleApplication(w http.ResponseWriter, req *http.Request) {
	var entry string
	var err error
	entryId := path.Base(req.URL.Path)
	if entryId != "/" {
		// load entry
		entry = db.Entries[entryId]
		if entry == "" { // redirect if entry doesn't exist (or is empty)
			http.Redirect(w, req, "/", http.StatusTemporaryRedirect)
		}
	}
	err = appTemplate.ExecuteTemplate(
		w, "app",
		TemplateData{
			TOC: db.Titles,
			Entry: entry,
			Title: MainTitle,
			EntryTitle: db.Titles[entryId],
			Footer: FooterContent,
			SiteDescription: SiteDescription,
		})
	if err != nil { logger.Println(err) }
}

func handleSearchAPI(w http.ResponseWriter, req *http.Request) {
	searchQuery := path.Base(req.URL.Path)
	if searchQuery == "" {
		w.WriteHeader(http.StatusBadRequest)
		return
	}
	results := db.searchForIds(searchQuery)
	for i, r := range results {
		results[i] = r + "|" + db.Titles[r]
	}
	w.Write([]byte(strings.Join(results, "\n")))
}

func main() {
	// get logger
	logger = log.Default()
	// load template
	loadTemplate()
	// build db
	db = BuildDB(EntriesDirectory)
	// handle static files
	staticHandler := http.StripPrefix("/static/", http.FileServer(http.Dir(StaticDirectory)))
	http.Handle("/static/", staticHandler)
	// handle application
	http.HandleFunc("/", handleApplication)
	http.HandleFunc("/search/", handleSearchAPI)
	// start server
	logger.Println("Starting server on", ServerListen)
	err := http.ListenAndServe(ServerListen, nil)
	if err != nil { logger.Panicln(err) }
}