feat: add interactive AI TUI mode (xsql-ai & xsql ai)

.github/workflows/ci.yml

@@ -229,16 +229,24 @@ jobs:
         working-directory: webui
 
       - name: Build Linux
-        run: GOOS=linux GOARCH=amd64 go build -ldflags "-s -w" -o xsql-linux-amd64 ./cmd/xsql
+        run: |
+          GOOS=linux GOARCH=amd64 go build -ldflags "-s -w" -o xsql-linux-amd64 ./cmd/xsql
+          GOOS=linux GOARCH=amd64 go build -ldflags "-s -w" -o xsql-ai-linux-amd64 ./cmd/xsql-ai
 
       - name: Build Windows
-        run: GOOS=windows GOARCH=amd64 go build -ldflags "-s -w" -o xsql-windows-amd64.exe ./cmd/xsql
+        run: |
+          GOOS=windows GOARCH=amd64 go build -ldflags "-s -w" -o xsql-windows-amd64.exe ./cmd/xsql
+          GOOS=windows GOARCH=amd64 go build -ldflags "-s -w" -o xsql-ai-windows-amd64.exe ./cmd/xsql-ai
 
       - name: Build macOS (amd64)
-        run: GOOS=darwin GOARCH=amd64 go build -ldflags "-s -w" -o xsql-darwin-amd64 ./cmd/xsql
+        run: |
+          GOOS=darwin GOARCH=amd64 go build -ldflags "-s -w" -o xsql-darwin-amd64 ./cmd/xsql
+          GOOS=darwin GOARCH=amd64 go build -ldflags "-s -w" -o xsql-ai-darwin-amd64 ./cmd/xsql-ai
 
       - name: Build macOS (arm64)
-        run: GOOS=darwin GOARCH=arm64 go build -ldflags "-s -w" -o xsql-darwin-arm64 ./cmd/xsql
+        run: |
+          GOOS=darwin GOARCH=arm64 go build -ldflags "-s -w" -o xsql-darwin-arm64 ./cmd/xsql
+          GOOS=darwin GOARCH=arm64 go build -ldflags "-s -w" -o xsql-ai-darwin-arm64 ./cmd/xsql-ai
 
       - name: Upload artifacts
         uses: actions/upload-artifact@ea165f8d65b6e75b540449e92b4886f43607fa02 # v4.6.2

.gitignore

@@ -11,3 +11,4 @@ coverage*.txt
 coverage*.html
 coverage.out
 stats_coverage.out
+bin/

.golangci.yml

@@ -1,23 +1,26 @@
-version: "2"
-
+version: "2"
+
 run:
   timeout: 5m
   tests: false
 
 linters:
   default: standard
   settings:
-    errcheck:
-      exclude-functions:
-        - (*database/sql.Rows).Close
-        - (*database/sql.DB).Close
-        - (*github.com/zx06/xsql/internal/ssh.Client).Close
-        - (io.Closer).Close
-
-formatters:
-  enable:
-    - goimports
-  settings:
-    goimports:
-      local-prefixes:
-        - github.com/zx06/xsql
+    errcheck:
+      exclude-functions:
+        - (*database/sql.Rows).Close
+        - (*database/sql.DB).Close
+        - (*github.com/zx06/xsql/internal/ssh.Client).Close
+        - (io.Closer).Close
+        - (*os.File).Close
+        - (*net/http.Response.Body).Close
+        - io.Copy
+
+formatters:
+  enable:
+    - goimports
+  settings:
+    goimports:
+      local-prefixes:
+        - github.com/zx06/xsql

.goreleaser.yaml

@@ -1,86 +1,105 @@
-version: 2
-
-project_name: xsql
-
+version: 2
+
+project_name: xsql
+
 before:
   hooks:
     - npm --prefix webui ci
     - npm --prefix webui run build
     - go mod tidy
