teafolio/main.go

package main

import (
	"context"
	"flag"
	"log"
	"net/http"
	"regexp"
	"strings"
	"sync"

	"github.com/BurntSushi/toml"
	"golang.org/x/sync/semaphore"
)

type Config struct {
	BindTo string
	Gitea  struct {
		URL, Org       string
		MaxConnections int64
	}
	Redirect map[string]string
	Template struct {
		AppName             string
		HomepageHeaderHTML  string
		CustomLogoPngBase64 string
	}
}

type Application struct {
	cfg Config

	rxRepoPage, rxRepoImage *regexp.Regexp
	apiSem                  *semaphore.Weighted

	reposMut               sync.RWMutex
	reposCache             []Repo // Sorted by recently-created-first
	reposAlphabeticalOrder map[string]int
}

func main() {
	app := Application{
		rxRepoPage:  regexp.MustCompile(`^/([^/]+)/?$`),
		rxRepoImage: regexp.MustCompile(`^/:banner/([^/]+)/?$`),
	}

	configFile := flag.String(`ConfigFile`, `config.toml`, `Configuration file in TOML format`)
	flag.Parse()

	_, err := toml.DecodeFile(*configFile, &app.cfg)
	if err != nil {
		log.Fatalf("toml.DecodeFile: %s", err.Error())
	}

	// Assert Gitea URL always has trailing slash
	if !strings.HasSuffix(app.cfg.Gitea.URL, `/`) {
		app.cfg.Gitea.URL += `/`
	}

	// Create semaphore
	if app.cfg.Gitea.MaxConnections == 0 { // unlimited
		app.apiSem = semaphore.NewWeighted(99999)
	} else {
		app.apiSem = semaphore.NewWeighted(app.cfg.Gitea.MaxConnections)
	}

	// Sync worker
	go app.syncWorker(context.Background())

	log.Printf("Starting web server on [%s]...", app.cfg.BindTo)
	log.Fatal(http.ListenAndServe(app.cfg.BindTo, &app))
}