summaryrefslogtreecommitdiff
path: root/services/AppSearch.qml
blob: d56213f3c4015c868af626b33d99d2ae230127c1 (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;
    }
}