podman
48 строк · 1.4 Кб
1// Copyright 2015 go-swagger maintainers
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15package middleware
16
17import (
18"net/http"
19"path"
20)
21
22// Spec creates a middleware to serve a swagger spec.
23// This allows for altering the spec before starting the http listener.
24// This can be useful if you want to serve the swagger spec from another path than /swagger.json
25//
26func Spec(basePath string, b []byte, next http.Handler) http.Handler {
27if basePath == "" {
28basePath = "/"
29}
30pth := path.Join(basePath, "swagger.json")
31
32return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
33if r.URL.Path == pth {
34rw.Header().Set("Content-Type", "application/json")
35rw.WriteHeader(http.StatusOK)
36//#nosec
37_, _ = rw.Write(b)
38return
39}
40
41if next == nil {
42rw.Header().Set("Content-Type", "application/json")
43rw.WriteHeader(http.StatusNotFound)
44return
45}
46next.ServeHTTP(rw, r)
47})
48}
49