From 73b3302f28602c57008fab86898de77cb0869fae Mon Sep 17 00:00:00 2001 From: Gabe Cook Date: Fri, 18 Aug 2023 23:40:46 -0500 Subject: [PATCH] fix(cli): Fix config during database and table completions --- internal/config/flags/database.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/internal/config/flags/database.go b/internal/config/flags/database.go index 6b0e1da..0e782be 100644 --- a/internal/config/flags/database.go +++ b/internal/config/flags/database.go @@ -148,6 +148,10 @@ func ExcludeTableData(cmd *cobra.Command, p *[]string) { func listTables(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) { conf := config.Exec{DisableHeaders: true} + if err := viper.Unmarshal(&conf); err != nil { + return nil, cobra.ShellCompDirectiveError + } + err := util.DefaultSetup(cmd, &conf.Global) if err != nil { return nil, cobra.ShellCompDirectiveError @@ -158,6 +162,10 @@ func listTables(cmd *cobra.Command, args []string, toComplete string) ([]string, func listDatabases(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) { conf := config.Exec{DisableHeaders: true} + if err := viper.Unmarshal(&conf); err != nil { + return nil, cobra.ShellCompDirectiveError + } + err := util.DefaultSetup(cmd, &conf.Global) if err != nil { return nil, cobra.ShellCompDirectiveError