From d1095f6103493089c71570c3547a31f964aa0b25 Mon Sep 17 00:00:00 2001
From: Reorx <novoreorx@gmail.com>
Date: Tue, 19 Apr 2022 01:08:19 +0800
Subject: [PATCH] use generateMarkdownLink to generate markdown links

---
 src/main.ts | 62 +++++++++++++++++++++++++++++------------------------
 1 file changed, 34 insertions(+), 28 deletions(-)

diff --git a/src/main.ts b/src/main.ts
index b0df193..f8ce6cc 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -85,22 +85,39 @@ export default class PasteImageRenamePlugin extends Plugin {
 	}
 
 	async renameImage(file: TFile, autoRename: boolean = false) {
-		const { stem, newName, isMeaningful }= this.generateNewName(file)
+		// get active file first
+		const activeFile = this.getActiveFile()
+		if (!activeFile) {
+			new Notice('Error: No active file found.')
+			return
+		}
+
+		const { stem, newName, isMeaningful }= this.generateNewName(file, activeFile)
 		debugLog('generated newName:', newName, isMeaningful)
 
 		if (!isMeaningful || !autoRename) {
-			this.openRenameModal(file, isMeaningful ? stem : '')
+			this.openRenameModal(file, isMeaningful ? stem : '', activeFile.path)
 			return
 		}
-		this.renameFile(file, newName)
+		this.renameFile(file, newName, activeFile.path)
 	}
 
-	async renameFile(file: TFile, newName: string) {
+	async renameFile(file: TFile, newName: string, sourcePath: string) {
 		// deduplicate name
 		newName = await this.deduplicateNewName(newName, file)
 		debugLog('deduplicated newName:', newName)
 		const originName = file.name
 
+		// get vault config, determine whether useMarkdownLinks is set
+		const vaultConfig = getVaultConfig(this.app)
+		let useMarkdownLinks = false
+		if (vaultConfig && vaultConfig.useMarkdownLinks) {
+			useMarkdownLinks = true
+		}
+
+		// get origin file link before renaming
+		const linkText = this.makeLinkText(originName, useMarkdownLinks, file, sourcePath)
+
 		// file system operation
 		const newPath = path.join(file.parent.path, newName)
 		try {
@@ -109,6 +126,8 @@ export default class PasteImageRenamePlugin extends Plugin {
 			new Notice(`Failed to rename ${newName}: ${err}`)
 			throw err
 		}
+		const newLinkText = this.makeLinkText(newName, useMarkdownLinks, this.app.vault.getAbstractFileByPath(newPath) as TFile, sourcePath)
+		debugLog('replace text', linkText, newLinkText)
 
 		// in case fileManager.renameFile may not update the internal link in the active file,
 		// we manually replace by manipulating the editor
@@ -118,20 +137,10 @@ export default class PasteImageRenamePlugin extends Plugin {
 			return
 		}
 
-		// get vault config, determine whether useMarkdownLinks is set
-		const vaultConfig = getVaultConfig(this.app)
-		let useMarkdownLinks = false
-		if (vaultConfig && vaultConfig.useMarkdownLinks) {
-			useMarkdownLinks = true
-		}
-
 		const cursor = editor.getCursor()
 		const line = editor.getLine(cursor.line)
 		debugLog('current line', line)
 		// console.log('editor context', cursor, )
-		const linkText = this.makeLinkText(originName, useMarkdownLinks),
-			newLinkText = this.makeLinkText(newName, useMarkdownLinks);
-		debugLog('replace text', linkText, newLinkText)
 		editor.transaction({
 			changes: [
 				{
@@ -145,20 +154,20 @@ export default class PasteImageRenamePlugin extends Plugin {
 		new Notice(`Renamed ${originName} to ${newName}`)
 	}
 
-	makeLinkText(fileName: string, useMarkdownLinks: boolean): string {
+	makeLinkText(fileName: string, useMarkdownLinks: boolean, file: TFile, sourcePath: string): string {
 		if (useMarkdownLinks) {
-			return `[](${encodeURI(fileName)})`
+			return this.app.fileManager.generateMarkdownLink(file, sourcePath)
 		} else {
 			return `[[${fileName}]]`
 		}
 	}
 
-	openRenameModal(file: TFile, newName: string) {
+	openRenameModal(file: TFile, newName: string, sourcePath: string) {
 		const modal = new ImageRenameModal(
 			this.app, file as TFile, newName,
 			(confirmedName: string) => {
 				debugLog('confirmedName:', confirmedName)
-				this.renameFile(file, confirmedName)
+				this.renameFile(file, confirmedName, sourcePath)
 			},
 			() => {
 				this.modals.splice(this.modals.indexOf(modal), 1)
@@ -170,17 +179,14 @@ export default class PasteImageRenamePlugin extends Plugin {
 	}
 
 	// returns a new name for the input file, with extension
-	generateNewName(file: TFile) {
+	generateNewName(file: TFile, activeFile: TFile) {
 		let imageNameKey = ''
-		const activeFile = this.getActiveFile()
-		if (activeFile) {
-			const fileCache = this.app.metadataCache.getFileCache(activeFile)
-			if (fileCache) {
-				debugLog('frontmatter', fileCache.frontmatter)
-				imageNameKey = fileCache.frontmatter?.imageNameKey || ''
-			} else {
-				console.warn('could not get file cache from active file', activeFile.name)
-			}
+		const fileCache = this.app.metadataCache.getFileCache(activeFile)
+		if (fileCache) {
+			debugLog('frontmatter', fileCache.frontmatter)
+			imageNameKey = fileCache.frontmatter?.imageNameKey || ''
+		} else {
+			console.warn('could not get file cache from active file', activeFile.name)
 		}
 
 		const stem = renderTemplate(this.settings.imageNamePattern, {