Add support for Base64 hardcoded icons in binary files

HardcodeTray/enum.py

@@ -69,6 +69,7 @@ class ApplicationType:
         Different Applications Type
     """
     ELECTRON = "ElectronApplication"
+    B64ELECTRON = "B64ElectronApplication"
     ZIP = "ZipApplication"
     PAK = "PakApplication"
     NWJS = "NWJSApplication"

HardcodeTray/modules/applications/b64electron.py

@@ -0,0 +1,67 @@
+"""
+Fixes Hardcoded tray icons in Linux.
+
+Author : Ivo Šmerek (ivo.smerek@gmail.com)
+Contributors : Bilal Elmoussaoui, Alexey Varfolomeev
+Website : https://github.com/bil-elmoussaoui/Hardcode-Tray
+Licence : The script is released under GPL, uses a modified script
+     form Chromium project released under BSD license
+This file is part of Hardcode-Tray.
+Hardcode-Tray is free software: you can redistribute it and/or
+modify it under the terms of the GNU General Public License as published
+by the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+Hardcode-Tray is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+You should have received a copy of the GNU General Public License
+along with Hardcode-Tray. If not, see <http://www.gnu.org/licenses/>.
+"""
+from base64 import b64encode
+from glob import glob
+from os import path, system
+from fileinput import FileInput
+
+from HardcodeTray.app import App
+from HardcodeTray.modules.applications.electron import ElectronApplication
+from HardcodeTray.modules.applications.helpers.asar import AsarFile
+from HardcodeTray.modules.log import Logger
+from HardcodeTray.utils import get_pngbytes
+
+
+class B64ElectronApplication(ElectronApplication):
+    """Electron application with Base64 hardcoded icons."""
+
+    def __init__(self, parser):
+        """Use the parent class, Application, modify only the (re)install."""
+        ElectronApplication.__init__(self, parser)
+
+    def install_icon(self, icon, icon_path):
+        """Install the icon."""
+        icon.icon_size = App.icon_size()
+
+        png_bytes = get_pngbytes(icon)
+
+        if png_bytes:
+            binary_file = path.join(str(icon_path), self.binary)
+            asar = AsarFile(binary_file)
+            target = B64ElectronApplication.get_real_path(self.file)
+            # Read a target file
+            file_content = asar.read_file(target).decode()
+            # Create new base64 binary file
+            base64_icon = b64encode(App.svg().to_bin(icon.theme, icon.icon_size, icon.icon_size))
+            # Build new icon
+            new_icon = "data:image/png;base64," + base64_icon.decode()
+            # Replace the original icon with newly built one
+            file_content = file_content.replace(icon.original, new_icon)
+            # Write changed file
+            asar.write(target, file_content.encode())
+        else:
+            Logger.error("Icon file was not found.")
+
+    def revert_icon(self, icon, icon_path):
+        """Revert to the original icon."""
+        # I don't know what is the best way to handle backup here.
+        # Maybe backup whole app.asar file?
+        Logger.error("Backup file of {0} was not found".format(self.name))

HardcodeTray/modules/applications/helpers/asar.py

@@ -3,7 +3,7 @@
 Fixes Hardcoded tray icons in Linux.
 
 Author : Bilal Elmoussaoui (bil.elmoussaoui@gmail.com)
-Contributors : Andreas Angerer, Joshua Fogg
+Contributors : Andreas Angerer, Joshua Fogg, Ivo Šmerek
 Website : https://github.com/bil-elmoussaoui/Hardcode-Tray
 Licence : The script is released under GPL, uses a modified script
      form Chromium project released under BSD license
@@ -119,3 +119,22 @@ def write(self, icon, pngbytes):
         asarfile = open(self._asar_file, 'wb')
         asarfile.write(asar_content)
         asarfile.close()
+
+    def read_file(self, path):
+        """Read a file in asar archive."""
+        self._keys = path.split("/")
+
+        fileinfo = get_from_dict(self._header, self._keys)
+
+        if (not isinstance(fileinfo, dict)
+                or "offset" not in fileinfo.keys()):
+            return
+
+        offset0 = int(fileinfo['offset'])
+        offset = offset0 + self._offset
+        size = int(fileinfo['size'])
+
+        with open(self._asar_file, 'rb') as asarfile:
+            asar = asarfile.read()
+
+        return asar[offset: offset + size]

data/database/slack.json

@@ -0,0 +1,28 @@
+{
+    "name": "Slack",
+    "app_path": [
+        "/usr/share/slack/",
+        "/usr/lib/slack/"
+    ],
+    "icons_path": [
+        "/usr/share/slack/resources/",
+        "/usr/lib/slack/resources/"
+    ],
+    "binary": "app.asar",
+    "script": "b64electron",
+    "file": "dist/main.bundle.js",
+    "icons": {
+        "highlight": {
+            "original": "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABYAAAAWCAYAAADEtGw7AAAAAXNSR0IArs4c6QAABBRJREFUOBGNlW1IW2cUx59c82KTNW0jNE2IXaadYytjhuLcmLF2MEM/DMqcjso6QSuI4FBaSj8UJkxmy1g/DBEGymQo2wqDzRfmW0GHK8R+sG6LRV1CbGPQCpqXapKbm3v3P1dviRplB355Xu5z/vd5zjn3CWOHWxYeHweqtGVa9I1p44zddIf0BST4PngHUD8GRoAJOAH5LYOfQQTsM27fzPaE02g0vjc5Ofmu3++/3NfXdxbTl0Bpf3//Gx6P52JRUVEuxhe3l///32uDg4P3pDRraWm5W1xc/CWmRJoeGxv7FXI3DpLMtGOa01ut1lPpTiUlJRabzabGnBy+l2Do05iM5naFNZOwiEXLXV1dbkEQePIiy8qiUO8z8v8I3NzhY7T0whdvpAUOYN+Z03Z2dm663e47lZWVJo1Gw42Ojq4j7nvV1Q6Ho7Cjo6OY4zhVc3OzHj6U4G7lKB/igaOxsdFgsVh0ePDCUqmUODw8HJqYmBAqKir2CrOhoaHP4PMyOVBizWZzF7p2Gh8BrVNTU7+n5WpvN1VXV3cHwu1YJ7W3t0vl5eVPcJI/cTLJ5/Mp60WcqhV652jH8i52koFhRuN0Ot2xgYGBt7xer4h4c2uBoF5Kifbu7m7W09PDamtrmdPpfBSJyGW9omTyU1TBmba2tlyTyaTfK434hrGzD1QqVfZt+3njZWvpkZPZNk5gvPR066n4xep47H7YHw+FQj6E7mv4/6IIv4JBDURZWVmZFgIqJCERCAQYVRXP865kMnn6wZtXcwpzCnflYHsTKukb30/Pby39EcWY6vtzJRnVTU1N5pGRkebq6uoLVVVVJXj4eG5ubrO3t/f67OysvSm7QHXJXKpnUoplQFV0vED3l2FDs7gWtMJ3nMrsKDiJUnEhjpRI2Sgb9fX1ppmZGQPF9BPz23omJtlBqEWRXTn6Gl1QGnCOhLEat0wstkWtYtPT01G1Ws2CwSDTMY5Zsk5wEoQP43RMJ8BfAiIJx8FiQ0PD2NLS0kI0Gg0h+/dgcYq1VqvFmyUWFzYRAX5bWGnpRUofbTgRIeF1EFY+kH7Up85ut/+ISTJaEEM0TPn5+QwfB/s3viKc1Z2iYx5oj54HQ3hI14CXdkxG2fwedIIfwF0QAMzlcsn3xFfhhzGEAXdb5nA84zcSt1cfrsLFDzyKMPpybJ6h9QGKt4Qyk/Ly8uTidyfWhFtPxsNJMb5bHCFYSazzNd7ffGGBp5O2kq9Sx+jvswuo6/Pz8/P1uEf0CMkCPgCbM7dAfyXn9WP5ooGFtqIJdyQY/279bz9EqQi+Bb2kdJgwld5VkEPVgSt0E30DqKGcIv7oMrpiN8ACaAX/ANkOE6YFVJevAjVYBBQi+gNwgRNgGTwGHiC/Ca1s/wHOaRbqM5pQKwAAAABJRU5ErkJggg==",
+            "theme": "slack-indicator-highlight"
+        },
+        "indicator": {
+            "original": "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABYAAAAWCAYAAADEtGw7AAAAAXNSR0IArs4c6QAAA8JJREFUOBGNVV9IU2EUv5tzq5nLFswmmhMKoQK3B1n+C+mlfFAjEwoiXxRkICiBDBERFCFRH5zsaZWGWi2DFAXTFxUN9KVI7EHBmc1YKjU33R+bd/3Otnu9bir94Lf7fec759zv+53z3THM6YjDchIoErhJMVYI5scOhQFCB0p4G7wJ0tgLfgSVYAFIcRvgW9AFxkAcYwkbChQKRd709HTO2trao4GBgesw3wNvjYyMXFtaWirKzs5Ow7wo7P7/v09HR0etQQHq6uq69Hp9C0wsmScnJz8gXf1JKY/bMdnkKSkpl4RB+fn56tTUVAlsIfnOARjTnEC2I7Iel5iF04bFYpkPBAL7FEWIiyOpY0Dx90FjhA/wpBfybyQHHaiJ2KRms3lvfn7+WXl5uTI+Pl48MTHxG7pHZ5fodDptT0+PXiwWi2pra+WIoQI/545SjAWdwWBIUKvVMizwODg4YMfHx51TU1OBsrKy6MTM2NjYE8SkUwAVNjk52YKhhhKfBXUzMzP6vLy8uxjHoKGhga2srOxwOp0kEw+lUslwScmoUqlScCrG5XIlkQShXUSKwQdFDcTYiWxubs6/v7/vp7Xl5WW7VEp35RCwfUFSMji4Sj5GF1xpbW1Nwy7kh67h0cLCwq+2tjYnZmxJScmZ3NxcRVNT07ZEImEGBwczyWt9fX3HaDQ6PB7PIqbvucQZmFTQ0QoLC6UiAEXw2+12hrqK6HA4SIY9JEssLi6OR02OdJTf7w+i96mLfoC9YAiGmpqaDp/P5+HuRGdnpxl92+x2u3cQ5MNO25C0eXNzc4PziX7abLZvyNYMZtJbE0EVWuWOTCajQoZAQVVVVUrsVgEtZRUVFRl4kQinUnE+0U/U4XJEdzl1xV9y8Hq9HqEjdHVnZWXxelOf4rsRbGxsNJWWlt6A7xEpWJYN9PX1fUVxSY5VSuwDV6qrqyX9/f1JtCP07ITVavVptdoErIUA2YNUrN7eXhf4CZpzS6R9AAyCW+AkuEOJCSOzs7MyjUbzOjwNOXqhBt0iHtipbGhoiK4vs7u762ppaXnR3t6+jqkJpOQ8uOO4YXkJmsFXYBdoB08EaV9fX/8QDudBdbQjt2Oy01E2IwzNIRfZQsCYXVxcJP3o6NyGuHYNOwl+OQeBiR9umUymP9vb2z9xm5z40KziZgWHh4ff4fsRQGvudXd3v4E3nZYXnIs+8Y1woNarBC9S0fAJ3cP4M5gjl8vjkJhBJ5DNCn4Hj+C0xORIH4OrIEm2AlJLXgDTQZLFBtL/YQz+ASRsjm4BTCa4AAAAAElFTkSuQmCC",
+            "theme": "slack-indicator"
+        },
+        "unread": {
+            "original": "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABYAAAAWCAYAAADEtGw7AAAAAXNSR0IArs4c6QAABAJJREFUOBGNlX9IW1cUx+97xpc2tumaFl0kXVNt/2lXaijOjprOFWootCC1Qgv9ATpBBDtlUMb2RwOzDf2hhU7yzwyTURnrGKz+oFb7h4Jji/+4lYZNxRBFpbWlS4w2L/H96Pc880qMRnrgk3vuefece+85570wtrFk4fEHgEtZJkA3p8zXVVMdUhdQwOPgCCA9Bh4DC3AC8psFv4AFsEb4NZYVg9NsNh8dGhr6NBQKne/s7DwAcwU41tXVtT8QCJwsLi7ehfnJleXv//tVT0/PAzVFmpqaWktKSr6DSSHzwMDA7wh3NVPI9U5MNlN+fv6HqU6lpaVWm81mgE1L3xYIdJqTkG1VWtcLrGDRbHt7u1+SpAR5kWRlUarXCPmfAV8nOYuRNny3Iy1wAHvSJni93iW/33+zqqrKkp2dzff3979G3tOjGxwOR1FbW1sJz/NcY2OjCT5UYJ9+ldN44Kivr8+xWq1GPHgnsiwrfX194cHBQamysjI9MOvt7b0En93kQIXNy8trh2qn+WbgHh4efpRSq3RVrqmpuYnAHqxTPR6PWl5ePo2b/IGbqcFgUF+v4FZuxDtMJ9ZOkSwGpusKbzQat3V3dx+anJxUkG9+ev6lSVJVu8/nYx0dHay6upo5nc6/Fxa0tn6uV/ICumBvc3PzLovFYkoPjfxGcLITjOM2GZtumeXz5zZLubk8k2R10+y0svXWtZjy5xMxHA4Hkbrb8P9ND7wHk8sIysrKygQOgiLEZ2ZmGHVVIpFwJZaXP+If+ncoHx9aVYPkIVTj3TuL8e+/jWJO/X0laWf1DQ0Nd0RRfKMnq6WlxYu+daNwi0VFRbLw5Y0FNiGqmeD+W1QLTpyiVz8EDlKbbQW5aBUX8kiF1IQ2qK2ttYyOjuZQTpUz50xMxqMMqOhcseIifaCywWEKvAxYLBZ7Q6MuIyMjUYPBwObm5pgiGJm008ozRUVgoI+kpzCft1uCP4xMocAimKirqxuYmpoaj0ajYVT/AUSkXAuCwDgJe4vxldPSe0mnpjENORKhwK9BRH9ButCfRrvd/jOMJLQghmxYCgsLGV4OJgQnpMS+A3TNjMI9HQ3juPQZmKQTk1A1fwRe8BNoBTOAuVwu7Tux5YfrMZxUzZRj9uplXPF5XsAlBAJ6YOhabuYxBgHlW0WbqQUFBVrzC0//krjWbyJcYhnBV+eWzT9PsKsXgmxRS4WbfPU+hr5GPkdffzY2NlaL74gJKRnHC2Db+ckxU9bpi9uWbHuZGInEpX/8IvvVG0JQaoJ74D5F2igwtd4XYAd1Bz6hS9BzwGWqKfIPVSvf/xjHgRs8A5psFJgWUF/uAwYwAShF9AfgAtvBLPgXBIC2E0ZN3gI4J+wlnIU7ngAAAABJRU5ErkJggg==",
+            "theme": "slack-indicator-unread"
+        }
+    }
+}