mirror of https://github.com/databricks/cli.git
73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
package flags
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/databricks/cli/libs/dyn/convert"
|
|
"github.com/databricks/cli/libs/dyn/jsonloader"
|
|
"github.com/databricks/databricks-sdk-go/marshal"
|
|
)
|
|
|
|
type JsonFlag struct {
|
|
raw []byte
|
|
}
|
|
|
|
func (j *JsonFlag) String() string {
|
|
return fmt.Sprintf("JSON (%d bytes)", len(j.raw))
|
|
}
|
|
|
|
// TODO: Command.MarkFlagFilename()
|
|
func (j *JsonFlag) Set(v string) error {
|
|
// Load request from file if it starts with '@' (like curl).
|
|
if v[0] != '@' {
|
|
j.raw = []byte(v)
|
|
return nil
|
|
}
|
|
buf, err := os.ReadFile(v[1:])
|
|
if err != nil {
|
|
return fmt.Errorf("read %s: %w", v, err)
|
|
}
|
|
j.raw = buf
|
|
return nil
|
|
}
|
|
|
|
func (j *JsonFlag) Unmarshal(v any) error {
|
|
if j.raw == nil {
|
|
return nil
|
|
}
|
|
|
|
dv, err := jsonloader.LoadJSON(j.raw)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// First normalize the input data.
|
|
// It will convert all the values to the correct types.
|
|
// For example string literals for booleans and integers will be converted to the correct types.
|
|
nv, diags := convert.Normalize(v, dv)
|
|
if len(diags) > 0 {
|
|
summary := ""
|
|
for _, diag := range diags {
|
|
summary += fmt.Sprintf("- %s\n", diag.Summary)
|
|
}
|
|
return fmt.Errorf("json input error:\n%v", summary)
|
|
}
|
|
|
|
// Then marshal the normalized data to the output.
|
|
// It will serialize all set data with the correct types.
|
|
data, err := json.Marshal(nv.AsAny())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Finally unmarshal the normalized data to the output.
|
|
// It will fill in the ForceSendFields field if the struct contains it.
|
|
return marshal.Unmarshal(data, v)
|
|
}
|
|
|
|
func (j *JsonFlag) Type() string {
|
|
return "JSON"
|
|
}
|