pgevolve-core 0.3.6

Postgres declarative schema management — core library (parser, IR, diff, planner) powering the pgevolve CLI.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

//! Lint engine. Spec §12.
//!
//! Two-stage check:
//! 1. [`universal::check_universal`] — rules that apply to every profile.
//! 2. [`profile::check_profile`] — path-shape rules specific to one layout
//!    profile (schema-mirror, kind-grouped, feature-grouped, free-form,
//!    custom).
//!
//! Callers compose both via [`run`].

pub mod finding;
pub mod profile;
pub(crate) mod rules;
pub mod source_tree;
pub mod universal;

use std::path::Path;

pub use finding::{Finding, Severity};
pub use profile::{CustomProfile, PathPattern, Profile, check_profile};
pub use source_tree::{ObjectKey, SourceTree};
pub use universal::{
    LINT_AT_PLAN_RULES, check_changeset, check_cluster_changeset, check_plan_time_catalog,
    check_universal, check_universal_with_cluster,
};

/// Inputs for [`run`].
#[derive(Debug, Clone)]
pub struct LintInputs<'a> {
    /// Parsed source tree.
    pub tree: &'a SourceTree,
    /// `[managed]` config block.
    pub managed: &'a ManagedConfig,
    /// Profile to apply.
    pub profile: &'a Profile,
    /// Source-tree root, used to compute relative paths for profile checks.
    pub schema_dir: &'a Path,
}

/// `[managed]` view passed to the lint engine.
///
/// Mirrors the binary's `ManagedConfig` but lives in the core crate so the
/// lint engine doesn't depend on the binary.
#[derive(Debug, Clone, Default)]
pub struct ManagedConfig {
    /// Schemas under pgevolve's control.
    pub schemas: Vec<crate::identifier::Identifier>,
}

/// Run universal rules + the configured profile against `inputs`.
pub fn run(inputs: &LintInputs<'_>) -> Vec<Finding> {
    let mut out = check_universal(inputs.tree, inputs.managed);
    out.extend(check_profile(
        inputs.profile,
        inputs.tree,
        inputs.schema_dir,
    ));
    out
}

impl SourceTree {
    /// Parse `root` into a `SourceTree`.
    pub fn parse(root: &Path, ignores: &[glob::Pattern]) -> Result<Self, crate::parse::ParseError> {
        let (catalog, string_locations) =
            crate::parse::parse_directory_with_locations(root, ignores)?;
        let mut object_locations = std::collections::HashMap::new();
        for s in &catalog.schemas {
            if let Some(loc) = string_locations.get(&s.name.to_string()) {
                object_locations.insert(ObjectKey::Schema(s.name.clone()), loc.clone());
            }
        }
        for t in &catalog.tables {
            if let Some(loc) = string_locations.get(&t.qname.to_string()) {
                object_locations.insert(ObjectKey::Table(t.qname.clone()), loc.clone());
            }
        }
        for i in &catalog.indexes {
            if let Some(loc) = string_locations.get(&i.qname.to_string()) {
                object_locations.insert(ObjectKey::Index(i.qname.clone()), loc.clone());
            }
        }
        for seq in &catalog.sequences {
            if let Some(loc) = string_locations.get(&seq.qname.to_string()) {
                object_locations.insert(ObjectKey::Sequence(seq.qname.clone()), loc.clone());
            }
        }
        Ok(Self::new(catalog, object_locations))
    }
}