feat: improve XDG specs implementation

cmd/lsdesktop/main.go

@@ -5,12 +5,21 @@ import (
 	"io/fs"
 	"os"
 	"path/filepath"
+	"slices"
 	"strings"
 
 	"github.com/FLchs/lsdesktop/internal/desktop"
 	"github.com/FLchs/lsdesktop/internal/history"
 )
 
+func currentDesktops() []string {
+	val := os.Getenv("XDG_CURRENT_DESKTOP")
+	if val == "" {
+		return nil
+	}
+	return strings.Split(val, ":")
+}
+
 func appDirs() []string {
 	dirs := make([]string, 0, 4)
 
@@ -38,6 +47,8 @@ func appDirs() []string {
 }
 
 func main() {
+	desktops := currentDesktops()
+
 	hist, err := history.ReadHistory()
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "Error: read history: %v\n", err)
@@ -76,6 +87,17 @@ func main() {
 				return nil
 			}
 
+			if len(entry.OnlyShowIn) > 0 && !slices.ContainsFunc(entry.OnlyShowIn, func(s string) bool {
+				return slices.Contains(desktops, s)
+			}) {
+				return nil
+			}
+			if len(entry.NotShowIn) > 0 && slices.ContainsFunc(entry.NotShowIn, func(s string) bool {
+				return slices.Contains(desktops, s)
+			}) {
+				return nil
+			}
+
 			if _, ok := seen[entry.Name]; ok {
 				return nil
 			}

internal/desktop/desktop.go

@@ -8,6 +8,7 @@ import (
 func ParseEntry(data string) (DesktopEntry, bool) {
 	var name string
 	visible := true
+	var entry DesktopEntry
 
 	for line := range strings.SplitSeq(data, "\n") {
 		line = strings.TrimSpace(line)
@@ -30,6 +31,20 @@ func ParseEntry(data string) (DesktopEntry, bool) {
 				visible = false
 			}
 		}
+		if v, ok := strings.CutPrefix(line, "OnlyShowIn="); ok {
+			for s := range strings.SplitSeq(v, ";") {
+				if s != "" {
+					entry.OnlyShowIn = append(entry.OnlyShowIn, s)
+				}
+			}
+		}
+		if v, ok := strings.CutPrefix(line, "NotShowIn="); ok {
+			for s := range strings.SplitSeq(v, ";") {
+				if s != "" {
+					entry.NotShowIn = append(entry.NotShowIn, s)
+				}
+			}
+		}
 	}
 
 	return DesktopEntry{Name: name}, visible && name != ""
@@ -49,8 +64,10 @@ func CleanExec(execLine string) string {
 }
 
 type DesktopEntry struct {
-	Name string
-	Path string
+	Name       string
+	Path       string
+	OnlyShowIn []string
+	NotShowIn  []string
 }
 
 func EntryString(name, path string) string {