Initial commit

backend/internal/api/storage.go

@@ -0,0 +1,98 @@
+package api
+
+import (
+	"fmt"
+	"net/http"
+	"os"
+	"path/filepath"
+	"syscall"
+
+	"clickploy/internal/db"
+	"clickploy/internal/models"
+
+	"github.com/gin-gonic/gin"
+)
+
+type CreateDatabaseRequest struct {
+	Name string `json:"name" binding:"required"`
+	Type string `json:"type" binding:"required,oneof=sqlite"`
+}
+
+type StorageStatsResponse struct {
+	TotalMB float64 `json:"total_mb"`
+	UsedMB  float64 `json:"used_mb"`
+}
+
+func (h *Handler) RegisterStorageRoutes(r *gin.Engine) {
+	api := r.Group("/api/storage")
+	api.Use(AuthMiddleware())
+	{
+		api.GET("/stats", h.handleGetStorageStats)
+		api.GET("/databases", h.handleListDatabases)
+		api.POST("/databases", h.handleCreateDatabase)
+	}
+}
+
+func (h *Handler) handleGetStorageStats(c *gin.Context) {
+	var stat syscall.Statfs_t
+	wd, _ := os.Getwd()
+
+	if err := syscall.Statfs(wd, &stat); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get disk stats"})
+		return
+	}
+
+	totalBytes := stat.Blocks * uint64(stat.Bsize)
+	availBytes := stat.Bavail * uint64(stat.Bsize)
+	usedBytes := totalBytes - availBytes
+
+	c.JSON(http.StatusOK, StorageStatsResponse{
+		TotalMB: float64(totalBytes) / 1024 / 1024,
+		UsedMB:  float64(usedBytes) / 1024 / 1024,
+	})
+}
+
+func (h *Handler) handleListDatabases(c *gin.Context) {
+	userId := c.GetUint("userID")
+	var dbs []models.Database
+	if err := db.DB.Where("owner_id = ?", userId).Find(&dbs).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to list databases"})
+		return
+	}
+	c.JSON(http.StatusOK, dbs)
+}
+
+func (h *Handler) handleCreateDatabase(c *gin.Context) {
+	userId := c.GetUint("userID")
+	var req CreateDatabaseRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	newDB := models.Database{
+		Name:    req.Name,
+		Type:    req.Type,
+		Status:  "available",
+		OwnerID: userId,
+		SizeMB:  0,
+	}
+
+	dataDir := "./data/user_dbs"
+	os.MkdirAll(dataDir, 0755)
+
+	dbPath := filepath.Join(dataDir, fmt.Sprintf("%d_%s.db", userId, req.Name))
+	file, err := os.Create(dbPath)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create database file"})
+		return
+	}
+	file.Close()
+
+	if err := db.DB.Create(&newDB).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to save database record"})
+		return
+	}
+
+	c.JSON(http.StatusOK, newDB)
+}