-
Notifications
You must be signed in to change notification settings - Fork 131
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
refactor: Reflect upstream
hcl-lang changes (#689)
* refactor: Reflect upstream hcl-lang changes * avoid race condition
- Loading branch information
1 parent
9c4b998
commit 6fb9f47
Showing
33 changed files
with
579 additions
and
750 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,115 @@ | ||
| package codelens | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
|
|
||
| "github.com/hashicorp/hcl-lang/decoder" | ||
| "github.com/hashicorp/hcl-lang/lang" | ||
| "github.com/hashicorp/hcl-lang/reference" | ||
| "github.com/hashicorp/hcl/v2" | ||
| ilsp "github.com/hashicorp/terraform-ls/internal/lsp" | ||
| lsp "github.com/hashicorp/terraform-ls/internal/protocol" | ||
| ) | ||
|
|
||
| func ReferenceCount(showReferencesCmdId string) lang.CodeLensFunc { | ||
| return func(ctx context.Context, path lang.Path, file string) ([]lang.CodeLens, error) { | ||
| lenses := make([]lang.CodeLens, 0) | ||
|
|
||
| pathCtx, err := decoder.PathCtx(ctx) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| refTargets := pathCtx.ReferenceTargets.OutermostInFile(file) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| // There can be two targets pointing to the same range | ||
| // e.g. when a block is targetable as type-less reference | ||
| // and as an object, which is important in most contexts | ||
| // but not here, where we present it to the user. | ||
| dedupedTargets := make(map[hcl.Range]reference.Targets, 0) | ||
| for _, refTarget := range refTargets { | ||
| rng := *refTarget.RangePtr | ||
| if _, ok := dedupedTargets[rng]; !ok { | ||
| dedupedTargets[rng] = make(reference.Targets, 0) | ||
| } | ||
| dedupedTargets[rng] = append(dedupedTargets[rng], refTarget) | ||
| } | ||
|
|
||
| for rng, refTargets := range dedupedTargets { | ||
| originCount := 0 | ||
| var defRange *hcl.Range | ||
| for _, refTarget := range refTargets { | ||
| if refTarget.DefRangePtr != nil { | ||
| defRange = refTarget.DefRangePtr | ||
| } | ||
|
|
||
| originCount += len(pathCtx.ReferenceOrigins.Targeting(refTarget)) | ||
| } | ||
|
|
||
| if originCount == 0 { | ||
| continue | ||
| } | ||
|
|
||
| var hclPos hcl.Pos | ||
| if defRange != nil { | ||
| hclPos = posMiddleOfRange(defRange) | ||
| } else { | ||
| hclPos = posMiddleOfRange(&rng) | ||
| } | ||
|
|
||
| lenses = append(lenses, lang.CodeLens{ | ||
| Range: rng, | ||
| Command: lang.Command{ | ||
| Title: getTitle("reference", "references", originCount), | ||
| ID: showReferencesCmdId, | ||
| Arguments: []lang.CommandArgument{ | ||
| Position(ilsp.HCLPosToLSP(hclPos)), | ||
| ReferenceContext(lsp.ReferenceContext{}), | ||
| }, | ||
| }, | ||
| }) | ||
| } | ||
| return lenses, nil | ||
| } | ||
| } | ||
|
|
||
| type Position lsp.Position | ||
|
|
||
| func (p Position) MarshalJSON() ([]byte, error) { | ||
| return json.Marshal(lsp.Position(p)) | ||
| } | ||
|
|
||
| type ReferenceContext lsp.ReferenceContext | ||
|
|
||
| func (rc ReferenceContext) MarshalJSON() ([]byte, error) { | ||
| return json.Marshal(lsp.ReferenceContext(rc)) | ||
| } | ||
|
|
||
| func posMiddleOfRange(rng *hcl.Range) hcl.Pos { | ||
| col := rng.Start.Column | ||
| byte := rng.Start.Byte | ||
|
|
||
| if rng.Start.Line == rng.End.Line && rng.End.Column > rng.Start.Column { | ||
| charsFromStart := (rng.End.Column - rng.Start.Column) / 2 | ||
| col += charsFromStart | ||
| byte += charsFromStart | ||
| } | ||
|
|
||
| return hcl.Pos{ | ||
| Line: rng.Start.Line, | ||
| Column: col, | ||
| Byte: byte, | ||
| } | ||
| } | ||
|
|
||
| func getTitle(singular, plural string, n int) string { | ||
| if n > 1 || n == 0 { | ||
| return fmt.Sprintf("%d %s", n, plural) | ||
| } | ||
| return fmt.Sprintf("%d %s", n, singular) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,21 @@ | ||
| package decoder | ||
|
|
||
| import ( | ||
| "context" | ||
|
|
||
| ilsp "github.com/hashicorp/terraform-ls/internal/lsp" | ||
| ) | ||
|
|
||
| type languageIdCtxKey struct{} | ||
|
|
||
| func WithLanguageId(ctx context.Context, langId ilsp.LanguageID) context.Context { | ||
| return context.WithValue(ctx, languageIdCtxKey{}, langId) | ||
| } | ||
|
|
||
| func LanguageId(ctx context.Context) (ilsp.LanguageID, bool) { | ||
| id, ok := ctx.Value(languageIdCtxKey{}).(ilsp.LanguageID) | ||
| if !ok { | ||
| return "", false | ||
| } | ||
| return id, true | ||
| } |
Oops, something went wrong.