Fix compilation errors in API handlers

api/internal/api/handlers.go

@@ -19,6 +19,7 @@ import (
 	"github.com/streamspace/streamspace/api/internal/tracker"
 	"github.com/streamspace/streamspace/api/internal/websocket"
 	"gopkg.in/yaml.v3"
+	corev1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime/schema"
 )
@@ -158,31 +159,39 @@ func (h *Handler) CreateSession(c *gin.Context) {
 	}
 
 	// Check user quota
-	quotaReq := &quota.SessionRequest{
-		UserID:  req.User,
-		Memory:  memory,
-		CPU:     cpu,
-		Storage: "50Gi", // Default storage quota check
-	}
-
-	quotaResult, err := h.quotaEnforcer.CheckSessionQuota(ctx, quotaReq)
+	// Parse CPU and memory to int64
+	requestedCPU, requestedMemory, err := h.quotaEnforcer.ValidateResourceRequest(cpu, memory)
 	if err != nil {
-		c.JSON(http.StatusInternalServerError, gin.H{
-			"error":   "Failed to check quota",
+		c.JSON(http.StatusBadRequest, gin.H{
+			"error":   "Invalid resource request",
 			"message": err.Error(),
 		})
 		return
 	}
 
-	if !quotaResult.Allowed {
+	// Get current usage by listing user's pods
+	podList, err := h.k8sClient.GetPods(ctx, h.namespace)
+	if err != nil {
+		log.Printf("Failed to get pods for quota check: %v", err)
+		// Continue with empty usage if we can't get pods
+		podList = &corev1.PodList{}
+	}
+
+	// Filter pods for this user
+	userPods := make([]corev1.Pod, 0)
+	for _, pod := range podList.Items {
+		if user, ok := pod.Labels["user"]; ok && user == req.User {
+			userPods = append(userPods, pod)
+		}
+	}
+
+	currentUsage := h.quotaEnforcer.CalculateUsage(userPods)
+
+	// Check if user can create session
+	if err := h.quotaEnforcer.CheckSessionCreation(ctx, req.User, requestedCPU, requestedMemory, 0, currentUsage); err != nil {
 		c.JSON(http.StatusForbidden, gin.H{
 			"error":   "Quota exceeded",
-			"message": quotaResult.Reason,
-			"quota": gin.H{
-				"current":   quotaResult.CurrentUsage,
-				"requested": quotaResult.RequestedUsage,
-				"available": quotaResult.AvailableQuota,
-			},
+			"message": err.Error(),
 		})
 		return
 	}
@@ -226,12 +235,6 @@ func (h *Handler) CreateSession(c *gin.Context) {
 		return
 	}
 
-	// Update quota usage
-	if err := h.quotaEnforcer.UpdateSessionQuota(ctx, req.User, memory, cpu, "50Gi", true); err != nil {
-		log.Printf("Failed to update quota usage: %v", err)
-		// Don't fail the request, but log the error
-	}
-
 	// Cache in database
 	if err := h.cacheSessionInDB(ctx, created); err != nil {
 		log.Printf("Failed to cache session in database: %v", err)
@@ -280,8 +283,8 @@ func (h *Handler) DeleteSession(c *gin.Context) {
 	ctx := context.Background()
 	sessionID := c.Param("id")
 
-	// Get session info before deletion (for quota tracking)
-	session, err := h.k8sClient.GetSession(ctx, h.namespace, sessionID)
+	// Verify session exists before deletion
+	_, err := h.k8sClient.GetSession(ctx, h.namespace, sessionID)
 	if err != nil {
 		c.JSON(http.StatusNotFound, gin.H{"error": "Session not found"})
 		return
@@ -293,15 +296,6 @@ func (h *Handler) DeleteSession(c *gin.Context) {
 		return
 	}
 
-	// Update quota usage (decrement)
-	if session.Resources.Memory != "" && session.Resources.CPU != "" {
-		if err := h.quotaEnforcer.UpdateSessionQuota(ctx, session.User,
-			session.Resources.Memory, session.Resources.CPU, "50Gi", false); err != nil {
-			log.Printf("Failed to update quota usage on session deletion: %v", err)
-			// Don't fail the request, quota will be cleaned up later
-		}
-	}
-
 	// Delete from database cache
 	if err := h.deleteSessionFromDB(ctx, sessionID); err != nil {
 		log.Printf("Failed to delete session from database: %v", err)

api/internal/api/stubs.go

@@ -13,6 +13,7 @@ import (
 	"github.com/gorilla/websocket"
 	corev1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
 	"k8s.io/apimachinery/pkg/runtime/schema"
 )
 
@@ -298,7 +299,7 @@ func (h *Handler) CreateResource(c *gin.Context) {
 
 // UpdateResource updates a K8s resource
 func (h *Handler) UpdateResource(c *gin.Context) {
-	resourceType := c.Param("type")     // e.g., "deployment", "service"
+	_ = c.Param("type") // Resource type not used; Kind from request body
 	resourceName := c.Param("name")
 	namespace := c.Query("namespace")
 	if namespace == "" {

api/internal/handlers/collaboration.go

@@ -254,26 +254,25 @@ import (
 	"fmt"
 	"net/http"
 	"strconv"
-	"strings"
 	"time"
 
 	"github.com/gin-gonic/gin"
 	"github.com/streamspace/streamspace/api/internal/db"
 )
 
 // Handler handles collaboration-related HTTP requests.
-type Handler struct {
+type CollaborationHandler struct {
 	// DB is the database connection for collaboration queries and updates.
 	DB *db.Database
 }
 
 // NewCollaborationHandler creates a new collaboration handler.
-func NewCollaborationHandler(database *db.Database) *Handler {
-	return &Handler{DB: database}
+func NewCollaborationHandler(database *db.Database) *CollaborationHandler {
+	return &CollaborationHandler{DB: database}
 }
 
 // canAccessSession checks if a user has access to a session.
-func (h *Handler) canAccessSession(userID, sessionID string) bool {
+func (h *CollaborationHandler) canAccessSession(userID, sessionID string) bool {
 	// Check if user owns the session
 	var owner string
 	err := h.DB.DB().QueryRow("SELECT user_id FROM sessions WHERE id = $1", sessionID).Scan(&owner)
@@ -417,7 +416,7 @@ type Point struct {
 }
 
 // CreateCollaborationSession creates a new collaboration session
-func (h *Handler) CreateCollaborationSession(c *gin.Context) {
+func (h *CollaborationHandler) CreateCollaborationSession(c *gin.Context) {
 	sessionID := c.Param("sessionId")
 	userID := c.GetString("user_id")
 
@@ -496,7 +495,7 @@ func (h *Handler) CreateCollaborationSession(c *gin.Context) {
 }
 
 // JoinCollaborationSession allows a user to join a collaboration
-func (h *Handler) JoinCollaborationSession(c *gin.Context) {
+func (h *CollaborationHandler) JoinCollaborationSession(c *gin.Context) {
 	collabID := c.Param("collabId")
 	userID := c.GetString("user_id")
 
@@ -616,7 +615,7 @@ func (h *Handler) JoinCollaborationSession(c *gin.Context) {
 }
 
 // LeaveCollaborationSession removes a user from collaboration
-func (h *Handler) LeaveCollaborationSession(c *gin.Context) {
+func (h *CollaborationHandler) LeaveCollaborationSession(c *gin.Context) {
 	collabID := c.Param("collabId")
 	userID := c.GetString("user_id")
 
@@ -650,7 +649,7 @@ func (h *Handler) LeaveCollaborationSession(c *gin.Context) {
 }
 
 // GetCollaborationParticipants lists all participants
-func (h *Handler) GetCollaborationParticipants(c *gin.Context) {
+func (h *CollaborationHandler) GetCollaborationParticipants(c *gin.Context) {
 	collabID := c.Param("collabId")
 	userID := c.GetString("user_id")
 
@@ -660,7 +659,7 @@ func (h *Handler) GetCollaborationParticipants(c *gin.Context) {
 		return
 	}
 
-	rows, err := h.DB.Query(`
+	rows, err := h.DB.DB().Query(`
 		SELECT cp.user_id, u.username, cp.role, cp.permissions, cp.cursor_position,
 		       cp.color, cp.is_active, cp.joined_at, cp.last_seen_at
 		FROM collaboration_participants cp
@@ -702,7 +701,7 @@ func (h *Handler) GetCollaborationParticipants(c *gin.Context) {
 }
 
 // UpdateParticipantRole updates a participant's role and permissions
-func (h *Handler) UpdateParticipantRole(c *gin.Context) {
+func (h *CollaborationHandler) UpdateParticipantRole(c *gin.Context) {
 	collabID := c.Param("collabId")
 	targetUserID := c.Param("userId")
 	userID := c.GetString("user_id")
@@ -741,7 +740,7 @@ func (h *Handler) UpdateParticipantRole(c *gin.Context) {
 // Chat Operations
 
 // SendChatMessage sends a message to the collaboration chat
-func (h *Handler) SendChatMessage(c *gin.Context) {
+func (h *CollaborationHandler) SendChatMessage(c *gin.Context) {
 	collabID := c.Param("collabId")
 	userID := c.GetString("user_id")
 
@@ -787,7 +786,7 @@ func (h *Handler) SendChatMessage(c *gin.Context) {
 }
 
 // GetChatHistory retrieves chat history
-func (h *Handler) GetChatHistory(c *gin.Context) {
+func (h *CollaborationHandler) GetChatHistory(c *gin.Context) {
 	collabID := c.Param("collabId")
 	userID := c.GetString("user_id")
 	limit, _ := strconv.Atoi(c.DefaultQuery("limit", "100"))
@@ -819,7 +818,7 @@ func (h *Handler) GetChatHistory(c *gin.Context) {
 	query += fmt.Sprintf(" ORDER BY cc.created_at DESC LIMIT $%d", argCount)
 	args = append(args, limit)
 
-	rows, err := h.DB.Query(query, args...)
+	rows, err := h.DB.DB().Query(query, args...)
 	if err != nil {
 		c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to retrieve chat"})
 		return
@@ -857,7 +856,7 @@ func (h *Handler) GetChatHistory(c *gin.Context) {
 // Annotation Operations
 
 // CreateAnnotation creates a new annotation
-func (h *Handler) CreateAnnotation(c *gin.Context) {
+func (h *CollaborationHandler) CreateAnnotation(c *gin.Context) {
 	collabID := c.Param("collabId")
 	userID := c.GetString("user_id")
 
@@ -906,7 +905,7 @@ func (h *Handler) CreateAnnotation(c *gin.Context) {
 }
 
 // GetAnnotations retrieves active annotations
-func (h *Handler) GetAnnotations(c *gin.Context) {
+func (h *CollaborationHandler) GetAnnotations(c *gin.Context) {
 	collabID := c.Param("collabId")
 	userID := c.GetString("user_id")
 
@@ -915,7 +914,7 @@ func (h *Handler) GetAnnotations(c *gin.Context) {
 		return
 	}
 
-	rows, err := h.DB.Query(`
+	rows, err := h.DB.DB().Query(`
 		SELECT id, session_id, user_id, type, color, thickness, points, text,
 		       is_persistent, created_at, expires_at
 		FROM collaboration_annotations
@@ -949,7 +948,7 @@ func (h *Handler) GetAnnotations(c *gin.Context) {
 }
 
 // DeleteAnnotation removes an annotation
-func (h *Handler) DeleteAnnotation(c *gin.Context) {
+func (h *CollaborationHandler) DeleteAnnotation(c *gin.Context) {
 	collabID := c.Param("collabId")
 	annotationID := c.Param("annotationId")
 	userID := c.GetString("user_id")
@@ -973,7 +972,7 @@ func (h *Handler) DeleteAnnotation(c *gin.Context) {
 }
 
 // ClearAllAnnotations removes all annotations
-func (h *Handler) ClearAllAnnotations(c *gin.Context) {
+func (h *CollaborationHandler) ClearAllAnnotations(c *gin.Context) {
 	collabID := c.Param("collabId")
 	userID := c.GetString("user_id")
 
@@ -994,7 +993,7 @@ func (h *Handler) ClearAllAnnotations(c *gin.Context) {
 
 // Helper functions
 
-func (h *Handler) isCollaborationParticipant(collabID, userID string) bool {
+func (h *CollaborationHandler) isCollaborationParticipant(collabID, userID string) bool {
 	var exists bool
 	h.DB.DB().QueryRow(`
 		SELECT EXISTS(SELECT 1 FROM collaboration_participants
@@ -1003,7 +1002,7 @@ func (h *Handler) isCollaborationParticipant(collabID, userID string) bool {
 	return exists
 }
 
-func (h *Handler) canManageCollaboration(collabID, userID string) bool {
+func (h *CollaborationHandler) canManageCollaboration(collabID, userID string) bool {
 	var permissions sql.NullString
 	h.DB.DB().QueryRow(`
 		SELECT permissions FROM collaboration_participants
@@ -1019,7 +1018,7 @@ func (h *Handler) canManageCollaboration(collabID, userID string) bool {
 	return perms.CanManage
 }
 
-func (h *Handler) hasCollaborationPermission(collabID, userID, permission string) bool {
+func (h *CollaborationHandler) hasCollaborationPermission(collabID, userID, permission string) bool {
 	var permissions sql.NullString
 	h.DB.DB().QueryRow(`
 		SELECT permissions FROM collaboration_participants
@@ -1048,7 +1047,7 @@ func (h *Handler) hasCollaborationPermission(collabID, userID, permission string
 }
 
 // GetCollaborationStats returns collaboration statistics
-func (h *Handler) GetCollaborationStats(c *gin.Context) {
+func (h *CollaborationHandler) GetCollaborationStats(c *gin.Context) {
 	collabID := c.Param("collabId")
 	userID := c.GetString("user_id")