PixivFE/main.go

package main

import (
	"context"
	"fmt"
	"log"
	"net"
	"net/http"
	"os"
	"os/exec"
	"runtime"
	"sync"
	"syscall"
	"time"

	"codeberg.org/vnpower/pixivfe/v2/config"
	"codeberg.org/vnpower/pixivfe/v2/server/audit"
	"codeberg.org/vnpower/pixivfe/v2/server/middleware"
	"codeberg.org/vnpower/pixivfe/v2/server/proxy_checker"
	"codeberg.org/vnpower/pixivfe/v2/server/template"
)

func main() {
	if err := config.GlobalConfig.LoadConfig(); err != nil {
		log.Fatalln(err)
	}

	audit.Init(config.GlobalConfig.InDevelopment)
	template.Init(config.GlobalConfig.InDevelopment, "assets/views")

	// Conditionally initialize and start the proxy checker
	if config.GlobalConfig.ProxyCheckEnabled {
		var wg_firstCheck sync.WaitGroup
		wg_firstCheck.Add(1)

		go func() {
			ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.ProxyCheckTimeout)
			proxy_checker.CheckProxies(ctx)
			cancel()
			wg_firstCheck.Done()
			if config.GlobalConfig.ProxyCheckInterval != 0 {
				for {
					time.Sleep(config.GlobalConfig.ProxyCheckInterval)
					ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.ProxyCheckTimeout)
					proxy_checker.CheckProxies(ctx)
					cancel()
				}
			}
		}()

		// Wait for the first proxy check to complete
		log.Println("Waiting for initial proxy check to complete...")
		wg_firstCheck.Wait()
		log.Println("Initial proxy check completed.")
	} else {
		log.Println("Skipping proxy checker initialization.")
	}

	log.Println("Starting server...")

	router := middleware.DefineRoutes()
	// the first middleware is the most outer / first executed one
	router.Use(middleware.ProvideUserContext) // needed for everything else
	router.Use(middleware.LogRequest)         // all pages need this
	router.Use(middleware.SetPrivacyHeaders)  // all pages need this
	router.Use(middleware.HandleError)        // if the inner handler fails, this shows the error page instead
	router.Use(middleware.InitializeRateLimiter())

	// watch and compile sass when in development mode
	if config.GlobalConfig.InDevelopment {
		go run_sass()
	}

	// Listen
	err := http.Serve(chooseListener(), router)
	if err != http.ErrServerClosed {
		log.Print(err)
	}
}

func run_sass() {
	cmd := exec.Command("sass", "--watch", "assets/css")
	cmd.Stdout = os.Stderr // Sass quirk
	cmd.Stderr = os.Stderr
	cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true, Pdeathsig: syscall.SIGHUP}
	runtime.LockOSThread() // Go quirk https://github.com/golang/go/issues/27505
	err := cmd.Run()
	if err != nil {
		log.Print(fmt.Errorf("when running sass: %w", err))
	}
}

func chooseListener() net.Listener {
	var l net.Listener
	if config.GlobalConfig.UnixSocket != "" {
		ln, err := net.Listen("unix", config.GlobalConfig.UnixSocket)
		if err != nil {
			panic(err)
		}
		l = ln
		log.Printf("Listening on domain socket %v\n", config.GlobalConfig.UnixSocket)
	} else {
		addr := config.GlobalConfig.Host + ":" + config.GlobalConfig.Port
		ln, err := net.Listen("tcp", addr)
		if err != nil {
			log.Panicf("failed to listen: %v", err)
		}
		l = ln
		addr = ln.Addr().String()
		log.Printf("Listening on http://%v/\n", addr)
	}
	return l
}