96 lines
2.6 KiB
Go
96 lines
2.6 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
var DEFAULT_EXTENSION = "txt"
|
||
|
|
|
||
|
|
func GetFile(filename string, w http.ResponseWriter, r *http.Request) {
|
||
|
|
filenames, err := filepath.Glob("data/" + filename + ".*") // .txt, .md, anything
|
||
|
|
if err != nil {
|
||
|
|
http.Error(w, "error finding file", http.StatusInternalServerError)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(filenames) == 0 {
|
||
|
|
http.Error(w, "no matching files found", http.StatusNotFound)
|
||
|
|
return
|
||
|
|
} else if len(filenames) > 1 {
|
||
|
|
http.Error(w, "several matching files found ("+strconv.Itoa(len(filenames))+")", http.StatusInternalServerError) // TODO: Better handling, duh
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
fileContents, err := os.ReadFile(filenames[0])
|
||
|
|
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)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// PostFile TODO: Save to trash to prevent malicious/accidental ovverrides?
|
||
|
|
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
|
||
|
|
}
|
||
|
|
|
||
|
|
filenames, err := filepath.Glob("data/" + filename + ".*") // .txt, .md, anything
|
||
|
|
if err != nil {
|
||
|
|
http.Error(w, "error searching for file", http.StatusInternalServerError)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
var filenameFinal string
|
||
|
|
if len(filenames) == 0 {
|
||
|
|
// Create new file and write
|
||
|
|
filenameFinal = "data/" + filename + "." + DEFAULT_EXTENSION
|
||
|
|
} else if len(filenames) > 1 {
|
||
|
|
http.Error(w, "several matching files found ("+strconv.Itoa(len(filenames))+")", http.StatusInternalServerError) // TODO: Better handling, duh
|
||
|
|
return
|
||
|
|
} else {
|
||
|
|
filenameFinal = filenames[0]
|
||
|
|
fmt.Println(filenameFinal)
|
||
|
|
fmt.Println(filenames)
|
||
|
|
}
|
||
|
|
|
||
|
|
f, err := os.OpenFile(filenameFinal, os.O_CREATE|os.O_WRONLY, 0644)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Println("Error opening/making file")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err := f.Write(body); err != nil {
|
||
|
|
fmt.Println("Error writing to the file")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// ListFiles returns JSON of filenames in a directory without extensions or path
|
||
|
|
func ListFiles(directory string, w http.ResponseWriter, r *http.Request) {
|
||
|
|
filenames, err := filepath.Glob("data/" + directory + "/*") // .txt, .md, anything
|
||
|
|
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)
|
||
|
|
}
|