fix(cli): Fix config during database and table completions

clevyr · Aug 19, 2023 · 73b3302 · 73b3302
1 parent cad3b03
commit 73b3302
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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