Added index, improved logic

This commit is contained in:
2025-08-07 22:08:48 +02:00
parent 84cdd52322
commit 6bcedf84f6
8 changed files with 159 additions and 43 deletions

3
.env Normal file
View File

@@ -0,0 +1,3 @@
PORT=8080
UPLOAD_DIR=./uploads
BASE_URL=http://localhost:8080

View File

@@ -1,3 +1,4 @@
PORT=8080
UPLOAD_DIR=./uploads
MAX_FILE_SIZE=10MB
BASE_URL=http://localhost:8080

4
.gitignore vendored
View File

@@ -202,3 +202,7 @@ $RECYCLE.BIN/
*.lnk
# End of https://www.toptal.com/developers/gitignore/api/jetbrains+all,go,macos,windows,linux,git
# Application
uploads/**

View File

@@ -9,14 +9,22 @@ import (
)
func HandleFileDownload(c *gin.Context) {
fileID := c.Param("fileID")
filename := c.Param("filename")
if fileID == "" {
c.JSON(http.StatusBadRequest, gin.H{"error": "File ID is required"})
return
}
if filename == "" {
c.JSON(http.StatusBadRequest, gin.H{"error": "Filename is required"})
return
}
uploadsDir := "./uploads"
filePath := filepath.Join(uploadsDir, filename)
fileDir := filepath.Join(uploadsDir, fileID)
filePath := filepath.Join(fileDir, filename)
if _, err := os.Stat(filePath); os.IsNotExist(err) {
c.JSON(http.StatusNotFound, gin.H{"error": "File not found"})
@@ -30,33 +38,3 @@ func HandleFileDownload(c *gin.Context) {
c.File(filePath)
}
func ListFiles(c *gin.Context) {
uploadsDir := "./uploads"
files, err := os.ReadDir(uploadsDir)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to read uploads directory"})
return
}
var fileList []gin.H
for _, file := range files {
if !file.IsDir() {
info, err := file.Info()
if err != nil {
continue
}
fileList = append(fileList, gin.H{
"filename": file.Name(),
"size": info.Size(),
"modified": info.ModTime(),
})
}
}
c.JSON(http.StatusOK, gin.H{
"files": fileList,
"count": len(fileList),
})
}

19
endpoints/index.go Normal file
View File

@@ -0,0 +1,19 @@
package endpoints
import (
"net/http"
"os"
"github.com/gin-gonic/gin"
)
func HandleIndex(c *gin.Context) {
baseURL := os.Getenv("BASE_URL")
if baseURL == "" {
baseURL = "http://localhost:8080"
}
c.HTML(http.StatusOK, "index.html", gin.H{
"BaseURL": baseURL,
})
}

View File

@@ -1,7 +1,10 @@
package endpoints
import (
"crypto/rand"
"encoding/hex"
"io"
"mime/multipart"
"net/http"
"os"
"path/filepath"
@@ -9,27 +12,52 @@ import (
"github.com/gin-gonic/gin"
)
func generateID() (string, error) {
bytes := make([]byte, 16)
if _, err := rand.Read(bytes); err != nil {
return "", err
}
return hex.EncodeToString(bytes), nil
}
func HandleFileUpload(c *gin.Context) {
file, header, err := c.Request.FormFile("file")
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "No file provided"})
return
}
defer file.Close()
defer func(file multipart.File) {
err := file.Close()
if err != nil {
uploadsDir := "./uploads"
if err := os.MkdirAll(uploadsDir, 0755); err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create uploads directory"})
}
}(file)
fileID, err := generateID()
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to generate file ID"})
return
}
dst := filepath.Join(uploadsDir, header.Filename)
uploadsDir := "./uploads"
fileDir := filepath.Join(uploadsDir, fileID)
if err := os.MkdirAll(fileDir, 0755); err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create file directory"})
return
}
dst := filepath.Join(fileDir, header.Filename)
out, err := os.Create(dst)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create file"})
return
}
defer out.Close()
defer func(out *os.File) {
err := out.Close()
if err != nil {
}
}(out)
_, err = io.Copy(out, file)
if err != nil {
@@ -37,9 +65,18 @@ func HandleFileUpload(c *gin.Context) {
return
}
baseURL := os.Getenv("BASE_URL")
if baseURL == "" {
baseURL = "http://localhost:8080"
}
fileURL := baseURL + "/files/" + fileID + "/" + header.Filename
c.JSON(http.StatusOK, gin.H{
"message": "File uploaded successfully",
"id": fileID,
"filename": header.Filename,
"size": header.Size,
"url": fileURL,
})
}

View File

@@ -10,10 +10,12 @@ import (
func main() {
r := gin.Default()
r.LoadHTMLGlob("views/*")
r.GET("/", endpoints.HandleIndex)
r.POST("/upload", endpoints.HandleFileUpload)
r.GET("/download/:filename", endpoints.HandleFileDownload)
r.GET("/files", endpoints.ListFiles)
r.GET("/uploads/:fileID/:filename", endpoints.HandleFileDownload)
r.GET("/health", func(c *gin.Context) {
c.JSON(200, gin.H{"status": "ok"})
})

72
views/index.html Normal file
View File

@@ -0,0 +1,72 @@
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>FITRA - File transfer API</title>
<style>
body { font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; line-height: 1.6; }
h1, h2 { color: #333; }
code { background: #f4f4f4; padding: 2px 6px; border-radius: 3px; font-family: 'Monaco', 'Consolas', monospace; }
pre { background: #f4f4f4; padding: 15px; border-radius: 5px; overflow-x: auto; }
.endpoint { margin: 20px 0; padding: 15px; border-left: 4px solid #007acc; background: #f9f9f9; }
.method { display: inline-block; padding: 3px 8px; border-radius: 3px; color: white; font-weight: bold; margin-right: 10px; }
.get { background: #61affe; }
.post { background: #49cc90; }
.step { margin: 10px 0; padding: 10px; background: #fff; border: 1px solid #ddd; border-radius: 3px; }
</style>
</head>
<body>
<h1>🚀 FITRA - File transfer API</h1>
<p><strong>Version:</strong> 1.0.0</p>
<p>Simple file upload and download service for developers using HTTP requests in CLI.</p>
<h2>📋 API endpoints</h2>
<div class="endpoint">
<h3><span class="method post">POST</span>/upload</h3>
<p><strong>Description:</strong> Upload a file</p>
<p><strong>cURL Example:</strong></p>
<pre><code>curl -X POST -F "file=@/path/to/your/file.txt" {{.BaseURL}}/upload</code></pre>
</div>
<div class="endpoint">
<h3><span class="method get">GET</span>/uploads/{fileID}/{filename}</h3>
<p><strong>Description:</strong> Download a file using the ID and filename from upload response</p>
<p><strong>cURL Example:</strong></p>
<pre><code>curl -O {{.BaseURL}}/uploads/{fileID}/{filename}</code></pre>
</div>
<div class="endpoint">
<h3><span class="method get">GET</span>/health</h3>
<p><strong>Description:</strong> Check service health</p>
<p><strong>cURL Example:</strong></p>
<pre><code>curl {{.BaseURL}}/health</code></pre>
</div>
<h2>🔄 Usage</h2>
<div class="step">
<strong>Step 1:</strong> Upload a file using POST /upload with form-data 'file' parameter
</div>
<div class="step">
<strong>Step 2:</strong> Use the returned 'id' and 'filename' to construct download URL
</div>
<div class="step">
<strong>Step 3:</strong> Download the file using GET /uploads/{id}/{filename}
</div>
<h2>💡 Examples</h2>
<pre><code># 1. Upload a file
curl -X POST -F "file=@myfile.txt" {{.BaseURL}}/upload
# Response will include:
# {
# "id": "abc123...",
# "filename": "myfile.txt",
# "url": "{{.BaseURL}}/uploads/abc123.../myfile.txt"
# }
# 2. Download the file
curl -O {{.BaseURL}}/uploads/abc123.../myfile.txt</code></pre>
</body>
</html>