1
0
mirror of https://github.com/lensapp/lens.git synced 2025-05-20 05:10:56 +00:00
lens/packages/generate-tray-icons/src/index.ts
Sebastian Malton 786e0e8ac9 chore: Fix build failing on non-windows platforms
Signed-off-by: Sebastian Malton <sebastian@malton.name>
2023-05-09 12:50:45 -04:00

182 lines
5.3 KiB
JavaScript

#!/usr/bin/env node
/**
* Copyright (c) OpenLens Authors. All rights reserved.
* Licensed under MIT License. See LICENSE in root directory for more information.
*/
import { mkdir, readFile } from "fs/promises";
import { JSDOM } from "jsdom";
import path from "path";
import sharp from "sharp";
import arg from "arg";
import assert from "assert";
import { platform } from "process";
const options = arg({
"--input": String,
"--output": String,
"--notice-icon": String,
"--spinner-icon": String,
"--output-size": Number,
});
type Options = typeof options;
const assertOption = <Key extends keyof Options>(key: Key): NonNullable<Options[Key]> => {
const raw = options[key];
if (raw === undefined) {
console.error(`missing ${key} option`);
process.exit(1);
}
return raw;
};
const joinWithInitCwd = (relativePath: string): string => {
const { INIT_CWD } = process.env;
if (!INIT_CWD) {
return relativePath;
}
return path.join(INIT_CWD, relativePath);
};
const resolve = async (input: string) => {
const filePath = await import.meta.resolve?.(input);
if (platform === "win32") {
return filePath?.replace("file:///", "");
} else {
return filePath?.replace("file://", "");
}
}
const size = options["--output-size"] ?? 16;
const outputFolder = joinWithInitCwd(assertOption("--output"));
const inputFile = await resolve(assertOption("--input"));
const noticeFile = await resolve(assertOption("--notice-icon"));
const spinnerFile = await resolve(assertOption("--spinner-icon"));
assert(inputFile, "input file not found");
assert(noticeFile, "notice icon file not found");
assert(spinnerFile, "spinner icon file not found");
const getSvgStyling = (colouring: "dark" | "light") => (
`
<style>
ellipse {
stroke: ${colouring === "dark" ? "white" : "black"} !important;
}
path, rect {
fill: ${colouring === "dark" ? "white" : "black"} !important;
}
</style>
`
);
type TargetSystems = "macos" | "windows-or-linux";
const getBaseIconImage = async (system: TargetSystems) => {
const svgData = await readFile(inputFile, { encoding: "utf-8" });
const dom = new JSDOM(`<body>${svgData}</body>`);
const root = dom.window.document.body.getElementsByTagName("svg")[0];
root.innerHTML += getSvgStyling(system === "macos" ? "light" : "dark");
return Buffer.from(root.outerHTML);
}
const generateImage = (image: Buffer, size: number, namePrefix: string) => (
sharp(image)
.resize({ width: size, height: size })
.png()
.toFile(path.join(outputFolder, `${namePrefix}.png`))
);
const generateImages = (image: Buffer, size: number, name: string) => Promise.all([
generateImage(image, size, name),
generateImage(image, size * 2, `${name}@2x`),
generateImage(image, size * 3, `${name}@3x`),
generateImage(image, size * 4, `${name}@4x`),
]);
async function generateImageWithSvg(baseImage: Buffer, system: TargetSystems, filePath: string) {
const svgFile = await getIconImage(system, filePath);
const circleBuffer = await sharp(Buffer.from(`
<svg viewBox="0 0 64 64">
<circle cx="32" cy="32" r="32" fill="black" />
</svg>
`))
.toBuffer();
return sharp(baseImage)
.resize({ width: 128, height: 128 })
.composite([
{
input: circleBuffer,
top: 64,
left: 64,
blend: "dest-out",
},
{
input: (
await sharp(svgFile)
.resize({
width: 60,
height: 60,
})
.toBuffer()
),
top: 66,
left: 66,
},
])
.toBuffer();
}
async function getIconImage(system: TargetSystems, filePath: string) {
const svgData = await readFile(filePath, { encoding: "utf-8" });
const root = new JSDOM(svgData).window.document.getElementsByTagName("svg")[0];
root.innerHTML += getSvgStyling(system === "macos" ? "light" : "dark");
return Buffer.from(root.outerHTML);
}
try {
console.log("Generating tray icon pngs");
await mkdir(outputFolder, {
recursive: true,
});
const baseIconTemplateImage = await getBaseIconImage("macos");
const baseIconImage = await getBaseIconImage("windows-or-linux");
const updateAvailableTemplateImage = await generateImageWithSvg(baseIconTemplateImage, "macos", noticeFile);
const updateAvailableImage = await generateImageWithSvg(baseIconImage, "windows-or-linux", noticeFile);
const checkingForUpdatesTemplateImage = await generateImageWithSvg(baseIconTemplateImage, "macos", spinnerFile);
const checkingForUpdatesImage = await generateImageWithSvg(baseIconImage, "windows-or-linux", spinnerFile);
await Promise.all([
// Templates are for macOS only
generateImages(baseIconTemplateImage, size, "trayIconTemplate"),
generateImages(updateAvailableTemplateImage, size, "trayIconUpdateAvailableTemplate"),
generateImages(updateAvailableTemplateImage, size, "trayIconUpdateAvailableTemplate"),
generateImages(checkingForUpdatesTemplateImage, size, "trayIconCheckingForUpdatesTemplate"),
// Non-templates are for windows and linux
generateImages(baseIconImage, size, "trayIcon"),
generateImages(updateAvailableImage, size, "trayIconUpdateAvailable"),
generateImages(checkingForUpdatesImage, size, "trayIconCheckingForUpdates"),
]);
console.log("Generated all images");
} catch (error) {
console.error(error);
process.exit(1);
}