src/utils/variables/selector.ts - main - githubmirror/obsidian-clipper

src/utils/variables/selector.ts
86 строк3 KB
Steph Ango
Open in Reader (#761)
08 апр 2026, 22:19
08 апр 2026, 22:19eeba739
import browser from '../browser-polyfill';
import { applyFilters } from '../filters';
import { selectorContentToString } from '../shared';
import { debugLog } from '../debug';

/**
 * Send an extractContent message to a tab via the background script,
 * avoiding direct use of browser.tabs which may be unavailable in
 * Firefox iframe/side-panel contexts.
 */
export async function sendExtractContent(tabId: number, selector: string, attribute: string | undefined, extractHtml: boolean): Promise<{ content: string | string[] } | undefined> {
	const response = await browser.runtime.sendMessage({
		action: "sendMessageToTab",
		tabId: tabId,
		message: {
			action: "extractContent",
			selector: selector,
			attribute: attribute,
			extractHtml: extractHtml
		}
	}) as { content: string | string[] };
	return response || undefined;
}

/**
 * Resolve a selector and return the raw content (array or string).
 * Used by the renderer for for loops and conditionals.
 */
export async function resolveSelector(tabId: number, selectorExpr: string): Promise<any> {
	// Parse the selector expression (selector:... or selectorHtml:...)
	// Format: selector:cssSelector or selectorHtml:cssSelector
	// May include attribute selector: selector:cssSelector?attr
	const selectorRegex = /^(selector|selectorHtml):(.*?)(?:\?(.*))?$/;
	const matches = selectorExpr.match(selectorRegex);
	if (!matches) {
		console.error('Invalid selector format:', selectorExpr);
		return undefined;
	}

	const [, selectorType, rawSelector, attribute] = matches;
	const extractHtml = selectorType === 'selectorHtml';

	// Unescape any escaped quotes and normalize whitespace in the selector
	const selector = rawSelector.replace(/\\"/g, '"').replace(/\s+/g, ' ').trim();

	try {
		const response = await sendExtractContent(tabId, selector, attribute, extractHtml);

		// Return the raw content (could be array or string)
		return response ? response.content : undefined;
	} catch (error) {
		console.error('Error extracting content by selector:', error, { selector, attribute, extractHtml });
		return undefined;
	}
}

export async function processSelector(tabId: number, match: string, currentUrl: string): Promise<string> {
	const selectorRegex = /{{(selector|selectorHtml):(.*?)(?:\?(.*?))?(?:\|(.*?))?}}/;
	const matches = match.match(selectorRegex);
	if (!matches) {
		console.error('Invalid selector format:', match);
		return match;
	}

	const [, selectorType, rawSelector, attribute, filtersString] = matches;
	const extractHtml = selectorType === 'selectorHtml';

	// Unescape any escaped quotes and normalize whitespace in the selector
	const selector = rawSelector.replace(/\\"/g, '"').replace(/\s+/g, ' ').trim();

	try {
		const response = await sendExtractContent(tabId, selector, attribute, extractHtml);

		let content = response ? response.content : '';

		const contentString = selectorContentToString(content);

		debugLog('ContentExtractor', 'Applying filters:', { selector, filterString: filtersString });
		const filteredContent = applyFilters(contentString, filtersString, currentUrl);

		return filteredContent;
	} catch (error) {
		console.error('Error extracting content by selector:', error, { selector, attribute, extractHtml });
		return '';
	}
}