feat: backend changes

Author: 35C4n0r

lib/httpapi/models.go

@@ -65,11 +65,11 @@ type MessageResponse struct {
 
 type UploadResponse struct {
 	Body struct {
-		Ok bool `json:"ok" doc:"Indicates whether the files were uploaded successfully."`
+		Ok       bool   `json:"ok" doc:"Indicates whether the files were uploaded successfully."`
+		FilePath string `json:"filePath" doc:"Path of the file"`
 	}
 }
 
 type UploadRequest struct {
-	UploadPath string        `form:"uploadPath" required:"true" doc:"location where all the files in the zip will be extracted to."`
-	Files      huma.FormFile `form:"file" contentType:"application/zip" required:"true" doc:"zip of all the files that needs to be uploaded"`
+	File huma.FormFile `form:"file" required:"true" doc:"file that needs to be uploaded"`
 }

lib/httpapi/server.go

@@ -1,9 +1,9 @@
 package httpapi
 
 import (
-	"archive/zip"
-	"bytes"
 	"context"
+	"crypto/sha256"
+	"encoding/hex"
 	"encoding/json"
 	"fmt"
 	"io"
@@ -45,6 +45,7 @@ type Server struct {
 	agentType    mf.AgentType
 	emitter      *EventEmitter
 	chatBasePath string
+	tempDir      string
 }
 
 func (s *Server) NormalizeSchema(schema any) any {
@@ -237,6 +238,14 @@ func NewServer(ctx context.Context, config ServerConfig) (*Server, error) {
 		FormatMessage:         formatMessage,
 	})
 	emitter := NewEventEmitter(1024)
+
+	// Create temporary directory for uploads
+	tempDir, err := os.MkdirTemp("", "agentapi-uploads-")
+	if err != nil {
+		return nil, xerrors.Errorf("failed to create temporary directory: %w", err)
+	}
+	logger.Info("Created temporary directory for uploads", "tempDir", tempDir)
+
 	s := &Server{
 		router:       router,
 		api:          api,
@@ -247,6 +256,7 @@ func NewServer(ctx context.Context, config ServerConfig) (*Server, error) {
 		agentType:    config.AgentType,
 		emitter:      emitter,
 		chatBasePath: strings.TrimSuffix(config.ChatBasePath, "/"),
+		tempDir:      tempDir,
 	}
 
 	// Register API routes
@@ -432,67 +442,36 @@ func (s *Server) uploadFiles(ctx context.Context, input *struct {
 }) (*UploadResponse, error) {
 	formData := input.RawBody.Data()
 
-	file := formData.Files.File
+	file := formData.File.File
 
 	buf, err := io.ReadAll(file)
 	if err != nil {
-		return nil, xerrors.Errorf("failed to upload files: %w", err)
+		return nil, xerrors.Errorf("failed to upload file: %w", err)
 	}
 
-	// Create a zip.Reader from the buffer
-	zipReader, err := zip.NewReader(bytes.NewReader(buf), int64(len(buf)))
+	// Calculate checksum of the uploaded file to create unique subdirectory
+	hash := sha256.Sum256(buf)
+	checksum := hex.EncodeToString(hash[:8]) // Use first 8 bytes (16 hex chars)
+
+	// Create checksum-based subdirectory in tempDir
+	uploadDir := filepath.Join(s.tempDir, checksum)
+	err = os.MkdirAll(uploadDir, 0755)
 	if err != nil {
-		return nil, xerrors.Errorf("failed to upload files: %w", err)
+		return nil, xerrors.Errorf("failed to create upload directory: %w", err)
 	}
 
-	for _, f := range zipReader.File {
-		outPath := filepath.Join(formData.UploadPath, f.Name)
-
-		if f.FileInfo().IsDir() {
-			err := os.MkdirAll(outPath, f.Mode())
-			if err != nil {
-				return nil, xerrors.Errorf("failed to upload files: %w", err)
-			}
-			continue
-		}
-
-		err := os.MkdirAll(filepath.Dir(outPath), f.Mode())
-		if err != nil {
-			return nil, xerrors.Errorf("failed to upload files: %w", err)
-		}
-
-		outFile, err := os.OpenFile(outPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
-		if err != nil {
-			return nil, xerrors.Errorf("failed to upload files: %w", err)
-		}
-
-		rc, err := f.Open()
-		if err != nil {
-			err := outFile.Close()
-			if err != nil {
-				return nil, xerrors.Errorf("failed to upload files: %w", err)
-			}
-			return nil, xerrors.Errorf("failed to upload files: %w", err)
-		}
-
-		_, err = io.Copy(outFile, rc)
-		if err != nil {
-			return nil, xerrors.Errorf("failed to upload files: %w", err)
-		}
-
-		err = rc.Close()
-		if err != nil {
-			return nil, xerrors.Errorf("failed to upload files: %w", err)
-		}
+	// Save individual file with original filename
+	filename := formData.File.Filename
 
-		err = outFile.Close()
-		if err != nil {
-			return nil, xerrors.Errorf("failed to upload files: %w", err)
-		}
+	outPath := filepath.Join(uploadDir, filename)
+	err = os.WriteFile(outPath, buf, 0644)
+	if err != nil {
+		return nil, xerrors.Errorf("failed to write file: %w", err)
 	}
 
 	resp := &UploadResponse{}
 	resp.Body.Ok = true
+	resp.Body.FilePath = outPath
 	return resp, nil
 }
 
@@ -579,12 +558,24 @@ func (s *Server) Start() error {
 
 // Stop gracefully stops the HTTP server
 func (s *Server) Stop(ctx context.Context) error {
+	// Clean up temporary directory
+	s.cleanupTempDir()
+
 	if s.srv != nil {
 		return s.srv.Shutdown(ctx)
 	}
 	return nil
 }
 
+// cleanupTempDir removes the temporary directory and all its contents
+func (s *Server) cleanupTempDir() {
+	if err := os.RemoveAll(s.tempDir); err != nil {
+		s.logger.Error("Failed to clean up temporary directory", "tempDir", s.tempDir, "error", err)
+	} else {
+		s.logger.Info("Cleaned up temporary directory", "tempDir", s.tempDir)
+	}
+}
+
 // registerStaticFileRoutes sets up routes for serving static files
 func (s *Server) registerStaticFileRoutes() {
 	chatHandler := FileServerWithIndexFallback(s.chatBasePath)

openapi.json

@@ -314,12 +314,17 @@
             "readOnly": true,
             "type": "string"
           },
+          "filePath": {
+            "description": "Path of the file",
+            "type": "string"
+          },
           "ok": {
             "description": "Indicates whether the files were uploaded successfully.",
             "type": "boolean"
           }
         },
         "required": [
+          "filePath",
           "ok"
         ],
         "type": "object"
@@ -527,28 +532,20 @@
             "multipart/form-data": {
               "encoding": {
                 "file": {
-                  "contentType": "application/zip"
-                },
-                "uploadPath": {
-                  "contentType": "text/plain"
+                  "contentType": "application/octet-stream"
                 }
               },
               "schema": {
                 "properties": {
                   "file": {
                     "contentEncoding": "binary",
                     "contentMediaType": "application/octet-stream",
-                    "description": "zip of all the files that needs to be uploaded",
+                    "description": "file that needs to be uploaded",
                     "format": "binary",
                     "type": "string"
-                  },
-                  "uploadPath": {
-                    "description": "location where all the files in the zip will be extracted to.",
-                    "type": "string"
                   }
                 },
                 "required": [
-                  "uploadPath",
                   "file"
                 ],
                 "type": "object"