Private
Public Access
1
0

style: Apply rustfmt with stable-only config
Some checks failed
CI Pipeline / Clippy Lints (push) Failing after 0s
CI Pipeline / Rust Unit Tests (push) Failing after 0s
CI Pipeline / Rust Format Check (push) Successful in 4s
CI Pipeline / Frontend Lint & Type Check (push) Failing after 0s
CI Pipeline / Security Audit (push) Failing after 3s
CI Pipeline / Build .deb & Release (push) Has been skipped

- Fixed rustfmt.toml to only use stable options (removed nightly-only)
- Applied cargo fmt --all to fix formatting violations
- Stable options: edition=2021, max_width=100, reorder_imports/modules, match_block_trailing_comma
This commit is contained in:
2026-04-24 15:32:50 +00:00
parent f0fe5f5fd1
commit 5a4d4d583e
44 changed files with 1498 additions and 1040 deletions

View File

@ -22,18 +22,15 @@
use std::time::Duration; use std::time::Duration;
use reqwest::{ use reqwest::{tls::Version, Certificate, ClientBuilder, Identity};
tls::Version,
Certificate, ClientBuilder, Identity,
};
use serde::{de::DeserializeOwned, Serialize}; use serde::{de::DeserializeOwned, Serialize};
use tracing::{debug, instrument}; use tracing::{debug, instrument};
use crate::{ use crate::{
error::AgentClientError, error::AgentClientError,
types::{ types::{
AgentEnvelope, HealthData, PackagesData, PatchesData, SystemInfoData, AgentEnvelope, AgentJobStatus, ApplyPatchesRequest, ApplyPatchesResponse, HealthData,
ApplyPatchesRequest, ApplyPatchesResponse, AgentJobStatus, RollbackResponse, PackagesData, PatchesData, RollbackResponse, SystemInfoData,
}, },
}; };
@ -151,11 +148,7 @@ impl AgentClient {
/// Execute a GET request against `{base_url}/{path}` with optional query /// Execute a GET request against `{base_url}/{path}` with optional query
/// parameters, deserialize the [`AgentEnvelope`], and extract the `data` /// parameters, deserialize the [`AgentEnvelope`], and extract the `data`
/// field — or propagate an [`AgentClientError::ApiError`]. /// field — or propagate an [`AgentClientError::ApiError`].
async fn get<T>( async fn get<T>(&self, path: &str, query: &[(&str, &str)]) -> Result<T, AgentClientError>
&self,
path: &str,
query: &[(&str, &str)],
) -> Result<T, AgentClientError>
where where
T: DeserializeOwned, T: DeserializeOwned,
{ {
@ -190,11 +183,7 @@ impl AgentClient {
// Fallback: use the HTTP status as the error indicator. // Fallback: use the HTTP status as the error indicator.
return Err(AgentClientError::ApiError { return Err(AgentClientError::ApiError {
code: status.as_str().to_string(), code: status.as_str().to_string(),
message: format!( message: format!("Agent returned HTTP {} for {}", status.as_u16(), url),
"Agent returned HTTP {} for {}",
status.as_u16(),
url
),
}); });
} }
@ -220,21 +209,16 @@ impl AgentClient {
/// `GET /api/v1/jobs/{id}` — poll an async agent job for status. /// `GET /api/v1/jobs/{id}` — poll an async agent job for status.
#[instrument(skip(self), fields(base_url = %self.base_url, job_id = %job_id))] #[instrument(skip(self), fields(base_url = %self.base_url, job_id = %job_id))]
pub async fn job_status( pub async fn job_status(&self, job_id: &str) -> Result<AgentJobStatus, AgentClientError> {
&self,
job_id: &str,
) -> Result<AgentJobStatus, AgentClientError> {
self.get(&format!("jobs/{}", job_id), &[]).await self.get(&format!("jobs/{}", job_id), &[]).await
} }
/// `POST /api/v1/jobs/{id}/rollback` — trigger rollback on the agent. /// `POST /api/v1/jobs/{id}/rollback` — trigger rollback on the agent.
#[instrument(skip(self), fields(base_url = %self.base_url, job_id = %job_id))] #[instrument(skip(self), fields(base_url = %self.base_url, job_id = %job_id))]
pub async fn rollback_job( pub async fn rollback_job(&self, job_id: &str) -> Result<RollbackResponse, AgentClientError> {
&self,
job_id: &str,
) -> Result<RollbackResponse, AgentClientError> {
let empty: serde_json::Value = serde_json::json!({}); let empty: serde_json::Value = serde_json::json!({});
self.post(&format!("jobs/{}/rollback", job_id), &empty).await self.post(&format!("jobs/{}/rollback", job_id), &empty)
.await
} }
// -------------------------------------------------------- // --------------------------------------------------------
@ -244,11 +228,7 @@ impl AgentClient {
/// Execute a POST request against `{base_url}/{path}`, serialize `body` as /// Execute a POST request against `{base_url}/{path}`, serialize `body` as
/// JSON, deserialize the [`AgentEnvelope`], and extract the `data` field — /// JSON, deserialize the [`AgentEnvelope`], and extract the `data` field —
/// or propagate an [`AgentClientError::ApiError`]. /// or propagate an [`AgentClientError::ApiError`].
async fn post<Req, Resp>( async fn post<Req, Resp>(&self, path: &str, body: &Req) -> Result<Resp, AgentClientError>
&self,
path: &str,
body: &Req,
) -> Result<Resp, AgentClientError>
where where
Req: Serialize, Req: Serialize,
Resp: DeserializeOwned, Resp: DeserializeOwned,

View File

@ -38,9 +38,6 @@ pub use error::AgentClientError;
/// Response envelope and all data types. /// Response envelope and all data types.
pub use types::{ pub use types::{
AgentEnvelope, AgentErrorBody, AgentEnvelope, AgentErrorBody, HealthData, Package, PackagesData, Patch, PatchesData,
HealthData,
Package, PackagesData,
Patch, PatchesData,
SystemInfoData, SystemInfoData,
}; };

View File

@ -91,10 +91,7 @@ pub fn issue_access_token(
} }
/// Validate and decode an access token using the Ed25519 public key PEM. /// Validate and decode an access token using the Ed25519 public key PEM.
pub fn validate_access_token( pub fn validate_access_token(token: &str, verify_key_pem: &str) -> Result<AccessClaims, JwtError> {
token: &str,
verify_key_pem: &str,
) -> Result<AccessClaims, JwtError> {
let key = DecodingKey::from_ed_pem(verify_key_pem.as_bytes()) let key = DecodingKey::from_ed_pem(verify_key_pem.as_bytes())
.map_err(|e| JwtError::KeyLoad(e.to_string()))?; .map_err(|e| JwtError::KeyLoad(e.to_string()))?;
@ -115,14 +112,12 @@ pub fn validate_access_token(
/// Load the Ed25519 signing key from a PEM file path. /// Load the Ed25519 signing key from a PEM file path.
pub fn load_signing_key(path: &str) -> Result<String, JwtError> { pub fn load_signing_key(path: &str) -> Result<String, JwtError> {
std::fs::read_to_string(path) std::fs::read_to_string(path).map_err(|e| JwtError::KeyLoad(format!("Cannot read {path}: {e}")))
.map_err(|e| JwtError::KeyLoad(format!("Cannot read {path}: {e}")))
} }
/// Load the Ed25519 verification (public) key from a PEM file path. /// Load the Ed25519 verification (public) key from a PEM file path.
pub fn load_verify_key(path: &str) -> Result<String, JwtError> { pub fn load_verify_key(path: &str) -> Result<String, JwtError> {
std::fs::read_to_string(path) std::fs::read_to_string(path).map_err(|e| JwtError::KeyLoad(format!("Cannot read {path}: {e}")))
.map_err(|e| JwtError::KeyLoad(format!("Cannot read {path}: {e}")))
} }
#[cfg(test)] #[cfg(test)]

View File

@ -66,9 +66,9 @@ fn build_totp(username: &str, secret_base32: &str) -> Result<TOTP, TotpError> {
// new(issuer, account_name, algorithm, digits, skew, step, secret) // new(issuer, account_name, algorithm, digits, skew, step, secret)
TOTP::new( TOTP::new(
Algorithm::SHA1, Algorithm::SHA1,
6, // digits 6, // digits
1, // skew 1, // skew
30, // step (seconds) 30, // step (seconds)
secret_bytes, secret_bytes,
Some(ISSUER.to_string()), Some(ISSUER.to_string()),
username.to_string(), username.to_string(),

View File

@ -7,17 +7,15 @@
//! - Parallelism: 1 //! - Parallelism: 1
use argon2::{ use argon2::{
password_hash::{ password_hash::{rand_core::OsRng, PasswordHash, PasswordHasher, PasswordVerifier, SaltString},
rand_core::OsRng, PasswordHash, PasswordHasher, PasswordVerifier, SaltString,
},
Argon2, Params, Version, Argon2, Params, Version,
}; };
use thiserror::Error; use thiserror::Error;
/// Argon2id parameters per spec. /// Argon2id parameters per spec.
const M_COST: u32 = 65536; // 64 MiB const M_COST: u32 = 65536; // 64 MiB
const T_COST: u32 = 3; // 3 iterations const T_COST: u32 = 3; // 3 iterations
const P_COST: u32 = 1; // 1 thread const P_COST: u32 = 1; // 1 thread
#[derive(Debug, Error)] #[derive(Debug, Error)]
pub enum PasswordError { pub enum PasswordError {
@ -33,7 +31,11 @@ pub enum PasswordError {
fn argon2() -> Result<Argon2<'static>, PasswordError> { fn argon2() -> Result<Argon2<'static>, PasswordError> {
let params = Params::new(M_COST, T_COST, P_COST, None) let params = Params::new(M_COST, T_COST, P_COST, None)
.map_err(|e| PasswordError::HashError(e.to_string()))?; .map_err(|e| PasswordError::HashError(e.to_string()))?;
Ok(Argon2::new(argon2::Algorithm::Argon2id, Version::V0x13, params)) Ok(Argon2::new(
argon2::Algorithm::Argon2id,
Version::V0x13,
params,
))
} }
/// Hash a plaintext password using Argon2id with a random salt. /// Hash a plaintext password using Argon2id with a random salt.
@ -54,8 +56,7 @@ pub fn hash_password(password: &str) -> Result<String, PasswordError> {
/// ///
/// Returns `Ok(true)` if the password matches, `Ok(false)` if not. /// Returns `Ok(true)` if the password matches, `Ok(false)` if not.
pub fn verify_password(password: &str, hash: &str) -> Result<bool, PasswordError> { pub fn verify_password(password: &str, hash: &str) -> Result<bool, PasswordError> {
let parsed_hash = let parsed_hash = PasswordHash::new(hash).map_err(|_| PasswordError::InvalidHash)?;
PasswordHash::new(hash).map_err(|_| PasswordError::InvalidHash)?;
let argon2 = argon2()?; let argon2 = argon2()?;

View File

@ -143,11 +143,7 @@ fn forbidden(message: &str) -> Response {
/// ///
/// Inserts `AuthUser` into request extensions on success. /// Inserts `AuthUser` into request extensions on success.
/// Rejects with 401 if token is missing/invalid, 403 if IP is blocked. /// Rejects with 401 if token is missing/invalid, 403 if IP is blocked.
pub async fn require_auth( pub async fn require_auth(auth_config: Arc<AuthConfig>, mut req: Request, next: Next) -> Response {
auth_config: Arc<AuthConfig>,
mut req: Request,
next: Next,
) -> Response {
// IP whitelist check // IP whitelist check
if let Some(ip) = extract_remote_ip(req.headers()) { if let Some(ip) = extract_remote_ip(req.headers()) {
if !auth_config.is_ip_allowed(&ip) { if !auth_config.is_ip_allowed(&ip) {
@ -168,7 +164,7 @@ pub async fn require_auth(
Err(e) => { Err(e) => {
tracing::debug!(error = %e, "JWT validation failed"); tracing::debug!(error = %e, "JWT validation failed");
return unauthorized("Invalid token"); return unauthorized("Invalid token");
} },
}; };
let role = match UserRole::from_str(&claims.role) { let role = match UserRole::from_str(&claims.role) {

View File

@ -123,12 +123,10 @@ pub async fn rotate(
} }
// Revoke old token // Revoke old token
sqlx::query( sqlx::query("UPDATE refresh_tokens SET revoked = TRUE, revoked_at = NOW() WHERE id = $1")
"UPDATE refresh_tokens SET revoked = TRUE, revoked_at = NOW() WHERE id = $1", .bind(stored.id)
) .execute(pool)
.bind(stored.id) .await?;
.execute(pool)
.await?;
// Issue new token // Issue new token
let new_token = issue(pool, stored.user_id, user_agent, ip_address).await?; let new_token = issue(pool, stored.user_id, user_agent, ip_address).await?;
@ -138,10 +136,7 @@ pub async fn rotate(
} }
/// Revoke all refresh tokens for a user (force logout). /// Revoke all refresh tokens for a user (force logout).
pub async fn revoke_all_for_user( pub async fn revoke_all_for_user(pool: &PgPool, user_id: Uuid) -> Result<u64, RefreshError> {
pool: &PgPool,
user_id: Uuid,
) -> Result<u64, RefreshError> {
let result = sqlx::query( let result = sqlx::query(
"UPDATE refresh_tokens SET revoked = TRUE, revoked_at = NOW() WHERE user_id = $1 AND revoked = FALSE", "UPDATE refresh_tokens SET revoked = TRUE, revoked_at = NOW() WHERE user_id = $1 AND revoked = FALSE",
) )
@ -154,10 +149,7 @@ pub async fn revoke_all_for_user(
} }
/// Revoke a single refresh token by its raw value. /// Revoke a single refresh token by its raw value.
pub async fn revoke( pub async fn revoke(pool: &PgPool, raw_token: &str) -> Result<(), RefreshError> {
pool: &PgPool,
raw_token: &str,
) -> Result<(), RefreshError> {
let hash = hex::encode(Sha256::digest(raw_token.as_bytes())); let hash = hex::encode(Sha256::digest(raw_token.as_bytes()));
sqlx::query( sqlx::query(

View File

@ -122,13 +122,12 @@ pub async fn login(
// Prevent timing-based username enumeration // Prevent timing-based username enumeration
let _ = password::hash_password("dummy-timing-fill"); let _ = password::hash_password("dummy-timing-fill");
return Err(SessionError::InvalidCredentials); return Err(SessionError::InvalidCredentials);
} },
}; };
// 2. Verify password // 2. Verify password
let hash = user.password_hash.as_deref().unwrap_or(""); let hash = user.password_hash.as_deref().unwrap_or("");
let valid = password::verify_password(&req.password, hash) let valid = password::verify_password(&req.password, hash).unwrap_or(false);
.unwrap_or(false);
if !valid { if !valid {
tracing::warn!(username = %req.username, "Login failed: invalid password"); tracing::warn!(username = %req.username, "Login failed: invalid password");
@ -146,8 +145,7 @@ pub async fn login(
let code = req.totp_code.as_deref().ok_or(SessionError::MfaRequired)?; let code = req.totp_code.as_deref().ok_or(SessionError::MfaRequired)?;
let secret = user.totp_secret.as_deref().unwrap_or(""); let secret = user.totp_secret.as_deref().unwrap_or("");
let mfa_ok = mfa_totp::verify_code(&user.username, secret, code) let mfa_ok = mfa_totp::verify_code(&user.username, secret, code).unwrap_or(false);
.unwrap_or(false);
if !mfa_ok { if !mfa_ok {
tracing::warn!(username = %req.username, "Login failed: invalid MFA code"); tracing::warn!(username = %req.username, "Login failed: invalid MFA code");
@ -246,19 +244,13 @@ pub async fn refresh_session(
} }
/// Logout: revoke the current refresh token. /// Logout: revoke the current refresh token.
pub async fn logout( pub async fn logout(pool: &PgPool, raw_refresh_token: &str) -> Result<(), SessionError> {
pool: &PgPool,
raw_refresh_token: &str,
) -> Result<(), SessionError> {
refresh::revoke(pool, raw_refresh_token).await?; refresh::revoke(pool, raw_refresh_token).await?;
Ok(()) Ok(())
} }
/// Force-logout: revoke all refresh tokens for a user. /// Force-logout: revoke all refresh tokens for a user.
pub async fn force_logout( pub async fn force_logout(pool: &PgPool, user_id: Uuid) -> Result<u64, SessionError> {
pool: &PgPool,
user_id: Uuid,
) -> Result<u64, SessionError> {
let count = refresh::revoke_all_for_user(pool, user_id).await?; let count = refresh::revoke_all_for_user(pool, user_id).await?;
Ok(count) Ok(count)
} }

View File

@ -13,8 +13,8 @@ use chrono::{DateTime, Duration as ChronoDuration, Utc};
use rand::RngCore; use rand::RngCore;
use rcgen::{ use rcgen::{
BasicConstraints, Certificate, CertificateParams, DistinguishedName, DnType, BasicConstraints, Certificate, CertificateParams, DistinguishedName, DnType,
ExtendedKeyUsagePurpose, Ia5String, IsCa, KeyPair, KeyUsagePurpose, SanType, ExtendedKeyUsagePurpose, Ia5String, IsCa, KeyPair, KeyUsagePurpose, SanType, SerialNumber,
SerialNumber, PKCS_ECDSA_P256_SHA256, PKCS_ECDSA_P256_SHA256,
}; };
use sqlx::{PgPool, Row}; use sqlx::{PgPool, Row};
use time::{Duration as TimeDuration, OffsetDateTime}; use time::{Duration as TimeDuration, OffsetDateTime};
@ -83,10 +83,7 @@ fn chrono_offset_days(days: i64) -> DateTime<Utc> {
/// Build a `CertificateParams` with common fields pre-filled. /// Build a `CertificateParams` with common fields pre-filled.
/// Caller still needs to set `is_ca`, `key_usages`, `extended_key_usages`, and `subject_alt_names`. /// Caller still needs to set `is_ca`, `key_usages`, `extended_key_usages`, and `subject_alt_names`.
fn base_params( fn base_params(cn: &str, validity_days: i64) -> Result<(CertificateParams, String, DateTime<Utc>)> {
cn: &str,
validity_days: i64,
) -> Result<(CertificateParams, String, DateTime<Utc>)> {
let (serial, serial_hex) = make_serial(); let (serial, serial_hex) = make_serial();
let expires_at = chrono_offset_days(validity_days); let expires_at = chrono_offset_days(validity_days);
@ -144,8 +141,7 @@ impl CertAuthority {
.context("read ca.crt")?; .context("read ca.crt")?;
// Validate that both PEMs parse without error. // Validate that both PEMs parse without error.
KeyPair::from_pem(&ca_key_pem) KeyPair::from_pem(&ca_key_pem).context("parse CA private-key PEM")?;
.context("parse CA private-key PEM")?;
CertificateParams::from_ca_cert_pem(&ca_cert_pem) CertificateParams::from_ca_cert_pem(&ca_cert_pem)
.context("parse CA certificate PEM")?; .context("parse CA certificate PEM")?;
@ -166,8 +162,8 @@ impl CertAuthority {
.await .await
.context("create CA directory")?; .context("create CA directory")?;
let ca_key = KeyPair::generate_for(&PKCS_ECDSA_P256_SHA256) let ca_key =
.context("generate CA key pair")?; KeyPair::generate_for(&PKCS_ECDSA_P256_SHA256).context("generate CA key pair")?;
let (serial, serial_hex) = make_serial(); let (serial, serial_hex) = make_serial();
let expires_at = chrono_offset_days(365 * 10); let expires_at = chrono_offset_days(365 * 10);
@ -177,20 +173,18 @@ impl CertAuthority {
params.not_after = odt_offset_days(365 * 10); params.not_after = odt_offset_days(365 * 10);
params.serial_number = Some(serial); params.serial_number = Some(serial);
params.is_ca = IsCa::Ca(BasicConstraints::Unconstrained); params.is_ca = IsCa::Ca(BasicConstraints::Unconstrained);
params.key_usages = vec![ params.key_usages = vec![KeyUsagePurpose::KeyCertSign, KeyUsagePurpose::CrlSign];
KeyUsagePurpose::KeyCertSign,
KeyUsagePurpose::CrlSign,
];
let mut dn = DistinguishedName::new(); let mut dn = DistinguishedName::new();
dn.push(DnType::CommonName, "Patch Manager Root CA"); dn.push(DnType::CommonName, "Patch Manager Root CA");
dn.push(DnType::OrganizationName, "Patch Manager"); dn.push(DnType::OrganizationName, "Patch Manager");
params.distinguished_name = dn; params.distinguished_name = dn;
let ca_cert_obj = params.self_signed(&ca_key) let ca_cert_obj = params
.self_signed(&ca_key)
.context("self-sign CA certificate")?; .context("self-sign CA certificate")?;
let ca_cert_pem = ca_cert_obj.pem(); let ca_cert_pem = ca_cert_obj.pem();
let ca_key_pem = ca_key.serialize_pem(); let ca_key_pem = ca_key.serialize_pem();
write_protected(&key_path, &ca_key_pem) write_protected(&key_path, &ca_key_pem)
.await .await
@ -256,8 +250,8 @@ impl CertAuthority {
) -> Result<IssuedCert> { ) -> Result<IssuedCert> {
tracing::info!(host_id = %host_id, hostname, "Issuing mTLS client certificate"); tracing::info!(host_id = %host_id, hostname, "Issuing mTLS client certificate");
let key = KeyPair::generate_for(&PKCS_ECDSA_P256_SHA256) let key =
.context("generate client key pair")?; KeyPair::generate_for(&PKCS_ECDSA_P256_SHA256).context("generate client key pair")?;
let (mut params, serial_hex, expires_at) = base_params(hostname, 365)?; let (mut params, serial_hex, expires_at) = base_params(hostname, 365)?;
params.is_ca = IsCa::ExplicitNoCa; params.is_ca = IsCa::ExplicitNoCa;
@ -270,7 +264,7 @@ impl CertAuthority {
.context("sign client cert with CA")?; .context("sign client cert with CA")?;
let cert_pem = cert.pem(); let cert_pem = cert.pem();
let key_pem = key.serialize_pem(); let key_pem = key.serialize_pem();
sqlx::query( sqlx::query(
"INSERT INTO certificates \ "INSERT INTO certificates \
@ -294,7 +288,12 @@ impl CertAuthority {
"Client certificate issued successfully" "Client certificate issued successfully"
); );
Ok(IssuedCert { cert_pem, key_pem, serial_number: serial_hex, expires_at }) Ok(IssuedCert {
cert_pem,
key_pem,
serial_number: serial_hex,
expires_at,
})
} }
/// Revoke a certificate by database ID. /// Revoke a certificate by database ID.
@ -328,18 +327,16 @@ impl CertAuthority {
tracing::info!(cert_id = %cert_id, "Renewing certificate"); tracing::info!(cert_id = %cert_id, "Renewing certificate");
// Fetch the existing cert's host_id and common_name. // Fetch the existing cert's host_id and common_name.
let row = sqlx::query( let row = sqlx::query("SELECT host_id, common_name FROM certificates WHERE id = $1")
"SELECT host_id, common_name FROM certificates WHERE id = $1", .bind(cert_id)
) .fetch_one(db)
.bind(cert_id) .await
.fetch_one(db) .context("fetch certificate for renewal")?;
.await
.context("fetch certificate for renewal")?;
let host_id: Uuid = row.try_get("host_id") let host_id: Uuid = row
.try_get("host_id")
.context("certificate has no host_id (cannot renew root CA)")?; .context("certificate has no host_id (cannot renew root CA)")?;
let common_name: String = row.try_get("common_name") let common_name: String = row.try_get("common_name").context("fetch common_name")?;
.context("fetch common_name")?;
// Revoke the old cert first. // Revoke the old cert first.
self.revoke_cert(cert_id, db).await?; self.revoke_cert(cert_id, db).await?;
@ -364,14 +361,11 @@ impl CertAuthority {
/// ///
/// Returns `(cert_pem, key_pem)`. This certificate is **not** stored in the /// Returns `(cert_pem, key_pem)`. This certificate is **not** stored in the
/// database; it is intended for runtime use only. /// database; it is intended for runtime use only.
pub async fn issue_web_tls_cert( pub async fn issue_web_tls_cert(&self, hostname: &str) -> Result<(String, String)> {
&self,
hostname: &str,
) -> Result<(String, String)> {
tracing::info!(hostname, "Issuing web TLS certificate"); tracing::info!(hostname, "Issuing web TLS certificate");
let key = KeyPair::generate_for(&PKCS_ECDSA_P256_SHA256) let key =
.context("generate web TLS key pair")?; KeyPair::generate_for(&PKCS_ECDSA_P256_SHA256).context("generate web TLS key pair")?;
let (mut params, serial_hex, expires_at) = base_params(hostname, 365)?; let (mut params, serial_hex, expires_at) = base_params(hostname, 365)?;
params.is_ca = IsCa::ExplicitNoCa; params.is_ca = IsCa::ExplicitNoCa;
@ -387,7 +381,7 @@ impl CertAuthority {
.context("sign web TLS cert with CA")?; .context("sign web TLS cert with CA")?;
let cert_pem = cert.pem(); let cert_pem = cert.pem();
let key_pem = key.serialize_pem(); let key_pem = key.serialize_pem();
tracing::info!( tracing::info!(
hostname, hostname,
@ -408,8 +402,8 @@ impl CertAuthority {
/// The returned `Certificate` is used solely as an issuer reference when /// The returned `Certificate` is used solely as an issuer reference when
/// signing leaf certificates; it is never distributed directly. /// signing leaf certificates; it is never distributed directly.
fn ca_objects(&self) -> Result<(KeyPair, Certificate)> { fn ca_objects(&self) -> Result<(KeyPair, Certificate)> {
let key = KeyPair::from_pem(&self.ca_key_pem) let key =
.context("reconstruct CA key pair from PEM")?; KeyPair::from_pem(&self.ca_key_pem).context("reconstruct CA key pair from PEM")?;
let params = CertificateParams::from_ca_cert_pem(&self.ca_cert_pem) let params = CertificateParams::from_ca_cert_pem(&self.ca_cert_pem)
.context("reconstruct CA params from PEM")?; .context("reconstruct CA params from PEM")?;
let cert = params let cert = params

View File

@ -101,8 +101,15 @@ pub async fn log_event(
request_id: Option<&str>, request_id: Option<&str>,
) { ) {
let result = write_audit_row( let result = write_audit_row(
pool, action, actor_user_id, actor_username, pool,
target_type, target_id, details, ip_address, request_id, action,
actor_user_id,
actor_username,
target_type,
target_id,
details,
ip_address,
request_id,
) )
.await; .await;
@ -123,11 +130,10 @@ async fn write_audit_row(
request_id: Option<&str>, request_id: Option<&str>,
) -> Result<(), sqlx::Error> { ) -> Result<(), sqlx::Error> {
// Fetch previous hash for chain // Fetch previous hash for chain
let prev_hash: Option<String> = sqlx::query_scalar( let prev_hash: Option<String> =
"SELECT row_hash FROM audit_log ORDER BY id DESC LIMIT 1", sqlx::query_scalar("SELECT row_hash FROM audit_log ORDER BY id DESC LIMIT 1")
) .fetch_optional(pool)
.fetch_optional(pool) .await?;
.await?;
let prev = prev_hash.unwrap_or_default(); let prev = prev_hash.unwrap_or_default();
let now = chrono::Utc::now().to_rfc3339(); let now = chrono::Utc::now().to_rfc3339();
@ -245,7 +251,7 @@ pub async fn verify_integrity(pool: &PgPool) -> IntegrityResult {
rows_checked: 0, rows_checked: 0,
errors: vec![], errors: vec![],
}; };
} },
}; };
let mut errors = Vec::new(); let mut errors = Vec::new();
@ -273,10 +279,7 @@ pub async fn verify_integrity(pool: &PgPool) -> IntegrityResult {
.unwrap_or_default(); .unwrap_or_default();
let ip_str = row.ip_address.as_deref().unwrap_or(""); let ip_str = row.ip_address.as_deref().unwrap_or("");
let rid = row.request_id.as_deref().unwrap_or(""); let rid = row.request_id.as_deref().unwrap_or("");
let created_str = row let created_str = row.created_at.map(|c| c.to_rfc3339()).unwrap_or_default();
.created_at
.map(|c| c.to_rfc3339())
.unwrap_or_default();
let mut hasher = Sha256::new(); let mut hasher = Sha256::new();
hasher.update(row.prev_hash.as_bytes()); hasher.update(row.prev_hash.as_bytes());

View File

@ -1,5 +1,5 @@
use serde::{Deserialize, Serialize};
use config::{Config, ConfigError, Environment, File}; use config::{Config, ConfigError, Environment, File};
use serde::{Deserialize, Serialize};
/// Top-level application configuration. /// Top-level application configuration.
#[derive(Debug, Clone, Deserialize, Serialize)] #[derive(Debug, Clone, Deserialize, Serialize)]

View File

@ -1,6 +1,6 @@
use crate::config::DatabaseConfig;
use sqlx::postgres::{PgPool, PgPoolOptions}; use sqlx::postgres::{PgPool, PgPoolOptions};
use std::time::Duration; use std::time::Duration;
use crate::config::DatabaseConfig;
/// Initialize and return a PostgreSQL connection pool. /// Initialize and return a PostgreSQL connection pool.
pub async fn init_pool(cfg: &DatabaseConfig) -> Result<PgPool, sqlx::Error> { pub async fn init_pool(cfg: &DatabaseConfig) -> Result<PgPool, sqlx::Error> {
@ -59,11 +59,9 @@ pub async fn run_migrations(pool: &PgPool) -> Result<(), sqlx::migrate::MigrateE
/// Check that the database schema is at the expected version. /// Check that the database schema is at the expected version.
/// Used by the worker to wait until migrations have been applied. /// Used by the worker to wait until migrations have been applied.
pub async fn check_schema_version(pool: &PgPool) -> Result<i64, sqlx::Error> { pub async fn check_schema_version(pool: &PgPool) -> Result<i64, sqlx::Error> {
let row: (i64,) = sqlx::query_as( let row: (i64,) = sqlx::query_as("SELECT COUNT(*) FROM _sqlx_migrations WHERE success = true")
"SELECT COUNT(*) FROM _sqlx_migrations WHERE success = true", .fetch_one(pool)
) .await?;
.fetch_one(pool)
.await?;
Ok(row.0) Ok(row.0)
} }

View File

@ -86,9 +86,11 @@ impl IntoResponse for AppError {
AppError::Forbidden(msg) => (StatusCode::FORBIDDEN, "forbidden", msg.clone()), AppError::Forbidden(msg) => (StatusCode::FORBIDDEN, "forbidden", msg.clone()),
AppError::BadRequest(msg) => (StatusCode::BAD_REQUEST, "bad_request", msg.clone()), AppError::BadRequest(msg) => (StatusCode::BAD_REQUEST, "bad_request", msg.clone()),
AppError::Conflict(msg) => (StatusCode::CONFLICT, "conflict", msg.clone()), AppError::Conflict(msg) => (StatusCode::CONFLICT, "conflict", msg.clone()),
AppError::UnprocessableEntity(msg) => { AppError::UnprocessableEntity(msg) => (
(StatusCode::UNPROCESSABLE_ENTITY, "unprocessable_entity", msg.clone()) StatusCode::UNPROCESSABLE_ENTITY,
} "unprocessable_entity",
msg.clone(),
),
AppError::Database(e) => { AppError::Database(e) => {
tracing::error!(error = %e, "Database error"); tracing::error!(error = %e, "Database error");
( (
@ -96,7 +98,7 @@ impl IntoResponse for AppError {
"internal_error", "internal_error",
"An internal error occurred".to_string(), "An internal error occurred".to_string(),
) )
} },
AppError::Internal(e) => { AppError::Internal(e) => {
tracing::error!(error = %e, "Internal error"); tracing::error!(error = %e, "Internal error");
( (
@ -104,7 +106,7 @@ impl IntoResponse for AppError {
"internal_error", "internal_error",
"An internal error occurred".to_string(), "An internal error occurred".to_string(),
) )
} },
AppError::Config(msg) => { AppError::Config(msg) => {
tracing::error!(error = %msg, "Configuration error"); tracing::error!(error = %msg, "Configuration error");
( (
@ -112,7 +114,7 @@ impl IntoResponse for AppError {
"config_error", "config_error",
"Server configuration error".to_string(), "Server configuration error".to_string(),
) )
} },
}; };
let body = ErrorResponse::new(code, message); let body = ErrorResponse::new(code, message);

View File

@ -1,20 +1,19 @@
pub mod audit;
pub mod config; pub mod config;
pub mod db; pub mod db;
pub mod error; pub mod error;
pub mod logging; pub mod logging;
pub mod models; pub mod models;
pub mod audit;
pub mod request_id; pub mod request_id;
// Re-export commonly used types // Re-export commonly used types
pub use error::{AppError, ErrorResponse};
pub use config::AppConfig; pub use config::AppConfig;
pub use error::{AppError, ErrorResponse};
pub use models::{ pub use models::{
Host, HostSummary, HostHealthStatus, CreateHostRequest, AuthProvider, CreateGroupRequest, CreateHostRequest, CreateUserRequest, DiscoveryCidrRequest,
Group, CreateGroupRequest, UpdateGroupRequest, DiscoveryResult, Group, Host, HostHealthStatus, HostSummary, RegisterDiscoveredRequest,
User, UserRole as DbUserRole, AuthProvider, CreateUserRequest, UpdateUserRequest, UpdateGroupRequest, UpdateUserRequest, User, UserRole as DbUserRole,
DiscoveryResult, DiscoveryCidrRequest, RegisterDiscoveredRequest,
}; };
// Re-export audit integrity types // Re-export audit integrity types
pub use audit::{verify_integrity, IntegrityResult, IntegrityError}; pub use audit::{verify_integrity, IntegrityError, IntegrityResult};

View File

@ -1,5 +1,5 @@
use tracing_subscriber::{fmt, layer::SubscriberExt, util::SubscriberInitExt, EnvFilter};
use crate::config::LoggingConfig; use crate::config::LoggingConfig;
use tracing_subscriber::{fmt, layer::SubscriberExt, util::SubscriberInitExt, EnvFilter};
/// Initialize the global tracing subscriber. /// Initialize the global tracing subscriber.
/// ///
@ -10,8 +10,7 @@ use crate::config::LoggingConfig;
/// Log level is controlled by `cfg.level` (e.g. `"info"`, `"debug"`). /// Log level is controlled by `cfg.level` (e.g. `"info"`, `"debug"`).
/// The `RUST_LOG` environment variable overrides `cfg.level`. /// The `RUST_LOG` environment variable overrides `cfg.level`.
pub fn init(cfg: &LoggingConfig) { pub fn init(cfg: &LoggingConfig) {
let filter = EnvFilter::try_from_default_env() let filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new(&cfg.level));
.unwrap_or_else(|_| EnvFilter::new(&cfg.level));
match cfg.format.as_str() { match cfg.format.as_str() {
"json" => { "json" => {
@ -19,13 +18,13 @@ pub fn init(cfg: &LoggingConfig) {
.with(filter) .with(filter)
.with(fmt::layer().json().with_current_span(true)) .with(fmt::layer().json().with_current_span(true))
.init(); .init();
} },
_ => { _ => {
tracing_subscriber::registry() tracing_subscriber::registry()
.with(filter) .with(filter)
.with(fmt::layer().pretty()) .with(fmt::layer().pretty())
.init(); .init();
} },
} }
tracing::info!(format = %cfg.format, level = %cfg.level, "Logging initialized"); tracing::info!(format = %cfg.format, level = %cfg.level, "Logging initialized");

View File

@ -211,11 +211,11 @@ pub enum JobStatus {
impl std::fmt::Display for JobStatus { impl std::fmt::Display for JobStatus {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self { match self {
Self::Queued => write!(f, "queued"), Self::Queued => write!(f, "queued"),
Self::Pending => write!(f, "pending"), Self::Pending => write!(f, "pending"),
Self::Running => write!(f, "running"), Self::Running => write!(f, "running"),
Self::Succeeded => write!(f, "succeeded"), Self::Succeeded => write!(f, "succeeded"),
Self::Failed => write!(f, "failed"), Self::Failed => write!(f, "failed"),
Self::Cancelled => write!(f, "cancelled"), Self::Cancelled => write!(f, "cancelled"),
} }
} }
@ -321,9 +321,9 @@ pub enum WindowRecurrence {
impl std::fmt::Display for WindowRecurrence { impl std::fmt::Display for WindowRecurrence {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self { match self {
Self::Once => write!(f, "once"), Self::Once => write!(f, "once"),
Self::Daily => write!(f, "daily"), Self::Daily => write!(f, "daily"),
Self::Weekly => write!(f, "weekly"), Self::Weekly => write!(f, "weekly"),
Self::Monthly => write!(f, "monthly"), Self::Monthly => write!(f, "monthly"),
} }
} }

View File

@ -1,9 +1,4 @@
use axum::{ use axum::{extract::Request, http::HeaderValue, middleware::Next, response::Response};
extract::Request,
http::HeaderValue,
middleware::Next,
response::Response,
};
use ulid::Ulid; use ulid::Ulid;
/// HTTP header name for request correlation IDs. /// HTTP header name for request correlation IDs.

View File

@ -4,10 +4,7 @@ use crate::{ReportParams, ReportType};
use anyhow::Context; use anyhow::Context;
/// Generate a CSV report and return the raw bytes. /// Generate a CSV report and return the raw bytes.
pub async fn generate_csv( pub async fn generate_csv(pool: &sqlx::PgPool, params: &ReportParams) -> anyhow::Result<Vec<u8>> {
pool: &sqlx::PgPool,
params: &ReportParams,
) -> anyhow::Result<Vec<u8>> {
match params.report_type { match params.report_type {
ReportType::Compliance => compliance_csv(pool, params).await, ReportType::Compliance => compliance_csv(pool, params).await,
ReportType::PatchHistory => patch_history_csv(pool, params).await, ReportType::PatchHistory => patch_history_csv(pool, params).await,
@ -20,12 +17,10 @@ pub async fn generate_csv(
// Compliance // Compliance
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
async fn compliance_csv( async fn compliance_csv(pool: &sqlx::PgPool, params: &ReportParams) -> anyhow::Result<Vec<u8>> {
pool: &sqlx::PgPool,
params: &ReportParams,
) -> anyhow::Result<Vec<u8>> {
let rows = if let Some(gid) = params.group_id { let rows = if let Some(gid) = params.group_id {
sqlx::query(" sqlx::query(
"
SELECT SELECT
h.id::text AS host_id, h.id::text AS host_id,
h.display_name, h.display_name,
@ -47,13 +42,15 @@ WHERE h.id IN (
) )
GROUP BY h.id, pd.total_packages, pd.pending_patches GROUP BY h.id, pd.total_packages, pd.pending_patches
ORDER BY compliance_pct ASC ORDER BY compliance_pct ASC
") ",
)
.bind(gid) .bind(gid)
.fetch_all(pool) .fetch_all(pool)
.await .await
.context("compliance query (group filter) failed")? .context("compliance query (group filter) failed")?
} else { } else {
sqlx::query(" sqlx::query(
"
SELECT SELECT
h.id::text AS host_id, h.id::text AS host_id,
h.display_name, h.display_name,
@ -72,7 +69,8 @@ LEFT JOIN host_groups hg ON hg.host_id = h.id
LEFT JOIN groups g ON g.id = hg.group_id LEFT JOIN groups g ON g.id = hg.group_id
GROUP BY h.id, pd.total_packages, pd.pending_patches GROUP BY h.id, pd.total_packages, pd.pending_patches
ORDER BY compliance_pct ASC ORDER BY compliance_pct ASC
") ",
)
.fetch_all(pool) .fetch_all(pool)
.await .await
.context("compliance query failed")? .context("compliance query failed")?
@ -80,23 +78,29 @@ ORDER BY compliance_pct ASC
let mut wtr = csv::Writer::from_writer(vec![]); let mut wtr = csv::Writer::from_writer(vec![]);
wtr.write_record(&[ wtr.write_record(&[
"host_id", "display_name", "fqdn", "group_names", "host_id",
"total_packages", "pending_patches", "compliance_pct", "display_name",
"last_patch_at", "health_status", "fqdn",
"group_names",
"total_packages",
"pending_patches",
"compliance_pct",
"last_patch_at",
"health_status",
])?; ])?;
for row in &rows { for row in &rows {
use sqlx::Row; use sqlx::Row;
let host_id: String = row.try_get("host_id").unwrap_or_default(); let host_id: String = row.try_get("host_id").unwrap_or_default();
let display_name: String = row.try_get("display_name").unwrap_or_default(); let display_name: String = row.try_get("display_name").unwrap_or_default();
let fqdn: String = row.try_get("fqdn").unwrap_or_default(); let fqdn: String = row.try_get("fqdn").unwrap_or_default();
let group_names: String = row.try_get("group_names").unwrap_or_default(); let group_names: String = row.try_get("group_names").unwrap_or_default();
let total_packages: i64 = row.try_get("total_packages").unwrap_or(0); let total_packages: i64 = row.try_get("total_packages").unwrap_or(0);
let pending_patches: i64 = row.try_get("pending_patches").unwrap_or(0); let pending_patches: i64 = row.try_get("pending_patches").unwrap_or(0);
let compliance_pct: f64 = row.try_get("compliance_pct").unwrap_or(0.0); let compliance_pct: f64 = row.try_get("compliance_pct").unwrap_or(0.0);
let last_patch_at: Option<chrono::DateTime<chrono::Utc>> = let last_patch_at: Option<chrono::DateTime<chrono::Utc>> =
row.try_get("last_patch_at").unwrap_or(None); row.try_get("last_patch_at").unwrap_or(None);
let health_status: String = row.try_get("health_status").unwrap_or_default(); let health_status: String = row.try_get("health_status").unwrap_or_default();
wtr.write_record(&[ wtr.write_record(&[
host_id, host_id,
@ -118,11 +122,9 @@ ORDER BY compliance_pct ASC
// Patch history // Patch history
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
async fn patch_history_csv( async fn patch_history_csv(pool: &sqlx::PgPool, params: &ReportParams) -> anyhow::Result<Vec<u8>> {
pool: &sqlx::PgPool, let rows = sqlx::query(
params: &ReportParams, "
) -> anyhow::Result<Vec<u8>> {
let rows = sqlx::query("
SELECT SELECT
pj.id::text AS job_id, pj.id::text AS job_id,
pj.kind::text AS job_kind, pj.kind::text AS job_kind,
@ -141,7 +143,8 @@ LEFT JOIN users u ON u.id = pj.created_by_user_id
WHERE ($1::timestamptz IS NULL OR pjh.started_at >= $1) WHERE ($1::timestamptz IS NULL OR pjh.started_at >= $1)
AND ($2::timestamptz IS NULL OR pjh.started_at <= $2) AND ($2::timestamptz IS NULL OR pjh.started_at <= $2)
ORDER BY pjh.started_at DESC ORDER BY pjh.started_at DESC
") ",
)
.bind(params.from) .bind(params.from)
.bind(params.to) .bind(params.to)
.fetch_all(pool) .fetch_all(pool)
@ -150,24 +153,32 @@ ORDER BY pjh.started_at DESC
let mut wtr = csv::Writer::from_writer(vec![]); let mut wtr = csv::Writer::from_writer(vec![]);
wtr.write_record(&[ wtr.write_record(&[
"job_id", "job_kind", "job_status", "host_display_name", "host_fqdn", "job_id",
"package_count", "started_at", "completed_at", "duration_seconds", "operator", "job_kind",
"job_status",
"host_display_name",
"host_fqdn",
"package_count",
"started_at",
"completed_at",
"duration_seconds",
"operator",
])?; ])?;
for row in &rows { for row in &rows {
use sqlx::Row; use sqlx::Row;
let job_id: String = row.try_get("job_id").unwrap_or_default(); let job_id: String = row.try_get("job_id").unwrap_or_default();
let job_kind: String = row.try_get("job_kind").unwrap_or_default(); let job_kind: String = row.try_get("job_kind").unwrap_or_default();
let job_status: String = row.try_get("job_status").unwrap_or_default(); let job_status: String = row.try_get("job_status").unwrap_or_default();
let display_name: String = row.try_get("display_name").unwrap_or_default(); let display_name: String = row.try_get("display_name").unwrap_or_default();
let fqdn: String = row.try_get("fqdn").unwrap_or_default(); let fqdn: String = row.try_get("fqdn").unwrap_or_default();
let package_count: i64 = row.try_get("package_count").unwrap_or(0); let package_count: i64 = row.try_get("package_count").unwrap_or(0);
let started_at: Option<chrono::DateTime<chrono::Utc>> = let started_at: Option<chrono::DateTime<chrono::Utc>> =
row.try_get("started_at").unwrap_or(None); row.try_get("started_at").unwrap_or(None);
let completed_at: Option<chrono::DateTime<chrono::Utc>> = let completed_at: Option<chrono::DateTime<chrono::Utc>> =
row.try_get("completed_at").unwrap_or(None); row.try_get("completed_at").unwrap_or(None);
let duration_seconds: Option<i64> = row.try_get("duration_seconds").unwrap_or(None); let duration_seconds: Option<i64> = row.try_get("duration_seconds").unwrap_or(None);
let operator: String = row.try_get("operator").unwrap_or_default(); let operator: String = row.try_get("operator").unwrap_or_default();
wtr.write_record(&[ wtr.write_record(&[
job_id, job_id,
@ -190,17 +201,21 @@ ORDER BY pjh.started_at DESC
// Vulnerability // Vulnerability
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
async fn vulnerability_csv( async fn vulnerability_csv(pool: &sqlx::PgPool, params: &ReportParams) -> anyhow::Result<Vec<u8>> {
pool: &sqlx::PgPool,
params: &ReportParams,
) -> anyhow::Result<Vec<u8>> {
let mut wtr = csv::Writer::from_writer(vec![]); let mut wtr = csv::Writer::from_writer(vec![]);
wtr.write_record(&[ wtr.write_record(&[
"host_id", "display_name", "fqdn", "cve_id", "host_id",
"package_name", "severity", "available_version", "last_seen_at", "display_name",
"fqdn",
"cve_id",
"package_name",
"severity",
"available_version",
"last_seen_at",
])?; ])?;
let result = sqlx::query(" let result = sqlx::query(
"
SELECT SELECT
h.id::text AS host_id, h.id::text AS host_id,
h.display_name, h.display_name,
@ -224,7 +239,8 @@ ORDER BY
ELSE 4 ELSE 4
END, END,
h.display_name h.display_name
") ",
)
.bind(params.from) .bind(params.from)
.bind(params.to) .bind(params.to)
.fetch_all(pool) .fetch_all(pool)
@ -234,12 +250,12 @@ ORDER BY
Ok(rows) => { Ok(rows) => {
for row in &rows { for row in &rows {
use sqlx::Row; use sqlx::Row;
let host_id: String = row.try_get("host_id").unwrap_or_default(); let host_id: String = row.try_get("host_id").unwrap_or_default();
let display_name: String = row.try_get("display_name").unwrap_or_default(); let display_name: String = row.try_get("display_name").unwrap_or_default();
let fqdn: String = row.try_get("fqdn").unwrap_or_default(); let fqdn: String = row.try_get("fqdn").unwrap_or_default();
let cve_id: String = row.try_get("cve_id").unwrap_or_default(); let cve_id: String = row.try_get("cve_id").unwrap_or_default();
let package_name: String = row.try_get("package_name").unwrap_or_default(); let package_name: String = row.try_get("package_name").unwrap_or_default();
let severity: String = row.try_get("severity").unwrap_or_default(); let severity: String = row.try_get("severity").unwrap_or_default();
let available_version: String = let available_version: String =
row.try_get("available_version").unwrap_or_default(); row.try_get("available_version").unwrap_or_default();
let last_seen_at: Option<chrono::DateTime<chrono::Utc>> = let last_seen_at: Option<chrono::DateTime<chrono::Utc>> =
@ -256,15 +272,21 @@ ORDER BY
last_seen_at.map(|d| d.to_rfc3339()).unwrap_or_default(), last_seen_at.map(|d| d.to_rfc3339()).unwrap_or_default(),
])?; ])?;
} }
} },
Err(e) => { Err(e) => {
tracing::warn!(error = %e, "vulnerability query failed — returning empty rows"); tracing::warn!(error = %e, "vulnerability query failed — returning empty rows");
// write a comment row indicating empty data // write a comment row indicating empty data
wtr.write_record(&[ wtr.write_record(&[
"(no data)", "", "", "", "", "", "", "(no data)",
"",
"",
"",
"",
"",
"",
&format!("query error: {}", e), &format!("query error: {}", e),
])?; ])?;
} },
} }
Ok(wtr.into_inner().context("csv flush failed")?) Ok(wtr.into_inner().context("csv flush failed")?)
@ -274,11 +296,9 @@ ORDER BY
// Audit // Audit
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
async fn audit_csv( async fn audit_csv(pool: &sqlx::PgPool, params: &ReportParams) -> anyhow::Result<Vec<u8>> {
pool: &sqlx::PgPool, let rows = sqlx::query(
params: &ReportParams, "
) -> anyhow::Result<Vec<u8>> {
let rows = sqlx::query("
SELECT SELECT
id::text AS id, id::text AS id,
created_at, created_at,
@ -293,7 +313,8 @@ WHERE ($1::timestamptz IS NULL OR created_at >= $1)
AND ($2::timestamptz IS NULL OR created_at <= $2) AND ($2::timestamptz IS NULL OR created_at <= $2)
ORDER BY created_at DESC ORDER BY created_at DESC
LIMIT 10000 LIMIT 10000
") ",
)
.bind(params.from) .bind(params.from)
.bind(params.to) .bind(params.to)
.fetch_all(pool) .fetch_all(pool)
@ -302,21 +323,27 @@ LIMIT 10000
let mut wtr = csv::Writer::from_writer(vec![]); let mut wtr = csv::Writer::from_writer(vec![]);
wtr.write_record(&[ wtr.write_record(&[
"id", "created_at", "action", "actor_username", "id",
"target_type", "target_id", "ip_address", "request_id", "created_at",
"action",
"actor_username",
"target_type",
"target_id",
"ip_address",
"request_id",
])?; ])?;
for row in &rows { for row in &rows {
use sqlx::Row; use sqlx::Row;
let id: String = row.try_get("id").unwrap_or_default(); let id: String = row.try_get("id").unwrap_or_default();
let created_at: Option<chrono::DateTime<chrono::Utc>> = let created_at: Option<chrono::DateTime<chrono::Utc>> =
row.try_get("created_at").unwrap_or(None); row.try_get("created_at").unwrap_or(None);
let action: String = row.try_get("action").unwrap_or_default(); let action: String = row.try_get("action").unwrap_or_default();
let actor_username: String = row.try_get("actor_username").unwrap_or_default(); let actor_username: String = row.try_get("actor_username").unwrap_or_default();
let target_type: String = row.try_get("target_type").unwrap_or_default(); let target_type: String = row.try_get("target_type").unwrap_or_default();
let target_id: String = row.try_get("target_id").unwrap_or_default(); let target_id: String = row.try_get("target_id").unwrap_or_default();
let ip_address: String = row.try_get("ip_address").unwrap_or_default(); let ip_address: String = row.try_get("ip_address").unwrap_or_default();
let request_id: String = row.try_get("request_id").unwrap_or_default(); let request_id: String = row.try_get("request_id").unwrap_or_default();
wtr.write_record(&[ wtr.write_record(&[
id, id,

View File

@ -6,9 +6,8 @@ use crate::{ReportParams, ReportType};
use anyhow::Context; use anyhow::Context;
use plotters::prelude::*; use plotters::prelude::*;
use printpdf::{ use printpdf::{
BuiltinFont, ColorBits, ColorSpace, Image, ImageTransform, ImageXObject, BuiltinFont, ColorBits, ColorSpace, Image, ImageTransform, ImageXObject, IndirectFontRef, Mm,
IndirectFontRef, Mm, PdfDocument, PdfLayerIndex, PdfLayerReference, PdfDocument, PdfLayerIndex, PdfLayerReference, PdfPageIndex, Px,
PdfPageIndex, Px,
}; };
const PAGE_W: f32 = 297.0; // A4 landscape width (mm) const PAGE_W: f32 = 297.0; // A4 landscape width (mm)
@ -23,15 +22,12 @@ const NEW_PAGE_THRESHOLD: f32 = 20.0;
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
/// Generate a PDF report and return the raw bytes. /// Generate a PDF report and return the raw bytes.
pub async fn generate_pdf( pub async fn generate_pdf(pool: &sqlx::PgPool, params: &ReportParams) -> anyhow::Result<Vec<u8>> {
pool: &sqlx::PgPool,
params: &ReportParams,
) -> anyhow::Result<Vec<u8>> {
match params.report_type { match params.report_type {
ReportType::Compliance => compliance_pdf(pool, params).await, ReportType::Compliance => compliance_pdf(pool, params).await,
ReportType::PatchHistory => patch_history_pdf(pool, params).await, ReportType::PatchHistory => patch_history_pdf(pool, params).await,
ReportType::Vulnerability => vulnerability_pdf(pool, params).await, ReportType::Vulnerability => vulnerability_pdf(pool, params).await,
ReportType::Audit => audit_pdf(pool, params).await, ReportType::Audit => audit_pdf(pool, params).await,
} }
} }
@ -51,15 +47,10 @@ fn render_bar_chart(
let mut pixel_buf = vec![0u8; (W * H * 3) as usize]; let mut pixel_buf = vec![0u8; (W * H * 3) as usize];
{ {
let root = BitMapBackend::with_buffer(&mut pixel_buf, (W, H)) let root = BitMapBackend::with_buffer(&mut pixel_buf, (W, H)).into_drawing_area();
.into_drawing_area();
root.fill(&WHITE)?; root.fill(&WHITE)?;
let max_val = values let max_val = values.iter().cloned().fold(0.0_f64, f64::max).max(1.0);
.iter()
.cloned()
.fold(0.0_f64, f64::max)
.max(1.0);
let n = labels.len().max(1); let n = labels.len().max(1);
let mut chart = ChartBuilder::on(&root) let mut chart = ChartBuilder::on(&root)
@ -76,7 +67,11 @@ fn render_bar_chart(
labels labels
.get(*idx) .get(*idx)
.map(|s| { .map(|s| {
if s.len() > 12 { s[..12].to_string() } else { s.clone() } if s.len() > 12 {
s[..12].to_string()
} else {
s.clone()
}
}) })
.unwrap_or_default() .unwrap_or_default()
}) })
@ -119,7 +114,7 @@ impl PdfBuilder {
fn new(title: &str) -> anyhow::Result<Self> { fn new(title: &str) -> anyhow::Result<Self> {
let doc = PdfDocument::empty(title); let doc = PdfDocument::empty(title);
let (page_idx, layer_idx) = doc.add_page(Mm(PAGE_W), Mm(PAGE_H), "Layer 1"); let (page_idx, layer_idx) = doc.add_page(Mm(PAGE_W), Mm(PAGE_H), "Layer 1");
let font = doc.add_builtin_font(BuiltinFont::Helvetica)?; let font = doc.add_builtin_font(BuiltinFont::Helvetica)?;
let font_bold = doc.add_builtin_font(BuiltinFont::HelveticaBold)?; let font_bold = doc.add_builtin_font(BuiltinFont::HelveticaBold)?;
Ok(Self { Ok(Self {
doc, doc,
@ -212,19 +207,31 @@ impl PdfBuilder {
fn write_title_page(pdf: &mut PdfBuilder, title: &str, params: &ReportParams) { fn write_title_page(pdf: &mut PdfBuilder, title: &str, params: &ReportParams) {
pdf.write_text(title, 24.0, MARGIN, 160.0, true); pdf.write_text(title, 24.0, MARGIN, 160.0, true);
pdf.write_text( pdf.write_text(
&format!("Generated: {}", chrono::Utc::now().format("%Y-%m-%d %H:%M UTC")), &format!(
11.0, MARGIN, 148.0, false, "Generated: {}",
chrono::Utc::now().format("%Y-%m-%d %H:%M UTC")
),
11.0,
MARGIN,
148.0,
false,
); );
if let Some(from) = params.from { if let Some(from) = params.from {
pdf.write_text( pdf.write_text(
&format!("From: {}", from.format("%Y-%m-%d")), &format!("From: {}", from.format("%Y-%m-%d")),
10.0, MARGIN, 140.0, false, 10.0,
MARGIN,
140.0,
false,
); );
} }
if let Some(to) = params.to { if let Some(to) = params.to {
pdf.write_text( pdf.write_text(
&format!("To: {}", to.format("%Y-%m-%d")), &format!("To: {}", to.format("%Y-%m-%d")),
10.0, MARGIN, 134.0, false, 10.0,
MARGIN,
134.0,
false,
); );
} }
if let Some(gid) = params.group_id { if let Some(gid) = params.group_id {
@ -237,13 +244,11 @@ fn write_title_page(pdf: &mut PdfBuilder, title: &str, params: &ReportParams) {
// Compliance PDF // Compliance PDF
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
async fn compliance_pdf( async fn compliance_pdf(pool: &sqlx::PgPool, params: &ReportParams) -> anyhow::Result<Vec<u8>> {
pool: &sqlx::PgPool,
params: &ReportParams,
) -> anyhow::Result<Vec<u8>> {
use sqlx::Row; use sqlx::Row;
let rows = if let Some(gid) = params.group_id { let rows = if let Some(gid) = params.group_id {
sqlx::query(" sqlx::query(
"
SELECT h.display_name, h.fqdn, SELECT h.display_name, h.fqdn,
COALESCE(pd.total_packages,0) AS total_packages, COALESCE(pd.total_packages,0) AS total_packages,
COALESCE(pd.pending_patches,0) AS pending_patches, COALESCE(pd.pending_patches,0) AS pending_patches,
@ -254,11 +259,15 @@ SELECT h.display_name, h.fqdn,
FROM hosts h LEFT JOIN host_patch_data pd ON pd.host_id=h.id FROM hosts h LEFT JOIN host_patch_data pd ON pd.host_id=h.id
WHERE h.id IN (SELECT host_id FROM host_groups WHERE group_id=$1) WHERE h.id IN (SELECT host_id FROM host_groups WHERE group_id=$1)
GROUP BY h.id,pd.total_packages,pd.pending_patches GROUP BY h.id,pd.total_packages,pd.pending_patches
ORDER BY compliance_pct ASC") ORDER BY compliance_pct ASC",
.bind(gid).fetch_all(pool).await )
.context("compliance PDF query (group) failed")? .bind(gid)
.fetch_all(pool)
.await
.context("compliance PDF query (group) failed")?
} else { } else {
sqlx::query(" sqlx::query(
"
SELECT h.display_name, h.fqdn, SELECT h.display_name, h.fqdn,
COALESCE(pd.total_packages,0) AS total_packages, COALESCE(pd.total_packages,0) AS total_packages,
COALESCE(pd.pending_patches,0) AS pending_patches, COALESCE(pd.pending_patches,0) AS pending_patches,
@ -268,26 +277,55 @@ SELECT h.display_name, h.fqdn,
h.health_status::text AS health_status h.health_status::text AS health_status
FROM hosts h LEFT JOIN host_patch_data pd ON pd.host_id=h.id FROM hosts h LEFT JOIN host_patch_data pd ON pd.host_id=h.id
GROUP BY h.id,pd.total_packages,pd.pending_patches GROUP BY h.id,pd.total_packages,pd.pending_patches
ORDER BY compliance_pct ASC") ORDER BY compliance_pct ASC",
.fetch_all(pool).await )
.context("compliance PDF query failed")? .fetch_all(pool)
.await
.context("compliance PDF query failed")?
}; };
let labels: Vec<String> = rows.iter().map(|r| r.try_get::<String,_>("display_name").unwrap_or_default()).collect(); let labels: Vec<String> = rows
let values: Vec<f64> = rows.iter().map(|r| r.try_get::<f64,_>("compliance_pct").unwrap_or(0.0)).collect(); .iter()
.map(|r| r.try_get::<String, _>("display_name").unwrap_or_default())
.collect();
let values: Vec<f64> = rows
.iter()
.map(|r| r.try_get::<f64, _>("compliance_pct").unwrap_or(0.0))
.collect();
let mut pdf = PdfBuilder::new("Compliance Report")?; let mut pdf = PdfBuilder::new("Compliance Report")?;
write_title_page(&mut pdf, "Compliance Report", params); write_title_page(&mut pdf, "Compliance Report", params);
let col_x: &[f32] = &[MARGIN, 65.0, 130.0, 165.0, 200.0, 235.0]; let col_x: &[f32] = &[MARGIN, 65.0, 130.0, 165.0, 200.0, 235.0];
pdf.table_row(&["Host","FQDN","Total Pkgs","Pending","Compliance %","Status"], col_x, 9.0, true); pdf.table_row(
&[
"Host",
"FQDN",
"Total Pkgs",
"Pending",
"Compliance %",
"Status",
],
col_x,
9.0,
true,
);
for row in &rows { for row in &rows {
let name: String = row.try_get("display_name").unwrap_or_default(); let name: String = row.try_get("display_name").unwrap_or_default();
let fqdn: String = row.try_get("fqdn").unwrap_or_default(); let fqdn: String = row.try_get("fqdn").unwrap_or_default();
let total: i64 = row.try_get("total_packages").unwrap_or(0); let total: i64 = row.try_get("total_packages").unwrap_or(0);
let pend: i64 = row.try_get("pending_patches").unwrap_or(0); let pend: i64 = row.try_get("pending_patches").unwrap_or(0);
let pct: f64 = row.try_get("compliance_pct").unwrap_or(0.0); let pct: f64 = row.try_get("compliance_pct").unwrap_or(0.0);
let status: String = row.try_get("health_status").unwrap_or_default(); let status: String = row.try_get("health_status").unwrap_or_default();
pdf.table_row( pdf.table_row(
&[&name,&fqdn,&total.to_string(),&pend.to_string(),&format!("{:.1}%",pct),&status], &[
col_x, 8.0, false, &name,
&fqdn,
&total.to_string(),
&pend.to_string(),
&format!("{:.1}%", pct),
&status,
],
col_x,
8.0,
false,
); );
} }
if !labels.is_empty() { if !labels.is_empty() {
@ -298,7 +336,7 @@ ORDER BY compliance_pct ASC")
if let Err(e) = pdf.embed_image(raw, w, h, MARGIN, 10.0, 0.18, 0.18) { if let Err(e) = pdf.embed_image(raw, w, h, MARGIN, 10.0, 0.18, 0.18) {
tracing::warn!(error = %e, "chart embed failed"); tracing::warn!(error = %e, "chart embed failed");
} }
} },
Err(e) => tracing::warn!(error = %e, "chart render failed"), Err(e) => tracing::warn!(error = %e, "chart render failed"),
} }
} }
@ -309,12 +347,10 @@ ORDER BY compliance_pct ASC")
// Patch history PDF // Patch history PDF
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
async fn patch_history_pdf( async fn patch_history_pdf(pool: &sqlx::PgPool, params: &ReportParams) -> anyhow::Result<Vec<u8>> {
pool: &sqlx::PgPool,
params: &ReportParams,
) -> anyhow::Result<Vec<u8>> {
use sqlx::Row; use sqlx::Row;
let rows = sqlx::query(" let rows = sqlx::query(
"
SELECT pj.kind::text AS job_kind, pj.status::text AS job_status, SELECT pj.kind::text AS job_kind, pj.status::text AS job_status,
h.display_name, h.fqdn, pjh.started_at, pjh.completed_at, h.display_name, h.fqdn, pjh.started_at, pjh.completed_at,
EXTRACT(EPOCH FROM (pjh.completed_at-pjh.started_at))::bigint AS duration_seconds, EXTRACT(EPOCH FROM (pjh.completed_at-pjh.started_at))::bigint AS duration_seconds,
@ -325,33 +361,71 @@ JOIN hosts h ON h.id=pjh.host_id
LEFT JOIN users u ON u.id=pj.created_by_user_id LEFT JOIN users u ON u.id=pj.created_by_user_id
WHERE ($1::timestamptz IS NULL OR pjh.started_at>=$1) WHERE ($1::timestamptz IS NULL OR pjh.started_at>=$1)
AND ($2::timestamptz IS NULL OR pjh.started_at<=$2) AND ($2::timestamptz IS NULL OR pjh.started_at<=$2)
ORDER BY pjh.started_at DESC") ORDER BY pjh.started_at DESC",
.bind(params.from).bind(params.to).fetch_all(pool).await )
.context("patch history PDF query failed")?; .bind(params.from)
let mut dc: std::collections::BTreeMap<String,f64> = std::collections::BTreeMap::new(); .bind(params.to)
.fetch_all(pool)
.await
.context("patch history PDF query failed")?;
let mut dc: std::collections::BTreeMap<String, f64> = std::collections::BTreeMap::new();
for row in &rows { for row in &rows {
if let Ok(Some(s)) = row.try_get::<Option<chrono::DateTime<chrono::Utc>>,_>("started_at") { if let Ok(Some(s)) = row.try_get::<Option<chrono::DateTime<chrono::Utc>>, _>("started_at") {
*dc.entry(s.format("%Y-%m-%d").to_string()).or_insert(0.0) += 1.0; *dc.entry(s.format("%Y-%m-%d").to_string()).or_insert(0.0) += 1.0;
} }
} }
let cl: Vec<String> = dc.keys().cloned().collect(); let cl: Vec<String> = dc.keys().cloned().collect();
let cv: Vec<f64> = dc.values().cloned().collect(); let cv: Vec<f64> = dc.values().cloned().collect();
let mut pdf = PdfBuilder::new("Patch History Report")?; let mut pdf = PdfBuilder::new("Patch History Report")?;
write_title_page(&mut pdf, "Patch History Report", params); write_title_page(&mut pdf, "Patch History Report", params);
let col_x: &[f32] = &[MARGIN,45.0,80.0,115.0,155.0,200.0,245.0,270.0]; let col_x: &[f32] = &[MARGIN, 45.0, 80.0, 115.0, 155.0, 200.0, 245.0, 270.0];
pdf.table_row(&["Kind","Status","Host","FQDN","Started","Completed","Dur(s)","Operator"], col_x, 9.0, true); pdf.table_row(
&[
"Kind",
"Status",
"Host",
"FQDN",
"Started",
"Completed",
"Dur(s)",
"Operator",
],
col_x,
9.0,
true,
);
for row in &rows { for row in &rows {
let kind: String = row.try_get("job_kind").unwrap_or_default(); let kind: String = row.try_get("job_kind").unwrap_or_default();
let status: String = row.try_get("job_status").unwrap_or_default(); let status: String = row.try_get("job_status").unwrap_or_default();
let name: String = row.try_get("display_name").unwrap_or_default(); let name: String = row.try_get("display_name").unwrap_or_default();
let fqdn: String = row.try_get("fqdn").unwrap_or_default(); let fqdn: String = row.try_get("fqdn").unwrap_or_default();
let started: String = row.try_get::<Option<chrono::DateTime<chrono::Utc>>,_>("started_at") let started: String = row
.unwrap_or(None).map(|d| d.format("%Y-%m-%d %H:%M").to_string()).unwrap_or_default(); .try_get::<Option<chrono::DateTime<chrono::Utc>>, _>("started_at")
let completed: String = row.try_get::<Option<chrono::DateTime<chrono::Utc>>,_>("completed_at") .unwrap_or(None)
.unwrap_or(None).map(|d| d.format("%Y-%m-%d %H:%M").to_string()).unwrap_or_default(); .map(|d| d.format("%Y-%m-%d %H:%M").to_string())
let dur: i64 = row.try_get("duration_seconds").unwrap_or(0); .unwrap_or_default();
let op: String = row.try_get("operator").unwrap_or_default(); let completed: String = row
pdf.table_row(&[&kind,&status,&name,&fqdn,&started,&completed,&dur.to_string(),&op], col_x, 8.0, false); .try_get::<Option<chrono::DateTime<chrono::Utc>>, _>("completed_at")
.unwrap_or(None)
.map(|d| d.format("%Y-%m-%d %H:%M").to_string())
.unwrap_or_default();
let dur: i64 = row.try_get("duration_seconds").unwrap_or(0);
let op: String = row.try_get("operator").unwrap_or_default();
pdf.table_row(
&[
&kind,
&status,
&name,
&fqdn,
&started,
&completed,
&dur.to_string(),
&op,
],
col_x,
8.0,
false,
);
} }
if !cl.is_empty() { if !cl.is_empty() {
match render_bar_chart(&cl, &cv, "Jobs per Day") { match render_bar_chart(&cl, &cv, "Jobs per Day") {
@ -361,7 +435,7 @@ ORDER BY pjh.started_at DESC")
if let Err(e) = pdf.embed_image(raw, w, h, MARGIN, 10.0, 0.18, 0.18) { if let Err(e) = pdf.embed_image(raw, w, h, MARGIN, 10.0, 0.18, 0.18) {
tracing::warn!(error = %e, "chart embed failed"); tracing::warn!(error = %e, "chart embed failed");
} }
} },
Err(e) => tracing::warn!(error = %e, "chart render failed"), Err(e) => tracing::warn!(error = %e, "chart render failed"),
} }
} }
@ -372,10 +446,7 @@ ORDER BY pjh.started_at DESC")
// Vulnerability PDF // Vulnerability PDF
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
async fn vulnerability_pdf( async fn vulnerability_pdf(pool: &sqlx::PgPool, params: &ReportParams) -> anyhow::Result<Vec<u8>> {
pool: &sqlx::PgPool,
params: &ReportParams,
) -> anyhow::Result<Vec<u8>> {
use sqlx::Row; use sqlx::Row;
// Query DB FIRST (before creating any non-Send PdfBuilder) // Query DB FIRST (before creating any non-Send PdfBuilder)
let query_result = sqlx::query(" let query_result = sqlx::query("
@ -393,61 +464,104 @@ ORDER BY CASE cve.severity WHEN 'critical' THEN 1 WHEN 'high' THEN 2 WHEN 'mediu
// Now create PdfBuilder (non-Send Rc types) after all awaits // Now create PdfBuilder (non-Send Rc types) after all awaits
let mut pdf = PdfBuilder::new("Vulnerability Report")?; let mut pdf = PdfBuilder::new("Vulnerability Report")?;
write_title_page(&mut pdf, "Vulnerability Exposure Report", params); write_title_page(&mut pdf, "Vulnerability Exposure Report", params);
let col_x: &[f32] = &[MARGIN,55.0,100.0,130.0,175.0,215.0,255.0]; let col_x: &[f32] = &[MARGIN, 55.0, 100.0, 130.0, 175.0, 215.0, 255.0];
pdf.table_row(&["Host","FQDN","CVE ID","Package","Severity","Fix Version","Last Seen"], col_x, 9.0, true); pdf.table_row(
&[
"Host",
"FQDN",
"CVE ID",
"Package",
"Severity",
"Fix Version",
"Last Seen",
],
col_x,
9.0,
true,
);
match query_result { match query_result {
Ok(rows) => { Ok(rows) => {
for row in &rows { for row in &rows {
let name: String = row.try_get("display_name").unwrap_or_default(); let name: String = row.try_get("display_name").unwrap_or_default();
let fqdn: String = row.try_get("fqdn").unwrap_or_default(); let fqdn: String = row.try_get("fqdn").unwrap_or_default();
let cve: String = row.try_get("cve_id").unwrap_or_default(); let cve: String = row.try_get("cve_id").unwrap_or_default();
let pkg: String = row.try_get("package_name").unwrap_or_default(); let pkg: String = row.try_get("package_name").unwrap_or_default();
let sev: String = row.try_get("severity").unwrap_or_default(); let sev: String = row.try_get("severity").unwrap_or_default();
let fix: String = row.try_get("available_version").unwrap_or_default(); let fix: String = row.try_get("available_version").unwrap_or_default();
let seen: String = row.try_get::<Option<chrono::DateTime<chrono::Utc>>,_>("last_seen_at") let seen: String = row
.unwrap_or(None).map(|d| d.format("%Y-%m-%d").to_string()).unwrap_or_default(); .try_get::<Option<chrono::DateTime<chrono::Utc>>, _>("last_seen_at")
pdf.table_row(&[&name,&fqdn,&cve,&pkg,&sev,&fix,&seen], col_x, 8.0, false); .unwrap_or(None)
.map(|d| d.format("%Y-%m-%d").to_string())
.unwrap_or_default();
pdf.table_row(
&[&name, &fqdn, &cve, &pkg, &sev, &fix, &seen],
col_x,
8.0,
false,
);
} }
} },
Err(e) => { Err(e) => {
tracing::warn!(error = %e, "vulnerability PDF query failed"); tracing::warn!(error = %e, "vulnerability PDF query failed");
let y = pdf.current_y; let y = pdf.current_y;
pdf.write_text(&format!("No data: {}", e), 10.0, MARGIN, y, false); pdf.write_text(&format!("No data: {}", e), 10.0, MARGIN, y, false);
} },
} }
pdf.save() pdf.save()
} }
async fn audit_pdf(pool: &sqlx::PgPool, params: &ReportParams) -> anyhow::Result<Vec<u8>> {
async fn audit_pdf(
pool: &sqlx::PgPool,
params: &ReportParams,
) -> anyhow::Result<Vec<u8>> {
use sqlx::Row; use sqlx::Row;
let rows = sqlx::query(" let rows = sqlx::query(
"
SELECT id::text AS id, created_at, action::text AS action, SELECT id::text AS id, created_at, action::text AS action,
actor_username, target_type, target_id, actor_username, target_type, target_id,
ip_address::text AS ip_address, request_id ip_address::text AS ip_address, request_id
FROM audit_log FROM audit_log
WHERE ($1::timestamptz IS NULL OR created_at>=$1) WHERE ($1::timestamptz IS NULL OR created_at>=$1)
AND ($2::timestamptz IS NULL OR created_at<=$2) AND ($2::timestamptz IS NULL OR created_at<=$2)
ORDER BY created_at DESC LIMIT 10000") ORDER BY created_at DESC LIMIT 10000",
.bind(params.from).bind(params.to).fetch_all(pool).await )
.context("audit PDF query failed")?; .bind(params.from)
.bind(params.to)
.fetch_all(pool)
.await
.context("audit PDF query failed")?;
let mut pdf = PdfBuilder::new("Audit Trail Report")?; let mut pdf = PdfBuilder::new("Audit Trail Report")?;
write_title_page(&mut pdf, "Audit Trail Report", params); write_title_page(&mut pdf, "Audit Trail Report", params);
let col_x: &[f32] = &[MARGIN,50.0,95.0,135.0,175.0,215.0,255.0]; let col_x: &[f32] = &[MARGIN, 50.0, 95.0, 135.0, 175.0, 215.0, 255.0];
pdf.table_row(&["Timestamp","Action","Actor","Target Type","Target ID","IP","Request ID"], col_x, 9.0, true); pdf.table_row(
&[
"Timestamp",
"Action",
"Actor",
"Target Type",
"Target ID",
"IP",
"Request ID",
],
col_x,
9.0,
true,
);
for row in &rows { for row in &rows {
let created: String = row.try_get::<Option<chrono::DateTime<chrono::Utc>>,_>("created_at") let created: String = row
.unwrap_or(None).map(|d| d.format("%Y-%m-%d %H:%M").to_string()).unwrap_or_default(); .try_get::<Option<chrono::DateTime<chrono::Utc>>, _>("created_at")
.unwrap_or(None)
.map(|d| d.format("%Y-%m-%d %H:%M").to_string())
.unwrap_or_default();
let action: String = row.try_get("action").unwrap_or_default(); let action: String = row.try_get("action").unwrap_or_default();
let actor: String = row.try_get("actor_username").unwrap_or_default(); let actor: String = row.try_get("actor_username").unwrap_or_default();
let ttype: String = row.try_get("target_type").unwrap_or_default(); let ttype: String = row.try_get("target_type").unwrap_or_default();
let tid: String = row.try_get("target_id").unwrap_or_default(); let tid: String = row.try_get("target_id").unwrap_or_default();
let ip: String = row.try_get("ip_address").unwrap_or_default(); let ip: String = row.try_get("ip_address").unwrap_or_default();
let req: String = row.try_get("request_id").unwrap_or_default(); let req: String = row.try_get("request_id").unwrap_or_default();
pdf.table_row(&[&created,&action,&actor,&ttype,&tid,&ip,&req], col_x, 8.0, false); pdf.table_row(
&[&created, &action, &actor, &ttype, &tid, &ip, &req],
col_x,
8.0,
false,
);
} }
pdf.save() pdf.save()
} }

View File

@ -2,37 +2,18 @@
mod routes; mod routes;
use axum::{ use axum::{extract::State, http::StatusCode, middleware, response::Json, routing::get, Router};
extract::State,
http::StatusCode,
middleware,
response::Json,
routing::get,
Router,
};
use dashmap::DashMap; use dashmap::DashMap;
use pm_core::{
config::AppConfig,
db,
logging,
request_id::request_id_middleware,
};
use pm_auth::{ use pm_auth::{
jwt, jwt,
rbac::{AuthConfig, require_auth}, rbac::{require_auth, AuthConfig},
}; };
use routes::ws::WsTicket; use pm_core::{config::AppConfig, db, logging, request_id::request_id_middleware};
use routes::azure_sso::SsoSession; use routes::azure_sso::SsoSession;
use routes::ws::WsTicket;
use serde_json::{json, Value}; use serde_json::{json, Value};
use std::{ use std::{net::SocketAddr, sync::Arc, time::Duration};
net::SocketAddr, use tower_http::{services::ServeDir, trace::TraceLayer};
sync::Arc,
time::Duration,
};
use tower_http::{
services::ServeDir,
trace::TraceLayer,
};
/// Shared application state threaded through Axum. /// Shared application state threaded through Axum.
#[derive(Clone)] #[derive(Clone)]
@ -60,7 +41,10 @@ async fn main() -> anyhow::Result<()> {
}); });
logging::init(&config.logging); logging::init(&config.logging);
tracing::info!(version = env!("CARGO_PKG_VERSION"), "patch-manager-web starting"); tracing::info!(
version = env!("CARGO_PKG_VERSION"),
"patch-manager-web starting"
);
let signing_key_pem = jwt::load_signing_key(&config.security.jwt_signing_key_path) let signing_key_pem = jwt::load_signing_key(&config.security.jwt_signing_key_path)
.unwrap_or_else(|e| { .unwrap_or_else(|e| {
@ -68,8 +52,8 @@ async fn main() -> anyhow::Result<()> {
String::new() String::new()
}); });
let verify_key_pem = jwt::load_verify_key(&config.security.jwt_verify_key_path) let verify_key_pem =
.unwrap_or_else(|e| { jwt::load_verify_key(&config.security.jwt_verify_key_path).unwrap_or_else(|e| {
tracing::warn!(error = %e, "JWT verify key not found (dev mode)"); tracing::warn!(error = %e, "JWT verify key not found (dev mode)");
String::new() String::new()
}); });
@ -159,7 +143,10 @@ pub fn build_router(state: AppState) -> Router {
// Patch jobs // Patch jobs
.nest("/jobs", routes::jobs::router()) .nest("/jobs", routes::jobs::router())
// Maintenance windows (nested under hosts path param) // Maintenance windows (nested under hosts path param)
.nest("/hosts/:host_id/maintenance-windows", routes::maintenance_windows::router()) .nest(
"/hosts/:host_id/maintenance-windows",
routes::maintenance_windows::router(),
)
// CA root certificate download // CA root certificate download
.nest("/ca", routes::ca::ca_router()) .nest("/ca", routes::ca::ca_router())
// Certificate list / renew / revoke // Certificate list / renew / revoke
@ -187,9 +174,7 @@ pub fn build_router(state: AppState) -> Router {
// WebSocket browser endpoint — ticket-authenticated, outside JWT middleware // WebSocket browser endpoint — ticket-authenticated, outside JWT middleware
.merge(routes::ws::ws_router()) .merge(routes::ws::ws_router())
// Serve React SPA // Serve React SPA
.fallback_service( .fallback_service(ServeDir::new(&static_dir).append_index_html_on_directories(true))
ServeDir::new(&static_dir).append_index_html_on_directories(true),
)
.layer(middleware::from_fn(request_id_middleware)) .layer(middleware::from_fn(request_id_middleware))
.layer(TraceLayer::new_for_http()) .layer(TraceLayer::new_for_http())
.with_state(state) .with_state(state)
@ -199,5 +184,9 @@ async fn health_handler(State(state): State<AppState>) -> Result<Json<Value>, St
let db_ok = sqlx::query("SELECT 1").execute(&state.db).await.is_ok(); let db_ok = sqlx::query("SELECT 1").execute(&state.db).await.is_ok();
let status = if db_ok { "healthy" } else { "degraded" }; let status = if db_ok { "healthy" } else { "degraded" };
let body = json!({ "service": "patch-manager-web", "version": env!("CARGO_PKG_VERSION"), "status": status, "database": if db_ok { "ok" } else { "error" } }); let body = json!({ "service": "patch-manager-web", "version": env!("CARGO_PKG_VERSION"), "status": status, "database": if db_ok { "ok" } else { "error" } });
if db_ok { Ok(Json(body)) } else { Err(StatusCode::SERVICE_UNAVAILABLE) } if db_ok {
Ok(Json(body))
} else {
Err(StatusCode::SERVICE_UNAVAILABLE)
}
} }

View File

@ -18,8 +18,8 @@ use axum::{
}; };
use pm_auth::{ use pm_auth::{
mfa_totp, mfa_totp,
session::{self, LoginRequest, LoginResponse},
rbac::AuthUser, rbac::AuthUser,
session::{self, LoginRequest, LoginResponse},
}; };
use serde::Deserialize; use serde::Deserialize;
use serde_json::{json, Value}; use serde_json::{json, Value};
@ -107,10 +107,17 @@ async fn login_handler(
), ),
_ => { _ => {
tracing::error!(error = %e, "Login error"); tracing::error!(error = %e, "Login error");
(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "An error occurred") (
} StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"An error occurred",
)
},
}; };
(status, Json(json!({ "error": { "code": code, "message": message } }))) (
status,
Json(json!({ "error": { "code": code, "message": message } })),
)
}) })
} }
@ -156,10 +163,17 @@ async fn refresh_handler(
), ),
_ => { _ => {
tracing::error!(error = %e, "Refresh error"); tracing::error!(error = %e, "Refresh error");
(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "An error occurred") (
} StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"An error occurred",
)
},
}; };
(status, Json(json!({ "error": { "code": code, "message": msg } }))) (
status,
Json(json!({ "error": { "code": code, "message": msg } })),
)
}) })
} }
@ -221,11 +235,13 @@ async fn mfa_verify_handler(
auth_user: AuthUser, auth_user: AuthUser,
Json(req): Json<MfaVerifyRequest>, Json(req): Json<MfaVerifyRequest>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
let valid = mfa_totp::verify_code(&auth_user.username, &req.secret_base32, &req.code) let valid =
.map_err(|e| ( mfa_totp::verify_code(&auth_user.username, &req.secret_base32, &req.code).map_err(|e| {
StatusCode::INTERNAL_SERVER_ERROR, (
Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } })), StatusCode::INTERNAL_SERVER_ERROR,
))?; Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } })),
)
})?;
if !valid { if !valid {
return Err(( return Err((

View File

@ -97,15 +97,19 @@ async fn azure_login(
None => { None => {
return Err(( return Err((
StatusCode::FORBIDDEN, StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Azure SSO is not configured" } })), Json(
json!({ "error": { "code": "forbidden", "message": "Azure SSO is not configured" } }),
),
)); ));
} },
}; };
if !enabled { if !enabled {
return Err(( return Err((
StatusCode::FORBIDDEN, StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Azure SSO is not enabled" } })), Json(
json!({ "error": { "code": "forbidden", "message": "Azure SSO is not enabled" } }),
),
)); ));
} }
@ -162,7 +166,9 @@ async fn azure_callback(
let desc = params.error_description.unwrap_or_default(); let desc = params.error_description.unwrap_or_default();
return Err(( return Err((
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
Json(json!({ "error": { "code": "sso_error", "message": format!("Azure AD error: {} - {}", error, desc) } })), Json(
json!({ "error": { "code": "sso_error", "message": format!("Azure AD error: {} - {}", error, desc) } }),
),
)); ));
} }
@ -176,7 +182,9 @@ async fn azure_callback(
let state_token = params.state.ok_or_else(|| { let state_token = params.state.ok_or_else(|| {
( (
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
Json(json!({ "error": { "code": "bad_request", "message": "Missing state parameter" } })), Json(
json!({ "error": { "code": "bad_request", "message": "Missing state parameter" } }),
),
) )
})?; })?;
@ -211,9 +219,11 @@ async fn azure_callback(
None => { None => {
return Err(( return Err((
StatusCode::INTERNAL_SERVER_ERROR, StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": "Azure SSO not configured" } })), Json(
json!({ "error": { "code": "internal_error", "message": "Azure SSO not configured" } }),
),
)); ));
} },
}; };
// Exchange code for tokens // Exchange code for tokens
@ -263,7 +273,9 @@ async fn azure_callback(
tracing::error!(status = %status, body = %body, "Token exchange failed"); tracing::error!(status = %status, body = %body, "Token exchange failed");
return Err(( return Err((
StatusCode::BAD_GATEWAY, StatusCode::BAD_GATEWAY,
Json(json!({ "error": { "code": "sso_error", "message": format!("Token exchange failed: HTTP {}", status) } })), Json(
json!({ "error": { "code": "sso_error", "message": format!("Token exchange failed: HTTP {}", status) } }),
),
)); ));
} }
@ -302,7 +314,9 @@ async fn azure_callback(
if email.is_empty() || oid.is_empty() { if email.is_empty() || oid.is_empty() {
return Err(( return Err((
StatusCode::BAD_GATEWAY, StatusCode::BAD_GATEWAY,
Json(json!({ "error": { "code": "sso_error", "message": "Missing email or oid in id_token" } })), Json(
json!({ "error": { "code": "sso_error", "message": "Missing email or oid in id_token" } }),
),
)); ));
} }
@ -326,9 +340,11 @@ async fn azure_callback(
Some(u) if !u.is_active => { Some(u) if !u.is_active => {
return Err(( return Err((
StatusCode::FORBIDDEN, StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "account_disabled", "message": "Account is disabled" } })), Json(
json!({ "error": { "code": "account_disabled", "message": "Account is disabled" } }),
),
)); ));
} },
Some(u) => u, Some(u) => u,
None => { None => {
// Auto-create user with role=operator, auth_provider=azure_sso // Auto-create user with role=operator, auth_provider=azure_sso
@ -372,22 +388,24 @@ async fn azure_callback(
is_active: true, is_active: true,
mfa_enabled: false, mfa_enabled: false,
} }
} },
}; };
// Update last_login_at and azure_oid // Update last_login_at and azure_oid
sqlx::query("UPDATE users SET last_login_at = NOW(), azure_oid = COALESCE(azure_oid, $1) WHERE id = $2") sqlx::query(
.bind(&oid) "UPDATE users SET last_login_at = NOW(), azure_oid = COALESCE(azure_oid, $1) WHERE id = $2",
.bind(user.id) )
.execute(&state.db) .bind(&oid)
.await .bind(user.id)
.map_err(|e| { .execute(&state.db)
tracing::error!(error = %e, "Failed to update last_login_at"); .await
( .map_err(|e| {
StatusCode::INTERNAL_SERVER_ERROR, tracing::error!(error = %e, "Failed to update last_login_at");
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })), (
) StatusCode::INTERNAL_SERVER_ERROR,
})?; Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })),
)
})?;
// Issue JWT access token + refresh token // Issue JWT access token + refresh token
let access_ttl = state.config.security.jwt_access_ttl_secs as i64; let access_ttl = state.config.security.jwt_access_ttl_secs as i64;
@ -466,6 +484,5 @@ fn decode_jwt_payload(token: &str) -> Result<IdTokenClaims, String> {
.decode(&payload_b64_padded) .decode(&payload_b64_padded)
.map_err(|e| format!("Base64 decode error: {}", e))?; .map_err(|e| format!("Base64 decode error: {}", e))?;
serde_json::from_slice(&payload_bytes) serde_json::from_slice(&payload_bytes).map_err(|e| format!("JSON parse error: {}", e))
.map_err(|e| format!("JSON parse error: {}", e))
} }

View File

@ -33,8 +33,7 @@ use crate::AppState;
/// Handles routes mounted at /api/v1/ca /// Handles routes mounted at /api/v1/ca
pub fn ca_router() -> Router<AppState> { pub fn ca_router() -> Router<AppState> {
Router::new() Router::new().route("/root.crt", get(download_root_ca))
.route("/root.crt", get(download_root_ca))
} }
/// Handles routes mounted at /api/v1/certificates /// Handles routes mounted at /api/v1/certificates
@ -84,10 +83,7 @@ struct IssueCertRequest {
// ── Helper: build PEM download response ────────────────────────────────────── // ── Helper: build PEM download response ──────────────────────────────────────
fn pem_response( fn pem_response(pem: String, filename: &str) -> Result<Response<Body>, (StatusCode, Json<Value>)> {
pem: String,
filename: &str,
) -> Result<Response<Body>, (StatusCode, Json<Value>)> {
let disposition = format!("attachment; filename=\"{filename}\""); let disposition = format!("attachment; filename=\"{filename}\"");
Response::builder() Response::builder()
.status(StatusCode::OK) .status(StatusCode::OK)
@ -174,7 +170,7 @@ async fn list_certificates(
.bind(st) .bind(st)
.fetch_all(&state.db) .fetch_all(&state.db)
.await .await
} },
(Some(hid), None) => { (Some(hid), None) => {
sqlx::query_as::<_, CertRow>( sqlx::query_as::<_, CertRow>(
r#"SELECT id, host_id, serial_number, common_name, r#"SELECT id, host_id, serial_number, common_name,
@ -187,7 +183,7 @@ async fn list_certificates(
.bind(hid) .bind(hid)
.fetch_all(&state.db) .fetch_all(&state.db)
.await .await
} },
(None, Some(st)) => { (None, Some(st)) => {
sqlx::query_as::<_, CertRow>( sqlx::query_as::<_, CertRow>(
r#"SELECT id, host_id, serial_number, common_name, r#"SELECT id, host_id, serial_number, common_name,
@ -200,7 +196,7 @@ async fn list_certificates(
.bind(st) .bind(st)
.fetch_all(&state.db) .fetch_all(&state.db)
.await .await
} },
(None, None) => { (None, None) => {
sqlx::query_as::<_, CertRow>( sqlx::query_as::<_, CertRow>(
r#"SELECT id, host_id, serial_number, common_name, r#"SELECT id, host_id, serial_number, common_name,
@ -211,7 +207,7 @@ async fn list_certificates(
) )
.fetch_all(&state.db) .fetch_all(&state.db)
.await .await
} },
} }
.map_err(db_error)?; .map_err(db_error)?;
@ -259,7 +255,7 @@ async fn download_client_cert(
) )
.await; .await;
pem_response(pem, "client.crt") pem_response(pem, "client.crt")
} },
None => Err(( None => Err((
StatusCode::NOT_FOUND, StatusCode::NOT_FOUND,
Json(json!({ Json(json!({
@ -328,25 +324,23 @@ async fn renew_cert(
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
require_admin(&auth)?; require_admin(&auth)?;
let issued = state let issued = state.ca.renew_cert(cert_id, &state.db).await.map_err(|e| {
.ca let msg = e.to_string();
.renew_cert(cert_id, &state.db) tracing::error!(error = %e, %cert_id, "Failed to renew cert");
.await if msg.contains("not found") {
.map_err(|e| { (
let msg = e.to_string(); StatusCode::NOT_FOUND,
tracing::error!(error = %e, %cert_id, "Failed to renew cert"); Json(
if msg.contains("not found") { json!({ "error": { "code": "not_found", "message": "Certificate not found" } }),
( ),
StatusCode::NOT_FOUND, )
Json(json!({ "error": { "code": "not_found", "message": "Certificate not found" } })), } else {
) (
} else { StatusCode::INTERNAL_SERVER_ERROR,
( Json(json!({ "error": { "code": "internal_error", "message": msg } })),
StatusCode::INTERNAL_SERVER_ERROR, )
Json(json!({ "error": { "code": "internal_error", "message": msg } })), }
) })?;
}
})?;
log_event( log_event(
&state.db, &state.db,

View File

@ -11,11 +11,11 @@ use axum::{
routing::{get, post}, routing::{get, post},
Router, Router,
}; };
use pm_auth::rbac::AuthUser;
use pm_core::{ use pm_core::{
audit::{log_event, AuditAction}, audit::{log_event, AuditAction},
models::{DiscoveryCidrRequest, DiscoveryResult, RegisterDiscoveredRequest}, models::{DiscoveryCidrRequest, DiscoveryResult, RegisterDiscoveredRequest},
}; };
use pm_auth::rbac::AuthUser;
use serde_json::{json, Value}; use serde_json::{json, Value};
use std::{ use std::{
net::{IpAddr, TcpStream}, net::{IpAddr, TcpStream},
@ -46,13 +46,18 @@ async fn start_cidr_scan(
Json(req): Json<DiscoveryCidrRequest>, Json(req): Json<DiscoveryCidrRequest>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
let cidr: ipnet::IpNet = req.cidr.parse().map_err(|_| ( let cidr: ipnet::IpNet = req.cidr.parse().map_err(|_| {
StatusCode::BAD_REQUEST, (
Json(json!({ "error": { "code": "bad_request", "message": "Invalid CIDR range" } })) StatusCode::BAD_REQUEST,
))?; Json(json!({ "error": { "code": "bad_request", "message": "Invalid CIDR range" } })),
)
})?;
let agent_port = req.agent_port.unwrap_or(12443) as u16; let agent_port = req.agent_port.unwrap_or(12443) as u16;
let scan_id = Uuid::new_v4(); let scan_id = Uuid::new_v4();
@ -67,13 +72,23 @@ async fn start_cidr_scan(
run_cidr_scan(pool, scan_id_clone, cidr, agent_port).await; run_cidr_scan(pool, scan_id_clone, cidr, agent_port).await;
}); });
log_event(&state.db, AuditAction::DiscoveryScanStarted, log_event(
Some(auth.user_id), Some(&auth.username), &state.db,
Some("discovery"), Some(&scan_id.to_string()), AuditAction::DiscoveryScanStarted,
json!({ "cidr": cidr_str }), None, None).await; Some(auth.user_id),
Some(&auth.username),
Some("discovery"),
Some(&scan_id.to_string()),
json!({ "cidr": cidr_str }),
None,
None,
)
.await;
tracing::info!(scan_id = %scan_id, cidr = %req.cidr, "CIDR scan started"); tracing::info!(scan_id = %scan_id, cidr = %req.cidr, "CIDR scan started");
Ok(Json(json!({ "scan_id": scan_id, "message": "Discovery scan started", "cidr": req.cidr }))) Ok(Json(
json!({ "scan_id": scan_id, "message": "Discovery scan started", "cidr": req.cidr }),
))
} }
/// Background CIDR scanner. /// Background CIDR scanner.
@ -103,12 +118,7 @@ async fn run_cidr_scan(pool: sqlx::PgPool, scan_id: Uuid, cidr: ipnet::IpNet, po
} }
/// Probe a single IP:port and store the result if the port is open. /// Probe a single IP:port and store the result if the port is open.
async fn probe_and_store( async fn probe_and_store(pool: sqlx::PgPool, scan_id: Uuid, ip: IpAddr, port: u16) -> Option<()> {
pool: sqlx::PgPool,
scan_id: Uuid,
ip: IpAddr,
port: u16,
) -> Option<()> {
let addr = format!("{ip}:{port}"); let addr = format!("{ip}:{port}");
// TCP connect probe (blocking, run in thread pool) // TCP connect probe (blocking, run in thread pool)
@ -116,9 +126,13 @@ async fn probe_and_store(
let addr_clone = addr.clone(); let addr_clone = addr.clone();
let open = task::spawn_blocking(move || { let open = task::spawn_blocking(move || {
TcpStream::connect_timeout( TcpStream::connect_timeout(
&match addr_clone.parse() { Ok(a) => a, Err(_) => return false }, &match addr_clone.parse() {
Ok(a) => a,
Err(_) => return false,
},
Duration::from_secs(PROBE_TIMEOUT_SECS), Duration::from_secs(PROBE_TIMEOUT_SECS),
).is_ok() )
.is_ok()
}) })
.await .await
.unwrap_or(false); .unwrap_or(false);
@ -132,7 +146,8 @@ async fn probe_and_store(
let fqdn = task::spawn_blocking(move || { let fqdn = task::spawn_blocking(move || {
use std::net::ToSocketAddrs; use std::net::ToSocketAddrs;
let addr = format!("{ip_clone}:{port}"); let addr = format!("{ip_clone}:{port}");
addr.to_socket_addrs().ok() addr.to_socket_addrs()
.ok()
.and_then(|mut a| a.next()) .and_then(|mut a| a.next())
.and_then(|_| dns_lookup_for_ip(ip_clone)) .and_then(|_| dns_lookup_for_ip(ip_clone))
}) })
@ -163,7 +178,10 @@ fn dns_lookup_for_ip(ip: IpAddr) -> Option<String> {
// Standard library doesn't have reverse lookup; use getaddrinfo via format // Standard library doesn't have reverse lookup; use getaddrinfo via format
let host = format!("{ip}"); let host = format!("{ip}");
// Best-effort: try to resolve numeric address to hostname // Best-effort: try to resolve numeric address to hostname
(host + ":0").to_socket_addrs().ok()?.next() (host + ":0")
.to_socket_addrs()
.ok()?
.next()
.map(|a| a.ip().to_string()) .map(|a| a.ip().to_string())
.filter(|s| s != &ip.to_string()) .filter(|s| s != &ip.to_string())
} }
@ -188,7 +206,10 @@ async fn get_scan_results(
.map(Json) .map(Json)
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e); tracing::error!(error = %e);
(StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": "Database error" } }))) (
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })),
)
}) })
} }
@ -201,7 +222,10 @@ async fn register_discovered_host(
Json(req): Json<RegisterDiscoveredRequest>, Json(req): Json<RegisterDiscoveredRequest>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
// Fetch discovery result // Fetch discovery result
@ -213,7 +237,12 @@ async fn register_discovered_host(
.bind(id) .bind(id)
.fetch_optional(&state.db) .fetch_optional(&state.db)
.await .await
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } }))))?; .map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } })),
)
})?;
let result = result.ok_or_else(|| ( let result = result.ok_or_else(|| (
StatusCode::NOT_FOUND, StatusCode::NOT_FOUND,
@ -235,7 +264,12 @@ async fn register_discovered_host(
.bind(result.agent_port) .bind(result.agent_port)
.fetch_one(&state.db) .fetch_one(&state.db)
.await .await
.map_err(|e| (StatusCode::CONFLICT, Json(json!({ "error": { "code": "conflict", "message": e.to_string() } }))))?; .map_err(|e| {
(
StatusCode::CONFLICT,
Json(json!({ "error": { "code": "conflict", "message": e.to_string() } })),
)
})?;
// Assign to groups // Assign to groups
if let Some(group_ids) = &req.group_ids { if let Some(group_ids) = &req.group_ids {
@ -247,10 +281,24 @@ async fn register_discovered_host(
// Mark as registered // Mark as registered
let _ = sqlx::query("UPDATE discovery_results SET registered = TRUE WHERE id = $1") let _ = sqlx::query("UPDATE discovery_results SET registered = TRUE WHERE id = $1")
.bind(id).execute(&state.db).await; .bind(id)
.execute(&state.db)
.await;
log_event(&state.db, AuditAction::HostRegistered, Some(auth.user_id), Some(&auth.username), log_event(
Some("host"), Some(&host_id.to_string()), json!({ "from_discovery": true, "ip": result.ip_address }), None, None).await; &state.db,
AuditAction::HostRegistered,
Some(auth.user_id),
Some(&auth.username),
Some("host"),
Some(&host_id.to_string()),
json!({ "from_discovery": true, "ip": result.ip_address }),
None,
None,
)
.await;
Ok(Json(json!({ "host_id": host_id, "message": "Host registered from discovery" }))) Ok(Json(
json!({ "host_id": host_id, "message": "Host registered from discovery" }),
))
} }

View File

@ -15,11 +15,11 @@ use axum::{
routing::{delete, get, post, put}, routing::{delete, get, post, put},
Router, Router,
}; };
use pm_auth::rbac::AuthUser;
use pm_core::{ use pm_core::{
audit::{log_event, AuditAction}, audit::{log_event, AuditAction},
models::{Group, CreateGroupRequest, UpdateGroupRequest}, models::{CreateGroupRequest, Group, UpdateGroupRequest},
}; };
use pm_auth::rbac::AuthUser;
use serde_json::{json, Value}; use serde_json::{json, Value};
use uuid::Uuid; use uuid::Uuid;
@ -28,8 +28,14 @@ use crate::AppState;
pub fn router() -> Router<AppState> { pub fn router() -> Router<AppState> {
Router::new() Router::new()
.route("/", get(list_groups).post(create_group)) .route("/", get(list_groups).post(create_group))
.route("/:id", get(get_group).put(update_group).delete(delete_group)) .route(
.route("/:id/users/:user_id", post(add_user_to_group).delete(remove_user_from_group)) "/:id",
get(get_group).put(update_group).delete(delete_group),
)
.route(
"/:id/users/:user_id",
post(add_user_to_group).delete(remove_user_from_group),
)
} }
async fn list_groups( async fn list_groups(
@ -37,14 +43,17 @@ async fn list_groups(
_auth: AuthUser, _auth: AuthUser,
) -> Result<Json<Vec<Group>>, (StatusCode, Json<Value>)> { ) -> Result<Json<Vec<Group>>, (StatusCode, Json<Value>)> {
sqlx::query_as::<_, Group>( sqlx::query_as::<_, Group>(
"SELECT id, name, description, created_at, updated_at FROM groups ORDER BY name" "SELECT id, name, description, created_at, updated_at FROM groups ORDER BY name",
) )
.fetch_all(&state.db) .fetch_all(&state.db)
.await .await
.map(Json) .map(Json)
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, "Failed to list groups"); tracing::error!(error = %e, "Failed to list groups");
(StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": "Database error" } }))) (
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })),
)
}) })
} }
@ -54,23 +63,42 @@ async fn create_group(
Json(req): Json<CreateGroupRequest>, Json(req): Json<CreateGroupRequest>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
let id: Uuid = sqlx::query_scalar( let id: Uuid =
"INSERT INTO groups (name, description) VALUES ($1, $2) RETURNING id" sqlx::query_scalar("INSERT INTO groups (name, description) VALUES ($1, $2) RETURNING id")
) .bind(&req.name)
.bind(&req.name) .bind(req.description.as_deref().unwrap_or(""))
.bind(req.description.as_deref().unwrap_or("")) .fetch_one(&state.db)
.fetch_one(&state.db) .await
.await .map_err(|e| {
.map_err(|e| { let msg = if e.to_string().contains("unique") {
let msg = if e.to_string().contains("unique") { "Group name already exists".to_string() } else { "Database error".to_string() }; "Group name already exists".to_string()
(StatusCode::CONFLICT, Json(json!({ "error": { "code": "conflict", "message": msg } }))) } else {
})?; "Database error".to_string()
};
(
StatusCode::CONFLICT,
Json(json!({ "error": { "code": "conflict", "message": msg } })),
)
})?;
log_event(&state.db, AuditAction::GroupCreated, Some(auth.user_id), Some(&auth.username), log_event(
Some("group"), Some(&id.to_string()), json!({ "name": req.name }), None, None).await; &state.db,
AuditAction::GroupCreated,
Some(auth.user_id),
Some(&auth.username),
Some("group"),
Some(&id.to_string()),
json!({ "name": req.name }),
None,
None,
)
.await;
Ok(Json(json!({ "id": id, "message": "Group created" }))) Ok(Json(json!({ "id": id, "message": "Group created" })))
} }
@ -81,24 +109,43 @@ async fn get_group(
Path(id): Path<Uuid>, Path(id): Path<Uuid>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
let group: Option<Group> = sqlx::query_as( let group: Option<Group> = sqlx::query_as(
"SELECT id, name, description, created_at, updated_at FROM groups WHERE id = $1" "SELECT id, name, description, created_at, updated_at FROM groups WHERE id = $1",
) )
.bind(id) .bind(id)
.fetch_optional(&state.db) .fetch_optional(&state.db)
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e); (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": "Database error" } }))) tracing::error!(error = %e);
(
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })),
)
})?; })?;
let group = group.ok_or_else(|| (StatusCode::NOT_FOUND, Json(json!({ "error": { "code": "not_found", "message": "Group not found" } }))))?; let group = group.ok_or_else(|| {
(
StatusCode::NOT_FOUND,
Json(json!({ "error": { "code": "not_found", "message": "Group not found" } })),
)
})?;
// Fetch member counts // Fetch member counts
let host_count: i64 = sqlx::query_scalar("SELECT COUNT(*) FROM host_groups WHERE group_id = $1") let host_count: i64 =
.bind(id).fetch_one(&state.db).await.unwrap_or(0); sqlx::query_scalar("SELECT COUNT(*) FROM host_groups WHERE group_id = $1")
let user_count: i64 = sqlx::query_scalar("SELECT COUNT(*) FROM user_groups WHERE group_id = $1") .bind(id)
.bind(id).fetch_one(&state.db).await.unwrap_or(0); .fetch_one(&state.db)
.await
.unwrap_or(0);
let user_count: i64 =
sqlx::query_scalar("SELECT COUNT(*) FROM user_groups WHERE group_id = $1")
.bind(id)
.fetch_one(&state.db)
.await
.unwrap_or(0);
Ok(Json(json!({ "group": group, "host_count": host_count, "user_count": user_count }))) Ok(Json(
json!({ "group": group, "host_count": host_count, "user_count": user_count }),
))
} }
async fn update_group( async fn update_group(
@ -108,7 +155,10 @@ async fn update_group(
Json(req): Json<UpdateGroupRequest>, Json(req): Json<UpdateGroupRequest>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
let rows = sqlx::query( let rows = sqlx::query(
@ -123,7 +173,10 @@ async fn update_group(
.rows_affected(); .rows_affected();
if rows == 0 { if rows == 0 {
return Err((StatusCode::NOT_FOUND, Json(json!({ "error": { "code": "not_found", "message": "Group not found" } })))); return Err((
StatusCode::NOT_FOUND,
Json(json!({ "error": { "code": "not_found", "message": "Group not found" } })),
));
} }
Ok(Json(json!({ "message": "Group updated" }))) Ok(Json(json!({ "message": "Group updated" })))
@ -135,20 +188,43 @@ async fn delete_group(
Path(id): Path<Uuid>, Path(id): Path<Uuid>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
let rows = sqlx::query("DELETE FROM groups WHERE id = $1") let rows = sqlx::query("DELETE FROM groups WHERE id = $1")
.bind(id).execute(&state.db).await .bind(id)
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } }))))? .execute(&state.db)
.await
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } })),
)
})?
.rows_affected(); .rows_affected();
if rows == 0 { if rows == 0 {
return Err((StatusCode::NOT_FOUND, Json(json!({ "error": { "code": "not_found", "message": "Group not found" } })))); return Err((
StatusCode::NOT_FOUND,
Json(json!({ "error": { "code": "not_found", "message": "Group not found" } })),
));
} }
log_event(&state.db, AuditAction::GroupDeleted, Some(auth.user_id), Some(&auth.username), log_event(
Some("group"), Some(&id.to_string()), json!({}), None, None).await; &state.db,
AuditAction::GroupDeleted,
Some(auth.user_id),
Some(&auth.username),
Some("group"),
Some(&id.to_string()),
json!({}),
None,
None,
)
.await;
Ok(Json(json!({ "message": "Group deleted" }))) Ok(Json(json!({ "message": "Group deleted" })))
} }
@ -159,16 +235,38 @@ async fn add_user_to_group(
Path((id, user_id)): Path<(Uuid, Uuid)>, Path((id, user_id)): Path<(Uuid, Uuid)>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
sqlx::query("INSERT INTO user_groups (user_id, group_id) VALUES ($1, $2) ON CONFLICT DO NOTHING") sqlx::query(
.bind(user_id).bind(id) "INSERT INTO user_groups (user_id, group_id) VALUES ($1, $2) ON CONFLICT DO NOTHING",
.execute(&state.db).await )
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } }))))?; .bind(user_id)
.bind(id)
.execute(&state.db)
.await
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } })),
)
})?;
log_event(&state.db, AuditAction::GroupMembershipChanged, Some(auth.user_id), Some(&auth.username), log_event(
Some("user_group"), Some(&id.to_string()), json!({ "user_id": user_id, "action": "added" }), None, None).await; &state.db,
AuditAction::GroupMembershipChanged,
Some(auth.user_id),
Some(&auth.username),
Some("user_group"),
Some(&id.to_string()),
json!({ "user_id": user_id, "action": "added" }),
None,
None,
)
.await;
Ok(Json(json!({ "message": "User added to group" }))) Ok(Json(json!({ "message": "User added to group" })))
} }
@ -179,16 +277,36 @@ async fn remove_user_from_group(
Path((id, user_id)): Path<(Uuid, Uuid)>, Path((id, user_id)): Path<(Uuid, Uuid)>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
sqlx::query("DELETE FROM user_groups WHERE user_id = $1 AND group_id = $2") sqlx::query("DELETE FROM user_groups WHERE user_id = $1 AND group_id = $2")
.bind(user_id).bind(id) .bind(user_id)
.execute(&state.db).await .bind(id)
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } }))))?; .execute(&state.db)
.await
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } })),
)
})?;
log_event(&state.db, AuditAction::GroupMembershipChanged, Some(auth.user_id), Some(&auth.username), log_event(
Some("user_group"), Some(&id.to_string()), json!({ "user_id": user_id, "action": "removed" }), None, None).await; &state.db,
AuditAction::GroupMembershipChanged,
Some(auth.user_id),
Some(&auth.username),
Some("user_group"),
Some(&id.to_string()),
json!({ "user_id": user_id, "action": "removed" }),
None,
None,
)
.await;
Ok(Json(json!({ "message": "User removed from group" }))) Ok(Json(json!({ "message": "User removed from group" })))
} }

