You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
2.3 KiB
110 lines
2.3 KiB
import { randomBytes } from "crypto";
|
|
import { promisify } from "util";
|
|
import DiscordBot from "./services/DiscordBot";
|
|
import Service from "./services/Service";
|
|
import TorrentClientIpc from "./services/TorrentClientIpc";
|
|
import WebServer from "./services/WebServer";
|
|
import { User, RegisterToken } from "./database/entities";
|
|
import Database from "./services/Database";
|
|
|
|
/**
|
|
* @TODO Not sure where to put this yet... here's fine for now
|
|
*/
|
|
async function createRegisterToken() {
|
|
let randomBytesPromise = promisify(randomBytes);
|
|
let token = new RegisterToken();
|
|
token.token = (await randomBytesPromise(48)).toString("hex");
|
|
await token.save();
|
|
return token;
|
|
}
|
|
|
|
/**
|
|
* The main application class
|
|
*/
|
|
export default class Application
|
|
{
|
|
/**
|
|
* All available services
|
|
*/
|
|
protected services: Service[];
|
|
|
|
/**
|
|
* The database service
|
|
*/
|
|
protected database!: Database;
|
|
|
|
/**
|
|
* The discord bot service
|
|
*/
|
|
protected discord!: DiscordBot;
|
|
|
|
/**
|
|
* The webserver service
|
|
*/
|
|
protected web!: WebServer;
|
|
|
|
/**
|
|
* The torrent client service
|
|
*/
|
|
protected torrent!: TorrentClientIpc;
|
|
|
|
/**
|
|
* Create a new application instance
|
|
*/
|
|
public constructor() {
|
|
this.services = [
|
|
this.database = new Database(this),
|
|
// this.discord = new DiscordBot(this),
|
|
this.web = new WebServer(this),
|
|
// this.torrent = new TorrentClientIpc(this)
|
|
]
|
|
|
|
// this.torrent.logging = false;
|
|
}
|
|
|
|
/**
|
|
* Boot all of the services
|
|
*/
|
|
protected async boot() {
|
|
return Promise.all(this.services.map(service => service.boot()));
|
|
}
|
|
|
|
/**
|
|
* Initialize the application if necessary
|
|
*/
|
|
protected async initialize() {
|
|
let numUsers = await User.count();
|
|
if (numUsers == 0) {
|
|
console.log("Found 0 users");
|
|
let token = await RegisterToken.findOne();
|
|
if (token === undefined) {
|
|
token = await createRegisterToken();
|
|
}
|
|
console.log("First time register with: ", token.token);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Shutdown the application
|
|
*/
|
|
protected shutdown() {
|
|
return Promise.all(this.services.map(service => service.shutdown()));
|
|
}
|
|
|
|
/**
|
|
* Start the application
|
|
*/
|
|
public async start() {
|
|
await this.boot();
|
|
await this.initialize();
|
|
this.services.forEach(service => service.start());
|
|
}
|
|
|
|
/**
|
|
* Quit the application
|
|
*/
|
|
public async quit(code: number = 0) {
|
|
await this.shutdown();
|
|
process.exit(code);
|
|
}
|
|
}
|