From ca2d657d184d62851fe17c1a2cfc166a8185548e Mon Sep 17 00:00:00 2001
From: anaximeno <anaximenobrito@gmail.com>
Date: Fri, 24 Feb 2023 00:16:02 -0100
Subject: [PATCH] feat: icon display for the workspace-switcher applet initial
 implementation

---
 .../workspace-switcher@cinnamon.org/applet.js | 52 +++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/files/usr/share/cinnamon/applets/workspace-switcher@cinnamon.org/applet.js b/files/usr/share/cinnamon/applets/workspace-switcher@cinnamon.org/applet.js
index ff99df24b3..19dcf23e34 100644
--- a/files/usr/share/cinnamon/applets/workspace-switcher@cinnamon.org/applet.js
+++ b/files/usr/share/cinnamon/applets/workspace-switcher@cinnamon.org/applet.js
@@ -11,9 +11,12 @@ const Tooltips = imports.ui.tooltips;
 const Settings = imports.ui.settings;
 const ModalDialog = imports.ui.modalDialog;
 const Pango = imports.gi.Pango;
+const Cinnamon = imports.gi.Cinnamon;
 
 const MIN_SWITCH_INTERVAL_MS = 220;
 
+const ICON_SIZE = 16; // TODO: size according to the size of the applet (or panel)
+
 class WorkspaceButton {
     constructor(index, applet) {
         this.index = index;
@@ -82,6 +85,41 @@ class WorkspaceGraph extends WorkspaceButton {
 
         this.graphArea.set_size(1, 1);
         this.graphArea.connect('repaint', Lang.bind(this, this.onRepaint));
+
+        this.window_icons = {};
+    }
+
+    getIconForMetaWindow (metaWindow) {
+        let metaWindowId = metaWindow.get_id();
+
+        if (this.window_icons[metaWindowId])
+            return this.window_icons[metaWindowId];
+
+        let iconActor = null;
+        let app = null;
+
+        if (metaWindow._expoApp) {
+            app = metaWindow._expoApp;
+        } else {
+            let tracker = Cinnamon.WindowTracker.get_default();
+            app = tracker.get_window_app(metaWindow);
+            metaWindow._expoApp = app;
+        }
+
+        if (app) {
+            iconActor = app.create_icon_texture_for_window(ICON_SIZE, metaWindow);
+        }
+
+        if (!iconActor) {
+            iconActor = new St.Icon({
+                icon_name: 'applications-other',
+                icon_type: St.IconType.FULLCOLOR,
+                icon_size: ICON_SIZE
+            });
+        }
+
+        this.window_icons[metaWindowId] = iconActor;
+        return this.window_icons[metaWindowId];
     }
 
     getSizeAdjustment (actor, vertical) {
@@ -153,10 +191,19 @@ class WorkspaceGraph extends WorkspaceButton {
 
         Clutter.cairo_set_source_color(cr, windowBorderColor);
         cr.rectangle(scaled_rect.x, scaled_rect.y, scaled_rect.width, scaled_rect.height);
+
         cr.strokePreserve();
 
         Clutter.cairo_set_source_color(cr, windowBackgroundColor);
         cr.fill();
+
+        //TODO: check settings for showing icon before
+        
+        let icon = this.getIconForMetaWindow(metaWindow);
+
+        if (icon) {
+            // TODO: draw icon in the frame
+        }
     }
 
     onRepaint(area) {
@@ -210,6 +257,11 @@ class WorkspaceGraph extends WorkspaceButton {
         else
             this.actor.remove_style_pseudo_class('active');
     }
+
+    destroy() {
+        super.destroy();
+        this.window_icons = null;
+    }
 }
 
 class SimpleButton extends WorkspaceButton {