hibiscus/files.go

77 lines
2 KiB
Go
Raw Normal View History

2024-03-15 18:34:24 +03:00
package main
import (
"encoding/json"
"errors"
2024-03-15 18:34:24 +03:00
"fmt"
"io"
"net/http"
"os"
"path/filepath"
"strings"
)
2024-03-17 17:12:41 +03:00
// GetFile returns raw contents of a txt file in data directory
2024-03-15 18:34:24 +03:00
func GetFile(filename string, w http.ResponseWriter, r *http.Request) {
2024-03-17 16:44:05 +03:00
path := "data/" + filename + ".txt" // Can we and should we sanitize this?
2024-03-15 19:22:22 +03:00
if _, err := os.Stat(path); errors.Is(err, os.ErrNotExist) {
NotFound(w, r)
2024-03-15 18:34:24 +03:00
return
}
2024-03-15 19:22:22 +03:00
fileContents, err := os.ReadFile(path)
2024-03-15 18:34:24 +03:00
if err != nil {
http.Error(w, "error reading file", http.StatusInternalServerError)
return
}
_, err = w.Write(fileContents)
if err != nil {
http.Error(w, "error sending file", http.StatusInternalServerError)
}
}
2024-03-17 17:12:41 +03:00
// PostFile Writes request's contents to a txt file in data directory
// TODO: Save to trash to prevent malicious/accidental overrides?
2024-03-15 18:34:24 +03:00
func PostFile(filename string, w http.ResponseWriter, r *http.Request) {
body, err := io.ReadAll(r.Body)
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
w.Write([]byte("error reading body"))
return
}
f, err := os.OpenFile("data/"+filename+".txt", os.O_CREATE|os.O_WRONLY, 0644)
2024-03-15 18:34:24 +03:00
if err != nil {
fmt.Println("error opening/making file")
w.Write([]byte("error opening or creating file"))
w.WriteHeader(http.StatusInternalServerError)
2024-03-15 18:34:24 +03:00
return
}
if _, err := f.Write(body); err != nil {
2024-03-17 18:11:20 +03:00
fmt.Println("error writing to the file")
w.Write([]byte("error writing to file"))
w.WriteHeader(http.StatusInternalServerError)
return
2024-03-15 18:34:24 +03:00
}
w.Write([]byte("wrote to file"))
w.WriteHeader(http.StatusOK)
2024-03-15 18:34:24 +03:00
}
2024-03-15 19:22:22 +03:00
// ListFiles returns JSON list of filenames in a directory without extensions or path
2024-03-15 18:34:24 +03:00
func ListFiles(directory string, w http.ResponseWriter, r *http.Request) {
filenames, err := filepath.Glob("data/" + directory + "/*.txt")
2024-03-15 18:34:24 +03:00
if err != nil {
http.Error(w, "error searching for files", http.StatusInternalServerError)
return
}
for i, file := range filenames {
file, _ := strings.CutSuffix(filepath.Base(file), filepath.Ext(file))
filenames[i] = file
}
filenamesJson, err := json.Marshal(filenames)
w.Write(filenamesJson)
}