74 lines
1.8 KiB
TypeScript
74 lines
1.8 KiB
TypeScript
import { format } from "node:util";
|
|
|
|
import type { PrismaClient } from "@prisma/client";
|
|
|
|
import type { BackendBaseClass } from "../backendimpl/base.js";
|
|
import { backendProviders } from "../backendimpl/index.js";
|
|
|
|
type Backend = {
|
|
id: number;
|
|
name: string;
|
|
description: string | null;
|
|
backend: string;
|
|
connectionDetails: string;
|
|
};
|
|
|
|
export async function backendInit(
|
|
backend: Backend,
|
|
backends: Record<number, BackendBaseClass>,
|
|
prisma: PrismaClient,
|
|
logger?: (arg: string) => void,
|
|
errorOut?: (arg: string) => void,
|
|
): Promise<boolean> {
|
|
const log = (...args: string[]) =>
|
|
logger ? logger(format(...args)) : console.log(...args);
|
|
|
|
const error = (...args: string[]) =>
|
|
errorOut ? errorOut(format(...args)) : log(...args);
|
|
|
|
const ourProvider = backendProviders[backend.backend];
|
|
|
|
if (!ourProvider) {
|
|
error(" - Error: Invalid backend recieved!");
|
|
return false;
|
|
}
|
|
|
|
log(" - Initializing backend...");
|
|
|
|
backends[backend.id] = new ourProvider(backend.connectionDetails);
|
|
const ourBackend = backends[backend.id];
|
|
|
|
if (!(await ourBackend.start())) {
|
|
error(" - Error initializing backend!");
|
|
error(" - " + ourBackend.logs.join("\n - "));
|
|
|
|
return false;
|
|
}
|
|
|
|
log(" - Initializing clients...");
|
|
|
|
const clients = await prisma.forwardRule.findMany({
|
|
where: {
|
|
destProviderID: backend.id,
|
|
enabled: true,
|
|
},
|
|
});
|
|
|
|
for (const client of clients) {
|
|
if (client.protocol != "tcp" && client.protocol != "udp") {
|
|
error(
|
|
` - Error: Client with ID of '${client.id}' has an invalid protocol! (must be either TCP or UDP)`,
|
|
);
|
|
continue;
|
|
}
|
|
|
|
ourBackend.addConnection(
|
|
client.sourceIP,
|
|
client.sourcePort,
|
|
client.destPort,
|
|
client.protocol,
|
|
);
|
|
}
|
|
|
|
return true;
|
|
}
|