wip

internal/langserver/handlers/initialize.go

@@ -155,6 +155,8 @@ func (svc *service) Initialize(ctx context.Context, params lsp.InitializeParams)
 
 	serverCaps.Capabilities.SemanticTokensProvider = semanticTokensOpts
 
+	svc.jrpcSvr = jrpc2.ServerFromContext(ctx)
+
 	// set commandPrefix for session
 	lsctx.SetCommandPrefix(ctx, out.Options.CommandPrefix)
 	// apply prefix to executeCommand handler names

internal/langserver/handlers/service.go

@@ -53,6 +53,8 @@ type service struct {
 	decoder          *decoder.Decoder
 	stateStore       *state.StateStore
 
+	jrpcSvr *jrpc2.Server
+
 	additionalHandlers map[string]rpch.Func
 }
 
@@ -446,6 +448,13 @@ func (svc *service) configureSessionDependencies(ctx context.Context, cfgOpts *s
 	svc.stateStore.SetLogger(svc.logger)
 	svc.stateStore.Modules.ChangeHooks = state.ModuleChangeHooks{
 		sendModuleTelemetry(svc.sessCtx, svc.stateStore, svc.telemetry),
+		func(_, newMod *state.Module) {
+			svc.logger.Printf("Sending refresh notification for %s", newMod.Path)
+			_, err := svc.jrpcSvr.Callback(svc.srvCtx, "workspace/semanticTokens/refresh", nil)
+			if err != nil {
+				svc.logger.Printf("Error refreshing %s: %s", newMod.Path, err)
+			}
+		},
 	}
 
 	svc.modStore = svc.stateStore.Modules

internal/state/module.go

@@ -2,6 +2,7 @@ package state
 
 import (
 	"path/filepath"
+	"runtime"
 
 	"github.com/hashicorp/go-memdb"
 	"github.com/hashicorp/go-version"
@@ -448,6 +449,13 @@ func (s *ModuleStore) UpdateModManifest(path string, manifest *datadir.ModuleMan
 		return err
 	}
 
+	txn.Defer(func() {
+		for _, hook := range s.ChangeHooks {
+			s.logger.Printf("%s Sending refresh for %s", funcName(), path)
+			hook(nil, mod)
+		}
+	})
+
 	txn.Commit()
 	return nil
 }
@@ -467,6 +475,13 @@ func (s *ModuleStore) SetTerraformVersionState(path string, state op.OpState) er
 		return err
 	}
 
+	txn.Defer(func() {
+		for _, hook := range s.ChangeHooks {
+			s.logger.Printf("%s Sending refresh for %s", funcName(), path)
+			hook(nil, mod)
+		}
+	})
+
 	txn.Commit()
 	return nil
 }
@@ -548,6 +563,13 @@ func (s *ModuleStore) UpdateTerraformVersion(modPath string, tfVer *version.Vers
 		return err
 	}
 
+	txn.Defer(func() {
+		for _, hook := range s.ChangeHooks {
+			s.logger.Printf("%s Sending refresh for %s", funcName(), modPath)
+			hook(nil, mod)
+		}
+	})
+
 	txn.Commit()
 	return nil
 }
@@ -820,3 +842,8 @@ func (s *ModuleStore) UpdateReferenceOrigins(path string, origins reference.Orig
 	txn.Commit()
 	return nil
 }
+
+func funcName() string {
+	pc, _, _, _ := runtime.Caller(1)
+	return runtime.FuncForPC(pc).Name()
+}