add initial lsp support

Author: kujtimiihoxha

README.md

@@ -91,6 +91,12 @@ go build -o termai
 ./termai
 ```
 
+## Acknowledgments
+
+TermAI builds upon the work of several open source projects and developers:
+
+- [@isaacphi](https://github.com/isaacphi) - LSP client implementation
+
 ## License
 
 [License information coming soon]

cmd/lsp/main.go

@@ -0,0 +1,4 @@
+package main
+
+func main() {
+}

cmd/root.go

@@ -36,6 +36,7 @@ var rootCmd = &cobra.Command{
 ctx := context.Background()
 
 app := app.New(ctx, conn)
+defer app.Close()
 app.Logger.Info("Starting termai...")
 zone.NewGlobal()
 tui := tea.NewProgram(

go.mod

@@ -1,6 +1,8 @@
 module github.com/kujtimiihoxha/termai
 
-go 1.23.5
+go 1.24.0
+
+toolchain go1.24.2
 
 require (
 github.com/anthropics/anthropic-sdk-go v0.2.0-beta.2
@@ -11,6 +13,7 @@ require (
 github.com/charmbracelet/glamour v0.9.1
 github.com/charmbracelet/huh v0.6.0
 github.com/charmbracelet/lipgloss v1.1.0
+github.com/fsnotify/fsnotify v1.8.0
 github.com/go-logfmt/logfmt v0.6.0
 github.com/golang-migrate/migrate/v4 v4.18.2
 github.com/google/generative-ai-go v0.19.0
@@ -53,7 +56,6 @@ require (
 github.com/dustin/go-humanize v1.0.1 // indirect
 github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
 github.com/felixge/httpsnoop v1.0.4 // indirect
-github.com/fsnotify/fsnotify v1.8.0 // indirect
 github.com/go-logr/logr v1.4.2 // indirect
 github.com/go-logr/stdr v1.2.2 // indirect
 github.com/go-viper/mapstructure/v2 v2.2.1 // indirect

internal/app/services.go

@@ -7,6 +7,8 @@ import (
 "github.com/kujtimiihoxha/termai/internal/config"
 "github.com/kujtimiihoxha/termai/internal/db"
 "github.com/kujtimiihoxha/termai/internal/logging"
+"github.com/kujtimiihoxha/termai/internal/lsp"
+"github.com/kujtimiihoxha/termai/internal/lsp/watcher"
 "github.com/kujtimiihoxha/termai/internal/message"
 "github.com/kujtimiihoxha/termai/internal/permission"
 "github.com/kujtimiihoxha/termai/internal/session"
@@ -19,23 +21,59 @@ type App struct {
 Messages message.Service
 Permissions permission.Service
 
+LSPClients map[string]*lsp.Client
+
 Logger logging.Interface
+
+ceanups []func()
 }
 
 func New(ctx context.Context, conn *sql.DB) *App {
+cfg := config.Get()
 q := db.New(conn)
 log := logging.NewLogger(logging.Options{
-Level: config.Get().Log.Level,
+Level: cfg.Log.Level,
 })
 sessions := session.NewService(ctx, q)
 messages := message.NewService(ctx, q)
 
-return &App{
+app := &App{
 Context: ctx,
 Sessions: sessions,
 Messages: messages,
 Permissions: permission.Default,
 Logger: log,
+LSPClients: make(map[string]*lsp.Client),
+}
+
+for name, client := range cfg.LSP {
+lspClient, err := lsp.NewClient(client.Command, client.Args...)
+app.ceanups = append(app.ceanups, func() {
+lspClient.Close()
+})
+workspaceWatcher := watcher.NewWorkspaceWatcher(lspClient)
+if err != nil {
+log.Error("Failed to create LSP client for", name, err)
+continue
+}
+
+_, err = lspClient.InitializeLSPClient(ctx, config.WorkingDirectory())
+if err != nil {
+log.Error("Initialize failed", "error", err)
+continue
+}
+go workspaceWatcher.WatchWorkspace(ctx, config.WorkingDirectory())
+app.LSPClients[name] = lspClient
 }
+return app
 }
 
+func (a *App) Close() {
+for _, cleanup := range a.ceanups {
+cleanup()
+}
+for _, client := range a.LSPClients {
+client.Close()
+}
+a.Logger.Info("App closed")
+}

internal/config/config.go

@@ -49,12 +49,21 @@ type Log struct {
 Level string `json:"level"`
 }
 
+type LSPConfig struct {
+Disabled bool `json:"enabled"`
+Command string `json:"command"`
+Args []string `json:"args"`
+Options any `json:"options"`
+}
+
 type Config struct {
 Data *Data `json:"data,omitempty"`
 Log *Log `json:"log,omitempty"`
 MCPServers map[string]MCPServer `json:"mcpServers,omitempty"`
 Providers map[models.ModelProvider]Provider `json:"providers,omitempty"`
 
+LSP map[string]LSPConfig `json:"lsp,omitempty"`
+
 Model *Model `json:"model,omitempty"`
 }
 

internal/db/messages.sql.go

@@ -23,11 +23,7 @@ CREATE TABLE IF NOT EXISTS messages (
  id TEXT PRIMARY KEY,
  session_id TEXT NOT NULL,
  role TEXT NOT NULL,
- content TEXT NOT NULL,
- thinking Text NOT NULL DEFAULT '',
- finished BOOLEAN NOT NULL DEFAULT 0,
- tool_calls TEXT,
- tool_results TEXT,
+ parts TEXT NOT NULL default '[]',
  created_at INTEGER NOT NULL, -- Unix timestamp in milliseconds
  updated_at INTEGER NOT NULL, -- Unix timestamp in milliseconds
  FOREIGN KEY (session_id) REFERENCES sessions (id) ON DELETE CASCADE

internal/db/migrations/000001_initial.up.sql

@@ -14,25 +14,18 @@ INSERT INTO messages (
  id,
  session_id,
  role,
- finished,
- content,
- tool_calls,
- tool_results,
+ parts,
  created_at,
  updated_at
 ) VALUES (
- ?, ?, ?, ?, ?, ?, ?, strftime('%s', 'now'), strftime('%s', 'now')
+ ?, ?, ?, ?, strftime('%s', 'now'), strftime('%s', 'now')
 )
 RETURNING *;
 
 -- name: UpdateMessage :exec
 UPDATE messages
 SET
- content = ?,
- thinking = ?,
- tool_calls = ?,
- tool_results = ?,
- finished = ?,
+ parts = ?,
  updated_at = strftime('%s', 'now')
 WHERE id = ?;