Return not found errors when applicable

Solves: #11
AlexGustafsson · Dec 16, 2024 · d468317 · d468317
1 parent c014e31
commit d468317
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/internal/api/server.go b/internal/api/server.go
@@ -98,6 +98,11 @@ func NewServer(api *store.Store, processQueue chan<- oci.Reference) *Server {
 		reference := query.Get("reference")
 
 		response, err := api.GetImage(r.Context(), reference)
+		if response == nil && err == nil {
+			s.handleGenericResponse(w, r, ErrNotFound)
+			return
+		}
+
 		s.handleJSONResponse(w, r, response, err)
 	})
 
@@ -107,6 +112,11 @@ func NewServer(api *store.Store, processQueue chan<- oci.Reference) *Server {
 		reference := query.Get("reference")
 
 		response, err := api.GetImageDescription(r.Context(), reference)
+		if response == nil && err == nil {
+			s.handleGenericResponse(w, r, ErrNotFound)
+			return
+		}
+
 		s.handleJSONResponse(w, r, response, err)
 	})
 
@@ -116,6 +126,11 @@ func NewServer(api *store.Store, processQueue chan<- oci.Reference) *Server {
 		reference := query.Get("reference")
 
 		response, err := api.GetImageReleaseNotes(r.Context(), reference)
+		if response == nil && err == nil {
+			s.handleGenericResponse(w, r, ErrNotFound)
+			return
+		}
+
 		s.handleJSONResponse(w, r, response, err)
 	})