@leafer/selector 1.0.0-rc.4 → 1.0.0-rc.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@leafer/selector",
-  "version": "1.0.0-rc.4",
+  "version": "1.0.0-rc.6",
   "description": "@leafer/selector",
   "author": "Chao (Leafer) Wan",
   "license": "MIT",
@@ -22,9 +22,9 @@
     "leaferjs"
   ],
   "dependencies": {
-    "@leafer/core": "1.0.0-rc.4"
+    "@leafer/core": "1.0.0-rc.6"
   },
   "devDependencies": {
-    "@leafer/interface": "1.0.0-rc.4"
+    "@leafer/interface": "1.0.0-rc.6"
   }
 }

package/src/{FindPath.ts → Pather.ts}

@@ -4,7 +4,7 @@ import { BoundsHelper, LeafList, LeafHelper } from '@leafer/core'
 
 const { hitRadiusPoint } = BoundsHelper
 
-export class FindPath {
+export class Pather {
 
     protected target: ILeaf
     protected selector: ISelector
@@ -104,7 +104,7 @@ export class FindPath {
     }
 
 
-    protected eachFind(children: Array<ILeaf>, hitMask: boolean): void {
+    protected eachFind(children: ILeaf[], hitMask: boolean): void {
         let child: ILeaf, hit: boolean
         const { point } = this, len = children.length
         for (let i = len - 1; i > -1; i--) {

package/src/Selector.ts

@@ -1,127 +1,136 @@
-import { ILeaf, ILeafArrayMap, ILeafMap, ISelector, ISelectPathResult, ISelectPathOptions, IPointData, IEventListenerId, ISelectorConfig } from '@leafer/interface'
-import { ChildEvent, LayoutEvent, DataHelper, Platform } from '@leafer/core'
+import { ILeaf, ILeafMap, ILeafList, ISelector, ISelectPathResult, ISelectPathOptions, IPointData, IEventListenerId, ISelectorConfig, IFindMethod } from '@leafer/interface'
+import { ChildEvent, LayoutEvent, DataHelper, Platform, PropertyEvent, LeafHelper } from '@leafer/core'
 
-import { FindPath } from './FindPath'
-
-
-interface IFind {
-    (leaf: ILeaf): boolean
-}
+import { Pather } from './Pather'
 
 
 export class Selector implements ISelector {
 
     public target: ILeaf
 
+    public list: ILeafList
+
     public config: ISelectorConfig = {}
 
-    protected findPath: FindPath
+    protected pather: Pather
+
+
+    protected innerIdMap: ILeafMap = {}
+    protected idMap: ILeafMap = {}
+
+    protected findLeaf: ILeaf
 
-    protected innerIdList: ILeafMap = {}
-    protected idList: ILeafMap = {}
-    protected classNameList: ILeafArrayMap = {}
-    protected tagNameList: ILeafArrayMap = {}
+    protected methods = {
+        id: (leaf: ILeaf, name: string) => leaf.id === name ? this.idMap[name] = leaf : 0,
+        innerId: (leaf: ILeaf, innerId: number) => leaf.innerId === innerId ? this.innerIdMap[innerId] = leaf : 0,
+        className: (leaf: ILeaf, name: string) => leaf.className === name ? 1 : 0,
+        tag: (leaf: ILeaf, name: string) => leaf.__tag === name ? 1 : 0
+    }
 
     protected __eventIds: IEventListenerId[]
 
+
     constructor(target: ILeaf, userConfig?: ISelectorConfig) {
         this.target = target
         if (userConfig) this.config = DataHelper.default(userConfig, this.config)
-        this.findPath = new FindPath(target, this)
+        this.pather = new Pather(target, this)
         this.__listenEvents()
     }
 
     public getByPoint(hitPoint: IPointData, hitRadius: number, options?: ISelectPathOptions): ISelectPathResult {
         if (Platform.name === 'node') this.target.emit(LayoutEvent.CHECK_UPDATE)
-        return this.findPath.getByPoint(hitPoint, hitRadius, options)
-    }
-
-    public find(name: number | string, branch?: ILeaf): ILeaf | ILeaf[] {
-        if (typeof name === 'number') {
-            return this.getByInnerId(name, branch)
-        } else if (name.startsWith('#')) {
-            return this.getById(name.substring(1), branch)
-        } else if (name.startsWith('.')) {
-            return this.getByClassName(name.substring(1), branch)
-        } else {
-            return this.getByTagName(name, branch)
+        return this.pather.getByPoint(hitPoint, hitRadius, options)
+    }
+
+    public getBy(condition: number | string | IFindMethod, branch?: ILeaf, one?: boolean, options?: any): ILeaf | ILeaf[] {
+        switch (typeof condition) {
+            case 'number':
+                const leaf = this.getByInnerId(condition, branch)
+                return one ? leaf : (leaf ? [leaf] : [])
+            case 'string':
+                switch (condition[0]) {
+                    case '#':
+                        const leaf = this.getById(condition.substring(1), branch)
+                        return one ? leaf : (leaf ? [leaf] : [])
+                    case '.':
+                        return this.getByMethod(this.methods.className, branch, one, condition.substring(1)) // className
+                    default:
+                        return this.getByMethod(this.methods.tag, branch, one, condition) // tagName
+                }
+            case 'function':
+                return this.getByMethod(condition as IFindMethod, branch, one, options)
         }
     }
 
-    public getByInnerId(name: number, branch?: ILeaf): ILeaf {
-        let cache = this.innerIdList[name]
+    public getByInnerId(innerId: number, branch?: ILeaf): ILeaf {
+        const cache = this.innerIdMap[innerId]
         if (cache) return cache
-        if (!branch) branch = this.target
-        let find: ILeaf
-        this.loopFind(branch, (leaf) => {
-            if (leaf.innerId === name) {
-                find = leaf
-                this.innerIdList[name] = find
-                return true
-            } else {
-                return false
-            }
-        })
-        return find
+        this.eachFind(this.toChildren(branch), this.methods.innerId, null, innerId)
+        return this.findLeaf
     }
 
-    public getById(name: string, branch?: ILeaf): ILeaf {
-        let cache = this.idList[name]
-        if (cache) return cache
-        if (!branch) branch = this.target
-        let find: ILeaf
-        this.loopFind(branch, (leaf) => {
-            if (leaf.id === name) {
-                find = leaf
-                this.idList[name] = find
-                return true
-            } else {
-                return false
-            }
-        })
-        return find
+    public getById(id: string, branch?: ILeaf): ILeaf {
+        const cache = this.idMap[id]
+        if (cache && LeafHelper.hasParent(cache, branch || this.target)) return cache
+        this.eachFind(this.toChildren(branch), this.methods.id, null, id)
+        return this.findLeaf
+    }
+
+    public getByClassName(className: string, branch?: ILeaf): ILeaf[] {
+        return this.getByMethod(this.methods.className, branch, false, className) as ILeaf[]
     }
 
-    public getByClassName(name: string, branch?: ILeaf): ILeaf[] {
-        if (!branch) branch = this.target
-        let find: Array<ILeaf | ILeaf> = []
-        this.loopFind(branch, (leaf) => {
-            if (leaf.className === name) find.push(leaf)
-            return false
-        })
-        return find
+    public getByTag(tag: string, branch?: ILeaf): ILeaf[] {
+        return this.getByMethod(this.methods.tag, branch, false, tag) as ILeaf[]
     }
 
-    public getByTagName(name: string, branch?: ILeaf): ILeaf[] {
-        if (!branch) branch = this.target
-        let find: Array<ILeaf | ILeaf> = []
-        this.loopFind(branch, (leaf) => {
-            if (leaf.__tag === name) find.push(leaf)
-            return false
-        })
-        return find
+    public getByMethod(method: IFindMethod, branch?: ILeaf, one?: boolean, options?: any): ILeaf[] | ILeaf {
+        const list: ILeaf[] = one ? null : []
+        this.eachFind(this.toChildren(branch), method, list, options)
+        return list || this.findLeaf
     }
 
-    protected loopFind(branch: ILeaf, find: IFind): void {
-        if (find(branch)) return
-        const { children } = branch
+
+    protected eachFind(children: ILeaf[], method: IFindMethod, list?: ILeaf[], options?: any,): void {
+        let child: ILeaf
         for (let i = 0, len = children.length; i < len; i++) {
-            branch = children[i] as ILeaf
-            if (find(branch)) return
-            if (branch.isBranch) this.loopFind(branch, find)
+            child = children[i]
+            if (method(child, options)) {
+                if (list) {
+                    list.push(child)
+                } else {
+                    this.findLeaf = child
+                    return
+                }
+            }
+            if (child.isBranch) this.eachFind(child.children, method, list, options)
         }
     }
 
+    protected toChildren(branch: ILeaf): ILeaf[] {
+        this.findLeaf = null
+        return [branch || this.target]
+    }
+
+
     protected __onRemoveChild(event: ChildEvent): void {
-        const target = event.target as ILeaf
-        if (this.idList[target.id]) this.idList[target.id] = null
-        if (this.innerIdList[target.id]) this.innerIdList[target.innerId] = null
+        const { id, innerId } = event.child
+        if (this.idMap[id]) delete this.idMap[id]
+        if (this.innerIdMap[innerId]) delete this.innerIdMap[innerId]
+    }
+
+    protected __checkIdChange(event: PropertyEvent): void {
+        if (event.attrName === 'id') {
+            const id = event.oldValue as string
+            if (this.idMap[id]) delete this.idMap[id]
+        }
     }
 
 
     protected __listenEvents(): void {
         this.__eventIds = [
-            this.target.on_(ChildEvent.REMOVE, this.__onRemoveChild, this)
+            this.target.on_(ChildEvent.REMOVE, this.__onRemoveChild, this),
+            this.target.on_(PropertyEvent.CHANGE, this.__checkIdChange, this)
         ]
     }
 
@@ -133,11 +142,10 @@ export class Selector implements ISelector {
     public destroy(): void {
         if (this.__eventIds.length) {
             this.__removeListenEvents()
-            this.findPath.destroy()
-            this.innerIdList = {}
-            this.idList = {}
-            this.classNameList = {}
-            this.tagNameList = {}
+            this.pather.destroy()
+            this.findLeaf = null
+            this.innerIdMap = {}
+            this.idMap = {}
         }
     }
 

package/types/index.d.ts

@@ -1,7 +1,7 @@
-import { ILeaf, ISelector, ILeafList, IRadiusPointData, IPointData, ISelectPathOptions, ISelectPathResult, ISelectorConfig, ILeafMap, ILeafArrayMap, IEventListenerId } from '@leafer/interface';
-import { LeafList, ChildEvent } from '@leafer/core';
+import { ILeaf, ISelector, ILeafList, IRadiusPointData, IPointData, ISelectPathOptions, ISelectPathResult, ISelectorConfig, ILeafMap, IEventListenerId, IFindMethod } from '@leafer/interface';
+import { LeafList, ChildEvent, PropertyEvent } from '@leafer/core';
 
-declare class FindPath {
+declare class Pather {
     protected target: ILeaf;
     protected selector: ISelector;
     protected findList: ILeaf[];
@@ -13,33 +13,39 @@ declare class FindPath {
     getPath(leaf: ILeaf): LeafList;
     getHitablePath(leaf: ILeaf): LeafList;
     getThroughPath(list: ILeaf[]): LeafList;
-    protected eachFind(children: Array<ILeaf>, hitMask: boolean): void;
+    protected eachFind(children: ILeaf[], hitMask: boolean): void;
     protected hitChild(child: ILeaf, point: IRadiusPointData): void;
     protected clear(): void;
     destroy(): void;
 }
 
-interface IFind {
-    (leaf: ILeaf): boolean;
-}
 declare class Selector implements ISelector {
     target: ILeaf;
+    list: ILeafList;
     config: ISelectorConfig;
-    protected findPath: FindPath;
-    protected innerIdList: ILeafMap;
-    protected idList: ILeafMap;
-    protected classNameList: ILeafArrayMap;
-    protected tagNameList: ILeafArrayMap;
+    protected pather: Pather;
+    protected innerIdMap: ILeafMap;
+    protected idMap: ILeafMap;
+    protected findLeaf: ILeaf;
+    protected methods: {
+        id: (leaf: ILeaf, name: string) => ILeaf | 0;
+        innerId: (leaf: ILeaf, innerId: number) => ILeaf | 0;
+        className: (leaf: ILeaf, name: string) => 1 | 0;
+        tag: (leaf: ILeaf, name: string) => 1 | 0;
+    };
     protected __eventIds: IEventListenerId[];
     constructor(target: ILeaf, userConfig?: ISelectorConfig);
     getByPoint(hitPoint: IPointData, hitRadius: number, options?: ISelectPathOptions): ISelectPathResult;
-    find(name: number | string, branch?: ILeaf): ILeaf | ILeaf[];
-    getByInnerId(name: number, branch?: ILeaf): ILeaf;
-    getById(name: string, branch?: ILeaf): ILeaf;
-    getByClassName(name: string, branch?: ILeaf): ILeaf[];
-    getByTagName(name: string, branch?: ILeaf): ILeaf[];
-    protected loopFind(branch: ILeaf, find: IFind): void;
+    getBy(condition: number | string | IFindMethod, branch?: ILeaf, one?: boolean, options?: any): ILeaf | ILeaf[];
+    getByInnerId(innerId: number, branch?: ILeaf): ILeaf;
+    getById(id: string, branch?: ILeaf): ILeaf;
+    getByClassName(className: string, branch?: ILeaf): ILeaf[];
+    getByTag(tag: string, branch?: ILeaf): ILeaf[];
+    getByMethod(method: IFindMethod, branch?: ILeaf, one?: boolean, options?: any): ILeaf[] | ILeaf;
+    protected eachFind(children: ILeaf[], method: IFindMethod, list?: ILeaf[], options?: any): void;
+    protected toChildren(branch: ILeaf): ILeaf[];
     protected __onRemoveChild(event: ChildEvent): void;
+    protected __checkIdChange(event: PropertyEvent): void;
     protected __listenEvents(): void;
     protected __removeListenEvents(): void;
     destroy(): void;