diff options
| author | littleblack111 <littleblack11111@gmail.com> | 2025-07-11 21:49:53 +0800 |
|---|---|---|
| committer | littleblack111 <littleblack11111@gmail.com> | 2025-07-11 21:49:53 +0800 |
| commit | 0e4bdc7ed815ef389d9ca95fc4865f46701b1aaf (patch) | |
| tree | 09c5e1609a4e3526416ae6bd294fb003d58ec70a | |
| parent | bf44811dced54246317c51183b2c77b551fafb79 (diff) | |
activewindow
| -rw-r--r-- | bar/ActiveWindow.qml | 23 | ||||
| -rw-r--r-- | bar/Bar.qml | 8 | ||||
| -rw-r--r-- | bar/Workspace.qml | 12 | ||||
| -rw-r--r-- | config/Bar.qml | 13 | ||||
| -rw-r--r-- | config/General.qml | 9 | ||||
| -rw-r--r-- | config/Style.qml | 5 | ||||
| -rw-r--r-- | services/SHyprland.qml | 38 | ||||
| -rw-r--r-- | utils/AppSearch.qml | 118 | ||||
| -rw-r--r-- | utils/fuzzysort.js | 692 | ||||
| -rw-r--r-- | utils/levendist.js | 142 |
10 files changed, 1000 insertions, 60 deletions
diff --git a/bar/ActiveWindow.qml b/bar/ActiveWindow.qml new file mode 100644 index 0000000..efdbfd9 --- /dev/null +++ b/bar/ActiveWindow.qml @@ -0,0 +1,23 @@ +import QtQuick +import Quickshell +import Quickshell.Wayland +import Quickshell.Widgets +import qs.utils + +Item { + id: root + property var toplevel: ToplevelManager.activeToplevel + property var icon: Quickshell.iconPath(AppSearch.guessIcon(toplevel?.appId), "image-missing") + + IconImage { + source: root.icon + implicitWidth: 32 + implicitHeight: 32 + anchors.verticalCenter: parent.verticalCenter + anchors.left: parent.left + anchors.leftMargin: 8 + } + Text { + text: root.toplevel?.title + } +} diff --git a/bar/Bar.qml b/bar/Bar.qml index 8bef666..0dc49f5 100644 --- a/bar/Bar.qml +++ b/bar/Bar.qml @@ -43,7 +43,7 @@ Scope { id: leftRow Layout.preferredWidth: 1 Layout.fillWidth: true - spacing: 40 + spacing: 0 // Layout.alignment: Qt.AlignLeft | Qt.AlignVCenter Workspace { @@ -64,11 +64,7 @@ Scope { Layout.fillWidth: true Layout.fillHeight: true Layout.preferredWidth: 1 - Rectangle { - Layout.fillWidth: true - implicitHeight: 1 - color: "#000000" - } + ActiveWindow {} } RowLayout { id: rightRow diff --git a/bar/Workspace.qml b/bar/Workspace.qml index de63eba..04ad537 100644 --- a/bar/Workspace.qml +++ b/bar/Workspace.qml @@ -1,7 +1,7 @@ import QtQuick.Layouts import QtQuick import Quickshell -import qs.services +import Quickshell.Hyprland import qs.config Item { @@ -11,23 +11,23 @@ Item { function updateWorkspaceOccupied() { // workspaceOccupied = Array.from({ length: Config.options.bar.workspaces.shown }, (_, i) => { - // return SHyprland.workspaces.values.some(ws => ws.id === workspaceGroup * Config.options.bar.workspaces.shown + i + 1); + // return Hyprland.workspaces.values.some(ws => ws.id === workspaceGroup * Config.options.bar.workspaces.shown + i + 1); // }) workspaceOccupied = Array.from({ length: Bar.workspaces }, (_, i) => { - return SHyprland.workspaces.values.some(ws => ws.id === (i + 1)); + return Hyprland.workspaces.values.some(ws => ws?.id === (i + 1)); }); } function isActive(index: int): bool { - SHyprland.focusedWorkspace.id.toString() == (index + 1).toString(); + Hyprland.focusedWorkspace?.id.toString() == (index + 1).toString(); } function isOccupied(index: int): bool { return workspaceOccupied[index] === true; } Connections { - target: SHyprland.workspaces + target: Hyprland.workspaces function onValuesChanged() { updateWorkspaceOccupied(); } @@ -47,7 +47,7 @@ Item { Rectangle { Text { - text: SHyprland.focusedWorkspace?.id.toString() == (index + 1).toString() + text: Hyprland.focusedWorkspace?.id.toString() == (index + 1).toString() // text: isActive(index) } implicitWidth: 10 diff --git a/config/Bar.qml b/config/Bar.qml index a851efc..5a5be3a 100644 --- a/config/Bar.qml +++ b/config/Bar.qml @@ -1,16 +1,15 @@ pragma Singleton import Quickshell -import QtQuick Singleton { id: root - readonly property int height: 45 - readonly property int borderRadius: Style.rounding.small - readonly property int topMargin: Style.spacing.small - readonly property int leftMargin: Style.spacing.larger - readonly property int rightMargin: Style.spacing.larger + readonly property int height: 45 + readonly property int borderRadius: Style.rounding.small + readonly property int topMargin: Style.spacing.small + readonly property int leftMargin: Style.spacing.larger + readonly property int rightMargin: Style.spacing.larger - readonly property int workspaces: 10 + readonly property int workspaces: 10 } diff --git a/config/General.qml b/config/General.qml new file mode 100644 index 0000000..80df622 --- /dev/null +++ b/config/General.qml @@ -0,0 +1,9 @@ +pragma Singleton + +import Quickshell + +Singleton { + id: root + + readonly property bool sloopySearch: false +} diff --git a/config/Style.qml b/config/Style.qml index c867ee9..dc2a0ca 100644 --- a/config/Style.qml +++ b/config/Style.qml @@ -36,8 +36,8 @@ Singleton { } component FontFamily: QtObject { - readonly property string symbols: "Symbols Nerd Font" - readonly property string iosevka: "Iosevka Nerd Font" + readonly property string symbols: "Symbols Nerd Font" + readonly property string iosevka: "Iosevka Nerd Font" readonly property string mono: "JetBrains Mono" } @@ -82,4 +82,3 @@ Singleton { readonly property AnimDurations durations: AnimDurations {} } } - diff --git a/services/SHyprland.qml b/services/SHyprland.qml deleted file mode 100644 index dfbc874..0000000 --- a/services/SHyprland.qml +++ /dev/null @@ -1,38 +0,0 @@ -pragma Singleton - -import Quickshell -import Quickshell.Hyprland -import QtQuick -import qs.config - -Singleton { - id: root - - readonly property var toplevels: Hyprland.toplevels - readonly property var workspaces: Hyprland.workspaces - readonly property var monitors: Hyprland.monitors - property var ws: ObjectModel - // readonly property HyprlandToplevel activeToplevel: Hyprland.activeToplevel - readonly property HyprlandWorkspace focusedWorkspace: Hyprland.focusedWorkspace - readonly property HyprlandMonitor focusedMonitor: Hyprland.focusedMonitor - - function dispatch(request: string): void { - Hyprland.dispatch(request); - } - - Connections { - target: Hyprland - - function onRawEvent(event: HyprlandEvent): void { - if (event.name.endsWith("v2")) - return; - - if (event.name.includes("mon")) - Hyprland.refreshMonitors(); - else if (event.name.includes("workspace")) - Hyprland.refreshWorkspaces(); - // else - // Hyprland.refreshToplevels(); - } - } -} diff --git a/utils/AppSearch.qml b/utils/AppSearch.qml new file mode 100644 index 0000000..6d1ad15 --- /dev/null +++ b/utils/AppSearch.qml @@ -0,0 +1,118 @@ +pragma Singleton + +import "fuzzysort.js" as Fuzzy +import "levendist.js" as Levendist +import Quickshell +import Quickshell.Io +import qs.config + +/** + * - Eases fuzzy searching for applications by name + * - Guesses icon name for window class name + */ +Singleton { + id: root + property bool sloppySearch: General.sloppySearch ?? false + property real scoreThreshold: 0.2 + property var substitutions: ({ + "code-url-handler": "visual-studio-code", + "Code": "visual-studio-code", + "gnome-tweaks": "org.gnome.tweaks", + "pavucontrol-qt": "pavucontrol", + "wps": "wps-office2019-kprometheus", + "wpsoffice": "wps-office2019-kprometheus", + "footclient": "foot", + "zen": "zen-browser", + "brave-browser": "brave-desktop" + }) + property var regexSubstitutions: [ + { + "regex": /^steam_app_(\d+)$/, + "replace": "steam_icon_$1" + }, + { + "regex": /Minecraft.*/, + "replace": "minecraft" + }, + { + "regex": /.*polkit.*/, + "replace": "system-lock-screen" + }, + { + "regex": /gcr.prompter/, + "replace": "system-lock-screen" + } + ] + + readonly property list<DesktopEntry> list: Array.from(DesktopEntries.applications.values).sort((a, b) => a.name.localeCompare(b.name)) + + readonly property var preppedNames: list.map(a => ({ + name: Fuzzy.prepare(`${a.name} `), + entry: a + })) + + function fuzzyQuery(search: string): var { // Idk why list<DesktopEntry> doesn't work + if (root.sloppySearch) { + const results = list.map(obj => ({ + entry: obj, + score: Levendist.computeScore(obj.name.toLowerCase(), search.toLowerCase()) + })).filter(item => item.score > root.scoreThreshold).sort((a, b) => b.score - a.score); + return results.map(item => item.entry); + } + + return Fuzzy.go(search, preppedNames, { + all: true, + key: "name" + }).map(r => { + return r.obj.entry; + }); + } + + function iconExists(iconName) { + if (!iconName || iconName.length == 0) + return false; + return (Quickshell.iconPath(iconName, true).length > 0) && !iconName.includes("image-missing"); + } + + function guessIcon(str) { + if (!str || str.length == 0) + return "image-missing"; + + // Normal substitutions + if (substitutions[str]) + return substitutions[str]; + + // Regex substitutions + for (let i = 0; i < regexSubstitutions.length; i++) { + const substitution = regexSubstitutions[i]; + const replacedName = str.replace(substitution.regex, substitution.replace); + if (replacedName != str) + return replacedName; + } + + // If it gets detected normally, no need to guess + if (iconExists(str)) + return str; + + let guessStr = str; + // Guess: Take only app name of reverse domain name notation + guessStr = str.split('.').slice(-1)[0].toLowerCase(); + if (iconExists(guessStr)) + return guessStr; + // Guess: normalize to kebab case + guessStr = str.toLowerCase().replace(/\s+/g, "-"); + if (iconExists(guessStr)) + return guessStr; + // Guess: First fuzzy desktop entry match + const searchResults = root.fuzzyQuery(str); + if (searchResults.length > 0) { + const firstEntry = searchResults[0]; + guessStr = firstEntry.icon; + if (iconExists(guessStr)) + return guessStr; + } + + // Give up + return str; + } +} diff --git a/utils/fuzzysort.js b/utils/fuzzysort.js new file mode 100644 index 0000000..0e373ce --- /dev/null +++ b/utils/fuzzysort.js @@ -0,0 +1,692 @@ +.pragma library + +// https://github.com/farzher/fuzzysort +// License: MIT | Copyright (c) 2018 Stephen Kamenar +// A copy of the license is available in the `licenses` folder of this repository + +var single = (search, target) => { + if (!search || !target) return NULL + + var preparedSearch = getPreparedSearch(search) + if (!isPrepared(target)) target = getPrepared(target) + + var searchBitflags = preparedSearch.bitflags + if ((searchBitflags & target._bitflags) !== searchBitflags) return NULL + + return algorithm(preparedSearch, target) +} + +var go = (search, targets, options) => { + if (!search) return options?.all ? all(targets, options) : noResults + + var preparedSearch = getPreparedSearch(search) + var searchBitflags = preparedSearch.bitflags + var containsSpace = preparedSearch.containsSpace + + var threshold = denormalizeScore(options?.threshold || 0) + var limit = options?.limit || INFINITY + + var resultsLen = 0; var limitedCount = 0 + var targetsLen = targets.length + + function push_result(result) { + if (resultsLen < limit) { q.add(result); ++resultsLen } + else { + ++limitedCount + if (result._score > q.peek()._score) q.replaceTop(result) + } + } + + // This code is copy/pasted 3 times for performance reasons [options.key, options.keys, no keys] + + // options.key + if (options?.key) { + var key = options.key + for (var i = 0; i < targetsLen; ++i) { + var obj = targets[i] + var target = getValue(obj, key) + if (!target) continue + if (!isPrepared(target)) target = getPrepared(target) + + if ((searchBitflags & target._bitflags) !== searchBitflags) continue + var result = algorithm(preparedSearch, target) + if (result === NULL) continue + if (result._score < threshold) continue + + result.obj = obj + push_result(result) + } + + // options.keys + } else if (options?.keys) { + var keys = options.keys + var keysLen = keys.length + + outer: for (var i = 0; i < targetsLen; ++i) { + var obj = targets[i] + + { // early out based on bitflags + var keysBitflags = 0 + for (var keyI = 0; keyI < keysLen; ++keyI) { + var key = keys[keyI] + var target = getValue(obj, key) + if (!target) { tmpTargets[keyI] = noTarget; continue } + if (!isPrepared(target)) target = getPrepared(target) + tmpTargets[keyI] = target + + keysBitflags |= target._bitflags + } + + if ((searchBitflags & keysBitflags) !== searchBitflags) continue + } + + if (containsSpace) for (let i = 0; i < preparedSearch.spaceSearches.length; i++) keysSpacesBestScores[i] = NEGATIVE_INFINITY + + for (var keyI = 0; keyI < keysLen; ++keyI) { + target = tmpTargets[keyI] + if (target === noTarget) { tmpResults[keyI] = noTarget; continue } + + tmpResults[keyI] = algorithm(preparedSearch, target, /*allowSpaces=*/false, /*allowPartialMatch=*/containsSpace) + if (tmpResults[keyI] === NULL) { tmpResults[keyI] = noTarget; continue } + + // todo: this seems weird and wrong. like what if our first match wasn't good. this should just replace it instead of averaging with it + // if our second match isn't good we ignore it instead of averaging with it + if (containsSpace) for (let i = 0; i < preparedSearch.spaceSearches.length; i++) { + if (allowPartialMatchScores[i] > -1000) { + if (keysSpacesBestScores[i] > NEGATIVE_INFINITY) { + var tmp = (keysSpacesBestScores[i] + allowPartialMatchScores[i]) / 4/*bonus score for having multiple matches*/ + if (tmp > keysSpacesBestScores[i]) keysSpacesBestScores[i] = tmp + } + } + if (allowPartialMatchScores[i] > keysSpacesBestScores[i]) keysSpacesBestScores[i] = allowPartialMatchScores[i] + } + } + + if (containsSpace) { + for (let i = 0; i < preparedSearch.spaceSearches.length; i++) { if (keysSpacesBestScores[i] === NEGATIVE_INFINITY) continue outer } + } else { + var hasAtLeast1Match = false + for (let i = 0; i < keysLen; i++) { if (tmpResults[i]._score !== NEGATIVE_INFINITY) { hasAtLeast1Match = true; break } } + if (!hasAtLeast1Match) continue + } + + var objResults = new KeysResult(keysLen) + for (let i = 0; i < keysLen; i++) { objResults[i] = tmpResults[i] } + + if (containsSpace) { + var score = 0 + for (let i = 0; i < preparedSearch.spaceSearches.length; i++) score += keysSpacesBestScores[i] + } else { + // todo could rewrite this scoring to be more similar to when there's spaces + // if we match multiple keys give us bonus points + var score = NEGATIVE_INFINITY + for (let i = 0; i < keysLen; i++) { + var result = objResults[i] + if (result._score > -1000) { + if (score > NEGATIVE_INFINITY) { + var tmp = (score + result._score) / 4/*bonus score for having multiple matches*/ + if (tmp > score) score = tmp + } + } + if (result._score > score) score = result._score + } + } + + objResults.obj = obj + objResults._score = score + if (options?.scoreFn) { + score = options.scoreFn(objResults) + if (!score) continue + score = denormalizeScore(score) + objResults._score = score + } + + if (score < threshold) continue + push_result(objResults) + } + + // no keys + } else { + for (var i = 0; i < targetsLen; ++i) { + var target = targets[i] + if (!target) continue + if (!isPrepared(target)) target = getPrepared(target) + + if ((searchBitflags & target._bitflags) !== searchBitflags) continue + var result = algorithm(preparedSearch, target) + if (result === NULL) continue + if (result._score < threshold) continue + + push_result(result) + } + } + + if (resultsLen === 0) return noResults + var results = new Array(resultsLen) + for (var i = resultsLen - 1; i >= 0; --i) results[i] = q.poll() + results.total = resultsLen + limitedCount + return results +} + + +// this is written as 1 function instead of 2 for minification. perf seems fine ... +// except when minified. the perf is very slow +var highlight = (result, open = '<b>', close = '</b>') => { + var callback = typeof open === 'function' ? open : undefined + + var target = result.target + var targetLen = target.length + var indexes = result.indexes + var highlighted = '' + var matchI = 0 + var indexesI = 0 + var opened = false + var parts = [] + + for (var i = 0; i < targetLen; ++i) { + var char = target[i] + if (indexes[indexesI] === i) { + ++indexesI + if (!opened) { + opened = true + if (callback) { + parts.push(highlighted); highlighted = '' + } else { + highlighted += open + } + } + + if (indexesI === indexes.length) { + if (callback) { + highlighted += char + parts.push(callback(highlighted, matchI++)); highlighted = '' + parts.push(target.substr(i + 1)) + } else { + highlighted += char + close + target.substr(i + 1) + } + break + } + } else { + if (opened) { + opened = false + if (callback) { + parts.push(callback(highlighted, matchI++)); highlighted = '' + } else { + highlighted += close + } + } + } + highlighted += char + } + + return callback ? parts : highlighted +} + + +var prepare = (target) => { + if (typeof target === 'number') target = '' + target + else if (typeof target !== 'string') target = '' + var info = prepareLowerInfo(target) + return new_result(target, { _targetLower: info._lower, _targetLowerCodes: info.lowerCodes, _bitflags: info.bitflags }) +} + +var cleanup = () => { preparedCache.clear(); preparedSearchCache.clear() } + + +// Below this point is only internal code +// Below this point is only internal code +// Below this point is only internal code +// Below this point is only internal code + + +class Result { + get ['indexes']() { return this._indexes.slice(0, this._indexes.len).sort((a, b) => a - b) } + set ['indexes'](indexes) { return this._indexes = indexes } + ['highlight'](open, close) { return highlight(this, open, close) } + get ['score']() { return normalizeScore(this._score) } + set ['score'](score) { this._score = denormalizeScore(score) } +} + +class KeysResult extends Array { + get ['score']() { return normalizeScore(this._score) } + set ['score'](score) { this._score = denormalizeScore(score) } +} + +var new_result = (target, options) => { + const result = new Result() + result['target'] = target + result['obj'] = options.obj ?? NULL + result._score = options._score ?? NEGATIVE_INFINITY + result._indexes = options._indexes ?? [] + result._targetLower = options._targetLower ?? '' + result._targetLowerCodes = options._targetLowerCodes ?? NULL + result._nextBeginningIndexes = options._nextBeginningIndexes ?? NULL + result._bitflags = options._bitflags ?? 0 + return result +} + + +var normalizeScore = score => { + if (score === NEGATIVE_INFINITY) return 0 + if (score > 1) return score + return Math.E ** (((-score + 1) ** .04307 - 1) * -2) +} +var denormalizeScore = normalizedScore => { + if (normalizedScore === 0) return NEGATIVE_INFINITY + if (normalizedScore > 1) return normalizedScore + return 1 - Math.pow((Math.log(normalizedScore) / -2 + 1), 1 / 0.04307) +} + + +var prepareSearch = (search) => { + if (typeof search === 'number') search = '' + search + else if (typeof search !== 'string') search = '' + search = search.trim() + var info = prepareLowerInfo(search) + + var spaceSearches = [] + if (info.containsSpace) { + var searches = search.split(/\s+/) + searches = [...new Set(searches)] // distinct + for (var i = 0; i < searches.length; i++) { + if (searches[i] === '') continue + var _info = prepareLowerInfo(searches[i]) + spaceSearches.push({ lowerCodes: _info.lowerCodes, _lower: searches[i].toLowerCase(), containsSpace: false }) + } + } + + return { lowerCodes: info.lowerCodes, _lower: info._lower, containsSpace: info.containsSpace, bitflags: info.bitflags, spaceSearches: spaceSearches } +} + + + +var getPrepared = (target) => { + if (target.length > 999) return prepare(target) // don't cache huge targets + var targetPrepared = preparedCache.get(target) + if (targetPrepared !== undefined) return targetPrepared + targetPrepared = prepare(target) + preparedCache.set(target, targetPrepared) + return targetPrepared +} +var getPreparedSearch = (search) => { + if (search.length > 999) return prepareSearch(search) // don't cache huge searches + var searchPrepared = preparedSearchCache.get(search) + if (searchPrepared !== undefined) return searchPrepared + searchPrepared = prepareSearch(search) + preparedSearchCache.set(search, searchPrepared) + return searchPrepared +} + + +var all = (targets, options) => { + var results = []; results.total = targets.length // this total can be wrong if some targets are skipped + + var limit = options?.limit || INFINITY + + if (options?.key) { + for (var i = 0; i < targets.length; i++) { + var obj = targets[i] + var target = getValue(obj, options.key) + if (target == NULL) continue + if (!isPrepared(target)) target = getPrepared(target) + var result = new_result(target.target, { _score: target._score, obj: obj }) + results.push(result); if (results.length >= limit) return results + } + } else if (options?.keys) { + for (var i = 0; i < targets.length; i++) { + var obj = targets[i] + var objResults = new KeysResult(options.keys.length) + for (var keyI = options.keys.length - 1; keyI >= 0; --keyI) { + var target = getValue(obj, options.keys[keyI]) + if (!target) { objResults[keyI] = noTarget; continue } + if (!isPrepared(target)) target = getPrepared(target) + target._score = NEGATIVE_INFINITY + target._indexes.len = 0 + objResults[keyI] = target + } + objResults.obj = obj + objResults._score = NEGATIVE_INFINITY + results.push(objResults); if (results.length >= limit) return results + } + } else { + for (var i = 0; i < targets.length; i++) { + var target = targets[i] + if (target == NULL) continue + if (!isPrepared(target)) target = getPrepared(target) + target._score = NEGATIVE_INFINITY + target._indexes.len = 0 + results.push(target); if (results.length >= limit) return results + } + } + + return results +} + + +var algorithm = (preparedSearch, prepared, allowSpaces = false, allowPartialMatch = false) => { + if (allowSpaces === false && preparedSearch.containsSpace) return algorithmSpaces(preparedSearch, prepared, allowPartialMatch) + + var searchLower = preparedSearch._lower + var searchLowerCodes = preparedSearch.lowerCodes + var searchLowerCode = searchLowerCodes[0] + var targetLowerCodes = prepared._targetLowerCodes + var searchLen = searchLowerCodes.length + var targetLen = targetLowerCodes.length + var searchI = 0 // where we at + var targetI = 0 // where you at + var matchesSimpleLen = 0 + + // very basic fuzzy match; to remove non-matching targets ASAP! + // walk through target. find sequential matches. + // if all chars aren't found then exit + for (; ;) { + var isMatch = searchLowerCode === targetLowerCodes[targetI] + if (isMatch) { + matchesSimple[matchesSimpleLen++] = targetI + ++searchI; if (searchI === searchLen) break + searchLowerCode = searchLowerCodes[searchI] + } + ++targetI; if (targetI >= targetLen) return NULL // Failed to find searchI + } + + var searchI = 0 + var successStrict = false + var matchesStrictLen = 0 + + var nextBeginningIndexes = prepared._nextBeginningIndexes + if (nextBeginningIndexes === NULL) nextBeginningIndexes = prepared._nextBeginningIndexes = prepareNextBeginningIndexes(prepared.target) + targetI = matchesSimple[0] === 0 ? 0 : nextBeginningIndexes[matchesSimple[0] - 1] + + // Our target string successfully matched all characters in sequence! + // Let's try a more advanced and strict test to improve the score + // only count it as a match if it's consecutive or a beginning character! + var backtrackCount = 0 + if (targetI !== targetLen) for (; ;) { + if (targetI >= targetLen) { + // We failed to find a good spot for this search char, go back to the previous search char and force it forward + if (searchI <= 0) break // We failed to push chars forward for a better match + + ++backtrackCount; if (backtrackCount > 200) break // exponential backtracking is taking too long, just give up and return a bad match + + --searchI + var lastMatch = matchesStrict[--matchesStrictLen] + targetI = nextBeginningIndexes[lastMatch] + + } else { + var isMatch = searchLowerCodes[searchI] === targetLowerCodes[targetI] + if (isMatch) { + matchesStrict[matchesStrictLen++] = targetI + ++searchI; if (searchI === searchLen) { successStrict = true; break } + ++targetI + } else { + targetI = nextBeginningIndexes[targetI] + } + } + } + + // check if it's a substring match + var substringIndex = searchLen <= 1 ? -1 : prepared._targetLower.indexOf(searchLower, matchesSimple[0]) // perf: this is slow + var isSubstring = !!~substringIndex + var isSubstringBeginning = !isSubstring ? false : substringIndex === 0 || prepared._nextBeginningIndexes[substringIndex - 1] === substringIndex + + // if it's a substring match but not at a beginning index, let's try to find a substring starting at a beginning index for a better score + if (isSubstring && !isSubstringBeginning) { + for (var i = 0; i < nextBeginningIndexes.length; i = nextBeginningIndexes[i]) { + if (i <= substringIndex) continue + + for (var s = 0; s < searchLen; s++) if (searchLowerCodes[s] !== prepared._targetLowerCodes[i + s]) break + if (s === searchLen) { substringIndex = i; isSubstringBeginning = true; break } + } + } + + // tally up the score & keep track of matches for highlighting later + // if it's a simple match, we'll switch to a substring match if a substring exists + // if it's a strict match, we'll switch to a substring match only if that's a better score + + var calculateScore = matches => { + var score = 0 + + var extraMatchGroupCount = 0 + for (var i = 1; i < searchLen; ++i) { + if (matches[i] - matches[i - 1] !== 1) { score -= matches[i]; ++extraMatchGroupCount } + } + var unmatchedDistance = matches[searchLen - 1] - matches[0] - (searchLen - 1) + + score -= (12 + unmatchedDistance) * extraMatchGroupCount // penality for more groups + + if (matches[0] !== 0) score -= matches[0] * matches[0] * .2 // penality for not starting near the beginning + + if (!successStrict) { + score *= 1000 + } else { + // successStrict on a target with too many beginning indexes loses points for being a bad target + var uniqueBeginningIndexes = 1 + for (var i = nextBeginningIndexes[0]; i < targetLen; i = nextBeginningIndexes[i]) ++uniqueBeginningIndexes + + if (uniqueBeginningIndexes > 24) score *= (uniqueBeginningIndexes - 24) * 10 // quite arbitrary numbers here ... + } + + score -= (targetLen - searchLen) / 2 // penality for longer targets + + if (isSubstring) score /= 1 + searchLen * searchLen * 1 // bonus for being a full substring + if (isSubstringBeginning) score /= 1 + searchLen * searchLen * 1 // bonus for substring starting on a beginningIndex + + score -= (targetLen - searchLen) / 2 // penality for longer targets + + return score + } + + if (!successStrict) { + if (isSubstring) for (var i = 0; i < searchLen; ++i) matchesSimple[i] = substringIndex + i // at this point it's safe to overwrite matchehsSimple with substr matches + var matchesBest = matchesSimple + var score = calculateScore(matchesBest) + } else { + if (isSubstringBeginning) { + for (var i = 0; i < searchLen; ++i) matchesSimple[i] = substringIndex + i // at this point it's safe to overwrite matchehsSimple with substr matches + var matchesBest = matchesSimple + var score = calculateScore(matchesSimple) + } else { + var matchesBest = matchesStrict + var score = calculateScore(matchesStrict) + } + } + + prepared._score = score + + for (var i = 0; i < searchLen; ++i) prepared._indexes[i] = matchesBest[i] + prepared._indexes.len = searchLen + + const result = new Result() + result.target = prepared.target + result._score = prepared._score + result._indexes = prepared._indexes + return result +} +var algorithmSpaces = (preparedSearch, target, allowPartialMatch) => { + var seen_indexes = new Set() + var score = 0 + var result = NULL + + var first_seen_index_last_search = 0 + var searches = preparedSearch.spaceSearches + var searchesLen = searches.length + var changeslen = 0 + + // Return _nextBeginningIndexes back to its normal state + var resetNextBeginningIndexes = () => { + for (let i = changeslen - 1; i >= 0; i--) target._nextBeginningIndexes[nextBeginningIndexesChanges[i * 2 + 0]] = nextBeginningIndexesChanges[i * 2 + 1] + } + + var hasAtLeast1Match = false + for (var i = 0; i < searchesLen; ++i) { + allowPartialMatchScores[i] = NEGATIVE_INFINITY + var search = searches[i] + + result = algorithm(search, target) + if (allowPartialMatch) { + if (result === NULL) continue + hasAtLeast1Match = true + } else { + if (result === NULL) { resetNextBeginningIndexes(); return NULL } + } + + // if not the last search, we need to mutate _nextBeginningIndexes for the next search + var isTheLastSearch = i === searchesLen - 1 + if (!isTheLastSearch) { + var indexes = result._indexes + + var indexesIsConsecutiveSubstring = true + for (let i = 0; i < indexes.len - 1; i++) { + if (indexes[i + 1] - indexes[i] !== 1) { + indexesIsConsecutiveSubstring = false; break; + } + } + + if (indexesIsConsecutiveSubstring) { + var newBeginningIndex = indexes[indexes.len - 1] + 1 + var toReplace = target._nextBeginningIndexes[newBeginningIndex - 1] + for (let i = newBeginningIndex - 1; i >= 0; i--) { + if (toReplace !== target._nextBeginningIndexes[i]) break + target._nextBeginningIndexes[i] = newBeginningIndex + nextBeginningIndexesChanges[changeslen * 2 + 0] = i + nextBeginningIndexesChanges[changeslen * 2 + 1] = toReplace + changeslen++ + } + } + } + + score += result._score / searchesLen + allowPartialMatchScores[i] = result._score / searchesLen + + // dock points based on order otherwise "c man" returns Manifest.cpp instead of CheatManager.h + if (result._indexes[0] < first_seen_index_last_search) { + score -= (first_seen_index_last_search - result._indexes[0]) * 2 + } + first_seen_index_last_search = result._indexes[0] + + for (var j = 0; j < result._indexes.len; ++j) seen_indexes.add(result._indexes[j]) + } + + if (allowPartialMatch && !hasAtLeast1Match) return NULL + + resetNextBeginningIndexes() + + // allows a search with spaces that's an exact substring to score well + var allowSpacesResult = algorithm(preparedSearch, target, /*allowSpaces=*/true) + if (allowSpacesResult !== NULL && allowSpacesResult._score > score) { + if (allowPartialMatch) { + for (var i = 0; i < searchesLen; ++i) { + allowPartialMatchScores[i] = allowSpacesResult._score / searchesLen + } + } + return allowSpacesResult + } + + if (allowPartialMatch) result = target + result._score = score + + var i = 0 + for (let index of seen_indexes) result._indexes[i++] = index + result._indexes.len = i + + return result +} + +// we use this instead of just .normalize('NFD').replace(/[\u0300-\u036f]/g, '') because that screws with japanese characters +var remove_accents = (str) => str.replace(/\p{Script=Latin}+/gu, match => match.normalize('NFD')).replace(/[\u0300-\u036f]/g, '') + +var prepareLowerInfo = (str) => { + str = remove_accents(str) + var strLen = str.length + var lower = str.toLowerCase() + var lowerCodes = [] // new Array(strLen) sparse array is too slow + var bitflags = 0 + var containsSpace = false // space isn't stored in bitflags because of how searching with a space works + + for (var i = 0; i < strLen; ++i) { + var lowerCode = lowerCodes[i] = lower.charCodeAt(i) + + if (lowerCode === 32) { + containsSpace = true + continue // it's important that we don't set any bitflags for space + } + + var bit = lowerCode >= 97 && lowerCode <= 122 ? lowerCode - 97 // alphabet + : lowerCode >= 48 && lowerCode <= 57 ? 26 // numbers + // 3 bits available + : lowerCode <= 127 ? 30 // other ascii + : 31 // other utf8 + bitflags |= 1 << bit + } + + return { lowerCodes: lowerCodes, bitflags: bitflags, containsSpace: containsSpace, _lower: lower } +} +var prepareBeginningIndexes = (target) => { + var targetLen = target.length + var beginningIndexes = []; var beginningIndexesLen = 0 + var wasUpper = false + var wasAlphanum = false + for (var i = 0; i < targetLen; ++i) { + var targetCode = target.charCodeAt(i) + var isUpper = targetCode >= 65 && targetCode <= 90 + var isAlphanum = isUpper || targetCode >= 97 && targetCode <= 122 || targetCode >= 48 && targetCode <= 57 + var isBeginning = isUpper && !wasUpper || !wasAlphanum || !isAlphanum + wasUpper = isUpper + wasAlphanum = isAlphanum + if (isBeginning) beginningIndexes[beginningIndexesLen++] = i + } + return beginningIndexes +} +var prepareNextBeginningIndexes = (target) => { + target = remove_accents(target) + var targetLen = target.length + var beginningIndexes = prepareBeginningIndexes(target) + var nextBeginningIndexes = [] // new Array(targetLen) sparse array is too slow + var lastIsBeginning = beginningIndexes[0] + var lastIsBeginningI = 0 + for (var i = 0; i < targetLen; ++i) { + if (lastIsBeginning > i) { + nextBeginningIndexes[i] = lastIsBeginning + } else { + lastIsBeginning = beginningIndexes[++lastIsBeginningI] + nextBeginningIndexes[i] = lastIsBeginning === undefined ? targetLen : lastIsBeginning + } + } + return nextBeginningIndexes +} + +var preparedCache = new Map() +var preparedSearchCache = new Map() + +// the theory behind these being globals is to reduce garbage collection by not making new arrays +var matchesSimple = []; var matchesStrict = [] +var nextBeginningIndexesChanges = [] // allows straw berry to match strawberry well, by modifying the end of a substring to be considered a beginning index for the rest of the search +var keysSpacesBestScores = []; var allowPartialMatchScores = [] +var tmpTargets = []; var tmpResults = [] + +// prop = 'key' 2.5ms optimized for this case, seems to be about as fast as direct obj[prop] +// prop = 'key1.key2' 10ms +// prop = ['key1', 'key2'] 27ms +// prop = obj => obj.tags.join() ??ms +var getValue = (obj, prop) => { + var tmp = obj[prop]; if (tmp !== undefined) return tmp + if (typeof prop === 'function') return prop(obj) // this should run first. but that makes string props slower + var segs = prop + if (!Array.isArray(prop)) segs = prop.split('.') + var len = segs.length + var i = -1 + while (obj && (++i < len)) obj = obj[segs[i]] + return obj +} + +var isPrepared = (x) => { return typeof x === 'object' && typeof x._bitflags === 'number' } +var INFINITY = Infinity; var NEGATIVE_INFINITY = -INFINITY +var noResults = []; noResults.total = 0 +var NULL = null + +var noTarget = prepare('') + +// Hacked version of https://github.com/lemire/FastPriorityQueue.js +var fastpriorityqueue = r => { var e = [], o = 0, a = {}, v = r => { for (var a = 0, v = e[a], c = 1; c < o;) { var s = c + 1; a = c, s < o && e[s]._score < e[c]._score && (a = s), e[a - 1 >> 1] = e[a], c = 1 + (a << 1) } for (var f = a - 1 >> 1; a > 0 && v._score < e[f]._score; f = (a = f) - 1 >> 1)e[a] = e[f]; e[a] = v }; return a.add = (r => { var a = o; e[o++] = r; for (var v = a - 1 >> 1; a > 0 && r._score < e[v]._score; v = (a = v) - 1 >> 1)e[a] = e[v]; e[a] = r }), a.poll = (r => { if (0 !== o) { var a = e[0]; return e[0] = e[--o], v(), a } }), a.peek = (r => { if (0 !== o) return e[0] }), a.replaceTop = (r => { e[0] = r, v() }), a } +var q = fastpriorityqueue() // reuse this + diff --git a/utils/levendist.js b/utils/levendist.js new file mode 100644 index 0000000..ed8a5d8 --- /dev/null +++ b/utils/levendist.js @@ -0,0 +1,142 @@ +// Original code from https://github.com/koeqaife/hyprland-material-you +// Original code license: GPLv3 +// Translated to Js from Cython with an LLM and reviewed + +function min3(a, b, c) { + return a < b && a < c ? a : b < c ? b : c; +} + +function max3(a, b, c) { + return a > b && a > c ? a : b > c ? b : c; +} + +function min2(a, b) { + return a < b ? a : b; +} + +function max2(a, b) { + return a > b ? a : b; +} + +function levenshteinDistance(s1, s2) { + let len1 = s1.length; + let len2 = s2.length; + + if (len1 === 0) return len2; + if (len2 === 0) return len1; + + if (len2 > len1) { + [s1, s2] = [s2, s1]; + [len1, len2] = [len2, len1]; + } + + let prev = new Array(len2 + 1); + let curr = new Array(len2 + 1); + + for (let j = 0; j <= len2; j++) { + prev[j] = j; + } + + for (let i = 1; i <= len1; i++) { + curr[0] = i; + for (let j = 1; j <= len2; j++) { + let cost = s1[i - 1] === s2[j - 1] ? 0 : 1; + curr[j] = min3(prev[j] + 1, curr[j - 1] + 1, prev[j - 1] + cost); + } + [prev, curr] = [curr, prev]; + } + + return prev[len2]; +} + +function partialRatio(shortS, longS) { + let lenS = shortS.length; + let lenL = longS.length; + let best = 0.0; + + if (lenS === 0) return 1.0; + + for (let i = 0; i <= lenL - lenS; i++) { + let sub = longS.slice(i, i + lenS); + let dist = levenshteinDistance(shortS, sub); + let score = 1.0 - (dist / lenS); + if (score > best) best = score; + } + + return best; +} + +function computeScore(s1, s2) { + if (s1 === s2) return 1.0; + + let dist = levenshteinDistance(s1, s2); + let maxLen = max2(s1.length, s2.length); + if (maxLen === 0) return 1.0; + + let full = 1.0 - (dist / maxLen); + let part = s1.length < s2.length ? partialRatio(s1, s2) : partialRatio(s2, s1); + + let score = 0.85 * full + 0.15 * part; + + if (s1 && s2 && s1[0] !== s2[0]) { + score -= 0.05; + } + + let lenDiff = Math.abs(s1.length - s2.length); + if (lenDiff >= 3) { + score -= 0.05 * lenDiff / maxLen; + } + + let commonPrefixLen = 0; + let minLen = min2(s1.length, s2.length); + for (let i = 0; i < minLen; i++) { + if (s1[i] === s2[i]) { + commonPrefixLen++; + } else { + break; + } + } + score += 0.02 * commonPrefixLen; + + if (s1.includes(s2) || s2.includes(s1)) { + score += 0.06; + } + + return Math.max(0.0, Math.min(1.0, score)); +} + +function computeTextMatchScore(s1, s2) { + if (s1 === s2) return 1.0; + + let dist = levenshteinDistance(s1, s2); + let maxLen = max2(s1.length, s2.length); + if (maxLen === 0) return 1.0; + + let full = 1.0 - (dist / maxLen); + let part = s1.length < s2.length ? partialRatio(s1, s2) : partialRatio(s2, s1); + + let score = 0.4 * full + 0.6 * part; + + let lenDiff = Math.abs(s1.length - s2.length); + if (lenDiff >= 10) { + score -= 0.02 * lenDiff / maxLen; + } + + let commonPrefixLen = 0; + let minLen = min2(s1.length, s2.length); + for (let i = 0; i < minLen; i++) { + if (s1[i] === s2[i]) { + commonPrefixLen++; + } else { + break; + } + } + score += 0.01 * commonPrefixLen; + + if (s1.includes(s2) || s2.includes(s1)) { + score += 0.2; + } + + return Math.max(0.0, Math.min(1.0, score)); +} + |
