This commit is contained in:
Yann Hamon 2025-05-10 23:56:10 +02:00
parent ecc042d7f9
commit c954a22d7d
6 changed files with 1392 additions and 300 deletions

View file

@ -1,9 +1,12 @@
package loader
import (
"bytes"
"errors"
"fmt"
"github.com/santhosh-tekuri/jsonschema/v6"
"github.com/yannh/kubeconform/pkg/cache"
"io"
gourl "net/url"
"os"
"path/filepath"
@ -23,28 +26,32 @@ func (l FileLoader) Load(url string) (any, error) {
}
if l.cache != nil {
if cached, err := l.cache.Get(path); err == nil {
return cached, nil
return jsonschema.UnmarshalJSON(bytes.NewReader(cached))
}
}
f, err := os.Open(path)
if err != nil {
if os.IsNotExist(err) {
msg := fmt.Sprintf("could not open file %s", path)
return nil, NewNotFoundError(errors.New(msg))
}
return nil, err
}
defer f.Close()
s, err := jsonschema.UnmarshalJSON(f)
content, err := io.ReadAll(f)
if err != nil {
return nil, err
}
if l.cache != nil {
if err = l.cache.Set(path, s); err != nil {
if err = l.cache.Set(path, content); err != nil {
return nil, fmt.Errorf("failed to write cache to disk: %s", err)
}
}
return s, nil
return jsonschema.UnmarshalJSON(bytes.NewReader(content))
}
// ToFile is helper method to convert file url to file path.
@ -54,7 +61,7 @@ func (l FileLoader) ToFile(url string) (string, error) {
return "", err
}
if u.Scheme != "file" {
return "", fmt.Errorf("invalid file url: %s", u)
return url, nil
}
path := u.Path
if runtime.GOOS == "windows" {

View file

@ -21,7 +21,7 @@ type HTTPURLLoader struct {
func (l *HTTPURLLoader) Load(url string) (any, error) {
if l.cache != nil {
if cached, err := l.cache.Get(url); err == nil {
return cached, nil
return jsonschema.UnmarshalJSON(bytes.NewReader(cached))
}
}
@ -47,17 +47,18 @@ func (l *HTTPURLLoader) Load(url string) (any, error) {
msg := fmt.Sprintf("failed parsing schema from %s: %s", url, err)
return nil, errors.New(msg)
}
if l.cache != nil {
if err = l.cache.Set(url, body); err != nil {
return nil, fmt.Errorf("failed to write cache to disk: %s", err)
}
}
s, err := jsonschema.UnmarshalJSON(bytes.NewReader(body))
if err != nil {
return nil, err
}
if l.cache != nil {
if err = l.cache.Set(url, s); err != nil {
return nil, fmt.Errorf("failed to write cache to disk: %s", err)
}
}
return s, nil
}