`;
};
FUNC.RenderTagQueryColumn = function (related_tags, tags_overlap) {
let is_empty = related_tags.tags.length === 0;
let display_name = related_tags.query.replace(/_/g, ' ');
let column_html = (!is_empty ? FUNC.RenderTaglist(related_tags.tags, display_name, tags_overlap) : "");
return FUNC.RenderTagColumn('irt-general-related-tags-column', column_html, is_empty);
};
FUNC.RenderChecklistColumn = function (checklist_tags, tag_name) {
let is_empty = checklist_tags.length === 0;
let display_name = "Checklist: " + tag_name.replace(/_/g, ' ');
let column_html = (!is_empty ? FUNC.RenderTaglist(checklist_tags, display_name) : "");
return FUNC.RenderTagColumn('irt-checklist-related-tags-column', column_html, is_empty);
};
FUNC.RenderWikiTagQueryColumns = function (wiki_page, other_wikis) {
let is_empty = wiki_page.tags.length === 0;
let display_name = wiki_page.title.replace(/_/g, ' ');
let column_html = (!is_empty ? FUNC.RenderTaglist(FUNC.FilterTagEntries(wiki_page.tags), JSPLib.danbooru.wikiLink(wiki_page.title, `wiki:${display_name}`, 'target="_blank"')) : "");
let html = FUNC.RenderTagColumn('irt-wiki-related-tags-column', column_html, is_empty);
other_wikis.forEach((other_wiki) => {
if (other_wiki.tags.length === 0) return;
let title_name = other_wiki.title.replace(/_/g, ' ');
column_html = FUNC.RenderTaglist(FUNC.FilterTagEntries(other_wiki.tags), JSPLib.danbooru.wikiLink(other_wiki.title, `wiki:${title_name}`, 'target="_blank"'));
html += FUNC.RenderTagColumn('irt-wiki-related-tags-column', column_html, false);
});
return html;
};
FUNC.RenderUserQueryColumns = function (recent_tags, frequent_tags, ai_tags) {
let is_empty = recent_tags.length === 0;
let column_html = (is_empty ? "" : FUNC.RenderTaglist(recent_tags, 'Recent'));
let html = FUNC.RenderTagColumn('irt-recent-related-tags-column', column_html, is_empty);
is_empty = frequent_tags.length === 0;
column_html = (!is_empty ? FUNC.RenderTaglist(frequent_tags, 'Frequent') : "");
html += FUNC.RenderTagColumn('irt-frequent-related-tags-column', column_html, is_empty);
is_empty = ai_tags.length === 0;
column_html = (!is_empty ? FUNC.RenderTaglist(ai_tags, 'Suggested') : "");
html += FUNC.RenderTagColumn('irt-ai-tags-related-tags-column', column_html, is_empty);
return html;
};
FUNC.RenderTranslatedColumn = function (translated_tags) {
let is_empty = translated_tags.length === 0;
let column_html = (!is_empty ? FUNC.RenderTaglist(translated_tags, 'Translated') : "");
return FUNC.RenderTagColumn('irt-translated-tags-related-tags-column', column_html, is_empty);
};
FUNC.RenderRelatedQueryCategoryControls = function () {
let html = '';
for (let category in RELATED_QUERY_CATEGORIES) {
if (!IRT.related_query_categories.includes(category)) continue;
let display_name = JSPLib.utility.displayCase(category);
html += `
`;
}
return `
${html}
`;
};
FUNC.RenderRelatedQueryTypeControls = function () {
let html = "";
RELATED_QUERY_ORDERS.forEach((type) => {
let checked = (IRT.related_query_order_default[0] === type ? 'checked' : "");
let display_name = JSPLib.utility.displayCase(type);
html += `
`;
});
return `
${html}
`;
};
//Network functions
FUNC.RandomPosts = async function (tag, batches, per_batch) {
let posts = [];
let url_addons = {
tags: tag + ' order:md5', // Gives us quasi-random results
only: 'id,md5,tag_string',
limit: per_batch,
};
for (let i = 1; i <= batches; i++) {
url_addons.page = i;
let result = await JSPLib.danbooru.submitRequest('posts', url_addons);
posts = JSPLib.utility.concat(posts, result);
if (result.length < per_batch) break;
}
return posts;
};
FUNC.TagsOverlapQuery = async function(self, tag) {
const [batches, per_batch] = [IRT.random_post_batches, IRT.random_posts_per_batch];
self.debuglog("Querying:", tag, batches, per_batch);
let [posts, count] = await Promise.all([
FUNC.RandomPosts(tag, batches, per_batch),
JSPLib.danbooru.submitRequest('counts/posts', {tags: tag}, {default_val: {counts: {posts: 0}}})
]);
let overlap = {};
for (let i = 0; i < posts.length; i++) {
let tag_names = posts[i].tag_string.split(' ');
tag_names.forEach((tag) => {
overlap[tag] = (overlap[tag] ?? 0) + 1;
});
}
let cutoff = Math.min(posts.length, batches * per_batch) / 20; // (5% or greater overlap)
for (let k in overlap) {
if (overlap[k] < cutoff) {
delete overlap[k];
}
}
return {value: {overlap, count: Math.min(count.counts.posts, batches * per_batch)}, expires: JSPLib.utility.getExpires(TAGS_OVERLAP_EXPIRES)};
};
FUNC.WikiPageTagsQuery = async function (self, title) {
self.debuglog("Querying:", title, (IRT.other_wikis_enabled ? "with" : "without", "other wikis"));
let url_addons = {
search: {title},
only: 'body,tag,dtext_links[link_target,link_type,linked_tag[name,category,is_deprecated]]'
};
let wikis_with_links = await JSPLib.danbooru.submitRequest('wiki_pages', url_addons);
let tags = (wikis_with_links.length ? FUNC.GetTagsEntryArray(wikis_with_links[0]) : []);
if (IRT.query_unknown_tags_enabled) {
let tag_names = tags.filter((tag) => tag[1] === NONEXISTENT_TAG_CATEGORY).map((tag) => tag[0]);
if (tag_names.length) {
let tag_data = await JSPLib.danbooru.submitRequest('tags', FUNC.GetTagQueryParams(tag_names));
let tag_array = FUNC.CreateTagArray(tag_names, tag_data);
tags = tags.map((tag_entry) => {
let unknown_entry = tag_array.find((item) => item[0] === tag_entry[0]);
if (unknown_entry) {
return unknown_entry;
} else {
return tag_entry;
}
});
}
}
let other_wikis = (!title.startsWith('list_of_') ?
(wikis_with_links?.[0]?.dtext_links || [])
.filter((link) => (link.link_type === 'wiki_link' && link.link_target.startsWith('list_of_')))
.map((link) => link.link_target) :
[]);
return {value: {title, tags, other_wikis}, expires: JSPLib.utility.getExpires(WIKI_PAGE_TAGS_EXPIRES)};
};
FUNC.RelatedTagsQuery = async function(self, tag, category, query_order) {
self.debuglog("Querying:", tag, category);
let url_addons = {search: {query: tag, order: query_order}, limit: IRT.related_results_limit || Danbooru.RelatedTag.MAX_RELATED_TAGS};
if (category in RELATED_QUERY_CATEGORIES) {
url_addons.search.category = RELATED_QUERY_CATEGORIES[category];
}
let html = await JSPLib.network.get('/related_tag.html', {data: url_addons});
let tagentry_array = $(html).find('tbody .name-column a[href^="/posts"]').toArray().map((link) => {
let name = link.innerText;
let category = Number(
link.className
.match(/tag-type-(\d)/)?.[1]
?? NONEXISTENT_TAG_CATEGORY
);
return [name, category];
});
let data = {
query: tag,
categories: (category ? [RELATED_QUERY_CATEGORIES[category]] : []),
tags: tagentry_array,
};
return {value: data, expires: JSPLib.utility.getExpires(RELATED_TAG_EXPIRES)};
};
//Network/storage wrappers
FUNC.GetCachedData = async function (self, {name = "", args = [], keyfunc = (() => {}), netfunc = (() => {}), expires = null} = {}) {
let key = keyfunc(...args);
self.debuglog("Checking", name, ':', key);
let cached = await (!IRT.network_only_mode ?
JSPLib.storage.checkLocalDB(key, FUNC.ValidateEntry, expires) :
Promise.resolve(null));
if (!cached) {
cached = await netfunc(...args);
JSPLib.storage.saveData(key, cached);
} else if (IRT.recheck_data_interval > 0) {
let recheck_time = cached.expires - (IRT.recheck_data_interval * JSPLib.utility.one_day);
if (!JSPLib.utility.validateExpires(recheck_time)) {
self.debuglog("Rechecking", name, key);
netfunc(...args).then((data) => {
JSPLib.storage.saveData(key, data);
});
}
}
self.debuglog("Found", name, ':', key, cached.value);
return cached.value;
};
FUNC.GetRelatedTags = function (tag, category, query_order) {
return FUNC.GetCachedData({
name: 'related tags',
args: [tag, category, query_order],
keyfunc: (tag, category, query_order) => (FUNC.GetRelatedKeyModifer(category, query_order) + '-' + tag),
netfunc: FUNC.RelatedTagsQuery,
expires: RELATED_TAG_EXPIRES,
});
};
FUNC.GetTagsOverlap = function (tag) {
return FUNC.GetCachedData({
name: 'tags overlap',
args: [tag],
keyfunc: (tag) => ('tagov-' + tag),
netfunc: FUNC.TagsOverlapQuery,
expires: TAGS_OVERLAP_EXPIRES,
});
};
FUNC.GetWikiPageTags = function(tag) {
return FUNC.GetCachedData({
name: 'wiki page tags',
args: [tag],
keyfunc: (tag) => ('wpt-' + tag),
netfunc: FUNC.WikiPageTagsQuery,
expires: WIKI_PAGE_TAGS_EXPIRES,
});
};
FUNC.GetAllWikiPageTags = async function (tag) {
let wiki_page = await FUNC.GetWikiPageTags(tag);
var other_wikis;
if (IRT.other_wikis_enabled) {
let promise_array = [];
wiki_page.other_wikis.forEach((title) => {
promise_array.push(FUNC.GetWikiPageTags(title));
});
other_wikis = await Promise.all(promise_array);
} else {
other_wikis = [];
}
return {wiki_page, other_wikis};
};
//Event handlers
FUNC.RelatedTagsButton = async function (event) {
event.preventDefault();
let currenttag = Danbooru.RelatedTag.current_tag().trim().toLowerCase();
let category = $(event.target).data('selector');
let query_order = (IRT.related_query_order_enabled ? JSPLib.menu.getCheckboxRadioSelected('.irt-program-checkbox') : IRT.related_query_order_default);
let promise_array = [FUNC.GetRelatedTags(currenttag, category, query_order[0])];
if (IRT.related_statistics_enabled) {
promise_array.push(FUNC.GetTagsOverlap(currenttag));
} else {
promise_array.push(Promise.resolve(null));
}
if (IRT.wiki_page_tags_enabled) {
promise_array.push(FUNC.GetAllWikiPageTags(currenttag));
} else {
promise_array.push(Promise.resolve(null));
}
let [related_tags, tags_overlap, wiki_result] = await Promise.all(promise_array);
if (!related_tags) {
return;
}
var tag_array;
if (IRT.checklist_tags_enabled) {
tag_array = FUNC.GetChecklistTagsArray(currenttag) ?? [];
}
$('#irt-related-tags-query-container').html(
FUNC.RenderTagQueryColumn(related_tags, tags_overlap) +
(IRT.checklist_tags_enabled ? FUNC.RenderChecklistColumn(tag_array, currenttag) : "") +
(IRT.wiki_page_tags_enabled ? FUNC.RenderWikiTagQueryColumns(wiki_result.wiki_page, wiki_result.other_wikis) : "")
);
FUNC.UpdateSelected();
FUNC.QueueRelatedTagColumnWidths();
};
FUNC.WikiPageButton = async function (event) {
event.preventDefault();
let currenttag = Danbooru.RelatedTag.current_tag().trim().toLowerCase();
let wiki_result = await FUNC.GetAllWikiPageTags(currenttag);
$('#irt-related-tags-query-container').html(FUNC.RenderWikiTagQueryColumns(wiki_result.wiki_page, wiki_result.other_wikis));
FUNC.UpdateSelected();
FUNC.QueueRelatedTagColumnWidths();
};
FUNC.ChecklistButton = async function (event) {
event.preventDefault();
let currenttag = Danbooru.RelatedTag.current_tag().trim().toLowerCase();
let tag_array = FUNC.GetChecklistTagsArray(currenttag);
if (tag_array === null) {
JSPLib.notice.error("Corrupted data: See debug console for details.");
} else {
$('#irt-related-tags-query-container').html(FUNC.RenderChecklistColumn(tag_array, currenttag));
FUNC.UpdateSelected();
FUNC.QueueRelatedTagColumnWidths();
}
};
FUNC.RelatedTagsEnter = function () {
$(document).on(PROGRAM_KEYDOWN + '.scroll', null, 'left right', FUNC.RelatedTagsScroll);
};
FUNC.RelatedTagsLeave = function () {
$(document).off(PROGRAM_KEYDOWN + '.scroll');
};
FUNC.RelatedTagsScroll = function (event) {
let $related_tags = $('.irt-related-tags');
let current_left = $related_tags.prop('scrollLeft');
if (event.originalEvent.key === 'ArrowLeft') {
current_left -= 40;
} else if (event.originalEvent.key === 'ArrowRight') {
current_left += 40;
}
$related_tags.prop('scrollLeft', current_left);
};
FUNC.ViewChecklistTag = function () {
let import_export = $('#irt-enable-import-export').prop('checked');
if (import_export) {
let tag_list = Object.keys(localStorage)
.filter((name) => name.startsWith('irt-checklist-'))
.map((name) => name.replace('irt-checklist-', ""));
let tag_data = tag_list.map((tag_name) => {
let tag_array = FUNC.GetChecklistTagsArray(tag_name);
if (Array.isArray(tag_array)) {
return {[tag_name]: tag_array.map((item) => item[0])};
}
return null;
})
.filter((data) => data != null);
$('#irt-checklist-frequent-tags textarea').val(JSON.stringify(Object.assign({}, ...tag_data), null, 4));
} else {
let tag_name = $('#irt-control-tag-name').val().split(/\s+/)[0];
if (!tag_name) return;
let tag_array = FUNC.GetChecklistTagsArray(tag_name);
if (tag_array === null) {
JSPLib.notice.error("Corrupted data: See debug console for details.");
} else {
let tag_list = tag_array.map((entry) => entry[0]);
$('#irt-checklist-frequent-tags textarea').val(tag_list.join('\n'));
}
}
};
FUNC.SaveChecklistTag = async function () {
let import_export = $('#irt-enable-import-export').prop('checked');
if (import_export) {
let text_input = $('#irt-checklist-frequent-tags textarea').val();
var data_input;
try {
data_input = JSON.parse(text_input);
} catch (e) {
data_input = null;
JSPLib.debug.debugerror("Error parsing data:", e);
}
if (JSPLib.validate.isHash(data_input)) {
let checklist_data = {};
let check_tags = [];
for (let key in data_input) {
let checklist = data_input[key];
if (!Array.isArray(checklist)) continue;
checklist = checklist.filter((item) => typeof item === "string");
if (checklist.length === 0) continue;
checklist_data[key] = checklist;
check_tags = JSPLib.utility.arrayUnion(check_tags, checklist);
}
if (check_tags.length > 0) {
JSPLib.notice.notice("Querying tags...");
let tag_data = [];
for (let i = 0; i < check_tags.length; i += 1000) {
let query_tags = check_tags.slice(i, i + 1000);
let tags = await JSPLib.danbooru.submitRequest('tags', FUNC.GetTagQueryParams(query_tags), {long_format: true});
tag_data = JSPLib.utility.concat(tag_data, tags);
}
for (let tag_name in checklist_data) {
let checklist = checklist_data[tag_name];
let tag_array = FUNC.CreateTagArray(checklist, tag_data);
JSPLib.storage.setStorageData('irt-checklist-' + tag_name, tag_array, localStorage);
}
JSPLib.notice.notice("Checklists imported.");
} else {
JSPLib.notice.error("No valid checklists found.");
}
} else {
JSPLib.notice.error("Error importing checklist.");
}
} else {
let tag_name = $('#irt-control-tag-name').val().split(/\s+/)[0];
if (!tag_name) return;
let checklist = $('#irt-checklist-frequent-tags textarea').val().split(/\s/).filter((name)=> name !== "");
if (checklist.length > 0) {
let tag_data = await JSPLib.danbooru.submitRequest('tags', FUNC.GetTagQueryParams(checklist));
let tag_array = FUNC.CreateTagArray(checklist, tag_data);
JSPLib.storage.setStorageData('irt-checklist-' + tag_name, tag_array, localStorage);
} else {
JSPLib.storage.removeStorageData('irt-checklist-' + tag_name, localStorage);
}
JSPLib.notice.notice("Checklist updated.");
}
};
FUNC.PopulateChecklistTag = function () {
let tag_name = $('#irt-control-tag-name').val().split(/\s+/)[0];
if (!tag_name) return;
JSPLib.notice.notice("Querying Danbooru...");
FUNC.WikiPageTagsQuery(tag_name).then((data) => {
let tag_list = data.value.tags.map((entry) => entry[0]);
$('#irt-checklist-frequent-tags textarea').val(tag_list.join('\n'));
});
};
FUNC.ListChecklistTags = function () {
let tag_list = Object.keys(localStorage)
.filter((name) => name.startsWith('irt-checklist-'))
.map((name) => name.replace('irt-checklist-', ""))
.sort();
$('#irt-checklist-frequent-tags textarea').val(tag_list.join('\n'));
};
//Initialization functions
FUNC.InitializeUserMediaTags = function (self) {
let recent_tags = $('.recent-related-tags-column [data-tag-name]').map((i, entry) => [[entry.dataset.tagName, Number(entry.className.match(/tag-type-(\d)/)?.[1])]]).toArray();
let frequent_tags = $('.frequent-related-tags-column [data-tag-name]').map((i, entry) => [[entry.dataset.tagName, Number(entry.className.match(/tag-type-(\d)/)?.[1])]]).toArray();
let ai_tags = $('.ai-tags-related-tags-column [data-tag-name]').map((i, entry) => [[entry.dataset.tagName, Number(entry.className.match(/tag-type-(\d)/)?.[1])]]).toArray();
self.debuglog("Media tags:", {recent_tags, frequent_tags, ai_tags});
$('#irt-frequent-recent-container').html(FUNC.RenderUserQueryColumns(recent_tags, frequent_tags, ai_tags));
FUNC.UpdateSelected();
FUNC.QueueRelatedTagColumnWidths();
};
FUNC.InitializeTranslatedTags = function (self) {
let translated_tags = $('.translated-tags-related-tags-column [data-tag-name]').map((i, entry) => [[entry.dataset.tagName, Number(entry.className.match(/tag-type-(\d)/)?.[1])]]).toArray();
self.debuglog("Translated tags:", translated_tags);
$('#irt-translated-tags-container').html(FUNC.RenderTranslatedColumn(translated_tags));
FUNC.UpdateSelected();
FUNC.QueueRelatedTagColumnWidths();
};
FUNC.UpdateSelected = function () {
const current_tags = Danbooru.RelatedTag.current_tags();
$('.irt-related-tags li').each((_, li) => {
const tag_name = $(li).find('a').attr('data-tag-name');
if (current_tags.includes(tag_name)) {
$(li).addClass('selected');
$(li).find('input').prop('checked', true);
} else {
$(li).removeClass('selected');
$(li).find('input').prop('checked', false);
}
});
};
FUNC.ToggleTag = function (event) {
const $field = $('#post_tag_string');
const tag = $(event.target).closest('li').find('a').attr('data-tag-name');
if (Danbooru.RelatedTag.current_tags().includes(tag)) {
const escaped_tag = Danbooru.Utility.regexp_escape(tag);
$field.val($field.val().replace(new RegExp('(^|\\s)' + escaped_tag + '($|\\s)', 'gi'), '$1$2'));
} else {
$field.val($field.val() + ' ' + tag);
}
$field.val($field.val().trim().replace(/ +/g, ' ') + ' ');
FUNC.UpdateSelected();
// The timeout is needed on Chrome since it will clobber the field attribute otherwise
setTimeout(() => {
$field.prop('selectionStart', $field.val().length);
}, 100);
event.preventDefault();
// Artificially trigger input event so the tag counter updates.
$field.trigger('input');
};
//Initialize functions
FUNC.InitializeRelatedTagsSection = function () {
Danbooru.Post.EDIT_DIALOG_MIN_HEIGHT = 800;
$(document).on(PROGRAM_CLICK, '.irt-related-tags a.search-tag', FUNC.ToggleTag);
$(document).on(PROGRAM_CLICK, '.irt-related-button', FUNC.RelatedTagsButton);
$(document).on(PROGRAM_CLICK, '.irt-wiki-button', FUNC.WikiPageButton);
$(document).on(PROGRAM_CLICK, '.irt-checklist-button', FUNC.ChecklistButton);
$(document).on('input.irt', '#post_tag_string', FUNC.UpdateSelected);
FUNC.InitialiazeRelatedQueryControls();
$('.related-tags').before(IRT_RELATED_TAGS_SECTION);
$('#related-tags-container').hide();
FUNC.InitializeTagColumns();
if (IRT.expandable_related_section_enabled) {
FUNC.InitialiazeRelatedExpandableSection();
}
};
FUNC.InitializeTagColumns = function (self) {
if (IRT.controller === 'posts') {
let media_asset_id = $("#related-tags-container").attr("data-media-asset-id");
JSPLib.network.get("/related_tag.js", {data: {user_tags: true, media_asset_id}});
}
if (!$('#related-tags-container .ai-tags-related-tags-column').html()?.trim()) {
self.debuglog("User/Media tags not loaded yet... setting up mutation observer.");
JSPLib.concurrency.setupMutationReplaceObserver('#related-tags-container', '.ai-tags-related-tags-column', () => {
FUNC.InitializeUserMediaTags();
});
} else {
FUNC.InitializeUserMediaTags();
}
if (!$('#related-tags-container .translated-tags-related-tags-column').html()?.trim()) {
self.debuglog("Translated tags not loaded yet... setting up mutation observer.");
JSPLib.concurrency.setupMutationReplaceObserver('#related-tags-container', '.translated-tags-related-tags-column', () => {
FUNC.InitializeTranslatedTags();
});
} else {
FUNC.InitializeTranslatedTags();
}
};
FUNC.InitialiazeRelatedQueryControls = function () {
$('#post_tag_string, #upload_tag_string').parent().after('');
$('#irt-related-tag-query-controls').append(FUNC.RenderRelatedQueryCategoryControls());
if (IRT.wiki_page_query_only_enabled) {
$('#irt-related-tag-query-controls').append(WIKI_PAGE_BUTTON);
}
if (IRT.checklist_query_only_enabled) {
$('#irt-related-tag-query-controls').append(CHECKLIST_BUTTON);
}
if (IRT.related_query_order_enabled) {
$('#irt-related-tag-query-controls').append(FUNC.RenderRelatedQueryTypeControls());
$('#irt-related-query-type .irt-program-checkbox').checkboxradio();
$('#irt-related-query-type .ui-state-hover').removeClass('ui-state-hover');
}
$('#post_tag_string').css('max-width', '80rem');
$('#post_tag_string').closest('.fixed-width-container').css('max-width', '80rem');
JSPLib.utility.setCSSStyle(RELATED_QUERY_CONTROL_CSS, 'related_query');
};
FUNC.InitialiazeRelatedExpandableSection = function () {
$('.irt-related-tags').before(IRT_SCROLL_WRAPPER);
$('.irt-related-tags').on(PROGRAM_MOUSEENTER, FUNC.RelatedTagsEnter);
$('.irt-related-tags').on(PROGRAM_MOUSELEAVE, FUNC.RelatedTagsLeave);
$('#irt-edit-scroll-wrapper').on(PROGRAM_SCROLL, () => {
$('.irt-related-tags').scrollLeft($('#irt-edit-scroll-wrapper').scrollLeft());
});
$('.irt-related-tags').on(PROGRAM_SCROLL, () => {
$('#irt-edit-scroll-wrapper').scrollLeft($('.irt-related-tags').scrollLeft());
});
let $container = $('#irt-related-tags-container');
new ResizeObserver(() => {
FUNC.QueueRelatedTagColumnWidths();
}).observe($container[0]);
JSPLib.utility.setCSSStyle(EXPANDABLE_RELATED_SECTION_CSS, 'expandable_related');
};
FUNC.InitializeRelatedTagColumnWidths = function () {
const em_size = 14;
const max_column_em = 20;
const min_column_em = 10;
const range = document.createRange();
const getChildWidth = (i, child) => {
if (child.nodeType === 3) {
range.selectNodeContents(child);
const rects = range.getClientRects();
return (rects.length > 0 ? rects[0].width : 0);
}
return $(child).outerWidth();
};
const getSum = (a, b) => (a + b);
let $related_tags = $('.irt-related-tags');
$('.tag-column', $related_tags[0]).each((i, column) => {
let $column = $(column);
$column.css('width', "");
let $container = $('>ul,>div', column);
let $children = $container.children();
if ($children.length === 0) {
return;
}
let line_tag = $container.children().get(0).tagName.toLowerCase();
let container_tag = $container.get(0).tagName.toLowerCase();
let line_selector = container_tag + '>' + line_tag;
let max_child_width = Math.max(...$(line_selector, column).map((i, entry) => {
let child_widths = $(entry).contents().map(getChildWidth).toArray();
return child_widths.reduce(getSum, 0);
}));
let max_column_width = max_column_em * em_size;
let column_width = Math.max(Math.min(max_child_width, max_column_width), min_column_em * em_size);
$column.width((Math.ceil(column_width / em_size) + 2) + 'em');
});
if ($related_tags.prop('scrollWidth') > ($related_tags.outerWidth() + (2 * em_size))) {
$('#irt-edit-scroll-wrapper').width($related_tags.outerWidth());
$('#irt-edit-scroll-bar').width($related_tags.prop('scrollWidth') - em_size);
$('#irt-edit-scroll-wrapper').show();
$related_tags.addClass('scrollable');
} else {
$('#irt-edit-scroll-wrapper').hide();
$related_tags.removeClass('scrollable');
}
};
FUNC.QueueRelatedTagColumnWidths = function () {
if (Number.isInteger(FUNC.QueueRelatedTagColumnWidths.timer)) {
clearTimeout(FUNC.QueueRelatedTagColumnWidths.timer);
}
FUNC.QueueRelatedTagColumnWidths.timer = setTimeout(() => {
FUNC.InitializeRelatedTagColumnWidths();
FUNC.QueueRelatedTagColumnWidths.timer = null;
}, 100);
};
//Main execution functions
FUNC.SetupInitializations = function () {
if (IRT.action !== 'show' || !['posts', 'uploads', 'upload-media-assets'].includes(IRT.controller)) return;
IRT.inititialization_started = false;
// Wait for event signal from Danbooru
$(document).on('danbooru:open-post-edit-tab.irt danbooru:open-post-edit-dialog.irt', () => {
IRT.inititialization_started = true;
$(document).off('danbooru:open-post-edit-tab.irt danbooru:open-post-edit-dialog.irt');
});
// Or check if the tags have already been rendered
JSPLib.utility.recheckTimer({
check: () => (IRT.inititialization_started || ($('#related-tags-container .ai-tags-related-tags-column .tag-list').html() || "").trim() !== ""),
exec: () => FUNC.InitializeRelatedTagsSection(),
fail: () => {
$(document)
.off('danbooru:open-post-edit-tab.irt danbooru:open-post-edit-dialog.irt')
.on('danbooru:open-post-edit-tab.irt danbooru:open-post-edit-dialog.irt', () => {
FUNC.InitializeRelatedTagsSection();
$(document).off('danbooru:open-post-edit-tab.irt danbooru:open-post-edit-dialog.irt');
});
},
}, 250, JSPLib.utility.one_second * 10);
$(document).on('danbooru:close-post-edit-dialog.irt', FUNC.QueueRelatedTagColumnWidths);
};
FUNC.CleanupTasks = function () {
JSPLib.storage.pruneEntries(PROGRAM_SHORTCUT, PROGRAM_DATA_REGEX, PRUNE_EXPIRES);
};
//Menu functions
FUNC.OptionCacheDataKey = function (data_type, data_value) {
if (data_type === 'related_tag') {
IRT.tag_category = $('#irt-control-tag-category').val();
IRT.query_order = $('#irt-control-query-order').val();
let modifier = FUNC.GetRelatedKeyModifer(IRT.related_category, IRT.query_order);
return `${modifier}-${data_value}`;
}
if (data_type === 'wiki_page') {
return `wpt-${data_value}`;
}
if (data_type === 'tag_overlap') {
return `tagov-${data_value}`;
}
};
FUNC.DataTypeChange = function () {
let data_type = $('#irt-control-data-type').val();
let action = (data_type === 'related_tag' ? 'show' : 'hide');
$('.irt-options[data-setting=tag_category]')[action]();
$('.irt-options[data-setting=query_order]')[action]();
};
FUNC.InitializeProgramValues = function(self) {
if (!JSPLib.storage.use_indexed_db) {
self.debugwarn("No Indexed DB! Exiting...");
return false;
}
JSPLib.utility.hashMerge(IRT, {
userid: Danbooru.CurrentUser.data('id'),
max_random_posts: IRT.random_post_batches * IRT.random_posts_per_batch,
}, PROGRAM_RESET_KEYS);
return true;
};
FUNC.RenderSettingsMenu = function() {
$('#indexed-related-tags').append(JSPLib.menu.renderMenuFramework(MENU_CONFIG));
$('#irt-general-settings').append(JSPLib.menu.renderDomainSelectors());
$('#irt-related-tag-settings-message').append(JSPLib.menu.renderExpandable("Additional setting details", RELATED_TAG_SETTINGS_DETAILS));
$('#irt-related-tag-settings').append(JSPLib.menu.renderInputSelectors('related_query_categories', 'checkbox'));
$('#irt-related-tag-settings').append(JSPLib.menu.renderCheckbox('related_query_order_enabled'));
$('#irt-related-tag-settings').append(JSPLib.menu.renderInputSelectors('related_query_order_default', 'radio'));
$('#irt-related-tag-settings').append(JSPLib.menu.renderTextinput('related_results_limit', 5));
$('#irt-related-tag-settings').append(JSPLib.menu.renderCheckbox('expandable_related_section_enabled'));
$('#irt-tag-statistic-settings').append(JSPLib.menu.renderCheckbox('related_statistics_enabled'));
$('#irt-tag-statistic-settings').append(JSPLib.menu.renderTextinput('random_post_batches', 5));
$('#irt-tag-statistic-settings').append(JSPLib.menu.renderTextinput('random_posts_per_batch', 5));
$('#irt-checklist-settings').append(JSPLib.menu.renderCheckbox('checklist_tags_enabled'));
$('#irt-checklist-settings').append(JSPLib.menu.renderCheckbox('checklist_query_only_enabled'));
$('#irt-wiki-page-settings').append(JSPLib.menu.renderCheckbox('wiki_page_tags_enabled'));
$('#irt-wiki-page-settings').append(JSPLib.menu.renderCheckbox('other_wikis_enabled'));
$('#irt-wiki-page-settings').append(JSPLib.menu.renderCheckbox('unique_wiki_tags_enabled'));
$('#irt-wiki-page-settings').append(JSPLib.menu.renderCheckbox('wiki_page_query_only_enabled'));
$('#irt-wiki-page-settings').append(JSPLib.menu.renderCheckbox('query_unknown_tags_enabled'));
$('#irt-network-settings-message').append(JSPLib.menu.renderExpandable("Additional setting details", NETWORK_SETTINGS_DETAILS));
$('#irt-network-settings').append(JSPLib.menu.renderTextinput('recheck_data_interval', 5));
$('#irt-network-settings').append(JSPLib.menu.renderCheckbox('network_only_mode'));
$('#irt-checklist-controls').append(JSPLib.menu.renderCheckbox('import_export', true));
$('#irt-checklist-controls').append(JSPLib.menu.renderTextinput('tag_name', 50, true));
$('#irt-control-tag-name').attr('data-autocomplete', 'tag-query');
$('#irt-checklist-controls').append(CHECKLIST_TEXTAREA);
$('#irt-controls').append(JSPLib.menu.renderCacheControls());
$('#irt-cache-controls-message').append(JSPLib.menu.renderExpandable("Cache Data details", CACHE_DATA_DETAILS));
$('#irt-cache-controls').append(JSPLib.menu.renderLinkclick('cache_info', true));
$('#irt-cache-controls').append(JSPLib.menu.renderCacheInfoTable());
$('#irt-cache-controls').append(JSPLib.menu.renderLinkclick('purge_cache', true));
$('#irt-controls').append(JSPLib.menu.renderCacheEditor(true));
$('#irt-cache-editor-message').append(JSPLib.menu.renderExpandable("Program Data details", PROGRAM_DATA_DETAILS));
$('#irt-cache-editor-controls').append(JSPLib.menu.renderKeyselect('data_source', true));
$('#irt-cache-editor-controls').append(JSPLib.menu.renderDataSourceSections());
$('#irt-section-indexed-db').append(JSPLib.menu.renderKeyselect('data_type', true));
$('#irt-section-indexed-db').append(JSPLib.menu.renderKeyselect('tag_category', true));
$('#irt-section-indexed-db').append(JSPLib.menu.renderKeyselect('query_order', true));
$('#irt-section-local-storage').append(JSPLib.menu.renderCheckbox('raw_data', true));
$('#irt-cache-editor-controls').append(JSPLib.menu.renderTextinput('data_name', 20, true));
JSPLib.menu.engageUI(true, true);
JSPLib.menu.saveUserSettingsClick();
JSPLib.menu.resetUserSettingsClick(LOCALSTORAGE_KEYS);
$('#irt-tag-name-view').on(PROGRAM_CLICK, FUNC.ViewChecklistTag);
$('#irt-tag-name-save').on(PROGRAM_CLICK, FUNC.SaveChecklistTag);
$('#irt-tag-name-populate').on(PROGRAM_CLICK, FUNC.PopulateChecklistTag);
$('#irt-tag-name-list').on(PROGRAM_CLICK, FUNC.ListChecklistTags);
JSPLib.menu.cacheInfoClick();
JSPLib.menu.purgeCacheClick();
JSPLib.menu.expandableClick();
JSPLib.menu.dataSourceChange();
$('#irt-control-data-type').on('change.irt', FUNC.DataTypeChange);
JSPLib.menu.rawDataChange();
JSPLib.menu.getCacheClick(FUNC.ValidateProgramData);
JSPLib.menu.saveCacheClick(FUNC.ValidateProgramData, FUNC.ValidateEntry);
JSPLib.menu.deleteCacheClick();
JSPLib.menu.listCacheClick();
JSPLib.menu.refreshCacheClick();
JSPLib.menu.cacheAutocomplete();
};
//Main program
FUNC.Main = function(self) {
self.debuglog("Initialize start:", JSPLib.utility.getProgramTime());
const preload = {
run_on_settings: true,
default_data: DEFAULT_VALUES,
initialize_func: FUNC.InitializeProgramValues,
menu_css: SETTINGS_MENU_CSS + '\n' + LIBRARY_MENU_CSS,
};
if (!JSPLib.menu.preloadScript(IRT, FUNC.RenderSettingsMenu, preload)) return;
FUNC.SetupInitializations();
JSPLib.utility.setCSSStyle(PROGRAM_CSS, 'program');
JSPLib.statistics.addPageStatistics(PROGRAM_NAME);
JSPLib.load.noncriticalTasks(FUNC.CleanupTasks);
};
/****Function decoration****/
JSPLib.debug.addProgramTimers(FUNC, {
sync_funcs: ['RenderTagQueryColumn', 'RenderSettingsMenu'],
async_funcs: ['RelatedTagsButton', 'WikiPageButton', 'RelatedTagsQuery'],
});
JSPLib.debug.addProgramLogs(FUNC, [
'Main', 'ValidateEntry', 'InitializeProgramValues',
'InitializeTagColumns', 'InitializeUserMediaTags', 'InitializeTranslatedTags',
'RelatedTagsQuery', 'TagsOverlapQuery', 'WikiPageTagsQuery',
'GetCachedData', 'BroadcastIRT',
]);
/****Initialization****/
//Variables for debug.js
JSPLib.debug.debug_console = false;
JSPLib.debug.level = JSPLib.debug.INFO;
JSPLib.debug.program_shortcut = PROGRAM_SHORTCUT;
//Variables for menu.js
JSPLib.menu.program_shortcut = PROGRAM_SHORTCUT;
JSPLib.menu.program_name = PROGRAM_NAME;
JSPLib.menu.program_data = IRT;
JSPLib.menu.program_reset_data = PROGRAM_RESET_KEYS;
JSPLib.menu.program_data_regex = PROGRAM_DATA_REGEX;
JSPLib.menu.program_data_key = FUNC.OptionCacheDataKey;
JSPLib.menu.settings_config = SETTINGS_CONFIG;
JSPLib.menu.control_config = CONTROL_CONFIG;
//Export JSPLib
JSPLib.load.exportData(PROGRAM_NAME, IRT, {datalist: ['cached_data'], other_data: {FUNC}});
/****Execution start****/
JSPLib.load.programInitialize(FUNC.Main, {program_name: PROGRAM_NAME, required_variables: PROGRAM_LOAD_REQUIRED_VARIABLES, required_selectors: PROGRAM_LOAD_REQUIRED_SELECTORS});