QOL: Move source code under the src directory. (#1318)

src/plugins/notifications/utils.ts

@@ -0,0 +1,88 @@
+import path from 'node:path';
+import fs from 'node:fs';
+
+import { app, NativeImage } from 'electron';
+
+import config from './config';
+
+import { cache } from '../../providers/decorators';
+import { SongInfo } from '../../providers/song-info';
+import { getAssetsDirectoryLocation } from '../utils';
+
+const defaultIcon = path.join(getAssetsDirectoryLocation(), 'youtube-music.png');
+const userData = app.getPath('userData');
+const temporaryIcon = path.join(userData, 'tempIcon.png');
+const temporaryBanner = path.join(userData, 'tempBanner.png');
+
+
+export const ToastStyles = {
+  logo: 1,
+  banner_centered_top: 2,
+  hero: 3,
+  banner_top_custom: 4,
+  banner_centered_bottom: 5,
+  banner_bottom: 6,
+  legacy: 7,
+};
+
+export const urgencyLevels = [
+  { name: 'Low', value: 'low' },
+  { name: 'Normal', value: 'normal' },
+  { name: 'High', value: 'critical' },
+];
+
+const nativeImageToLogo = cache((nativeImage: NativeImage) => {
+  const temporaryImage = nativeImage.resize({ height: 256 });
+  const margin = Math.max(temporaryImage.getSize().width - 256, 0);
+
+  return temporaryImage.crop({
+    x: Math.round(margin / 2),
+    y: 0,
+    width: 256,
+    height: 256,
+  });
+});
+
+export const notificationImage = (songInfo: SongInfo) => {
+  if (!songInfo.image) {
+    return defaultIcon;
+  }
+
+  if (!config.get('interactive')) {
+    return nativeImageToLogo(songInfo.image);
+  }
+
+  switch (config.get('toastStyle')) {
+    case ToastStyles.logo:
+    case ToastStyles.legacy: {
+      return saveImage(nativeImageToLogo(songInfo.image), temporaryIcon);
+    }
+
+    default: {
+      return saveImage(songInfo.image, temporaryBanner);
+    }
+  }
+};
+
+export const saveImage = cache((img: NativeImage, savePath: string) => {
+  try {
+    fs.writeFileSync(savePath, img.toPNG());
+  } catch (error: unknown) {
+    console.log(`Error writing song icon to disk:\n${String(error)}`);
+    return defaultIcon;
+  }
+
+  return savePath;
+});
+
+export const snakeToCamel = (string_: string) => string_.replaceAll(/([-_][a-z]|^[a-z])/g, (group) =>
+  group.toUpperCase()
+    .replace('-', ' ')
+    .replace('_', ' '),
+);
+
+export const secondsToMinutes = (seconds: number) => {
+  const minutes = Math.floor(seconds / 60);
+  const secondsLeft = seconds % 60;
+  return `${minutes}:${secondsLeft < 10 ? '0' : ''}${secondsLeft}`;
+};