add custom frontmatter

src/upload/uploadCommand.ts

@@ -2,10 +2,12 @@ import { i18nConfig } from "../lang/I18n";
 import { App, Notice } from "obsidian";
 import { Upload2NotionNext } from "./upload_next/Upload2NotionNext";
 import { Upload2NotionGeneral } from "./upload_general/Upload2NotionGeneral";
+import { Upload2NotionCustom } from "./upoload_custom/Upload2NotionCustom";
 import { PluginSettings } from "../ui/settingTabs";
 import ObsidianSyncNotionPlugin from "../main";
 import { getNowFileMarkdownContentNext } from "./upload_next/getMarkdownNext";
 import { getNowFileMarkdownContentGeneral } from "./upload_general/getMarkdownGeneral";
+import {getNowFileMarkdownContentCustom} from "./upoload_custom/getMarkdownCustom";
 
 export async function uploadCommandNext(
     plugin: ObsidianSyncNotionPlugin,
@@ -78,3 +80,36 @@ export async function uploadCommandGeneral(
 
     }
 }
+
+
+export async function uploadCommandCustom(
+    plugin: ObsidianSyncNotionPlugin,
+    settings: PluginSettings,
+    app: App,
+) {
+
+    const { notionAPIGeneral, databaseIDGeneral } = settings;
+
+    // Check if the user has set up the Notion API and database ID
+    if (notionAPIGeneral === "" || databaseIDGeneral === "") {
+        const setAPIMessage = i18nConfig["set-api-id"];
+        new Notice(setAPIMessage);
+        return;
+    }
+
+    const { markDownData, nowFile, cover, tags ,customValues} = await getNowFileMarkdownContentCustom(app, settings)
+
+    if (markDownData) {
+        const { basename } = nowFile;
+
+        const upload = new Upload2NotionCustom(plugin);
+        const res = await upload.syncMarkdownToNotionCustom(basename, cover, tags, customValues, markDownData, nowFile, this.app);
+
+        if (res.status === 200) {
+            new Notice(`${i18nConfig["sync-success"]}${basename}`);
+        } else {
+            new Notice(`${i18nConfig["sync-fail"]}${basename}`, 5000);
+        }
+
+    }
+}

src/upload/upoload_custom/BaseUpload2NotionCustom.ts

@@ -0,0 +1,90 @@
+import {App, Notice, requestUrl, TFile} from "obsidian";
+import {Client} from '@notionhq/client';
+import {markdownToBlocks,} from "@tryfabric/martian";
+import * as yamlFrontMatter from "yaml-front-matter";
+// import * as yaml from "yaml"
+import MyPlugin from "src/main";
+
+export class UploadBaseCustom {
+    plugin: MyPlugin;
+    notion: Client;
+    agent: any;
+
+    constructor(plugin: MyPlugin) {
+        this.plugin = plugin;
+    }
+
+    async deletePage(notionID: string) {
+        return requestUrl({
+            url: `https://api.notion.com/v1/blocks/${notionID}`,
+            method: 'DELETE',
+            headers: {
+                'Content-Type': 'application/json',
+                'Authorization': 'Bearer ' + this.plugin.settings.notionAPINext,
+                'Notion-Version': '2022-06-28',
+            },
+            body: ''
+        });
+    }
+
+    async getDataBase(databaseID: string) {
+        const response = await requestUrl({
+                url: `https://api.notion.com/v1/databases/${databaseID}`,
+                method: 'GET',
+                headers: {
+                    'Authorization': 'Bearer ' + this.plugin.settings.notionAPINext,
+                    'Notion-Version': '2022-06-28',
+                }
+            }
+        )
+
+        // Check if cover is present in the JSON response and then get the URL
+        if (response.json.cover && response.json.cover.external) {
+            return response.json.cover.external.url;
+        } else {
+            return null;  // or some other default value, if you prefer
+        }
+    }
+
+    // async updateYamlInfo(yamlContent: string, nowFile: TFile, res: any, app: App, settings: any) {
+    //     let {url, id} = res.json
+    //     // replace www to notionID
+    //     const {notionUser} = this.plugin.settings;
+    //
+    //     if (notionUser !== "") {
+    //         // replace url str "www" to notionID
+    //         url = url.replace("www.notion.so", `${notionUser}.notion.site`)
+    //     }
+    //
+    //     await app.fileManager.processFrontMatter(nowFile, yamlContent => {
+    //         if (yamlContent['notionID']) {
+    //             delete yamlContent['notionID']
+    //         }
+    //         if (yamlContent['link']) {
+    //             delete yamlContent['link']
+    //         }
+    //         // add new notionID and link
+    //         yamlContent.notionID = id;
+    //         yamlContent.link = url;
+    //     });
+    //
+    //     try {
+    //         await navigator.clipboard.writeText(url)
+    //     } catch (error) {
+    //         new Notice(`复制链接失败，请手动复制${error}`)
+    //     }
+    //     // const __content = yamlContent.__content;
+    //     // delete yamlContent.__content
+    //     // const yamlhead = yaml.stringify(yamlContent)
+    //     // //  if yamlhead hava last \n  remove it
+    //     // const yamlhead_remove_n = yamlhead.replace(/\n$/, '')
+    //     // // if __content have start \n remove it
+    //     // const __content_remove_n = __content.replace(/^\n/, '')
+    //     // const content = '---\n' +yamlhead_remove_n +'\n---\n' + __content_remove_n;
+    //     // try {
+    //     // 	await nowFile.vault.modify(nowFile, content)
+    //     // } catch (error) {
+    //     // 	new Notice(`write file error ${error}`)
+    //     // }
+    // }
+}

src/upload/upoload_custom/Upload2NotionCustom.ts

