admin panel added
This commit is contained in:
commit
534a84e51a
|
|
@ -0,0 +1,5 @@
|
|||
module upload
|
||||
|
||||
go 1.22.4
|
||||
|
||||
require github.com/google/uuid v1.6.0 // indirect
|
||||
|
|
@ -0,0 +1,2 @@
|
|||
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||
|
|
@ -0,0 +1,49 @@
|
|||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>Admin Panel</title>
|
||||
</head>
|
||||
<body>
|
||||
<h1>Admin Panel</h1>
|
||||
<form id="adminForm">
|
||||
<label for="maxSize">Max File Size:</label>
|
||||
<input type="number" id="maxSize" name="maxSize" required>
|
||||
<select id="sizeUnit" name="sizeUnit">
|
||||
<option value="MB">MB</option>
|
||||
<option value="GB">GB</option>
|
||||
</select><br><br>
|
||||
|
||||
<label for="lifetime">Link Lifetime:</label>
|
||||
<input type="number" id="lifetime" name="lifetime" required><br><br>
|
||||
|
||||
<button type="button" onclick="generateLink()">Generate</button>
|
||||
</form>
|
||||
<p id="uploadLink"></p>
|
||||
|
||||
<script>
|
||||
async function generateLink() {
|
||||
const maxSize = document.getElementById('maxSize').value;
|
||||
const sizeUnit = document.getElementById('sizeUnit').value;
|
||||
const lifetime = document.getElementById('lifetime').value;
|
||||
|
||||
const response = await fetch('/generate', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json'
|
||||
},
|
||||
body: JSON.stringify({ maxSize, sizeUnit, lifetime })
|
||||
});
|
||||
|
||||
if (response.ok) {
|
||||
const data = await response.json();
|
||||
document.getElementById('uploadLink').innerText = `Upload Link (copied to clipboard): ${data.uploadLink}`;
|
||||
navigator.clipboard.writeText(data.uploadLink);
|
||||
} else {
|
||||
document.getElementById('uploadLink').innerText = 'Error generating link';
|
||||
}
|
||||
}
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
|
|
@ -0,0 +1,45 @@
|
|||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>Chunked File Upload</title>
|
||||
</head>
|
||||
<body>
|
||||
<input type="file" id="fileInput">
|
||||
<button onclick="uploadFile()">Upload</button>
|
||||
|
||||
<script>
|
||||
async function uploadFile() {
|
||||
const fileInput = document.getElementById('fileInput');
|
||||
const file = fileInput.files[0];
|
||||
const chunkSize = 10 * 1024 * 1024; // 10 MB
|
||||
const totalChunks = Math.ceil(file.size / chunkSize);
|
||||
|
||||
for (let chunkIndex = 0; chunkIndex < totalChunks; chunkIndex++) {
|
||||
const start = chunkIndex * chunkSize;
|
||||
const end = Math.min(start + chunkSize, file.size);
|
||||
const chunk = file.slice(start, end);
|
||||
|
||||
const formData = new FormData();
|
||||
formData.append('chunk', chunk);
|
||||
formData.append('filename', file.name);
|
||||
formData.append('chunkIndex', chunkIndex);
|
||||
formData.append('totalChunks', totalChunks);
|
||||
|
||||
const response = await fetch('/upload', {
|
||||
method: 'POST',
|
||||
body: formData,
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
console.error('Failed to upload chunk', chunkIndex);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
console.log('File uploaded successfully');
|
||||
}
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
|
|
@ -0,0 +1,166 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"os"
|
||||
"strconv"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
type Admin struct {
|
||||
name string
|
||||
}
|
||||
|
||||
type UploadState int
|
||||
|
||||
const (
|
||||
inactive UploadState = iota
|
||||
active
|
||||
completed
|
||||
)
|
||||
|
||||
type UploadContext struct {
|
||||
maxSize int64
|
||||
created time.Time
|
||||
state UploadState
|
||||
finalName string
|
||||
}
|
||||
|
||||
type UploadDatabase struct {
|
||||
data map[uuid.UUID]UploadContext
|
||||
mtx sync.Mutex
|
||||
}
|
||||
|
||||
var db UploadDatabase
|
||||
|
||||
func (d *UploadDatabase) commit(size int64, finalName string, lifefile time.Duration) uuid.UUID {
|
||||
id := uuid.New()
|
||||
d.mtx.Lock()
|
||||
defer d.mtx.Unlock()
|
||||
|
||||
_, found := d.data[id]
|
||||
|
||||
if found {
|
||||
return uuid.Max
|
||||
}
|
||||
|
||||
d.data[id] = UploadContext{
|
||||
created: time.Now(),
|
||||
finalName: finalName,
|
||||
maxSize: size,
|
||||
state: inactive,
|
||||
}
|
||||
|
||||
return id
|
||||
}
|
||||
|
||||
func (d *UploadDatabase) activate(id uuid.UUID) *UploadContext {
|
||||
d.mtx.Lock()
|
||||
defer d.mtx.Unlock()
|
||||
|
||||
// i, found := d.data[id]
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *UploadDatabase) deactivate(id uuid.UUID) {
|
||||
|
||||
}
|
||||
|
||||
func initialize() {
|
||||
db.data = make(map[uuid.UUID]UploadContext)
|
||||
}
|
||||
|
||||
func adminHandler(w http.ResponseWriter, r *http.Request) {
|
||||
http.ServeFile(w, r, "./html/admin.html")
|
||||
}
|
||||
|
||||
func generateHandler(w http.ResponseWriter, r *http.Request) {
|
||||
var maxSizeBytes int64
|
||||
var linkLifetime int64
|
||||
var id uuid.UUID
|
||||
var uploadLink string
|
||||
|
||||
if r.Method != "POST" {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
}
|
||||
|
||||
var linkSettings struct {
|
||||
Lifetime string `json:"lifetime"`
|
||||
MaxSize string `json:"maxSize"`
|
||||
SizeUnit string `json:"sizeUnit"`
|
||||
}
|
||||
err := json.NewDecoder(r.Body).Decode(&linkSettings)
|
||||
if err != nil {
|
||||
http.Error(w, fmt.Sprintf("Invalid input: %v", err), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
linkLifetime, err = strconv.ParseInt(linkSettings.Lifetime, 10, 64)
|
||||
if err != nil {
|
||||
http.Error(w, fmt.Sprintf("Invalid input: %v", err), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
maxSizeBytes, err = strconv.ParseInt(linkSettings.MaxSize, 10, 64)
|
||||
if err != nil {
|
||||
http.Error(w, fmt.Sprintf("Invalid input: %v", err), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
if linkSettings.SizeUnit == "GB" {
|
||||
maxSizeBytes *= 1024 * 1024 * 1024
|
||||
} else if linkSettings.SizeUnit == "MB" {
|
||||
maxSizeBytes *= 1024 * 1024
|
||||
} else {
|
||||
http.Error(w, fmt.Sprintf("Invalid input: %v", err), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
id = db.commit(int64(maxSizeBytes), "untitled"+time.Now().String(), time.Duration(linkLifetime))
|
||||
if id == uuid.Max {
|
||||
http.Error(w, "Commit failed", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
uploadLink = fmt.Sprintf("https://localhost:8080/upload?key=%s", id)
|
||||
|
||||
response := map[string]string{"uploadLink": uploadLink}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
if err := json.NewEncoder(w).Encode(response); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
func uploadHandler(w http.ResponseWriter, r *http.Request) {
|
||||
|
||||
}
|
||||
|
||||
func fileHandler(w http.ResponseWriter, r *http.Request) {
|
||||
|
||||
}
|
||||
|
||||
func main() {
|
||||
initialize()
|
||||
|
||||
os.MkdirAll("uploads", os.ModePerm)
|
||||
|
||||
http.HandleFunc("/admin", adminHandler)
|
||||
http.HandleFunc("/generate", generateHandler)
|
||||
http.HandleFunc("/upload", uploadHandler)
|
||||
http.HandleFunc("/file", fileHandler)
|
||||
|
||||
// http.Handle("/", http.FileServer(http.Dir(".")))
|
||||
|
||||
// Start the server
|
||||
fmt.Println("Starting server on :8080")
|
||||
err := http.ListenAndServe(":8080", nil)
|
||||
if err != nil {
|
||||
fmt.Printf("Error starting server: %v", err)
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue