89 lines
2.0 KiB
TypeScript
89 lines
2.0 KiB
TypeScript
import { AllowNull, BelongsTo, Column, CreatedAt, DataType, ForeignKey, Table, UpdatedAt } from 'sequelize-typescript'
|
|
import { MUserImport } from '@server/types/models/index.js'
|
|
import { SequelizeModel } from '../shared/index.js'
|
|
import { UserModel } from './user.js'
|
|
import type { UserImportResultSummary, UserImportStateType } from '@peertube/peertube-models'
|
|
import { getSort } from '../shared/sort.js'
|
|
import { USER_IMPORT_STATES } from '@server/initializers/constants.js'
|
|
|
|
@Table({
|
|
tableName: 'userImport',
|
|
indexes: [
|
|
{
|
|
fields: [ 'userId' ]
|
|
},
|
|
{
|
|
fields: [ 'filename' ],
|
|
unique: true
|
|
}
|
|
]
|
|
})
|
|
export class UserImportModel extends SequelizeModel<UserImportModel> {
|
|
@CreatedAt
|
|
createdAt: Date
|
|
|
|
@UpdatedAt
|
|
updatedAt: Date
|
|
|
|
@AllowNull(true)
|
|
@Column
|
|
filename: string
|
|
|
|
@AllowNull(false)
|
|
@Column
|
|
state: UserImportStateType
|
|
|
|
@AllowNull(true)
|
|
@Column(DataType.TEXT)
|
|
error: string
|
|
|
|
@AllowNull(true)
|
|
@Column(DataType.JSONB)
|
|
resultSummary: UserImportResultSummary
|
|
|
|
@ForeignKey(() => UserModel)
|
|
@Column
|
|
userId: number
|
|
|
|
@BelongsTo(() => UserModel, {
|
|
foreignKey: {
|
|
allowNull: false
|
|
},
|
|
onDelete: 'CASCADE'
|
|
})
|
|
User: Awaited<UserModel>
|
|
|
|
static load (id: number | string) {
|
|
return UserImportModel.findByPk<MUserImport>(id)
|
|
}
|
|
|
|
static loadLatestByUserId (userId: number) {
|
|
return UserImportModel.findOne<MUserImport>({
|
|
where: {
|
|
userId
|
|
},
|
|
order: getSort('-createdAt')
|
|
})
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
generateAndSetFilename () {
|
|
if (!this.userId) throw new Error('Cannot generate filename without userId')
|
|
if (!this.createdAt) throw new Error('Cannot generate filename without createdAt')
|
|
|
|
this.filename = `user-import-${this.userId}-${this.createdAt.toISOString()}.zip`
|
|
}
|
|
|
|
toFormattedJSON () {
|
|
return {
|
|
id: this.id,
|
|
state: {
|
|
id: this.state,
|
|
label: USER_IMPORT_STATES[this.state]
|
|
},
|
|
createdAt: this.createdAt.toISOString()
|
|
}
|
|
}
|
|
}
|