audiobookshelf/server/Database.js

107 lines
3.4 KiB
JavaScript
Raw Normal View History

const Path = require('path')
const { Sequelize } = require('sequelize')
const Logger = require('./Logger')
class Database {
constructor() {
this.sequelize = null
}
2023-03-15 23:42:35 +01:00
get models() {
return this.sequelize?.models || {}
}
async init() {
if (!await this.connect()) {
throw new Error('Database connection failed')
}
await this.buildModels()
Logger.info(`[Database] Db initialized`, Object.keys(this.sequelize.models))
}
async connect() {
const dbPath = Path.join(global.ConfigPath, 'database.sqlite')
Logger.info(`[Database] Initializing db at "${dbPath}"`)
this.sequelize = new Sequelize({
dialect: 'sqlite',
storage: dbPath,
logging: false
})
try {
await this.sequelize.authenticate()
Logger.info(`[Database] Db connection was successful`)
return true
} catch (error) {
Logger.error(`[Database] Failed to connect to db`, error)
return false
}
}
buildModels() {
require('./models/User')(this.sequelize)
2023-03-12 20:51:45 +01:00
require('./models/FileMetadata')(this.sequelize)
require('./models/Library')(this.sequelize)
require('./models/LibraryFolder')(this.sequelize)
require('./models/LibraryItem')(this.sequelize)
require('./models/EBookFile')(this.sequelize)
require('./models/Book')(this.sequelize)
require('./models/Podcast')(this.sequelize)
require('./models/PodcastEpisode')(this.sequelize)
require('./models/MediaProgress')(this.sequelize)
require('./models/LibraryFile')(this.sequelize)
require('./models/Person')(this.sequelize)
require('./models/AudioBookmark')(this.sequelize)
2023-03-15 23:42:35 +01:00
require('./models/MediaFile')(this.sequelize)
require('./models/MediaStream')(this.sequelize)
2023-03-12 20:51:45 +01:00
require('./models/AudioTrack')(this.sequelize)
require('./models/BookAuthor')(this.sequelize)
require('./models/BookChapter')(this.sequelize)
require('./models/Genre')(this.sequelize)
require('./models/BookGenre')(this.sequelize)
require('./models/BookNarrator')(this.sequelize)
require('./models/Series')(this.sequelize)
require('./models/BookSeries')(this.sequelize)
require('./models/Tag')(this.sequelize)
require('./models/BookTag')(this.sequelize)
require('./models/PodcastTag')(this.sequelize)
require('./models/Collection')(this.sequelize)
require('./models/CollectionBook')(this.sequelize)
require('./models/Playlist')(this.sequelize)
require('./models/PlaylistMediaItem')(this.sequelize)
require('./models/Device')(this.sequelize)
require('./models/PlaybackSession')(this.sequelize)
2023-03-13 23:13:31 +01:00
require('./models/PlaybackSessionListenTime')(this.sequelize)
require('./models/Feed')(this.sequelize)
require('./models/FeedEpisode')(this.sequelize)
require('./models/Setting')(this.sequelize)
require('./models/LibrarySetting')(this.sequelize)
2023-03-15 23:42:35 +01:00
require('./models/Notification')(this.sequelize)
require('./models/UserPermission')(this.sequelize)
2023-03-15 23:50:47 +01:00
return this.sequelize.sync({ force: true })
}
async createTestUser() {
const User = this.sequelize.models.User
let user = await User.findOne({
where: {
username: 'Tester'
}
})
if (user) {
Logger.info(`[Database] Tester user was found`, user.toJSON())
} else {
user = await User.create({ username: 'Tester' })
Logger.info(`[Database] Created Tester user`, user.toJSON())
}
}
}
module.exports = new Database()