72 lines
1.3 KiB
TypeScript
72 lines
1.3 KiB
TypeScript
import { PrimaryColumn, Entity, Index, JoinColumn, Column, OneToOne } from 'typeorm';
|
|
import { id } from '../id';
|
|
import { Note } from './note';
|
|
import { User } from './user';
|
|
|
|
@Entity()
|
|
export class Poll {
|
|
@PrimaryColumn(id())
|
|
public noteId: Note['id'];
|
|
|
|
@OneToOne(type => Note, {
|
|
onDelete: 'CASCADE'
|
|
})
|
|
@JoinColumn()
|
|
public note: Note | null;
|
|
|
|
@Column('timestamp with time zone', {
|
|
nullable: true
|
|
})
|
|
public expiresAt: Date | null;
|
|
|
|
@Column('boolean')
|
|
public multiple: boolean;
|
|
|
|
@Column('varchar', {
|
|
length: 128, array: true, default: '{}'
|
|
})
|
|
public choices: string[];
|
|
|
|
@Column('integer', {
|
|
array: true,
|
|
})
|
|
public votes: number[];
|
|
|
|
//#region Denormalized fields
|
|
@Column('enum', {
|
|
enum: ['public', 'home', 'followers', 'specified'],
|
|
comment: '[Denormalized]'
|
|
})
|
|
public noteVisibility: 'public' | 'home' | 'followers' | 'specified';
|
|
|
|
@Index()
|
|
@Column({
|
|
...id(),
|
|
comment: '[Denormalized]'
|
|
})
|
|
public userId: User['id'];
|
|
|
|
@Index()
|
|
@Column('varchar', {
|
|
length: 128, nullable: true,
|
|
comment: '[Denormalized]'
|
|
})
|
|
public userHost: string | null;
|
|
//#endregion
|
|
|
|
constructor(data: Partial<Poll>) {
|
|
if (data == null) return;
|
|
|
|
for (const [k, v] of Object.entries(data)) {
|
|
(this as any)[k] = v;
|
|
}
|
|
}
|
|
}
|
|
|
|
export type IPoll = {
|
|
choices: string[];
|
|
votes?: number[];
|
|
multiple: boolean;
|
|
expiresAt: Date | null;
|
|
};
|