Obsidian/.obsidian/plugins/obsidian-icon-folder-1.0.0/src/util.ts

import * as remixicons from 'react-icons/ri/index';
// @ts-ignore
import { renderToString } from 'react-dom/server';
import IconFolderPlugin from './main';

export const waitForNode = (selector: string): Promise<Element> => {
  return new Promise((resolve) => {
    if (document.querySelector(selector)) {
      return resolve(document.querySelector(selector));
    }

    const observer = new MutationObserver(() => {
      if (document.querySelector(selector)) {
        resolve(document.querySelector(selector));
        observer.disconnect();
      }
    });

    observer.observe(document.body, {
      childList: true,
      subtree: true,
    });
  });
};

export const removeFromDOM = (path: string) => {
  const node = document.querySelector(`[data-path="${path}"]`);
  if (!node) {
    console.error('element with data path not found', path);
    return;
  }

  const iconNode = node.querySelector('.obsidian-icon-folder-icon');
  if (!iconNode) {
    console.error('icon element does not exist', path);
    return;
  }

  iconNode.remove();
};

export const addToDOMWithElement = (
  plugin: IconFolderPlugin,
  path: string,
  iconId: string,
  node: Element,
): void => {
  const titleNode = node.querySelector('.nav-folder-title-content');
  if (!titleNode) {
    console.error('element with title not found');
    return;
  }

  const iconNode = document.createElement('div');
  iconNode.classList.add('obsidian-icon-folder-icon');
  iconNode.innerHTML = renderToString(
    // @ts-ignore
    remixicons[iconId]({
      size: '16px',
    }),
  );

  node.insertBefore(iconNode, titleNode);
};

export const addToDOM = (
  plugin: IconFolderPlugin,
  path: string,
  iconId: string,
): void => {
  const node = document.querySelector(`[data-path="${path}"]`);
  if (!node) {
    console.error('element with data path not found', path);
    return;
  }

  addToDOMWithElement(plugin, path, iconId, node);
};