supermemory/apps/browser-extension/utils/route-detection.ts

/**
 * Route Detection Utilities
 * Shared logic for detecting route changes across different AI chat platforms
 */

import { UI_CONFIG } from "./constants"

export interface RouteDetectionConfig {
	platform: string
	selectors: string[]
	reinitCallback: () => void
	checkInterval?: number
	observerThrottleDelay?: number
}

export interface RouteDetectionCleanup {
	observer: MutationObserver | null
	urlCheckInterval: NodeJS.Timeout | null
	observerThrottle: NodeJS.Timeout | null
}

export function createRouteDetection(
	config: RouteDetectionConfig,
	cleanup: RouteDetectionCleanup,
): void {
	if (cleanup.observer) {
		cleanup.observer.disconnect()
	}
	if (cleanup.urlCheckInterval) {
		clearInterval(cleanup.urlCheckInterval)
	}
	if (cleanup.observerThrottle) {
		clearTimeout(cleanup.observerThrottle)
		cleanup.observerThrottle = null
	}

	let currentUrl = window.location.href

	const checkForRouteChange = () => {
		if (window.location.href !== currentUrl) {
			currentUrl = window.location.href
			console.log(`${config.platform} route changed, re-initializing`)
			setTimeout(config.reinitCallback, 1000)
		}
	}

	cleanup.urlCheckInterval = setInterval(
		checkForRouteChange,
		config.checkInterval || UI_CONFIG.ROUTE_CHECK_INTERVAL,
	)

	cleanup.observer = new MutationObserver((mutations) => {
		if (cleanup.observerThrottle) {
			return
		}

		let shouldRecheck = false
		mutations.forEach((mutation) => {
			if (mutation.type === "childList" && mutation.addedNodes.length > 0) {
				mutation.addedNodes.forEach((node) => {
					if (node.nodeType === Node.ELEMENT_NODE) {
						const element = node as Element

						for (const selector of config.selectors) {
							if (
								element.querySelector?.(selector) ||
								element.matches?.(selector)
							) {
								shouldRecheck = true
								break
							}
						}
					}
				})
			}
		})

		if (shouldRecheck) {
			cleanup.observerThrottle = setTimeout(() => {
				try {
					cleanup.observerThrottle = null
					config.reinitCallback()
				} catch (error) {
					console.error(`Error in ${config.platform} observer callback:`, error)
				}
			}, config.observerThrottleDelay || UI_CONFIG.OBSERVER_THROTTLE_DELAY)
		}
	})

	try {
		cleanup.observer.observe(document.body, {
			childList: true,
			subtree: true,
		})
	} catch (error) {
		console.error(`Failed to set up ${config.platform} route observer:`, error)
		if (cleanup.urlCheckInterval) {
			clearInterval(cleanup.urlCheckInterval)
		}
		cleanup.urlCheckInterval = setInterval(checkForRouteChange, 1000)
	}
}

export function cleanupRouteDetection(cleanup: RouteDetectionCleanup): void {
	if (cleanup.observer) {
		cleanup.observer.disconnect()
		cleanup.observer = null
	}
	if (cleanup.urlCheckInterval) {
		clearInterval(cleanup.urlCheckInterval)
		cleanup.urlCheckInterval = null
	}
	if (cleanup.observerThrottle) {
		clearTimeout(cleanup.observerThrottle)
		cleanup.observerThrottle = null
	}
}