summaryrefslogtreecommitdiff
path: root/vendor/github.com/google/yamlfmt/formatter.go
diff options
context:
space:
mode:
authormo khan <mo@mokhan.ca>2025-05-11 21:12:57 -0600
committermo khan <mo@mokhan.ca>2025-05-11 21:12:57 -0600
commit60440f90dca28e99a31dd328c5f6d5dc0f9b6a2e (patch)
tree2f54adf55086516f162f0a55a5347e6b25f7f176 /vendor/github.com/google/yamlfmt/formatter.go
parent05ca9b8d3a9c7203a3a3b590beaa400900bd9007 (diff)
chore: vendor go dependencies
Diffstat (limited to 'vendor/github.com/google/yamlfmt/formatter.go')
-rw-r--r--vendor/github.com/google/yamlfmt/formatter.go65
1 files changed, 65 insertions, 0 deletions
diff --git a/vendor/github.com/google/yamlfmt/formatter.go b/vendor/github.com/google/yamlfmt/formatter.go
new file mode 100644
index 0000000..ce432ee
--- /dev/null
+++ b/vendor/github.com/google/yamlfmt/formatter.go
@@ -0,0 +1,65 @@
+// Copyright 2024 Google LLC
+//
+// 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 yamlfmt
+
+import "fmt"
+
+type Formatter interface {
+ Type() string
+ Format(yamlContent []byte) ([]byte, error)
+ ConfigMap() (map[string]any, error)
+}
+
+type Factory interface {
+ Type() string
+ NewFormatter(config map[string]interface{}) (Formatter, error)
+}
+
+type Registry struct {
+ registry map[string]Factory
+ defaultType string
+}
+
+func NewFormatterRegistry(defaultFactory Factory) *Registry {
+ return &Registry{
+ registry: map[string]Factory{
+ defaultFactory.Type(): defaultFactory,
+ },
+ defaultType: defaultFactory.Type(),
+ }
+}
+
+func (r *Registry) Add(f Factory) {
+ r.registry[f.Type()] = f
+}
+
+func (r *Registry) GetFactory(fType string) (Factory, error) {
+ if fType == "" {
+ return r.GetDefaultFactory()
+ }
+ factory, ok := r.registry[fType]
+ if !ok {
+ return nil, fmt.Errorf("no formatter registered with type \"%s\"", fType)
+ }
+ return factory, nil
+}
+
+func (r *Registry) GetDefaultFactory() (Factory, error) {
+ factory, ok := r.registry[r.defaultType]
+ if !ok {
+ return nil, fmt.Errorf("no default formatter registered for type \"%s\"", r.defaultType)
+ }
+ return factory, nil
+}