feat: add /last for checking last uploaded file

abyss.go

@@ -51,11 +51,12 @@ func main() {
 	}
 
 	mux := http.NewServeMux()
-	mux.HandleFunc("/", app.fileHandler)
+	mux.HandleFunc("/", app.indexHandler)
 	mux.HandleFunc(
 		"/tree/",
 		app.basicAuth(app.treeHandler),
 	)
+	mux.HandleFunc("/last", app.lastHandler)
 
 	srv := &http.Server{
 		Addr:         app.port,

handlers.go

@@ -20,9 +20,14 @@ type Application struct {
 	key              string
 	filesDir         string
 	port             string
+	lastUploadedFile string
 }
 
-func (app *Application) fileHandler(w http.ResponseWriter, r *http.Request) {
+func (app *Application) treeHandler(w http.ResponseWriter, r *http.Request) {
+	http.StripPrefix("/tree/", http.FileServer(http.Dir(app.filesDir))).ServeHTTP(w, r)
+}
+
+func (app *Application) indexHandler(w http.ResponseWriter, r *http.Request) {
 	if r.Method == http.MethodPost {
 		app.uploadHandler(w, r)
 		return
@@ -39,12 +44,16 @@ func (app *Application) fileHandler(w http.ResponseWriter, r *http.Request) {
 	if fileInfo, err := os.Stat(path); err == nil && !fileInfo.IsDir() {
 		http.ServeFile(w, r, path)
 	} else {
-		http.NotFound(w, r)
+		http.StripPrefix("/", http.FileServer(http.Dir("static"))).ServeHTTP(w, r)
 	}
 }
 
-func (app *Application) treeHandler(w http.ResponseWriter, r *http.Request) {
-	http.StripPrefix("/tree/", http.FileServer(http.Dir(app.filesDir))).ServeHTTP(w, r)
+func (app *Application) lastHandler(w http.ResponseWriter, r *http.Request) {
+	if app.lastUploadedFile == "" {
+		http.Error(w, "No new files uploaded yet", http.StatusNotFound)
+		return
+	}
+	http.ServeFile(w, r, app.lastUploadedFile)
 }
 
 func (app *Application) uploadHandler(w http.ResponseWriter, r *http.Request) {
@@ -87,6 +96,8 @@ func (app *Application) uploadHandler(w http.ResponseWriter, r *http.Request) {
 		http.Error(w, "Error copying the file", http.StatusInternalServerError)
 	}
 
+	app.lastUploadedFile = filepath
+
 	if app.url == "" {
 		fmt.Fprintf(w, "http://localhost%s/%s\n", app.port, filename)
 	} else {