Add Total Track Length property in response json in album routes

Author: lakshaykamat

controllers/albumController.js

@@ -1,15 +1,12 @@
 const Album = require("../models/Album");
 const Song = require("../models/Song");
-const { isAdmin } = require("../utils");
+const {
+  isAdmin,
+  calculateTotalDuration,
+  cleanUpSongsData,
+  enhanceAlbumWithSongsData,
+} = require("../utils");
 const errorHandler = require("../middlewares/errorMiddleware");
-const cleanUpSongs = (songs, album) => {
-  return songs
-    .filter((song) => song !== null)
-    .map(({ albumId, ...songWithoutAlbumId }) => ({
-      ...songWithoutAlbumId,
-      album: album.title,
-    }));
-};
 
 const getAlbum = async (req, res) => {
   try {
@@ -23,15 +20,7 @@ const getAlbum = async (req, res) => {
       return res.status(404).send("Album not found");
     }
 
-    const songPromises = album.tracks.map((songId) =>
-      Song.findById(songId.toString()).select("-__v").select("-_id").lean()
-    );
-    const songs = await Promise.all(songPromises);
-
-    album.tracks = cleanUpSongs(songs, album);
-    album.songs = album.tracks.length;
-
-    res.json(album);
+    res.json(await enhanceAlbumWithSongsData(album));
   } catch (error) {
     errorHandler(res, error);
   }
@@ -41,14 +30,8 @@ const getAllAlbums = async (req, res) => {
   try {
     const albums = await Album.find({}).lean().select("-__v").select("-_id");
 
-    for (const album of albums) {
-      const songPromises = album.tracks.map((songId) =>
-        Song.findById(songId.toString()).select("-__v").select("-_id").lean()
-      );
-      const songs = await Promise.all(songPromises);
-
-      album.tracks = cleanUpSongs(songs, album);
-      album.songs = album.tracks.length;
+    for (let album of albums) {
+      album = await enhanceAlbumWithSongsData(album);
     }
 
     return res.json(albums);
@@ -66,19 +49,13 @@ const searchAlbum = async (req, res) => {
       .select("-__v -_id")
       .lean();
 
-    for (const album of albums) {
-      const songPromises = album.tracks.map((songId) =>
-        Song.findById(songId.toString()).select("-__v").select("-_id").lean()
-      );
-      const songs = await Promise.all(songPromises);
-
-      album.tracks = cleanUpSongs(songs, album);
-      album.songs = album.tracks.length;
+    for (let album of albums) {
+      album = await enhanceAlbumWithSongsData(album);
     }
 
     res.json(albums);
   } catch (error) {
-    console.error("Error searching songs:", error.message);
+    console.error("Error searching albums:", error.message);
     res.status(500).json({ error: "Internal Server Error" });
   }
 };

utils/index.js

@@ -1,4 +1,18 @@
 const Album = require("../models/Album");
+const Song = require("../models/Song");
+
+const enhanceAlbumWithSongsData = async (album) => {
+  const songPromises = album.tracks.map((songId) =>
+    Song.findById(songId.toString()).select("-__v").select("-_id").lean()
+  );
+  const songs = await Promise.all(songPromises);
+
+  album.tracks = cleanUpSongsData(songs, album);
+  album.trackLength = calculateTotalDuration(album.tracks);
+  album.songCount = album.tracks.length;
+  return album;
+};
+
 const isAdmin = (usernameToCheck, passwordToCheck) => {
   const USER = [
     {
@@ -20,17 +34,6 @@ const isAdmin = (usernameToCheck, passwordToCheck) => {
   return false;
 };
 
-const getAlbumName = async (albumId) => {
-  try {
-    const album = await Album.findById(albumId).select("title").lean();
-
-    return album ? album.title : null;
-  } catch (error) {
-    console.error("Error fetching album name:", error.message);
-    return null;
-  }
-};
-
 const getAlbumDetails = async (albumId) => {
   try {
     const album = await Album.findById(albumId).select("-__v -_id").lean();
@@ -47,4 +50,47 @@ const getAlbumDetails = async (albumId) => {
     return null;
   }
 };
-module.exports = { isAdmin, getAlbumName, getAlbumDetails };
+
+const durationToSeconds = (duration) => {
+  const [minutes, seconds] = duration.split(" ").map((part) => {
+    if (part.includes("m")) {
+      return parseInt(part.replace("m", ""), 10) * 60;
+    } else {
+      return parseInt(part.replace("s", ""), 10);
+    }
+  });
+  return minutes + seconds;
+};
+const calculateTotalDuration = (tracks) => {
+  const totalSeconds = tracks.reduce((sum, track) => {
+    return sum + durationToSeconds(track.duration);
+  }, 0);
+
+  const hours = Math.floor(totalSeconds / 3600);
+  const minutes = Math.floor((totalSeconds % 3600) / 60);
+  const seconds = totalSeconds % 60;
+
+  if (hours > 0) {
+    return `${hours}h ${minutes}m ${seconds}s`;
+  } else {
+    return `${minutes}m ${seconds}s`;
+  }
+};
+
+// Helper function to clean up songs data
+const cleanUpSongsData = (songs, album) => {
+  return songs
+    .filter((song) => song !== null)
+    .map(({ albumId, ...songWithoutAlbumId }) => ({
+      ...songWithoutAlbumId,
+      album: album.title,
+    }));
+};
+
+module.exports = {
+  isAdmin,
+  getAlbumDetails,
+  calculateTotalDuration,
+  cleanUpSongsData,
+  enhanceAlbumWithSongsData,
+};