From f3740e5f37dcdaf921329a9e9c90ccf499a14ed2 Mon Sep 17 00:00:00 2001
From: Paul Bienkowski <pb@opatut.de>
Date: Wed, 18 Nov 2020 22:17:00 +0100
Subject: [PATCH] feat: generate TrackData points when saving tracks through
 PUT, and return raw body through GET /:track

---
 models/Track.js      |  5 ++--
 routes/api/tracks.js | 58 +++++++++++++++++++++++---------------------
 2 files changed, 34 insertions(+), 29 deletions(-)

diff --git a/models/Track.js b/models/Track.js
index edb165d..557e633 100644
--- a/models/Track.js
+++ b/models/Track.js
@@ -30,7 +30,7 @@ TrackSchema.methods.slugify = function() {
   this.slug = slug(this.title) + '-' + (Math.random() * Math.pow(36, 6) | 0).toString(36);
 };
 
-TrackSchema.methods.toJSONFor = function(user){
+TrackSchema.methods.toJSONFor = function (user, include) {
   return {
     slug: this.slug,
     title: this.title,
@@ -39,7 +39,8 @@ TrackSchema.methods.toJSONFor = function(user){
     updatedAt: this.updatedAt,
     visibleForAll: this.author ? this.author.areTracksVisibleForAll : false,
     visible: this.visible,
-    author: this.author.toProfileJSONFor(user)
+    author: this.author.toProfileJSONFor(user),
+    ...(include?.body ? { body: this.body } : {}),
   };
 };
 
diff --git a/routes/api/tracks.js b/routes/api/tracks.js
index 019d279..eead1cf 100644
--- a/routes/api/tracks.js
+++ b/routes/api/tracks.js
@@ -313,43 +313,47 @@ router.get('/:track', auth.optional, function (req, res, next) {
     .then(function (results) {
       const user = results[0];
 
-      return res.json({ track: req.track.toJSONFor(user) });
+      return res.json({ track: req.track.toJSONFor(user, { body: true }) });
     })
     .catch(next);
 });
 
 // update track
-router.put('/:track', auth.required, function (req, res, next) {
-  User.findById(req.payload.id).then(function (user) {
-    if (req.track.author._id.toString() === req.payload.id.toString()) {
-      if (typeof req.body.track.title !== 'undefined') {
-        req.track.title = req.body.track.title;
-      }
+router.put('/:track', auth.required, async function (req, res, next) {
+  const user = await User.findById(req.payload.id);
 
-      if (typeof req.body.track.description !== 'undefined') {
-        req.track.description = req.body.track.description;
-      }
+  if (req.track.author._id.toString() !== req.payload.id.toString()) {
+    return res.sendStatus(403);
+  }
 
-      if (typeof req.body.track.body !== 'undefined') {
-        req.track.body = req.body.track.body;
-      }
+  if (typeof req.body.track.title !== 'undefined') {
+    req.track.title = req.body.track.title;
+  }
 
-      if (typeof req.body.track.tagList !== 'undefined') {
-        req.track.tagList = req.body.track.tagList;
-      }
-      req.track.visible = req.body.track.visible;
-      console.log('saving track');
+  if (typeof req.body.track.description !== 'undefined') {
+    req.track.description = req.body.track.description;
+  }
 
-      req.track
-        .save()
-        .then(function (track) {
-          return res.json({ track: track.toJSONFor(user) });
-        })
-        .catch(next);
-    } else {
-      return res.sendStatus(403);
+  if (req.body.track.body?.trim()) {
+    req.track.body = req.body.track.body.trim();
+
+    let trackData = await TrackData.findById(req.track.trackData);
+    if (!trackData) {
+      trackData = new TrackData();
+      req.track.trackData = trackData._id;
     }
-  });
+    trackData.points = [];
+    addPointsToTrack({ trackData }, req.track.body);
+    await trackData.save();
+  }
+
+  if (typeof req.body.track.tagList !== 'undefined') {
+    req.track.tagList = req.body.track.tagList;
+  }
+  req.track.visible = req.body.track.visible;
+
+  const track = await req.track.save();
+  return res.json({ track: track.toJSONFor(user) });
 });
 
 // delete track