123 lines
4.4 KiB
JavaScript
123 lines
4.4 KiB
JavaScript
import {
|
|
ADD as MSSQL_ADD_CATEGORYPIC,
|
|
GET_NEXT_SORT as MSSQL_GET_NEXT_SORT,
|
|
} from '../../providers/mssql/queries/categorypic.js';
|
|
import {
|
|
ADD as SQLITE_ADD_CATEGORYPIC,
|
|
GET_NEXT_SORT as SQLITE_GET_NEXT_SORT,
|
|
} from '../../providers/sqlite3/queries/categorypic.js';
|
|
import { CREATE as MSSQL_CREATE_PIC } from '../../providers/mssql/queries/pic.js';
|
|
import { CREATE as SQLITE_CREATE_PIC } from '../../providers/sqlite3/queries/pic.js';
|
|
import processAvifImage from '../../util/processAvifImage.js';
|
|
import newFullHash from '../../util/newFullHash.js';
|
|
import resolveCategoryAccess from './resolveCategoryAccess.js';
|
|
import makeRunner from '../runner.js';
|
|
|
|
export default function addImage(provider) {
|
|
const access = resolveCategoryAccess(provider);
|
|
const isSqlite = provider?.dialect === 'sqlite3';
|
|
|
|
const getSortSql = isSqlite ? SQLITE_GET_NEXT_SORT : MSSQL_GET_NEXT_SORT;
|
|
const createPicSql = isSqlite ? SQLITE_CREATE_PIC : MSSQL_CREATE_PIC;
|
|
const addCategoryPicSql = isSqlite ? SQLITE_ADD_CATEGORYPIC : MSSQL_ADD_CATEGORYPIC;
|
|
const touchSql = isSqlite
|
|
? "UPDATE category SET updated_at = strftime('%Y-%m-%dT%H:%M:%fZ', 'now') WHERE id = @categoryId"
|
|
: 'UPDATE category SET updated_at = SYSUTCDATETIME() WHERE id = @categoryId';
|
|
|
|
/**
|
|
* Adds an image to a category, processing full, thumb, and 32px-edge small-thumb AVIF variants.
|
|
*
|
|
* @param {number} googleAuthId - The id of the Google auth user.
|
|
* @param {number} categoryId - The id of the category.
|
|
* @param {Buffer} image - The source image buffer.
|
|
* @param {object} [tx] - Optional transaction to run within.
|
|
* @returns {Promise<{id: number, fullHash: string, thumbW: number, thumbH: number, thumbHash: string, smallThumbW: number, smallThumbH: number}>} The added image info.
|
|
*
|
|
* @example
|
|
* const result = await store.category.addImage(42, 5, imageBuffer);
|
|
* // => { id, fullHash, thumbW, thumbH, thumbHash }
|
|
*/
|
|
return async (googleAuthId, categoryId, image, tx) => {
|
|
const processed = await processAvifImage(image);
|
|
const fullHash = newFullHash(processed.full.buffer);
|
|
|
|
const runAdd = async (runner) => {
|
|
await access(googleAuthId, categoryId, runner);
|
|
|
|
const sortResult = await runner.query(getSortSql, { categoryId });
|
|
const sort = sortResult.rows[0].sort;
|
|
|
|
let picId;
|
|
const existingPic = await runner.query(
|
|
'SELECT id FROM pic WHERE hash = @fullHash',
|
|
{ fullHash },
|
|
);
|
|
|
|
if (existingPic.rows[0]) {
|
|
picId = existingPic.rows[0].id;
|
|
} else {
|
|
if (isSqlite) {
|
|
if (runner.writePicFull) {
|
|
await runner.writePicFull(fullHash, processed.full.buffer);
|
|
await runner.writePicThumb(fullHash, processed.thumb.buffer);
|
|
} else if (provider?.writePicFull) {
|
|
await provider.writePicFull(fullHash, processed.full.buffer);
|
|
await provider.writePicThumb(fullHash, processed.thumb.buffer);
|
|
}
|
|
}
|
|
|
|
const picResult = await runner.query(createPicSql, {
|
|
w: processed.full.w,
|
|
h: processed.full.h,
|
|
size: processed.full.size,
|
|
hash: fullHash,
|
|
thumbw: processed.thumb.w,
|
|
thumbh: processed.thumb.h,
|
|
thumbhash: processed.thumb.hash,
|
|
smallthumbw: processed.smallThumb.w,
|
|
smallthumbh: processed.smallThumb.h,
|
|
smallthumbbinary: processed.smallThumb.buffer,
|
|
source: '',
|
|
fullbinary: processed.full.buffer,
|
|
thumbbinary: processed.thumb.buffer,
|
|
});
|
|
|
|
picId = picResult.rows[0]?.id ?? picResult.insertId;
|
|
}
|
|
|
|
const existingCategoryPic = await runner.query(
|
|
'SELECT id, sort FROM categorypic WHERE category_id = @categoryId AND pic_id = @picId',
|
|
{ categoryId, picId },
|
|
);
|
|
|
|
if (!existingCategoryPic.rows[0]) {
|
|
await runner.query(addCategoryPicSql, {
|
|
categoryId,
|
|
picId,
|
|
sort,
|
|
});
|
|
}
|
|
|
|
await runner.query(touchSql, { categoryId });
|
|
|
|
return {
|
|
id: categoryId,
|
|
fullHash,
|
|
thumbW: processed.thumb.w,
|
|
thumbH: processed.thumb.h,
|
|
thumbHash: processed.thumb.hash,
|
|
smallThumbW: processed.smallThumb.w,
|
|
smallThumbH: processed.smallThumb.h,
|
|
};
|
|
};
|
|
|
|
if (tx) {
|
|
return runAdd(makeRunner(provider, tx));
|
|
}
|
|
if (typeof provider?.transaction === 'function') {
|
|
return provider.transaction(runAdd);
|
|
}
|
|
return runAdd(makeRunner(provider));
|
|
};
|
|
}
|