a1111-sd-webui-tagcomplete/javascript/__globals.js

// Create our TAC namespace
var TAC = TAC || {};

/**
 * @typedef {Object} TAC.CFG
 * @property {string} tagFile - Tag filename
 * @property {{ global: boolean, txt2img: boolean, img2img: boolean, negativePrompts: boolean, thirdParty: boolean, modelList: string, modelListMode: "Blacklist"|"Whitelist" }} activeIn - Settings for which parts of the UI the tag completion is active in.
 * @property {boolean} slidingPopup - Move completion popup together with text cursor
 * @property {number} maxResults - Maximum results
 * @property {boolean} showAllResults - Show all results
 * @property {number} resultStepLength - How many results to load at once
 * @property {number} delayTime - Time in ms to wait before triggering completion again
 * @property {boolean} useWildcards - Search for wildcards
 * @property {boolean} sortWildcardResults - Sort wildcard file contents alphabetically
 * @property {boolean} useEmbeddings - Search for embeddings
 * @property {boolean} includeEmbeddingsInNormalResults - Include embeddings in normal tag results
 * @property {boolean} useHypernetworks - Search for hypernetworks
 * @property {boolean} useLoras - Search for Loras
 * @property {boolean} useLycos - Search for LyCORIS/LoHa
 * @property {boolean} useLoraPrefixForLycos - Use the '<lora:' prefix instead of '<lyco:' for models in the LyCORIS folder
 * @property {boolean} showWikiLinks - Show '?' next to tags, linking to its Danbooru or e621 wiki page
 * @property {boolean} showExtraNetworkPreviews - Show preview thumbnails for extra networks if available
 * @property {string} modelSortOrder - Model sort order
 * @property {boolean} frequencySort - Locally record tag usage and sort frequent tags higher
 * @property {string} frequencyFunction - Function to use for frequency sorting
 * @property {number} frequencyMinCount - Minimum number of uses for a tag to be considered frequent
 * @property {number} frequencyMaxAge - Maximum days since last use for a tag to be considered frequent
 * @property {number} frequencyRecommendCap - Maximum number of recommended tags
 * @property {boolean} frequencyIncludeAlias - Frequency sorting matches aliases for frequent tags
 * @property {boolean} useStyleVars - Search for webui style names
 * @property {boolean} replaceUnderscores - Replace underscores with spaces on insertion
 * @property {string} replaceUnderscoresExclusionList - Underscore replacement exclusion list
 * @property {boolean} escapeParentheses - Escape parentheses on insertion
 * @property {boolean} appendComma - Append comma on tag autocompletion
 * @property {boolean} appendSpace - Append space on tag autocompletion
 * @property {boolean} alwaysSpaceAtEnd - Always append space if inserting at the end of the textbox
 * @property {string} wildcardCompletionMode - How to complete nested wildcard paths
 * @property {string} modelKeywordCompletion - Try to add known trigger words for LORA/LyCO models
 * @property {string} modelKeywordLocation - Where to insert the trigger keyword
 * @property {string} wcWrap - Wrapper characters for wildcard tags.
 * @property {{ searchByAlias: boolean, onlyShowAlias: boolean }} alias - Alias-related settings.
 * @property {{ translationFile: string, oldFormat: boolean, searchByTranslation: boolean, liveTranslation: boolean }} translation - Translation-related settings.
 * @property {{ extraFile: string, addMode: "Insert before"|"Insert after" }} extra - Extra file-related settings.
 * @property {string} chantFile - Chant filename
 * @property {number} extraNetworksDefaultMultiplier - Default multiplier for extra networks.
 * @property {string} extraNetworksSeparator - Separator used for extra networks.
 * @property {{ MoveUp: string, MoveDown: string, JumpUp: string, JumpDown: string, JumpToStart: string, JumpToEnd: string, ChooseSelected: string, ChooseFirstOrSelected: string, Close: string }} keymap - Custom key mappings for tag completion.
 * @property {{ [filename: string]: { [category: string]: string[] } }} colorMap - Color mapping for tag categories.
 */
