Start rewriting YAML impementation

ammario · ammario · commit 5c4c1775d3b6 · 2023-02-24T20:55:29.000Z
diff --git a/cli/deployment/config.go b/cli/deployment/config.go
@@ -19,7 +19,7 @@ import (
 	"github.com/coder/coder/codersdk"
 )
 
-func newConfig() *codersdk.DeploymentConfig {
+func DefaultConfig() *codersdk.DeploymentConfig {
 	return &codersdk.DeploymentConfig{
 		AccessURL: &codersdk.DeploymentConfigField[string]{
 			Name:  "Access URL",
@@ -587,7 +587,7 @@ func newConfig() *codersdk.DeploymentConfig {
 
 //nolint:revive
 func Config(configPath string, vip *viper.Viper) (*codersdk.DeploymentConfig, error) {
-	dc := newConfig()
+	dc := DefaultConfig()
 	vip.SetEnvPrefix("CODER")
 
 	if configPath != "" {
@@ -756,7 +756,7 @@ func readSliceFromViper[T any](vip *viper.Viper, key string, value any) []T {
 }
 
 func NewViper() *viper.Viper {
-	dc := newConfig()
+	dc := DefaultConfig()
 	vip := viper.New()
 	vip.SetEnvPrefix("coder")
 	vip.SetEnvKeyReplacer(strings.NewReplacer("-", "_", ".", "_"))
@@ -800,7 +800,7 @@ func setViperDefaults(prefix string, vip *viper.Viper, target interface{}) {
 
 //nolint:revive
 func AttachFlags(flagset *pflag.FlagSet, vip *viper.Viper, enterprise bool) {
-	setFlags("", flagset, vip, newConfig(), enterprise)
+	setFlags("", flagset, vip, DefaultConfig(), enterprise)
 }
 
 //nolint:revive
diff --git a/cli/deployment/yaml.go b/cli/deployment/yaml.go
@@ -0,0 +1,69 @@
+package deployment
+
+import (
+	"reflect"
+	"strconv"
+	"strings"
+	"time"
+
+	"golang.org/x/xerrors"
+	"gopkg.in/yaml.v3"
+
+	"github.com/hashicorp/go-multierror"
+
+	"github.com/coder/coder/codersdk"
+)
+
+type yamlSchema map[string]interface{}
+
+func yamlDeploymentField(field reflect.Value) (*yaml.Node, error) {
+	valueField := field.FieldByName("Value")
+	var valueStr string
+	switch v := valueField.Interface().(type) {
+	case time.Duration:
+		valueStr = v.String()
+	case bool:
+		valueStr = strconv.FormatBool(v)
+	case string:
+		valueStr = v
+	default:
+		return nil, xerrors.Errorf(
+			"unsupported DeploymentField type: %s", valueField.Type().String(),
+		)
+	}
+	return &yaml.Node{
+		Kind:  yaml.ScalarNode,
+		Value: valueStr,
+	}, nil
+}
+
+// MarshalYAML converts the deployment config to its MarshalYAML representation.
+func MarshalYAML(config *codersdk.DeploymentConfig) ([]byte, error) {
+	var (
+		schema      = yamlSchema{}
+		configValue = reflect.ValueOf(config).Elem()
+		merr        *multierror.Error
+	)
+	for i := 0; i < configValue.NumField(); i++ {
+		var (
+			configField = configValue.Field(i).Elem()
+			typeName    = configField.Type().String()
+			fieldName   = configValue.Type().Field(i).Name
+		)
+
+		switch {
+		case strings.HasPrefix(typeName, "codersdk.DeploymentConfigField["):
+			node, err := yamlDeploymentField(configField)
+			if err != nil {
+				merr = multierror.Append(merr, err)
+				continue
+			}
+			schema[fieldName] = node
+		default:
+			merr = multierror.Append(merr, xerrors.Errorf("unsupported type: %s", typeName))
+		}
+	}
+	byt, err := yaml.Marshal(schema)
+	merr = multierror.Append(merr, err)
+	return byt, merr.ErrorOrNil()
+}
diff --git a/cli/deployment/yaml_test.go b/cli/deployment/yaml_test.go
@@ -0,0 +1,20 @@
+package deployment_test
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/coder/cli/deployment"
+)
+
+func TestMarshalYAML(t *testing.T) {
+	t.Parallel()
+
+	t.Run("Default", func(t *testing.T) {
+		t.Parallel()
+		byt, err := deployment.MarshalYAML(deployment.DefaultConfig())
+		t.Logf("yaml:\n%s", string(byt))
+		require.NoError(t, err)
+	})
+}