-
-builds:
-  - id: xsql
-    main: ./cmd/xsql
-    binary: xsql
-    env:
-      - CGO_ENABLED=0
-    goos:
-      - linux
-      - darwin
-      - windows
-    goarch:
-      - amd64
-      - arm64
-    ldflags:
-      - -s -w
-      - -X main.version={{.Version}}
-      - -X main.commit={{.ShortCommit}}
-      - -X main.date={{.Date}}
-
-archives:
-  - id: default
-    name_template: "{{ .ProjectName }}_{{ .Version }}_{{ .Os }}_{{ .Arch }}"
-    format_overrides:
-      - goos: windows
-        format: zip
-    files:
-      - README.md
-      - LICENSE*
-      - docs/*
-
-checksum:
-  name_template: "checksums.txt"
-
-snapshot:
-  version_template: "{{ incpatch .Version }}-next"
-
-changelog:
-  sort: asc
-  filters:
-    exclude:
-      - "^docs:"
-      - "^test:"
-      - "^chore:"
-      - "Merge pull request"
-      - "Merge branch"
-
-release:
-  github:
-    owner: zx06
-    name: xsql
-  draft: false
-  prerelease: auto
-  name_template: "v{{.Version}}"
-
-brews:
-  - repository:
-      owner: zx06
-      name: homebrew-tap
-      token: "{{ .Env.HOMEBREW_TAP_TOKEN }}"
-    directory: Formula
-    homepage: "https://github.com/zx06/xsql"
-    description: "AI-first cross-database CLI tool with SSH proxy support"
-    license: "MIT"
-    install: |
-      bin.install "xsql"
-    test: |
-      system "#{bin}/xsql", "version"
-
-scoops:
-  - repository:
-      owner: zx06
-      name: scoop-bucket
-      token: "{{ .Env.SCOOP_BUCKET_TOKEN }}"
-    homepage: "https://github.com/zx06/xsql"
-    description: "AI-first cross-database CLI tool with SSH proxy support"
-    license: "MIT"
+
+builds:
+  - id: xsql
+    main: ./cmd/xsql
+    binary: xsql
+    env:
+      - CGO_ENABLED=0
+    goos:
+      - linux
+      - darwin
+      - windows
+    goarch:
+      - amd64
+      - arm64
+    ldflags:
+      - -s -w
+      - -X main.version={{.Version}}
+      - -X main.commit={{.ShortCommit}}
+      - -X main.date={{.Date}}
+
+  - id: xsql-ai
+    main: ./cmd/xsql-ai
+    binary: xsql-ai
+    env:
+      - CGO_ENABLED=0
+    goos:
+      - linux
+      - darwin
+      - windows
+    goarch:
+      - amd64
+      - arm64
+    ldflags:
+      - -s -w
+      - -X main.version={{.Version}}
+      - -X main.commit={{.ShortCommit}}
+      - -X main.date={{.Date}}
+
+archives:
+  - id: default
+    name_template: "{{ .ProjectName }}_{{ .Version }}_{{ .Os }}_{{ .Arch }}"
+    format_overrides:
+      - goos: windows
+        format: zip
+    files:
+      - README.md
+      - LICENSE*
+      - docs/*
+
+checksum:
+  name_template: "checksums.txt"
+
+snapshot:
+  version_template: "{{ incpatch .Version }}-next"
+
+changelog:
+  sort: asc
+  filters:
+    exclude:
+      - "^docs:"
+      - "^test:"
+      - "^chore:"
+      - "Merge pull request"
+      - "Merge branch"
+
+release:
+  github:
+    owner: zx06
+    name: xsql
+  draft: false
+  prerelease: auto
+  name_template: "v{{.Version}}"
+
+brews:
+  - repository:
+      owner: zx06
+      name: homebrew-tap
+      token: "{{ .Env.HOMEBREW_TAP_TOKEN }}"
+    directory: Formula
+    homepage: "https://github.com/zx06/xsql"
+    description: "AI-first cross-database CLI tool with SSH proxy support"
+    license: "MIT"
+    install: |
+      bin.install "xsql"
+      bin.install "xsql-ai"
+    test: |
+      system "#{bin}/xsql", "version"
+
+scoops:
+  - repository:
+      owner: zx06
+      name: scoop-bucket
+      token: "{{ .Env.SCOOP_BUCKET_TOKEN }}"
+    homepage: "https://github.com/zx06/xsql"
+    description: "AI-first cross-database CLI tool with SSH proxy support"
+    license: "MIT"

cmd/xsql-ai/main.go

@@ -0,0 +1,107 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"strings"
+
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/spf13/cobra"
+
+	"github.com/zx06/xsql/internal/ai"
+	"github.com/zx06/xsql/internal/config"
+	_ "github.com/zx06/xsql/internal/db/mysql"
+	_ "github.com/zx06/xsql/internal/db/pg"
+	"github.com/zx06/xsql/internal/secret"
+	"github.com/zx06/xsql/internal/tui"
+)
+
+type AIFlags struct {
+	ConfigPath       string
+	Profile          string
+	Model            string
+	BaseURL          string
+	APIKey           string
+	UnsafeAllowWrite bool
+	Prompt           string
+}
+
+var newProgramFunc = func(model tea.Model) *tea.Program {
+	return tea.NewProgram(model, tea.WithAltScreen())
+}
+
+func newRootCmd() *cobra.Command {
+	flags := &AIFlags{}
+
+	rootCmd := &cobra.Command{
+		Use:   "xsql-ai [PROMPT]",
+		Short: "xsql AI interactive database query tool (TUI)",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if len(args) > 0 {
+				flags.Prompt = strings.Join(args, " ")
+			}
+			return runAI(cmd, flags)
+		},
+	}
+
+	rootCmd.Flags().StringVar(&flags.ConfigPath, "config", "", "Config file path (YAML)")
+	rootCmd.Flags().StringVarP(&flags.Profile, "profile", "p", "", "Profile name (default: 'default')")
+	rootCmd.Flags().StringVar(&flags.Model, "model", "", "AI model name (default: gpt-4o)")
+	rootCmd.Flags().StringVar(&flags.BaseURL, "base-url", "", "AI service base URL")
+	rootCmd.Flags().StringVar(&flags.APIKey, "api-key", "", "AI service API key")
+	rootCmd.Flags().BoolVar(&flags.UnsafeAllowWrite, "unsafe-allow-write", false, "Allow write operations (bypasses read-only protection)")
+	rootCmd.Flags().StringVar(&flags.Prompt, "prompt", "", "Initial prompt for AI query")
+
+	return rootCmd
+}
+
+func main() {
+	rootCmd := newRootCmd()
+	if err := rootCmd.Execute(); err != nil {
+		os.Exit(1)
+	}
+}
+
+func runAI(cmd *cobra.Command, flags *AIFlags) error {
+	opts := config.Options{
+		ConfigPath:      flags.ConfigPath,
+		CLIProfile:      flags.Profile,
+		CLIProfileSet:   flags.Profile != "",
+		CLIAIModel:      flags.Model,
+		CLIAIModelSet:   cmd.Flags().Changed("model"),
+		CLIAIBaseURL:    flags.BaseURL,
+		CLIAIBaseURLSet: cmd.Flags().Changed("base-url"),
+		CLIAIAPIKey:     flags.APIKey,
+		CLIAIAPIKeySet:  cmd.Flags().Changed("api-key"),
+	}
+
+	resolved, xe := config.Resolve(opts)
+	if xe != nil {
+		return fmt.Errorf("config error [%s]: %s", xe.Code, xe.Message)
+	}
+	if resolved.ProfileName == "" || resolved.Profile.DB == "" {
+		return fmt.Errorf("config error [XSQL_CFG_INVALID]: no profile specified and no 'default' profile found in config")
+	}
+
+	// Resolve API key if keyring reference or plaintext
+	apiKey := resolved.AI.APIKey
+	if secret.IsKeyringRef(apiKey) {
+		resolvedKey, xe := secret.Resolve(apiKey, secret.Options{AllowPlaintext: true})
+		if xe == nil {
+			apiKey = resolvedKey
+		}
+	}
+	resolved.AI.APIKey = apiKey
+
+	aiClient := ai.NewClient(resolved.AI, nil)
+	aiService := ai.NewService(resolved.AI, aiClient)
+
+	model := tui.NewModel(opts, resolved, aiService, flags.Prompt, flags.UnsafeAllowWrite)
+
+	p := newProgramFunc(model)
+	if _, err := p.Run(); err != nil {
+		return fmt.Errorf("error running TUI: %w", err)
+	}
+
+	return nil
+}