diff --git a/main.go b/main.go index 34e05d9..99c3e07 100644 --- a/main.go +++ b/main.go @@ -64,24 +64,19 @@ func handleRequest(w http.ResponseWriter, r *http.Request) { w.WriteHeader(http.StatusMethodNotAllowed) } case "DELETE": - if strings.HasPrefix(path, "/delete") { - urlItem := strings.Split(path, "/") - if urlExist("/" + urlItem[2]) { - secret := r.URL.Query().Get("secret") - if secret == db.HGet(ctx, "/"+urlItem[2], "secret").Val() { - err := db.Del(ctx, "/"+urlItem[2]) - if err != nil { - log.Println(err) - } - w.WriteHeader(http.StatusNoContent) - } else { - w.WriteHeader(http.StatusForbidden) + 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.StatusNotFound) + w.WriteHeader(http.StatusForbidden) } } else { - w.WriteHeader(http.StatusMethodNotAllowed) + w.WriteHeader(http.StatusNotFound) } } }