diff options
Diffstat (limited to 'util')
| -rw-r--r-- | util/bind_config.go | 43 |
1 files changed, 41 insertions, 2 deletions
diff --git a/util/bind_config.go b/util/bind_config.go index f849398..2ab6cc9 100644 --- a/util/bind_config.go +++ b/util/bind_config.go @@ -36,10 +36,18 @@ type SearchConfig struct { SearchNSFW bool } +type ListConfig struct { + ResultsLength int + ResultsOffset int + IncludeNSFW bool +} + // handles prompt-length, search-length, search-offset and search-nsfw func BindSearchConfig(flags *pflag.FlagSet) (SearchConfig, error) { - var conf SearchConfig - var err error + var ( + conf SearchConfig + err error + ) if flags.Lookup("prompt-length").Changed { conf.PromptLength, err = flags.GetInt("prompt-length") @@ -71,3 +79,34 @@ func BindSearchConfig(flags *pflag.FlagSet) (SearchConfig, error) { return conf, nil } + +// handles results-length, results-offset, include-nsfw +func BindListConfig(flags *pflag.FlagSet) (ListConfig, error) { + var ( + conf ListConfig + err error + ) + + if flags.Lookup("results-length").Changed { + conf.ResultsLength, err = flags.GetInt("results-length") + if err != nil {return conf, err} + } else { + conf.ResultsLength = viper.GetInt("lists.list_length") + } + + if flags.Lookup("results-offset").Changed { + conf.ResultsOffset, err = flags.GetInt("results-offset") + if err != nil {return conf, err} + } else { + conf.ResultsOffset = viper.GetInt("lists.list_offset") + } + + if flags.Lookup("include-nsfw").Changed { + conf.IncludeNSFW, err = flags.GetBool("include-nsfw") + if err != nil {return conf, err} + } else { + conf.IncludeNSFW = viper.GetBool("lists.include_nsfw_results") + } + + return conf, nil +} |