refactor(internal/lsp): 调整 semanticTokens 的实现方式

现在每个元素都带有 Location.URI 字段，不再需要先比较 api.URI
再比较元素的范围，可以直接与元素的 URI 字段作比较。

Author: caixw

internal/lsp/semantic.go

@@ -8,8 +8,6 @@ import (
 	"sort"
 	"unicode"
 
-	"github.com/issue9/sliceutil"
-
 	"github.com/caixw/apidoc/v7/core"
 	"github.com/caixw/apidoc/v7/internal/ast"
 	"github.com/caixw/apidoc/v7/internal/lsp/protocol"
@@ -18,7 +16,8 @@ import (
 )
 
 type tokenBuilder struct {
-	tag, attr, value int // 可用的 token
+	uri              core.URI // 当前 builder 对应的文件地址
+	tag, attr, value int      // 可用的 token
 
 	// 每个二级数组长度为 5，表示一组 semanticToken 数据。
 	// 数据分别为 绝对行号，当前行的绝对起始位置，长度，以及 token 和 modifier。
@@ -44,25 +43,14 @@ func (s *server) textDocumentSemanticTokens(notify bool, in *protocol.SemanticTo
 // value 表示属性的颜色值；
 func semanticTokens(doc *ast.APIDoc, uri core.URI, tag, attr, value int) []int {
 	b := &tokenBuilder{
+		uri:    uri,
 		tag:    tag,
 		attr:   attr,
 		value:  value,
 		tokens: make([][]int, 0, 100),
 	}
 
-	if doc.URI == uri {
-		b.parse(reflect.ValueOf(doc), "APIs")
-	}
-
-	for _, api := range doc.APIs {
-		matched := api.URI == uri || (api.URI == "" && doc.URI == uri)
-		if !matched {
-			continue
-		}
-
-		b.parse(reflect.ValueOf(api))
-	}
-
+	b.parse(reflect.ValueOf(doc))
 	b.sort()
 	return b.build()
 }
@@ -108,20 +96,18 @@ func (b *tokenBuilder) sort() {
 	})
 }
 
-func (b *tokenBuilder) parse(v reflect.Value, exclude ...string) {
+func (b *tokenBuilder) parse(v reflect.Value) {
 	v = node.RealValue(v)
-	t := v.Type()
-	if t.Kind() != reflect.Struct {
+	if !b.matched(v) {
 		return
 	}
 
 	b.parseAnonymous(v)
 
+	t := v.Type()
 	for i := 0; i < t.NumField(); i++ {
 		tf := t.Field(i)
-		if tf.Anonymous ||
-			unicode.IsLower(rune(tf.Name[0])) ||
-			sliceutil.Count(exclude, func(i int) bool { return exclude[i] == tf.Name }) > 0 { // 需要过滤的字段
+		if tf.Anonymous || unicode.IsLower(rune(tf.Name[0])) {
 			continue
 		}
 
@@ -136,6 +122,23 @@ func (b *tokenBuilder) parse(v reflect.Value, exclude ...string) {
 	}
 }
 
+func (b *tokenBuilder) matched(v reflect.Value) bool {
+	if v.Kind() != reflect.Struct {
+		return false
+	}
+
+	if s, ok := v.Interface().(core.Searcher); ok && s.Loc().URI == b.uri {
+		return true
+	}
+
+	if !v.CanAddr() {
+		return false
+	}
+
+	s, ok := v.Addr().Interface().(core.Searcher)
+	return ok && s.Loc().URI == b.uri
+}
+
 func (b *tokenBuilder) parseAnonymous(v reflect.Value) {
 	t := v.Type()
 	switch elem := v.Interface().(type) {

internal/lsp/semantic_test.go

@@ -90,10 +90,11 @@ func TestTokenBuilder_sort(t *testing.T) {
 	})
 }
 
-func TestTokens(t *testing.T) {
+func TestSemanticTokens(t *testing.T) {
 	a := assert.New(t)
 
-	b := `<apidoc version="1.1.1">
+	// NOTE 此处的 apidoc 属性值必须与当前的文档主版本号相同
+	b := `<apidoc version="1.1.1" apidoc="6.0.0" created="2020-01-02T13:12:11+08:00">
 	<title>标题</title>
 	<mimetype>xml</mimetype>
 	<api method="GET">
@@ -103,7 +104,7 @@ func TestTokens(t *testing.T) {
 
 	<api method="POST">
 		<path path="/users" />
-		<response status="200" />
+		<response status="200" type="number" />
 	</api>
 </apidoc>`
 	blk := core.Block{Data: []byte(b), Location: core.Location{URI: "doc.go"}}
@@ -117,6 +118,10 @@ func TestTokens(t *testing.T) {
 		0, 1, 6, 1, 0, // apidoc
 		0, 7, 7, 2, 0,
 		0, 9, 5, 3, 0,
+		0, 7, 6, 2, 0,
+		0, 8, 5, 3, 0,
+		0, 7, 7, 2, 0,
+		0, 9, 25, 3, 0,
 
 		1, 2, 5, 1, 0, // <title>
 		// {1, 8, 2, 0, 0}, // 元素内容不作解析，直接采用默认的注释颜色
@@ -151,6 +156,8 @@ func TestTokens(t *testing.T) {
 		1, 3, 8, 1, 0, // response
 		0, 9, 6, 2, 0,
 		0, 8, 3, 3, 0,
+		0, 5, 4, 2, 0,
+		0, 6, 6, 3, 0,
 
 		1, 3, 3, 1, 0, // </api>