Q-Tea/db.js at master · tartufu/Q-Tea · GitHub

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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
/*
 * ===================================================
 * ===================================================
 * ===================================================
 * ===================================================
 * ======             CONFIGURATION          =========
 * ===================================================
 * ===================================================
 * ===================================================
 * ===================================================
 */


const pg = require('pg');
const url = require('url');

var configs;

if( process.env.DATABASE_URL ){

  const params = url.parse(process.env.DATABASE_URL);
  const auth = params.auth.split(':');

  configs = {
    user: auth[0],
    password: auth[1],
    host: params.hostname,
    port: params.port,
    database: params.pathname.split('/')[1],
    ssl: true
  };

}else{
  configs = {
    user: 'changhaoteo',
    host: '127.0.0.1',
    database: 'q-tea',
    port: 5432
  };
}


const pool = new pg.Pool(configs);

pool.on('error', function (err) {
  console.log('idle client error', err.message, err.stack);
});


/*
 * ===================================================
 * ===================================================
 * ===================================================
 * ===================================================
 * ======        REQUIRE MODEL FILES         =========
 * ===================================================
 * ===================================================
 * ===================================================
 * ===================================================
 */

  const allAdminModelsFunction = require('./models/admin');
  const adminModelsObject = allAdminModelsFunction(pool);


  const allAdminUpdateModelsFunction = require('./models/adminupdate');
  const adminUpdateModelsObject = allAdminUpdateModelsFunction(pool);

  const allSearchModelsFunction = require('./models/search');
  const searchModelsObject = allSearchModelsFunction(pool);

  const allReceiptModelsFunction = require('./models/receipt');
  const receiptModelsObject = allReceiptModelsFunction(pool);

  const allUsersModelsFunction = require('./models/user');
  const userModelsObject = allUsersModelsFunction(pool);

/*
 * ===================================================
 * ===================================================
 * ===================================================
 * ===================================================
 * ======          MODULE EXPORTS            =========
 * ===================================================
 * ===================================================
 * ===================================================
 * ===================================================
 */


module.exports = {
  //make queries directly from here
  queryInterface: (text, params, callback) => {
    return pool.query(text, params, callback);
  },

  // get a reference to end the connection pool at server end
  pool:pool,

  /*
   * ADD APP MODELS HERE
   */

  // users: userModelsObject,
  // pokemon: pokemonModelsObject
  admin: adminModelsObject,
  adminupdate: adminUpdateModelsObject,
  search: searchModelsObject,
  receipt: receiptModelsObject,
  user: userModelsObject
};