Representing collected and deduplicated definitions and usages

structure Lean.Server.Reference : Type

Global reference. Used by the language server to figure out which identifiers refer to which other identifiers across the whole project.

• ident : Lean.Lsp.RefIdent

  Identifier of this reference.

• aliases : Array Lean.Lsp.RefIdent

  Identifiers that are logically identical to this reference.

• range : Lean.Lsp.Range

  Range where this reference occurs.

• stx : Lean.Syntax

  Syntax of this reference.

• ci : Lean.Elab.ContextInfo

  ContextInfo at the point of elaboration of this reference.

• info : Lean.Elab.Info

  Additional InfoTree information for this reference.

• isBinder : Bool

  Whether this reference declares ident.

structure Lean.Server.RefInfo : Type

Definition and usages of an identifier within a single module.

• definition : Option Lean.Server.Reference

  Definition Reference of the identifier. Is equal to none if e.g. the definition is outside of the module where this RefInfo is used.

• usages : Array Lean.Server.Reference

  All usage References of the identifier in a single module.

def Lean.Server.RefInfo.empty : Lean.Server.RefInfo

No definition, no usages.

Equations

• Lean.Server.RefInfo.empty = { definition := none, usages := #[] }

def Lean.Server.RefInfo.addRef (i : Lean.Server.RefInfo) (ref : Lean.Server.Reference) : Lean.Server.RefInfo

Adds ref to i. If i has no definition and ref is a declaration, it becomes the definition. If i already has a definition and ref is also a declaration, it is not added to i. Otherwise, ref is added to i.usages.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.RefInfo.toLspRefInfo (i : Lean.Server.RefInfo) : BaseIO Lean.Lsp.RefInfo

Converts i to a JSON-serializable Lsp.RefInfo.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.ModuleRefs : Type

All references from within a module for all identifiers used in a single module.

Equations

• Lean.Server.ModuleRefs = Std.HashMap Lean.Lsp.RefIdent Lean.Server.RefInfo

def Lean.Server.ModuleRefs.addRef (self : Lean.Server.ModuleRefs) (ref : Lean.Server.Reference) : Lean.Server.ModuleRefs

Adds ref to the RefInfo corresponding to ref.ident in self. See RefInfo.addRef.

Equations

• self.addRef ref = Std.HashMap.insert self ref.ident ((Std.HashMap.getD self ref.ident Lean.Server.RefInfo.empty).addRef ref)

def Lean.Server.ModuleRefs.toLspModuleRefs (refs : Lean.Server.ModuleRefs) : BaseIO Lean.Lsp.ModuleRefs

Converts refs to a JSON-serializable Lsp.ModuleRefs.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Lsp.RefInfo.empty : Lean.Lsp.RefInfo

No definition, no usages

Equations

• Lean.Lsp.RefInfo.empty = { definition? := none, usages := #[] }

def Lean.Lsp.RefInfo.merge (a : Lean.Lsp.RefInfo) (b : Lean.Lsp.RefInfo) : Lean.Lsp.RefInfo

Combines the usages of a and b and prefers the definition? of b over that of a.

Equations

• a.merge b = { definition? := b.definition?.orElse fun (x : Unit) => a.definition?, usages := a.usages.append b.usages }

def Lean.Lsp.RefInfo.findReferenceLocation? (self : Lean.Lsp.RefInfo) (pos : Lean.Lsp.Position) (includeStop : optParam Bool false) : Option Lean.Lsp.RefInfo.Location

Finds the first definition or usage in self where the RefInfo.Location.range contains the given pos. The includeStop parameter can be used to toggle between closed-interval and half-open-interval behavior for the range. Closed-interval behavior matches the expectation of VSCode when selecting an identifier at a cursor position (see #767).

Equations

• One or more equations did not get rendered due to their size.

def Lean.Lsp.RefInfo.findReferenceLocation?.contains (includeStop : optParam Bool false) (range : Lean.Lsp.Range) (pos : Lean.Lsp.Position) : Bool

Equations

• Lean.Lsp.RefInfo.findReferenceLocation?.contains includeStop range pos = (decide (range.start ≤ pos) && if includeStop = true then decide (pos ≤ range.end) else decide (pos < range.end))

def Lean.Lsp.RefInfo.contains (self : Lean.Lsp.RefInfo) (pos : Lean.Lsp.Position) (includeStop : optParam Bool false) : Bool

Checks whether any of the ranges in self.definition? or self.usages contains pos.

Equations

• self.contains pos includeStop = Id.run (self.findReferenceLocation? pos includeStop).isSome

def Lean.Lsp.ModuleRefs.findAt (self : Lean.Lsp.ModuleRefs) (pos : Lean.Lsp.Position) (includeStop : optParam Bool false) : Array Lean.Lsp.RefIdent

Find all identifiers in self with a reference in this module that contains pos in its range.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Lsp.ModuleRefs.findRange? (self : Lean.Lsp.ModuleRefs) (pos : Lean.Lsp.Position) (includeStop : optParam Bool false) : Option Lean.Lsp.Range

Finds the first range in self that contains pos.

Equations

• One or more equations did not get rendered due to their size.

structure Lean.Server.Ilean : Type

Content of individual .ilean files

• version : Nat

  Version number of the ilean format.

• module : Lake.Name

  Name of the module that this ilean data has been collected for.

• references : Lean.Lsp.ModuleRefs

  All references of this module.

instance Lean.Server.instFromJsonIlean : Lean.FromJson Lean.Server.Ilean

Equations

• Lean.Server.instFromJsonIlean = { fromJson? := Lean.Server.fromJsonIlean✝ }

instance Lean.Server.instToJsonIlean : Lean.ToJson Lean.Server.Ilean

Equations

• Lean.Server.instToJsonIlean = { toJson := Lean.Server.toJsonIlean✝ }

def Lean.Server.Ilean.load (path : Lake.FilePath) : IO Lean.Server.Ilean

Reads and parses the .ilean file at path.

Equations

• One or more equations did not get rendered due to their size.

Collecting and deduplicating definitions and usages

def Lean.Server.getModuleContainingDecl? (env : Lean.Environment) (declName : Lake.Name) : Option Lake.Name

Gets the name of the module that contains declName.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.identOf (ci : Lean.Elab.ContextInfo) (i : Lean.Elab.Info) : Option (Lean.Lsp.RefIdent × Bool)

Determines the RefIdent for the Info i of an identifier in module and whether it is a declaration.

Equations

• One or more equations did not get rendered due to their size.
• Lean.Server.identOf ci i = none

def Lean.Server.findReferences (text : Lean.FileMap) (trees : Array Lean.Elab.InfoTree) : Array Lean.Server.Reference

Finds all references in trees.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.combineIdents (trees : Array Lean.Elab.InfoTree) (refs : Array Lean.Server.Reference) : Array Lean.Server.Reference

There are several different identifiers that should be considered equal for the purpose of finding all references of an identifier:

• FVarIds of a function parameter in the function's signature and body
• Chains of helper definitions like those created for do-reassignment x := e
• Overlapping definitions like those defined by where declarations that define both an FVar (for local usage) and a constant (for non-local usage)
• Identifiers connected by FVarAliasInfo such as variables before and after match generalization

In the first three cases that are not explicitly denoted as aliases with an FVarAliasInfo, the corresponding References have the exact same range. This function finds all definitions that have the exact same range as another definition or usage and collapses them into a single identifier. It also collapses identifiers connected by an FVarAliasInfo. When collapsing identifiers, it prefers using a RefIdent.const name over a RefIdent.fvar id for all identifiers that are being collapsed into one.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.combineIdents.useConstRepresentatives (idMap : Std.HashMap Lean.Lsp.RefIdent Lean.Lsp.RefIdent) : Std.HashMap Lean.Lsp.RefIdent Lean.Lsp.RefIdent

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.combineIdents.findCanonicalRepresentative (idMap : Std.HashMap Lean.Lsp.RefIdent Lean.Lsp.RefIdent) (id : Lean.Lsp.RefIdent) : Lean.Lsp.RefIdent

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.combineIdents.buildIdMap (trees : Array Lean.Elab.InfoTree) (refs : Array Lean.Server.Reference) (posMap : Std.HashMap Lean.Lsp.Range Lean.Lsp.RefIdent) : Std.HashMap Lean.Lsp.RefIdent Lean.Lsp.RefIdent

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.combineIdents.insertIdMap (id : Lean.Lsp.RefIdent) (baseId : Lean.Lsp.RefIdent) : StateT (Std.HashMap Lean.Lsp.RefIdent Lean.Lsp.RefIdent) Id Unit

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.dedupReferences (refs : Array Lean.Server.Reference) (allowSimultaneousBinderUse : optParam Bool false) : Array Lean.Server.Reference

Groups refs by identifier and range s.t. references with the same identifier and range are added to the aliases of the representative of the group. Yields to separate groups for declaration and usages if allowSimultaneousBinderUse is set.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.findModuleRefs (text : Lean.FileMap) (trees : Array Lean.Elab.InfoTree) (localVars : optParam Bool true) (allowSimultaneousBinderUse : optParam Bool false) : Lean.Server.ModuleRefs

Finds all references in trees and deduplicates the result. See dedupReferences and combineIdents.

Equations

• One or more equations did not get rendered due to their size.

Collecting and maintaining reference info from different sources

structure Lean.Server.References : Type

References from ilean files and current ilean information from file workers.

• ileans : Std.HashMap Lake.Name (Lake.FilePath × Lean.Lsp.ModuleRefs)

  References loaded from ilean files

• workers : Std.HashMap Lake.Name (Nat × Lean.Lsp.ModuleRefs)

  References from workers, overriding the corresponding ilean files

def Lean.Server.References.empty : Lean.Server.References

No ilean files, no information from workers.

Equations

• Lean.Server.References.empty = { ileans := Std.HashMap.empty, workers := Std.HashMap.empty }

def Lean.Server.References.addIlean (self : Lean.Server.References) (path : Lake.FilePath) (ilean : Lean.Server.Ilean) : Lean.Server.References

Adds the contents of an ilean file ilean at path to self.

Equations

• self.addIlean path ilean = { ileans := self.ileans.insert ilean.module (path, ilean.references), workers := self.workers }

def Lean.Server.References.removeIlean (self : Lean.Server.References) (path : Lake.FilePath) : Lean.Server.References

Removes the ilean file data at path from self.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.References.updateWorkerRefs (self : Lean.Server.References) (name : Lake.Name) (version : Nat) (refs : Lean.Lsp.ModuleRefs) : Lean.Server.References

Updates the worker references in self with the refs of the worker managing the module name. Replaces the current references with refs if version is newer than the current version managed in refs and otherwise merges the reference data if version is equal to the current version.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.References.finalizeWorkerRefs (self : Lean.Server.References) (name : Lake.Name) (version : Nat) (refs : Lean.Lsp.ModuleRefs) : Lean.Server.References

Replaces the worker references in self with the refs of the worker managing the module name if version is newer than the current version managed in refs.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.References.removeWorkerRefs (self : Lean.Server.References) (name : Lake.Name) : Lean.Server.References

Erases all worker references in self for the worker managing name.

Equations

• self.removeWorkerRefs name = { ileans := self.ileans, workers := self.workers.erase name }

def Lean.Server.References.allRefs (self : Lean.Server.References) : Std.HashMap Lake.Name Lean.Lsp.ModuleRefs

Yields a map from all modules to all of their references.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.References.allRefsFor (self : Lean.Server.References) (srcSearchPath : Lean.SearchPath) (ident : Lean.Lsp.RefIdent) : IO (Array (Lean.Lsp.DocumentUri × Lean.Lsp.RefInfo))

Yields all references in self for ident, as well as the DocumentUri that each reference occurs in.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.References.findAt (self : Lean.Server.References) (module : Lake.Name) (pos : Lean.Lsp.Position) (includeStop : optParam Bool false) : Array Lean.Lsp.RefIdent

Yields all references in module at pos.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.References.findRange? (self : Lean.Server.References) (module : Lake.Name) (pos : Lean.Lsp.Position) (includeStop : optParam Bool false) : Option Lean.Lsp.Range

Yields the first reference in module at pos.

Equations

• self.findRange? module pos includeStop = do let refs ← self.allRefs[module]? refs.findRange? pos includeStop

structure Lean.Server.References.DocumentRefInfo : Type

Location and parent declaration of a reference.

• location : Lean.Lsp.Location

  Location of the reference.

• parentInfo? : Option Lean.Lsp.RefInfo.ParentDecl

  Parent declaration of the reference.

def Lean.Server.References.referringTo (self : Lean.Server.References) (srcSearchPath : Lean.SearchPath) (ident : Lean.Lsp.RefIdent) (includeDefinition : optParam Bool true) : IO (Array Lean.Server.References.DocumentRefInfo)

Yields locations and parent declaration for all references referring to ident.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.References.definitionOf? (self : Lean.Server.References) (ident : Lean.Lsp.RefIdent) (srcSearchPath : Lean.SearchPath) : IO (Option Lean.Server.References.DocumentRefInfo)

Yields the definition location of ident.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Server.References.definitionsMatching {α : Type} (self : Lean.Server.References) (srcSearchPath : Lean.SearchPath) (filter : Lake.Name → Option α) (maxAmount? : optParam (Option Nat) none) : IO (Array (α × Lean.Lsp.Location))

Yields all definitions matching the given filter.

Equations

• One or more equations did not get rendered due to their size.