-import { AllowNull, Column, CreatedAt, HasMany, Is, Model, Table, UpdatedAt } from 'sequelize-typescript'
+import { Transaction } from 'sequelize'
+import { AllowNull, Column, CreatedAt, Default, HasMany, Is, Model, Table, UpdatedAt } from 'sequelize-typescript'
+import { MServer, MServerFormattable } from '@server/types/models/server'
+import { AttributesOnly } from '@shared/typescript-utils'
import { isHostValid } from '../../helpers/custom-validators/servers'
-import { ActorModel } from '../activitypub/actor'
+import { ActorModel } from '../actor/actor'
import { throwIfNotValid } from '../utils'
+import { ServerBlocklistModel } from './server-blocklist'
@Table({
tableName: 'server',
}
]
})
-export class ServerModel extends Model<ServerModel> {
+export class ServerModel extends Model<Partial<AttributesOnly<ServerModel>>> {
@AllowNull(false)
@Is('Host', value => throwIfNotValid(value, isHostValid, 'valid host'))
@Column
host: string
+ @AllowNull(false)
+ @Default(false)
+ @Column
+ redundancyAllowed: boolean
+
@CreatedAt
createdAt: Date
hooks: true
})
Actors: ActorModel[]
+
+ @HasMany(() => ServerBlocklistModel, {
+ foreignKey: {
+ allowNull: false
+ },
+ onDelete: 'CASCADE'
+ })
+ BlockedBy: ServerBlocklistModel[]
+
+ static load (id: number, transaction?: Transaction): Promise<MServer> {
+ const query = {
+ where: {
+ id
+ },
+ transaction
+ }
+
+ return ServerModel.findOne(query)
+ }
+
+ static loadByHost (host: string): Promise<MServer> {
+ const query = {
+ where: {
+ host
+ }
+ }
+
+ return ServerModel.findOne(query)
+ }
+
+ static async loadOrCreateByHost (host: string) {
+ let server = await ServerModel.loadByHost(host)
+ if (!server) server = await ServerModel.create({ host })
+
+ return server
+ }
+
+ isBlocked () {
+ return this.BlockedBy && this.BlockedBy.length !== 0
+ }
+
+ toFormattedJSON (this: MServerFormattable) {
+ return {
+ host: this.host
+ }
+ }
}