@@ -0,0 +1,157 @@
+import { App, Notice, requestUrl, TFile } from "obsidian";
+import { Client } from "@notionhq/client";
+import { markdownToBlocks } from "@tryfabric/martian";
+import * as yamlFrontMatter from "yaml-front-matter";
+// import * as yaml from "yaml"
+import MyPlugin from "src/main";
+import { PluginSettings } from "../../ui/settingTabs";
+import { updateYamlInfo } from "../updateYaml";
+import {UploadBaseCustom} from "./BaseUpload2NotionCustom";
+
+export class Upload2NotionCustom extends UploadBaseCustom {
+	settings: PluginSettings;
+
+	constructor(plugin: MyPlugin) {
+		super(plugin);
+	}
+
+	// 因为需要解析notion的block进行对比，非常的麻烦，
+	// 暂时就直接删除，新建一个page
+	async updatePage(
+		notionID: string,
+		title: string,
+		cover: string,
+		tags: string[],
+		customValues: Record<string, string>,
+		childArr: any,
+	) {
+		await this.deletePage(notionID);
+
+		const databasecover = await this.getDataBase(
+			this.plugin.settings.databaseIDGeneral,
+		);
+
+		if (cover == null) {
+			cover = databasecover;
+		}
+
+		return await this.createPage(title, cover, tags, customValues, childArr);
+	}
+
+	async createPage(
+		title: string,
+		cover: string,
+		tags: string[],
+		customValues: Record<string, string>,
+		childArr: any,
+	) {
+		const bodyString: any = {
+			parent: {
+				database_id: this.plugin.settings.databaseIDGeneral,
+			},
+			properties: {
+				[this.plugin.settings.CustomTitleButton
+					? this.plugin.settings.CustomTitleName
+					: "title"]: {
+					title: [
+						{
+							text: {
+								content: title,
+							},
+						},
+					],
+				},
+				...(Object.keys(customValues).reduce((acc, key) => {
+					acc[key] = {
+						rich_text: [
+							{
+								text: {
+									content: customValues[key] || '',
+								},
+							},
+						],
+					};
+					return acc;
+				}, {} as Record<string, { rich_text: { text: { content: string } }[] } >)),
+			},
+			children: childArr,
+		};
+
+		if (cover) {
+			bodyString.cover = {
+				type: "external",
+				external: {
+					url: cover,
+				},
+			};
+		}
+
+		if (!bodyString.cover && this.plugin.settings.bannerUrl) {
+			bodyString.cover = {
+				type: "external",
+				external: {
+					url: this.plugin.settings.bannerUrl,
+				},
+			};
+		}
+
+		try {
+			return await requestUrl({
+				url: `https://api.notion.com/v1/pages`,
+				method: "POST",
+				headers: {
+					"Content-Type": "application/json",
+					// 'User-Agent': 'obsidian.md',
+					Authorization:
+						"Bearer " + this.plugin.settings.notionAPIGeneral,
+					"Notion-Version": "2022-06-28",
+				},
+				body: JSON.stringify(bodyString),
+			});
+		} catch (error) {
+			new Notice(`network error ${error}`);
+		}
+	}
+
+	async syncMarkdownToNotionCustom(
+		title: string,
+		cover: string,
+		tags: string[],
+		customValues: Record<string, string>,
+		markdown: string,
+		nowFile: TFile,
+		app: App,
+	): Promise<any> {
+		const options = {
+			notionLimits: {
+				truncate: false,
+			}
+		}
+		let res: any;
+		const yamlContent: any = yamlFrontMatter.loadFront(markdown);
+		const __content = yamlContent.__content;
+		const file2Block = markdownToBlocks(__content, options);
+		const frontmasster =
+			app.metadataCache.getFileCache(nowFile)?.frontmatter;
+		const notionID = frontmasster ? frontmasster.notionID : null;
+
+		if (notionID) {
+			res = await this.updatePage(
+				notionID,
+				title,
+				cover,
+				tags,
+				customValues,
+				file2Block,
+			);
+		} else {
+			res = await this.createPage(title, cover, tags, customValues, file2Block);
+		}
+		if (res.status === 200) {
+			await updateYamlInfo(markdown, nowFile, res, app, this.plugin);
+		} else {
+			new Notice(`${res.text}`);
+		}
+		return res;
+	}
+}

src/upload/upoload_custom/getMarkdownCustom.ts

@@ -0,0 +1,43 @@
+import {App, Notice} from "obsidian";
+import {i18nConfig} from "../../lang/I18n";
+import {PluginSettings} from "../../ui/settingTabs";
+
+export async function getNowFileMarkdownContentCustom(
+    app: App,
+    settings: PluginSettings,
+) {
+    const nowFile = app.workspace.getActiveFile();
+    let cover = '';
+    let tags = [];
+    let customValues: Record<string, string> = {};
+
+    const FileCache = app.metadataCache.getFileCache(nowFile);
+    try {
+            cover = FileCache.frontmatter.coverurl;
+            tags = FileCache.frontmatter.tags;
+
+            // split the CustomValues into an array
+            const customValuesNames = settings.CustomValues.split('\n').map(value => value.trim());
+
+            // get the custom values from the frontmatter
+            customValuesNames.forEach(valueName => {
+                customValues[valueName] = FileCache.frontmatter[valueName];
+            });
+    } catch (error) {
+        new Notice(i18nConfig["set-tags-fail"]);
+    }
+
+    if (nowFile) {
+        const markDownData = await nowFile.vault.read(nowFile);
+        return {
+            markDownData,
+            nowFile,
+            cover,
+            tags,
+            customValues,
+        };
+    } else {
+        new Notice(i18nConfig["open-file"]);
+        return;
+    }
+}