File size: 3,927 Bytes
5528541
faca43f
5528541
faca43f
 
5528541
 
 
 
 
 
faca43f
 
 
5528541
 
 
 
 
faca43f
 
 
5528541
 
 
 
 
 
 
 
0015338
 
 
5528541
 
 
faca43f
 
5528541
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
faca43f
 
 
5528541
 
faca43f
 
 
5528541
 
 
 
 
 
 
 
faca43f
 
 
5528541
 
 
 
 
 
 
 
faca43f
 
5528541
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
faca43f
 
 
5528541
 
 
 
 
 
 
 
faca43f
 
 
5528541
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
faca43f
 
 
5528541
 
 
 
 
faca43f
 
 
 
5528541
 
 
 
 
 
faca43f
 
 
5528541
 
 
 
 
 
 
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
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
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;
	model: string;
	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: null,
		});
		this.version(17).stores({
			chats: null,		
		});
		this.version(18).stores({
			chats: "++index, title, createdAt, id, message, model",
		});
	}
}

export async function createChat(
	id_chat: string,
	msg: MessageDb | undefined,
	model: string,
	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(),
			model: model,
		};
		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, model: string) {
	const chat_ret = db.chats.where("id").equals(id_chat);
	let count = await chat_ret.count();
	if (count < 1) {
		createChat(id_chat, msg, model);
	} 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 getModel(id_chat: string) {
	try {
		const chat_ret = await db.chats.where("id").equals(id_chat).first();
		let model = chat_ret?.model;
		if (model === undefined) return "";
		return model;
	} catch (err) {
		console.log(err);
	}
	return "";
}

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();