server/initializers/database.js

   1 'use strict'
   2
   3 const fs = require('fs')
   4 const path = require('path')
   5 const Sequelize = require('sequelize')
   6
   7 const constants = require('../initializers/constants')
   8 const logger = require('../helpers/logger')
   9 const utils = require('../helpers/utils')
  10
  11 const database = {}
  12
  13 const sequelize = new Sequelize(constants.CONFIG.DATABASE.DBNAME, 'peertube', 'peertube', {
  14   dialect: 'postgres',
  15   host: constants.CONFIG.DATABASE.HOSTNAME,
  16   port: constants.CONFIG.DATABASE.PORT,
  17   benchmark: utils.isTestInstance(),
  18
  19   logging: function (message, benchmark) {
  20     let newMessage = message
  21     if (benchmark !== undefined) {
  22       newMessage += ' | ' + benchmark + 'ms'
  23     }
  24
  25     logger.debug(newMessage)
  26   }
  27 })
  28
  29 const modelDirectory = path.join(__dirname, '..', 'models')
  30 fs.readdir(modelDirectory, function (err, files) {
  31   if (err) throw err
  32
  33   files.filter(function (file) {
  34     if (file === 'utils.js') return false
  35
  36     return true
  37   })
  38   .forEach(function (file) {
  39     const model = sequelize.import(path.join(modelDirectory, file))
  40
  41     database[model.name] = model
  42   })
  43
  44   Object.keys(database).forEach(function (modelName) {
  45     if ('associate' in database[modelName]) {
  46       database[modelName].associate(database)
  47     }
  48   })
  49
  50   logger.info('Database is ready.')
  51 })
  52
  53 database.sequelize = sequelize
  54 database.Sequelize = Sequelize
  55
  56 // ---------------------------------------------------------------------------
  57
  58 module.exports = database