summaryrefslogtreecommitdiff
path: root/services/AppSearch.qml
blob: b8294b6bb04281d88200901b563c659c653aaae6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
pragma Singleton

import Quickshell
import Quickshell.Io
import qs.config
import qs.components

Searchable {
	id: root

	list: Array.from(DesktopEntries.applications.values).sort((a, b) => a.name.localeCompare(b.name))
	scoreThreshold: General.appSearchFuzzySearchThreshold
	algorithm: Searchable.SearchAlgorithm.Levendist
	keys: ["name", "genericName", "execString", "keywords", "comment", "categories", "icon"]

	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"
		}
	]

	property var iconExistsCache: ({})
	property var guessIconCache: ({})

	function iconExists(iconName) {
		if (!iconName || iconName.length === 0) {
			return false;
		}
		if (root.iconExistsCache.hasOwnProperty(iconName)) {
			return root.iconExistsCache[iconName];
		}
		const exists = Quickshell.iconPath(iconName, true).length > 0 && !iconName.includes("image-missing");
		root.iconExistsCache[iconName] = exists;
		return exists;
	}

	function guessIcon(str) {
		if (!str || str.length === 0) {
			return "image-missing";
		}
		if (root.guessIconCache[str]) {
			return root.guessIconCache[str];
		}

		var result;

		if (substitutions[str]) {
			result = substitutions[str];
		} else {
			result = null;
			for (var i = 0; i < regexSubstitutions.length; i++) {
				var {
					regex,
					replace
				} = regexSubstitutions[i];
				var replaced = str.replace(regex, replace);
				if (replaced !== str) {
					result = replaced;
					break;
				}
			}
		}

		if (!result && root.iconExists(str)) {
			result = str;
		}

		if (!result) {
			var guessStr = str.split(".").slice(-1)[0].toLowerCase();
			if (root.iconExists(guessStr)) {
				result = guessStr;
			}
		}

		if (!result) {
			var kebab = str.toLowerCase().replace(/\s+/g, "-");
			if (root.iconExists(kebab)) {
				result = kebab;
			}
		}

		if (!result) {
			const matches = root.query(str);
			if (matches.length > 0) {
				const candidate = matches[0].icon;
				if (root.iconExists(candidate)) {
					result = candidate;
				}
			}
		}

		if (!result) {
			result = str;
		}

		root.guessIconCache[str] = result;
		return result;
	}
}