115 lines
2.1 KiB
TypeScript
115 lines
2.1 KiB
TypeScript
import { Transaction } from "sequelize";
|
|
import { v4 as uuidv4 } from "uuid";
|
|
import { AttachmentPreset } from "@shared/types";
|
|
import { Attachment, Event, User } from "@server/models";
|
|
import AttachmentHelper from "@server/models/helpers/AttachmentHelper";
|
|
import FileStorage from "@server/storage/files";
|
|
|
|
type BaseProps = {
|
|
id?: string;
|
|
name: string;
|
|
user: User;
|
|
source?: "import";
|
|
preset: AttachmentPreset;
|
|
ip?: string;
|
|
transaction?: Transaction;
|
|
};
|
|
|
|
type UrlProps = BaseProps & {
|
|
url: string;
|
|
};
|
|
|
|
type BufferProps = BaseProps & {
|
|
buffer: Buffer;
|
|
type: string;
|
|
};
|
|
|
|
type Props = UrlProps | BufferProps;
|
|
|
|
export default async function attachmentCreator({
|
|
id,
|
|
name,
|
|
user,
|
|
source,
|
|
preset,
|
|
ip,
|
|
transaction,
|
|
...rest
|
|
}: Props): Promise<Attachment | undefined> {
|
|
const acl = AttachmentHelper.presetToAcl(preset);
|
|
const key = AttachmentHelper.getKey({
|
|
acl,
|
|
id: uuidv4(),
|
|
name,
|
|
userId: user.id,
|
|
});
|
|
|
|
let attachment;
|
|
|
|
if ("url" in rest) {
|
|
const { url } = rest;
|
|
const res = await FileStorage.storeFromUrl(url, key, acl);
|
|
|
|
if (!res) {
|
|
return;
|
|
}
|
|
attachment = await Attachment.create(
|
|
{
|
|
id,
|
|
key,
|
|
acl,
|
|
size: res.contentLength,
|
|
contentType: res.contentType,
|
|
teamId: user.teamId,
|
|
userId: user.id,
|
|
},
|
|
{
|
|
transaction,
|
|
}
|
|
);
|
|
} else {
|
|
const { buffer, type } = rest;
|
|
await FileStorage.store({
|
|
body: buffer,
|
|
contentType: type,
|
|
contentLength: buffer.length,
|
|
key,
|
|
acl,
|
|
});
|
|
|
|
attachment = await Attachment.create(
|
|
{
|
|
id,
|
|
key,
|
|
acl,
|
|
size: buffer.length,
|
|
contentType: type,
|
|
teamId: user.teamId,
|
|
userId: user.id,
|
|
},
|
|
{
|
|
transaction,
|
|
}
|
|
);
|
|
}
|
|
|
|
await Event.create(
|
|
{
|
|
name: "attachments.create",
|
|
data: {
|
|
name,
|
|
source,
|
|
},
|
|
modelId: attachment.id,
|
|
teamId: user.teamId,
|
|
actorId: user.id,
|
|
ip,
|
|
},
|
|
{
|
|
transaction,
|
|
}
|
|
);
|
|
|
|
return attachment;
|
|
}
|