2023-03-02 15:37:05 -05:00
|
|
|
// Copyright (c) HashiCorp, Inc.
|
2023-08-10 15:53:29 -07:00
|
|
|
// SPDX-License-Identifier: BUSL-1.1
|
2023-03-02 15:37:05 -05:00
|
|
|
|
2015-05-26 09:58:04 -07:00
|
|
|
package kvflag
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"encoding/json"
|
|
|
|
|
"fmt"
|
|
|
|
|
"os"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// FlagJSON is a flag.Value implementation for parsing user variables
|
|
|
|
|
// from the command-line using JSON files.
|
|
|
|
|
type FlagJSON map[string]string
|
|
|
|
|
|
|
|
|
|
func (v *FlagJSON) String() string {
|
|
|
|
|
return ""
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (v *FlagJSON) Set(raw string) error {
|
|
|
|
|
f, err := os.Open(raw)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|
defer f.Close()
|
|
|
|
|
|
|
|
|
|
if *v == nil {
|
|
|
|
|
*v = make(map[string]string)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if err := json.NewDecoder(f).Decode(v); err != nil {
|
|
|
|
|
return fmt.Errorf(
|
|
|
|
|
"Error reading variables in '%s': %s", raw, err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
}
|