Merge pull request #16 from edrichhans/master

README.md

@@ -104,6 +104,15 @@ source: file
 ```
 ~~~
 
+#### Show all words using a folder/file
+
+~~~markdown
+```wordcloud
+source: file
+query: 'Folder/File'
+```
+~~~
+
 ### Options
 
 | **Name**  | **Description**                                                                                                 | **Possible Values** | **Default** |

src/functions.ts

@@ -10,7 +10,7 @@ export function removeMarkdown(text: string): string {
 		.replace(/\[\[(.*(?=\|))(.*)\]\]/g, '$2') //wikilinks with alias
 		.replace(/\[\[([\s\S]*?)\]\]/gm, '$1') //wikilinks
 		.replace(/- ?\[.?\]/gm, '') //tasks
-		.replace(/%%.*?%%/gm, '')//Obsidian Comments
+		.replace(/%%(.|\n)*?%%/gm, '')//Obsidian Comments
 		.replace(/`([\s\S]*?)`/gm, '') //codeblocks, inline & normal
 		.replace(/\[\^[[\s\S]]*\]/g, '') //footnotes
 		.replace(/\^\[([\s\S]*?)\]/g, '$1') //inline footnotes
@@ -36,20 +36,13 @@ export function removeStopwords(words: Record<string, number>, customStopwords:
 }
 
 export async function getWords(text: string): Promise<string[]> {
-	const words = text.split(/[\n\s]/g);
-	const output: string[] = [];
-	for (let word of words) {
-		const result = removeMarkdown(word).toLocaleLowerCase();
-		if(result.length > 0) {
-			output.push(result);
-		}
-	}
-	return output;
+	const cleanText = removeMarkdown(text).toLocaleLowerCase();
+	return cleanText.split(/[\n\s]/g);
 }
 
 export async function convertToMap(words: string[]): Promise<Record<string, number>> {
 	const record: Record<string, number> = {};
-	for (let word of words) {
+	for (const word of words) {
 		const element = record[word];
 		if(element) {
 			record[word] = element + 1;

src/wordcloud.ts

@@ -1,5 +1,6 @@
-import TagCloudPlugin from "./main";
+import TagCloudPlugin, {logger} from "./main";
 import {MarkdownPostProcessorContext, TFile} from "obsidian";
+import {getAPI} from "obsidian-dataview";
 import {convertToMap, getWords, recordToArray, removeStopwords} from "./functions";
 import WordCloud from "wordcloud";
 
@@ -50,32 +51,41 @@ export class Wordcloud {
 			}
 		}
 		if (options.source === 'query') {
-			el.createEl("p", {cls: "cloud-error"}).setText("Queries are not supported in a wordcloud");
-			return;
-		}
-		/*this part is really not performant, need to find a better solution.
-		if(options.source === 'query') {
 			const dataviewAPI = getAPI();
-			if(dataviewAPI === undefined) {
-				el.createEl("p").setText("Dataview is not installed, but is required to use queries");
+			if (dataviewAPI === undefined) {
+				el.createEl("p", {cls: "cloud-error"}).setText("Dataview is not installed, but is required to use queries");
 				return;
 			}
 
-			const pages = dataviewAPI.pages(options.query, ctx.sourcePath);
-			const words : string[] = [];
-			for (let page of pages) {
-				const file = this.app.vault.getAbstractFileByPath(page.file.path);
-				if (file === undefined) return;
-				if (!(file instanceof TFile)) return;
-				const fileContent = await this.app.vault.read(file);
-				words.push(...this.getWords(fileContent));
-			}
-			if(options.stopwords) {
-				content = this.convertToMap(this.removeStopwords(words));
-			}else {
-				content = this.convertToMap(words);
+			try {
+				const query = options.query;
+				if(!query) {
+					el.createEl('p', {cls: "cloud-error"}).setText("query option is required");
+					return;
+				}
+
+				const page = dataviewAPI.page(query);
+
+				if(!page) {
+					el.createEl('p', {cls: "cloud-error"}).setText("Page not found");
+					return;
+				}
+
+				const rawContent = await dataviewAPI.io.load(page.file.path)
+				const parsedWords = await getWords(rawContent)
+				content = await convertToMap(parsedWords);
+
+				if(options.stopwords) {
+					const tmp = this.plugin.settings.stopwords.split("\n");
+					const customStopwords = new Set<string>(tmp);
+					content = removeStopwords(content, customStopwords)
+				}
+			} catch (error: any) { // eslint-disable-line @typescript-eslint/no-explicit-any
+				el.createEl('p', {cls: "cloud-error"}).setText(error.toString());
+				logger.error(error);
+				return;
 			}
-		}*/
+		}
 
 		el.empty();