View File

@ -16,13 +16,11 @@ use axum::{
routing::{delete, get, post}, routing::{delete, get, post},
Router, Router,
}; };
use pm_auth::rbac::AuthUser;
use pm_core::{ use pm_core::{
audit::{log_event, AuditAction}, audit::{log_event, AuditAction},
models::{ models::{CreateHostRequest, Group, HostSummary},
CreateHostRequest, HostSummary, Group,
},
}; };
use pm_auth::rbac::AuthUser;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use serde_json::{json, Value}; use serde_json::{json, Value};
use uuid::Uuid; use uuid::Uuid;
@ -88,12 +86,11 @@ async fn operator_can_access_host(
} }
// Ungrouped hosts are accessible to any operator // Ungrouped hosts are accessible to any operator
let ungrouped: bool = sqlx::query_scalar( let ungrouped: bool =
"SELECT NOT EXISTS (SELECT 1 FROM host_groups WHERE host_id = $1)", sqlx::query_scalar("SELECT NOT EXISTS (SELECT 1 FROM host_groups WHERE host_id = $1)")
) .bind(host_id)
.bind(host_id) .fetch_one(pool)
.fetch_one(pool) .await?;
.await?;
Ok(ungrouped) Ok(ungrouped)
} }
@ -162,7 +159,12 @@ async fn list_hosts(
.await .await
.unwrap_or(0); .unwrap_or(0);
Ok(Json(HostListResponse { hosts, total, limit, offset })) Ok(Json(HostListResponse {
hosts,
total,
limit,
offset,
}))
} }
// ── POST /api/v1/hosts ──────────────────────────────────────────────────────── // ── POST /api/v1/hosts ────────────────────────────────────────────────────────
@ -244,7 +246,8 @@ async fn register_host(
json!({ "fqdn": req.fqdn, "ip": ip_address }), json!({ "fqdn": req.fqdn, "ip": ip_address }),
None, None,
None, None,
).await; )
.await;
tracing::info!(host_id = %host_id, fqdn = %req.fqdn, "Host registered"); tracing::info!(host_id = %host_id, fqdn = %req.fqdn, "Host registered");
Ok(Json(json!({ "id": host_id, "message": "Host registered" }))) Ok(Json(json!({ "id": host_id, "message": "Host registered" })))
@ -291,10 +294,12 @@ async fn get_host(
) )
})?; })?;
host.map(Json).ok_or_else(|| ( host.map(Json).ok_or_else(|| {
StatusCode::NOT_FOUND, (
Json(json!({ "error": { "code": "not_found", "message": "Host not found" } })), StatusCode::NOT_FOUND,
)) Json(json!({ "error": { "code": "not_found", "message": "Host not found" } })),
)
})
} }
// ── DELETE /api/v1/hosts/:id ────────────────────────────────────────────────── // ── DELETE /api/v1/hosts/:id ──────────────────────────────────────────────────
@ -347,7 +352,8 @@ async fn remove_host(
json!({ "fqdn": fqdn }), json!({ "fqdn": fqdn }),
None, None,
None, None,
).await; )
.await;
tracing::info!(host_id = %id, "Host removed"); tracing::info!(host_id = %id, "Host removed");
Ok(Json(json!({ "message": "Host removed" }))) Ok(Json(json!({ "message": "Host removed" })))
@ -362,10 +368,13 @@ async fn list_host_groups(
) -> Result<Json<Vec<Group>>, (StatusCode, Json<Value>)> { ) -> Result<Json<Vec<Group>>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
let can_access = operator_can_access_host(&state.db, auth.user_id, id) let can_access = operator_can_access_host(&state.db, auth.user_id, id)
.await.unwrap_or(false); .await
.unwrap_or(false);
if !can_access { if !can_access {
return Err((StatusCode::FORBIDDEN, return Err((
Json(json!({ "error": { "code": "forbidden", "message": "Access denied" } })))); StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Access denied" } })),
));
} }
} }
@ -381,8 +390,10 @@ async fn list_host_groups(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, "Failed to list host groups"); tracing::error!(error = %e, "Failed to list host groups");
(StatusCode::INTERNAL_SERVER_ERROR, (
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } }))) StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })),
)
})?; })?;
Ok(Json(groups)) Ok(Json(groups))
@ -391,7 +402,9 @@ async fn list_host_groups(
// ── POST /api/v1/hosts/:id/groups ───────────────────────────────────────────── // ── POST /api/v1/hosts/:id/groups ─────────────────────────────────────────────
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
struct AddToGroupRequest { group_id: Uuid } struct AddToGroupRequest {
group_id: Uuid,
}
async fn add_host_to_group( async fn add_host_to_group(
State(state): State<AppState>, State(state): State<AppState>,
@ -400,8 +413,10 @@ async fn add_host_to_group(
Json(req): Json<AddToGroupRequest>, Json(req): Json<AddToGroupRequest>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, return Err((
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
sqlx::query( sqlx::query(
@ -413,13 +428,24 @@ async fn add_host_to_group(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, "Failed to add host to group"); tracing::error!(error = %e, "Failed to add host to group");
(StatusCode::INTERNAL_SERVER_ERROR, (
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } }))) StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })),
)
})?; })?;
log_event(&state.db, AuditAction::GroupMembershipChanged, log_event(
Some(auth.user_id), Some(&auth.username), Some("host"), Some(&id.to_string()), &state.db,
json!({ "group_id": req.group_id, "action": "added" }), None, None).await; AuditAction::GroupMembershipChanged,
Some(auth.user_id),
Some(&auth.username),
Some("host"),
Some(&id.to_string()),
json!({ "group_id": req.group_id, "action": "added" }),
None,
None,
)
.await;
Ok(Json(json!({ "message": "Host added to group" }))) Ok(Json(json!({ "message": "Host added to group" })))
} }
@ -432,22 +458,37 @@ async fn remove_host_from_group(
Path((id, group_id)): Path<(Uuid, Uuid)>, Path((id, group_id)): Path<(Uuid, Uuid)>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, return Err((
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
sqlx::query("DELETE FROM host_groups WHERE host_id = $1 AND group_id = $2") sqlx::query("DELETE FROM host_groups WHERE host_id = $1 AND group_id = $2")
.bind(id).bind(group_id) .bind(id)
.execute(&state.db).await .bind(group_id)
.execute(&state.db)
.await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, "Failed to remove host from group"); tracing::error!(error = %e, "Failed to remove host from group");
(StatusCode::INTERNAL_SERVER_ERROR, (
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } }))) StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })),
)
})?; })?;
log_event(&state.db, AuditAction::GroupMembershipChanged, log_event(
Some(auth.user_id), Some(&auth.username), Some("host"), Some(&id.to_string()), &state.db,
json!({ "group_id": group_id, "action": "removed" }), None, None).await; AuditAction::GroupMembershipChanged,
Some(auth.user_id),
Some(&auth.username),
Some("host"),
Some(&id.to_string()),
json!({ "group_id": group_id, "action": "removed" }),
None,
None,
)
.await;
Ok(Json(json!({ "message": "Host removed from group" }))) Ok(Json(json!({ "message": "Host removed from group" })))
} }

