colormap.ts

import { Tag } from "model";
import { index, space } from "$sb/silverbullet-syscall/mod.ts";
import { readGraphviewSettings } from "utils";
import { PageMeta } from "../silverbullet/common/types";

export class ColorMap {
  page: string;
  color: string;

  constructor(page: string, color: string) {
    this.page = page;
    this.color = color;
  }
}

export class ColorMapBuilder {
  colorMapSettings: any;
  colorMapPathSettings: any;
  colorMapTagSettings: any;
  spacetags: Tag[];
  spacepages: PageMeta[]
  taggedPages: string[];
  individuallyTaggedPages: any;
  default_color: any;

  async init(): Promise<void> {
    // Read settings
    this.colorMapSettings = await readGraphviewSettings("colormap");
    console.log(this.colorMapSettings);
    this.colorMapPathSettings = this.colorMapSettings ? this.colorMapSettings["path"] : [];
    this.colorMapTagSettings = this.colorMapSettings ? this.colorMapSettings["tag"] : [];

    // Get all tags
    this.spacetags = await index.queryPrefix("tag:");
    this.taggedPages = [...new Set(this.spacetags.map((tag) => tag.page))];
    this.individuallyTaggedPages = await index.queryPrefix("tag:node_color=");

    // Get all pages
    this.spacepages = await space.listPages();

    // Get default color
    this.default_color = await readGraphviewSettings("default_color");
  }

  build(): ColorMap[] {
    // Iterate over all pages
    return this.spacepages.map((page) => {
      // Get all tags of page
      const pageTags = this.spacetags.filter((tag) => tag.page === page.name);

      // If page has tag with "tag:node_color=" → use color from tag and continue
      if (this.individuallyTaggedPages.find((t) => t.page === page.name)) {
        return { "page": page.name, "color": this.individuallyTaggedPages.find((t) => t.page === page.name).value.split("=")[1] };
      }

      // If page has a tag from colorMapSettings ["tag"] →  map color code to page name and continue
      if (this.colorMapTagSettings) {
        // check, if any of the tags is in colorMapSettings
        const pageTagsInColorMapSettings = pageTags.filter((tag) =>
          this.colorMapTagSettings[tag.value] !== undefined,
        );
        // if yes, use color from colorMapSettings
        if (pageTagsInColorMapSettings.length > 0) {
          return { "page": page.name, "color": this.colorMapTagSettings[pageTagsInColorMapSettings[0].value] };
        }
      }

      // If page name begins with element colorMapSettings ["path"] → map color code to page name and continue
      if (this.colorMapPathSettings) {
        // console.log(Object.keys(this.colorMapPathSettings))
        const pageNameBeginsWithPath = Object.keys(this.colorMapPathSettings)
          .find((path: string) => page.name.startsWith(path));
        if (pageNameBeginsWithPath) {
          return { "page": page.name, "color": this.colorMapPathSettings[pageNameBeginsWithPath] };
        }
      }

      // Use default color
      return { "page": page.name, "color": this.default_color ? this.default_color : "000000" };
    });
  }
}