TidaLuna-Plugins/plugins/obsidian-theme-luna/src/index.ts

import { type LunaUnload, Tracer } from "@luna/core";
import { StyleTag, observePromise, PlayState } from "@luna/lib";
import { settings, Settings } from "./Settings";

// Import CSS files directly using Luna's file:// syntax - Took me a while to figure out <3
import darkTheme from "file://dark-theme.css?minify";
import oledFriendlyTheme from "file://oled-friendly.css?minify";
import lightTheme from "file://light-theme.css?minify";

export const { trace } = Tracer("[Obsidian]");
export { Settings };

// called when plugin is unloaded.
// clean up resources
export const unloads = new Set<LunaUnload>();

// StyleTag instance for theme management
const themeStyleTag = new StyleTag("Obsidian", unloads);

// Quality color mapping
const QUALITY_COLORS = {
	MAX: "#FED330", // Max/HiFi
	HIGH: "#31FFEE", // High
	LOW: "#FFFFFE", // Low
};

// Function to get quality color based on audio quality
const getQualityColor = (audioQuality: string): string => {
	const quality = audioQuality?.toUpperCase();
	if (quality?.includes("HI_RES_LOSSLESS")) {
		return QUALITY_COLORS.MAX;
	} else if (quality?.includes("LOSSLESS")) {
		return QUALITY_COLORS.HIGH;
	} else if (quality?.includes("HIGH")) {
		return QUALITY_COLORS.HIGH;
	} else if (quality?.includes("LOW")) {
		return QUALITY_COLORS.LOW;
	}
	return QUALITY_COLORS.LOW;
};

// Interval tracking for quality monitoring
let qualityMonitoringIntervalId: number | null = null;

// Function to Reset Seek Bar Color (if setting gets disabled while playing)
const resetSeekBarColor = async (): Promise<void> => {
	try {
		const progressBarWrapper = await observePromise<HTMLElement>(
			unloads,
			`[class^="_progressBarWrapper"]`,
		);
		if (!progressBarWrapper) return;
		progressBarWrapper.style.removeProperty("color");
		progressBarWrapper
			.querySelectorAll('[class*="progress"], [class*="bar"]')
			.forEach((el) => {
				if (el instanceof HTMLElement) el.style.removeProperty("color");
			});
	} catch (error) {
		trace.msg.err(`Failed to reset seek bar color: ${error}`);
	}
};

// Function to apply quality-based seek bar coloring (if enabled)
const applyQualityColors = async (): Promise<void> => {
	if (!settings.qualityColorMatchedSeekBar) return;
	try {
		const progressBarWrapper = await observePromise<HTMLElement>(
			unloads,
			`[class^="_progressBarWrapper"]`,
		);
		if (!progressBarWrapper) return;
		const audioQuality = PlayState.playbackContext?.actualAudioQuality;
		if (!audioQuality) return;
		const qualityColor = getQualityColor(audioQuality);
		progressBarWrapper.style.setProperty("color", qualityColor, "important");
		progressBarWrapper
			.querySelectorAll('[class*="progress"], [class*="bar"]')
			.forEach((el) => {
				if (el instanceof HTMLElement)
					el.style.setProperty("color", qualityColor, "important");
			});
		//trace.msg.log(`Applied quality color ${qualityColor}`);
	} catch (error) {
		trace.msg.err(`Failed to apply quality colors: ${error}`);
	}
};

// Function to monitor track changes using track ID
const setupQualityMonitoring = (): void => {
	if (qualityMonitoringIntervalId != null) return;
	let lastTrackId: string | null = null;
	qualityMonitoringIntervalId = window.setInterval(() => {
		if (!settings.qualityColorMatchedSeekBar) return;
		const currentTrackId = PlayState.playbackContext?.actualProductId;
		if (currentTrackId && currentTrackId !== lastTrackId) {
			lastTrackId = currentTrackId;
			applyQualityColors();
		}
	}, 250);

	// Initial color application (if a track is already loaded)
	const currentTrackId = PlayState.playbackContext?.actualProductId;
	if (settings.qualityColorMatchedSeekBar && currentTrackId) {
		lastTrackId = currentTrackId;
		applyQualityColors();
	}
};

const stopQualityMonitoring = (): void => {
	if (qualityMonitoringIntervalId != null) {
		window.clearInterval(qualityMonitoringIntervalId);
		qualityMonitoringIntervalId = null;
	}
};

// Function to apply theme styles based on current settings
const applyThemeStyles = (): void => {
	// Choose the appropriate CSS file based on settings
	let selectedStyle: string;

	if (settings.lightMode) {
		// Light mode - (OLED friendly doesn't apply to light theme)
		selectedStyle = lightTheme;
	} else {
		// Dark mode
		selectedStyle = settings.oledFriendlyButtons
			? oledFriendlyTheme
			: darkTheme;
	}

	// Remove SeekBar coloring if Quality Color Matched Seek Bar is enabled
	// This allows our manual coloring to take precedence
	if (settings.qualityColorMatchedSeekBar) {
		selectedStyle = selectedStyle.replace(
			/\[class\^="_progressBarWrapper"\]\s*\{[^}]*\}/g,
			"",
		);
		setupQualityMonitoring();
	} else {
		// If disabling, reset the seek bar color and stop monitoring
		resetSeekBarColor();
		stopQualityMonitoring();
	}

	// Apply the selected theme using StyleTag
	themeStyleTag.css = selectedStyle;
};

// Make this function available globally so Settings can call it
declare global {
	interface Window {
		updateObsidianStyles?: () => void;
	}
}

window.updateObsidianStyles = applyThemeStyles;

// Apply the Obsidian theme initially
applyThemeStyles();

// Ensure interval is cleared on unload
unloads.add(() => {
	stopQualityMonitoring();
});