Merge branch 'refactor/serpents'

This commit is contained in:
Victor Lacasse-Beaudoin 2023-10-08 01:38:27 -04:00
commit 4b42f8d215
4 changed files with 47 additions and 76 deletions

View file

@ -31,7 +31,7 @@ func init() {
rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.bottin-ag.yaml)") rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.bottin-ag.yaml)")
if err := config.RegisterFlags(rootCmd); err != nil { if err := config.RegisterFlags(rootCmd.PersistentFlags()); err != nil {
log.Fatal(err) log.Fatal(err)
} }
} }

View file

@ -5,7 +5,8 @@ config options and flags
package config package config
import ( import (
"github.com/spf13/cobra" "codeberg.org/vlbeaudoin/serpents"
"github.com/spf13/pflag"
"github.com/spf13/viper" "github.com/spf13/viper"
) )
@ -55,152 +56,119 @@ func UnmarshalConfig() (cfg Config, err error) {
return cfg, viper.Unmarshal(&cfg) return cfg, viper.Unmarshal(&cfg)
} }
// RegisterBool registers a new cobra bool flag and associated viper config option
func RegisterBool(cmd *cobra.Command, isPersistent bool, viperName, cobraName, cobraDescription string, cobraDefault bool) error {
if isPersistent {
cmd.PersistentFlags().Bool(cobraName, cobraDefault, cobraDescription)
return viper.BindPFlag(viperName, cmd.PersistentFlags().Lookup(cobraName))
} else {
cmd.Flags().Bool(cobraName, cobraDefault, cobraDescription)
return viper.BindPFlag(viperName, cmd.Flags().Lookup(cobraName))
}
}
// RegisterInt registers a new cobra int flag and associated viper config option
func RegisterInt(cmd *cobra.Command, isPersistent bool, viperName, cobraName, cobraDescription string, cobraDefault int) error {
if isPersistent {
cmd.PersistentFlags().Int(cobraName, cobraDefault, cobraDescription)
return viper.BindPFlag(viperName, cmd.PersistentFlags().Lookup(cobraName))
} else {
cmd.Flags().Int(cobraName, cobraDefault, cobraDescription)
return viper.BindPFlag(viperName, cmd.Flags().Lookup(cobraName))
}
}
// RegisterString registers a new cobra string flag and associated viper config option
func RegisterString(cmd *cobra.Command, isPersistent bool, viperName, cobraName, cobraDescription string, cobraDefault string) error {
if isPersistent {
cmd.PersistentFlags().String(cobraName, cobraDefault, cobraDescription)
return viper.BindPFlag(viperName, cmd.PersistentFlags().Lookup(cobraName))
} else {
cmd.Flags().String(cobraName, cobraDefault, cobraDescription)
return viper.BindPFlag(viperName, cmd.Flags().Lookup(cobraName))
}
}
// RegisterFlags adds persistent flags necessary to the application to the specified *cobra.Command // RegisterFlags adds persistent flags necessary to the application to the specified *cobra.Command
func RegisterFlags(cmd *cobra.Command) error { func RegisterFlags(flagSet *pflag.FlagSet) error {
// api.port ; --api-port // api.port ; --api-port
cmd.PersistentFlags().Int("api-port", 3182, "API server port") if err := serpents.Int(flagSet,
if err := viper.BindPFlag("api.port", cmd.PersistentFlags().Lookup("api-port")); err != nil { "api.port", "api-port", 3182, "API server port"); err != nil {
return err return err
} }
// api.key ; --api-key // api.key ; --api-key
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"api.key", "api-key", "API server key", "bottinag"); err != nil { "api.key", "api-key", "bottinag", "API server key"); err != nil {
return err return err
} }
// bottin.api.host // bottin.api.host
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"bottin.api.host", "bottin-api-host", "Bottin API server host", "localhost"); err != nil { "bottin.api.host", "bottin-api-host", "localhost", "Bottin API server host"); err != nil {
return err return err
} }
// bottin.api.key // bottin.api.key
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"bottin.api.key", "bottin-api-key", "Bottin API server key", "bottin"); err != nil { "bottin.api.key", "bottin-api-key", "bottin", "Bottin API server key"); err != nil {
return err return err
} }
// bottin.api.port // bottin.api.port
if err := RegisterInt(cmd, true, if err := serpents.Int(flagSet,
"bottin.api.port", "bottin-api-port", "Bottin API server port", 1312); err != nil { "bottin.api.port", "bottin-api-port", 1312, "Bottin API server port"); err != nil {
return err return err
} }
// bottin.api.protocol // bottin.api.protocol
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"bottin.api.protocol", "bottin-api-protocol", "Bottin API server protocol", "http"); err != nil { "bottin.api.protocol", "bottin-api-protocol", "http", "Bottin API server protocol"); err != nil {
return err return err
} }
// db.database // db.database
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"db.database", "db-database", "PostgreSQL database", "bottinag"); err != nil { "db.database", "db-database", "bottinag", "PostgreSQL database"); err != nil {
return err return err
} }
// db.host // db.host
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"db.host", "db-host", "PostgreSQL host", "db"); err != nil { "db.host", "db-host", "db", "PostgreSQL host"); err != nil {
return err return err
} }
// db.password // db.password
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"db.password", "db-password", "PostgreSQL password", "bottinag"); err != nil { "db.password", "db-password", "bottinag", "PostgreSQL password"); err != nil {
return err return err
} }
// db.port // db.port
if err := RegisterInt(cmd, true, if err := serpents.Int(flagSet,
"db.port", "db-port", "PostgreSQL port", 5432); err != nil { "db.port", "db-port", 5432, "PostgreSQL port"); err != nil {
return err return err
} }
// db.user // db.user
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"db.user", "db-user", "PostgreSQL user", "bottinag"); err != nil { "db.user", "db-user", "bottinag", "PostgreSQL user"); err != nil {
return err return err
} }
// db.use_ssl // db.use_ssl
if err := RegisterBool(cmd, true, if err := serpents.Bool(flagSet,
"db.use_ssl", "db-use-ssl", "PostgreSQL use_ssl", false); err != nil { "db.use_ssl", "db-use-ssl", false, "PostgreSQL use_ssl"); err != nil {
return err return err
} }
// web.api.host // web.api.host
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"web.api.host", "web-api-host", "Webserver API client host", "localhost"); err != nil { "web.api.host", "web-api-host", "localhost", "Webserver API client host"); err != nil {
return err return err
} }
// web.api.key // web.api.key
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"web.api.key", "web-api-key", "Webserver API client key", "bottinag"); err != nil { "web.api.key", "web-api-key", "bottinag", "Webserver API client key"); err != nil {
return err return err
} }
// web.api.port // web.api.port
if err := RegisterInt(cmd, true, if err := serpents.Int(flagSet,
"web.api.port", "web-api-port", "Webserver API client port", 3182); err != nil { "web.api.port", "web-api-port", 3182, "Webserver API client port"); err != nil {
return err return err
} }
// web.api.protocol // web.api.protocol
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"web.api.protocol", "web-api-protocol", "Webserver API client protocol", "http"); err != nil { "web.api.protocol", "web-api-protocol", "http", "Webserver API client protocol"); err != nil {
return err return err
} }
// web.password ; --web-password // web.password ; --web-password
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"web.password", "web-password", "Webserver basic auth password", "bottinag"); err != nil { "web.password", "web-password", "bottinag", "Webserver basic auth password"); err != nil {
return err return err
} }
// web.port ; --web-port // web.port ; --web-port
cmd.PersistentFlags().Int("web-port", 3183, "Webserver port") if err := serpents.Int(flagSet,
if err := viper.BindPFlag("web.port", cmd.PersistentFlags().Lookup("web-port")); err != nil { "web.port", "web-port", 3183, "Webserver port"); err != nil {
return err return err
} }
// web.user ; --web-user // web.user ; --web-user
if err := RegisterString(cmd, true, if err := serpents.String(flagSet,
"web.user", "web-user", "Webserver basic auth username", "bottinag"); err != nil { "web.user", "web-user", "bottinag", "Webserver basic auth username"); err != nil {
return err return err
} }

