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
use crate::cli::parser::PgschemaArgs;
use crate::commands::base::{Command, CommandContext};
use anyhow::Result;
/// Implementation of the `pgschema` command.
///
/// This struct holds the specific arguments parsed by `clap` and
/// implements the [Command] trait to execute Property Graph Schema logic.
pub struct PgschemaCommand {
/// Arguments specific to Property Graph Schema command.
args: PgschemaArgs,
}
impl PgschemaCommand {
pub fn new(args: PgschemaArgs) -> Self {
Self { args }
}
}
impl Command for PgschemaCommand {
/// Returns the unique identifier for this command.
fn name(&self) -> &'static str {
"pgschema"
}
/// Executes the Property Graph Schema command logic.
///
/// With no `--schema`, there is nothing new to load, so this just
/// re-serializes whatever PGSchema is already loaded in the session
/// (useful in the interactive shell, where state persists across
/// commands).
fn execute(&self, ctx: &mut CommandContext) -> Result<()> {
let pg_schema_format = self.args.schema_format.into();
let result_pg_schema_format = self.args.result_schema_format.into();
if let Some(schema) = &self.args.schema {
ctx.rudof
.load_pg_schema(schema)
.with_pg_schema_format(&pg_schema_format)
.execute()?;
}
ctx.rudof
.serialize_pg_schema(&mut ctx.writer)
.with_result_pg_schema_format(&result_pg_schema_format)
.execute()?;
Ok(())
}
}