View File

@ -149,7 +149,11 @@ async fn create_job(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, "create_job: insert patch_jobs failed"); tracing::error!(error = %e, "create_job: insert patch_jobs failed");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
// Insert one patch_job_hosts row per requested host. // Insert one patch_job_hosts row per requested host.
@ -170,7 +174,11 @@ async fn create_job(
error = %e, %job_id, %host_id, error = %e, %job_id, %host_id,
"create_job: insert patch_job_hosts failed" "create_job: insert patch_job_hosts failed"
); );
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
} }
@ -310,7 +318,12 @@ async fn list_jobs(
.unwrap_or(0) .unwrap_or(0)
}; };
Ok(Json(JobListResponse { jobs, total, limit, offset })) Ok(Json(JobListResponse {
jobs,
total,
limit,
offset,
}))
} }
// ── GET /api/v1/jobs/:id ───────────────────────────────────────────────────── // ── GET /api/v1/jobs/:id ─────────────────────────────────────────────────────
@ -325,11 +338,7 @@ async fn get_job(
.await .await
.unwrap_or(false); .unwrap_or(false);
if !allowed { if !allowed {
return Err(err( return Err(err(StatusCode::FORBIDDEN, "forbidden", "Access denied"));
StatusCode::FORBIDDEN,
"forbidden",
"Access denied",
));
} }
} }
@ -350,12 +359,14 @@ async fn get_job(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, %id, "get_job: failed to fetch job"); tracing::error!(error = %e, %id, "get_job: failed to fetch job");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
let job = job.ok_or_else(|| { let job = job.ok_or_else(|| err(StatusCode::NOT_FOUND, "not_found", "Job not found"))?;
err(StatusCode::NOT_FOUND, "not_found", "Job not found")
})?;
// Fetch per-host status rows joined to the host display name. // Fetch per-host status rows joined to the host display name.
let hosts: Vec<JobHostRow> = sqlx::query_as( let hosts: Vec<JobHostRow> = sqlx::query_as(
@ -383,7 +394,11 @@ async fn get_job(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, %id, "get_job: failed to fetch host rows"); tracing::error!(error = %e, %id, "get_job: failed to fetch host rows");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
Ok(Json(json!({ "job": job, "hosts": hosts }))) Ok(Json(json!({ "job": job, "hosts": hosts })))
@ -397,20 +412,22 @@ async fn cancel_job(
Path(id): Path<Uuid>, Path(id): Path<Uuid>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
// Fetch the job to verify it exists and check ownership. // Fetch the job to verify it exists and check ownership.
let row: Option<(String, Option<Uuid>)> = sqlx::query_as( let row: Option<(String, Option<Uuid>)> =
"SELECT status::text, created_by_user_id FROM patch_jobs WHERE id = $1", sqlx::query_as("SELECT status::text, created_by_user_id FROM patch_jobs WHERE id = $1")
) .bind(id)
.bind(id) .fetch_optional(&state.db)
.fetch_optional(&state.db) .await
.await .map_err(|e| {
.map_err(|e| { tracing::error!(error = %e, %id, "cancel_job: db fetch failed");
tracing::error!(error = %e, %id, "cancel_job: db fetch failed"); err(
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") StatusCode::INTERNAL_SERVER_ERROR,
})?; "internal_error",
"Database error",
)
})?;
let (status_str, creator_id) = row.ok_or_else(|| { let (status_str, creator_id) =
err(StatusCode::NOT_FOUND, "not_found", "Job not found") row.ok_or_else(|| err(StatusCode::NOT_FOUND, "not_found", "Job not found"))?;
})?;
// Only admin or the job creator may cancel. // Only admin or the job creator may cancel.
if !auth.role.is_admin() { if !auth.role.is_admin() {
@ -437,16 +454,18 @@ async fn cancel_job(
} }
// Cancel the parent job. // Cancel the parent job.
sqlx::query( sqlx::query("UPDATE patch_jobs SET status = 'cancelled'::job_status WHERE id = $1")
"UPDATE patch_jobs SET status = 'cancelled'::job_status WHERE id = $1", .bind(id)
) .execute(&state.db)
.bind(id) .await
.execute(&state.db) .map_err(|e| {
.await tracing::error!(error = %e, %id, "cancel_job: update patch_jobs failed");
.map_err(|e| { err(
tracing::error!(error = %e, %id, "cancel_job: update patch_jobs failed"); StatusCode::INTERNAL_SERVER_ERROR,
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") "internal_error",
})?; "Database error",
)
})?;
// Cancel all queued/pending host rows for this job. // Cancel all queued/pending host rows for this job.
sqlx::query( sqlx::query(
@ -462,7 +481,11 @@ async fn cancel_job(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, %id, "cancel_job: update patch_job_hosts failed"); tracing::error!(error = %e, %id, "cancel_job: update patch_job_hosts failed");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
log_event( log_event(
@ -506,7 +529,11 @@ async fn rollback_job(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, %id, "rollback_job: existence check failed"); tracing::error!(error = %e, %id, "rollback_job: existence check failed");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
if !original_exists { if !original_exists {
@ -521,7 +548,11 @@ async fn rollback_job(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, %id, "rollback_job: host fetch failed"); tracing::error!(error = %e, %id, "rollback_job: host fetch failed");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
if host_ids.is_empty() { if host_ids.is_empty() {
@ -552,7 +583,11 @@ async fn rollback_job(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, parent_job_id = %id, "rollback_job: insert failed"); tracing::error!(error = %e, parent_job_id = %id, "rollback_job: insert failed");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
// Replicate host list into the rollback job. // Replicate host list into the rollback job.
@ -573,7 +608,11 @@ async fn rollback_job(
error = %e, %rollback_job_id, %host_id, error = %e, %rollback_job_id, %host_id,
"rollback_job: insert patch_job_hosts failed" "rollback_job: insert patch_job_hosts failed"
); );
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
} }

View File

@ -15,9 +15,7 @@ use axum::{
use pm_auth::rbac::AuthUser; use pm_auth::rbac::AuthUser;
use pm_core::{ use pm_core::{
audit::{log_event, AuditAction}, audit::{log_event, AuditAction},
models::{ models::{CreateMaintenanceWindowRequest, MaintenanceWindow, UpdateMaintenanceWindowRequest},
CreateMaintenanceWindowRequest, MaintenanceWindow, UpdateMaintenanceWindowRequest,
},
}; };
use serde_json::{json, Value}; use serde_json::{json, Value};
use uuid::Uuid; use uuid::Uuid;
@ -56,15 +54,18 @@ async fn list_windows(
Path(host_id): Path<Uuid>, Path(host_id): Path<Uuid>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
// Verify host exists. // Verify host exists.
let host_exists: bool = let host_exists: bool = sqlx::query_scalar("SELECT EXISTS(SELECT 1 FROM hosts WHERE id = $1)")
sqlx::query_scalar("SELECT EXISTS(SELECT 1 FROM hosts WHERE id = $1)") .bind(host_id)
.bind(host_id) .fetch_one(&state.db)
.fetch_one(&state.db) .await
.await .map_err(|e| {
.map_err(|e| { tracing::error!(error = %e, %host_id, "list_windows: host existence check failed");
tracing::error!(error = %e, %host_id, "list_windows: host existence check failed"); err(
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") StatusCode::INTERNAL_SERVER_ERROR,
})?; "internal_error",
"Database error",
)
})?;
if !host_exists { if !host_exists {
return Err(err(StatusCode::NOT_FOUND, "not_found", "Host not found")); return Err(err(StatusCode::NOT_FOUND, "not_found", "Host not found"));
@ -84,7 +85,11 @@ async fn list_windows(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, %host_id, "list_windows: query failed"); tracing::error!(error = %e, %host_id, "list_windows: query failed");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
Ok(Json(json!({ "windows": windows }))) Ok(Json(json!({ "windows": windows })))
@ -101,41 +106,44 @@ async fn create_window(
// Validate: weekly requires recurrence_day 0-6 // Validate: weekly requires recurrence_day 0-6
if req.recurrence == pm_core::models::WindowRecurrence::Weekly { if req.recurrence == pm_core::models::WindowRecurrence::Weekly {
match req.recurrence_day { match req.recurrence_day {
Some(d) if (0..=6).contains(&d) => {} Some(d) if (0..=6).contains(&d) => {},
_ => { _ => {
return Err(err( return Err(err(
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
"bad_request", "bad_request",
"Weekly recurrence requires recurrence_day 0-6 (0=Sunday)", "Weekly recurrence requires recurrence_day 0-6 (0=Sunday)",
)); ));
} },
} }
} }
// Validate: monthly requires recurrence_day 1-31 // Validate: monthly requires recurrence_day 1-31
if req.recurrence == pm_core::models::WindowRecurrence::Monthly { if req.recurrence == pm_core::models::WindowRecurrence::Monthly {
match req.recurrence_day { match req.recurrence_day {
Some(d) if (1..=31).contains(&d) => {} Some(d) if (1..=31).contains(&d) => {},
_ => { _ => {
return Err(err( return Err(err(
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
"bad_request", "bad_request",
"Monthly recurrence requires recurrence_day 1-31", "Monthly recurrence requires recurrence_day 1-31",
)); ));
} },
} }
} }
// Verify host exists. // Verify host exists.
let host_exists: bool = let host_exists: bool = sqlx::query_scalar("SELECT EXISTS(SELECT 1 FROM hosts WHERE id = $1)")
sqlx::query_scalar("SELECT EXISTS(SELECT 1 FROM hosts WHERE id = $1)") .bind(host_id)
.bind(host_id) .fetch_one(&state.db)
.fetch_one(&state.db) .await
.await .map_err(|e| {
.map_err(|e| { tracing::error!(error = %e, %host_id, "create_window: host existence check failed");
tracing::error!(error = %e, %host_id, "create_window: host existence check failed"); err(
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") StatusCode::INTERNAL_SERVER_ERROR,
})?; "internal_error",
"Database error",
)
})?;
if !host_exists { if !host_exists {
return Err(err(StatusCode::NOT_FOUND, "not_found", "Host not found")); return Err(err(StatusCode::NOT_FOUND, "not_found", "Host not found"));
@ -165,7 +173,11 @@ async fn create_window(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, %host_id, "create_window: insert failed"); tracing::error!(error = %e, %host_id, "create_window: insert failed");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
log_event( log_event(
@ -219,44 +231,52 @@ async fn update_window(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, %win_id, "update_window: fetch failed"); tracing::error!(error = %e, %win_id, "update_window: fetch failed");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
let existing = existing.ok_or_else(|| { let existing = existing.ok_or_else(|| {
err(StatusCode::NOT_FOUND, "not_found", "Maintenance window not found") err(
StatusCode::NOT_FOUND,
"not_found",
"Maintenance window not found",
)
})?; })?;
// Apply partial updates using existing values as defaults. // Apply partial updates using existing values as defaults.
let new_label = req.label.unwrap_or(existing.label); let new_label = req.label.unwrap_or(existing.label);
let new_recurrence = req.recurrence.unwrap_or(existing.recurrence); let new_recurrence = req.recurrence.unwrap_or(existing.recurrence);
let new_start_at = req.start_at.unwrap_or(existing.start_at); let new_start_at = req.start_at.unwrap_or(existing.start_at);
let new_duration = req.duration_minutes.unwrap_or(existing.duration_minutes); let new_duration = req.duration_minutes.unwrap_or(existing.duration_minutes);
let new_rec_day = req.recurrence_day.or(existing.recurrence_day); let new_rec_day = req.recurrence_day.or(existing.recurrence_day);
let new_enabled = req.enabled.unwrap_or(existing.enabled); let new_enabled = req.enabled.unwrap_or(existing.enabled);
// Validate recurrence_day for the final recurrence type. // Validate recurrence_day for the final recurrence type.
if new_recurrence == pm_core::models::WindowRecurrence::Weekly { if new_recurrence == pm_core::models::WindowRecurrence::Weekly {
match new_rec_day { match new_rec_day {
Some(d) if (0..=6).contains(&d) => {} Some(d) if (0..=6).contains(&d) => {},
_ => { _ => {
return Err(err( return Err(err(
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
"bad_request", "bad_request",
"Weekly recurrence requires recurrence_day 0-6", "Weekly recurrence requires recurrence_day 0-6",
)); ));
} },
} }
} }
if new_recurrence == pm_core::models::WindowRecurrence::Monthly { if new_recurrence == pm_core::models::WindowRecurrence::Monthly {
match new_rec_day { match new_rec_day {
Some(d) if (1..=31).contains(&d) => {} Some(d) if (1..=31).contains(&d) => {},
_ => { _ => {
return Err(err( return Err(err(
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
"bad_request", "bad_request",
"Monthly recurrence requires recurrence_day 1-31", "Monthly recurrence requires recurrence_day 1-31",
)); ));
} },
} }
} }
@ -287,7 +307,11 @@ async fn update_window(
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e, %win_id, "update_window: update failed"); tracing::error!(error = %e, %win_id, "update_window: update failed");
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") err(
StatusCode::INTERNAL_SERVER_ERROR,
"internal_error",
"Database error",
)
})?; })?;
log_event( log_event(
@ -320,17 +344,19 @@ async fn delete_window(
auth: AuthUser, auth: AuthUser,
Path((host_id, win_id)): Path<(Uuid, Uuid)>, Path((host_id, win_id)): Path<(Uuid, Uuid)>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
let result = sqlx::query( let result = sqlx::query("DELETE FROM maintenance_windows WHERE id = $1 AND host_id = $2")
"DELETE FROM maintenance_windows WHERE id = $1 AND host_id = $2", .bind(win_id)
) .bind(host_id)
.bind(win_id) .execute(&state.db)
.bind(host_id) .await
.execute(&state.db) .map_err(|e| {
.await tracing::error!(error = %e, %win_id, "delete_window: delete failed");
.map_err(|e| { err(
tracing::error!(error = %e, %win_id, "delete_window: delete failed"); StatusCode::INTERNAL_SERVER_ERROR,
err(StatusCode::INTERNAL_SERVER_ERROR, "internal_error", "Database error") "internal_error",
})?; "Database error",
)
})?;
if result.rows_affected() == 0 { if result.rows_affected() == 0 {
return Err(err( return Err(err(

View File

@ -1,15 +1,15 @@
//! Route modules for the pm-web API. //! Route modules for the pm-web API.
pub mod auth; pub mod auth;
pub mod azure_sso;
pub mod ca; pub mod ca;
pub mod discovery; pub mod discovery;
pub mod groups; pub mod groups;
pub mod hosts; pub mod hosts;
pub mod maintenance_windows;
pub mod jobs; pub mod jobs;
pub mod maintenance_windows;
pub mod settings;
pub mod status; pub mod status;
pub mod users; pub mod users;
pub mod settings;
pub mod azure_sso;
pub mod ws; pub mod ws;
pub mod reports; pub mod reports;

View File

@ -28,10 +28,10 @@ struct ReportQuery {
pub fn router() -> Router<AppState> { pub fn router() -> Router<AppState> {
Router::new() Router::new()
.route("/compliance", get(compliance_report)) .route("/compliance", get(compliance_report))
.route("/patch-history", get(patch_history_report)) .route("/patch-history", get(patch_history_report))
.route("/vulnerability", get(vulnerability_report)) .route("/vulnerability", get(vulnerability_report))
.route("/audit", get(audit_report)) .route("/audit", get(audit_report))
} }
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
@ -58,21 +58,22 @@ async fn run_report(
match result { match result {
Ok(bytes) => { Ok(bytes) => {
let mut headers = HeaderMap::new(); let mut headers = HeaderMap::new();
headers.insert( headers.insert(header::CONTENT_TYPE, HeaderValue::from_static(ct));
header::CONTENT_TYPE,
HeaderValue::from_static(ct),
);
headers.insert( headers.insert(
header::CONTENT_DISPOSITION, header::CONTENT_DISPOSITION,
HeaderValue::from_str(&disposition) HeaderValue::from_str(&disposition)
.unwrap_or_else(|_| HeaderValue::from_static("attachment")), .unwrap_or_else(|_| HeaderValue::from_static("attachment")),
); );
(headers, Bytes::from(bytes)).into_response() (headers, Bytes::from(bytes)).into_response()
} },
Err(e) => { Err(e) => {
tracing::error!(error = %e, "report generation failed"); tracing::error!(error = %e, "report generation failed");
(StatusCode::INTERNAL_SERVER_ERROR, format!("Report error: {}", e)).into_response() (
} StatusCode::INTERNAL_SERVER_ERROR,
format!("Report error: {}", e),
)
.into_response()
},
} }
} }
@ -92,10 +93,13 @@ async fn compliance_report(
}; };
let use_pdf = matches!(q.format.as_deref(), Some("pdf")); let use_pdf = matches!(q.format.as_deref(), Some("pdf"));
run_report( run_report(
state.db, params, use_pdf, state.db,
params,
use_pdf,
"compliance-report.csv", "compliance-report.csv",
"compliance-report.pdf", "compliance-report.pdf",
).await )
.await
} }
async fn patch_history_report( async fn patch_history_report(
@ -110,10 +114,13 @@ async fn patch_history_report(
}; };
let use_pdf = matches!(q.format.as_deref(), Some("pdf")); let use_pdf = matches!(q.format.as_deref(), Some("pdf"));
run_report( run_report(
state.db, params, use_pdf, state.db,
params,
use_pdf,
"patch-history-report.csv", "patch-history-report.csv",
"patch-history-report.pdf", "patch-history-report.pdf",
).await )
.await
} }
async fn vulnerability_report( async fn vulnerability_report(
@ -128,16 +135,16 @@ async fn vulnerability_report(
}; };
let use_pdf = matches!(q.format.as_deref(), Some("pdf")); let use_pdf = matches!(q.format.as_deref(), Some("pdf"));
run_report( run_report(
state.db, params, use_pdf, state.db,
params,
use_pdf,
"vulnerability-report.csv", "vulnerability-report.csv",
"vulnerability-report.pdf", "vulnerability-report.pdf",
).await )
.await
} }
async fn audit_report( async fn audit_report(State(state): State<AppState>, Query(q): Query<ReportQuery>) -> Response {
State(state): State<AppState>,
Query(q): Query<ReportQuery>,
) -> Response {
let params = ReportParams { let params = ReportParams {
report_type: ReportType::Audit, report_type: ReportType::Audit,
from: q.from, from: q.from,
@ -146,8 +153,11 @@ async fn audit_report(
}; };
let use_pdf = matches!(q.format.as_deref(), Some("pdf")); let use_pdf = matches!(q.format.as_deref(), Some("pdf"));
run_report( run_report(
state.db, params, use_pdf, state.db,
params,
use_pdf,
"audit-report.csv", "audit-report.csv",
"audit-report.pdf", "audit-report.pdf",
).await )
.await
} }

View File

@ -20,8 +20,8 @@ use lettre::{
transport::smtp::authentication::Credentials, transport::smtp::authentication::Credentials,
AsyncSmtpTransport, AsyncTransport, Message, Tokio1Executor, AsyncSmtpTransport, AsyncTransport, Message, Tokio1Executor,
}; };
use pm_core::audit::{log_event, verify_integrity, AuditAction};
use pm_auth::rbac::AuthUser; use pm_auth::rbac::AuthUser;
use pm_core::audit::{log_event, verify_integrity, AuditAction};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use serde_json::{json, Value}; use serde_json::{json, Value};
use std::collections::HashMap; use std::collections::HashMap;
@ -132,7 +132,10 @@ pub fn router() -> Router<AppState> {
.route("/", get(get_settings).put(update_settings)) .route("/", get(get_settings).put(update_settings))
.route("/azure-sso/test", post(test_azure_sso)) .route("/azure-sso/test", post(test_azure_sso))
.route("/smtp/test", post(test_smtp)) .route("/smtp/test", post(test_smtp))
.route("/ip-whitelist", get(get_ip_whitelist).put(update_ip_whitelist)) .route(
"/ip-whitelist",
get(get_ip_whitelist).put(update_ip_whitelist),
)
.route("/audit-integrity", post(audit_integrity)) .route("/audit-integrity", post(audit_integrity))
} }
@ -155,26 +158,28 @@ fn admin_only(auth: &AuthUser) -> Result<(), (StatusCode, Json<Value>)> {
async fn load_system_config( async fn load_system_config(
pool: &sqlx::PgPool, pool: &sqlx::PgPool,
) -> Result<HashMap<String, String>, (StatusCode, Json<Value>)> { ) -> Result<HashMap<String, String>, (StatusCode, Json<Value>)> {
let rows: Vec<(String, String)> = sqlx::query_as( let rows: Vec<(String, String)> = sqlx::query_as("SELECT key, value FROM system_config")
"SELECT key, value FROM system_config", .fetch_all(pool)
) .await
.fetch_all(pool) .map_err(|e| {
.await tracing::error!(error = %e, "Failed to load system_config");
.map_err(|e| { (
tracing::error!(error = %e, "Failed to load system_config"); StatusCode::INTERNAL_SERVER_ERROR,
( Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })),
StatusCode::INTERNAL_SERVER_ERROR, )
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })), })?;
)
})?;
Ok(rows.into_iter().collect()) Ok(rows.into_iter().collect())
} }
fn build_settings_response(cfg: &HashMap<String, String>, azure: AzureSsoConfig) -> SettingsResponse { fn build_settings_response(
cfg: &HashMap<String, String>,
azure: AzureSsoConfig,
) -> SettingsResponse {
let get = |key: &str| -> String { cfg.get(key).cloned().unwrap_or_default() }; let get = |key: &str| -> String { cfg.get(key).cloned().unwrap_or_default() };
let recipients: Vec<String> = serde_json::from_str(&get("notification_email_recipients")).unwrap_or_default(); let recipients: Vec<String> =
serde_json::from_str(&get("notification_email_recipients")).unwrap_or_default();
SettingsResponse { SettingsResponse {
azure_sso: azure, azure_sso: azure,
@ -517,7 +522,7 @@ async fn test_azure_sso(
"success": false, "success": false,
"message": "Azure SSO is not configured" "message": "Azure SSO is not configured"
}))); })));
} },
}; };
if tenant_id.is_empty() { if tenant_id.is_empty() {
@ -560,7 +565,7 @@ async fn test_azure_sso(
"issuer": issuer "issuer": issuer
}))) })))
} }
} },
Ok(resp) => Ok(Json(json!({ Ok(resp) => Ok(Json(json!({
"success": false, "success": false,
"message": format!("Failed to reach Azure AD: HTTP {}", resp.status()) "message": format!("Failed to reach Azure AD: HTTP {}", resp.status())
@ -593,11 +598,17 @@ async fn test_smtp(
} }
let host = cfg.get("smtp_host").cloned().unwrap_or_default(); let host = cfg.get("smtp_host").cloned().unwrap_or_default();
let port: u16 = cfg.get("smtp_port").and_then(|v| v.parse().ok()).unwrap_or(587); let port: u16 = cfg
.get("smtp_port")
.and_then(|v| v.parse().ok())
.unwrap_or(587);
let username = cfg.get("smtp_username").cloned().unwrap_or_default(); let username = cfg.get("smtp_username").cloned().unwrap_or_default();
let password = cfg.get("smtp_password").cloned().unwrap_or_default(); let password = cfg.get("smtp_password").cloned().unwrap_or_default();
let from_addr = cfg.get("smtp_from").cloned().unwrap_or_default(); let from_addr = cfg.get("smtp_from").cloned().unwrap_or_default();
let tls_mode = cfg.get("smtp_tls_mode").cloned().unwrap_or_else(|| "starttls".to_string()); let tls_mode = cfg
.get("smtp_tls_mode")
.cloned()
.unwrap_or_else(|| "starttls".to_string());
if host.is_empty() || from_addr.is_empty() { if host.is_empty() || from_addr.is_empty() {
return Ok(Json(json!({ return Ok(Json(json!({
@ -628,7 +639,9 @@ async fn send_smtp_test(
from_addr: &str, from_addr: &str,
tls_mode: &str, tls_mode: &str,
) -> Result<(), String> { ) -> Result<(), String> {
let from_mailbox: Mailbox = from_addr.parse().map_err(|e| format!("Invalid from address: {}", e))?; let from_mailbox: Mailbox = from_addr
.parse()
.map_err(|e| format!("Invalid from address: {}", e))?;
let email = Message::builder() let email = Message::builder()
.from(from_mailbox.clone()) .from(from_mailbox.clone())
@ -644,33 +657,39 @@ async fn send_smtp_test(
.map_err(|e| format!("TLS relay error: {}", e))?; .map_err(|e| format!("TLS relay error: {}", e))?;
builder = builder.port(port); builder = builder.port(port);
if !username.is_empty() { if !username.is_empty() {
builder = builder.credentials(Credentials::new(username.to_string(), password.to_string())); builder = builder
.credentials(Credentials::new(username.to_string(), password.to_string()));
} }
let transport = builder.build(); let transport = builder.build();
transport.send(email).await transport.send(email).await
} },
"starttls" => { "starttls" => {
let mut builder = AsyncSmtpTransport::<Tokio1Executor>::starttls_relay(host) let mut builder = AsyncSmtpTransport::<Tokio1Executor>::starttls_relay(host)
.map_err(|e| format!("STARTTLS relay error: {}", e))?; .map_err(|e| format!("STARTTLS relay error: {}", e))?;
builder = builder.port(port); builder = builder.port(port);
if !username.is_empty() { if !username.is_empty() {
builder = builder.credentials(Credentials::new(username.to_string(), password.to_string())); builder = builder
.credentials(Credentials::new(username.to_string(), password.to_string()));
} }
let transport = builder.build(); let transport = builder.build();
transport.send(email).await transport.send(email).await
} },
_ => { _ => {
// "none" — plaintext / no TLS // "none" — plaintext / no TLS
let mut builder = AsyncSmtpTransport::<Tokio1Executor>::builder_dangerous(host).port(port); let mut builder =
AsyncSmtpTransport::<Tokio1Executor>::builder_dangerous(host).port(port);
if !username.is_empty() { if !username.is_empty() {
builder = builder.credentials(Credentials::new(username.to_string(), password.to_string())); builder = builder
.credentials(Credentials::new(username.to_string(), password.to_string()));
} }
let transport = builder.build(); let transport = builder.build();
transport.send(email).await transport.send(email).await
} },
}; };
result.map(|_| ()).map_err(|e| format!("SMTP send error: {}", e)) result
.map(|_| ())
.map_err(|e| format!("SMTP send error: {}", e))
} }
// ============================================================ // ============================================================
@ -713,12 +732,12 @@ async fn update_ip_whitelist(
// Validate each entry // Validate each entry
for entry in &req.entries { for entry in &req.entries {
if entry.parse::<ipnet::IpNet>().is_err() if entry.parse::<ipnet::IpNet>().is_err() && entry.parse::<std::net::IpAddr>().is_err() {
&& entry.parse::<std::net::IpAddr>().is_err()
{
return Err(( return Err((
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
Json(json!({ "error": { "code": "bad_request", "message": format!("Invalid CIDR or IP: {}", entry) } })), Json(
json!({ "error": { "code": "bad_request", "message": format!("Invalid CIDR or IP: {}", entry) } }),
),
)); ));
} }
} }

View File

@ -2,13 +2,7 @@
//! //!
//! GET /api/v1/status/fleet — aggregate health and patch summary across all hosts. //! GET /api/v1/status/fleet — aggregate health and patch summary across all hosts.
use axum::{ use axum::{extract::State, http::StatusCode, response::Json, routing::get, Router};
extract::State,
http::StatusCode,
response::Json,
routing::get,
Router,
};
use serde::Serialize; use serde::Serialize;
use serde_json::{json, Value}; use serde_json::{json, Value};

View File

@ -15,11 +15,11 @@ use axum::{
routing::{delete, get, post, put}, routing::{delete, get, post, put},
Router, Router,
}; };
use pm_auth::{hash_password, rbac::AuthUser, session::force_logout};
use pm_core::{ use pm_core::{
audit::{log_event, AuditAction}, audit::{log_event, AuditAction},
models::{User, CreateUserRequest, UpdateUserRequest}, models::{CreateUserRequest, UpdateUserRequest, User},
}; };
use pm_auth::{hash_password, rbac::AuthUser, session::force_logout};
use serde_json::{json, Value}; use serde_json::{json, Value};
use uuid::Uuid; use uuid::Uuid;
@ -38,7 +38,10 @@ async fn list_users(
auth: AuthUser, auth: AuthUser,
) -> Result<Json<Vec<User>>, (StatusCode, Json<Value>)> { ) -> Result<Json<Vec<User>>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
sqlx::query_as::<_, User>( sqlx::query_as::<_, User>(
@ -52,7 +55,10 @@ async fn list_users(
.map(Json) .map(Json)
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e); tracing::error!(error = %e);
(StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": "Database error" } }))) (
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })),
)
}) })
} }
@ -62,14 +68,24 @@ async fn create_user(
Json(req): Json<CreateUserRequest>, Json(req): Json<CreateUserRequest>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
let hash = hash_password(&req.password).map_err(|e| { let hash = hash_password(&req.password).map_err(|e| {
(StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } }))) (
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } })),
)
})?; })?;
let role = if req.role == "admin" { "admin" } else { "operator" }; let role = if req.role == "admin" {
"admin"
} else {
"operator"
};
let id: Uuid = sqlx::query_scalar( let id: Uuid = sqlx::query_scalar(
r#"INSERT INTO users (username, display_name, email, role, auth_provider, password_hash) r#"INSERT INTO users (username, display_name, email, role, auth_provider, password_hash)
@ -84,12 +100,29 @@ async fn create_user(
.fetch_one(&state.db) .fetch_one(&state.db)
.await .await
.map_err(|e| { .map_err(|e| {
let msg = if e.to_string().contains("unique") { "Username or email already exists".to_string() } else { "Database error".to_string() }; let msg = if e.to_string().contains("unique") {
(StatusCode::CONFLICT, Json(json!({ "error": { "code": "conflict", "message": msg } }))) "Username or email already exists".to_string()
} else {
"Database error".to_string()
};
(
StatusCode::CONFLICT,
Json(json!({ "error": { "code": "conflict", "message": msg } })),
)
})?; })?;
log_event(&state.db, AuditAction::UserCreated, Some(auth.user_id), Some(&auth.username), log_event(
Some("user"), Some(&id.to_string()), json!({ "username": req.username }), None, None).await; &state.db,
AuditAction::UserCreated,
Some(auth.user_id),
Some(&auth.username),
Some("user"),
Some(&id.to_string()),
json!({ "username": req.username }),
None,
None,
)
.await;
Ok(Json(json!({ "id": id, "message": "User created" }))) Ok(Json(json!({ "id": id, "message": "User created" })))
} }
@ -108,12 +141,18 @@ async fn get_user(
) -> Result<Json<User>, (StatusCode, Json<Value>)> { ) -> Result<Json<User>, (StatusCode, Json<Value>)> {
// Users can see themselves; admin can see anyone // Users can see themselves; admin can see anyone
if !auth.role.is_admin() && auth.user_id != id { if !auth.role.is_admin() && auth.user_id != id {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Access denied" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Access denied" } })),
));
} }
fetch_user(&state.db, id).await fetch_user(&state.db, id).await
} }
async fn fetch_user(pool: &sqlx::PgPool, id: Uuid) -> Result<Json<User>, (StatusCode, Json<Value>)> { async fn fetch_user(
pool: &sqlx::PgPool,
id: Uuid,
) -> Result<Json<User>, (StatusCode, Json<Value>)> {
let user: Option<User> = sqlx::query_as( let user: Option<User> = sqlx::query_as(
r#"SELECT id, username, display_name, email, role, auth_provider, r#"SELECT id, username, display_name, email, role, auth_provider,
mfa_enabled, is_active, force_password_reset, last_login_at, mfa_enabled, is_active, force_password_reset, last_login_at,
@ -125,10 +164,18 @@ async fn fetch_user(pool: &sqlx::PgPool, id: Uuid) -> Result<Json<User>, (Status
.await .await
.map_err(|e| { .map_err(|e| {
tracing::error!(error = %e); tracing::error!(error = %e);
(StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": "Database error" } }))) (
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": "Database error" } })),
)
})?; })?;
user.map(Json).ok_or_else(|| (StatusCode::NOT_FOUND, Json(json!({ "error": { "code": "not_found", "message": "User not found" } })))) user.map(Json).ok_or_else(|| {
(
StatusCode::NOT_FOUND,
Json(json!({ "error": { "code": "not_found", "message": "User not found" } })),
)
})
} }
async fn update_user( async fn update_user(
@ -138,14 +185,25 @@ async fn update_user(
Json(req): Json<UpdateUserRequest>, Json(req): Json<UpdateUserRequest>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() && auth.user_id != id { if !auth.role.is_admin() && auth.user_id != id {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Access denied" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Access denied" } })),
));
} }
// Only admins can change role or active status // Only admins can change role or active status
if (req.role.is_some() || req.is_active.is_some()) && !auth.role.is_admin() { if (req.role.is_some() || req.is_active.is_some()) && !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required to change role or status" } })))); return Err((
StatusCode::FORBIDDEN,
Json(
json!({ "error": { "code": "forbidden", "message": "Admin role required to change role or status" } }),
),
));
} }
let role_str = req.role.as_deref().map(|r| if r == "admin" { "admin" } else { "operator" }); let role_str = req
.role
.as_deref()
.map(|r| if r == "admin" { "admin" } else { "operator" });
let rows = sqlx::query( let rows = sqlx::query(
r#"UPDATE users SET r#"UPDATE users SET
@ -163,15 +221,33 @@ async fn update_user(
.bind(id) .bind(id)
.execute(&state.db) .execute(&state.db)
.await .await
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } }))))? .map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } })),
)
})?
.rows_affected(); .rows_affected();
if rows == 0 { if rows == 0 {
return Err((StatusCode::NOT_FOUND, Json(json!({ "error": { "code": "not_found", "message": "User not found" } })))); return Err((
StatusCode::NOT_FOUND,
Json(json!({ "error": { "code": "not_found", "message": "User not found" } })),
));
} }
log_event(&state.db, AuditAction::UserUpdated, Some(auth.user_id), Some(&auth.username), log_event(
Some("user"), Some(&id.to_string()), json!({}), None, None).await; &state.db,
AuditAction::UserUpdated,
Some(auth.user_id),
Some(&auth.username),
Some("user"),
Some(&id.to_string()),
json!({}),
None,
None,
)
.await;
Ok(Json(json!({ "message": "User updated" }))) Ok(Json(json!({ "message": "User updated" })))
} }
@ -182,23 +258,51 @@ async fn delete_user(
Path(id): Path<Uuid>, Path(id): Path<Uuid>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
if auth.user_id == id { if auth.user_id == id {
return Err((StatusCode::BAD_REQUEST, Json(json!({ "error": { "code": "bad_request", "message": "Cannot delete your own account" } })))); return Err((
StatusCode::BAD_REQUEST,
Json(
json!({ "error": { "code": "bad_request", "message": "Cannot delete your own account" } }),
),
));
} }
let rows = sqlx::query("DELETE FROM users WHERE id = $1") let rows = sqlx::query("DELETE FROM users WHERE id = $1")
.bind(id).execute(&state.db).await .bind(id)
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } }))))? .execute(&state.db)
.await
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } })),
)
})?
.rows_affected(); .rows_affected();
if rows == 0 { if rows == 0 {
return Err((StatusCode::NOT_FOUND, Json(json!({ "error": { "code": "not_found", "message": "User not found" } })))); return Err((
StatusCode::NOT_FOUND,
Json(json!({ "error": { "code": "not_found", "message": "User not found" } })),
));
} }
log_event(&state.db, AuditAction::UserDeleted, Some(auth.user_id), Some(&auth.username), log_event(
Some("user"), Some(&id.to_string()), json!({}), None, None).await; &state.db,
AuditAction::UserDeleted,
Some(auth.user_id),
Some(&auth.username),
Some("user"),
Some(&id.to_string()),
json!({}),
None,
None,
)
.await;
Ok(Json(json!({ "message": "User deleted" }))) Ok(Json(json!({ "message": "User deleted" })))
} }
@ -209,11 +313,20 @@ async fn revoke_user_sessions(
Path(id): Path<Uuid>, Path(id): Path<Uuid>,
) -> Result<Json<Value>, (StatusCode, Json<Value>)> { ) -> Result<Json<Value>, (StatusCode, Json<Value>)> {
if !auth.role.is_admin() { if !auth.role.is_admin() {
return Err((StatusCode::FORBIDDEN, Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })))); return Err((
StatusCode::FORBIDDEN,
Json(json!({ "error": { "code": "forbidden", "message": "Admin role required" } })),
));
} }
let count = force_logout(&state.db, id).await let count = force_logout(&state.db, id).await.map_err(|e| {
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } }))))?; (
StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": { "code": "internal_error", "message": e.to_string() } })),
)
})?;
Ok(Json(json!({ "message": "Sessions revoked", "count": count }))) Ok(Json(
json!({ "message": "Sessions revoked", "count": count }),
))
} }

