paulmataruso/patchmon.net
1
0
Fork 0
mirror of https://github.com/9technologygroup/patchmon.net.git synced 2025-11-07 15:33:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Script actually downloads on deb systems (i only use deb, sm1 else will have to test the other distros)

Browse Source
This commit is contained in:
AdamT20054
2025-09-21 07:54:43 +01:00
parent d2bf201f1e
commit a96439596d
1 changed files with 75 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

146
backend/src/routes/hostRoutes.js
View File

@@ -14,61 +14,62 @@ const prisma = new PrismaClient();
// Public endpoint to download the agent script // Public endpoint to download the agent script
router.get('/agent/download', async (req, res) => { router.get('/agent/download', async (req, res) => {
try { try {
const { version } = req.query; const {version} = req.query;
let agentVersion;
if (version) {
// Download specific version
agentVersion = await prisma.agent_versions.findUnique({
where: { version }
});
if (!agentVersion) {
return res.status(404).json({ error: 'Agent version not found' });
}
} else {
// Download current version (latest)
agentVersion = await prisma.agent_versions.findFirst({
where: { is_current: true },
orderBy: { created_at: 'desc' }
});
if (!agentVersion) { let agentVersion;
// Fallback to default version
agentVersion = await prisma.agent_versions.findFirst({
where: { is_default: true },
orderBy: { created_at: 'desc' }
});
}
}
if (!agentVersion) {
return res.status(404).json({ error: 'No agent version available' });
}
// Use script content from database if available, otherwise fallback to file
if (agentVersion.script_content) {
res.setHeader('Content-Type', 'application/x-shellscript');
res.setHeader('Content-Disposition', `attachment; filename="patchmon-agent-${agentVersion.version}.sh"`);
res.send(agentVersion.script_content);
} else {
// Fallback to file system
const agentPath = path.join(__dirname, '../../../agents/patchmon-agent.sh');
if (!fs.existsSync(agentPath)) {
return res.status(404).json({ error: 'Agent script not found' });
}
res.setHeader('Content-Type', 'application/x-shellscript'); if (version) {
res.setHeader('Content-Disposition', `attachment; filename="patchmon-agent-${agentVersion.version}.sh"`); // Download specific version
res.sendFile(path.resolve(agentPath)); agentVersion = await prisma.agent_versions.findUnique({
where: {version}
});
if (!agentVersion) {
return res.status(404).json({error: 'Agent version not found'});
}
} else {
// Download current version (latest)
agentVersion = await prisma.agent_versions.findFirst({
where: {is_current: true},
orderBy: {created_at: 'desc'}
});
if (!agentVersion) {
// Fallback to default version
agentVersion = await prisma.agent_versions.findFirst({
where: {is_default: true},
orderBy: {created_at: 'desc'}
});
}
}
if (!agentVersion) {
return res.status(404).json({error: 'No agent version available'});
}
// Use script content from database if available, otherwise fallback to file
if (agentVersion.script_content) {
// Convert Windows line endings to Unix line endings
const scriptContent = agentVersion.script_content.replace(/\r\n/g, '\n').replace(/\r/g, '\n');
res.setHeader('Content-Type', 'application/x-shellscript');
res.setHeader('Content-Disposition', `attachment; filename="patchmon-agent-${agentVersion.version}.sh"`);
res.send(scriptContent);
} else {
// Fallback to file system
const agentPath = path.join(__dirname, '../../../agents/patchmon-agent.sh');
if (!fs.existsSync(agentPath)) {
return res.status(404).json({error: 'Agent script not found'});
}
// Read file and convert line endings
const scriptContent = fs.readFileSync(agentPath, 'utf8').replace(/\r\n/g, '\n').replace(/\r/g, '\n');
res.setHeader('Content-Type', 'application/x-shellscript');
res.setHeader('Content-Disposition', `attachment; filename="patchmon-agent-${agentVersion.version}.sh"`);
res.send(scriptContent);
}
} catch (error) {
console.error('Agent download error:', error);
} }
} catch (error) {
console.error('Agent download error:', error);
res.status(500).json({ error: 'Failed to download agent script' });
}
}); });
// Version check endpoint for agents // Version check endpoint for agents
@@ -829,26 +830,29 @@ router.get('/install', async (req, res) => {
if (!fs.existsSync(scriptPath)) { if (!fs.existsSync(scriptPath)) {
return res.status(404).json({ error: 'Installation script not found' }); return res.status(404).json({ error: 'Installation script not found' });
} }
let script = fs.readFileSync(scriptPath, 'utf8'); let script = fs.readFileSync(scriptPath, 'utf8');
// Get the configured server URL from settings // Convert Windows line endings to Unix line endings
try { script = script.replace(/\r\n/g, '\n').replace(/\r/g, '\n');
const settings = await prisma.settings.findFirst();
if (settings) { // Get the configured server URL from settings
// Replace the default server URL in the script with the configured one try {
script = script.replace( const settings = await prisma.settings.findFirst();
/PATCHMON_URL="[^"]*"/g, if (settings) {
`PATCHMON_URL="${settings.server_url}"` // Replace the default server URL in the script with the configured one
); script = script.replace(
/PATCHMON_URL="[^"]*"/g,
`PATCHMON_URL="${settings.server_url}"`
);
}
} catch (settingsError) {
console.warn('Could not fetch settings, using default server URL:', settingsError.message);
} }
} catch (settingsError) {
console.warn('Could not fetch settings, using default server URL:', settingsError.message); res.setHeader('Content-Type', 'text/plain');
} res.setHeader('Content-Disposition', 'inline; filename="patchmon_install.sh"');
res.send(script);
res.setHeader('Content-Type', 'text/plain');
res.setHeader('Content-Disposition', 'inline; filename="patchmon_install.sh"');
res.send(script);
} catch (error) { } catch (error) {
console.error('Installation script error:', error); console.error('Installation script error:', error);
res.status(500).json({ error: 'Failed to serve installation script' }); res.status(500).json({ error: 'Failed to serve installation script' });