Add ssl support

Allow disabling ssl certificate verification.

Author: foxylion

README.md

@@ -26,6 +26,9 @@ Usage: java -jar build/libs/psql-multi-all.jar [options]
        Default: 127.0.0.1:5432
     --include, -i
        Include databases with the following regex pattern
+    --no-ssl-verify
+       Verify SSL certificate
+       Default: true
     --pass, -p
        Password to authenticate
        Default: postgres
@@ -35,6 +38,9 @@ Usage: java -jar build/libs/psql-multi-all.jar [options]
        When using a SELECT statement as the command results will be printed to
        console
        Default: false
+    --ssl
+       Use a SSL connection to the database server
+       Default: false
     --user, -u
        User to authenticate
        Default: postgres

src/main/java/de/foxylion/psql/multi/CommandlineParams.java

@@ -31,4 +31,10 @@ class CommandlineParams {
 
 	@Parameter(names = { "--force", "-f" }, description = "Force to continue when executed query fails on a database")
 	boolean force = false;
-}
+
+	@Parameter(names = { "--ssl" }, description = "Use a SSL connection to the database server")
+	boolean ssl = false;
+
+	@Parameter(names = { "--no-ssl-verify" }, description = "Verify SSL certificate")
+	boolean noSslVerify = false;
+}

src/main/java/de/foxylion/psql/multi/PsqlFactory.java

@@ -2,7 +2,7 @@
 
 public class PsqlFactory {
 
-	public Psql getConnection(String host, String database, String user, String password) {
-		return new PsqlImpl(host, database, user, password);
+	public Psql getConnection(String host, String database, String user, String password, boolean ssl, boolean verifySsl) {
+		return new PsqlImpl(host, database, user, password, ssl, verifySsl);
 	}
 }

src/main/java/de/foxylion/psql/multi/PsqlImpl.java

@@ -14,9 +14,10 @@ public class PsqlImpl implements Psql {
 
 	private final Connection connection;
 
-	public PsqlImpl(String host, String db, String user, String pass) {
+	public PsqlImpl(String host, String db, String user, String pass, boolean ssl, boolean noSslVerify) {
 		try {
-			connection = DriverManager.getConnection("jdbc:postgresql://" + host + "/" + db, user, pass);
+			String sslFactory = noSslVerify ? "&sslfactory=org.postgresql.ssl.NonValidatingFactory" : "";
+			connection = DriverManager.getConnection("jdbc:postgresql://" + host + "/" + db + "?ssl=" + ssl + sslFactory, user, pass);
 		} catch (SQLException e) {
 			throw new RuntimeException("Failed to open database connection: " + e.getMessage());
 		}

src/main/java/de/foxylion/psql/multi/PsqlMulti.java

@@ -27,7 +27,7 @@ public void run() {
 	}
 
 	private void runQuery(String db) {
-		try (Psql psql = psqlFactory.getConnection(params.host, db, params.user, params.pass)) {
+		try (Psql psql = psqlFactory.getConnection(params.host, db, params.user, params.pass, params.ssl, params.noSslVerify)) {
 			if (!params.showResults) {
 				psql.execute(params.command);
 			} else {
@@ -44,7 +44,7 @@ private void runQuery(String db) {
 	}
 
 	private List<String> collectDatabases() {
-		try (Psql psql = psqlFactory.getConnection(params.host, "postgres", params.user, params.pass)) {
+		try (Psql psql = psqlFactory.getConnection(params.host, "postgres", params.user, params.pass, params.ssl, params.noSslVerify)) {
 			return psql.fetch("SELECT datname FROM pg_database WHERE datistemplate = false;").stream()
 					.map((row) -> row.get("datname")) //
 					.collect(Collectors.toList());

src/test/java/de/foxylion/psql/multi/PsqlMultiTest.java

@@ -1,6 +1,7 @@
 package de.foxylion.psql.multi;
 
 import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.anyBoolean;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
@@ -19,7 +20,7 @@ void setup() {
 		psqlMocks = new HashMap<>();
 
 		psqlFactory = mock(PsqlFactory.class);
-		when(psqlFactory.getConnection(any(), any(), any(), any())).then((a) -> {
+		when(psqlFactory.getConnection(any(), any(), any(), any(), anyBoolean(), anyBoolean())).then((a) -> {
 			String dbName = a.getArgumentAt(2, String.class);
 			if(!psqlMocks.containsKey(dbName)) {
 				Psql psql = mock(Psql.class);