hcl: Introduce TokenAtPosition

hashicorp · Jun 2, 2020 · 010655f · 010655f
1 parent f7fc4bd
commit 010655f
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/internal/hcl/errors.go b/internal/hcl/errors.go
@@ -27,3 +27,11 @@ func IsNoBlockFoundErr(err error) bool {
 	_, ok := err.(*NoBlockFoundErr)
 	return ok
 }
+
+type NoTokenFoundErr struct {
+	AtPos hcllib.Pos
+}
+
+func (e *NoTokenFoundErr) Error() string {
+	return fmt.Sprintf("no token found at %#v", e.AtPos)
+}
diff --git a/internal/hcl/hcl.go b/internal/hcl/hcl.go
@@ -11,6 +11,7 @@ import (
 
 type File interface {
 	BlockTokensAtPosition(hcl.Pos) (hclsyntax.Tokens, error)
+	TokenAtPosition(hcl.Pos) (hclsyntax.Token, error)
 }
 
 type file struct {
@@ -87,6 +88,18 @@ func (f *file) BlockTokensAtPosition(pos hcllib.Pos) (hclsyntax.Tokens, error) {
 	return pf.Tokens, &NoBlockFoundErr{pos}
 }
 
+func (f *file) TokenAtPosition(pos hcllib.Pos) (hclsyntax.Token, error) {
+	pf, _ := f.parse()
+
+	for _, t := range pf.Tokens {
+		if rangeContainsOffset(t.Range, pos.Byte) {
+			return t, nil
+		}
+	}
+
+	return hclsyntax.Token{}, &NoTokenFoundErr{pos}
+}
+
 func tokensInRange(tokens hclsyntax.Tokens, rng hcllib.Range) hclsyntax.Tokens {
 	var ts hclsyntax.Tokens