init

src/main.go

@@ -0,0 +1,57 @@
+package main
+
+import (
+	"log"
+	"os"
+
+	"github.com/gin-gonic/gin"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+)
+
+var db *gorm.DB
+
+func main() {
+	_ = os.MkdirAll("uploads", 0755)
+
+	var err error
+	db, err = gorm.Open(sqlite.Open("files.db"), &gorm.Config{})
+	if err != nil {
+		log.Fatal("DB Connection failed:", err)
+	}
+	db.AutoMigrate(&FileRecord{})
+
+	go cleanupWorker()
+
+	router := gin.Default()
+	router.MaxMultipartMemory = 100 << 20 // 100 MiB limit
+	router.LoadHTMLGlob("templates/*")
+
+	// Public Routes
+	router.GET("/", func(c *gin.Context) { c.HTML(200, "index.html", nil) })
+	router.POST("/api/upload", uploadHandler)
+	router.GET("/f/:id", downloadHandler)
+	router.DELETE("/api/file/:del_id", deleteHandler)
+
+	// Protected Admin Routes
+	admin := router.Group("/admin", gin.BasicAuth(gin.Accounts{
+		"admin": "password123", // CHANGE THIS
+	}))
+
+	admin.GET("/", func(c *gin.Context) {
+		var files []FileRecord
+		db.Order("created_at desc").Find(&files)
+		c.HTML(200, "admin.html", gin.H{"Files": files})
+	})
+
+	admin.POST("/delete/:id", func(c *gin.Context) {
+		var record FileRecord
+		if err := db.First(&record, "id = ?", c.Param("id")).Error; err == nil {
+			performDeletion(&record)
+		}
+		c.Redirect(303, "/admin")
+	})
+
+	log.Println("Server starting at http://localhost:8080")
+	router.Run(":8080")
+}