Added index, improved logic
This commit is contained in:
3
.env
Normal file
3
.env
Normal file
@@ -0,0 +1,3 @@
|
||||
PORT=8080
|
||||
UPLOAD_DIR=./uploads
|
||||
BASE_URL=http://localhost:8080
|
||||
@@ -1,3 +1,4 @@
|
||||
PORT=8080
|
||||
UPLOAD_DIR=./uploads
|
||||
MAX_FILE_SIZE=10MB
|
||||
BASE_URL=http://localhost:8080
|
||||
4
.gitignore
vendored
4
.gitignore
vendored
@@ -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/**
|
||||
@@ -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
19
endpoints/index.go
Normal 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,
|
||||
})
|
||||
}
|
||||
@@ -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,
|
||||
})
|
||||
}
|
||||
6
main.go
6
main.go
@@ -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
72
views/index.html
Normal 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>
|
||||
Reference in New Issue
Block a user