pangolin_mirror/pangolin
1
0
Fork 0
mirror of https://github.com/fosrl/pangolin.git synced 2026-03-08 03:36:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor files and add func to private traefik config generator file

Browse Source
This commit is contained in:
Pallavi Kumari
2025-11-02 21:16:50 +05:30
committed by Owen Schwartz
parent 096a2bfa10
commit 795302a351
5 changed files with 220 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server/index.ts
View File

@@ -23,7 +23,7 @@ import { initCleanup } from "#dynamic/cleanup";
import license from "#dynamic/license/license"; import license from "#dynamic/license/license";
import { initLogCleanupInterval } from "@server/lib/cleanupLogs"; import { initLogCleanupInterval } from "@server/lib/cleanupLogs";
import { fetchServerIp } from "@server/lib/serverIpService"; import { fetchServerIp } from "@server/lib/serverIpService";
import { startMaintenanceServer } from "./lib/traefik/maintenance-server.js"; import { startMaintenanceServer } from "./maintenance/maintenance-server.js";
async function startServers() { async function startServers() {
await setHostMeta(); await setHostMeta();

103
server/lib/traefik/getTraefikConfig.ts
View File

@@ -19,109 +19,6 @@ import { sanitize, validatePathRewriteConfig } from "./utils";
const redirectHttpsMiddlewareName = "redirect-to-https"; const redirectHttpsMiddlewareName = "redirect-to-https";
const badgerMiddlewareName = "badger"; const badgerMiddlewareName = "badger";
function escapeHtml(text: string): string {
const map: Record<string, string> = {
'&': '&amp;',
'<': '&lt;',
'>': '&gt;',
'"': '&quot;',
"'": '&#039;'
};
return text.replace(/[&<>"']/g, (char) => map[char]);
}
export function generateMaintenanceHTML(
title: string | null,
message: string | null,
estimatedTime: string | null
): string {
const safeTitle = escapeHtml(title || 'Service Temporarily Unavailable');
const safeMessage = escapeHtml(message || 'We are currently experiencing technical difficulties. Please check back soon.');
const safeEstimatedTime = estimatedTime ? escapeHtml(estimatedTime) : null;
return `<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<meta name="robots" content="noindex, nofollow">
<title>${safeTitle}</title>
<style>
* { margin: 0; padding: 0; box-sizing: border-box; }
body {
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
display: flex;
justify-content: center;
align-items: center;
min-height: 100vh;
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
color: #fff;
padding: 1rem;
line-height: 1.6;
}
.container {
text-align: center;
padding: 3rem 2rem;
max-width: 600px;
background: rgba(255, 255, 255, 0.1);
backdrop-filter: blur(10px);
border-radius: 20px;
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.1);
border: 1px solid rgba(255, 255, 255, 0.2);
}
.icon {
font-size: 4rem;
margin-bottom: 1.5rem;
animation: pulse 2s ease-in-out infinite;
}
@keyframes pulse {
0%, 100% { transform: scale(1); }
50% { transform: scale(1.1); }
}
h1 {
font-size: 2.5rem;
margin-bottom: 1rem;
font-weight: 700;
line-height: 1.2;
}
.message {
font-size: 1.2rem;
margin-bottom: 1rem;
opacity: 0.95;
}
.time {
margin-top: 1.5rem;
padding: 1rem;
background: rgba(255, 255, 255, 0.15);
border-radius: 10px;
font-size: 1rem;
font-weight: 500;
}
@media (max-width: 640px) {
h1 { font-size: 2rem; }
.message { font-size: 1rem; }
.container { padding: 2rem 1.5rem; }
}
</style>
</head>
<body>
<div class="container">
<div class="icon">🔧</div>
<h1>${safeTitle}</h1>
<p class="message">${safeMessage}</p>
${safeEstimatedTime ?
`<div class="time">
<strong>Estimated completion:</strong><br>
${safeEstimatedTime}
</div>`
: ''}
</div>
</body>
</html>`;
}
export async function getTraefikConfig( export async function getTraefikConfig(
exitNodeId: number, exitNodeId: number,
siteTypes: string[], siteTypes: string[],

2
server/lib/traefik/maintenance-server.ts → server/maintenance/maintenance-server.ts
View File

@@ -1,7 +1,7 @@
import express from 'express'; import express from 'express';
import { db, resources } from '@server/db'; import { db, resources } from '@server/db';
import { eq } from 'drizzle-orm'; import { eq } from 'drizzle-orm';
import { generateMaintenanceHTML } from './getTraefikConfig'; import { generateMaintenanceHTML } from './maintenanceUI';
import config from '@server/lib/config'; import config from '@server/lib/config';
import logger from '@server/logger'; import logger from '@server/logger';
import path from 'path'; import path from 'path';

102
server/maintenance/maintenanceUI.ts Normal file
View File

@@ -0,0 +1,102 @@
function escapeHtml(text: string): string {
const map: Record<string, string> = {
'&': '&amp;',
'<': '&lt;',
'>': '&gt;',
'"': '&quot;',
"'": '&#039;'
};
return text.replace(/[&<>"']/g, (char) => map[char]);
}
export function generateMaintenanceHTML(
title: string | null,
message: string | null,
estimatedTime: string | null
): string {
const safeTitle = escapeHtml(title || 'Service Temporarily Unavailable');
const safeMessage = escapeHtml(message || 'We are currently experiencing technical difficulties. Please check back soon.');
const safeEstimatedTime = estimatedTime ? escapeHtml(estimatedTime) : null;
return `<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<meta name="robots" content="noindex, nofollow">
<title>${safeTitle}</title>
<style>
* { margin: 0; padding: 0; box-sizing: border-box; }
body {
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
display: flex;
justify-content: center;
align-items: center;
min-height: 100vh;
background: #fff;
color: #000;
padding: 1rem;
line-height: 1.6;
}
.container {
text-align: center;
padding: 3rem 2rem;
max-width: 600px;
background: rgba(255, 255, 255, 0.1);
backdrop-filter: blur(10px);
border-radius: 20px;
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.1);
border: 1px solid rgba(255, 255, 255, 0.2);
}
.icon {
font-size: 4rem;
margin-bottom: 1.5rem;
animation: pulse 2s ease-in-out infinite;
}
@keyframes pulse {
0%, 100% { transform: scale(1); }
50% { transform: scale(1.1); }
}
h1 {
font-size: 2.5rem;
margin-bottom: 1rem;
font-weight: 700;
line-height: 1.2;
}
.message {
font-size: 1.2rem;
margin-bottom: 1rem;
opacity: 0.95;
}
.time {
margin-top: 1.5rem;
padding: 1rem;
background: rgba(255, 255, 255, 0.15);
border-radius: 10px;
font-size: 1rem;
font-weight: 500;
}
@media (max-width: 640px) {
h1 { font-size: 2rem; }
.message { font-size: 1rem; }
.container { padding: 2rem 1.5rem; }
}
</style>
</head>
<body>
<div class="container">
<div class="icon">🔧</div>
<h1>${safeTitle}</h1>
<p class="message">${safeMessage}</p>
${safeEstimatedTime ?
`<div class="time">
<strong>Estimated completion:</strong><br>
${safeEstimatedTime}
</div>`
: ''}
</div>
</body>
</html>`;
}

142
server/private/lib/traefik/getTraefikConfig.ts
View File

@@ -308,6 +308,97 @@ export async function getTraefikConfig(
config_output.http.services = {}; config_output.http.services = {};
} }
const availableServers = (targets as TargetWithSite[]).filter(
(target: TargetWithSite) => {
if (!target.enabled) return false;
const anySitesOnline = (targets as TargetWithSite[]).some(
(t: TargetWithSite) => t.site.online
);
if (anySitesOnline && !target.site.online) return false;
if (target.site.type === "local" || target.site.type === "wireguard") {
return target.ip && target.port && target.method;
} else if (target.site.type === "newt") {
return target.internalPort && target.method && target.site.subnet;
}
return false;
}
);
const hasHealthyServers = availableServers.length > 0;
let showMaintenancePage = false;
if (resource.maintenanceModeEnabled) {
if (resource.maintenanceModeType === "forced") {
showMaintenancePage = true;
logger.debug(
`Resource ${resource.name} (${fullDomain}) is in FORCED maintenance mode`
);
} else if (resource.maintenanceModeType === "automatic") {
showMaintenancePage = !hasHealthyServers;
if (showMaintenancePage) {
logger.warn(
`Resource ${resource.name} (${fullDomain}) has no healthy servers - showing maintenance page (AUTOMATIC mode)`
);
}
}
}
if (showMaintenancePage) {
const maintenanceServiceName = `${key}-maintenance-service`;
const maintenanceRouterName = `${key}-maintenance-router`;
const maintenancePort = config.getRawConfig().traefik.maintenance_port || 8888;
const entrypointHttp = config.getRawConfig().traefik.http_entrypoint;
const entrypointHttps = config.getRawConfig().traefik.https_entrypoint;
const fullDomain = resource.fullDomain;
const domainParts = fullDomain.split(".");
const wildCard = resource.subdomain
? `*.${domainParts.slice(1).join(".")}`
: fullDomain;
const tls = {
certResolver: resource.domainCertResolver?.trim() ||
config.getRawConfig().traefik.cert_resolver,
...(config.getRawConfig().traefik.prefer_wildcard_cert
? { domains: [{ main: wildCard }] }
: {})
};
const maintenanceHost = config.getRawConfig().traefik?.maintenance_host || 'pangolin';
config_output.http.services[maintenanceServiceName] = {
loadBalancer: {
servers: [{ url: `http://${maintenanceHost}:${maintenancePort}` }],
passHostHeader: true
}
};
const rule = `Host(\`${fullDomain}\`)`;
config_output.http.routers[maintenanceRouterName] = {
entryPoints: [resource.ssl ? entrypointHttps : entrypointHttp],
service: maintenanceServiceName,
rule,
priority: 2000,
...(resource.ssl ? { tls } : {})
};
if (resource.ssl) {
config_output.http.routers[`${maintenanceRouterName}-redirect`] = {
entryPoints: [entrypointHttp],
middlewares: [redirectHttpsMiddlewareName],
service: maintenanceServiceName,
rule,
priority: 2000
};
}
continue;
}
const domainParts = fullDomain.split("."); const domainParts = fullDomain.split(".");
let wildCard; let wildCard;
if (domainParts.length <= 2) { if (domainParts.length <= 2) {
@@ -366,12 +457,12 @@ export async function getTraefikConfig(
certResolver: resolverName, certResolver: resolverName,
...(preferWildcard ...(preferWildcard
? { ? {
domains: [ domains: [
{ {
main: wildCard main: wildCard
} }
] ]
} }
: {}) : {})
}; };
} else { } else {
@@ -624,14 +715,14 @@ export async function getTraefikConfig(
})(), })(),
...(resource.stickySession ...(resource.stickySession
? { ? {
sticky: { sticky: {
cookie: { cookie: {
name: "p_sticky", // TODO: make this configurable via config.yml like other cookies name: "p_sticky", // TODO: make this configurable via config.yml like other cookies
secure: resource.ssl, secure: resource.ssl,
httpOnly: true httpOnly: true
} }
} }
} }
: {}) : {})
} }
}; };
@@ -734,18 +825,18 @@ export async function getTraefikConfig(
})(), })(),
...(resource.proxyProtocol && protocol == "tcp" // proxy protocol only works for tcp ...(resource.proxyProtocol && protocol == "tcp" // proxy protocol only works for tcp
? { ? {
serversTransport: `${ppPrefix}${resource.proxyProtocolVersion || 1}@file` // TODO: does @file here cause issues? serversTransport: `${ppPrefix}${resource.proxyProtocolVersion || 1}@file` // TODO: does @file here cause issues?
} }
: {}), : {}),
...(resource.stickySession ...(resource.stickySession
? { ? {
sticky: { sticky: {
ipStrategy: { ipStrategy: {
depth: 0, depth: 0,
sourcePort: true sourcePort: true
} }
} }
} }
: {}) : {})
} }
}; };
@@ -793,10 +884,9 @@ export async function getTraefikConfig(
loadBalancer: { loadBalancer: {
servers: [ servers: [
{ {
url: `http://${ url: `http://${config.getRawConfig().server
config.getRawConfig().server
.internal_hostname .internal_hostname
}:${config.getRawConfig().server.next_port}` }:${config.getRawConfig().server.next_port}`
} }
] ]
} }