73 lines
1.4 KiB
JavaScript
73 lines
1.4 KiB
JavaScript
// @flow
|
|
import { DataTypes, sequelize } from "../sequelize";
|
|
import { deleteFromS3 } from "../utils/s3";
|
|
|
|
const FileOperation = sequelize.define("file_operations", {
|
|
id: {
|
|
type: DataTypes.UUID,
|
|
defaultValue: DataTypes.UUIDV4,
|
|
primaryKey: true,
|
|
},
|
|
type: {
|
|
type: DataTypes.ENUM("import", "export"),
|
|
allowNull: false,
|
|
},
|
|
state: {
|
|
type: DataTypes.ENUM(
|
|
"creating",
|
|
"uploading",
|
|
"complete",
|
|
"error",
|
|
"expired"
|
|
),
|
|
allowNull: false,
|
|
},
|
|
key: {
|
|
type: DataTypes.STRING,
|
|
},
|
|
url: {
|
|
type: DataTypes.STRING,
|
|
},
|
|
size: {
|
|
type: DataTypes.BIGINT,
|
|
allowNull: false,
|
|
},
|
|
});
|
|
|
|
FileOperation.prototype.expire = async function () {
|
|
this.state = "expired";
|
|
await deleteFromS3(this.key);
|
|
this.save();
|
|
};
|
|
|
|
FileOperation.associate = (models) => {
|
|
FileOperation.belongsTo(models.User, {
|
|
as: "user",
|
|
foreignKey: "userId",
|
|
});
|
|
FileOperation.belongsTo(models.Collection, {
|
|
as: "collection",
|
|
foreignKey: "collectionId",
|
|
});
|
|
FileOperation.belongsTo(models.Team, {
|
|
as: "team",
|
|
foreignKey: "teamId",
|
|
});
|
|
FileOperation.addScope("defaultScope", {
|
|
include: [
|
|
{
|
|
model: models.User,
|
|
as: "user",
|
|
paranoid: false,
|
|
},
|
|
{
|
|
model: models.Collection,
|
|
as: "collection",
|
|
paranoid: false,
|
|
},
|
|
],
|
|
});
|
|
};
|
|
|
|
export default FileOperation;
|