mirror of
https://github.com/fosrl/pangolin.git
synced 2025-08-19 08:48:34 +02:00
Updated gerbil key handeling
This commit is contained in:
parent
6cee5703b5
commit
7feb21e727
5 changed files with 25 additions and 17 deletions
|
@ -20,9 +20,10 @@ export async function ensureActions() {
|
||||||
|
|
||||||
// Add new actions
|
// Add new actions
|
||||||
for (const actionId of actionsToAdd) {
|
for (const actionId of actionsToAdd) {
|
||||||
|
logger.debug(`Adding action: ${actionId}`);
|
||||||
await db.insert(actions).values({ actionId }).execute();
|
await db.insert(actions).values({ actionId }).execute();
|
||||||
// Add new actions to the Default role
|
// Add new actions to the Default role
|
||||||
if (defaultRoles.length === 0) {
|
if (defaultRoles.length != 0) {
|
||||||
await db.insert(roleActions)
|
await db.insert(roleActions)
|
||||||
.values(defaultRoles.map(role => ({ roleId: role.roleId!, actionId, orgId: role.orgId! })))
|
.values(defaultRoles.map(role => ({ roleId: role.roleId!, actionId, orgId: role.orgId! })))
|
||||||
.execute();
|
.execute();
|
||||||
|
@ -31,6 +32,7 @@ export async function ensureActions() {
|
||||||
|
|
||||||
// Remove deprecated actions
|
// Remove deprecated actions
|
||||||
if (actionsToRemove.length > 0) {
|
if (actionsToRemove.length > 0) {
|
||||||
|
logger.debug(`Removing actions: ${actionsToRemove.join(', ')}`);
|
||||||
await db.delete(actions).where(inArray(actions.actionId, actionsToRemove)).execute();
|
await db.delete(actions).where(inArray(actions.actionId, actionsToRemove)).execute();
|
||||||
await db.delete(roleActions).where(inArray(roleActions.actionId, actionsToRemove)).execute();
|
await db.delete(roleActions).where(inArray(roleActions.actionId, actionsToRemove)).execute();
|
||||||
}
|
}
|
||||||
|
|
|
@ -6,7 +6,13 @@ import { eq, and } from "drizzle-orm";
|
||||||
import { __DIRNAME } from "@server/config";
|
import { __DIRNAME } from "@server/config";
|
||||||
|
|
||||||
// Load the names from the names.json file
|
// Load the names from the names.json file
|
||||||
const file = join(__DIRNAME, "names.json");
|
const dev = process.env.ENVIRONMENT !== "prod";
|
||||||
|
let file;
|
||||||
|
if (!dev) {
|
||||||
|
file = join(__DIRNAME, "names.json");
|
||||||
|
} else {
|
||||||
|
file = join(__DIRNAME, "/db/names.json");
|
||||||
|
}
|
||||||
export const names = JSON.parse(readFileSync(file, "utf-8"));
|
export const names = JSON.parse(readFileSync(file, "utf-8"));
|
||||||
|
|
||||||
export async function getUniqueSiteName(orgId: string): Promise<string> {
|
export async function getUniqueSiteName(orgId: string): Promise<string> {
|
||||||
|
|
|
@ -27,7 +27,7 @@ export type GetConfigResponse = {
|
||||||
export async function getConfig(req: Request, res: Response, next: NextFunction): Promise<any> {
|
export async function getConfig(req: Request, res: Response, next: NextFunction): Promise<any> {
|
||||||
try {
|
try {
|
||||||
// Validate request parameters
|
// Validate request parameters
|
||||||
const parsedParams = getConfigSchema.safeParse(req.query);
|
const parsedParams = getConfigSchema.safeParse(req.body);
|
||||||
if (!parsedParams.success) {
|
if (!parsedParams.success) {
|
||||||
return next(
|
return next(
|
||||||
createHttpError(
|
createHttpError(
|
||||||
|
@ -44,9 +44,9 @@ export async function getConfig(req: Request, res: Response, next: NextFunction)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Fetch exit node
|
// Fetch exit node
|
||||||
let exitNode = await db.select().from(exitNodes).where(eq(exitNodes.publicKey, publicKey));
|
let exitNodeQuery = await db.select().from(exitNodes).where(eq(exitNodes.publicKey, publicKey));
|
||||||
|
let exitNode;
|
||||||
if (!exitNode) {
|
if (exitNodeQuery.length === 0) {
|
||||||
const address = await getNextAvailableSubnet();
|
const address = await getNextAvailableSubnet();
|
||||||
const listenPort = await getNextAvailablePort();
|
const listenPort = await getNextAvailablePort();
|
||||||
const subEndpoint = await getUniqueExitNodeEndpointName();
|
const subEndpoint = await getUniqueExitNodeEndpointName();
|
||||||
|
@ -61,6 +61,8 @@ export async function getConfig(req: Request, res: Response, next: NextFunction)
|
||||||
}).returning().execute();
|
}).returning().execute();
|
||||||
|
|
||||||
logger.info(`Created new exit node ${exitNode[0].name} with address ${exitNode[0].address} and port ${exitNode[0].listenPort}`);
|
logger.info(`Created new exit node ${exitNode[0].name} with address ${exitNode[0].address} and port ${exitNode[0].listenPort}`);
|
||||||
|
} else {
|
||||||
|
exitNode = exitNodeQuery;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!exitNode) {
|
if (!exitNode) {
|
||||||
|
@ -98,16 +100,11 @@ export async function getConfig(req: Request, res: Response, next: NextFunction)
|
||||||
peers,
|
peers,
|
||||||
};
|
};
|
||||||
|
|
||||||
return response(res, {
|
logger.debug("Sending config: ", configResponse);
|
||||||
data: configResponse,
|
|
||||||
success: true,
|
|
||||||
error: false,
|
|
||||||
message: "Configuration retrieved successfully",
|
|
||||||
status: HttpCode.OK,
|
|
||||||
});
|
|
||||||
|
|
||||||
|
return res.status(HttpCode.OK).send(configResponse);
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
logger.error('Error from getConfig:', error);
|
logger.error(error);
|
||||||
return next(createHttpError(HttpCode.INTERNAL_SERVER_ERROR, "An error occurred..."));
|
return next(createHttpError(HttpCode.INTERNAL_SERVER_ERROR, "An error occurred..."));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -119,10 +116,13 @@ async function getNextAvailableSubnet(): Promise<string> {
|
||||||
}).from(exitNodes);
|
}).from(exitNodes);
|
||||||
|
|
||||||
const addresses = existingAddresses.map(a => a.address);
|
const addresses = existingAddresses.map(a => a.address);
|
||||||
const subnet = findNextAvailableCidr(addresses, config.gerbil.block_size, config.gerbil.subnet_group);
|
let subnet = findNextAvailableCidr(addresses, config.gerbil.block_size, config.gerbil.subnet_group);
|
||||||
if (!subnet) {
|
if (!subnet) {
|
||||||
throw new Error('No available subnets remaining in space');
|
throw new Error('No available subnets remaining in space');
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// replace the last octet with 1
|
||||||
|
subnet = subnet.split('.').slice(0, 3).join('.') + '.1';
|
||||||
return subnet;
|
return subnet;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -17,7 +17,7 @@ internalRouter.get("/traefik-config", traefik.traefikConfigProvider);
|
||||||
const gerbilRouter = Router();
|
const gerbilRouter = Router();
|
||||||
internalRouter.use("/gerbil", gerbilRouter);
|
internalRouter.use("/gerbil", gerbilRouter);
|
||||||
|
|
||||||
gerbilRouter.get("/get-config", gerbil.getConfig);
|
gerbilRouter.post("/get-config", gerbil.getConfig);
|
||||||
gerbilRouter.post("/receive-bandwidth", gerbil.receiveBandwidth);
|
gerbilRouter.post("/receive-bandwidth", gerbil.receiveBandwidth);
|
||||||
|
|
||||||
// Badger routes
|
// Badger routes
|
||||||
|
|
|
@ -75,7 +75,7 @@ export async function createResource(req: Request, res: Response, next: NextFunc
|
||||||
}
|
}
|
||||||
|
|
||||||
// Generate a unique resourceId
|
// Generate a unique resourceId
|
||||||
const fullDomain = `${subdomain}.${org[0].orgId}.${org[0].domain}`;
|
const fullDomain = `${subdomain}.${org[0].domain}`;
|
||||||
|
|
||||||
// Create new resource in the database
|
// Create new resource in the database
|
||||||
const newResource = await db.insert(resources).values({
|
const newResource = await db.insert(resources).values({
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue