Merge pull request #170 from bvp/master

Initial PostgreSQL support

README.md

@@ -17,7 +17,7 @@ Shiori is a simple bookmarks manager written in Go language. Intended as a simpl
 - Simple and clean command line interface.
 - Simple and pretty web interface for those who don't want to use a command line app.
 - Portable, thanks to its single binary format.
-- Support sqlite3 and MySQL as its database.
+- Support sqlite3, PostgreSQL and MySQL as its database.
 - Where possible, by default `shiori` will parse the readable content and create an offline archive of the webpage.
 - [BETA] [web extension](https://github.com/go-shiori/shiori-web-ext) support for Firefox and Chrome.
 

go.mod

@@ -11,7 +11,7 @@ require (
 	github.com/gofrs/uuid v3.2.0+incompatible
 	github.com/jmoiron/sqlx v1.2.0
 	github.com/julienschmidt/httprouter v1.2.0
-	github.com/lib/pq v1.1.1 // indirect
+	github.com/lib/pq v1.1.1
 	github.com/mattn/go-colorable v0.1.1 // indirect
 	github.com/mattn/go-isatty v0.0.7 // indirect
 	github.com/mattn/go-sqlite3 v1.10.0

internal/cmd/root.go

@@ -111,7 +111,17 @@ func openDatabase() (database.DB, error) {
 		connString := fmt.Sprintf("%s:%s@%s/%s", user, password, dbAddress, dbName)
 		return database.OpenMySQLDatabase(connString)
 	}
-
+	// Check if it uses PostgreSQL
+	if dbms, _ := os.LookupEnv("SHIORI_DBMS"); dbms == "postgresql" {
+		host, _ := os.LookupEnv("SHIORI_PG_HOST")
+		port, _ := os.LookupEnv("SHIORI_PG_PORT")
+		user, _ := os.LookupEnv("SHIORI_PG_USER")
+		password, _ := os.LookupEnv("SHIORI_PG_PASS")
+		dbName, _ := os.LookupEnv("SHIORI_PG_NAME")
+
+		connString := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbName)
+		return database.OpenPGDatabase(connString)
+	}
 	// If not, just uses SQLite
 	dbPath := fp.Join(dataDir, "shiori.db")
 	return database.OpenSQLiteDatabase(dbPath)