forked from forbole/juno
-
Notifications
You must be signed in to change notification settings - Fork 0
/
types.go
50 lines (40 loc) · 1.56 KB
/
types.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package init
import (
"github.com/spf13/cobra"
"github.com/gotabit/juno/v5/types/config"
)
// WritableConfig represents a configuration that can be written to a file
type WritableConfig interface {
// GetBytes returns the bytes to be written to the config file when initializing it
GetBytes() ([]byte, error)
}
// ConfigCreator represents a function that builds a Config instance from the flags that have been specified by the
// user inside the given command.
type ConfigCreator = func(cmd *cobra.Command) WritableConfig
// DefaultConfigCreator represents the default configuration creator that builds a Config instance using the values
// specified using the default flags.
func DefaultConfigCreator(_ *cobra.Command) WritableConfig {
return config.DefaultConfig()
}
// --------------------------------------------------------------------------------------------------------------------
// Config contains the configuration data for the init command
type Config struct {
createConfig ConfigCreator
}
// NewConfig allows to build a new Config instance
func NewConfig() *Config {
return &Config{}
}
// WithConfigCreator sets the given setup function as the configuration creator
func (c *Config) WithConfigCreator(creator ConfigCreator) *Config {
c.createConfig = creator
return c
}
// GetConfigCreator return the function that should be run to create a configuration from a set of
// flags specified by the user with the "init" command
func (c *Config) GetConfigCreator() ConfigCreator {
if c.createConfig == nil {
return DefaultConfigCreator
}
return c.createConfig
}