feat: update custom mode draft

2026-05-14 19:41:09 +00:00 · 2025-04-28 16:58:29 +08:00
parent 5558c96aa1
commit 497a9739d7
12 changed files with 2539 additions and 124 deletions
--- a/src/database/json/custom-mode/CustomModeManager.ts
+++ b/src/database/json/custom-mode/CustomModeManager.ts
@@ -0,0 +1,153 @@
+import fuzzysort from 'fuzzysort'
+import { App } from 'obsidian'
+import { v4 as uuidv4 } from 'uuid'
+
+import { AbstractJsonRepository } from '../base'
+import { CUSTOM_MODE_DIR, ROOT_DIR } from '../constants'
+import {
+	DuplicateCustomModeException,
+	EmptyCustomModeNameException,
+} from '../exception'
+
+import { CUSTOM_MODE_SCHEMA_VERSION, CustomMode, CustomModeMetadata } from './types'
+
+export class CustomModeManager extends AbstractJsonRepository<
+	CustomMode,
+	CustomModeMetadata
+> {
+	constructor(app: App) {
+		super(app, `${ROOT_DIR}/${CUSTOM_MODE_DIR}`)
+	}
+
+	protected generateFileName(mode: CustomMode): string {
+		// Format: v{schemaVersion}_name_id.json (with name encoded)
+		const encodedName = encodeURIComponent(mode.name)
+		return `v${CUSTOM_MODE_SCHEMA_VERSION}_${encodedName}_${mode.id}.json`
+	}
+
+	protected parseFileName(fileName: string): CustomModeMetadata | null {
+		const match = fileName.match(
+			new RegExp(`^v${CUSTOM_MODE_SCHEMA_VERSION}_(.+)_([0-9a-f-]+)\\.json$`),
+		)
+		if (!match) return null
+
+		const encodedName = match[1]
+		const id = match[2]
+		const name = decodeURIComponent(encodedName)
+
+		return {
+			id,
+			name,
+			updatedAt: Date.now(),
+			schemaVersion: CUSTOM_MODE_SCHEMA_VERSION,
+		}
+	}
+
+	public async createCustomMode(
+		customMode: Omit<
+			CustomMode,
+			'id' | 'slug' | 'createdAt' | 'updatedAt' | 'schemaVersion'
+		>,
+	): Promise<CustomMode> {
+		if (customMode.name !== undefined && customMode.name.length === 0) {
+			throw new EmptyCustomModeNameException()
+		}
+
+		const existingCustomMode = await this.findByName(customMode.name)
+		if (existingCustomMode) {
+			throw new DuplicateCustomModeException(customMode.name)
+		}
+
+		const newCustomMode: CustomMode = {
+			id: uuidv4(),
+			...customMode,
+			slug: customMode.name.toLowerCase().replace(/ /g, '-'),
+			updatedAt: Date.now(),
+			schemaVersion: CUSTOM_MODE_SCHEMA_VERSION,
+		}
+
+		await this.create(newCustomMode)
+		return newCustomMode
+	}
+
+	public async ListCustomModes(): Promise<CustomMode[]> {
+		const allMetadata = await this.listMetadata()
+		const allCustomModes = await Promise.all(allMetadata.map(async (meta) => this.read(meta.fileName)))
+		return allCustomModes.sort((a, b) => b.updatedAt - a.updatedAt)
+	}
+
+	public async findById(id: string): Promise<CustomMode | null> {
+		const allMetadata = await this.listMetadata()
+		const targetMetadata = allMetadata.find((meta) => meta.id === id)
+
+		if (!targetMetadata) return null
+
+		return this.read(targetMetadata.fileName)
+	}
+
+	public async findByName(name: string): Promise<CustomMode | null> {
+		const allMetadata = await this.listMetadata()
+		const targetMetadata = allMetadata.find((meta) => meta.name === name)
+
+		if (!targetMetadata) return null
+
+		return this.read(targetMetadata.fileName)
+	}
+
+	public async updateCustomMode(
+		id: string,
+		updates: Partial<
+			Omit<CustomMode, 'id' | 'slug' | 'createdAt' | 'updatedAt' | 'schemaVersion'>
+		>,
+	): Promise<CustomMode | null> {
+		if (updates.name !== undefined && updates.name.length === 0) {
+			throw new EmptyCustomModeNameException()
+		}
+
+		const customMode = await this.findById(id)
+		if (!customMode) return null
+
+		if (updates.name && updates.name !== customMode.name) {
+			const existingCustomMode = await this.findByName(updates.name)
+			if (existingCustomMode) {
+				throw new DuplicateCustomModeException(updates.name)
+			}
+		}
+
+		const updatedCustomMode: CustomMode = {
+			...customMode,
+			...updates,
+			updatedAt: Date.now(),
+		}
+
+		await this.update(customMode, updatedCustomMode)
+		return updatedCustomMode
+	}
+
+	public async deleteCustomMode(id: string): Promise<boolean> {
+		const customMode = await this.findById(id)
+		if (!customMode) return false
+
+		const fileName = this.generateFileName(customMode)
+		await this.delete(fileName)
+		return true
+	}
+
+	public async searchCustomModes(query: string): Promise<CustomMode[]> {
+		const allMetadata = await this.listMetadata()
+		const results = fuzzysort.go(query, allMetadata, {
+			keys: ['name'],
+			threshold: 0.2,
+			limit: 20,
+			all: true,
+		})
+
+		const customModes = (
+			await Promise.all(
+				results.map(async (result) => this.read(result.obj.fileName)),
+			)
+		).filter((customMode): customMode is CustomMode => customMode !== null)
+
+		return customModes
+	}
+}