packages/desktop/src/main/spellchecker/index.ts - develop - githubmirror/marktext

packages/desktop/src/main/spellchecker/index.ts
91 строка3 KB
Ran Luo
chore: convert repo to pnpm monorepo (packages/desktop, muyajs, website) (#4302)
29 май 2026, 05:25
29 май 2026, 05:25565bfcd
import { BrowserWindow, ipcMain } from 'electron'
import log from 'electron-log'
import { isOsx } from '../config'

/**
 * Add the given word to the spellchecker dictionary.
 */
export const addToDictionary = (win: BrowserWindow, word: string): boolean => {
  return win.webContents.session.addWordToSpellCheckerDictionary(word)
}

/**
 * Remove the given word from the spellchecker dictionary.
 */
export const removeFromDictionary = (win: BrowserWindow, word: string): boolean => {
  return win.webContents.session.removeWordFromSpellCheckerDictionary(word)
}

/**
 * Returns a list of all words in the custom dictionary.
 */
export const getCustomDictionaryWords = async(win: BrowserWindow): Promise<string[]> => {
  return win.webContents.session.listWordsInSpellCheckerDictionary()
}

/**
 * Sets whether to enable the builtin spell checker.
 */
export const setSpellCheckerEnabled = (win: BrowserWindow, enabled: boolean): boolean => {
  win.webContents.session.setSpellCheckerEnabled(enabled)
  return win.webContents.session.isSpellCheckerEnabled() === enabled
}

/**
 * Switch the spellchecker to the given language and enable the builtin spell checker.
 */
export const switchLanguage = (win: BrowserWindow, lang: string): void => {
  win.webContents.session.setSpellCheckerLanguages([lang])
}

/**
 * Returns the list of available spellchecker languages, or empty on macOS
 * where the OS spellchecker is used and language is auto-detected.
 */
export const getAvailableDictionaries = (win: BrowserWindow): string[] => {
  if (!win.webContents.session.isSpellCheckerEnabled) {
    console.warn('Spell Checker not available but dictionaries requested.')
    return []
  } else if (isOsx) {
    return []
  }

  const availableLanguages = win.webContents.session.availableSpellCheckerLanguages

  return availableLanguages.length > 0 ? availableLanguages : ['en-US']
}

const registerSpellcheckerHandlers = (): void => {
  ipcMain.handle('mt::spellchecker-remove-word', async(e, word: string) => {
    const win = BrowserWindow.fromWebContents(e.sender)
    if (!win) return false
    return removeFromDictionary(win, word)
  })
  ipcMain.handle('mt::spellchecker-switch-language', async(e, lang: string) => {
    const win = BrowserWindow.fromWebContents(e.sender)
    if (win) switchLanguage(win, lang)
    return null
  })
  ipcMain.handle('mt::spellchecker-get-available-dictionaries', async(e) => {
    const win = BrowserWindow.fromWebContents(e.sender)
    if (!win) return []
    return getAvailableDictionaries(win)
  })
  // We have to set a language or call `switchLanguage` on Linux and Windows.
  ipcMain.handle('mt::spellchecker-set-enabled', async(e, enabled: boolean) => {
    const win = BrowserWindow.fromWebContents(e.sender)
    if (!win) return false
    if (!setSpellCheckerEnabled(win, enabled)) {
      log.warn(`Failed to (de-)activate spell checking on editor (id=${win.id}).`)
      return false
    }
    return true
  })
  ipcMain.handle('mt::spellchecker-get-custom-dictionary-words', async(e) => {
    const win = BrowserWindow.fromWebContents(e.sender)
    if (!win) return []
    return getCustomDictionaryWords(win)
  })
}

export default registerSpellcheckerHandlers