mirror of https://github.com/nocodb/nocodb
Wing-Kam Wong
2 years ago
2 changed files with 88 additions and 0 deletions
@ -0,0 +1,87 @@ |
|||||||
|
// ref : https://medium.com/weekly-webtips/js-implementing-auto-complete-f4c5a5d5c009
|
||||||
|
|
||||||
|
interface Node { |
||||||
|
value: [] |
||||||
|
isLeaf: boolean |
||||||
|
children: Record<string, Node> |
||||||
|
} |
||||||
|
|
||||||
|
export class NcAutocompleteTree { |
||||||
|
trie: Record<string, any> |
||||||
|
suggestions: string[] |
||||||
|
|
||||||
|
constructor() { |
||||||
|
this.trie = {} |
||||||
|
this.suggestions = [] |
||||||
|
} |
||||||
|
|
||||||
|
newNode(): Node { |
||||||
|
return { |
||||||
|
value: [], |
||||||
|
isLeaf: false, |
||||||
|
children: {}, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
add(word: Record<string, any>) { |
||||||
|
if (!this.trie) { |
||||||
|
this.trie = this.newNode() |
||||||
|
} |
||||||
|
|
||||||
|
let root = this.trie |
||||||
|
for (const letter of word.text.toLowerCase()) { |
||||||
|
if (!(letter in root.children)) { |
||||||
|
root.children[letter] = this.newNode() |
||||||
|
} |
||||||
|
root = root.children[letter] |
||||||
|
} |
||||||
|
root.value = root.value || [] |
||||||
|
root.value.push(word) |
||||||
|
} |
||||||
|
|
||||||
|
find(word: string) { |
||||||
|
let root = this.trie |
||||||
|
for (const letter of word) { |
||||||
|
if (letter in root.children) { |
||||||
|
root = root.children[letter] |
||||||
|
} else { |
||||||
|
return null // if not found return null
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return root // return the root where it ends search
|
||||||
|
} |
||||||
|
|
||||||
|
traverse(root: Node) { |
||||||
|
if (root.value && root.value.length) { |
||||||
|
this.suggestions.push(...root.value) |
||||||
|
} |
||||||
|
|
||||||
|
for (const letter in root.children) { |
||||||
|
this.traverse(root.children[letter]) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
complete(word: string, CHILDREN = null) { |
||||||
|
this.suggestions = [] |
||||||
|
const root = this.find(word.toLowerCase()) |
||||||
|
|
||||||
|
if (!root) { |
||||||
|
return this.suggestions |
||||||
|
} // cannot suggest anything
|
||||||
|
|
||||||
|
const children = root.children |
||||||
|
|
||||||
|
let spread = 0 |
||||||
|
for (const letter in children) { |
||||||
|
this.traverse(children[letter]) |
||||||
|
spread++ |
||||||
|
|
||||||
|
if (CHILDREN && spread === CHILDREN) { |
||||||
|
break |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return this.suggestions |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue