wof 1.0.0

The Who's On First rust library and command line.
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
90
91

use crate::postgres;
use crate::utils::ResultExit;
use clap::Parser;
use log::info;

#[derive(Debug, Parser)]
pub struct Postgres {
  /// WOF data directories to import
  #[arg(default_value = ".")]
  pub directories: Vec<String>,
  /// The IP or hostname of the postgreSQL database.
  #[arg(long = "host", default_value = "127.0.0.1", env = "WOF_PG_HOST")]
  pub host: String,
  /// The postgreSQL user name to use.
  #[arg(
    short = 'u',
    long = "user",
    default_value = "wof",
    env = "WOF_PG_USERNAME"
  )]
  pub user: String,
  /// The postgreSQL database name to use.
  #[arg(
    short = 'd',
    long = "dbname",
    default_value = "gis",
    env = "WOF_PG_DBNAME"
  )]
  pub dbname: String,
  /// The postgreSQL database port to use.
  #[arg(
    short = 'p',
    long = "port",
    default_value = "5432",
    env = "WOF_PG_PORT"
  )]
  pub port: u16,
  /// The postgreSQL database port to use.
  #[arg(short = 'W', long = "password", env = "WOF_PG_PASSWORD")]
  pub password: Option<String>,
  /// The SIRID to use for geometry storage. Default value is 4326, common usage is also 3857.
  #[arg(
    short = 's',
    long = "srid",
    default_value = "4326",
    env = "WOF_PG_SRID"
  )]
  pub srid: i32,
  /// Don't insert deprecated features.
  #[arg(long = "no-deprecated")]
  pub no_deprecated: bool,
  /// Display timings during the build process, implies verbose.
  #[arg(long = "timings")]
  pub timings: bool,
  /// Activate verbose mode.
  #[arg(short = 'v', long = "verbose")]
  pub verbose: bool,
}

impl Postgres {
  pub fn exec(&self) {
    crate::utils::logger::set_verbose(self.verbose || self.timings, "wof::build::postgres")
      .expect_exit("Can't init logger.");
    let mut config = postgres::Config::new();

    config.dbname(&self.dbname);
    config.user(&self.user);
    config.host(&self.host);
    config.port(self.port);
    if let Some(password) = &self.password {
      config.password(&password);
    }

    info!("Connecting to database: `{:?}`", config.get_hosts());
    let mut postgres =
      postgres::Postgres::new(config, Some(self.srid)).expect_exit("Can't open the database");

    info!("Creating tables and indexes.");
    postgres.create_tables().expect_exit("Can't create tables");

    crate::commands::build::build_database(&self.directories, self.timings, &mut |buffer, file| {
      if let Some(buffer) = buffer {
        postgres.add_string(buffer)
      } else if let Some(file) = file {
        postgres.add_file(file)
      } else {
        Ok(())
      }
    });
  }
}