107 lines
3.3 KiB
JavaScript
107 lines
3.3 KiB
JavaScript
const { collection, doc, addDoc, getDoc, getDocs, updateDoc, deleteDoc, query, where, orderBy, Timestamp } = require('firebase/firestore');
|
|
const { getFirestoreInstance } = require('../../services/firebaseClient');
|
|
const { createEncryptedConverter } = require('../firestoreConverter');
|
|
const encryptionService = require('../../services/encryptionService');
|
|
|
|
const userPresetConverter = createEncryptedConverter(['prompt', 'title']);
|
|
|
|
const defaultPresetConverter = {
|
|
toFirestore: (data) => data,
|
|
fromFirestore: (snapshot, options) => {
|
|
const data = snapshot.data(options);
|
|
return { ...data, id: snapshot.id };
|
|
}
|
|
};
|
|
|
|
function userPresetsCol() {
|
|
const db = getFirestoreInstance();
|
|
return collection(db, 'prompt_presets').withConverter(userPresetConverter);
|
|
}
|
|
|
|
function defaultPresetsCol() {
|
|
const db = getFirestoreInstance();
|
|
// Path must have an odd number of segments. 'v1' is a placeholder document.
|
|
return collection(db, 'defaults/v1/prompt_presets').withConverter(defaultPresetConverter);
|
|
}
|
|
|
|
async function getPresets(uid) {
|
|
const userPresetsQuery = query(userPresetsCol(), where('uid', '==', uid));
|
|
const defaultPresetsQuery = query(defaultPresetsCol()); // Defaults have no owner
|
|
|
|
const [userSnapshot, defaultSnapshot] = await Promise.all([
|
|
getDocs(userPresetsQuery),
|
|
getDocs(defaultPresetsQuery)
|
|
]);
|
|
|
|
const presets = [
|
|
...defaultSnapshot.docs.map(d => d.data()),
|
|
...userSnapshot.docs.map(d => d.data())
|
|
];
|
|
|
|
return presets.sort((a, b) => {
|
|
if (a.is_default && !b.is_default) return -1;
|
|
if (!a.is_default && b.is_default) return 1;
|
|
return a.title.localeCompare(b.title);
|
|
});
|
|
}
|
|
|
|
async function getPresetTemplates() {
|
|
const q = query(defaultPresetsCol(), orderBy('title', 'asc'));
|
|
const snapshot = await getDocs(q);
|
|
return snapshot.docs.map(doc => doc.data());
|
|
}
|
|
|
|
async function create({ uid, title, prompt }) {
|
|
const now = Timestamp.now();
|
|
const newPreset = {
|
|
uid: uid,
|
|
title,
|
|
prompt,
|
|
is_default: 0,
|
|
created_at: now,
|
|
};
|
|
const docRef = await addDoc(userPresetsCol(), newPreset);
|
|
return { id: docRef.id };
|
|
}
|
|
|
|
async function update(id, { title, prompt }, uid) {
|
|
const docRef = doc(userPresetsCol(), id);
|
|
const docSnap = await getDoc(docRef);
|
|
|
|
if (!docSnap.exists() || docSnap.data().uid !== uid || docSnap.data().is_default) {
|
|
throw new Error("Preset not found or permission denied to update.");
|
|
}
|
|
|
|
// Encrypt sensitive fields before sending to Firestore because `updateDoc` bypasses converters.
|
|
const updates = {};
|
|
if (title !== undefined) {
|
|
updates.title = encryptionService.encrypt(title);
|
|
}
|
|
if (prompt !== undefined) {
|
|
updates.prompt = encryptionService.encrypt(prompt);
|
|
}
|
|
updates.updated_at = Timestamp.now();
|
|
|
|
await updateDoc(docRef, updates);
|
|
return { changes: 1 };
|
|
}
|
|
|
|
async function del(id, uid) {
|
|
const docRef = doc(userPresetsCol(), id);
|
|
const docSnap = await getDoc(docRef);
|
|
|
|
if (!docSnap.exists() || docSnap.data().uid !== uid || docSnap.data().is_default) {
|
|
throw new Error("Preset not found or permission denied to delete.");
|
|
}
|
|
|
|
await deleteDoc(docRef);
|
|
return { changes: 1 };
|
|
}
|
|
|
|
module.exports = {
|
|
getPresets,
|
|
getPresetTemplates,
|
|
create,
|
|
update,
|
|
delete: del,
|
|
};
|