View File

@ -4,8 +4,8 @@
//! GET /api/v1/ws/jobs — browser WebSocket endpoint (ticket-authenticated) //! GET /api/v1/ws/jobs — browser WebSocket endpoint (ticket-authenticated)
use axum::{ use axum::{
extract::{Query, State, WebSocketUpgrade},
extract::ws::{Message, WebSocket}, extract::ws::{Message, WebSocket},
extract::{Query, State, WebSocketUpgrade},
http::StatusCode, http::StatusCode,
response::{Json, Response}, response::{Json, Response},
routing::{get, post}, routing::{get, post},
@ -59,7 +59,6 @@ fn err(
// ── POST /api/v1/ws/ticket ──────────────────────────────────────────────────── // ── POST /api/v1/ws/ticket ────────────────────────────────────────────────────
/// Issue a single-use WebSocket authentication ticket (60 s expiry). /// Issue a single-use WebSocket authentication ticket (60 s expiry).
pub async fn create_ticket_handler( pub async fn create_ticket_handler(
State(state): State<AppState>, State(state): State<AppState>,
@ -109,7 +108,7 @@ pub async fn ws_handler(
"invalid_ticket", "invalid_ticket",
"WebSocket ticket not found or already used", "WebSocket ticket not found or already used",
)); ));
} },
Some(t) => { Some(t) => {
if t.expires_at < Utc::now() { if t.expires_at < Utc::now() {
drop(t); drop(t);
@ -121,7 +120,7 @@ pub async fn ws_handler(
)); ));
} }
t.clone() t.clone()
} },
} }
}; };
// Single-use: remove immediately after validation. // Single-use: remove immediately after validation.
@ -140,11 +139,7 @@ pub async fn ws_handler(
// ── WebSocket handler ───────────────────────────────────────────────────────── // ── WebSocket handler ─────────────────────────────────────────────────────────
/// Drive the browser WebSocket: LISTEN on `job_update` and forward payloads. /// Drive the browser WebSocket: LISTEN on `job_update` and forward payloads.
async fn handle_browser_ws( async fn handle_browser_ws(mut socket: WebSocket, db: sqlx::PgPool, ticket: WsTicket) {
mut socket: WebSocket,
db: sqlx::PgPool,
ticket: WsTicket,
) {
// Acquire a dedicated PG listener connection. // Acquire a dedicated PG listener connection.
let mut listener = match PgListener::connect_with(&db).await { let mut listener = match PgListener::connect_with(&db).await {
Ok(l) => l, Ok(l) => l,
@ -156,7 +151,7 @@ async fn handle_browser_ws(
)) ))
.await; .await;
return; return;
} },
}; };
if let Err(e) = listener.listen("job_update").await { if let Err(e) = listener.listen("job_update").await {

View File

@ -34,12 +34,12 @@ pub fn load_agent_certs(security: &SecurityConfig) -> anyhow::Result<AgentCerts>
})?; })?;
let ca_cert = std::fs::read(&security.ca_cert_path).map_err(|e| { let ca_cert = std::fs::read(&security.ca_cert_path).map_err(|e| {
anyhow::anyhow!( anyhow::anyhow!("Failed to read CA cert '{}': {}", security.ca_cert_path, e)
"Failed to read CA cert '{}': {}",
security.ca_cert_path,
e
)
})?; })?;
Ok(AgentCerts { client_cert, client_key, ca_cert }) Ok(AgentCerts {
client_cert,
client_key,
ca_cert,
})
} }

