Talon/chat-with-gpt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add support for deleting chats

Browse Source
This commit is contained in:
Cogent Apps
2023-03-17 20:45:27 +00:00
parent 6415e3032c
commit 1be33f5d0b
10 changed files with 193 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/src/backend.ts
View File

@@ -78,6 +78,12 @@ export class Backend extends EventEmitter {
id: chatID,
messages: new MessageTree(),
} as Chat;
if (response[chatID].deleted) {
chatManager.deleteChat(chatID);
continue;
}
chat.title = response[chatID].title || chat.title;
chat.messages.addMessages(response[chatID].messages);
chatManager.loadChat(chat);
@@ -131,6 +137,14 @@ export class Backend extends EventEmitter {
return null;
}
async deleteChat(id: string) {
if (!this.isAuthenticated) {
return;
}
return this.post(endpoint + '/delete', { id });
}
async get(url: string) {
const response = await fetch(url);
if (!response.ok) {

58
app/src/chat-manager.ts
View File

@@ -28,13 +28,20 @@ export class ChatManager extends EventEmitter {
});
channel.onmessage = (message: {
type: 'chat-update',
type: 'chat-update' | 'chat-delete',
data: string,
}) => {
const chat = deserializeChat(message.data);
const id = chat.id;
this.chats.set(id, chat);
this.emit(id);
switch (message.type) {
case 'chat-update':
const chat = deserializeChat(message.data);
const id = chat.id;
this.chats.set(id, chat);
this.emit(id);
break;
case 'chat-delete':
this.deleteChat(message.data, false);
break;
}
};
(async () => {
@@ -69,7 +76,7 @@ export class ChatManager extends EventEmitter {
public async sendMessage(message: UserSubmittedMessage) {
const chat = this.chats.get(message.chatID);
if (!chat) {
if (!chat || chat.deleted) {
throw new Error('Chat not found');
}
@@ -101,7 +108,7 @@ export class ChatManager extends EventEmitter {
public async regenerate(message: Message, requestedParameters: Parameters) {
const chat = this.chats.get(message.chatID);
if (!chat) {
if (!chat || chat.deleted) {
throw new Error('Chat not found');
}
@@ -116,7 +123,7 @@ export class ChatManager extends EventEmitter {
const latestMessage = messages[messages.length - 1];
const chat = this.chats.get(latestMessage.chatID);
if (!chat) {
if (!chat || chat.deleted) {
throw new Error('Chat not found');
}
@@ -250,12 +257,19 @@ export class ChatManager extends EventEmitter {
const serialized = await idb.get('chats');
if (serialized) {
for (const chat of serialized) {
const messages = new MessageTree();
for (const m of chat.messages) {
messages.addMessage(m);
try {
if (chat.deleted) {
continue;
}
const messages = new MessageTree();
for (const m of chat.messages) {
messages.addMessage(m);
}
chat.messages = messages;
this.loadChat(chat);
} catch (e) {
console.error(e);
}
chat.messages = messages;
this.loadChat(chat);
}
this.emit('update');
}
@@ -268,6 +282,11 @@ export class ChatManager extends EventEmitter {
}
const existing = this.chats.get(chat.id);
if (existing && existing.deleted) {
return;
}
if (existing && existing.title && !chat.title) {
chat.title = existing.title;
}
@@ -283,6 +302,15 @@ export class ChatManager extends EventEmitter {
public get(id: string): Chat | undefined {
return this.chats.get(id);
}
public deleteChat(id: string, broadcast = true) {
this.chats.delete(id);
this.search.delete(id);
this.emit(id);
if (broadcast) {
channel.postMessage({ type: 'chat-delete', data: id });
}
}
}
export class Search {
@@ -309,6 +337,10 @@ export class Search {
}
}
public delete(id: string) {
this.index.remove({ id });
}
public query(query: string) {
if (!query?.trim().length) {
const searchResults = Array.from(this.chats.values())

5
app/src/components/sidebar/index.tsx
View File

@@ -103,6 +103,7 @@ export default function Sidebar(props: {
className?: string;
}) {
const intl = useIntl();
const context = useAppContext();
const dispatch = useAppDispatch();
const sidebarOpen = useAppSelector(selectSidebarOpen);
const onBurgerClick = useCallback(() => dispatch(toggleSidebar()), [dispatch]);
@@ -147,13 +148,13 @@ export default function Sidebar(props: {
<Menu.Divider />
<Menu.Item color="red" onClick={() => backend.current?.logout()} icon={<i className="fas fa-sign-out-alt" />}>
<FormattedMessage defaultMessage={"Sign out"} />
</Menu.Item>
</Menu.Item>
*/}
</Menu.Dropdown>
</Menu>
)}
</Container>
), [sidebarOpen, width, ref, burgerLabel, onBurgerClick, dispatch]);
), [sidebarOpen, width, ref, burgerLabel, onBurgerClick, dispatch, context.chat.chats.size]);
return elem;
}

91
app/src/components/sidebar/recent-chats.tsx
View File

@@ -1,10 +1,13 @@
import styled from '@emotion/styled';
import { useCallback, useEffect } from 'react';
import { FormattedMessage } from 'react-intl';
import { Link } from 'react-router-dom';
import { Link, useNavigate } from 'react-router-dom';
import { useAppContext } from '../../context';
import { useAppDispatch } from '../../store';
import { toggleSidebar } from '../../store/sidebar';
import { ActionIcon, Menu } from '@mantine/core';
import { useModals } from '@mantine/modals';
import { backend } from '../../backend';
const Container = styled.div`
margin: calc(1.618rem - 1rem);
@@ -19,8 +22,9 @@ const Empty = styled.p`
const ChatList = styled.div``;
const ChatListItem = styled(Link)`
const ChatListItemLink = styled(Link)`
display: block;
position: relative;
padding: 0.4rem 1rem;
margin: 0.218rem 0;
line-height: 1.7;
@@ -35,15 +39,13 @@ const ChatListItem = styled(Link)`
background: #2b3d54;
}
&, * {
color: white;
}
strong {
display: block;
font-weight: 400;
font-size: 1rem;
line-height: 1.6;
padding-right: 1rem;
color: white;
}
p {
@@ -51,8 +53,77 @@ const ChatListItem = styled(Link)`
font-weight: 200;
opacity: 0.8;
}
.mantine-ActionIcon-root {
position: absolute;
right: 0.5rem;
top: 50%;
margin-top: -14px;
}
`;
function ChatListItem(props: { chat: any, onClick: any, selected: boolean }) {
const c = props.chat;
const context = useAppContext();
const modals = useModals();
const navigate = useNavigate();
const onDelete = useCallback(() => {
modals.openConfirmModal({
title: "Are you sure you want to delete this chat?",
children: <p style={{ lineHeight: 1.7 }}>The chat "{c.title}" will be permanently deleted. This cannot be undone.</p>,
labels: {
confirm: "Delete permanently",
cancel: "Cancel",
},
confirmProps: {
color: 'red',
},
onConfirm: async () => {
try {
await backend.current?.deleteChat(c.chatID);
context.chat.deleteChat(c.chatID);
navigate('/');
} catch (e) {
console.error(e);
modals.openConfirmModal({
title: "Something went wrong",
children: <p style={{ lineHeight: 1.7 }}>The chat "{c.title}" could not be deleted.</p>,
labels: {
confirm: "Try again",
cancel: "Cancel",
},
onConfirm: onDelete,
});
}
},
});
}, [c.chatID, c.title]);
return (
<ChatListItemLink to={'/chat/' + c.chatID}
onClick={props.onClick}
data-chat-id={c.chatID}
className={props.selected ? 'selected' : ''}>
<strong>{c.title || <FormattedMessage defaultMessage={"Untitled"} description="default title for untitled chat sessions" />}</strong>
{props.selected && (
<Menu>
<Menu.Target>
<ActionIcon>
<i className="fas fa-ellipsis" />
</ActionIcon>
</Menu.Target>
<Menu.Dropdown>
<Menu.Item onClick={onDelete} color="red" icon={<i className="fa fa-trash" />}>
<FormattedMessage defaultMessage={"Delete this chat"} />
</Menu.Item>
</Menu.Dropdown>
</Menu>
)}
</ChatListItemLink>
);
}
export default function RecentChats(props: any) {
const context = useAppContext();
const dispatch = useAppDispatch();
@@ -79,13 +150,7 @@ export default function RecentChats(props: any) {
<Container>
{recentChats.length > 0 && <ChatList>
{recentChats.map(c => (
<ChatListItem key={c.chatID}
to={'/chat/' + c.chatID}
onClick={onClick}
data-chat-id={c.chatID}
className={c.chatID === currentChatID ? 'selected' : ''}>
<strong>{c.title || <FormattedMessage defaultMessage={"Untitled"} description="default title for untitled chat sessions" />}</strong>
</ChatListItem>
<ChatListItem key={c.chatID} chat={c} onClick={onClick} selected={c.chatID === currentChatID} />
))}
</ChatList>}
{recentChats.length === 0 && <Empty>

1
app/src/types.ts
View File

@@ -44,6 +44,7 @@ export interface Chat {
title?: string | null;
created: number;
updated: number;
deleted?: boolean;
}
export function serializeChat(chat: Chat): string {