From 52c835053007c5991903e778b5dbf64a25aea0a4 Mon Sep 17 00:00:00 2001
From: Brent Baude <bbaude@redhat.com>
Date: Wed, 18 Mar 2020 14:33:39 -0500
Subject: serve swagger when present

register the swagger endpoint and add some error handling for when the swagger file does not exist

Signed-off-by: Brent Baude <bbaude@redhat.com>
---
 pkg/api/handlers/libpod/swagger.go | 29 ++++++++++++++++++++++++++++-
 pkg/api/server/register_swagger.go | 15 ++-------------
 pkg/api/server/server.go           |  1 +
 3 files changed, 31 insertions(+), 14 deletions(-)

(limited to 'pkg')

diff --git a/pkg/api/handlers/libpod/swagger.go b/pkg/api/handlers/libpod/swagger.go
index f6a26134b..149fa10dc 100644
--- a/pkg/api/handlers/libpod/swagger.go
+++ b/pkg/api/handlers/libpod/swagger.go
@@ -1,6 +1,16 @@
 package libpod
 
-import "github.com/containers/image/v5/manifest"
+import (
+	"net/http"
+	"os"
+
+	"github.com/containers/image/v5/manifest"
+	"github.com/containers/libpod/pkg/api/handlers/utils"
+	"github.com/pkg/errors"
+)
+
+// DefaultPodmanSwaggerSpec provides the default path to the podman swagger spec file
+const DefaultPodmanSwaggerSpec = "/usr/share/containers/podman/swagger.yaml"
 
 // List Containers
 // swagger:response ListContainers
@@ -15,3 +25,20 @@ type swagInspectManifestResponse struct {
 	// in:body
 	Body manifest.List
 }
+
+func ServeSwagger(w http.ResponseWriter, r *http.Request) {
+	path := DefaultPodmanSwaggerSpec
+	if p, found := os.LookupEnv("PODMAN_SWAGGER_SPEC"); found {
+		path = p
+	}
+	if _, err := os.Stat(path); err != nil {
+		if os.IsNotExist(err) {
+			utils.InternalServerError(w, errors.Errorf("file %q does not exist", path))
+			return
+		}
+		utils.InternalServerError(w, err)
+		return
+	}
+	w.Header().Set("Content-Type", "text/yaml")
+	http.ServeFile(w, r, path)
+}
diff --git a/pkg/api/server/register_swagger.go b/pkg/api/server/register_swagger.go
index 5564ec096..9048c1951 100644
--- a/pkg/api/server/register_swagger.go
+++ b/pkg/api/server/register_swagger.go
@@ -2,25 +2,14 @@ package server
 
 import (
 	"net/http"
-	"os"
 
+	"github.com/containers/libpod/pkg/api/handlers/libpod"
 	"github.com/gorilla/mux"
 )
 
-// DefaultPodmanSwaggerSpec provides the default path to the podman swagger spec file
-const DefaultPodmanSwaggerSpec = "/usr/share/containers/podman/swagger.yaml"
-
 // RegisterSwaggerHandlers maps the swagger endpoint for the server
 func (s *APIServer) RegisterSwaggerHandlers(r *mux.Router) error {
 	// This handler does _*NOT*_ provide an UI rather just a swagger spec that an UI could render
-	r.PathPrefix("/swagger/").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		path := DefaultPodmanSwaggerSpec
-		if p, found := os.LookupEnv("PODMAN_SWAGGER_SPEC"); found {
-			path = p
-		}
-		w.Header().Set("Content-Type", "text/yaml")
-
-		http.ServeFile(w, r, path)
-	})
+	r.HandleFunc(VersionedPath("/libpod/swagger"), s.APIHandler(libpod.ServeSwagger)).Methods(http.MethodGet)
 	return nil
 }
diff --git a/pkg/api/server/server.go b/pkg/api/server/server.go
index 8496cd11c..3d4974535 100644
--- a/pkg/api/server/server.go
+++ b/pkg/api/server/server.go
@@ -109,6 +109,7 @@ func newServer(runtime *libpod.Runtime, duration time.Duration, listener *net.Li
 		server.registerPingHandlers,
 		server.registerPluginsHandlers,
 		server.registerPodsHandlers,
+		server.RegisterSwaggerHandlers,
 		server.registerSwarmHandlers,
 		server.registerSystemHandlers,
 		server.registerVersionHandlers,
-- 
cgit v1.2.3-54-g00ecf