aliasvault/browser-extensions/chrome/vite-plugins/dictionary-loader.ts

import fs from 'fs';
import path from 'path';

/**
 * Vite plugin to load dictionary files for the identity generator and
 * inject the content into the code during the build process.
 *
 * This allows to keep separate dictionary files for improved maintainability
 * but include the contents in the resulting extension code without needing to bundle
 * the files separately.
 */
export default function dictionaryLoader() {
  return {
    name: 'dictionary-loader',
    transform(code: string, id: string) {
      // Only transform identity generator files
      // TODO: refactor to dynamically load all dictionaries for all languages
      // instead of hardcoding the languages here.
      if (id.includes('IdentityGeneratorEn.ts') || id.includes('IdentityGeneratorNl.ts')) {
        const lang = id.includes('IdentityGeneratorEn.ts') ? 'en' : 'nl';
        // Load dictionaries from the repository root 'dictionaries' folder. These dictionaries
        // are used by both .NET and JavaScript code.
        const dictionaryPath = path.resolve(__dirname, `../../../dictionaries/${lang}`);

        try {
          // Read dictionary files and clean up entries
          const firstNamesMale = fs.readFileSync(path.join(dictionaryPath, 'firstnames_male'), 'utf-8')
            .split('\n')
            .filter(name => name.trim())
            .map(name => name.trim());

          const firstNamesFemale = fs.readFileSync(path.join(dictionaryPath, 'firstnames_female'), 'utf-8')
            .split('\n')
            .filter(name => name.trim())
            .map(name => name.trim());

          const lastNames = fs.readFileSync(path.join(dictionaryPath, 'lastnames'), 'utf-8')
            .split('\n')
            .filter(name => name.trim())
            .map(name => name.trim());

          // Replace the placeholder strings with stringified arrays
          code = code.replace(
            new RegExp(`['"\`]__FIRSTNAMES_MALE_${lang.toUpperCase()}__['"\`]`, 'g'),
            `[${firstNamesMale.map(name => `"${name}"`).join(',')}]`
          );

          code = code.replace(
            new RegExp(`['"\`]__FIRSTNAMES_FEMALE_${lang.toUpperCase()}__['"\`]`, 'g'),
            `[${firstNamesFemale.map(name => `"${name}"`).join(',')}]`
          );

          code = code.replace(
            new RegExp(`['"\`]__LASTNAMES_${lang.toUpperCase()}__['"\`]`, 'g'),
            `[${lastNames.map(name => `"${name}"`).join(',')}]`
          );

          return {
            code,
            map: null
          };
        } catch (error) {
          console.error(`Error loading dictionary files for ${lang}:`, error);
          return null;
        }
      }
    }
  };
}