From 5e9617f6ccd2a679c6b5276b2c5b4a8b1f7802c8 Mon Sep 17 00:00:00 2001
From: Nariman Jelveh <n.jelveh@gmail.com>
Date: Wed, 5 Jun 2024 14:39:52 -0700
Subject: [PATCH] Add support for `ContextMenu` icons via Puter.js

---
 src/IPC.js | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/IPC.js b/src/IPC.js
index 4f02e61e6..f00b6a13a 100644
--- a/src/IPC.js
+++ b/src/IPC.js
@@ -397,6 +397,13 @@ window.addEventListener('message', async (event) => {
         let items = value.items ?? [];
         const sanitize_items = items => {
             return items.map(item => {
+                // make sure item.icon and item.icon_active are valid base64 strings
+                if (item.icon && !item.icon.startsWith('data:image')) {
+                    item.icon = undefined;
+                }
+                if (item.icon_active && !item.icon_active.startsWith('data:image')) {
+                    item.icon_active = undefined;
+                }
                 // Check if the item is just '-'
                 if (item === '-') {
                     return '-';
@@ -404,9 +411,10 @@ window.addEventListener('message', async (event) => {
                 // Otherwise, proceed as before
                 return {
                     html: item.label,
+                    icon: item.icon ? `<img style="width: 15px; height: 15px; position: absolute; top: 4px; left: 6px;" src="${html_encode(item.icon)}" />` : undefined,
+                    icon_active: item.icon_active ? `<img style="width: 15px; height: 15px; position: absolute; top: 4px; left: 6px;" src="${html_encode(item.icon_active)}" />` : undefined,
                     onClick: () => {
                         if (item.action !== undefined) {
-                            console.log('item.action', item.action);
                             item.action();
                         }
                     },