View File

@ -80,7 +80,8 @@ async fn load_notification_settings(pool: &PgPool) -> NotificationSettings {
.unwrap_or_default() .unwrap_or_default()
}; };
let recipients: Vec<String> = serde_json::from_str(&get("notification_email_recipients")).unwrap_or_default(); let recipients: Vec<String> =
serde_json::from_str(&get("notification_email_recipients")).unwrap_or_default();
NotificationSettings { NotificationSettings {
email_enabled: get("notification_email_enabled") == "true", email_enabled: get("notification_email_enabled") == "true",
@ -90,9 +91,7 @@ async fn load_notification_settings(pool: &PgPool) -> NotificationSettings {
} }
/// Build an async SMTP transport from settings. /// Build an async SMTP transport from settings.
fn build_transport( fn build_transport(settings: &SmtpSettings) -> Result<AsyncSmtpTransport<Tokio1Executor>, String> {
settings: &SmtpSettings,
) -> Result<AsyncSmtpTransport<Tokio1Executor>, String> {
match settings.tls_mode.as_str() { match settings.tls_mode.as_str() {
"tls" => { "tls" => {
let mut builder = AsyncSmtpTransport::<Tokio1Executor>::relay(&settings.host) let mut builder = AsyncSmtpTransport::<Tokio1Executor>::relay(&settings.host)
@ -105,7 +104,7 @@ fn build_transport(
)); ));
} }
Ok(builder.build()) Ok(builder.build())
} },
"starttls" => { "starttls" => {
let mut builder = AsyncSmtpTransport::<Tokio1Executor>::starttls_relay(&settings.host) let mut builder = AsyncSmtpTransport::<Tokio1Executor>::starttls_relay(&settings.host)
.map_err(|e| format!("STARTTLS relay error: {}", e))?; .map_err(|e| format!("STARTTLS relay error: {}", e))?;
@ -117,11 +116,12 @@ fn build_transport(
)); ));
} }
Ok(builder.build()) Ok(builder.build())
} },
_ => { _ => {
// "none" — plaintext / no TLS // "none" — plaintext / no TLS
let mut builder = AsyncSmtpTransport::<Tokio1Executor>::builder_dangerous(&settings.host) let mut builder =
.port(settings.port); AsyncSmtpTransport::<Tokio1Executor>::builder_dangerous(&settings.host)
.port(settings.port);
if !settings.username.is_empty() { if !settings.username.is_empty() {
builder = builder.credentials(Credentials::new( builder = builder.credentials(Credentials::new(
settings.username.clone(), settings.username.clone(),
@ -129,21 +129,17 @@ fn build_transport(
)); ));
} }
Ok(builder.build()) Ok(builder.build())
} },
} }
} }
/// Send an email notification. Returns true if the email was sent successfully. /// Send an email notification. Returns true if the email was sent successfully.
async fn send_email( async fn send_email(pool: &PgPool, subject: &str, body: &str) -> bool {
pool: &PgPool,
subject: &str,
body: &str,
) -> bool {
let smtp = match load_smtp_settings(pool).await { let smtp = match load_smtp_settings(pool).await {
s if !s.enabled => { s if !s.enabled => {
tracing::debug!("SMTP not enabled, skipping email notification"); tracing::debug!("SMTP not enabled, skipping email notification");
return false; return false;
} },
s => s, s => s,
}; };
@ -169,7 +165,7 @@ async fn send_email(
Err(e) => { Err(e) => {
tracing::error!(error = %e, "Invalid from address for email notification"); tracing::error!(error = %e, "Invalid from address for email notification");
return false; return false;
} },
}; };
let mut builder = Message::builder() let mut builder = Message::builder()
@ -184,7 +180,7 @@ async fn send_email(
Err(e) => { Err(e) => {
tracing::error!(error = %e, recipient = %recipient, "Invalid recipient address"); tracing::error!(error = %e, recipient = %recipient, "Invalid recipient address");
continue; continue;
} },
}; };
builder = builder.to(mailbox); builder = builder.to(mailbox);
} }
@ -194,7 +190,7 @@ async fn send_email(
Err(e) => { Err(e) => {
tracing::error!(error = %e, "Failed to build email message"); tracing::error!(error = %e, "Failed to build email message");
return false; return false;
} },
}; };
let transport = match build_transport(&smtp) { let transport = match build_transport(&smtp) {
@ -202,18 +198,18 @@ async fn send_email(
Err(e) => { Err(e) => {
tracing::error!(error = %e, "Failed to build SMTP transport"); tracing::error!(error = %e, "Failed to build SMTP transport");
return false; return false;
} },
}; };
match transport.send(email).await { match transport.send(email).await {
Ok(_) => { Ok(_) => {
tracing::info!(subject, "Email notification sent successfully"); tracing::info!(subject, "Email notification sent successfully");
true true
} },
Err(e) => { Err(e) => {
tracing::error!(error = %e, subject, "Failed to send email notification"); tracing::error!(error = %e, subject, "Failed to send email notification");
false false
} },
} }
} }
@ -300,7 +296,10 @@ pub async fn send_maintenance_window_reminder_email(
window_label: &str, window_label: &str,
start_at: &str, start_at: &str,
) { ) {
let subject = format!("[Patch Manager] Upcoming Maintenance Window: {}", window_label); let subject = format!(
"[Patch Manager] Upcoming Maintenance Window: {}",
window_label
);
let body = format!( let body = format!(
"Maintenance window reminder:\n\ "Maintenance window reminder:\n\
Host: {host_fqdn}\n\ Host: {host_fqdn}\n\

View File

@ -7,15 +7,9 @@
use std::sync::Arc; use std::sync::Arc;
use pm_agent_client::{AgentClient, AgentClientError}; use pm_agent_client::{AgentClient, AgentClientError};
use pm_core::{ use pm_core::{config::AppConfig, models::HostHealthStatus};
config::AppConfig,
models::HostHealthStatus,
};
use sqlx::{FromRow, PgPool}; use sqlx::{FromRow, PgPool};
use tokio::{ use tokio::{sync::Semaphore, time};
sync::Semaphore,
time,
};
use uuid::Uuid; use uuid::Uuid;
use crate::agent_loader::load_agent_certs; use crate::agent_loader::load_agent_certs;
@ -37,10 +31,7 @@ pub async fn run_health_poller(pool: PgPool, config: Arc<AppConfig>) {
let interval_secs = config.worker.health_poll_interval_secs; let interval_secs = config.worker.health_poll_interval_secs;
let mut ticker = time::interval(std::time::Duration::from_secs(interval_secs)); let mut ticker = time::interval(std::time::Duration::from_secs(interval_secs));
tracing::info!( tracing::info!(interval_secs, "Health poller started");
interval_secs,
"Health poller started"
);
loop { loop {
ticker.tick().await; ticker.tick().await;
@ -51,7 +42,7 @@ pub async fn run_health_poller(pool: PgPool, config: Arc<AppConfig>) {
Err(e) => { Err(e) => {
tracing::error!(error = %e, "Health poller: failed to load agent certs — skipping cycle"); tracing::error!(error = %e, "Health poller: failed to load agent certs — skipping cycle");
continue; continue;
} },
}; };
let client_cert = Arc::new(certs.client_cert); let client_cert = Arc::new(certs.client_cert);
@ -69,7 +60,7 @@ pub async fn run_health_poller(pool: PgPool, config: Arc<AppConfig>) {
Err(e) => { Err(e) => {
tracing::error!(error = %e, "Health poller: failed to fetch hosts"); tracing::error!(error = %e, "Health poller: failed to fetch hosts");
continue; continue;
} },
}; };
if hosts.is_empty() { if hosts.is_empty() {
@ -107,7 +98,7 @@ pub async fn run_health_poller(pool: PgPool, config: Arc<AppConfig>) {
Ok(HostHealthStatus::Healthy) => healthy += 1, Ok(HostHealthStatus::Healthy) => healthy += 1,
Ok(HostHealthStatus::Degraded) => degraded += 1, Ok(HostHealthStatus::Degraded) => degraded += 1,
Ok(HostHealthStatus::Unreachable) => unreachable += 1, Ok(HostHealthStatus::Unreachable) => unreachable += 1,
Ok(_) => {} Ok(_) => {},
Err(e) => tracing::error!(error = %e, "Health poller task panicked"), Err(e) => tracing::error!(error = %e, "Health poller task panicked"),
} }
} }
@ -144,25 +135,37 @@ async fn poll_host_health(
error = %e, error = %e,
"Health poller: failed to build AgentClient" "Health poller: failed to build AgentClient"
); );
(HostHealthStatus::Unreachable, serde_json::Value::Object(Default::default())) (
} HostHealthStatus::Unreachable,
serde_json::Value::Object(Default::default()),
)
},
Ok(client) => match client.health().await { Ok(client) => match client.health().await {
Ok(data) => { Ok(data) => {
let payload = serde_json::to_value(&data).unwrap_or_default(); let payload = serde_json::to_value(&data).unwrap_or_default();
(HostHealthStatus::Healthy, payload) (HostHealthStatus::Healthy, payload)
} },
Err(AgentClientError::Timeout) => { Err(AgentClientError::Timeout) => {
tracing::warn!(host_id = %host.id, "Health poller: agent timed out"); tracing::warn!(host_id = %host.id, "Health poller: agent timed out");
(HostHealthStatus::Unreachable, serde_json::Value::Object(Default::default())) (
} HostHealthStatus::Unreachable,
serde_json::Value::Object(Default::default()),
)
},
Err(AgentClientError::Connect(_)) => { Err(AgentClientError::Connect(_)) => {
tracing::warn!(host_id = %host.id, "Health poller: agent connection refused"); tracing::warn!(host_id = %host.id, "Health poller: agent connection refused");
(HostHealthStatus::Unreachable, serde_json::Value::Object(Default::default())) (
} HostHealthStatus::Unreachable,
serde_json::Value::Object(Default::default()),
)
},
Err(e) => { Err(e) => {
tracing::warn!(host_id = %host.id, error = %e, "Health poller: agent error"); tracing::warn!(host_id = %host.id, error = %e, "Health poller: agent error");
(HostHealthStatus::Degraded, serde_json::Value::Object(Default::default())) (
} HostHealthStatus::Degraded,
serde_json::Value::Object(Default::default()),
)
},
}, },
}; };

