API redesign, better styling for the blog page

api/main.go

@@ -1,42 +1,78 @@
 package main
 
 import (
-	"database/sql"
 	"log"
+	"net/http"
 
 	"github.com/gofiber/fiber/v2"
+	"github.com/ngn13/website/api/database"
 	"github.com/ngn13/website/api/routes"
 )
 
-func CorsMiddleware(c *fiber.Ctx) error {
-  c.Set("Access-Control-Allow-Origin", "*")
-  c.Set("Access-Control-Allow-Headers", 
-  "Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization, accept, origin, Cache-Control, X-Requested-With")
-  c.Set("Access-Control-Allow-Methods", "PUT, DELETE, GET")
-  return c.Next()
-}
-
 func main() {
-  app := fiber.New(fiber.Config{
-    DisableStartupMessage: true,
-  })
-  app.Use(CorsMiddleware)
+	var (
+		app *fiber.App
+		db  database.Type
+		err error
+	)
 
-  db, err := sql.Open("sqlite3", "data.db")
-  if err != nil {
-    log.Fatal("Cannot connect to the database: "+err.Error())
-  }
+	if err = db.Open("data.db"); err != nil {
+		log.Fatalf("Cannot access the database: %s", err.Error())
+		return
+	}
+	defer db.Close()
 
-  log.Println("Creating tables")
-  routes.BlogDb(db)
-  routes.ServicesDb(db)
-  routes.Setup(app, db)
+	app = fiber.New(fiber.Config{
+		DisableStartupMessage: true,
+	})
 
-  log.Println("Starting web server at port 7001")
-  err = app.Listen("0.0.0.0:7001")
-  if err != nil {
-    log.Printf("Error starting the webserver: %s", err.Error())
-  }
+	app.Use("*", func(c *fiber.Ctx) error {
+		c.Set("Access-Control-Allow-Origin", "*")
+		c.Set("Access-Control-Allow-Headers",
+			"Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization, accept, origin, Cache-Control, X-Requested-With")
+		c.Set("Access-Control-Allow-Methods", "PUT, DELETE, GET")
+		c.Locals("database", &db)
+		return c.Next()
+	})
 
-  defer db.Close()
+	// index route
+	app.Get("/", func(c *fiber.Ctx) error {
+		return c.Send([]byte("o/"))
+	})
+
+	// blog routes
+	blog_routes := app.Group("/blog")
+	blog_routes.Get("/feed.atom", routes.GetAtomFeed)
+	blog_routes.Get("/feed.rss", routes.GetRSSFeed)
+	blog_routes.Get("/feed.json", routes.GetJSONFeed)
+	blog_routes.Get("/sum", routes.SumPost)
+	blog_routes.Get("/get", routes.GetPost)
+	blog_routes.Get("/vote/set", routes.VoteSet)
+	blog_routes.Get("/vote/status", routes.VoteStat)
+
+	// service routes
+	service_routes := app.Group("services")
+	service_routes.Get("/all", routes.GetServices)
+
+	// admin routes
+	admin_routes := app.Group("admin")
+	admin_routes.Use("*", routes.AuthMiddleware)
+	admin_routes.Get("/login", routes.Login)
+	admin_routes.Get("/logout", routes.Logout)
+	admin_routes.Put("/service/add", routes.AddService)
+	admin_routes.Delete("/service/remove", routes.RemoveService)
+	admin_routes.Put("/blog/add", routes.AddPost)
+	admin_routes.Delete("/blog/remove", routes.RemovePost)
+
+	// 404 routes
+	app.All("*", func(c *fiber.Ctx) error {
+		return c.Status(http.StatusNotFound).JSON(fiber.Map{
+			"error": "Requested endpoint not found",
+		})
+	})
+
+	log.Println("Starting web server at port 7001")
+	if err = app.Listen("0.0.0.0:7001"); err != nil {
+		log.Printf("Error starting the webserver: %s", err.Error())
+	}
 }