multi dataset

internal/server/server.go

@@ -271,6 +271,36 @@ func (s *Server) HandleUploadStream(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	// Check if dataset is allowed for this client
+	dataset, err := s.db.GetDatasetByName(clientID, datasetName)
+	if err != nil || dataset == nil {
+		// Auto-create dataset if not exists
+		log.Printf("Dataset %s not found for client %s, creating...", datasetName, clientID)
+		newDataset := &DatasetConfig{
+			ClientID:    clientID,
+			DatasetName: datasetName,
+			StorageType: "s3",
+			Enabled:     true,
+		}
+		if err := s.db.SaveDataset(newDataset); err != nil {
+			log.Printf("Error creating dataset: %v", err)
+			respondJSON(w, http.StatusForbidden, UploadResponse{
+				Success: false,
+				Message: "Dataset not configured for this client",
+			})
+			return
+		}
+		dataset = newDataset
+	}
+
+	if !dataset.Enabled {
+		respondJSON(w, http.StatusForbidden, UploadResponse{
+			Success: false,
+			Message: "Dataset is disabled",
+		})
+		return
+	}
+
 	ctx := context.Background()
 
 	// Upload to S3
@@ -438,13 +468,6 @@ func (s *Server) HandleDownload(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	// Find snapshot metadata
-	client, err := s.db.GetClient(clientID)
-	if err != nil || client == nil {
-		http.Error(w, "Client not found", http.StatusNotFound)
-		return
-	}
-
 	targetSnapshot, err := s.db.GetSnapshotByID(clientID, snapshotID)
 	if err != nil || targetSnapshot == nil {
 		http.Error(w, "Snapshot not found", http.StatusNotFound)
@@ -560,6 +583,8 @@ func (s *Server) RegisterRoutes(mux *http.ServeMux) {
 	mux.HandleFunc("/admin/snapshots", s.handleAdminGetSnapshots)
 	mux.HandleFunc("/admin/snapshot/delete", s.handleAdminDeleteSnapshot)
 	mux.HandleFunc("/admin/stats", s.handleAdminGetStats)
+	mux.HandleFunc("/admin/datasets", s.handleAdminGetDatasets)
+	mux.HandleFunc("/admin/datasets/{id}", s.handleAdminUpdateDeleteDataset)
 	mux.HandleFunc("/admin/admins", s.handleAdminGetAdmins)
 	mux.HandleFunc("/admin/admin/create", s.handleAdminCreateAdmin)
 	mux.HandleFunc("/admin/admin/delete", s.handleAdminDeleteAdmin)