97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func handleRoot(w http.ResponseWriter, r *http.Request) {
|
||
|
|
if r.URL.Path != "/" {
|
||
|
|
http.NotFound(w, r)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
progressMutex.Lock()
|
||
|
|
jobs := make(map[string]*ProgressInfo)
|
||
|
|
for k, v := range progress {
|
||
|
|
jobs[k] = v
|
||
|
|
}
|
||
|
|
progressMutex.Unlock()
|
||
|
|
|
||
|
|
err := templates.ExecuteTemplate(w, "index", struct{ Jobs map[string]*ProgressInfo }{jobs})
|
||
|
|
if err != nil {
|
||
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func handleUpload(w http.ResponseWriter, r *http.Request) {
|
||
|
|
file, header, err := r.FormFile("file")
|
||
|
|
if err != nil {
|
||
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer file.Close()
|
||
|
|
|
||
|
|
filename := fmt.Sprintf("%d_%s", time.Now().UnixNano(), header.Filename)
|
||
|
|
filepath := filepath.Join(uploadDir, filename)
|
||
|
|
|
||
|
|
newFile, err := os.Create(filepath)
|
||
|
|
if err != nil {
|
||
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer newFile.Close()
|
||
|
|
|
||
|
|
_, err = io.Copy(newFile, file)
|
||
|
|
if err != nil {
|
||
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
err := processInputFile(filepath)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Error processing file: %v\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
os.Remove(filepath)
|
||
|
|
}()
|
||
|
|
|
||
|
|
http.Redirect(w, r, "/progress?filename="+filename, http.StatusSeeOther)
|
||
|
|
}
|
||
|
|
|
||
|
|
func handleProgress(w http.ResponseWriter, r *http.Request) {
|
||
|
|
filename := r.URL.Query().Get("filename")
|
||
|
|
fmt.Printf("Handling progress request for filename: %s\n", filename)
|
||
|
|
|
||
|
|
if r.Header.Get("Accept") == "application/json" {
|
||
|
|
progressInfo := getProgress(filename)
|
||
|
|
fmt.Printf("Progress info for %s: %+v\n", filename, progressInfo)
|
||
|
|
|
||
|
|
if progressInfo == nil {
|
||
|
|
w.WriteHeader(http.StatusNotFound)
|
||
|
|
json.NewEncoder(w).Encode(map[string]string{"error": "No progress information found"})
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
w.Header().Set("Content-Type", "application/json")
|
||
|
|
err := json.NewEncoder(w).Encode(progressInfo)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Error encoding progress info: %v\n", err)
|
||
|
|
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
err := templates.ExecuteTemplate(w, "progress", struct{ Filename string }{filename})
|
||
|
|
if err != nil {
|
||
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||
|
|
}
|
||
|
|
}
|