index.js 997 B

12345678910111213141516171819202122232425262728293031323334353637383940
  1. 'use strict';
  2. const mongoose = require('mongoose');
  3. let lib = {
  4. connection: null,
  5. schemas: {},
  6. models: {},
  7. init: (url, cb) => {
  8. lib.connection = mongoose.connect(url).connection;
  9. lib.connection.on('error', err => console.log('Database error: ' + err.message));
  10. lib.connection.once('open', _ => {
  11. lib.schemas = {
  12. song: new mongoose.Schema(require(`./schemas/song`)),
  13. queueSong: new mongoose.Schema(require(`./schemas/queueSong`)),
  14. station: new mongoose.Schema(require(`./schemas/station`)),
  15. user: new mongoose.Schema(require(`./schemas/user`)),
  16. news: new mongoose.Schema(require(`./schemas/news`))
  17. };
  18. lib.models = {
  19. song: mongoose.model('song', lib.schemas.song),
  20. queueSong: mongoose.model('queueSong', lib.schemas.queueSong),
  21. station: mongoose.model('station', lib.schemas.station),
  22. user: mongoose.model('user', lib.schemas.user),
  23. news: mongoose.model('news', lib.schemas.news)
  24. };
  25. cb();
  26. });
  27. }
  28. };
  29. module.exports = lib;