pjeby_note-folder-autorename/plugin.js
2021-02-07 00:25:53 -05:00

110 lines
4.3 KiB
JavaScript

import {Plugin, Notice} from "obsidian";
import * as obsidian from "obsidian";
export default class NotesAsFolders extends Plugin {
onload() {
this.addCommand(
{id: "make-folder-note", name: "Make this note a folder note", checkCallback: this.noteToFolder.bind(this)}
)
// Ensure these can't be called until their previous call completes
this.register(serializeInstanceMethod(this, "onRenameFile"));
this.register(serializeInstanceMethod(this.app.fileManager, "renameFile"));
this.register(serializeInstanceMethod(this.app.fileManager, "updateInternalLinks"));
this.registerEvent(this.app.vault.on("rename", this.onRenameFile.bind(this)));
// Track changes that might mean a post-rename link update is still in progress
this.metaDataIsClean = Promise.resolve();
this.markClean = () => {};
this.registerEvent(this.app.vault.on("modify", () => this.metaDataIsClean = new Promise((res) => { this.markClean = res; })));
this.registerEvent(this.app.metadataCache.on("resolved", () => this.markClean()));
}
noteToFolder(check) {
const note = this.app.workspace.getActiveFile();
if ( !note || note.extension !== "md" || isFolderNote(note.path)) return false;
if (check) return true;
return (async () => {
const destination = dirname(note.path) + "/" + note.basename;
const existing = this.app.vault.getAbstractFileByPath(destination);
if ( !existing ) {
await this.app.vault.createFolder(destination);
} else {
return new Notice(`A file or folder named ${note.basename} is in the way; can't create folder`);
}
// Move the note into the folder
await this.safeRename(note, destination + "/" + note.name);
})()
}
async safeRename(file, newName) {
// Let any current rename in-progress finish
await this.app.fileManager.renameFile.after();
while(document.body.find(".modal-container")) {
// if modal is displayed, need to wait for it to close
await new Promise((res) => setTimeout(res, 250));
}
// Let the link updates run, and wait for the cache to catch up
await this.app.fileManager.updateInternalLinks.after();
await this.metaDataIsClean;
return await this.app.fileManager.renameFile(file, newName);
}
async onRenameFile(file, oldName) {
// We only care about notes that were already folder notes
if (file.extension !== "md" || !isFolderNote(oldName)) return;
const
oldDir = folderBasename(oldName),
oldPath = dirname(oldName),
newPath = dirname(file.path),
oldFolder = this.app.vault.getAbstractFileByPath(oldPath)
;
// If a folder note was renamed inside its folder, rename the folder to match
if (oldPath === newPath) {
const destination = dirname(oldPath) + "/" + file.basename;
await this.safeRename(oldFolder, destination);
}
// If a folder note was moved to a new folder,
else if (file.basename === oldDir) {
const destination = newPath + "/" + oldDir;
// Move the folder alongside the note,
await this.safeRename(oldFolder, destination);
// Then move the note back into the folder
await this.safeRename(file, destination + "/" + file.name);
return;
}
}
}
function folderBasename(path) { return path.split("/").slice(-2, -1).pop(); }
function basename(path) { return path.split("/").pop(); }
function dirname(path) { return path.split("/").slice(0, -1).join("/"); }
function isFolderNote(path) { return basename(path) === folderBasename(path) + ".md"; }
function serializeInstanceMethod(obj, method) {
let lastRun = Promise.resolve();
obj[method] = function(...args) {
return lastRun = new Promise((res, rej) => {
const wrap = () => {
this.constructor.prototype[method].apply(this, args).then(res, rej);
};
lastRun.then(wrap, wrap);
});
}
obj[method].after = function () {
return lastRun = new Promise((res, rej) => { lastRun.then(res, res); });
}
return () => { delete obj[method]; }
}