feat(selector): export connectKeys

badest192 · Jan 8, 2024 · 8a8de5d · 8a8de5d
1 parent 7d96c5f
commit 8a8de5d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/selector/src/commonMain/kotlin/li/songe/selector/CommonSelector.kt b/selector/src/commonMain/kotlin/li/songe/selector/CommonSelector.kt
@@ -11,6 +11,7 @@ class CommonSelector private constructor(
 ) {
     val tracks = selector.tracks
     val trackIndex = selector.trackIndex
+    val connectKeys = selector.connectKeys
 
     fun <T : Any> match(node: T, transform: CommonTransform<T>): T? {
         return selector.match(node, transform.transform)

diff --git a/selector/src/commonMain/kotlin/li/songe/selector/Selector.kt b/selector/src/commonMain/kotlin/li/songe/selector/Selector.kt
@@ -23,6 +23,17 @@ class Selector internal constructor(private val propertyWrapper: PropertyWrapper
         if (i < 0) 0 else i
     }
 
+    val connectKeys by lazy {
+        var c = propertyWrapper.to
+        val keys = mutableListOf<String>()
+        while (c != null) {
+            c!!.connectSegment.connectExpression
+            keys.add(c!!.connectSegment.operator.key)
+            c = c?.to?.to
+        }
+        keys.toTypedArray()
+    }
+
     fun <T> match(
         node: T,
         transform: Transform<T>,