diff options
Diffstat (limited to 'pkg/ext/router.go')
| -rw-r--r-- | pkg/ext/router.go | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/pkg/ext/router.go b/pkg/ext/router.go index e12a40c..ce4c126 100644 --- a/pkg/ext/router.go +++ b/pkg/ext/router.go @@ -8,6 +8,7 @@ import ( "git.gabrielgio.me/cerrado/pkg/service" "git.gabrielgio.me/cerrado/templates" + "github.com/go-git/go-git/v5/plumbing" ) type ( @@ -36,7 +37,8 @@ func (r *Router) AddMiddleware(middleware Middleware) { func wrapError(next ErrorRequestHandler) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { if err := next(w, r); err != nil { - if errors.Is(err, service.ErrRepositoryNotFound) { + if errors.Is(err, service.ErrRepositoryNotFound) || + errors.Is(err, plumbing.ErrReferenceNotFound) { NotFound(w, r) } else { slog.Error("Internal Server Error", "error", err) |