Function registerCodeEmbedProcessor

src/code_embed.ts:610–776

(plugin: CodeSpacePlugin)

}

export function registerCodeEmbedProcessor(plugin: CodeSpacePlugin) {
	// Install observer for the main window document to catch any embeds that the post processor misses.
	// This is necessary because registerMarkdownPostProcessor may be called before the embed element
	// is attached to the workspace leaf, causing resolveSourcePathForEmbed to fail.
	const mainWindow = activeWindow;
	const mainDoc = activeDocument;
	ensureCodeSpaceStylesInDocument(mainDoc, plugin);
	installEmbedObserverForDocument(mainDoc, mainWindow, plugin);
	installPrintRefreshForDocument(mainDoc, mainWindow, plugin);

	// Use Obsidian's official markdown post processor so we always get a correct ctx.sourcePath.
	// This avoids races where MutationObserver runs before embed link attributes are stable.
	plugin.registerMarkdownPostProcessor((el, ctx) => {
		if (ctx.sourcePath && !ctx.sourcePath.startsWith("Untitled")) {
			rememberSourcePath(el, ctx.sourcePath);
		}

		ctx.addChild(new CodeEmbedSectionObserverChild(el, plugin, ctx.sourcePath ?? ""));

		// Obsidian renders embedded code files as div.file-embed (edit mode) or span.file-embed (reading mode)
		// with div.file-embed-title
		const embeds = el.querySelectorAll('.file-embed');

		for (let i = 0; i < embeds.length; i++) {
			const embedEl = embeds[i] as HTMLElement;
			// Prefer ctx.sourcePath when available. When unreliable, the main window's
			// MutationObserver (installed above) will catch it after DOM stabilizes.
			const sourcePath = ctx.sourcePath;
			if (sourcePath && !sourcePath.startsWith("Untitled")) {
				rememberSourcePath(embedEl, sourcePath);
				scheduleProcessCodeEmbed(embedEl, plugin, sourcePath);
			}
			// If sourcePath is unreliable, rely on the main window observer to pick it up.
		}
	});

	// Listen for file changes so we can update embeds when their source files change on disk.
	plugin.registerEvent(
		plugin.app.vault.on("modify", (changedFile) => {
			// 遍历所有文档，查找嵌入了变更文件的嵌入块
			plugin.app.workspace.iterateAllLeaves((leaf) => {
				const view = leaf.view as unknown as {
					containerEl?: HTMLElement
					contentEl?: HTMLElement
					file?: TFile
				} | null
				if (!view) return

				const possibleContainers = [
					view.contentEl,
					view.containerEl,
					view.containerEl?.querySelector(".markdown-preview-view"),
					view.containerEl?.querySelector(".markdown-source-view"),
				]

				for (const container of possibleContainers) {
					if (!container) continue