5
go.mod
View file

@ -1,13 +1,15 @@
module git.agecem.com/agecem/bottin-ag module git.agecem.com/agecem/bottin-ag
go 1.21.0 go 1.21.1
require ( require (
codeberg.org/vlbeaudoin/serpents v1.0.0
codeberg.org/vlbeaudoin/voki v1.3.1 codeberg.org/vlbeaudoin/voki v1.3.1
git.agecem.com/agecem/bottin/v5 v5.3.0 git.agecem.com/agecem/bottin/v5 v5.3.0
github.com/jmoiron/sqlx v1.3.5 github.com/jmoiron/sqlx v1.3.5
github.com/labstack/echo/v4 v4.11.1 github.com/labstack/echo/v4 v4.11.1
github.com/spf13/cobra v1.7.0 github.com/spf13/cobra v1.7.0
github.com/spf13/pflag v1.0.5
github.com/spf13/viper v1.16.0 github.com/spf13/viper v1.16.0
) )
@ -27,7 +29,6 @@ require (
github.com/spf13/afero v1.9.5 // indirect github.com/spf13/afero v1.9.5 // indirect
github.com/spf13/cast v1.5.1 // indirect github.com/spf13/cast v1.5.1 // indirect
github.com/spf13/jwalterweatherman v1.1.0 // indirect github.com/spf13/jwalterweatherman v1.1.0 // indirect
github.com/spf13/pflag v1.0.5 // indirect
github.com/subosito/gotenv v1.4.2 // indirect github.com/subosito/gotenv v1.4.2 // indirect
github.com/valyala/bytebufferpool v1.0.0 // indirect github.com/valyala/bytebufferpool v1.0.0 // indirect
github.com/valyala/fasttemplate v1.2.2 // indirect github.com/valyala/fasttemplate v1.2.2 // indirect

2
go.sum
View file

@ -35,6 +35,8 @@ cloud.google.com/go/storage v1.6.0/go.mod h1:N7U0C8pVQ/+NIKOBQyamJIeKQKkZ+mxpohl
cloud.google.com/go/storage v1.8.0/go.mod h1:Wv1Oy7z6Yz3DshWRJFhqM/UCfaWIRTdp0RXyy7KQOVs= cloud.google.com/go/storage v1.8.0/go.mod h1:Wv1Oy7z6Yz3DshWRJFhqM/UCfaWIRTdp0RXyy7KQOVs=
cloud.google.com/go/storage v1.10.0/go.mod h1:FLPqc6j+Ki4BU591ie1oL6qBQGu2Bl/tZ9ullr3+Kg0= cloud.google.com/go/storage v1.10.0/go.mod h1:FLPqc6j+Ki4BU591ie1oL6qBQGu2Bl/tZ9ullr3+Kg0=
cloud.google.com/go/storage v1.14.0/go.mod h1:GrKmX003DSIwi9o29oFT7YDnHYwZoctc3fOKtUw0Xmo= cloud.google.com/go/storage v1.14.0/go.mod h1:GrKmX003DSIwi9o29oFT7YDnHYwZoctc3fOKtUw0Xmo=
codeberg.org/vlbeaudoin/serpents v1.0.0 h1:3mN5QdVojmd3U0ISh/jgSslNOd7By0H3u9enRwgimkE=
codeberg.org/vlbeaudoin/serpents v1.0.0/go.mod h1:3bE/R0ToABwcUJtS1VcGEBa86K5FYhrZGAbFl2qL8kQ=
codeberg.org/vlbeaudoin/voki v1.3.1 h1:TxJj3qmOys0Pbq1dPKnOEXMXKqQLQqrBYd4QqiWWXcw= codeberg.org/vlbeaudoin/voki v1.3.1 h1:TxJj3qmOys0Pbq1dPKnOEXMXKqQLQqrBYd4QqiWWXcw=
codeberg.org/vlbeaudoin/voki v1.3.1/go.mod h1:5XTLx/KiW/OfiupF3o7PAAAU/UhsPdKSrVMmtHbmkPI= codeberg.org/vlbeaudoin/voki v1.3.1/go.mod h1:5XTLx/KiW/OfiupF3o7PAAAU/UhsPdKSrVMmtHbmkPI=
dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU= dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU=