test

main.go

@@ -1,28 +1,40 @@
 package main
 
 import (
-	"Hyst2Auth/app"
+	"Hyst2Node/app"
 	"log"
 	"net/http"
+	"time"
+
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+	"github.com/joho/godotenv"
 )
 
 func main() {
-	app.LoadIP()
-	app.LoadConfig()
-	_, err := app.LoadProvider()
-	if err != nil {
-		panic(err)
+	if err := godotenv.Load(".env"); err != nil {
+		log.Println("Error loading .env file. If you use docker, then you can ignore this.")
 	}
 
-	r := http.NewServeMux()
-	r.HandleFunc("/add", app.AddUser)
-	r.HandleFunc("/delete", app.DeleteUser)
-	r.HandleFunc("/users", app.AllUsers)
+	app.LoadConfig()
 
-	r.HandleFunc("/connect", app.GetUserURL)
+	r := chi.NewRouter()
+	r.Use(middleware.RequestID)
+	r.Use(middleware.RealIP)
+	r.Use(middleware.Logger)
+	r.Use(middleware.Recoverer)
+	r.Use(middleware.Timeout(60 * time.Second))
 
-	r.HandleFunc("/auth", app.DoAuth)
+	r.Post("/add", app.AddUser)
+	r.Post("/delete", app.DeleteUser)
+	r.Get("/users", app.AllUsers)
 
+	r.Post("/connect", app.GetUserURL)
+	r.Get("/sub/{id}", app.Sub)
+
+	r.Post("/auth", app.DoAuth)
+
+	defer app.ClosePsql()
 	log.Println("Listening on :8080")
 	if err := http.ListenAndServe(":8080", r); err != nil {
 		log.Fatal(err)