Init

server/ApiController.js

@@ -0,0 +1,143 @@
+const express = require('express')
+const Logger = require('./Logger')
+
+class ApiController {
+  constructor(db, scanner, auth, streamManager, emitter) {
+    this.db = db
+    this.scanner = scanner
+    this.auth = auth
+    this.streamManager = streamManager
+    this.emitter = emitter
+
+    this.router = express()
+    this.init()
+  }
+
+  init() {
+    this.router.get('/find/:method', this.find.bind(this))
+
+    this.router.get('/audiobooks', this.getAudiobooks.bind(this))
+    this.router.get('/audiobook/:id', this.getAudiobook.bind(this))
+    this.router.delete('/audiobook/:id', this.deleteAudiobook.bind(this))
+    this.router.patch('/audiobook/:id/tracks', this.updateAudiobookTracks.bind(this))
+    this.router.patch('/audiobook/:id', this.updateAudiobook.bind(this))
+
+    this.router.get('/metadata/:id/:trackIndex', this.getMetadata.bind(this))
+    this.router.patch('/match/:id', this.match.bind(this))
+
+    this.router.delete('/user/audiobook/:id', this.resetUserAudiobookProgress.bind(this))
+
+    this.router.post('/authorize', this.authorize.bind(this))
+  }
+
+  find(req, res) {
+    this.scanner.find(req, res)
+  }
+
+  async getMetadata(req, res) {
+    var metadata = await this.scanner.fetchMetadata(req.params.id, req.params.trackIndex)
+    res.json(metadata)
+  }
+
+  authorize(req, res) {
+    if (!req.user) {
+      Logger.error('Invalid user in authorize')
+      return res.sendStatus(401)
+    }
+    res.json({ user: req.user })
+  }
+
+  getAudiobooks(req, res) {
+    Logger.info('Get Audiobooks')
+    var audiobooksMinified = this.db.audiobooks.map(ab => ab.toJSONMinified())
+    res.json(audiobooksMinified)
+  }
+
+  getAudiobook(req, res) {
+    var audiobook = this.db.audiobooks.find(a => a.id === req.params.id)
+    if (!audiobook) return res.sendStatus(404)
+    res.json(audiobook.toJSONExpanded())
+  }
+
+  async deleteAudiobook(req, res) {
+    var audiobook = this.db.audiobooks.find(a => a.id === req.params.id)
+    if (!audiobook) return res.sendStatus(404)
+
+    // Remove audiobook from users
+    for (let i = 0; i < this.db.users.length; i++) {
+      var user = this.db.users[i]
+      var madeUpdates = user.resetAudiobookProgress(audiobook.id)
+      if (madeUpdates) {
+        await this.db.updateEntity('user', user)
+      }
+    }
+
+    // remove any streams open for this audiobook
+    var streams = this.streamManager.streams.filter(stream => stream.audiobookId === audiobook.id)
+    for (let i = 0; i < streams.length; i++) {
+      var stream = streams[i]
+      var client = stream.client
+      await stream.close()
+      if (client && client.user) {
+        client.user.stream = null
+        client.stream = null
+        this.db.updateUserStream(client.user.id, null)
+      }
+    }
+
+    await this.db.removeEntity('audiobook', audiobook.id)
+
+    this.emitter('audiobook_removed', audiobook.toJSONMinified())
+    res.sendStatus(200)
+  }
+
+  async updateAudiobookTracks(req, res) {
+    var audiobook = this.db.audiobooks.find(a => a.id === req.params.id)
+    if (!audiobook) return res.sendStatus(404)
+    var files = req.body.files
+    Logger.info(`Updating audiobook tracks called ${audiobook.id}`)
+    audiobook.updateAudioTracks(files)
+    await this.db.updateAudiobook(audiobook)
+    this.emitter('audiobook_updated', audiobook.toJSONMinified())
+    res.json(audiobook.toJSON())
+  }
+
+  async updateAudiobook(req, res) {
+    var audiobook = this.db.audiobooks.find(a => a.id === req.params.id)
+    if (!audiobook) return res.sendStatus(404)
+    var hasUpdates = audiobook.update(req.body)
+    if (hasUpdates) {
+      await this.db.updateAudiobook(audiobook)
+    }
+    this.emitter('audiobook_updated', audiobook.toJSONMinified())
+    res.json(audiobook.toJSON())
+  }
+
+  async match(req, res) {
+    var body = req.body
+    var audiobookId = req.params.id
+    var audiobook = this.db.audiobooks.find(ab => ab.id === audiobookId)
+    var bookData = {
+      olid: body.id,
+      publish_year: body.first_publish_year,
+      description: body.description,
+      title: body.title,
+      author: body.author,
+      cover: body.cover
+    }
+    audiobook.setBook(bookData)
+    await this.db.updateAudiobook(audiobook)
+
+    this.emitter('audiobook_updated', audiobook.toJSONMinified())
+
+    res.sendStatus(200)
+  }
+
+  async resetUserAudiobookProgress(req, res) {
+    req.user.resetAudiobookProgress(req.params.id)
+    await this.db.updateEntity('user', req.user)
+    this.emitter('user_updated', req.user.toJSONForBrowser())
+    res.sendStatus(200)
+  }
+}
+module.exports = ApiController