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
  10 const database = {}
  11
  12 const sequelize = new Sequelize(constants.CONFIG.DATABASE.DBNAME, 'peertube', 'peertube', {
  13   dialect: 'postgres',
  14   host: constants.CONFIG.DATABASE.HOSTNAME,
  15   port: constants.CONFIG.DATABASE.PORT
  16 })
  17
  18 const modelDirectory = path.join(__dirname, '..', 'models')
  19 fs.readdir(modelDirectory, function (err, files) {
  20   if (err) throw err
  21
  22   files.filter(function (file) {
  23     if (file === 'utils.js') return false
  24
  25     return true
  26   })
  27   .forEach(function (file) {
  28     const model = sequelize.import(path.join(modelDirectory, file))
  29
  30     database[model.name] = model
  31   })
  32
  33   Object.keys(database).forEach(function (modelName) {
  34     if ('associate' in database[modelName]) {
  35       database[modelName].associate(database)
  36     }
  37   })
  38
  39   logger.info('Database is ready.')
  40 })
  41
  42 database.sequelize = sequelize
  43 database.Sequelize = Sequelize
  44
  45 // ---------------------------------------------------------------------------
  46
  47 module.exports = database