vscode-front-matter/src/commands/Folders.ts

import { commands, Uri, workspace, window } from "vscode";
import { CONFIG_KEY, EXTENSION_NAME, SETTINGS_CONTENT_FOLDERS } from "../constants";
import { basename } from "path";
import { ContentFolder } from "../models";
import uniqBy = require("lodash.uniqby");
import { VscodeCollapsible } from "@bendera/vscode-webview-elements/dist/vscode-collapsible";
import { Template } from "./Template";

export class Folders {

  public static async create() {
    const folders = Folders.get();

    if (!folders || folders.length === 0) {
      window.showWarningMessage(`${EXTENSION_NAME}: There are no known content locations defined in this project.`);
      return;
    }

    const selectedFolder = await window.showQuickPick(folders.map(f => f.title), {
      placeHolder: `Select where you want to create your content`
    });

    if (!selectedFolder) {
      window.showWarningMessage(`${EXTENSION_NAME}: You didn't select a place where you wanted to create your content.`);
      return;
    }

    const location = folders.find(f => f.title === selectedFolder);
    if (location) {
      const folderPath = Folders.getFolderPath(Uri.file(location.fsPath));
      if (folderPath) {
        Template.create(folderPath);
      }
    }
  }

  /**
   * Register the new folder path
   * @param folder 
   */
  public static async register(folder: Uri) {
    if (folder && folder.path) {
      const wslPath = folder.path.replace(/\//g, '\\');

      let folders = Folders.get();

      const exists = folders.find(f => f.paths.includes(folder.path) || f.paths.includes(wslPath));

      if (exists) {
        return window.showInformationMessage(`Folder is already registered`);
      }

      const folderName = await window.showInputBox({  
        prompt: `Which name would you like to specify for this folder?`,
        placeHolder: `Folder name`,
        value: basename(folder.fsPath)
      });

      folders.push({
        title: folderName,
        fsPath: folder.fsPath,
        paths: [
          folder.path, 
          wslPath
        ]
      } as ContentFolder);

      folders = uniqBy(folders, f => f.fsPath);
      await Folders.update(folders);
    }

		Folders.updateVsCodeCtx();
  }

  /**
   * Unregister a folder path
   * @param folder 
   */
  public static async unregister(folder: Uri) {
    if (folder && folder.path) {
      let folders = Folders.get();
      folders = folders.filter(f => f.fsPath !== folder.fsPath);
      await Folders.update(folders);
    }
    
    Folders.updateVsCodeCtx();
  }

  /**
   * Update the registered folders context
   */
  public static updateVsCodeCtx() {
    const folders = Folders.get();
    let allFolders: string[] = [];
    for (const folder of folders) {
      allFolders = [...allFolders, ...folder.paths]
    }
    commands.executeCommand('setContext', 'frontMatter.registeredFolders', allFolders);
  }

  /**
   * Retrieve the folder path
   * @param folder 
   * @returns 
   */
  public static getFolderPath(folder: Uri) {
    let folderPath = "";
		if (folder && folder.fsPath) {
			folderPath = folder.fsPath;
		} else if (workspace.workspaceFolders && workspace.workspaceFolders.length > 0) {
			folderPath = workspace.workspaceFolders[0].uri.fsPath;
		}
    return folderPath;
  }

  /**
   * Get the folder settings
   * @returns 
   */
  private static get() {
    const config = workspace.getConfiguration(CONFIG_KEY);
    const folders: ContentFolder[] = config.get(SETTINGS_CONTENT_FOLDERS) as ContentFolder[];
    return folders;
  }
  
  /**
   * Update the folder settings
   * @param folders 
   */
  private static async update(folders: ContentFolder[]) {
    const config = workspace.getConfiguration(CONFIG_KEY);
    await config.update(SETTINGS_CONTENT_FOLDERS, folders);
  }
}