From a1fd5d7f61160584b4ac166cc2dae2ff9579d4a2 Mon Sep 17 00:00:00 2001
From: amazedkoumei <amazed.koumei@gmail.com>
Date: Mon, 26 Jun 2017 01:13:31 +0900
Subject: [PATCH] more free pgconfig by .env (#3909)

* more free pgconfig for streaming by .env

* fix wrong default values

* database.yml read ENV as same as streaming server
---
 config/database.yml | 8 ++++++++
 streaming/index.js  | 6 +++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/config/database.yml b/config/database.yml
index 39393e93ad..079ea7b4a9 100644
--- a/config/database.yml
+++ b/config/database.yml
@@ -7,6 +7,10 @@ default: &default
 development:
   <<: *default
   database: mastodon_development
+  username: <%= ENV['DB_USER'] %>
+  password: <%= ENV['DB_PASS'] %>
+  host: <%= ENV['DB_HOST'] %>
+  port: <%= ENV['DB_PORT'] %>
 
 # Warning: The database defined as "test" will be erased and
 # re-generated from your development database when you run "rake".
@@ -14,6 +18,10 @@ development:
 test:
   <<: *default
   database: mastodon_test<%= ENV['TEST_ENV_NUMBER'] %>
+  username: <%= ENV['TEST_DB_USER'] %>
+  password: <%= ENV['TEST_DB_PASS'] %>
+  host: <%= ENV['TEST_DB_HOST'] %>
+  port: <%= ENV['TEST_DB_PORT'] %>
 
 production:
   <<: *default
diff --git a/streaming/index.js b/streaming/index.js
index 156e1d4bc5..701cb2f557 100644
--- a/streaming/index.js
+++ b/streaming/index.js
@@ -78,7 +78,11 @@ const startWorker = (workerId) => {
 
   const pgConfigs = {
     development: {
-      database: 'mastodon_development',
+      user:     process.env.DB_USER || pg.defaults.user,
+      password: process.env.DB_PASS || pg.defaults.password,
+      database: process.env.DB_NAME || 'mastodon_development',
+      host:     process.env.DB_HOST || pg.defaults.host,
+      port:     process.env.DB_PORT || pg.defaults.port,
       max:      10,
     },