main.js

'use strict';

let obsidian = require('obsidian');
let DEFAULT_SETTINGS = {
	'keyboard_shortcuts': []
};

class HighlightOpenFiles extends obsidian.Plugin {
    async onload() {
		// await this.loadSettings();
		const workspace = this.app.workspace;
		const addMarker = (el,leaf_id) => {
			let marker_container = document.createElement('div');
				marker_container.classList.add('mark_open_files_container');
			let marker = document.createElement('span');
				marker.classList.add('mark_open_files');
				marker.dataset.leaf_id = leaf_id;
				marker.textContent = '•';
			if ( !el.querySelector('.mark_open_files_container') ) {
				el.insertAdjacentElement('afterbegin',marker_container);									// add marker container if needed
			}
			el.querySelector('.mark_open_files_container').insertAdjacentElement('beforeend',marker);		// add marker
		}
		const awaitFileExplorer = () => {
			return new Promise( (resolve) => {
				resolve(workspace.getLeavesOfType('file-explorer')[0]);
			})
		}
		const highlight_open_files = obsidian.debounce( async () => {
			awaitFileExplorer().then( () => {
 				const file_explorer_tree = workspace.getLeavesOfType('file-explorer')[0].view.tree;
				const getOpenItems = () => { 
					let open_leaves = [], open_leaves_ids = [];
					workspace.iterateRootLeaves( leaf => { leaf.loadIfDeferred(); open_leaves.push(leaf); open_leaves_ids.push(leaf.id); } );
					return [open_leaves,open_leaves_ids];
				};
				let open_items = getOpenItems(), open_leaves = open_items[0], open_leaves_ids = open_items[1], open_leaves_paths = open_items[2];
				let getMarkers = () => { return file_explorer_tree?.containerEl?.querySelectorAll('.mark_open_files') };
				let markers = getMarkers();
				if ( !markers ) { return }
				open_leaves.forEach( open_leaf => {
					let tree_item = file_explorer_tree?.containerEl?.querySelector('.tree-item-self[data-path="'+open_leaf?.view?.file?.path+'"]') || null;	// find tree item by leaf file path
					if ( tree_item !== null && !tree_item.querySelector('.mark_open_files[data-leaf_id="'+open_leaf.id+'"]') ) {
						addMarker(tree_item,open_leaf.id );												// if no matching leaf marker, add leaf marker
					}
				});
				markers.forEach( marker => {
					if ( !open_leaves_ids.includes(marker.dataset.leaf_id) 																	// if no open leaf id matches the marker id
					|| workspace.getLeafById(marker.dataset.leaf_id).view.file.path !== marker.closest('.tree-item-self').dataset.path ) { 	// reused leaves keep the same id, so check path instead
						marker.remove();																									// remove the marker
					}					
				});
			});
		},100);
		workspace.onLayoutReady( () => { sleep(100).then( () => { highlight_open_files(); }); });												// initialize
		this.registerDomEvent(document,'mousedown', (e) => {
			if ( /mark_open_files/.test(e.target.className) ) {
				e.stopPropagation(); e.stopImmediatePropagation; e.preventDefault();
				e.target.addEventListener('mouseup', function(e) { 
					e.stopPropagation(); e.stopImmediatePropagation(); e.preventDefault();
				});
				e.target.addEventListener('click', function(e) { 
					e.stopPropagation(); e.stopImmediatePropagation(); e.preventDefault();
					workspace.setActiveLeaf(workspace.getLeafById(e.srcElement.dataset.leaf_id),{focus:true});
					workspace.getLeafById(e.srcElement.dataset.leaf_id).tabHeaderEl.click();												// trigger Continuous Mode plugin scroll into view
				});
			}
		});
		this.registerEvent(
			this.app.workspace.on('active-leaf-change', () => {
				highlight_open_files();
			})
		);
		this.registerEvent(
			this.app.workspace.on('layout-change', () => {
				highlight_open_files();
			})
		);
	} 
    // end onload
    // on plugin unload
	onunload() {
		this.app.workspace.getLeavesOfType('file-explorer')[0].view.tree.containerEl.querySelectorAll('.mark_open_files_container').forEach( 
			el => el.remove() 													// remove marker elements
		);
    }
	// load settings
    async loadSettings() {
		// this.settings = Object.assign({}, DEFAULT_SETTINGS, await this.loadData());
    }
    // save settings
    async saveSettings() {
		// await this.saveData(this.settings);
    }
}
module.exports = HighlightOpenFiles;