add basic auto completion using postgres keyword

Signed-off-by: Balaji J <j.balaji2468@gmail.com>

Author: balajz

internal/app/app.go

@@ -30,7 +30,7 @@ type PgxCLI struct {
 	logger  *slog.Logger
 }
 
-func NewPgxCLI(config *config.Config, printer cliio.Printer, logger *slog.Logger) (*PgxCLI, error) {
+func New(config *config.Config, printer cliio.Printer, logger *slog.Logger) (*PgxCLI, error) {
 	history, entries := newHistory(config.Main.HistoryFile, logger)
 	reader, err := NewPgxReader()
 	if err != nil {
@@ -126,6 +126,10 @@ func (p *PgxCLI) Start(ctx context.Context, client *database.Client) {
 	}
 }
 
+func (p *PgxCLI) SetAutocompleter(keywords []string) {
+	p.prompt.SetAutocompleter(keywords)
+}
+
 func (p *PgxCLI) Close() error {
 	p.History.saveHistory(p.prompt.History())
 	return nil

internal/app/reader.go

@@ -24,6 +24,7 @@ var chromaFormatter = detectTerminalColorProfile()
 type Reader interface {
 	Read(prefix string, ctx context.Context) (string, error)
 	History() []prompt.HistoryCommand
+	SetAutocompleter(keywords []string)
 }
 
 type PgxReader struct {
@@ -48,6 +49,14 @@ func (r *PgxReader) Read(prefix string, ctx context.Context) (string, error) {
 	return text, nil
 }
 
+func (r *PgxReader) SetAutocompleter(keywords []string) {
+	suggestions := make([]prompt.Suggestion, len(keywords))
+	for i, kw := range keywords {
+		suggestions[i] = prompt.Suggestion{Value: kw}
+	}
+	r.prompt.SetAutoCompleterContextual(prompt.AutoCompleteSimple(suggestions, true))
+}
+
 func (r *PgxReader) History() []prompt.HistoryCommand {
 	return r.prompt.History()
 }

internal/cli/root.go

@@ -12,6 +12,7 @@ import (
 	"github.com/balaji01-4d/pgxcli/internal/config"
 	"github.com/balaji01-4d/pgxcli/internal/database"
 	"github.com/balaji01-4d/pgxcli/internal/logger"
+	"github.com/balaji01-4d/pgxcli/internal/parser"
 	"github.com/balaji01-4d/pgxcli/internal/ui"
 	"github.com/spf13/cobra"
 )
@@ -41,6 +42,8 @@ func NewRootCmd(ctx context.Context, cliCtx *CliContext) *cobra.Command {
 		finalPassword string
 	)
 
+	var pgKws []string
+
 	rootCmd := &cobra.Command{
 		Use:     "pgxcli [DBNAME] [USERNAME]",
 		Short:   "Interactive PostgreSQL command-line client for querying and managing databases.",
@@ -64,6 +67,8 @@ func NewRootCmd(ctx context.Context, cliCtx *CliContext) *cobra.Command {
 
 			cliCtx.Logger = logger
 
+			pgKws = parser.LoadPgKeywords()
+
 			return nil
 		},
 
@@ -190,11 +195,12 @@ func NewRootCmd(ctx context.Context, cliCtx *CliContext) *cobra.Command {
 				return err
 			}
 
-			app, err := app.NewPgxCLI(cliCtx.config, cliCtx.Printer, cliCtx.Logger.Logger)
+			app, err := app.New(cliCtx.config, cliCtx.Printer, cliCtx.Logger.Logger)
 			if err != nil {
 				cliCtx.Logger.Error("Failed to initialize app", "error", err)
 				return err
 			}
+			app.SetAutocompleter(pgKws)
 			cliCtx.App = app
 			return nil
 		},