forked from mirrors/iceshrimp
83 lines
1.4 KiB
TypeScript
83 lines
1.4 KiB
TypeScript
import {
|
|
PrimaryColumn,
|
|
Entity,
|
|
Index,
|
|
JoinColumn,
|
|
Column,
|
|
OneToOne,
|
|
} from "typeorm";
|
|
import { id } from "../id.js";
|
|
import { Note } from "./note.js";
|
|
import type { User } from "./user.js";
|
|
import { noteVisibilities } from "../../types.js";
|
|
|
|
@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: 256,
|
|
array: true,
|
|
default: "{}",
|
|
})
|
|
public choices: string[];
|
|
|
|
@Column("integer", {
|
|
array: true,
|
|
})
|
|
public votes: number[];
|
|
|
|
//#region Denormalized fields
|
|
@Column("enum", {
|
|
enum: noteVisibilities,
|
|
comment: "[Denormalized]",
|
|
})
|
|
public noteVisibility: typeof noteVisibilities[number];
|
|
|
|
@Index()
|
|
@Column({
|
|
...id(),
|
|
comment: "[Denormalized]",
|
|
})
|
|
public userId: User["id"];
|
|
|
|
@Index()
|
|
@Column("varchar", {
|
|
length: 512,
|
|
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;
|
|
};
|