mirror of
https://github.com/thegeeklab/wp-plugin-go.git
synced 2024-11-21 14:10:39 +00:00
feat: add custom cli types for string maps (#18)
BREAKING CHANGE: The existing type `StringSliceFlag` was moved to the new `types` package.
This commit is contained in:
parent
b9e185044c
commit
535fbe4771
74
types/stringmap.go
Normal file
74
types/stringmap.go
Normal file
@ -0,0 +1,74 @@
|
|||||||
|
package types
|
||||||
|
|
||||||
|
import "encoding/json"
|
||||||
|
|
||||||
|
type DeepStringMapFlag struct {
|
||||||
|
parts map[string]map[string]string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DeepStringMapFlag) String() string {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DeepStringMapFlag) Get() map[string]map[string]string {
|
||||||
|
return d.parts
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DeepStringMapFlag) Set(value string) error {
|
||||||
|
d.parts = map[string]map[string]string{}
|
||||||
|
|
||||||
|
err := json.Unmarshal([]byte(value), &d.parts)
|
||||||
|
if err != nil {
|
||||||
|
single := map[string]string{}
|
||||||
|
|
||||||
|
err := json.Unmarshal([]byte(value), &single)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
d.parts["*"] = single
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type StringMapFlag struct {
|
||||||
|
parts map[string]string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *StringMapFlag) String() string {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *StringMapFlag) Get() map[string]string {
|
||||||
|
return s.parts
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *StringMapFlag) Set(value string) error {
|
||||||
|
s.parts = map[string]string{}
|
||||||
|
|
||||||
|
err := json.Unmarshal([]byte(value), &s.parts)
|
||||||
|
if err != nil {
|
||||||
|
s.parts["*"] = value
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type MapFlag struct {
|
||||||
|
parts map[string]string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MapFlag) String() string {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MapFlag) Get() map[string]string {
|
||||||
|
return m.parts
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *MapFlag) Set(value string) error {
|
||||||
|
m.parts = map[string]string{}
|
||||||
|
|
||||||
|
return json.Unmarshal([]byte(value), &m.parts)
|
||||||
|
}
|
@ -1,8 +1,6 @@
|
|||||||
package plugin
|
package types
|
||||||
|
|
||||||
import (
|
import "strings"
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
// StringSliceFlag is a flag type which support comma separated values and escaping to not split at unwanted lines.
|
// StringSliceFlag is a flag type which support comma separated values and escaping to not split at unwanted lines.
|
||||||
type StringSliceFlag struct {
|
type StringSliceFlag struct {
|
@ -1,4 +1,4 @@
|
|||||||
package plugin
|
package types
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"reflect"
|
"reflect"
|
Loading…
Reference in New Issue
Block a user