[refactoring] internal position package
This commit is contained in:
Vadym Slizov
204
internal/position/position.go
Normal file
204
internal/position/position.go
Normal file
@@ -0,0 +1,204 @@
|
||||
package position
|
||||
|
||||
import (
|
||||
"github.com/z7zmey/php-parser/internal/scanner"
|
||||
"github.com/z7zmey/php-parser/pkg/ast"
|
||||
"github.com/z7zmey/php-parser/pkg/position"
|
||||
)
|
||||
|
||||
type startPos struct {
|
||||
startLine int
|
||||
startPos int
|
||||
}
|
||||
|
||||
type endPos struct {
|
||||
endLine int
|
||||
endPos int
|
||||
}
|
||||
|
||||
func getListStartPos(l []ast.Vertex) startPos {
|
||||
if l == nil {
|
||||
return startPos{-1, -1}
|
||||
}
|
||||
|
||||
if len(l) == 0 {
|
||||
return startPos{-1, -1}
|
||||
}
|
||||
|
||||
return getNodeStartPos(l[0])
|
||||
}
|
||||
|
||||
func getNodeStartPos(n ast.Vertex) startPos {
|
||||
sl := -1
|
||||
sp := -1
|
||||
|
||||
if n == nil {
|
||||
return startPos{-1, -1}
|
||||
}
|
||||
|
||||
p := n.GetNode().Position
|
||||
if p != nil {
|
||||
sl = p.StartLine
|
||||
sp = p.StartPos
|
||||
}
|
||||
|
||||
return startPos{sl, sp}
|
||||
}
|
||||
|
||||
func getListEndPos(l []ast.Vertex) endPos {
|
||||
if l == nil {
|
||||
return endPos{-1, -1}
|
||||
}
|
||||
|
||||
if len(l) == 0 {
|
||||
return endPos{-1, -1}
|
||||
}
|
||||
|
||||
return getNodeEndPos(l[len(l)-1])
|
||||
}
|
||||
|
||||
func getNodeEndPos(n ast.Vertex) endPos {
|
||||
el := -1
|
||||
ep := -1
|
||||
|
||||
if n == nil {
|
||||
return endPos{-1, -1}
|
||||
}
|
||||
|
||||
p := n.GetNode().Position
|
||||
if p != nil {
|
||||
el = p.EndLine
|
||||
ep = p.EndPos
|
||||
}
|
||||
|
||||
return endPos{el, ep}
|
||||
}
|
||||
|
||||
// NewNodeListPosition returns new Position
|
||||
func NewNodeListPosition(list []ast.Vertex) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: getListStartPos(list).startLine,
|
||||
EndLine: getListEndPos(list).endLine,
|
||||
StartPos: getListStartPos(list).startPos,
|
||||
EndPos: getListEndPos(list).endPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewNodePosition returns new Position
|
||||
func NewNodePosition(n ast.Vertex) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: getNodeStartPos(n).startLine,
|
||||
EndLine: getNodeEndPos(n).endLine,
|
||||
StartPos: getNodeStartPos(n).startPos,
|
||||
EndPos: getNodeEndPos(n).endPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewTokenPosition returns new Position
|
||||
func NewTokenPosition(t *scanner.Token) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: t.Position.StartLine,
|
||||
EndLine: t.Position.EndLine,
|
||||
StartPos: t.Position.StartPos,
|
||||
EndPos: t.Position.EndPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewTokensPosition returns new Position
|
||||
func NewTokensPosition(startToken *scanner.Token, endToken *scanner.Token) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: startToken.Position.StartLine,
|
||||
EndLine: endToken.Position.EndLine,
|
||||
StartPos: startToken.Position.StartPos,
|
||||
EndPos: endToken.Position.EndPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewTokenNodePosition returns new Position
|
||||
func NewTokenNodePosition(t *scanner.Token, n ast.Vertex) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: t.Position.StartLine,
|
||||
EndLine: getNodeEndPos(n).endLine,
|
||||
StartPos: t.Position.StartPos,
|
||||
EndPos: getNodeEndPos(n).endPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewNodeTokenPosition returns new Position
|
||||
func NewNodeTokenPosition(n ast.Vertex, t *scanner.Token) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: getNodeStartPos(n).startLine,
|
||||
EndLine: t.Position.EndLine,
|
||||
StartPos: getNodeStartPos(n).startPos,
|
||||
EndPos: t.Position.EndPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewNodesPosition returns new Position
|
||||
func NewNodesPosition(startNode ast.Vertex, endNode ast.Vertex) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: getNodeStartPos(startNode).startLine,
|
||||
EndLine: getNodeEndPos(endNode).endLine,
|
||||
StartPos: getNodeStartPos(startNode).startPos,
|
||||
EndPos: getNodeEndPos(endNode).endPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewNodeListTokenPosition returns new Position
|
||||
func NewNodeListTokenPosition(list []ast.Vertex, t *scanner.Token) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: getListStartPos(list).startLine,
|
||||
EndLine: t.Position.EndLine,
|
||||
StartPos: getListStartPos(list).startPos,
|
||||
EndPos: t.Position.EndPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewTokenNodeListPosition returns new Position
|
||||
func NewTokenNodeListPosition(t *scanner.Token, list []ast.Vertex) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: t.Position.StartLine,
|
||||
EndLine: getListEndPos(list).endLine,
|
||||
StartPos: t.Position.StartPos,
|
||||
EndPos: getListEndPos(list).endPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewNodeNodeListPosition returns new Position
|
||||
func NewNodeNodeListPosition(n ast.Vertex, list []ast.Vertex) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: getNodeStartPos(n).startLine,
|
||||
EndLine: getListEndPos(list).endLine,
|
||||
StartPos: getNodeStartPos(n).startPos,
|
||||
EndPos: getListEndPos(list).endPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewNodeListNodePosition returns new Position
|
||||
func NewNodeListNodePosition(list []ast.Vertex, n ast.Vertex) *position.Position {
|
||||
return &position.Position{
|
||||
StartLine: getListStartPos(list).startLine,
|
||||
EndLine: getNodeEndPos(n).endLine,
|
||||
StartPos: getListStartPos(list).startPos,
|
||||
EndPos: getNodeEndPos(n).endPos,
|
||||
}
|
||||
}
|
||||
|
||||
// NewOptionalListTokensPosition returns new Position
|
||||
func NewOptionalListTokensPosition(list []ast.Vertex, t *scanner.Token, endToken *scanner.Token) *position.Position {
|
||||
if list == nil {
|
||||
return &position.Position{
|
||||
StartLine: t.Position.StartLine,
|
||||
EndLine: endToken.Position.EndLine,
|
||||
StartPos: t.Position.StartPos,
|
||||
EndPos: endToken.Position.EndPos,
|
||||
}
|
||||
}
|
||||
|
||||
return &position.Position{
|
||||
StartLine: getListStartPos(list).startLine,
|
||||
EndLine: endToken.Position.EndLine,
|
||||
StartPos: getListStartPos(list).startPos,
|
||||
EndPos: endToken.Position.EndPos,
|
||||
}
|
||||
}
|
||||
475
internal/position/position_test.go
Normal file
475
internal/position/position_test.go
Normal file
@@ -0,0 +1,475 @@
|
||||
package position_test
|
||||
|
||||
import (
|
||||
"gotest.tools/assert"
|
||||
"testing"
|
||||
|
||||
builder "github.com/z7zmey/php-parser/internal/position"
|
||||
"github.com/z7zmey/php-parser/internal/scanner"
|
||||
"github.com/z7zmey/php-parser/pkg/ast"
|
||||
"github.com/z7zmey/php-parser/pkg/position"
|
||||
)
|
||||
|
||||
func TestNewTokenPosition(t *testing.T) {
|
||||
tkn := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 3,
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewTokenPosition(tkn)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 1, 0, 3}, pos)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 1, 0, 3}, pos)
|
||||
}
|
||||
|
||||
func TestNewTokensPosition(t *testing.T) {
|
||||
token1 := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 3,
|
||||
},
|
||||
}
|
||||
token2 := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 4,
|
||||
EndPos: 6,
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewTokensPosition(token1, token2)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 2, 0, 6}, pos)
|
||||
}
|
||||
|
||||
func TestNewNodePosition(t *testing.T) {
|
||||
n := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 3,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodePosition(n)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 1, 0, 3}, pos)
|
||||
}
|
||||
|
||||
func TestNewTokenNodePosition(t *testing.T) {
|
||||
tkn := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 3,
|
||||
},
|
||||
}
|
||||
n := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 4,
|
||||
EndPos: 12,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewTokenNodePosition(tkn, n)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 2, 0, 12}, pos)
|
||||
}
|
||||
|
||||
func TestNewNodeTokenPosition(t *testing.T) {
|
||||
n := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 9,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
tkn := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 10,
|
||||
EndPos: 12,
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodeTokenPosition(n, tkn)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 2, 0, 12}, pos)
|
||||
}
|
||||
|
||||
func TestNewNodeListPosition(t *testing.T) {
|
||||
n1 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 9,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
n2 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 10,
|
||||
EndPos: 19,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodeListPosition([]ast.Vertex{n1, n2})
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 2, 0, 19}, pos)
|
||||
}
|
||||
|
||||
func TestNewNodesPosition(t *testing.T) {
|
||||
n1 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 9,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
n2 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 10,
|
||||
EndPos: 19,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodesPosition(n1, n2)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 2, 0, 19}, pos)
|
||||
}
|
||||
|
||||
func TestNewNodeListTokenPosition(t *testing.T) {
|
||||
n1 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 9,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
n2 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 10,
|
||||
EndPos: 19,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
tkn := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 3,
|
||||
EndLine: 3,
|
||||
StartPos: 20,
|
||||
EndPos: 22,
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodeListTokenPosition([]ast.Vertex{n1, n2}, tkn)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 3, 0, 22}, pos)
|
||||
}
|
||||
|
||||
func TestNewTokenNodeListPosition(t *testing.T) {
|
||||
tkn := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 2,
|
||||
},
|
||||
}
|
||||
|
||||
n1 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 3,
|
||||
EndPos: 10,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
n2 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 3,
|
||||
EndLine: 3,
|
||||
StartPos: 11,
|
||||
EndPos: 20,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewTokenNodeListPosition(tkn, []ast.Vertex{n1, n2})
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 3, 0, 20}, pos)
|
||||
}
|
||||
|
||||
func TestNewNodeNodeListPosition(t *testing.T) {
|
||||
n1 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 8,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
n2 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 9,
|
||||
EndPos: 17,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
n3 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 3,
|
||||
EndLine: 3,
|
||||
StartPos: 18,
|
||||
EndPos: 26,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodeNodeListPosition(n1, []ast.Vertex{n2, n3})
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 3, 0, 26}, pos)
|
||||
}
|
||||
|
||||
func TestNewNodeListNodePosition(t *testing.T) {
|
||||
n1 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 8,
|
||||
},
|
||||
},
|
||||
}
|
||||
n2 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 9,
|
||||
EndPos: 17,
|
||||
},
|
||||
},
|
||||
}
|
||||
n3 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 3,
|
||||
EndLine: 3,
|
||||
StartPos: 18,
|
||||
EndPos: 26,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodeListNodePosition([]ast.Vertex{n1, n2}, n3)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 3, 0, 26}, pos)
|
||||
}
|
||||
|
||||
func TestNewOptionalListTokensPosition(t *testing.T) {
|
||||
token1 := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 3,
|
||||
},
|
||||
}
|
||||
token2 := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 4,
|
||||
EndPos: 6,
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewOptionalListTokensPosition(nil, token1, token2)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, 2, 0, 6}, pos)
|
||||
}
|
||||
|
||||
func TestNewOptionalListTokensPosition2(t *testing.T) {
|
||||
n2 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 2,
|
||||
EndLine: 2,
|
||||
StartPos: 9,
|
||||
EndPos: 17,
|
||||
},
|
||||
},
|
||||
}
|
||||
n3 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 3,
|
||||
EndLine: 3,
|
||||
StartPos: 18,
|
||||
EndPos: 26,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
token1 := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 4,
|
||||
EndLine: 4,
|
||||
StartPos: 27,
|
||||
EndPos: 29,
|
||||
},
|
||||
}
|
||||
token2 := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 5,
|
||||
EndLine: 5,
|
||||
StartPos: 30,
|
||||
EndPos: 32,
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewOptionalListTokensPosition([]ast.Vertex{n2, n3}, token1, token2)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{2, 5, 9, 32}, pos)
|
||||
}
|
||||
|
||||
func TestNilNodePos(t *testing.T) {
|
||||
pos := builder.NewNodesPosition(nil, nil)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{-1, -1, -1, -1}, pos)
|
||||
}
|
||||
|
||||
func TestNilNodeListPos(t *testing.T) {
|
||||
n1 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 8,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodeNodeListPosition(n1, nil)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, -1, 0, -1}, pos)
|
||||
}
|
||||
|
||||
func TestNilNodeListTokenPos(t *testing.T) {
|
||||
token := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 3,
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodeListTokenPosition(nil, token)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{-1, 1, -1, 3}, pos)
|
||||
}
|
||||
|
||||
func TestEmptyNodeListPos(t *testing.T) {
|
||||
n1 := &ast.Identifier{
|
||||
Node: ast.Node{
|
||||
Position: &position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 8,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodeNodeListPosition(n1, []ast.Vertex{})
|
||||
|
||||
assert.DeepEqual(t, &position.Position{1, -1, 0, -1}, pos)
|
||||
}
|
||||
|
||||
func TestEmptyNodeListTokenPos(t *testing.T) {
|
||||
token := &scanner.Token{
|
||||
Value: []byte(`foo`),
|
||||
Position: position.Position{
|
||||
StartLine: 1,
|
||||
EndLine: 1,
|
||||
StartPos: 0,
|
||||
EndPos: 3,
|
||||
},
|
||||
}
|
||||
|
||||
pos := builder.NewNodeListTokenPosition([]ast.Vertex{}, token)
|
||||
|
||||
assert.DeepEqual(t, &position.Position{-1, 1, -1, 3}, pos)
|
||||
}
|
||||
Reference in New Issue
Block a user