postgresfixture 0.2.2

Easily create and manage PostgreSQL clusters on demand for testing and development.
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

#[macro_use]
extern crate clap;
extern crate postgresfixture;

use std::env;
use std::path::PathBuf;
use std::process::exit;

fn main() {
    exit(match parse_args().subcommand() {
        ("shell", Some(args)) => {
            let database_dir: PathBuf = args.value_of("database-dir").unwrap().into();
            let database_name = match args.value_of("database-name") {
                Some(name) => name.into(),
                None => match env::var("PGDATABASE") {
                    Ok(name) => name,
                    Err(_) => "data".into(),
                },
            };
            shell(database_dir, &database_name)
        }
        (command, _) => {
            unreachable!("subcommand branch for {:?} is missing", command);
        }
    });
}

/// Parse command-line arguments.
fn parse_args() -> clap::ArgMatches<'static> {
    use clap::{App, AppSettings, Arg, SubCommand};
    App::new("rust-postgresfixture")
        .setting(AppSettings::DisableHelpSubcommand)
        .setting(AppSettings::SubcommandRequired)
        .setting(AppSettings::VersionlessSubcommands)
        .version(crate_version!())
        .author(crate_authors!())
        .about("Work with ephemeral PostgreSQL clusters.")
        .after_help(concat!(
            "Based on the Python postgresfixture library ",
            "<https://pypi.python.org/pypi/postgresfixture>."
        ))
        .subcommand(
            SubCommand::with_name("shell")
                .about("Start a psql shell, creating and starting the cluster as necessary.")
                .arg(
                    Arg::with_name("database-dir")
                        .help("The directory in which to place, or find, the cluster.")
                        .short("D")
                        .long("datadir")
                        .value_name("DIR")
                        .default_value("db"),
                )
                .arg(
                    Arg::with_name("database-name")
                        .help(concat!(
                    "The database to connect to. The default is taken from the PGDATABASE ",
                    "environment variable. If that is not set, a database named 'data' will ",
                    "be created."))
                        .value_name("PGDATABASE"),
                ),
        )
        .get_matches()
}

fn shell(database_dir: PathBuf, database_name: &str) -> i32 {
    let cluster = postgresfixture::Cluster::new(
        match database_dir.is_absolute() {
            false => env::current_dir().unwrap().join(database_dir),
            true => database_dir,
        },
        postgresfixture::Runtime::default(),
    );
    cluster.start().expect("could not start cluster");
    if !cluster
        .databases()
        .unwrap()
        .contains(&database_name.to_string())
    {
        cluster.createdb(database_name).unwrap();
    }
    cluster.shell(&database_name).expect("shell failed");
    cluster.stop().expect("could not stop cluster");
    0
}