Enable Semantic Token Refresh (#630)

* Enable Semantic Token Refresh Add ModuleChangeHooks to trigger semantic token refresh when specified parts of the schema are discovered after initialization. Fixes #344
hashicorp · Feb 9, 2022 · 9ac397b · 9ac397b
1 parent 0c14b7a
commit 9ac397b
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 0 deletions.
diff --git a/internal/langserver/handlers/hooks_module.go b/internal/langserver/handlers/hooks_module.go
@@ -3,7 +3,9 @@ package handlers
 import (
 	"context"
 	"fmt"
+	"log"
 
+	"github.com/creachadair/jrpc2"
 	"github.com/hashicorp/terraform-ls/internal/langserver/diagnostics"
 	"github.com/hashicorp/terraform-ls/internal/langserver/session"
 	"github.com/hashicorp/terraform-ls/internal/state"
@@ -169,3 +171,13 @@ func refreshCodeLens(ctx context.Context, clientRequester session.ClientCaller)
 		}
 	}
 }
+
+func refreshSemanticTokens(ctx context.Context, svrCtx context.Context, logger *log.Logger) state.ModuleChangeHook {
+	return func(_, newMod *state.Module) {
+		jrpcsvc := jrpc2.ServerFromContext(ctx)
+		_, err := jrpcsvc.Callback(svrCtx, "workspace/semanticTokens/refresh", nil)
+		if err != nil {
+			logger.Printf("Error refreshing %s: %s", newMod.Path, err)
+		}
+	}
+}
diff --git a/internal/langserver/handlers/service.go b/internal/langserver/handlers/service.go
@@ -446,6 +446,7 @@ func (svc *service) configureSessionDependencies(ctx context.Context, cfgOpts *s
 	svc.stateStore.Modules.ChangeHooks = state.ModuleChangeHooks{
 		updateDiagnostics(svc.sessCtx, svc.diagsNotifier),
 		sendModuleTelemetry(svc.sessCtx, svc.stateStore, svc.telemetry),
+		refreshSemanticTokens(ctx, svc.srvCtx, svc.logger),
 	}
 
 	cc, err := ilsp.ClientCapabilities(ctx)

diff --git a/internal/state/module.go b/internal/state/module.go
@@ -456,6 +456,11 @@ func (s *ModuleStore) UpdateModManifest(path string, manifest *datadir.ModuleMan
 		return err
 	}
 
+	txn.Defer(func() {
+		s.logger.Printf("Queuing refresh for %s", path)
+		go s.ChangeHooks.notifyModuleChange(nil, mod)
+	})
+
 	txn.Commit()
 	return nil
 }
@@ -475,6 +480,10 @@ func (s *ModuleStore) SetTerraformVersionState(path string, state op.OpState) er
 		return err
 	}
 
+	txn.Defer(func() {
+		go s.ChangeHooks.notifyModuleChange(nil, mod)
+	})
+
 	txn.Commit()
 	return nil
 }
@@ -556,6 +565,10 @@ func (s *ModuleStore) UpdateTerraformVersion(modPath string, tfVer *version.Vers
 		return err
 	}
 
+	txn.Defer(func() {
+		go s.ChangeHooks.notifyModuleChange(nil, mod)
+	})
+
 	txn.Commit()
 	return nil
 }