-
Notifications
You must be signed in to change notification settings - Fork 132
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
- Loading branch information
1 parent
9c4b998
commit 589dfdf
Showing
33 changed files
with
577 additions
and
748 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.