Spaces:
Runtime error
Runtime error
File size: 3,676 Bytes
faca43f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 |
import Dexie, { type Table, liveQuery } from 'dexie';
import { refresh_chats_writable_empty, refresh_chats_writable } from "../routes/LayoutWritable";
import { env } from '$env/dynamic/public';
export interface Chat {
index?: number;
title: string,
id: string;
createdAt: Date,
message?: Array<MessageDb>;
}
export interface MessageDb {
content: string;
from: string;
id: string;
createdAt: Date;
updatedAt: Date;
}
export class ChatDatabase extends Dexie {
chats!: Table<Chat>;
constructor() {
super('blindchat');
this.version(16).stores({
chats: '++index, title, createdAt, id, message'
});
}
}
export async function createChat(id_chat: string, msg: MessageDb | undefined, title?: string) {
try {
let title_f = ""
if (title === undefined) {
let count = (await db.chats.count()) + 1
title_f = "Untitled " + count
}
else
title_f = title
const chat = {
id: id_chat,
title: title_f,
message: msg === undefined ? undefined : [msg],
createdAt: new Date(),
}
const id = await db.chats.add(chat);
} catch (error) {
console.log(error)
}
let push = await getChats()
refresh_chats_writable.set(push)
}
export async function deleteAllChats() {
const chat_ret = await db.chats.clear()
refresh_chats_writable_empty.set(true)
}
export async function deleteChat(id_chat: string) {
const chat_ret = await db.chats.where("id").equals(id_chat).delete()
let count = await db.chats.count()
if (count > 0) {
let push = await getChats()
refresh_chats_writable.set(push)
}
else {
refresh_chats_writable_empty.set(true)
}
}
export async function modifyTitle(id_chat: string, newTitle: string) {
const chat_ret = db.chats.where("id").equals(id_chat)
let count = await chat_ret.count()
if (count > 0) {
let res = await chat_ret.first()
chat_ret.modify({title: newTitle})
let push = await getChats()
refresh_chats_writable.set(push)
}
}
export async function addMessageToChat(id_chat: string, msg: MessageDb) {
const chat_ret = db.chats.where("id").equals(id_chat)
let count = await chat_ret.count()
if (count < 1) {
createChat(id_chat, msg, )
}
else {
let msgs: MessageDb[]
chat_ret.first().then((res) => {
if (res?.message == undefined) {
msgs.push(msg)
res.message = msgs
}
res.message.push(msg)
chat_ret.modify({id: id_chat, message: res.message})
})
}
}
export async function getTitle(id_chat: string) {
let title_ret = env.PUBLIC_APP_NAME
try {
const chat_ret = await (db.chats.where("id").equals(id_chat).first())
title_ret = chat_ret!.title
}
catch (err) {
console.log(err)
}
return title_ret
}
export async function getMessages(id_chat: string) {
try {
const chat_ret = await db.chats.where("id").equals(id_chat).first()
const msg = chat_ret?.message
return [...msg]
}
catch (err) {
console.log(err)
}
return undefined
}
export async function getChats() {
let titles = []
try {
const all = (await db.chats.orderBy('createdAt').toArray()).forEach(function (chat) {
titles.push({
title: chat.title,
model: "", // Hardcoded for now
id: chat.id,
updatedAt: chat.createdAt,
createdAt: chat.createdAt,
})
});
}
catch (err) {
console.log(err)
}
return titles;
}
export async function getChat(id_chat: string) {
const chat_ret = db.chats.where("id").equals(id_chat).first().then((res) => {
return res;
})
}
export const db = new ChatDatabase();
|