From d5a424ebe1a306e68aa9187db2e2ba2c84a66ca2 Mon Sep 17 00:00:00 2001
From: Pujit Mehrotra <pujitm2009@gmail.com>
Date: Wed, 9 Oct 2024 10:20:45 -0400
Subject: [PATCH] refactor(api): directly accept importance level in
 UnraidLocalNotifier

---
 api/src/core/notifiers/notifier.ts     |  2 ++
 api/src/core/notifiers/unraid-local.ts | 39 ++++++++++++++++++++++++++
 2 files changed, 41 insertions(+)
 create mode 100644 api/src/core/notifiers/unraid-local.ts

diff --git a/api/src/core/notifiers/notifier.ts b/api/src/core/notifiers/notifier.ts
index 26309aa04..e3678f519 100644
--- a/api/src/core/notifiers/notifier.ts
+++ b/api/src/core/notifiers/notifier.ts
@@ -1,10 +1,12 @@
 import Mustache from 'mustache';
 import { type LooseObject } from '@app/core/types';
+import { type NotificationIni } from '../types/states/notification';
 
 export type NotifierLevel = 'info' | 'warn' | 'error';
 
 export type NotifierOptions = Partial<{
 	level: NotifierLevel;
+	importance?: NotificationIni['importance'];
 	helpers?: Record<string, unknown>;
 	template?: string;
 }>;
diff --git a/api/src/core/notifiers/unraid-local.ts b/api/src/core/notifiers/unraid-local.ts
new file mode 100644
index 000000000..e40c24145
--- /dev/null
+++ b/api/src/core/notifiers/unraid-local.ts
@@ -0,0 +1,39 @@
+import { logger } from '@app/core/log';
+import { Notifier, type NotifierSendOptions, type NotifierOptions } from '@app/core/notifiers/notifier';
+import { execa } from 'execa';
+
+type ValidLocalLevels = 'alert' | 'warning' | 'normal';
+
+export class UnraidLocalNotifier extends Notifier {
+	private convertNotifierLevel(level: NotifierOptions['level']): ValidLocalLevels {
+		switch (level) {
+			case 'error':
+				return 'alert';
+			case 'warn':
+				return 'warning';
+			case 'info':
+				return 'normal';
+			default:
+				return 'normal';
+		}
+	}
+
+	constructor(options: NotifierOptions = {}) {
+		super(options);
+
+		this.level = options.importance ?? this.convertNotifierLevel(options.level ?? 'info');
+		this.template = options.template ?? '{{ message }}';
+	}
+
+	async send(options: NotifierSendOptions) {
+		const { title, data } = options;
+		const { level } = this;
+
+		const template = this.render(data);
+		try {
+			await execa('/usr/local/emhttp/webGui/scripts/notify', ['-i', `${level}`, '-s', 'Unraid API', '-d', `${template}`, '-e', `${title}`]);
+		} catch (error: unknown) {
+			logger.warn(`Error sending unraid notification: ${error instanceof Error ? error.message : 'No Error Information'}`);
+		}
+	}
+}