Bookmarks support for groups

This commit is contained in:
SillyLossy
2023-05-02 19:56:29 +03:00
parent 7a94ddfa9c
commit 89e088a6d1
3 changed files with 84 additions and 43 deletions

View File

@@ -3611,7 +3611,7 @@ async function saveMetadata() {
}
}
async function saveChatConditional() {
export async function saveChatConditional() {
if (selected_group) {
await saveGroupChat(selected_group, true);
}
@@ -4272,10 +4272,9 @@ $(document).ready(function () {
//open the history view again after 100ms
//hide option popup menu
setTimeout(function () {
await delay(100);
$("#option_select_chat").click();
$("#options").hide();
}, 100);
}
if (popup_type == "del_ch") {
console.log(

View File

@@ -12,9 +12,17 @@ import {
getThumbnailUrl,
getCharacters,
chat,
saveChatConditional,
} from "../script.js";
import { humanizedDateTime } from "./RossAscends-mods.js";
import { group_activation_strategy, groups, selected_group } from "./group-chats.js";
import {
getGroupPastChats,
group_activation_strategy,
groups,
openGroupChat,
saveGroupBookmarkChat,
selected_group,
} from "./group-chats.js";
import { createTagMapFromList } from "./tags.js";
import {
@@ -30,6 +38,10 @@ export {
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(),
@@ -40,20 +52,20 @@ async function getExistingChatNames() {
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 = `<h3>Enter the bookmark name:<h3>
<small>Using existing name will overwrite your bookmark chat.
<br>Leave empty to auto-generate.</small>`;
<small>Leave empty to auto-generate.</small>`;
let name = await callPopup(popupText, 'input');
if (name === false) {
return null;
}
else if (name === '') {
for (let i = 0; i < 1000; i++) {
for (let i = chatNames.length; i < 1000; i++) {
name = bookmarkNameToken + i;
if (!chatNames.includes(name)) {
break;
@@ -61,7 +73,7 @@ async function getBookmarkName() {
}
}
return name;
return `${name} - ${humanizedDateTime()}`;
}
function getMainChatName() {
@@ -69,6 +81,10 @@ function getMainChatName() {
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();
@@ -80,24 +96,24 @@ function getMainChatName() {
function showBookmarksButtons() {
try {
// In groups or without an active chat
if (selected_group || !characters[this_chid].chat) {
$("#option_back_to_main").hide();
$("#option_new_bookmark").hide();
if (selected_group) {
$("#option_convert_to_group").hide();
}
// In main chat
else if (!chat_metadata['main_chat']) {
$("#option_back_to_main").hide();
$("#option_new_bookmark").show();
} else {
$("#option_convert_to_group").show();
}
if (chat_metadata['main_chat']) {
// In bookmark chat
else {
$("#option_back_to_main").show();
$("#option_new_bookmark").show();
$("#option_convert_to_group").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 {
@@ -108,30 +124,36 @@ function showBookmarksButtons() {
}
async function createNewBookmark() {
if (selected_group) {
alert('Chat bookmarks unsupported for groups');
throw new Error();
}
let name = await getBookmarkName(characters[this_chid].chat);
let name = await getBookmarkName();
if (!name) {
return;
}
const newMetadata = { main_chat: characters[this_chid].chat };
saveChat(name, newMetadata);
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);
} else {
await saveChat(name, newMetadata);
}
let mainMessage = stringFormat(system_messages[system_message_types.BOOKMARK_CREATED].mes, name, name);
sendSystemMessage(system_message_types.BOOKMARK_CREATED, mainMessage);
saveChat();
await saveChatConditional();
}
async function backToMainChat() {
const mainChatName = getMainChatName(characters[this_chid].chat);
const mainChatName = getMainChatName();
const allChats = await getExistingChatNames();
if (allChats.includes(mainChatName)) {
openCharacterChat(mainChatName);
if (selected_group) {
await openGroupChat(selected_group, mainChatName);
} else {
await openCharacterChat(mainChatName);
}
}
}
@@ -158,6 +180,7 @@ async function convertSoloToGroupChat() {
const allowSelfResponses = false;
const favChecked = character.fav == 'true';
const metadata = Object.assign({}, chat_metadata);
delete metadata.main_chat;
const createGroupResponse = await fetch("/creategroup", {
method: "POST",

View File

@@ -1181,6 +1181,25 @@ export async function deleteGroupChat(groupId, chatId) {
}
}
export async function saveGroupBookmarkChat(groupId, name, metadata) {
const group = groups.find(x => x.id === groupId);
if (!group) {
return;
}
group.past_metadata[name] = { ...chat_metadata, ...(metadata || {}) };
group.chats.push(name);
await editGroup(groupId, true);
await fetch("/savegroupchat", {
method: "POST",
headers: getRequestHeaders(),
body: JSON.stringify({ id: name, chat: [...chat] }),
});
}
$(document).ready(() => {
$(document).on("click", ".group_select", selectGroup);
$("#rm_group_filter").on("input", filterGroupMembers);