From d0a7696b8081cb8c7f577b61e21bbdd85066793b Mon Sep 17 00:00:00 2001
From: Lynn Smeria <ae@cephea.de>
Date: Wed, 29 Aug 2018 14:45:46 +0200
Subject: [PATCH] Include track info for next/prev track

---
 package.json    |  2 +-
 src/generate.js | 23 ++++++++++++++++++-----
 2 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/package.json b/package.json
index 2b5dbb8..97ac816 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "publikator",
-  "version": "0.11.0",
+  "version": "0.12.0",
   "main": "index.js",
   "repository": "https://github.com/aengl/publikator.git",
   "author": "Lynn Smeria <ae@cephea.de>",
diff --git a/src/generate.js b/src/generate.js
index ff5a777..ff98be4 100644
--- a/src/generate.js
+++ b/src/generate.js
@@ -31,6 +31,21 @@ const getAlbumInfo = (root, tracks) => ({
   tracks,
 });
 
+/**
+ * Creates release information for a single track.
+ */
+const getTrackInfo = (albumInfo, trackIndex) => {
+  const nextTrackIndex = (trackIndex + 1) % albumInfo.trackCount;
+  const previousTrackIndex =
+    trackIndex === 0 ? albumInfo.trackCount - 1 : trackIndex - 1;
+  return {
+    layout: 'track',
+    ...albumInfo.tracks[trackIndex],
+    nextTrack: albumInfo.tracks[nextTrackIndex],
+    previousTrack: albumInfo.tracks[previousTrackIndex],
+  };
+};
+
 module.exports = {
   /**
    * Generates Jekyll-compatible release data
@@ -60,7 +75,8 @@ module.exports = {
 
         // Write track collection
         await Promise.all(
-          tracks.map(async track => {
+          tracks.map(async (_0, i) => {
+            const track = getTrackInfo(albumInfo, i);
             const trackInfoPath = path.resolve(
               trackCollectionRoot,
               baseName,
@@ -69,10 +85,7 @@ module.exports = {
             await fs.ensureFile(trackInfoPath);
             await fs.writeFile(
               trackInfoPath,
-              `---\n${yaml.safeDump({
-                layout: 'track',
-                ...track,
-              })}---\n`
+              `---\n${yaml.safeDump(track)}---\n`
             );
           })
         );