package main import ( "fmt" "github.com/redis/go-redis/v9" "html/template" "io" "log" "net/http" "strings" ) var currentConfig Config var db *redis.Client type pasteView struct { Content string Key string } func handleRequest(w http.ResponseWriter, r *http.Request) { path := r.URL.Path clearPath := strings.ReplaceAll(r.URL.Path, "/raw", "") staticResource := "/static/" switch r.Method { case "GET": if path == "/" { http.ServeFile(w, r, "./static/index.html") } else if strings.HasPrefix(path, staticResource) { fs := http.FileServer(http.Dir("./static")) http.Handle(staticResource, http.StripPrefix(staticResource, fs)) } else { if UrlExist(clearPath) { if strings.HasSuffix(path, "/raw") { pasteContent := getContent(clearPath) w.Header().Set("Content-Type", "text/plain; charset=utf-8") _, err := io.WriteString(w, pasteContent) if err != nil { log.Println(err) } } else { pasteContent := getContent(path) s := pasteView{Content: pasteContent, Key: strings.TrimPrefix(path, "/")} t, err := template.ParseFiles("templates/paste.html") if err != nil { log.Println(err) } err = t.Execute(w, s) 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) } case "DELETE": if UrlExist(path) { secret := r.URL.Query().Get("secret") if secret == db.HGet(ctx, path, "secret").Val() { err := db.Del(ctx, path) if err != nil { log.Println(err) } w.WriteHeader(http.StatusNoContent) } else { w.WriteHeader(http.StatusForbidden) } } else { w.WriteHeader(http.StatusNotFound) } } } func main() { db = ConnectDB() currentConfig = GetConfig() listen := currentConfig.host + ":" + currentConfig.port http.HandleFunc("/", handleRequest) if currentConfig.host == "" { fmt.Println("Listening on port " + listen) } else { fmt.Println("Listening on " + listen) } err := http.ListenAndServe(listen, nil) if err != nil { log.Fatal(err) } }