-
Notifications
You must be signed in to change notification settings - Fork 214
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: Pavel Okhlopkov <[email protected]>
- Loading branch information
Pavel Okhlopkov
committed
Nov 14, 2024
1 parent
631dc0e
commit 9068de4
Showing
20 changed files
with
2,274 additions
and
17 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,144 @@ | ||
| package objectpatch | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "encoding/json" | ||
| "fmt" | ||
| "io" | ||
|
|
||
| "github.com/deckhouse/module-sdk/pkg/app" | ||
| "github.com/deckhouse/module-sdk/pkg/jq" | ||
| "gopkg.in/yaml.v3" | ||
| "k8s.io/apimachinery/pkg/apis/meta/v1/unstructured" | ||
| "k8s.io/apimachinery/pkg/runtime" | ||
| k8yaml "sigs.k8s.io/yaml" | ||
|
|
||
| "github.com/flant/kube-client/manifest" | ||
| ) | ||
|
|
||
| func unmarshalFromJSONOrYAML(specs []byte) ([]OperationSpec, error) { | ||
| fromJsonSpecs, err := unmarshalFromJson(specs) | ||
| if err != nil { | ||
| return unmarshalFromYaml(specs) | ||
| } | ||
|
|
||
| return fromJsonSpecs, nil | ||
| } | ||
|
|
||
| func unmarshalFromJson(jsonSpecs []byte) ([]OperationSpec, error) { | ||
| var specSlice []OperationSpec | ||
|
|
||
| dec := json.NewDecoder(bytes.NewReader(jsonSpecs)) | ||
| for { | ||
| var doc OperationSpec | ||
| err := dec.Decode(&doc) | ||
| if err == io.EOF { | ||
| break | ||
| } | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| specSlice = append(specSlice, doc) | ||
| } | ||
|
|
||
| return specSlice, nil | ||
| } | ||
|
|
||
| func unmarshalFromYaml(yamlSpecs []byte) ([]OperationSpec, error) { | ||
| var specSlice []OperationSpec | ||
|
|
||
| dec := yaml.NewDecoder(bytes.NewReader(yamlSpecs)) | ||
| for { | ||
| var doc OperationSpec | ||
| err := dec.Decode(&doc) | ||
| if err == io.EOF { | ||
| break | ||
| } | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| specSlice = append(specSlice, doc) | ||
| } | ||
|
|
||
| return specSlice, nil | ||
| } | ||
|
|
||
| func applyJQPatch(jqFilter string, obj *unstructured.Unstructured) (*unstructured.Unstructured, error) { | ||
| objBytes, err := obj.MarshalJSON() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| filterResult, err := jq.ApplyJqFilter(jqFilter, objBytes, app.JqLibraryPath) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to apply jqFilter:\n%sto Object:\n%s\n"+ | ||
| "error: %s", jqFilter, obj, err) | ||
| } | ||
|
|
||
| retObj := &unstructured.Unstructured{} | ||
| _, _, err = unstructured.UnstructuredJSONScheme.Decode([]byte(filterResult), nil, retObj) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to convert filterResult:\n%s\nto Unstructured Object\nerror: %s", filterResult, err) | ||
| } | ||
|
|
||
| return retObj, nil | ||
| } | ||
|
|
||
| func generateSubresources(subresource string) (ret []string) { | ||
| if subresource != "" { | ||
| ret = append(ret, subresource) | ||
| } | ||
|
|
||
| return | ||
| } | ||
|
|
||
| func toUnstructured(obj interface{}) (*unstructured.Unstructured, error) { | ||
| switch v := obj.(type) { | ||
| case []byte: | ||
| mft, err := manifest.NewFromYAML(string(v)) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| return mft.Unstructured(), nil | ||
| case string: | ||
| mft, err := manifest.NewFromYAML(v) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| return mft.Unstructured(), nil | ||
| case map[string]interface{}: | ||
| return &unstructured.Unstructured{Object: v}, nil | ||
| default: | ||
| objectContent, err := runtime.DefaultUnstructuredConverter.ToUnstructured(obj) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("convert to unstructured: %v", err) | ||
| } | ||
| return &unstructured.Unstructured{Object: objectContent}, nil | ||
| } | ||
| } | ||
|
|
||
| func convertPatchToBytes(patch interface{}) ([]byte, error) { | ||
| var err error | ||
| var intermediate interface{} | ||
| switch v := patch.(type) { | ||
| case []byte: | ||
| err = k8yaml.Unmarshal(v, &intermediate) | ||
| case string: | ||
| err = k8yaml.Unmarshal([]byte(v), &intermediate) | ||
| default: | ||
| intermediate = v | ||
| } | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| // Try to encode to JSON. | ||
| var patchBytes []byte | ||
| patchBytes, err = json.Marshal(intermediate) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| return patchBytes, nil | ||
| } |
Oops, something went wrong.