add endpoint to check if password exists

main.go

@@ -26,6 +26,7 @@ func run() error {
 	mux.Handle("GET /", routes.ServeFiles(embedFS, "static"))
 	mux.Handle("POST /api/password", routes.CreatePassword(storage, 12*1024, base64.StdEncoding))
 	mux.Handle("GET /api/password/{id}", routes.GetPassword(storage, base64.StdEncoding))
+	mux.Handle("HEAD /api/password/{id}", routes.HasPassword(storage))
 
 	address := os.Getenv("PASSED_ADDRESS")
 	if address == "" {

routes/has_password.go

@@ -0,0 +1,24 @@
+package routes
+
+import (
+	"net/http"
+
+	"git.1e99.eu/1e99/passed/storage"
+)
+
+func HasPassword(store storage.Store) http.HandlerFunc {
+	return func(res http.ResponseWriter, req *http.Request) {
+		id := req.PathValue("id")
+		found, err := store.HasPassword(id)
+		if err != nil {
+			http.Error(res, "", http.StatusInternalServerError)
+			return
+		}
+
+		if found {
+			res.WriteHeader(http.StatusNoContent)
+		} else {
+			res.WriteHeader(http.StatusNotFound)
+		}
+	}
+}

storage/ram.go

@@ -60,6 +60,14 @@ func (store *ram) GetPassword(id string) ([]byte, error) {
 	return password.password, nil
 }
 
+func (store *ram) HasPassword(id string) (bool, error) {
+	store.lock.Lock()
+	defer store.lock.Unlock()
+
+	_, found := store.passwords[id]
+	return found, nil
+}
+
 func (store *ram) Close() error {
 	store.close <- true
 	return nil
@@ -75,7 +83,6 @@ func (store *ram) clearExpired() error {
 			return nil
 		case <-ticker.C:
 			store.lock.Lock()
-			defer store.lock.Unlock()
 			time := time.Now()
 
 			for id, password := range store.passwords {
@@ -83,6 +90,8 @@ func (store *ram) clearExpired() error {
 					delete(store.passwords, id)
 				}
 			}
+
+			store.lock.Unlock()
 		}
 	}
 }

storage/storage.go

@@ -21,6 +21,7 @@ type entry struct {
 type Store interface {
 	CreatePassword(password []byte, expiresAt time.Time) (string, error)
 	GetPassword(id string) ([]byte, error)
+	HasPassword(id string) (bool, error)
 	Close() error
 }