import { characters, saveChat, system_messages, system_message_types, this_chid, openCharacterChat, chat_metadata, callPopup, getRequestHeaders, getThumbnailUrl, getCharacters, chat, saveChatConditional, } from "../script.js"; import { humanizedDateTime } from "./RossAscends-mods.js"; import { getGroupPastChats, group_activation_strategy, groups, openGroupById, openGroupChat, saveGroupBookmarkChat, selected_group, } from "./group-chats.js"; import { createTagMapFromList } from "./tags.js"; import { delay, getUniqueName, stringFormat, } from "./utils.js"; export { createNewBookmark, showBookmarksButtons, } const bookmarkNameToken = 'Bookmark #'; async function getExistingChatNames() { if (selected_group) { const data = await getGroupPastChats(selected_group); return data.map(x => x.file_name); } else { const response = await fetch("/getallchatsofcharacter", { method: 'POST', headers: getRequestHeaders(), body: JSON.stringify({ avatar_url: characters[this_chid].avatar }) }); if (response.ok) { const data = await response.json(); return Object.values(data).map(x => x.file_name.replace('.jsonl', '')); } } } async function getBookmarkName() { const chatNames = await getExistingChatNames(); const popupText = `

Enter the bookmark name:

Leave empty to auto-generate.`; let name = await callPopup(popupText, 'input'); if (name === false) { return null; } else if (name === '') { for (let i = chatNames.length; i < 1000; i++) { name = bookmarkNameToken + i; if (!chatNames.includes(name)) { break; } } } return `${name} - ${humanizedDateTime()}`; } function getMainChatName() { if (chat_metadata) { if (chat_metadata['main_chat']) { return chat_metadata['main_chat']; } // groups didn't support bookmarks before chat metadata was introduced else if (selected_group) { return null; } else if (characters[this_chid].chat && characters[this_chid].chat.includes(bookmarkNameToken)) { const tokenIndex = characters[this_chid].chat.lastIndexOf(bookmarkNameToken); chat_metadata['main_chat'] = characters[this_chid].chat.substring(0, tokenIndex).trim(); return chat_metadata['main_chat']; } } return null; } function showBookmarksButtons() { try { if (selected_group) { $("#option_convert_to_group").hide(); } else { $("#option_convert_to_group").show(); } if (chat_metadata['main_chat']) { // In bookmark chat $("#option_back_to_main").show(); $("#option_new_bookmark").show(); } else if (!selected_group && !characters[this_chid].chat) { // No chat recorded on character $("#option_back_to_main").hide(); $("#option_new_bookmark").hide(); } else { // In main chat $("#option_back_to_main").hide(); $("#option_new_bookmark").show(); } } catch { $("#option_back_to_main").hide(); $("#option_new_bookmark").hide(); $("#option_convert_to_group").hide(); } } async function saveBookmarkMenu() { if (!chat.length) { toastr.warning('The chat is empty.', 'Bookmark creation failed'); return; } return createNewBookmark(chat.length - 1); } async function createNewBookmark(mesId) { if (!chat.length) { toastr.warning('The chat is empty.', 'Bookmark creation failed'); return; } if (mesId < 0 || mesId >= chat.length) { toastr.warning('Invalid message ID.', 'Bookmark creation failed'); return; } const lastMes = chat[mesId]; if (typeof lastMes.extra !== 'object') { lastMes.extra = {}; } if (lastMes.extra.bookmark_link) { const confirm = await callPopup('Bookmark checkpoint for the last message already exists. Would you like to replace it?', 'confirm'); if (!confirm) { return; } } await delay(250); let name = await getBookmarkName(); if (!name) { return; } const mainChat = selected_group ? groups?.find(x => x.id == selected_group)?.chat_id : characters[this_chid].chat; const newMetadata = { main_chat: mainChat }; if (selected_group) { await saveGroupBookmarkChat(selected_group, name, newMetadata, mesId); } else { await saveChat(name, newMetadata, mesId); } lastMes.extra['bookmark_link'] = name; $(`.mes[mesid="${mesId}"]`).attr('bookmark_link', name); await saveChatConditional(); toastr.success('Click the bookmark icon in the last message to open the checkpoint chat.', 'Bookmark created', { timeOut: 10000 }); } async function backToMainChat() { const mainChatName = getMainChatName(); const allChats = await getExistingChatNames(); if (allChats.includes(mainChatName)) { if (selected_group) { await openGroupChat(selected_group, mainChatName); } else { await openCharacterChat(mainChatName); } } } async function convertSoloToGroupChat() { if (selected_group) { console.log('Already in group. No need for conversion'); return; } if (this_chid === undefined) { console.log('Need to have a character selected'); return; } const character = characters[this_chid]; // Populate group required fields const name = getUniqueName(`Group: ${character.name}`, y => groups.findIndex(x => x.name === y) !== -1); const avatar = getThumbnailUrl('avatar', character.avatar); const chatName = humanizedDateTime(); const chats = [chatName]; const members = [character.avatar]; const activationStrategy = group_activation_strategy.NATURAL; const allowSelfResponses = false; const favChecked = character.fav || character.fav == 'true'; const metadata = Object.assign({}, chat_metadata); delete metadata.main_chat; const createGroupResponse = await fetch("/creategroup", { method: "POST", headers: getRequestHeaders(), body: JSON.stringify({ name: name, members: members, avatar_url: avatar, allow_self_responses: activationStrategy, activation_strategy: allowSelfResponses, disabled_members: [], chat_metadata: metadata, fav: favChecked, chat_id: chatName, chats: chats, }), }); if (!createGroupResponse.ok) { console.error('Group creation unsuccessful'); return; } const group = await createGroupResponse.json(); // Convert tags list and assign to group createTagMapFromList("#tagList", group.id); // Update chars list await getCharacters(); // Convert chat to group format const groupChat = chat.slice(); const genIdFirst = Date.now(); // Add something if the chat is empty if (groupChat.length === 0) { const newMessage = { ...system_messages[system_message_types.GROUP], send_date: humanizedDateTime(), extra: { type: system_message_types.GROUP } }; groupChat.push(newMessage); } for (let index = 0; index < groupChat.length; index++) { const message = groupChat[index]; // Save group-chat marker if (index == 0) { message.is_group = true; } // Skip messages we don't care about if (message.is_user || message.is_system || message.extra?.type === system_message_types.NARRATOR || message.force_avatar !== undefined) { continue; } // Set force fields for solo character message.name = character.name; message.original_avatar = character.avatar; message.force_avatar = getThumbnailUrl('avatar', character.avatar); message.is_name = true; // Allow regens of a single message in group if (typeof message.extra !== 'object') { message.extra = { gen_id: genIdFirst + index }; } } // Save group chat const createChatResponse = await fetch("/savegroupchat", { method: "POST", headers: getRequestHeaders(), body: JSON.stringify({ id: chatName, chat: groupChat }), }); if (!createChatResponse.ok) { console.error('Group chat creation unsuccessful'); return; } // Click on the freshly selected group to open it await openGroupById(group.id); toastr.success('The chat has been successfully converted!'); } jQuery(function () { $('#option_new_bookmark').on('click', saveBookmarkMenu); $('#option_back_to_main').on('click', backToMainChat); $('#option_convert_to_group').on('click', convertSoloToGroupChat); });