Improve file serving using built-ins

This commit is contained in:
raul 2024-04-30 12:11:26 +02:00
parent 2153ea7599
commit c9c8a03f8f
1 changed files with 16 additions and 16 deletions

32
main.go
View File

@ -8,22 +8,21 @@ import (
"os" "os"
) )
func getRoot(w http.ResponseWriter, r *http.Request) { // func getRoot(w http.ResponseWriter, r *http.Request) {
fmt.Printf("Received connection\n") // fmt.Printf("Received connection\n")
file, err := os.Open("./index.html") // file, err := os.Open("./index.html")
if err != nil { // if err != nil {
log.Printf("Error happened opening index.html: %v\n", err) // log.Printf("Error happened opening index.html: %v\n", err)
} // }
defer file.Close() // defer file.Close()
data, err := io.ReadAll(file) // data, err := io.ReadAll(file)
if err != nil { // if err != nil {
log.Printf("Error happened reading data: %v\n", err) // log.Printf("Error happened reading data: %v\n", err)
} // }
fmt.Fprint(w, string(data)) // fmt.Fprint(w, string(data))
} // }
func uploadFile(w http.ResponseWriter, r *http.Request) { func uploadFile(w http.ResponseWriter, r *http.Request) {
fmt.Printf("Received upload!\n")
fmt.Fprintln(w, "You have uploaded something") fmt.Fprintln(w, "You have uploaded something")
file, header, err := r.FormFile("file") file, header, err := r.FormFile("file")
@ -31,18 +30,19 @@ func uploadFile(w http.ResponseWriter, r *http.Request) {
if err != nil { if err != nil {
log.Printf("Error happened receiving file: %v\n", err) log.Printf("Error happened receiving file: %v\n", err)
} }
fmt.Fprintf(w, "%v", header.Header)
f, err := os.OpenFile("./temp/"+header.Filename, os.O_WRONLY|os.O_CREATE, 0660) f, err := os.OpenFile("./temp/"+header.Filename, os.O_WRONLY|os.O_CREATE, 0660)
if err != nil { if err != nil {
log.Printf("Error happened opening file: %v\n", err) log.Printf("Error happened opening file: %v\n", err)
} }
log.Printf("[%v] Received file \"%v\" from %v\n", r.RemoteAddr, header.Filename, r.UserAgent())
io.Copy(f, file) io.Copy(f, file)
defer file.Close() defer file.Close()
} }
func main() { func main() {
http.HandleFunc("/", getRoot) http.Handle("/", http.FileServer(http.Dir("./assets")))
http.HandleFunc("/api/upload", uploadFile) http.HandleFunc("/api/upload", uploadFile)
fmt.Println("Listening on port 8080...") fmt.Println("Listening on port 8080...")