DankMaterialShell/Services/AppSearchService.qml

pragma Singleton
pragma ComponentBehavior: Bound

import QtQuick
import Quickshell
import Quickshell.Io
import Quickshell.Widgets
import "../Common/fuzzysort.js" as Fuzzy

Singleton {
    id: root
    
    property var applications: DesktopEntries.applications.values
    
    property var applicationsByName: {
        var byName = {}
        for (var i = 0; i < applications.length; i++) {
            var app = applications[i]
            byName[app.name.toLowerCase()] = app
        }
        return byName
    }
    
    property var applicationsByExec: {
        var byExec = {}
        for (var i = 0; i < applications.length; i++) {
            var app = applications[i]
            var execProp = app.execString || ""
            var cleanExec = execProp ? execProp.replace(/%[fFuU]/g, "").trim() : ""
            if (cleanExec) {
                byExec[cleanExec] = app
            }
        }
        return byExec
    }
    
    property var preppedApps: applications.map(app => ({
        name: Fuzzy.prepare(app.name || ""),
        comment: Fuzzy.prepare(app.comment || ""),
        entry: app
    }))
    
    function searchApplications(query) {
        if (!query || query.length === 0) {
            return applications
        }
        
        if (preppedApps.length === 0) {
            return []
        }
        
        var results = Fuzzy.go(query, preppedApps, {
            all: false,
            keys: ["name", "comment"],
            scoreFn: r => {
                var nameScore = r[0] ? r[0].score : 0
                var commentScore = r[1] ? r[1].score : 0
                var appName = r.obj.entry.name || ""
                var finalScore = 0
                
                if (nameScore > 0) {
                    var queryLower = query.toLowerCase()
                    var nameLower = appName.toLowerCase()
                    
                    if (nameLower === queryLower) {
                        finalScore = nameScore * 100
                    }
                    else if (nameLower.startsWith(queryLower)) {
                        finalScore = nameScore * 50
                    }
                    else if (nameLower.includes(" " + queryLower) || nameLower.includes(queryLower + " ") || nameLower.endsWith(" " + queryLower)) {
                        finalScore = nameScore * 25
                    }
                    else if (nameLower.includes(queryLower)) {
                        finalScore = nameScore * 10
                    }
                    else {
                        finalScore = nameScore * 2 + commentScore * 0.1
                    }
                } else {
                    finalScore = commentScore * 0.1
                }
                
                return finalScore
            },
            limit: 50
        })
        
        return results.map(r => r.obj.entry)
    }
    
    function getAppByName(name) {
        return applicationsByName[name.toLowerCase()] || null
    }
    
    function getAppByExec(exec) {
        var cleanExec = exec.replace(/%[fFuU]/g, "").trim()
        return applicationsByExec[cleanExec] || null
    }
    
    function getCategoriesForApp(app) {
        if (!app || !app.categories) return []
        
        var categoryMap = {
            "AudioVideo": "Media",
            "Audio": "Media",
            "Video": "Media",
            "Development": "Development",
            "TextEditor": "Development",
            "IDE": "Development",
            "Education": "Education",
            "Game": "Games",
            "Graphics": "Graphics",
            "Photography": "Graphics",
            "Network": "Internet",
            "WebBrowser": "Internet",
            "Email": "Internet",
            "Office": "Office",
            "WordProcessor": "Office",
            "Spreadsheet": "Office",
            "Presentation": "Office",
            "Science": "Science",
            "Settings": "Settings",
            "System": "System",
            "Utility": "Utilities",
            "Accessories": "Utilities",
            "FileManager": "Utilities",
            "TerminalEmulator": "Utilities"
        }
        
        var mappedCategories = new Set()
        
        for (var i = 0; i < app.categories.length; i++) {
            var cat = app.categories[i]
            if (categoryMap[cat]) {
                mappedCategories.add(categoryMap[cat])
            }
        }
        
        return Array.from(mappedCategories)
    }
    
    // Category icon mappings
    property var categoryIcons: ({
        "All": "apps",
        "Media": "music_video",
        "Development": "code",
        "Games": "sports_esports",
        "Graphics": "photo_library", 
        "Internet": "web",
        "Office": "content_paste",
        "Settings": "settings",
        "System": "host",
        "Utilities": "build"
    })
    
    function getCategoryIcon(category) {
        return categoryIcons[category] || "folder"
    }
    
    function getAllCategories() {
        var categories = new Set(["All"])
        
        for (var i = 0; i < applications.length; i++) {
            var appCategories = getCategoriesForApp(applications[i])
            appCategories.forEach(cat => categories.add(cat))
        }
        
        return Array.from(categories).sort()
    }
    
    function getAppsInCategory(category) {
        if (category === "All") {
            return applications
        }
        
        return applications.filter(app => {
            var appCategories = getCategoriesForApp(app)
            return appCategories.includes(category)
        })
    }
    
}