plakken/main.go

73 lines
1.6 KiB
Go
Raw Normal View History

package main
import (
"fmt"
"io"
"log"
"net/http"
"strings"
)
var currentConfig config
func handleRequest(w http.ResponseWriter, r *http.Request) {
2023-10-06 00:16:55 +02:00
path := r.URL.Path
clearPath := strings.ReplaceAll(r.URL.Path, "/raw", "")
db := connectDB()
switch r.Method {
case "GET":
if path == "/" {
http.ServeFile(w, r, "./static/index.html")
2023-10-06 00:16:55 +02:00
} else if strings.HasPrefix(path, "/static/") {
fs := http.FileServer(http.Dir("./static"))
http.Handle("/static/", http.StripPrefix("/static/", fs))
} else {
2023-10-06 00:16:55 +02:00
if urlExist(clearPath) {
if strings.HasSuffix(path, "/raw") {
content := db.HGet(ctx, clearPath, "content").Val()
_, err := io.WriteString(w, content)
if err != nil {
log.Println(err)
}
} else {
content := db.HGet(ctx, path, "content").Val()
_, err := io.WriteString(w, content)
if err != nil {
log.Println(err)
}
}
} else {
w.WriteHeader(http.StatusNotFound)
}
}
case "POST":
if path == "/" {
secret := generateSecret()
url := "/" + generateUrl()
content := r.FormValue("content")
insertPaste(url, content, secret, -1)
http.Redirect(w, r, url, http.StatusSeeOther)
} else {
w.WriteHeader(http.StatusMethodNotAllowed)
}
}
}
func main() {
currentConfig = getConfig()
2023-10-02 15:03:04 +02:00
listen := currentConfig.host + ":" + currentConfig.port
http.HandleFunc("/", handleRequest)
if currentConfig.host == "" {
fmt.Println("Listening on port " + listen)
} else {
fmt.Println("Listening on " + listen)
}
2023-10-02 15:03:04 +02:00
err := http.ListenAndServe(listen, nil)
if err != nil {
log.Fatal(err)
}
}