mirror of
https://gitea.com/Lydanne/buildx.git
synced 2025-05-18 00:47:48 +08:00
121 lines
3.7 KiB
Go
121 lines
3.7 KiB
Go
/*
|
|
Copyright 2020 The Compose Specification Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package transform
|
|
|
|
import (
|
|
"github.com/compose-spec/compose-go/v2/tree"
|
|
)
|
|
|
|
type transformFunc func(data any, p tree.Path, ignoreParseError bool) (any, error)
|
|
|
|
var transformers = map[tree.Path]transformFunc{}
|
|
|
|
func init() {
|
|
transformers["services.*"] = transformService
|
|
transformers["services.*.build.secrets.*"] = transformFileMount
|
|
transformers["services.*.build.additional_contexts"] = transformKeyValue
|
|
transformers["services.*.depends_on"] = transformDependsOn
|
|
transformers["services.*.env_file"] = transformEnvFile
|
|
transformers["services.*.label_file"] = transformStringOrList
|
|
transformers["services.*.extends"] = transformExtends
|
|
transformers["services.*.gpus"] = transformGpus
|
|
transformers["services.*.networks"] = transformServiceNetworks
|
|
transformers["services.*.volumes.*"] = transformVolumeMount
|
|
transformers["services.*.dns"] = transformStringOrList
|
|
transformers["services.*.devices.*"] = transformDeviceMapping
|
|
transformers["services.*.secrets.*"] = transformFileMount
|
|
transformers["services.*.configs.*"] = transformFileMount
|
|
transformers["services.*.ports"] = transformPorts
|
|
transformers["services.*.build"] = transformBuild
|
|
transformers["services.*.build.ssh"] = transformSSH
|
|
transformers["services.*.ulimits.*"] = transformUlimits
|
|
transformers["services.*.build.ulimits.*"] = transformUlimits
|
|
transformers["volumes.*"] = transformMaybeExternal
|
|
transformers["networks.*"] = transformMaybeExternal
|
|
transformers["secrets.*"] = transformMaybeExternal
|
|
transformers["configs.*"] = transformMaybeExternal
|
|
transformers["include.*"] = transformInclude
|
|
}
|
|
|
|
func transformStringOrList(data any, _ tree.Path, _ bool) (any, error) {
|
|
switch t := data.(type) {
|
|
case string:
|
|
return []any{t}, nil
|
|
default:
|
|
return data, nil
|
|
}
|
|
}
|
|
|
|
// Canonical transforms a compose model into canonical syntax
|
|
func Canonical(yaml map[string]any, ignoreParseError bool) (map[string]any, error) {
|
|
canonical, err := transform(yaml, tree.NewPath(), ignoreParseError)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return canonical.(map[string]any), nil
|
|
}
|
|
|
|
func transform(data any, p tree.Path, ignoreParseError bool) (any, error) {
|
|
for pattern, transformer := range transformers {
|
|
if p.Matches(pattern) {
|
|
t, err := transformer(data, p, ignoreParseError)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return t, nil
|
|
}
|
|
}
|
|
switch v := data.(type) {
|
|
case map[string]any:
|
|
a, err := transformMapping(v, p, ignoreParseError)
|
|
if err != nil {
|
|
return a, err
|
|
}
|
|
return v, nil
|
|
case []any:
|
|
a, err := transformSequence(v, p, ignoreParseError)
|
|
if err != nil {
|
|
return a, err
|
|
}
|
|
return v, nil
|
|
default:
|
|
return data, nil
|
|
}
|
|
}
|
|
|
|
func transformSequence(v []any, p tree.Path, ignoreParseError bool) ([]any, error) {
|
|
for i, e := range v {
|
|
t, err := transform(e, p.Next("[]"), ignoreParseError)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
v[i] = t
|
|
}
|
|
return v, nil
|
|
}
|
|
|
|
func transformMapping(v map[string]any, p tree.Path, ignoreParseError bool) (map[string]any, error) {
|
|
for k, e := range v {
|
|
t, err := transform(e, p.Next(k), ignoreParseError)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
v[k] = t
|
|
}
|
|
return v, nil
|
|
}
|