main.go

package main

import (
	"errors"
	"fmt"
	"io"
	"net/http"
	"os"
	"strconv"
	"strings"
	"time"

	"server-go/common"
	"server-go/database"
	"server-go/legacy_routes"
	"server-go/modules"
	"server-go/routes"

	"github.com/go-chi/chi"
	"github.com/go-chi/httprate"
	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promhttp"
)

type Mux struct {
	*chi.Mux
}

var Counters = map[string]prometheus.Counter{}
var TotalRequestCounter = prometheus.NewCounter(prometheus.CounterOpts{
	Name: "total_request",
	Help: "Total request count",
})
var URUserCounter = prometheus.NewCounterFunc(prometheus.CounterOpts{
	Name: "user_count",
	Help: "Count of user reviews users",
}, func() float64 {
	userCount, err := modules.GetURUserCount()

	if err != nil {
		return 0
	}

	return float64(userCount)
})

var ReviewCounter = prometheus.NewCounterFunc(prometheus.CounterOpts{
	Name: "review_count",
	Help: "Count of total user reviews",
}, func() float64 {
	count, err := modules.GetReviewCount()

	if err != nil {
		return 0
	}

	return float64(count)
})

func (c *Mux) HandleFunc(pattern string, handler func(http.ResponseWriter, *http.Request)) {

	metric := strings.NewReplacer("{", "", "}", "", "/", "", "*", "", "-", "").Replace(pattern)

	if metric == "" {
		metric = "root"
	}

	if _, exists := Counters[metric]; !exists {
		Counters[metric] = prometheus.NewCounter(prometheus.CounterOpts{
			Name: metric,
			Help: "Number of requests on " + pattern,
		})
		prometheus.MustRegister(Counters[metric])
	}

	c.Mux.HandleFunc(pattern, func(w http.ResponseWriter, r *http.Request) {
		TotalRequestCounter.Inc()
		Counters[metric].Inc()

		handler(w, r)
	})
}

func cors(handler http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Access-Control-Allow-Origin", "*")
		w.Header().Set("Access-Control-Allow-Headers", "*")
		w.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE, PATCH")
		if r.Method == "OPTIONS" {
			w.WriteHeader(http.StatusOK)
			return
		}

		handler.ServeHTTP(w, r)
	})
}

func main() {

	common.InitCache()
	database.InitDB()

	prometheus.MustRegister(ReviewCounter)
	prometheus.MustRegister(URUserCounter)
	prometheus.MustRegister(TotalRequestCounter)

	optedOutUsers, err := modules.GetOptedOutUsers()
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

	common.OptedOut = append(common.OptedOut, optedOutUsers...)

	mux := Mux{chi.NewRouter()}

	mux.Use(cors)
	mux.Use(httprate.LimitByRealIP(2,1 * time.Second))

	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, "artgallery/index.html")
	})

	mux.Handle("/static/*", http.StripPrefix("/static/", http.FileServer(http.Dir("artgallery/static"))))

	mux.Handle("/assets/*", http.StripPrefix("/assets/", http.FileServer(http.Dir("artgallery/assets"))))

	mux.HandleFunc("/interactions", routes.HandleInteractions)

	mux.HandleFunc("/receiveToken/{token}", routes.ReceiveToken)

	//StupidityDB

	mux.HandleFunc("/vote", routes.VoteStupidity)

	mux.HandleFunc("/getuser", routes.GetStupidity)

	mux.HandleFunc("/auth", routes.StupidityDBAuth)

	//ReviewDB

	mux.HandleFunc("/URauth", legacy_routes.ReviewDBAuth)

	mux.HandleFunc("/api/reviewdb/users", routes.GetUserInfo)

	mux.HandleFunc("/admins", routes.Admins)

	mux.HandleFunc("/api/reviewdb/auth", routes.ReviewDBAuth)

	mux.HandleFunc("/api/reviewdb/users/{discordid}/reviews", routes.HandleReviews)

	mux.HandleFunc("/api/reviewdb/badges", routes.GetAllBadges)

	mux.HandleFunc("/api/reviewdb/reports", routes.ReportReview)

	mux.HandleFunc("/api/reviewdb/reviews", routes.SearchReview)

	mux.HandleFunc("/api/reviewdb/settings", routes.Settings)

	mux.HandleFunc("/api/reviewdb/authweb", routes.ReviewDBAuthWeb)

	mux.Put("/api/reviewdb/appeals", routes.AppealReview)

	mux.Group(func(r chi.Router) {
		r.Use(routes.AdminMiddleware)

		r.Route(("/api/reviewdb/admin"), func(r chi.Router) {
			r.Get("/filters", routes.GetFilters)
			r.Put("/filters", routes.AddFilter)
			r.Delete("/filters", routes.DeleteFilter)
			r.Get("/reload", routes.ReloadConfig)
		})
	})

	mux.HandleFunc("/api/reviewdb-twitter/auth", routes.ReviewDBTwitterAuth)
	mux.HandleFunc("/api/reviewdb-twitter/users/{profileid}/reviews", routes.HandleTwitterRoutes)
	mux.HandleFunc("/api/reviewdb-twitter/reports", routes.ReportTwitterReview)

	mux.HandleFunc("/error", func(w http.ResponseWriter, r *http.Request) {
		io.WriteString(w, "An Error occurred\n")
	})

	mux.HandleFunc("/getLastReviewID", func(w http.ResponseWriter, r *http.Request) {
		id := r.URL.Query().Get("discordid")

		w.Write([]byte(strconv.Itoa(int(modules.GetLastReviewID(id)))))
	})

	mux.Handle("/metrics", promhttp.Handler())

	err = http.ListenAndServe(":"+common.Config.Port, mux)
	if errors.Is(err, http.ErrServerClosed) {
		fmt.Printf("server closed\n")
	} else if err != nil {
		fmt.Printf("error starting server: %s\n", err)
		os.Exit(1)
	}
}