/** @type {TAC.CFG}  */
TAC.CFG = {
    // Main tag file
    tagFile,
    // Active in settings
    activeIn: {
        global,
        txt2img,
        img2img,
        negativePrompts,
        thirdParty,
        modelList,
        modelListMode,
    },
    // Results related settings
    slidingPopup,
    maxResults,
    showAllResults,
    resultStepLength,
    delayTime,
    useWildcards,
    sortWildcardResults,
    useEmbeddings,
    includeEmbeddingsInNormalResults,
    useHypernetworks,
    useLoras,
    useLycos,
    useLoraPrefixForLycos,
    showWikiLinks,
    showExtraNetworkPreviews,
    modelSortOrder,
    frequencySort,
    frequencyFunction,
    frequencyMinCount,
    frequencyMaxAge,
    frequencyRecommendCap,
    frequencyIncludeAlias,
    useStyleVars,
    // Insertion related settings
    replaceUnderscores,
    replaceUnderscoresExclusionList,
    escapeParentheses,
    appendComma,
    appendSpace,
    alwaysSpaceAtEnd,
    wildcardCompletionMode,
    modelKeywordCompletion,
    modelKeywordLocation,
    wcWrap, // to support custom wrapper chars set by dp_parser
    // Alias settings
    alias: {
        searchByAlias,
        onlyShowAlias,
    },
    // Translation settings
    translation: {
        translationFile,
        oldFormat,
        searchByTranslation,
        liveTranslation,
    },
    // Extra file settings
    extra: {
        extraFile,
        addMode,
    },
    // Chant file settings
    chantFile,
    // Settings not from tac but still used by the script
    extraNetworksDefaultMultiplier,
    extraNetworksSeparator,
    // Custom mapping settings
    keymap: {
        MoveUp,
        MoveDown,
        JumpUp,
        JumpDown,
        JumpToStart,
        JumpToEnd,
        ChooseSelected,
        ChooseFirstOrSelected,
        Close,
    },
    colorMap: {
        [filename]: { [category]: ["light", "dark"] },
    },
};

TAC.Globals = new (function () {
    // Core components
    this.tagBasePath = "";
    this.modelKeywordPath = "";
    this.selfTrigger = false;

    // Tag completion data loaded from files
    this.allTags = [];
    this.translations = new Map();
    this.extras = [];
    // Same for tag-likes
    this.wildcardFiles = [];
    this.wildcardExtFiles = [];
    this.yamlWildcards = [];
    this.umiWildcards = [];
    this.embeddings = [];
    this.hypernetworks = [];
    this.loras = [];
    this.lycos = [];
    this.modelKeywordDict = new Map();
    this.chants = [];
    this.styleNames = [];

    // Selected model info for black/whitelisting
    this.currentModelHash = "";
    this.currentModelName = "";

    // Current results
    this.results = [];
    this.resultCount = 0;

    // Relevant for parsing
    this.previousTags = [];
    this.tagword = "";
    this.originalTagword = "";
    this.hideBlocked = false;

    // Tag selection for keyboard navigation
    this.selectedTag = null;
    this.oldSelectedTag = null;
    this.resultCountBeforeNormalTags = 0;

    // Lora keyword undo/redo history
    this.textBeforeKeywordInsertion = "";
    this.textAfterKeywordInsertion = "";
    this.lastEditWasKeywordInsertion = false;
    this.keywordInsertionUndone = false;

    // UMI
    this.umiPreviousTags = [];
})();

/// Extendability system:
/// Provides "queues" for other files of the script (or really any js)
/// to add functions to be called at certain points in the script.
/// Similar to a callback system, but primitive.
TAC.Ext = new (function () {
    // Queues
    this.QUEUE_AFTER_INSERT = [];
    this.QUEUE_AFTER_SETUP = [];
    this.QUEUE_FILE_LOAD = [];
    this.QUEUE_AFTER_CONFIG_CHANGE = [];
    this.QUEUE_SANITIZE = [];

    // List of parsers to try
    this.PARSERS = [];
})();