Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-26 11:44:28 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

refactor(ocr): change provider listing to include db data

Browse Source 
Replace simple registry key listing with combined db query to filter available providers
This commit is contained in:
icarus 2025-10-20 06:47:44 +08:00
parent 78126c3d0b
commit 4dcfe276ac
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/main/services/ocr/OcrService.ts
View File

@ -1,5 +1,7 @@
import { dbService } from '@data/db/DbService'
import { ocrProviderTable } from '@data/db/schemas/ocr/provider'
import { loggerService } from '@logger'
import type { OcrParams, OcrResult, SupportedOcrFile } from '@types'
import type { ListOcrProvidersResponse, OcrParams, OcrResult, SupportedOcrFile } from '@types'
import { BuiltinOcrProviderIds } from '@types'
import type { OcrBaseService } from './builtin/OcrBaseService'
@ -28,19 +30,23 @@ export class OcrService {
}
}
register(providerId: string, service: OcrBaseService): void {
private register(providerId: string, service: OcrBaseService): void {
if (this.registry.has(providerId)) {
logger.warn(`Provider ${providerId} has existing handler. Overwrited.`)
}
this.registry.set(providerId, service)
}
unregister(providerId: string): void {
// @ts-expect-error not used for now, but just keep it.
private unregister(providerId: string): void {
this.registry.delete(providerId)
}
public listProviderIds(): string[] {
return Array.from(this.registry.keys())
public async listProviders(): Promise<ListOcrProvidersResponse> {
const registeredKeys = Array.from(this.registry.keys())
const providers = await dbService.getDb().select().from(ocrProviderTable)
return { data: providers.filter((p) => registeredKeys.includes(p.id)) }
}
public async ocr(file: SupportedOcrFile, params: OcrParams): Promise<OcrResult> {