fileserver/src/entities.ts

import {
	Entity,
	Column,
	PrimaryGeneratedColumn,
	ManyToOne,
	OneToMany,
	OneToOne
} from 'typeorm';
import { UserRole } from '../dto';

export enum tfaTypes {
	NONE = 0,
	EMAIL = 1,
	TOTP = 2
}

@Entity()
export class INode {
	@PrimaryGeneratedColumn()
	id: number;
	@Column()
	isFile: boolean;
	@Column()
	name: string;
	@Column({ nullable: true })
	size: number;

	@Column({ nullable: true })
	parentId: number;
	@ManyToOne(() => INode, (node) => node.children)
	parent: Promise<INode>;
	@OneToMany(() => INode, (node) => node.parent)
	children: Promise<INode[]>;

	@Column({ nullable: true })
	ownerId: number;
	@ManyToOne(() => User)
	owner: Promise<User>;
}

@Entity()
export class User {
	@PrimaryGeneratedColumn()
	id: number;
	@Column({ default: false })
	isGitlabUser: boolean;
	@Column()
	name: string;
	@Column()
	password: string;
	@Column({
		type: 'int',
		default: UserRole.DISABLED,
		transformer: {
			from: (db: number): UserRole => db,
			to: (role: UserRole): number => role
		}
	})
	role: UserRole;

	@Column({ nullable: true })
	rootId: number;
	@OneToOne(() => INode)
	root: Promise<INode>;

	@Column({
		type: 'int',
		default: tfaTypes.NONE,
		transformer: {
			from: (db: number): tfaTypes => db,
			to: (type: tfaTypes): number => type
		}
	})
	tfaType: tfaTypes;

	@Column({ nullable: true })
	tfaSecret: string;

	@Column({ nullable: true })
	gitlabAT: string;
	@Column({ nullable: true })
	gitlabRT: string;
}

@Entity()
export class JWTToken {
	@PrimaryGeneratedColumn()
	id: number;

	@Column()
	ownerId: number;

	@Column({ nullable: true })
	exp: number;
}