When new files are added to the sketch do a complete build to synchronize all files

Author: cmaglie

ls/builder.go

@@ -27,14 +27,14 @@ import (
 
 type SketchRebuilder struct {
 	ls      *INOLanguageServer
-	trigger chan bool
+	trigger chan chan<- bool
 	cancel  func()
 	mutex   sync.Mutex
 }
 
 func NewSketchBuilder(ls *INOLanguageServer) *SketchRebuilder {
 	res := &SketchRebuilder{
-		trigger: make(chan bool, 1),
+		trigger: make(chan chan<- bool, 1),
 		cancel:  func() {},
 		ls:      ls,
 	}
@@ -45,25 +45,33 @@ func NewSketchBuilder(ls *INOLanguageServer) *SketchRebuilder {
 	return res
 }
 
+func (ls *INOLanguageServer) triggerRebuildAndWait(logger jsonrpc.FunctionLogger) {
+	completed := make(chan bool)
+	ls.sketchRebuilder.TriggerRebuild(completed)
+	ls.writeUnlock(logger)
+	<-completed
+	ls.writeLock(logger, true)
+}
+
 func (ls *INOLanguageServer) triggerRebuild() {
-	ls.sketchRebuilder.TriggerRebuild()
+	ls.sketchRebuilder.TriggerRebuild(nil)
 }
 
-func (r *SketchRebuilder) TriggerRebuild() {
+func (r *SketchRebuilder) TriggerRebuild(completed chan<- bool) {
 	r.mutex.Lock()
 	defer r.mutex.Unlock()
 
 	r.cancel() // Stop possibly already running builds
 	select {
-	case r.trigger <- true:
+	case r.trigger <- completed:
 	default:
 	}
 }
 
 func (r *SketchRebuilder) rebuilderLoop() {
 	logger := NewLSPFunctionLogger(color.HiMagentaString, "SKETCH REBUILD: ")
 	for {
-		<-r.trigger
+		completed := <-r.trigger
 
 		for {
 			// Concede a 200ms delay to accumulate bursts of changes
@@ -90,6 +98,9 @@ func (r *SketchRebuilder) rebuilderLoop() {
 
 		cancel()
 		r.ls.progressHandler.End("arduinoLanguageServerRebuild", &lsp.WorkDoneProgressEnd{Message: "done"})
+		if completed != nil {
+			close(completed)
+		}
 	}
 }
 

ls/ls.go

@@ -961,10 +961,20 @@ func (ls *INOLanguageServer) TextDocumentDidOpenNotifFromIDE(logger jsonrpc.Func
 	ls.writeLock(logger, true)
 	defer ls.writeUnlock(logger)
 
-	ls.triggerRebuild()
+	ideTextDocItem := ideParam.TextDocument
+	clangURI, _, err := ls.ide2ClangDocumentURI(logger, ideTextDocItem.URI)
+	if err != nil {
+		logger.Logf("Error: %s", err)
+		return
+	}
+
+	if ls.ideURIIsPartOfTheSketch(ideTextDocItem.URI) {
+		if !clangURI.AsPath().Exist() {
+			ls.triggerRebuildAndWait(logger)
+		}
+	}
 
 	// Add the TextDocumentItem in the tracked files list
-	ideTextDocItem := ideParam.TextDocument
 	ls.trackedIdeDocs[ideTextDocItem.URI.AsPath().String()] = ideTextDocItem
 
 	// If we are tracking a .ino...
@@ -979,11 +989,6 @@ func (ls *INOLanguageServer) TextDocumentDidOpenNotifFromIDE(logger jsonrpc.Func
 		}
 	}
 
-	clangURI, _, err := ls.ide2ClangDocumentURI(logger, ideTextDocItem.URI)
-	if err != nil {
-		logger.Logf("Error: %s", err)
-		return
-	}
 	clangTextDocItem := lsp.TextDocumentItem{
 		URI: clangURI,
 	}