Moving fuzzy-native to core

package.json

@@ -30,6 +30,7 @@
   "dependencies": {
     "@atom/source-map-support": "^0.3.4",
     "@babel/core": "7.18.6",
+    "@pulsar-edit/fuzzy-native": "https://github.com/pulsar-edit/fuzzy-native.git#13d2e5f",
     "about": "file:packages/about",
     "archive-view": "file:packages/archive-view",
     "async": "3.2.4",

packages/autocomplete-plus/lib/autocomplete-manager.js

@@ -1,7 +1,5 @@
 const {CompositeDisposable, Disposable, Point, Range} = require('atom')
 const path = require('path')
-const fuzzaldrin = require('fuzzaldrin')
-const fuzzaldrinPlus = require('fuzzaldrin-plus')
 
 const ProviderManager = require('./provider-manager')
 const SuggestionList = require('./suggestion-list')
@@ -190,7 +188,6 @@ class AutocompleteManager {
     this.subscriptions.add(atom.config.observe('autocomplete-plus.enableAutoActivation', (value) => { this.autoActivationEnabled = value }))
     this.subscriptions.add(atom.config.observe('autocomplete-plus.enableAutoConfirmSingleSuggestion', (value) => { this.autoConfirmSingleSuggestionEnabled = value }))
     this.subscriptions.add(atom.config.observe('autocomplete-plus.consumeSuffix', (value) => { this.consumeSuffix = value }))
-    this.subscriptions.add(atom.config.observe('autocomplete-plus.useAlternateScoring', (value) => { this.useAlternateScoring = value }))
     this.subscriptions.add(atom.config.observe('autocomplete-plus.fileBlacklist', (value) => {
       if (value) {
         this.fileBlacklist = value.map((s) => { return s.trim() })
@@ -365,7 +362,6 @@ class AutocompleteManager {
 
   filterSuggestions (suggestions, {prefix}) {
     const results = []
-    const fuzzaldrinProvider = this.useAlternateScoring ? fuzzaldrinPlus : fuzzaldrin
     for (let i = 0; i < suggestions.length; i++) {
       // sortScore mostly preserves in the original sorting. The function is
       // chosen such that suggestions with a very high match score can break out.
@@ -382,7 +378,7 @@ class AutocompleteManager {
       if (prefixIsEmpty) {
         results.push(suggestion)
       }
-      if (firstCharIsMatch && (score = fuzzaldrinProvider.score(text, suggestionPrefix)) > 0) {
+      if (firstCharIsMatch && (score = atom.ui.fuzzyMatcher.score(text, suggestionPrefix)) > 0) {
         suggestion.score = score * suggestion.sortScore
         results.push(suggestion)
       }

packages/autocomplete-plus/lib/suggestion-list-element.js

@@ -1,7 +1,6 @@
 const {CompositeDisposable} = require('atom')
 const SnippetParser = require('./snippet-parser')
 const {isString} = require('./type-helpers')
-const fuzzaldrinPlus = require('fuzzaldrin-plus')
 const {marked} = require('marked')
 const createDOMPurify = require('dompurify')
 
@@ -83,9 +82,6 @@ module.exports = class SuggestionListElement {
     this.subscriptions.add(atom.config.observe('autocomplete-plus.maxVisibleSuggestions', maxVisibleSuggestions => {
       this.maxVisibleSuggestions = maxVisibleSuggestions
     }))
-    this.subscriptions.add(atom.config.observe('autocomplete-plus.useAlternateScoring', useAlternateScoring => {
-      this.useAlternateScoring = useAlternateScoring
-    }))
     this.subscriptions.add(atom.config.observe('autocomplete-plus.moveToCancel', moveToCancel => {
       this.moveToCancel = moveToCancel
     }))
@@ -544,12 +540,8 @@ module.exports = class SuggestionListElement {
 
     if (!characterMatchIndices) {
       characterMatchIndices = this.findCharacterMatchIndices(replacementText, replacementPrefix)
-    } else {
-      characterMatchIndices = characterMatchIndices.reduce((matches, index) => {
-        matches[index] = true
-        return matches
-      }, {})
     }
+    characterMatchIndices = new Set(characterMatchIndices);
 
     const appendNonMatchChars = (el, nonMatchChars) => {
       if (nonMatchChars) {
@@ -572,7 +564,7 @@ module.exports = class SuggestionListElement {
         workingEl = s
       }
 
-      if (characterMatchIndices && characterMatchIndices[index]) {
+      if (characterMatchIndices && characterMatchIndices.has(index)) {
         appendNonMatchChars(workingEl, nonMatchChars)
         nonMatchChars = ''
 
@@ -668,26 +660,11 @@ module.exports = class SuggestionListElement {
   //
   // Returns an {Object}
   findCharacterMatchIndices (text, replacementPrefix) {
-    if (!text || !text.length || !replacementPrefix || !replacementPrefix.length) { return }
+    if (!text?.length || !replacementPrefix?.length) { return }
     const matches = {}
-    if (this.useAlternateScoring) {
-      const matchIndices = fuzzaldrinPlus.match(text, replacementPrefix)
-      for (const i of matchIndices) {
-        matches[i] = true
-      }
-    } else {
-      let wordIndex = 0
-      for (let i = 0; i < replacementPrefix.length; i++) {
-        const ch = replacementPrefix[i]
-        while (wordIndex < text.length && text[wordIndex].toLowerCase() !== ch.toLowerCase()) {
-          wordIndex += 1
-        }
-        if (wordIndex >= text.length) { break }
-        matches[wordIndex] = true
-        wordIndex += 1
-      }
-    }
-    return matches
+    return atom.ui.fuzzyMatcher.match(
+      text, replacementPrefix, {recordMatchIndexes: true}
+    )?.matchIndexes || [];
   }
 
   dispose () {

packages/autocomplete-plus/package.json

@@ -11,8 +11,6 @@
   "dependencies": {
     "atom-slick": "^2.0.0",
     "dompurify": "^3.0.4",
-    "fuzzaldrin": "^2.1.0",
-    "fuzzaldrin-plus": "^0.6.0",
     "grim": "^2.0.1",
     "marked": "^5.1.1",
     "minimatch": "^3.0.3",
@@ -200,12 +198,6 @@
       "default": true,
       "order": 18
     },
-    "useAlternateScoring": {
-      "description": "Prefers runs of consecutive characters, acronyms and start of words. (Experimental)",
-      "type": "boolean",
-      "default": true,
-      "order": 19
-    },
     "useLocalityBonus": {
       "description": "Gives words near the cursor position a higher score than those far away",
       "type": "boolean",

packages/autocomplete-plus/spec/suggestion-list-element-spec.js

@@ -1,6 +1,7 @@
 /* eslint-env jasmine */
 /* eslint-disable no-template-curly-in-string */
 const SuggestionListElement = require('../lib/suggestion-list-element')
+const { conditionPromise } = require('./spec-helper')
 
 const fragmentToHtml = fragment => {
   const el = document.createElement('span')
@@ -63,11 +64,12 @@ describe('Suggestion List Element', () => {
   })
 
   describe('itemChanged', () => {
-    beforeEach(() => jasmine.attachToDOM(suggestionListElement.element))
-
-    it('updates the list item', async () => {
+    beforeEach(() => {
       jasmine.useRealClock()
+      jasmine.attachToDOM(suggestionListElement.element)
+    })
 
+    it('updates the list item', async () => {
       const suggestion = {text: 'foo'}
       const newSuggestion = {text: 'foo', description: 'foobar', rightLabel: 'foo'}
       suggestionListElement.model = {items: [newSuggestion]}
@@ -77,6 +79,9 @@ describe('Suggestion List Element', () => {
 
       suggestionListElement.itemChanged({suggestion: newSuggestion, index: 0})
 
+      await conditionPromise(() =>
+        suggestionListElement.element.querySelector('.right-label').innerText
+      )
       expect(suggestionListElement.element.querySelector('.right-label').innerText)
         .toBe('foo')
 
@@ -190,14 +195,15 @@ describe('Suggestion List Element', () => {
       let snippets = suggestionListElement.snippetParser.findSnippets(text)
       text = suggestionListElement.removeSnippetsFromText(snippets, text)
       let matches = suggestionListElement.findCharacterMatchIndices(text, replacementPrefix)
+      matches = new Set(matches)
 
       for (var i = 0; i <= text.length; i++) {
         if (truthyIndices.indexOf(i) !== -1) {
-          expect(matches[i]).toBeTruthy()
+          expect(matches.has(i)).toBeTruthy()
         } else {
           let m = matches
           if (m) {
-            m = m[i]
+            m = m.has(i)
           }
           expect(m).toBeFalsy()
         }
@@ -208,14 +214,14 @@ describe('Suggestion List Element', () => {
       assertMatches('hello', '', [])
       assertMatches('hello', 'h', [0])
       assertMatches('hello', 'hl', [0, 2])
-      assertMatches('hello', 'hlo', [0, 2, 4])
+      assertMatches('hello', 'hlo', [0, 3, 4])
     })
 
     it('finds matches when snippets exist', () => {
       assertMatches('${0:hello}', '', [])
       assertMatches('${0:hello}', 'h', [0])
       assertMatches('${0:hello}', 'hl', [0, 2])
-      assertMatches('${0:hello}', 'hlo', [0, 2, 4])
+      assertMatches('${0:hello}', 'hlo', [0, 3, 4])
       assertMatches('${0:hello}world', '', [])
       assertMatches('${0:hello}world', 'h', [0])
       assertMatches('${0:hello}world', 'hw', [0, 5])

packages/command-palette/lib/command-palette-package.js

@@ -15,9 +15,6 @@ class CommandPalettePackage {
         this.commandPaletteView.show(true)
       }
     }))
-    this.disposables.add(atom.config.observe('command-palette.useAlternateScoring', (newValue) => {
-      this.commandPaletteView.update({useAlternateScoring: newValue})
-    }))
     this.disposables.add(atom.config.observe('command-palette.preserveLastSearch', (newValue) => {
       this.commandPaletteView.update({preserveLastSearch: newValue})
     }))

packages/command-palette/lib/command-palette-view.js

@@ -2,8 +2,6 @@
 
 import SelectListView from 'atom-select-list'
 import {humanizeKeystroke} from 'underscore-plus'
-import fuzzaldrin from 'fuzzaldrin'
-import fuzzaldrinPlus from 'fuzzaldrin-plus'
 
 export default class CommandPaletteView {
   constructor (initiallyVisibleItemCount = 10) {
@@ -55,7 +53,7 @@ export default class CommandPaletteView {
 
           if (Array.isArray(item.tags)) {
             const matchingTags = item.tags
-              .map(t => [t, this.fuzz.score(t, query)])
+              .map(t => [t, atom.ui.fuzzyMatcher.score(t, query)])
               .filter(([t, s]) => s > 0)
               .sort((a, b) => a.s - b.s)
               .map(([t, s]) => t)
@@ -132,21 +130,13 @@ export default class CommandPaletteView {
     if (props.hasOwnProperty('preserveLastSearch')) {
       this.preserveLastSearch = props.preserveLastSearch
     }
-
-    if (props.hasOwnProperty('useAlternateScoring')) {
-      this.useAlternateScoring = props.useAlternateScoring
-    }
-  }
-
-  get fuzz () {
-    return this.useAlternateScoring ? fuzzaldrinPlus : fuzzaldrin
   }
 
   highlightMatchesInElement (text, query, el) {
-    const matches = this.fuzz.match(text, query)
+    const matches = atom.ui.fuzzyMatcher.match(text, query, {recordMatchIndexes: true})
     let matchedChars = []
     let lastIndex = 0
-    for (const matchIndex of matches) {
+    matches.matchIndexes.forEach(matchIndex => {
       const unmatched = text.substring(lastIndex, matchIndex)
       if (unmatched) {
         if (matchedChars.length > 0) {
@@ -162,7 +152,7 @@ export default class CommandPaletteView {
 
       matchedChars.push(text[matchIndex])
       lastIndex = matchIndex + 1
-    }
+    })
 
     if (matchedChars.length > 0) {
       const matchSpan = document.createElement('span')
@@ -184,15 +174,15 @@ export default class CommandPaletteView {
 
     const scoredItems = []
     for (const item of items) {
-      let score = this.fuzz.score(item.displayName, query)
+      let score = atom.ui.fuzzyMatcher.score(item.displayName, query)
       if (item.tags) {
         score += item.tags.reduce(
-          (currentScore, tag) => currentScore + this.fuzz.score(tag, query),
+          (currentScore, tag) => currentScore + atom.ui.fuzzyMatcher.score(tag, query),
           0
         )
       }
       if (item.description) {
-        score += this.fuzz.score(item.description, query)
+        score += atom.ui.fuzzyMatcher.score(item.description, query)
       }
 
       if (score > 0) {

packages/command-palette/package.json

@@ -16,8 +16,6 @@
   "atomTestRunner": "atom-mocha-test-runner",
   "dependencies": {
     "atom-select-list": "^0.7.1",
-    "fuzzaldrin": "^2.1.0",
-    "fuzzaldrin-plus": "^0.6.0",
     "underscore-plus": "^1.0.0"
   },
   "devDependencies": {
@@ -27,11 +25,6 @@
     "sinon": "^3.2.1"
   },
   "configSchema": {
-    "useAlternateScoring": {
-      "type": "boolean",
-      "default": true,
-      "description": "Use an alternative scoring approach which prefers run of consecutive characters, acronyms and start of words."
-    },
     "preserveLastSearch": {
       "type": "boolean",
       "default": false,

packages/fuzzy-finder/lib/fuzzy-finder-view.js

@@ -1,6 +1,5 @@
 const {Point, CompositeDisposable} = require('atom')
 const fs = require('fs')
-const NativeFuzzy = require('@pulsar-edit/fuzzy-native')
 
 const path = require('path')
 const SelectListView = require('atom-select-list')
@@ -79,7 +78,9 @@ module.exports = class FuzzyFinderView {
       elementForItem: ({filePath, label, ownerGitHubUsername}) => {
         const filterQuery = this.selectListView.getFilterQuery()
 
-        this.nativeFuzzyForResults.setCandidates([0], [label])
+        atom.ui.fuzzyMatcher.setCandidates(
+          this.nativeFuzzyForResults, [label]
+        );
         const items = this.nativeFuzzyForResults.match(
           filterQuery,
           {maxResults: 1, recordMatchIndexes: true}
@@ -125,14 +126,13 @@ module.exports = class FuzzyFinderView {
     })
 
     if (!this.nativeFuzzy) {
-      this.nativeFuzzy = new NativeFuzzy.Matcher(
-        indexArray(this.items.length),
+      this.nativeFuzzy = atom.ui.fuzzyMatcher.setCandidates(
         this.items.map(el => el.label)
-      )
+      );
       // We need a separate instance of the fuzzy finder to calculate the
       // matched paths only for the returned results. This speeds up considerably
       // the filtering of items.
-      this.nativeFuzzyForResults = new NativeFuzzy.Matcher([], [])
+      this.nativeFuzzyForResults = atom.ui.fuzzyMatcher.setCandidates([]);
     }
     this.selectListView.update({ filter: this.filterFn })
   }
@@ -290,10 +290,10 @@ module.exports = class FuzzyFinderView {
 
   setItems (items) {
     this.items = items
-    this.nativeFuzzy.setCandidates(
-      indexArray(this.items.length),
+    atom.ui.fuzzyMatcher.setCandidates(
+      this.nativeFuzzy,
       this.items.map(item => item.label)
-    )
+    );
 
     if (this.isQueryALineJump()) {
       this.selectListView.update({
@@ -382,14 +382,6 @@ function highlight (path, matches, offsetIndex) {
   return fragment
 }
 
-function indexArray (length) {
-  const array = []
-  for (let i = 0; i < length; i++) {
-    array[i] = i
-  }
-  return array
-}
-
 class FuzzyFinderItem {
   constructor ({filePath, label, ownerGitHubUsername, filterQuery, matches, repository}) {
     this.filePath = filePath