Merge pull request #3266 from d-ber/character_import_speedup Character import speedup by delaying tag import

d78577c46da72b100cd758a04a53a1d331eecd99

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed

Parents: 1807af355b3e4b4aedc1bd37edad95fa845c3221 2ab59f5a7a157a98ab7bcc83b319b6e740f84f3e

public/script.js+58 -17 Viewed

     // Set a timeout so multiple flashes don't overlap
     clearTimeout(importFlashTimeout);
     importFlashTimeout = setTimeout(function () {
         if (type === 'char_import' || type === 'char_create' || type === 'char_import_no_toast') {
             // Find the page at which the character is located
             const avatarFileName = charId;
             const charData = getEntitiesList({ doFilter: true });

         'charx',
     ];
 +    const avatarFileNames = [];
     for (const file of files) {
         const extension = file.name.split('.').pop().toLowerCase();
         if (allowedMimeTypes.some(x => file.type.startsWith(x)) || allowedExtensions.includes(extension)) {
             const preservedName = data instanceof Map && data.get(file);
             const avatarFileName = await importCharacter(file, { preserveFileName: preservedName });
 +            if (avatarFileName !== undefined) {
 +                avatarFileNames.push(avatarFileName);
 +            }
         } else {
             toastr.warning(t`Unsupported file type: ` + file.name);
+        }
+    }
++
 +    if (avatarFileNames.length > 0) {
 +        await importCharactersTags(avatarFileNames);
 +        selectImportedChar(avatarFileNames[avatarFileNames.length - 1]);
 +    }
 +}
++
 +/**
 + * Imports tags for the given characters
 + * @param {string[]} avatarFileNames character avatar filenames whose tags are to import
 + */
 +async function importCharactersTags(avatarFileNames) {
 +    await getCharacters();
 +    for (let i = 0; i < avatarFileNames.length; i++) {
 +        if (power_user.tag_import_setting !== tag_import_setting.NONE) {
 +            const importedCharacter = characters.find(character => character.avatar === avatarFileNames[i]);
 +            await importTags(importedCharacter);
 +        }
 +    }
 +}
++
 +/**
 + * Selects the given imported char
 + * @param {string} charId char to select
 + */
 +function selectImportedChar(charId) {
 +    let oldSelectedChar = null;
 +    if (this_chid !== undefined) {
 +        oldSelectedChar = characters[this_chid].avatar;
 +    }
 +    select_rm_info('char_import_no_toast', charId, oldSelectedChar);
+}
 /**
  * Imports a character from a file.
  * @param {File} file File to import
  * @param {string?object} preserveFileName Whether to preserve original[options] file- nameOptions
 - * @returns {Promise<void>}
 + * @param {string} [options.preserveFileName] Whether to preserve original file name
 + * @param {Boolean} [options.importTags=false] Whether to import tags
 + * @returns {Promise<string>}
  */
 async function importCharacter(file, { preserveFileName = '', importTags = false } = {}) {
     if (is_group_generating || is_send_press) {
         toastr.error(t`Cannot import characters while generating. Stop the request and try again.`, t`Import aborted`);
         throw new Error('Cannot import character while generating');

     if (data.file_name !== undefined) {
         $('#character_search_bar').val('').trigger('input');
 -        let oldSelectedChar = null;
 +        toastr.success(t`Character Created: ${String(data.file_name).replace('.png', '')}`);
 -        if (this_chid !== undefined) {
 +        let avatarFileName = `${data.file_name}.png`;
 -            oldSelectedChar = characters[this_chid].avatar;
 +        if (importTags) {
 -        }
 +            await importCharactersTags([avatarFileName]);
 -        await getCharacters();
 +            selectImportedChar(data.file_name);
 -        select_rm_info('char_import', data.file_name, oldSelectedChar);
 -        if (power_user.tag_import_setting !== tag_import_setting.NONE) {
 -            let currentContext = getContext();
 -            let avatarFileName = `${data.file_name}.png`;
 -            let importedCharacter = currentContext.characters.find(character => character.avatar === avatarFileName);
 -            await importTags(importedCharacter);
+        }
 +        return avatarFileName;
+    }
+}

             return;
+        }
 +        const avatarFileNames = [];
         for (const file of e.target.files) {
             const avatarFileName = await importCharacter(file);
 +            if (avatarFileName !== undefined) {
 +                avatarFileNames.push(avatarFileName);
 +            }
 +        }
++
 +        if (avatarFileNames.length > 0) {
 +            await importCharactersTags(avatarFileNames);
 +            selectImportedChar(avatarFileNames[avatarFileNames.length - 1]);
+        }
     });