b2700903ae
As deepmap/oapi-codegen didn't work with this newer version, upgrade to oapi-codegen/oapi-codegen v2. Mitigating CVE-2025-30153
78 lines
2.1 KiB
Go
78 lines
2.1 KiB
Go
package loader
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/speakeasy-api/openapi-overlay/pkg/overlay"
|
|
"gopkg.in/yaml.v3"
|
|
"net/url"
|
|
"os"
|
|
)
|
|
|
|
// GetOverlayExtendsPath returns the path to file if the extends URL is a file
|
|
// URL. Otherwise, returns an empty string and an error. The error may occur if
|
|
// no extends URL is present or if the URL is not a file URL or if the URL is
|
|
// malformed.
|
|
func GetOverlayExtendsPath(o *overlay.Overlay) (string, error) {
|
|
if o.Extends == "" {
|
|
return "", fmt.Errorf("overlay does not specify an extends URL")
|
|
}
|
|
|
|
specUrl, err := url.Parse(o.Extends)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to parse URL %q: %w", o.Extends, err)
|
|
}
|
|
|
|
if specUrl.Scheme != "file" {
|
|
return "", fmt.Errorf("only file:// extends URLs are supported, not %q", o.Extends)
|
|
}
|
|
|
|
return specUrl.Path, nil
|
|
}
|
|
|
|
// LoadExtendsSpecification will load and parse a YAML or JSON file as specified
|
|
// in the extends parameter of the overlay. Currently, this only supports file
|
|
// URLs.
|
|
func LoadExtendsSpecification(o *overlay.Overlay) (*yaml.Node, error) {
|
|
path, err := GetOverlayExtendsPath(o)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return LoadSpecification(path)
|
|
}
|
|
|
|
// LoadSpecification will load and parse a YAML or JSON file from the given path.
|
|
func LoadSpecification(path string) (*yaml.Node, error) {
|
|
rs, err := os.Open(path)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to open schema from path %q: %w", path, err)
|
|
}
|
|
|
|
var ys yaml.Node
|
|
err = yaml.NewDecoder(rs).Decode(&ys)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to parse schema at path %q: %w", path, err)
|
|
}
|
|
|
|
return &ys, nil
|
|
}
|
|
|
|
// LoadEitherSpecification is a convenience function that will load a
|
|
// specification from the given file path if it is non-empty. Otherwise, it will
|
|
// attempt to load the path from the overlay's extends URL. Also returns the name
|
|
// of the file loaded.
|
|
func LoadEitherSpecification(path string, o *overlay.Overlay) (*yaml.Node, string, error) {
|
|
var (
|
|
y *yaml.Node
|
|
err error
|
|
)
|
|
|
|
if path != "" {
|
|
y, err = LoadSpecification(path)
|
|
} else {
|
|
path, _ = GetOverlayExtendsPath(o)
|
|
y, err = LoadExtendsSpecification(o)
|
|
}
|
|
|
|
return y, path, err
|
|
}
|