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

use serde::{Deserialize, Serialize};

#[derive(Debug, Deserialize, Serialize)]
pub struct ConnectionParams {
  server: String,
  port: Option<i32>,
  database: String,
  user: String,
  password: Option<String>,
  tls: bool
}

impl ConnectionParams {
  pub fn new(server: String, port: Option<i32>, database: String, user: String, password: Option<String>, tls: bool) -> ConnectionParams {
    ConnectionParams {
      server,
      port,
      database,
      user,
      password,
      tls
    }
  }

  // pub fn server(&self) -> &String { &self.server }
  // pub fn port(&self) -> &Option<i32> { &self.port }
  // pub fn database(&self) -> &String { &self.database }
  // pub fn user(&self) -> &String { &self.user }
  // pub fn tls(&self) -> bool { self.tls }

  pub fn url(&self) -> String {
    let port_str = self.port.map(|x| format!(":{}", x)).unwrap_or_else(|| "".to_string());
    let user_pass = self
      .password
      .as_ref()
      .map(|p| format!("{}:{}", &self.user, p))
      .unwrap_or_else(|| self.user.to_string());
    format!("postgres://{}@{}{}", &user_pass, &self.server, &port_str)
  }
}

impl Default for ConnectionParams {
  fn default() -> ConnectionParams {
    ConnectionParams {
      server: String::from("localhost"),
      port: None,
      database: String::from("postgres"),
      user: String::from("postgres"),
      password: None,
      tls: false
    }
  }
}