View File

@ -15,7 +15,7 @@
use std::sync::Arc; use std::sync::Arc;
use chrono::{Duration as ChronoDuration, Utc}; use chrono::{Duration as ChronoDuration, Utc};
use pm_agent_client::{AgentClient, types::ApplyPatchesRequest}; use pm_agent_client::{types::ApplyPatchesRequest, AgentClient};
use pm_core::config::AppConfig; use pm_core::config::AppConfig;
use sqlx::{FromRow, PgPool}; use sqlx::{FromRow, PgPool};
use tokio::{sync::Semaphore, time}; use tokio::{sync::Semaphore, time};
@ -71,13 +71,13 @@ struct RetryRow {
#[derive(Debug, FromRow)] #[derive(Debug, FromRow)]
struct StatusCounts { struct StatusCounts {
running_count: i64, running_count: i64,
pending_count: i64, pending_count: i64,
queued_count: i64, queued_count: i64,
succeeded_count: i64, succeeded_count: i64,
failed_count: i64, failed_count: i64,
cancelled_count: i64, cancelled_count: i64,
total_count: i64, total_count: i64,
} }
// ───────────────────────────────────────────────────────────────────────────── // ─────────────────────────────────────────────────────────────────────────────
@ -125,12 +125,8 @@ async fn run_notify_listener(pool: PgPool, config: Arc<AppConfig>) {
/// Inner NOTIFY loop — returns `Err` only on a fatal connection error so the /// Inner NOTIFY loop — returns `Err` only on a fatal connection error so the
/// outer loop can reconnect. /// outer loop can reconnect.
async fn notify_listen_loop( async fn notify_listen_loop(pool: &PgPool, config: &Arc<AppConfig>) -> anyhow::Result<()> {
pool: &PgPool, let mut listener = sqlx::postgres::PgListener::connect(&config.database.url).await?;
config: &Arc<AppConfig>,
) -> anyhow::Result<()> {
let mut listener =
sqlx::postgres::PgListener::connect(&config.database.url).await?;
listener.listen("job_enqueued").await?; listener.listen("job_enqueued").await?;
tracing::debug!("Job executor NOTIFY listener connected"); tracing::debug!("Job executor NOTIFY listener connected");
@ -148,7 +144,7 @@ async fn notify_listen_loop(
"Job executor: invalid UUID in job_enqueued payload" "Job executor: invalid UUID in job_enqueued payload"
); );
continue; continue;
} },
}; };
let (p, c) = (pool.clone(), config.clone()); let (p, c) = (pool.clone(), config.clone());
@ -301,7 +297,7 @@ pub async fn process_job(pool: PgPool, config: Arc<AppConfig>, job_id: Uuid) {
Err(e) => { Err(e) => {
tracing::error!(%job_id, error = %e, "process_job: failed to fetch queued hosts"); tracing::error!(%job_id, error = %e, "process_job: failed to fetch queued hosts");
return; return;
} },
}; };
if hosts.is_empty() { if hosts.is_empty() {
@ -317,11 +313,11 @@ pub async fn process_job(pool: PgPool, config: Arc<AppConfig>, job_id: Uuid) {
Err(e) => { Err(e) => {
tracing::error!(%job_id, error = %e, "process_job: semaphore closed"); tracing::error!(%job_id, error = %e, "process_job: semaphore closed");
break; break;
} },
}; };
let (p, c) = (pool.clone(), config.clone()); let (p, c) = (pool.clone(), config.clone());
let pjh_id = host.id; let pjh_id = host.id;
let host_id = host.host_id; let host_id = host.host_id;
tokio::spawn(async move { tokio::spawn(async move {
@ -338,11 +334,11 @@ pub async fn process_job(pool: PgPool, config: Arc<AppConfig>, job_id: Uuid) {
/// Connect to a single host agent, submit the patch job, and record the /// Connect to a single host agent, submit the patch job, and record the
/// agent-assigned async job ID for later polling. /// agent-assigned async job ID for later polling.
async fn execute_host_job( async fn execute_host_job(
pool: PgPool, pool: PgPool,
config: Arc<AppConfig>, config: Arc<AppConfig>,
job_id: Uuid, job_id: Uuid,
host_id: Uuid, host_id: Uuid,
pjh_id: Uuid, pjh_id: Uuid,
) { ) {
tracing::info!(%job_id, %host_id, %pjh_id, "execute_host_job: starting"); tracing::info!(%job_id, %host_id, %pjh_id, "execute_host_job: starting");
@ -364,34 +360,33 @@ async fn execute_host_job(
) )
.await; .await;
return; return;
} },
Err(e) => { Err(e) => {
tracing::error!(%host_id, error = %e, "execute_host_job: DB error fetching host"); tracing::error!(%host_id, error = %e, "execute_host_job: DB error fetching host");
handle_host_failure(pool, pjh_id, format!("DB error fetching host: {e}")).await; handle_host_failure(pool, pjh_id, format!("DB error fetching host: {e}")).await;
return; return;
} },
}; };
// ── 2. Fetch the job's patch_selection ────────────────────────────────── // ── 2. Fetch the job's patch_selection ──────────────────────────────────
let patch_sel: JobPatchSelection = match sqlx::query_as( let patch_sel: JobPatchSelection =
"SELECT patch_selection FROM patch_jobs WHERE id = $1", match sqlx::query_as("SELECT patch_selection FROM patch_jobs WHERE id = $1")
) .bind(job_id)
.bind(job_id) .fetch_optional(&pool)
.fetch_optional(&pool) .await
.await {
{ Ok(Some(row)) => row,
Ok(Some(row)) => row, Ok(None) => {
Ok(None) => { tracing::error!(%job_id, "execute_host_job: parent job not found");
tracing::error!(%job_id, "execute_host_job: parent job not found"); handle_host_failure(pool, pjh_id, format!("Parent job {job_id} not found")).await;
handle_host_failure(pool, pjh_id, format!("Parent job {job_id} not found")).await; return;
return; },
} Err(e) => {
Err(e) => { tracing::error!(%job_id, error = %e, "execute_host_job: DB error fetching job");
tracing::error!(%job_id, error = %e, "execute_host_job: DB error fetching job"); handle_host_failure(pool, pjh_id, format!("DB error fetching job: {e}")).await;
handle_host_failure(pool, pjh_id, format!("DB error fetching job: {e}")).await; return;
return; },
} };
};
let packages: Vec<String> = let packages: Vec<String> =
serde_json::from_value(patch_sel.patch_selection).unwrap_or_default(); serde_json::from_value(patch_sel.patch_selection).unwrap_or_default();
@ -403,7 +398,7 @@ async fn execute_host_job(
tracing::error!(%host_id, error = %e, "execute_host_job: failed to load agent certs"); tracing::error!(%host_id, error = %e, "execute_host_job: failed to load agent certs");
handle_host_failure(pool, pjh_id, format!("Failed to load agent certs: {e}")).await; handle_host_failure(pool, pjh_id, format!("Failed to load agent certs: {e}")).await;
return; return;
} },
}; };
// ── 4. Build AgentClient ───────────────────────────────────────────────── // ── 4. Build AgentClient ─────────────────────────────────────────────────
@ -419,7 +414,7 @@ async fn execute_host_job(
tracing::error!(%host_id, error = %e, "execute_host_job: failed to build AgentClient"); tracing::error!(%host_id, error = %e, "execute_host_job: failed to build AgentClient");
handle_host_failure(pool, pjh_id, format!("Failed to build agent client: {e}")).await; handle_host_failure(pool, pjh_id, format!("Failed to build agent client: {e}")).await;
return; return;
} },
}; };
// ── 5. Mark pjh as running ─────────────────────────────────────────────── // ── 5. Mark pjh as running ───────────────────────────────────────────────
@ -439,7 +434,10 @@ async fn execute_host_job(
} }
// ── 6. Submit the patch job to the agent ───────────────────────────────── // ── 6. Submit the patch job to the agent ─────────────────────────────────
let req = ApplyPatchesRequest { packages, allow_reboot: true }; let req = ApplyPatchesRequest {
packages,
allow_reboot: true,
};
match client.apply_patches(&req).await { match client.apply_patches(&req).await {
Ok(resp) => { Ok(resp) => {
@ -450,13 +448,12 @@ async fn execute_host_job(
); );
// ── 7. Store agent_job_id; status stays 'running' (agent is async) ── // ── 7. Store agent_job_id; status stays 'running' (agent is async) ──
if let Err(e) = sqlx::query( if let Err(e) =
"UPDATE patch_job_hosts SET agent_job_id = $1 WHERE id = $2", sqlx::query("UPDATE patch_job_hosts SET agent_job_id = $1 WHERE id = $2")
) .bind(&resp.job_id)
.bind(&resp.job_id) .bind(pjh_id)
.bind(pjh_id) .execute(&pool)
.execute(&pool) .await
.await
{ {
tracing::error!( tracing::error!(
%pjh_id, %pjh_id,
@ -464,11 +461,11 @@ async fn execute_host_job(
"execute_host_job: failed to store agent_job_id" "execute_host_job: failed to store agent_job_id"
); );
} }
} },
Err(e) => { Err(e) => {
tracing::warn!(%pjh_id, error = %e, "execute_host_job: agent rejected job"); tracing::warn!(%pjh_id, error = %e, "execute_host_job: agent rejected job");
handle_host_failure(pool, pjh_id, format!("Agent error: {e}")).await; handle_host_failure(pool, pjh_id, format!("Agent error: {e}")).await;
} },
} }
} }
@ -498,7 +495,7 @@ pub async fn poll_running_jobs(pool: PgPool, config: Arc<AppConfig>) {
Err(e) => { Err(e) => {
tracing::error!(error = %e, "poll_running_jobs: DB query failed"); tracing::error!(error = %e, "poll_running_jobs: DB query failed");
return; return;
} },
}; };
for row in rows { for row in rows {
@ -510,11 +507,7 @@ pub async fn poll_running_jobs(pool: PgPool, config: Arc<AppConfig>) {
} }
/// Poll one running host entry and update its status from the agent response. /// Poll one running host entry and update its status from the agent response.
async fn poll_single_host( async fn poll_single_host(pool: PgPool, config: Arc<AppConfig>, row: PatchJobHostRunning) {
pool: PgPool,
config: Arc<AppConfig>,
row: PatchJobHostRunning,
) {
let certs = match load_agent_certs(&config.security) { let certs = match load_agent_certs(&config.security) {
Ok(c) => c, Ok(c) => c,
Err(e) => { Err(e) => {
@ -524,7 +517,7 @@ async fn poll_single_host(
"poll_single_host: failed to load agent certs" "poll_single_host: failed to load agent certs"
); );
return; return;
} },
}; };
let client = match AgentClient::new( let client = match AgentClient::new(
@ -542,7 +535,7 @@ async fn poll_single_host(
"poll_single_host: failed to build AgentClient" "poll_single_host: failed to build AgentClient"
); );
return; return;
} },
}; };
let status = match client.job_status(&row.agent_job_id).await { let status = match client.job_status(&row.agent_job_id).await {
@ -555,7 +548,7 @@ async fn poll_single_host(
"poll_single_host: agent status call failed" "poll_single_host: agent status call failed"
); );
return; return;
} },
}; };
match status.status.as_str() { match status.status.as_str() {
@ -578,14 +571,14 @@ async fn poll_single_host(
tracing::error!(pjh_id = %row.id, error = %e, "poll_single_host: update failed"); tracing::error!(pjh_id = %row.id, error = %e, "poll_single_host: update failed");
} }
sync_job_status(&pool, row.job_id).await; sync_job_status(&pool, row.job_id).await;
} },
"failed" => { "failed" => {
tracing::warn!(pjh_id = %row.id, "poll_single_host: agent job failed"); tracing::warn!(pjh_id = %row.id, "poll_single_host: agent job failed");
let err_msg = status let err_msg = status
.error .error
.unwrap_or_else(|| "Agent reported failure (no detail)".to_string()); .unwrap_or_else(|| "Agent reported failure (no detail)".to_string());
handle_host_failure(pool, row.id, err_msg).await; handle_host_failure(pool, row.id, err_msg).await;
} },
"running" | "queued" => { "running" | "queued" => {
// Still in progress — nothing to update; will poll again next cycle. // Still in progress — nothing to update; will poll again next cycle.
tracing::debug!( tracing::debug!(
@ -593,14 +586,14 @@ async fn poll_single_host(
agent_status = %status.status, agent_status = %status.status,
"poll_single_host: job still in progress" "poll_single_host: job still in progress"
); );
} },
other => { other => {
tracing::warn!( tracing::warn!(
pjh_id = %row.id, pjh_id = %row.id,
agent_status = %other, agent_status = %other,
"poll_single_host: unexpected agent status — ignoring" "poll_single_host: unexpected agent status — ignoring"
); );
} },
} }
} }
@ -624,7 +617,7 @@ async fn handle_host_failure(pool: PgPool, pjh_id: Uuid, error_msg: String) {
Err(e) => { Err(e) => {
tracing::error!(%pjh_id, error = %e, "handle_host_failure: DB error fetching retry row"); tracing::error!(%pjh_id, error = %e, "handle_host_failure: DB error fetching retry row");
return; return;
} },
}; };
let row = match row { let row = match row {
@ -632,7 +625,7 @@ async fn handle_host_failure(pool: PgPool, pjh_id: Uuid, error_msg: String) {
None => { None => {
tracing::error!(%pjh_id, "handle_host_failure: pjh row not found"); tracing::error!(%pjh_id, "handle_host_failure: pjh row not found");
return; return;
} },
}; };
if row.retry_count < 3 { if row.retry_count < 3 {
@ -736,7 +729,7 @@ async fn sync_job_status(pool: &PgPool, job_id: Uuid) {
Err(e) => { Err(e) => {
tracing::error!(%job_id, error = %e, "sync_job_status: DB query failed"); tracing::error!(%job_id, error = %e, "sync_job_status: DB query failed");
return; return;
} },
}; };
// Determine the aggregate status. // Determine the aggregate status.
@ -745,19 +738,19 @@ async fn sync_job_status(pool: &PgPool, job_id: Uuid) {
if counts.running_count > 0 || counts.pending_count > 0 || counts.queued_count > 0 { if counts.running_count > 0 || counts.pending_count > 0 || counts.queued_count > 0 {
// Still work in flight — keep parent running. // Still work in flight — keep parent running.
new_status = "running"; new_status = "running";
set_completed = false; set_completed = false;
} else if counts.total_count > 0 && counts.succeeded_count == counts.total_count { } else if counts.total_count > 0 && counts.succeeded_count == counts.total_count {
// Every host succeeded. // Every host succeeded.
new_status = "succeeded"; new_status = "succeeded";
set_completed = true; set_completed = true;
} else if counts.total_count > 0 && counts.cancelled_count == counts.total_count { } else if counts.total_count > 0 && counts.cancelled_count == counts.total_count {
// Every host cancelled. // Every host cancelled.
new_status = "cancelled"; new_status = "cancelled";
set_completed = true; set_completed = true;
} else if counts.failed_count > 0 { } else if counts.failed_count > 0 {
// At least one failure and nothing still active → failed (partial counts too). // At least one failure and nothing still active → failed (partial counts too).
new_status = "failed"; new_status = "failed";
set_completed = true; set_completed = true;
} else { } else {
// Fallback: nothing actionable yet. // Fallback: nothing actionable yet.
@ -789,13 +782,11 @@ async fn sync_job_status(pool: &PgPool, job_id: Uuid) {
.execute(pool) .execute(pool)
.await .await
} else { } else {
sqlx::query( sqlx::query("UPDATE patch_jobs SET status = $2 WHERE id = $1")
"UPDATE patch_jobs SET status = $2 WHERE id = $1", .bind(job_id)
) .bind(new_status)
.bind(job_id) .execute(pool)
.bind(new_status) .await
.execute(pool)
.await
}; };
if let Err(e) = result { if let Err(e) = result {
@ -812,13 +803,8 @@ async fn sync_job_status(pool: &PgPool, job_id: Uuid) {
let failed = counts.failed_count; let failed = counts.failed_count;
tokio::spawn(async move { tokio::spawn(async move {
email::send_job_completion_email( email::send_job_completion_email(&pool_clone, &job_id_str, total, succeeded, failed)
&pool_clone, .await;
&job_id_str,
total,
succeeded,
failed,
).await;
// If there are failures, also send failure emails per host // If there are failures, also send failure emails per host
if failed > 0 { if failed > 0 {
@ -838,16 +824,12 @@ async fn sync_job_status(pool: &PgPool, job_id: Uuid) {
Err(e) => { Err(e) => {
tracing::error!(%job_id, error = %e, "sync_job_status: failed to fetch failed hosts for email"); tracing::error!(%job_id, error = %e, "sync_job_status: failed to fetch failed hosts for email");
Vec::new() Vec::new()
} },
}; };
for (fqdn, error_msg) in failed_hosts { for (fqdn, error_msg) in failed_hosts {
email::send_patch_failure_email( email::send_patch_failure_email(&pool_clone, &fqdn, &job_id_str, &error_msg)
&pool_clone, .await;
&fqdn,
&job_id_str,
&error_msg,
).await;
} }
} }
}); });
@ -878,7 +860,7 @@ pub async fn retry_pending_jobs(pool: PgPool, config: Arc<AppConfig>) {
Err(e) => { Err(e) => {
tracing::error!(error = %e, "retry_pending_jobs: DB query failed"); tracing::error!(error = %e, "retry_pending_jobs: DB query failed");
return; return;
} },
}; };
for row in rows { for row in rows {

View File

@ -7,27 +7,23 @@ mod agent_loader;
mod audit_verifier; mod audit_verifier;
mod email; mod email;
mod health_poller; mod health_poller;
mod job_executor;
mod maintenance_scheduler; mod maintenance_scheduler;
mod patch_poller; mod patch_poller;
mod refresh_listener; mod refresh_listener;
mod job_executor;
mod ws_relay; mod ws_relay;
use pm_core::{ use pm_core::{config::AppConfig, db, logging};
config::AppConfig,
db,
logging,
};
use sqlx::PgPool; use sqlx::PgPool;
use std::{sync::Arc, time::Duration}; use std::{sync::Arc, time::Duration};
use tokio::time; use tokio::time;
use audit_verifier::run_audit_verifier; use audit_verifier::run_audit_verifier;
use health_poller::run_health_poller; use health_poller::run_health_poller;
use job_executor::run_job_executor;
use maintenance_scheduler::run_maintenance_scheduler; use maintenance_scheduler::run_maintenance_scheduler;
use patch_poller::run_patch_poller; use patch_poller::run_patch_poller;
use refresh_listener::run_refresh_listener; use refresh_listener::run_refresh_listener;
use job_executor::run_job_executor;
use ws_relay::run_ws_relay; use ws_relay::run_ws_relay;
/// Minimum number of applied migrations the worker requires before /// Minimum number of applied migrations the worker requires before
@ -44,16 +40,18 @@ async fn main() -> anyhow::Result<()> {
let config_path = std::env::var("PATCH_MANAGER_CONFIG") let config_path = std::env::var("PATCH_MANAGER_CONFIG")
.unwrap_or_else(|_| "/etc/patch-manager/config.toml".to_string()); .unwrap_or_else(|_| "/etc/patch-manager/config.toml".to_string());
let config = AppConfig::load(&config_path) let config = AppConfig::load(&config_path).unwrap_or_else(|_| {
.unwrap_or_else(|_| { eprintln!("Config file not found or invalid, using defaults");
eprintln!("Config file not found or invalid, using defaults"); AppConfig::default()
AppConfig::default() });
});
// Initialize logging // Initialize logging
logging::init(&config.logging); logging::init(&config.logging);
tracing::info!(version = env!("CARGO_PKG_VERSION"), "patch-manager-worker starting"); tracing::info!(
version = env!("CARGO_PKG_VERSION"),
"patch-manager-worker starting"
);
// Initialize database pool // Initialize database pool
let pool = db::init_pool(&config.database).await?; let pool = db::init_pool(&config.database).await?;
@ -114,17 +112,17 @@ async fn wait_for_schema(pool: &PgPool) -> anyhow::Result<()> {
Ok(count) if count >= REQUIRED_MIGRATION_COUNT => { Ok(count) if count >= REQUIRED_MIGRATION_COUNT => {
tracing::info!(migration_count = count, "Schema version check passed"); tracing::info!(migration_count = count, "Schema version check passed");
return Ok(()); return Ok(());
} },
Ok(count) => { Ok(count) => {
tracing::warn!( tracing::warn!(
migration_count = count, migration_count = count,
required = REQUIRED_MIGRATION_COUNT, required = REQUIRED_MIGRATION_COUNT,
"Schema not ready, waiting..." "Schema not ready, waiting..."
); );
} },
Err(e) => { Err(e) => {
tracing::warn!(error = %e, "Schema version check failed, retrying..."); tracing::warn!(error = %e, "Schema version check failed, retrying...");
} },
} }
if tokio::time::Instant::now() >= deadline { if tokio::time::Instant::now() >= deadline {

View File

@ -144,7 +144,7 @@ async fn dispatch_open_window_jobs(pool: PgPool, config: Arc<AppConfig>) {
"dispatch_open_window_jobs: queued jobs query failed" "dispatch_open_window_jobs: queued jobs query failed"
); );
continue; continue;
} },
}; };
for job in job_ids { for job in job_ids {

View File

@ -9,10 +9,7 @@ use std::sync::Arc;
use pm_agent_client::AgentClient; use pm_agent_client::AgentClient;
use pm_core::config::AppConfig; use pm_core::config::AppConfig;
use sqlx::{FromRow, PgPool}; use sqlx::{FromRow, PgPool};
use tokio::{ use tokio::{sync::Semaphore, time};
sync::Semaphore,
time,
};
use uuid::Uuid; use uuid::Uuid;
use crate::agent_loader::load_agent_certs; use crate::agent_loader::load_agent_certs;
@ -34,10 +31,7 @@ pub async fn run_patch_poller(pool: PgPool, config: Arc<AppConfig>) {
let interval_secs = config.worker.patch_poll_interval_secs; let interval_secs = config.worker.patch_poll_interval_secs;
let mut ticker = time::interval(std::time::Duration::from_secs(interval_secs)); let mut ticker = time::interval(std::time::Duration::from_secs(interval_secs));
tracing::info!( tracing::info!(interval_secs, "Patch poller started");
interval_secs,
"Patch poller started"
);
loop { loop {
ticker.tick().await; ticker.tick().await;
@ -47,7 +41,7 @@ pub async fn run_patch_poller(pool: PgPool, config: Arc<AppConfig>) {
Err(e) => { Err(e) => {
tracing::error!(error = %e, "Patch poller: failed to load agent certs — skipping cycle"); tracing::error!(error = %e, "Patch poller: failed to load agent certs — skipping cycle");
continue; continue;
} },
}; };
let client_cert = Arc::new(certs.client_cert); let client_cert = Arc::new(certs.client_cert);
@ -64,7 +58,7 @@ pub async fn run_patch_poller(pool: PgPool, config: Arc<AppConfig>) {
Err(e) => { Err(e) => {
tracing::error!(error = %e, "Patch poller: failed to fetch hosts"); tracing::error!(error = %e, "Patch poller: failed to fetch hosts");
continue; continue;
} },
}; };
if hosts.is_empty() { if hosts.is_empty() {
@ -102,16 +96,11 @@ pub async fn run_patch_poller(pool: PgPool, config: Arc<AppConfig>) {
Err(e) => { Err(e) => {
tracing::error!(error = %e, "Patch poller task panicked"); tracing::error!(error = %e, "Patch poller task panicked");
failed += 1; failed += 1;
} },
} }
} }
tracing::info!( tracing::info!(total, succeeded, failed, "Patch poll cycle complete");
total,
succeeded,
failed,
"Patch poll cycle complete"
);
} }
} }
@ -135,7 +124,7 @@ async fn poll_host_patches(
Err(e) => { Err(e) => {
tracing::warn!(host_id = %host.id, error = %e, "Patch poller: failed to build AgentClient"); tracing::warn!(host_id = %host.id, error = %e, "Patch poller: failed to build AgentClient");
return false; return false;
} },
}; };
// Fetch patches and packages concurrently. // Fetch patches and packages concurrently.
@ -147,7 +136,7 @@ async fn poll_host_patches(
Err(e) => { Err(e) => {
tracing::warn!(host_id = %host.id, error = %e, "Patch poller: patches() failed"); tracing::warn!(host_id = %host.id, error = %e, "Patch poller: patches() failed");
return false; return false;
} },
}; };
let packages_data = match packages_result { let packages_data = match packages_result {
@ -155,7 +144,7 @@ async fn poll_host_patches(
Err(e) => { Err(e) => {
tracing::warn!(host_id = %host.id, error = %e, "Patch poller: packages_upgradable() failed"); tracing::warn!(host_id = %host.id, error = %e, "Patch poller: packages_upgradable() failed");
return false; return false;
} },
}; };
let available_patches = serde_json::to_value(&patches_data.patches).unwrap_or_default(); let available_patches = serde_json::to_value(&patches_data.patches).unwrap_or_default();
@ -188,12 +177,10 @@ async fn poll_host_patches(
} }
// Update hosts.last_patch_at. // Update hosts.last_patch_at.
if let Err(e) = sqlx::query( if let Err(e) = sqlx::query("UPDATE hosts SET last_patch_at = NOW() WHERE id = $1")
"UPDATE hosts SET last_patch_at = NOW() WHERE id = $1", .bind(host.id)
) .execute(&pool)
.bind(host.id) .await
.execute(&pool)
.await
{ {
tracing::error!(host_id = %host.id, error = %e, "Patch poller: failed to update last_patch_at"); tracing::error!(host_id = %host.id, error = %e, "Patch poller: failed to update last_patch_at");
} }

