Merge pull request #63 from cmoog/initconfig

allow LSP client to set database connection config

Author: lighttiger2505

internal/handler/handler.go

@@ -30,6 +30,12 @@ type Server struct {
 	curDBName          string
 	curConnectionIndex int
 
+	// The initOptionDBConfig is an optional param
+	// sent by the client as part of the LSP InitializationOptions
+	// payload. If non-nil, the server will ignore all
+	// other configuration sources (workspace and user).
+	initOptionDBConfig *database.DBConfig
+
 	worker *database.Worker
 	files  map[string]*File
 }
@@ -159,6 +165,8 @@ func (s *Server) handleInitialize(ctx context.Context, conn *jsonrpc2.Conn, req
 		},
 	}
 
+	s.initOptionDBConfig = params.InitializationOptions.ConnectionConfig
+
 	// Initialize database database connection
 	// NOTE: If no connection is found at this point, it is possible that the connection settings are sent to workspace config, so don't make an error
 	messenger := lsp.NewLspMessenger(conn)
@@ -378,6 +386,11 @@ func (s *Server) newDBRepository(ctx context.Context) (database.DBRepository, er
 }
 
 func (s *Server) topConnection() *database.DBConfig {
+	// if the init config is set, ignore all other connection configs
+	if s.initOptionDBConfig != nil {
+		return s.initOptionDBConfig
+	}
+
 	cfg := s.getConfig()
 	if cfg == nil || len(cfg.Connections) == 0 {
 		return nil

internal/lsp/lsp.go

@@ -1,6 +1,9 @@
 package lsp
 
-import "github.com/lighttiger2505/sqls/internal/config"
+import (
+	"github.com/lighttiger2505/sqls/internal/config"
+	"github.com/lighttiger2505/sqls/internal/database"
+)
 
 // https://microsoft.github.io/language-server-protocol/specifications/specification-3-14/#initialize
 
@@ -14,6 +17,10 @@ type InitializeParams struct {
 }
 
 type InitializeOptions struct {
+	// Allow the LSP client to choose a database configuration.
+	// If set, the LSP server will ignore all other configuration
+	// sources, including the workspace and user configuration files.
+	ConnectionConfig *database.DBConfig `json:"connectionConfig,omitempty"`
 }
 
 type ClientCapabilities struct {