Support postgres with SSL requirements

This stupidly just strips any ?sslmode= arguments from the postgres URI
and enables SSL without verification regardless of which mode the user
specifies.

Author: ananace

Cargo.lock

@@ -9,9 +9,12 @@ edition = "2018"
 clap = "2.33.0"
 indicatif = "0.14.0"
 jemallocator = "0.3.2"
+openssl = "0.10.32"
 postgres = "0.17.0"
+postgres-openssl = "0.3.0"
 rand = "0.7.2"
 rayon = "1.3.0"
+regex = "1.4.3"
 string_cache = "0.8.0"
 
 [dependencies.state-map]

Cargo.toml

@@ -14,7 +14,10 @@
 
 use indicatif::{ProgressBar, ProgressStyle};
 use postgres::{fallible_iterator::FallibleIterator, Client};
+use openssl::ssl::{SslConnector, SslMethod, SslVerifyMode};
+use postgres_openssl::MakeTlsConnector;
 use rand::{distributions::Alphanumeric, thread_rng, Rng};
+use regex::Regex;
 use std::{borrow::Cow, collections::BTreeMap, fmt, iter};
 
 use super::StateGroupEntry;
@@ -26,7 +29,18 @@ pub fn get_data_from_db(
     room_id: &str,
     max_state_group: Option<i64>,
 ) -> BTreeMap<i64, StateGroupEntry> {
-    let mut client = Client::connect(db_url, postgres::NoTls).unwrap();
+    let mut client : postgres::Client;
+
+    if db_url.contains("sslmode=") {
+        let mut builder = SslConnector::builder(SslMethod::tls()).unwrap();
+        builder.set_verify(SslVerifyMode::NONE);
+        let connector = MakeTlsConnector::new(builder.build());
+
+        let re = Regex::new(r"(?:sslmode=[^&]+&|\??sslmode=[^&]+)").unwrap();
+        client = Client::connect(&re.replace(db_url, ""), connector).unwrap();
+    } else {
+        client = Client::connect(db_url, postgres::NoTls).unwrap();
+    }
 
     let mut state_group_map = get_initial_data_from_db(&mut client, room_id, max_state_group);