mod format;
mod integrity;
pub mod position;
use crate::hashing::{checksum_managed_body_line_refs, normalize_line_endings};
use crate::lockfile::{Lockfile, ManagedSpanRecord};
use crate::render::render_project;
use crate::resolve::ResolvedProject;
use crate::{PrayError, PrayResult};
use format::format_drift_report;
use integrity::{push_package_lock_findings, push_provisioned_and_local_findings};
use position::{format_position_drift_message, summarize_position_drift};
use std::collections::{BTreeMap, BTreeSet, HashSet};
use std::fs;
pub use format::format_verification_report;
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct VerificationFinding {
pub kind: String,
pub message: String,
}
#[derive(Debug, Clone, Default)]
pub struct VerificationReport {
pub findings: Vec<VerificationFinding>,
}
impl VerificationReport {
pub fn is_clean(&self) -> bool {
self.findings.is_empty()
}
pub fn has_warnings(&self) -> bool {
self.findings.iter().any(VerificationFinding::is_warning)
}
pub fn has_errors(&self) -> bool {
self.findings.iter().any(VerificationFinding::is_error)
}
}
impl VerificationFinding {
pub fn is_warning(&self) -> bool {
matches!(self.kind.as_str(), "orphan_marker")
}
pub fn is_error(&self) -> bool {
!self.is_warning()
}
}
pub fn inspect_project(
project: &ResolvedProject,
lockfile: &Lockfile,
) -> PrayResult<VerificationReport> {
let (report, _, _) = collect_verification_report(project, lockfile)?;
Ok(report)
}
pub fn verify_project(
project: &ResolvedProject,
lockfile: &Lockfile,
strict: bool,
) -> PrayResult<VerificationReport> {
let report = inspect_project(project, lockfile)?;
if report.is_clean() {
return Ok(report);
}
if strict || report.has_errors() {
Err(PrayError::Verify(format_verification_report(&report)))
} else {
Ok(report)
}
}
type CollectedVerification = (
VerificationReport,
BTreeMap<String, String>,
BTreeMap<String, String>,
);
fn collect_verification_report(
project: &ResolvedProject,
lockfile: &Lockfile,
) -> PrayResult<CollectedVerification> {
let mut report = VerificationReport::default();
let mut rendered_targets = BTreeMap::new();
let fresh_targets: BTreeMap<String, String> = render_project(project)?
.into_iter()
.map(|target| (target.path.to_string_lossy().to_string(), target.content))
.collect();
if project.manifest_hash != lockfile.manifest_hash {
report.findings.push(VerificationFinding {
kind: "verify_error".to_string(),
message:
"Prayfile changed since `Prayfile.lock` was generated. Run `pray install` to refresh the lockfile."
.to_string(),
});
}
push_package_lock_findings(project, lockfile, &mut report.findings);
let mut target_spans: BTreeMap<String, Vec<&ManagedSpanRecord>> = BTreeMap::new();
for span in &lockfile.managed_span {
target_spans
.entry(span.target.clone())
.or_default()
.push(span);
}
for (target_path, spans) in target_spans {
let absolute_path = project.project_root.join(&target_path);
if !absolute_path.exists() {
report.findings.push(VerificationFinding {
kind: "verify_error".to_string(),
message: format!(
"Rendered file `{}` is missing. Run `pray install` to generate it.",
target_path
),
});
continue;
}
let text = fs::read_to_string(&absolute_path)?;
rendered_targets.insert(target_path.clone(), text.clone());
let lines: Vec<&str> = text.lines().collect();
let markers = marker_positions(&lines);
for span in &spans {
match markers.get(&span.id) {
None => report.findings.push(VerificationFinding {
kind: "removed_prayer".to_string(),
message: format!(
"`{}` is missing managed marker `{}` for `{}::{}`. Run `pray install` to restore the managed span.",
target_path, span.id, span.package, span.export
),
}),
Some((_, _, checksum)) => {
if checksum != &span.ideal_checksum {
report.findings.push(VerificationFinding {
kind: "custom_implementation".to_string(),
message: format!(
"`{}` marker `{}` (`{}::{}`) was edited. Restore the managed block or run `pray install` to regenerate it.",
target_path, span.id, span.package, span.export
),
});
}
}
}
}
let fresh_lines: Vec<&str> = fresh_targets
.get(&target_path)
.map(|fresh| fresh.lines().collect())
.unwrap_or_default();
if let Some(summary) = summarize_position_drift(
&target_path,
&spans,
&markers,
&lines,
fresh_targets
.contains_key(&target_path)
.then_some(fresh_lines.as_slice()),
&project.local_files,
) {
report.findings.push(VerificationFinding {
kind: "position_drift".to_string(),
message: format_position_drift_message(&summary),
});
}
for finding in find_orphan_marker_findings_from_markers(&spans, &markers, &target_path) {
report.findings.push(finding);
}
}
push_provisioned_and_local_findings(project, &mut report.findings)?;
Ok((report, rendered_targets, fresh_targets))
}
pub fn find_orphan_marker_findings(
spans: &[&ManagedSpanRecord],
lines: &[&str],
target_path: &str,
) -> Vec<VerificationFinding> {
let markers = marker_positions(lines);
find_orphan_marker_findings_from_markers(spans, &markers, target_path)
}
fn find_orphan_marker_findings_from_markers(
spans: &[&ManagedSpanRecord],
markers: &BTreeMap<String, (usize, usize, String)>,
target_path: &str,
) -> Vec<VerificationFinding> {
let tracked_ids: HashSet<&str> = spans.iter().map(|span| span.id.as_str()).collect();
let mut findings = Vec::new();
for marker_id in markers.keys() {
if marker_id != "0" && !tracked_ids.contains(marker_id.as_str()) {
findings.push(VerificationFinding {
kind: "orphan_marker".to_string(),
message: format!(
"`{}` contains marker `{}` that is not tracked in `Prayfile.lock`. Remove the marker or run `pray install` to reconcile.",
target_path, marker_id
),
});
}
}
findings
}
pub fn drift_project(
project: &ResolvedProject,
lockfile: &Lockfile,
) -> PrayResult<VerificationReport> {
let (mut report, rendered_targets, fresh_targets) =
collect_verification_report(project, lockfile)?;
let lock_targets = lockfile_targets(lockfile);
for (path, fresh_content) in &fresh_targets {
let normalized_fresh = normalize_line_endings(fresh_content);
let on_disk = rendered_targets
.get(path)
.map(|text| normalize_line_endings(text));
let matches = on_disk.as_ref() == Some(&normalized_fresh);
if !matches {
report.findings.push(VerificationFinding {
kind: "renderer_drift".to_string(),
message: format!("{path} differs from fresh render"),
});
}
if !lock_targets.contains(path) {
report.findings.push(VerificationFinding {
kind: "renderer_drift".to_string(),
message: format!("{path} is not tracked in lockfile"),
});
}
}
if report.findings.is_empty() {
Ok(report)
} else {
Err(PrayError::Verify(format_drift_report(&report)))
}
}
fn marker_positions(lines: &[&str]) -> BTreeMap<String, (usize, usize, String)> {
let mut markers = BTreeMap::new();
let mut active: Option<(String, usize, Vec<&str>)> = None;
for (index, line) in lines.iter().enumerate() {
match parse_marker(line) {
None => {
if let Some((_, _, body)) = active.as_mut() {
body.push(line);
}
}
Some(ParsedMarker::Ignore) => {}
Some(ParsedMarker::Id(id)) => match active.take() {
None => {
active = Some((id.to_string(), index + 1, Vec::new()));
}
Some((open_id, open_line, body)) if open_id == id => {
let checksum = checksum_managed_body_line_refs(&body);
markers.insert(open_id, (open_line, index + 1, checksum));
}
Some(previous) => {
active = Some(previous);
}
},
}
}
markers
}
enum ParsedMarker<'a> {
Ignore,
Id(&'a str),
}
fn parse_marker(line: &str) -> Option<ParsedMarker<'_>> {
let trimmed = line.trim();
let remainder = trimmed.strip_prefix("<!-- pray:")?;
let id = remainder.strip_suffix(" -->")?;
if id == "0 ignore-comments" {
return Some(ParsedMarker::Ignore);
}
if id
.chars()
.all(|character| character.is_ascii_lowercase() || character.is_ascii_digit())
{
return Some(ParsedMarker::Id(id));
}
None
}
fn lockfile_targets(lockfile: &Lockfile) -> BTreeSet<String> {
lockfile
.target
.iter()
.flat_map(|target| target.outputs.iter().cloned())
.collect()
}