View File

@ -8,10 +8,7 @@
use std::sync::Arc; use std::sync::Arc;
use pm_agent_client::{AgentClient, AgentClientError}; use pm_agent_client::{AgentClient, AgentClientError};
use pm_core::{ use pm_core::{config::AppConfig, models::HostHealthStatus};
config::AppConfig,
models::HostHealthStatus,
};
use sqlx::{FromRow, PgPool}; use sqlx::{FromRow, PgPool};
use tokio::time; use tokio::time;
use uuid::Uuid; use uuid::Uuid;
@ -46,8 +43,7 @@ pub async fn run_refresh_listener(pool: PgPool, config: Arc<AppConfig>) {
/// Inner loop — returns `Err` only on a fatal listener error so the outer /// Inner loop — returns `Err` only on a fatal listener error so the outer
/// loop can reconnect. /// loop can reconnect.
async fn listen_loop(pool: &PgPool, config: &AppConfig) -> anyhow::Result<()> { async fn listen_loop(pool: &PgPool, config: &AppConfig) -> anyhow::Result<()> {
let mut listener = let mut listener = sqlx::postgres::PgListener::connect(&config.database.url).await?;
sqlx::postgres::PgListener::connect(&config.database.url).await?;
listener.listen("refresh_requested").await?; listener.listen("refresh_requested").await?;
@ -68,7 +64,7 @@ async fn listen_loop(pool: &PgPool, config: &AppConfig) -> anyhow::Result<()> {
"Refresh listener: invalid UUID in notification payload" "Refresh listener: invalid UUID in notification payload"
); );
continue; continue;
} },
}; };
// Fetch the host from the database. // Fetch the host from the database.
@ -85,7 +81,7 @@ async fn listen_loop(pool: &PgPool, config: &AppConfig) -> anyhow::Result<()> {
None => { None => {
tracing::warn!(%host_id, "Refresh listener: host not found"); tracing::warn!(%host_id, "Refresh listener: host not found");
continue; continue;
} },
}; };
// Load certs for this refresh. // Load certs for this refresh.
@ -98,7 +94,7 @@ async fn listen_loop(pool: &PgPool, config: &AppConfig) -> anyhow::Result<()> {
"Refresh listener: failed to load agent certs" "Refresh listener: failed to load agent certs"
); );
continue; continue;
} },
}; };
// Spawn the actual work so the listener loop is not blocked. // Spawn the actual work so the listener loop is not blocked.
@ -137,7 +133,7 @@ async fn refresh_host(
); );
persist_health_unreachable(&pool, host.id).await; persist_health_unreachable(&pool, host.id).await;
return; return;
} },
}; };
// ── Health ──────────────────────────────────────────────────────────── // ── Health ────────────────────────────────────────────────────────────
@ -145,15 +141,21 @@ async fn refresh_host(
Ok(data) => { Ok(data) => {
let payload = serde_json::to_value(&data).unwrap_or_default(); let payload = serde_json::to_value(&data).unwrap_or_default();
(HostHealthStatus::Healthy, payload) (HostHealthStatus::Healthy, payload)
} },
Err(AgentClientError::Timeout) | Err(AgentClientError::Connect(_)) => { Err(AgentClientError::Timeout) | Err(AgentClientError::Connect(_)) => {
tracing::warn!(host_id = %host.id, "Refresh: agent unreachable"); tracing::warn!(host_id = %host.id, "Refresh: agent unreachable");
(HostHealthStatus::Unreachable, serde_json::Value::Object(Default::default())) (
} HostHealthStatus::Unreachable,
serde_json::Value::Object(Default::default()),
)
},
Err(e) => { Err(e) => {
tracing::warn!(host_id = %host.id, error = %e, "Refresh: health error"); tracing::warn!(host_id = %host.id, error = %e, "Refresh: health error");
(HostHealthStatus::Degraded, serde_json::Value::Object(Default::default())) (
} HostHealthStatus::Degraded,
serde_json::Value::Object(Default::default()),
)
},
}; };
persist_health(&pool, host.id, &health_status, &health_payload).await; persist_health(&pool, host.id, &health_status, &health_payload).await;
@ -164,8 +166,7 @@ async fn refresh_host(
match (patches_result, packages_result) { match (patches_result, packages_result) {
(Ok(patches_data), Ok(packages_data)) => { (Ok(patches_data), Ok(packages_data)) => {
let available_patches = let available_patches = serde_json::to_value(&patches_data.patches).unwrap_or_default();
serde_json::to_value(&patches_data.patches).unwrap_or_default();
let installed_packages = let installed_packages =
serde_json::to_value(&packages_data.packages).unwrap_or_default(); serde_json::to_value(&packages_data.packages).unwrap_or_default();
let patch_count = patches_data.total as i32; let patch_count = patches_data.total as i32;
@ -196,12 +197,10 @@ async fn refresh_host(
"Refresh: failed to insert patch data" "Refresh: failed to insert patch data"
); );
} else { } else {
let _ = sqlx::query( let _ = sqlx::query("UPDATE hosts SET last_patch_at = NOW() WHERE id = $1")
"UPDATE hosts SET last_patch_at = NOW() WHERE id = $1", .bind(host.id)
) .execute(&pool)
.bind(host.id) .await;
.execute(&pool)
.await;
tracing::info!( tracing::info!(
host_id = %host.id, host_id = %host.id,
@ -210,14 +209,14 @@ async fn refresh_host(
"On-demand refresh complete" "On-demand refresh complete"
); );
} }
} },
(Err(e), _) | (_, Err(e)) => { (Err(e), _) | (_, Err(e)) => {
tracing::warn!( tracing::warn!(
host_id = %host.id, host_id = %host.id,
error = %e, error = %e,
"Refresh: failed to collect patch data" "Refresh: failed to collect patch data"
); );
} },
} }
} }
@ -252,13 +251,12 @@ async fn persist_health(
); );
} }
if let Err(e) = sqlx::query( if let Err(e) =
"UPDATE hosts SET health_status = $2, last_health_at = NOW() WHERE id = $1", sqlx::query("UPDATE hosts SET health_status = $2, last_health_at = NOW() WHERE id = $1")
) .bind(host_id)
.bind(host_id) .bind(status)
.bind(status) .execute(pool)
.execute(pool) .await
.await
{ {
tracing::error!(%host_id, error = %e, "Refresh: failed to update host health_status"); tracing::error!(%host_id, error = %e, "Refresh: failed to update host health_status");
} }

