29 lines
2.0 KiB
TypeScript
29 lines
2.0 KiB
TypeScript
|
import { database, withTransaction, type QueryContext } from '../database.js';
|
||
|
|
import type { NoteImage } from '../../shared/types.js';
|
||
|
|
|
||
|
|
interface ImageRow extends Omit<NoteImage, 'id' | 'note_id' | 'order_index'> { id: number | string; note_id: number | string; order_index: number | string }
|
||
|
|
function toImage(row: ImageRow): NoteImage { return { ...row, id: Number(row.id), note_id: Number(row.note_id), order_index: Number(row.order_index), url: /^https?:\/\//.test(row.url) || row.url.startsWith('/') ? row.url : `/uploads/${row.url}` }; }
|
||
|
|
|
||
|
|
export const imagesRepository = {
|
||
|
|
async listByNote(noteId: number, versionNumber?: number): Promise<NoteImage[]> {
|
||
|
|
const rows = await database.all<ImageRow>(`SELECT id, note_id, url, width, height, order_index, storage_provider, storage_key FROM images
|
||
|
|
WHERE note_id = ?${versionNumber ? ' AND version_number = ?' : ''} ORDER BY order_index ASC, id ASC`, versionNumber ? [noteId, versionNumber] : [noteId]);
|
||
|
|
return rows.map(toImage);
|
||
|
|
},
|
||
|
|
async createMany(noteId: number, images: { url: string; width: number; height: number; storageProvider: 'local' | 'tencent_cos'; storageKey: string }[], versionNumber = 1, existing?: QueryContext): Promise<number[]> {
|
||
|
|
const insert = async (tx: QueryContext) => {
|
||
|
|
const ids: number[] = [];
|
||
|
|
for (let index = 0; index < images.length; index += 1) {
|
||
|
|
const image = images[index];
|
||
|
|
ids.push(await tx.insertId('INSERT INTO images (note_id, url, width, height, order_index, storage_provider, storage_key, version_number) VALUES (?, ?, ?, ?, ?, ?, ?, ?)', [noteId, image.url, image.width, image.height, index, image.storageProvider, image.storageKey, versionNumber]));
|
||
|
|
}
|
||
|
|
return ids;
|
||
|
|
};
|
||
|
|
return existing ? insert(existing) : withTransaction(insert);
|
||
|
|
},
|
||
|
|
async findById(id: number): Promise<NoteImage | null> {
|
||
|
|
const row = await database.one<ImageRow>('SELECT id, note_id, url, width, height, order_index, storage_provider, storage_key FROM images WHERE id = ?', [id]);
|
||
|
|
return row ? toImage(row) : null;
|
||
|
|
},
|
||
|
|
};
|