View File

@ -5,27 +5,18 @@
//! DB row, and fire `pg_notify('job_update', payload_json)` so the browser WS //! DB row, and fire `pg_notify('job_update', payload_json)` so the browser WS
//! handler can forward the event to connected clients. //! handler can forward the event to connected clients.
use std::{ use std::{collections::HashSet, sync::Arc, time::Duration};
collections::HashSet,
sync::Arc,
time::Duration,
};
use anyhow::Context; use anyhow::Context;
use futures::StreamExt; use futures::StreamExt;
use rustls::{ use rustls::{
pki_types::{CertificateDer, PrivateKeyDer}, pki_types::{CertificateDer, PrivateKeyDer},
ClientConfig as TlsClientConfig, ClientConfig as TlsClientConfig, RootCertStore,
RootCertStore,
}; };
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use sqlx::PgPool; use sqlx::PgPool;
use tokio::sync::Mutex; use tokio::sync::Mutex;
use tokio_tungstenite::{ use tokio_tungstenite::{connect_async_tls_with_config, tungstenite::protocol::Message, Connector};
connect_async_tls_with_config,
tungstenite::protocol::Message,
Connector,
};
use uuid::Uuid; use uuid::Uuid;
use pm_agent_client::client::DEFAULT_AGENT_PORT; use pm_agent_client::client::DEFAULT_AGENT_PORT;
@ -84,7 +75,7 @@ pub async fn run_ws_relay(pool: PgPool, config: Arc<AppConfig>) {
Err(e) => { Err(e) => {
tracing::error!(error = %e, "ws_relay: DB poll failed"); tracing::error!(error = %e, "ws_relay: DB poll failed");
continue; continue;
} },
}; };
for row in rows { for row in rows {
@ -101,12 +92,12 @@ pub async fn run_ws_relay(pool: PgPool, config: Arc<AppConfig>) {
Err(e) => { Err(e) => {
tracing::error!(error = %e, "ws_relay: TLS config error"); tracing::error!(error = %e, "ws_relay: TLS config error");
continue; continue;
} },
}; };
active.lock().await.insert(key); active.lock().await.insert(key);
let pool_c = pool.clone(); let pool_c = pool.clone();
let active_c = active.clone(); let active_c = active.clone();
tokio::spawn(async move { tokio::spawn(async move {
@ -164,12 +155,15 @@ async fn query_running_jobs(pool: &PgPool) -> anyhow::Result<Vec<RunningHostJob>
async fn build_tls_config(config: &AppConfig) -> anyhow::Result<TlsClientConfig> { async fn build_tls_config(config: &AppConfig) -> anyhow::Result<TlsClientConfig> {
let sec = &config.security; let sec = &config.security;
let cert_pem = tokio::fs::read(&sec.agent_client_cert_path).await let cert_pem = tokio::fs::read(&sec.agent_client_cert_path)
.await
.with_context(|| format!("read agent client cert '{}'", sec.agent_client_cert_path))?; .with_context(|| format!("read agent client cert '{}'", sec.agent_client_cert_path))?;
let key_pem = tokio::fs::read(&sec.agent_client_key_path).await let key_pem = tokio::fs::read(&sec.agent_client_key_path)
.with_context(|| format!("read agent client key '{}'" , sec.agent_client_key_path))?; .await
let ca_pem = tokio::fs::read(&sec.ca_cert_path).await .with_context(|| format!("read agent client key '{}'", sec.agent_client_key_path))?;
.with_context(|| format!("read CA cert '{}'", sec.ca_cert_path))?; let ca_pem = tokio::fs::read(&sec.ca_cert_path)
.await
.with_context(|| format!("read CA cert '{}'", sec.ca_cert_path))?;
// Parse client certificate chain. // Parse client certificate chain.
let client_certs: Vec<CertificateDer<'static>> = { let client_certs: Vec<CertificateDer<'static>> = {
@ -207,8 +201,8 @@ async fn build_tls_config(config: &AppConfig) -> anyhow::Result<TlsClientConfig>
// ── Per-job relay ───────────────────────────────────────────────────────────── // ── Per-job relay ─────────────────────────────────────────────────────────────
async fn relay_one_job( async fn relay_one_job(
pool: &PgPool, pool: &PgPool,
row: &RunningHostJob, row: &RunningHostJob,
tls_config: Arc<TlsClientConfig>, tls_config: Arc<TlsClientConfig>,
) -> anyhow::Result<()> { ) -> anyhow::Result<()> {
let url = format!( let url = format!(
@ -229,7 +223,7 @@ async fn relay_one_job(
while let Some(frame) = stream.next().await { while let Some(frame) = stream.next().await {
let frame = match frame { let frame = match frame {
Ok(f) => f, Ok(f) => f,
Err(e) => { Err(e) => {
tracing::warn!( tracing::warn!(
error = %e, error = %e,
@ -238,16 +232,16 @@ async fn relay_one_job(
"WS relay: stream error" "WS relay: stream error"
); );
break; break;
} },
}; };
let text = match frame { let text = match frame {
Message::Text(t) => t.to_string(), Message::Text(t) => t.to_string(),
Message::Binary(b) => String::from_utf8(b.into()).unwrap_or_default(), Message::Binary(b) => String::from_utf8(b.into()).unwrap_or_default(),
Message::Close(_) => { Message::Close(_) => {
tracing::info!(job_id = %row.job_id, "Agent WS closed cleanly"); tracing::info!(job_id = %row.job_id, "Agent WS closed cleanly");
break; break;
} },
_ => continue, _ => continue,
}; };
@ -256,14 +250,14 @@ async fn relay_one_job(
} }
let event: AgentWsEvent = match serde_json::from_str(&text) { let event: AgentWsEvent = match serde_json::from_str(&text) {
Ok(e) => e, Ok(e) => e,
Err(e) => { Err(e) => {
tracing::warn!( tracing::warn!(
error = %e, raw = %text, error = %e, raw = %text,
"WS relay: unparseable agent frame" "WS relay: unparseable agent frame"
); );
continue; continue;
} },
}; };
process_event(pool, row, &event).await; process_event(pool, row, &event).await;
@ -287,17 +281,17 @@ async fn relay_one_job(
async fn process_event(pool: &PgPool, row: &RunningHostJob, event: &AgentWsEvent) { async fn process_event(pool: &PgPool, row: &RunningHostJob, event: &AgentWsEvent) {
// Map agent status string to DB job_status enum value. // Map agent status string to DB job_status enum value.
let db_status = match event.status.as_str() { let db_status = match event.status.as_str() {
"running" => "running", "running" => "running",
"succeeded" => "succeeded", "succeeded" => "succeeded",
"failed" => "failed", "failed" => "failed",
"cancelled" => "cancelled", "cancelled" => "cancelled",
other => { other => {
tracing::warn!(status = %other, "WS relay: unknown agent status"); tracing::warn!(status = %other, "WS relay: unknown agent status");
return; return;
} },
}; };
let output = event.output.as_deref().unwrap_or(""); let output = event.output.as_deref().unwrap_or("");
let error_msg = event.error.as_deref(); let error_msg = event.error.as_deref();
// Determine timestamps based on terminal state. // Determine timestamps based on terminal state.
@ -359,20 +353,20 @@ async fn process_event(pool: &PgPool, row: &RunningHostJob, event: &AgentWsEvent
// Fire pg_notify so browser WS handlers forward the event. // Fire pg_notify so browser WS handlers forward the event.
let payload = NotifyPayload { let payload = NotifyPayload {
job_id: row.job_id.to_string(), job_id: row.job_id.to_string(),
host_id: row.host_id.to_string(), host_id: row.host_id.to_string(),
status: db_status.to_string(), status: db_status.to_string(),
output: event.output.clone(), output: event.output.clone(),
error_message: event.error.clone(), error_message: event.error.clone(),
agent_job_id: row.agent_job_id.clone(), agent_job_id: row.agent_job_id.clone(),
}; };
let payload_json = match serde_json::to_string(&payload) { let payload_json = match serde_json::to_string(&payload) {
Ok(s) => s, Ok(s) => s,
Err(e) => { Err(e) => {
tracing::error!(error = %e, "WS relay: failed to serialize notify payload"); tracing::error!(error = %e, "WS relay: failed to serialize notify payload");
return; return;
} },
}; };
if let Err(e) = sqlx::query("SELECT pg_notify('job_update', $1)") if let Err(e) = sqlx::query("SELECT pg_notify('job_update', $1)")
@ -418,11 +412,11 @@ async fn update_parent_job_status(pool: &PgPool, job_id: Uuid) {
.fetch_one(pool) .fetch_one(pool)
.await .await
{ {
Ok(n) => n, Ok(n) => n,
Err(e) => { Err(e) => {
tracing::error!(error = %e, %job_id, "update_parent_job_status: count query failed"); tracing::error!(error = %e, %job_id, "update_parent_job_status: count query failed");
return; return;
} },
}; };
if pending > 0 { if pending > 0 {
@ -437,14 +431,18 @@ async fn update_parent_job_status(pool: &PgPool, job_id: Uuid) {
.fetch_one(pool) .fetch_one(pool)
.await .await
{ {
Ok(n) => n, Ok(n) => n,
Err(e) => { Err(e) => {
tracing::error!(error = %e, %job_id, "update_parent_job_status: failed-count query failed"); tracing::error!(error = %e, %job_id, "update_parent_job_status: failed-count query failed");
return; return;
} },
}; };
let final_status = if failed_count > 0 { "failed" } else { "succeeded" }; let final_status = if failed_count > 0 {
"failed"
} else {
"succeeded"
};
if let Err(e) = sqlx::query( if let Err(e) = sqlx::query(
"UPDATE patch_jobs SET status = $1::job_status, completed_at = NOW() WHERE id = $2", "UPDATE patch_jobs SET status = $1::job_status, completed_at = NOW() WHERE id = $2",

View File

@ -1,5 +1,6 @@
# Linux Patch Manager - Rust Formatting Configuration # Linux Patch Manager - Rust Formatting Configuration
# Run: cargo fmt --check (CI) or cargo fmt (fix) # Run: cargo fmt --check (CI) or cargo fmt (fix)
# Only stable options - nightly-only options removed
edition = "2021" edition = "2021"
max_width = 100 max_width = 100
@ -10,15 +11,4 @@ use_small_heuristics = "Default"
reorder_imports = true reorder_imports = true
reorder_modules = true reorder_modules = true
remove_nested_parens = true remove_nested_parens = true
fn_single_line = false
where_single_line = false
imports_granularity = "Crate"
group_imports = "StdExternalCrate"
normalize_doc_attributes = true
wrap_comments = true
comment_width = 80
indent_style = "Block"
trailing_comma = "Vertical"
match_block_trailing_comma = true match_block_trailing_comma = true
blank_lines_lower_bound = 0
blank_lines_upper_bound = 1