From 0f4d6d3213ccd09425fce2107e64d8100ab4005a Mon Sep 17 00:00:00 2001
From: SrGooglo <srgooglo@ragestudio.net>
Date: Wed, 5 Jul 2023 19:06:02 +0000
Subject: [PATCH] added fixments

---
 .../fixments/00_fix_audio_metadata.js         | 148 ++++++++++++++++++
 .../fixments/00_sanitize_tracks_metadata.js   |  40 +++++
 2 files changed, 188 insertions(+)
 create mode 100644 packages/file_server/fixments/00_fix_audio_metadata.js
 create mode 100644 packages/music_server/fixments/00_sanitize_tracks_metadata.js

diff --git a/packages/file_server/fixments/00_fix_audio_metadata.js b/packages/file_server/fixments/00_fix_audio_metadata.js
new file mode 100644
index 00000000..4ffc91b6
--- /dev/null
+++ b/packages/file_server/fixments/00_fix_audio_metadata.js
@@ -0,0 +1,148 @@
+global.toBoolean = (value) => {
+    if (typeof value === "boolean") {
+        return value
+    }
+
+    if (typeof value === "string") {
+        return value.toLowerCase() === "true"
+    }
+
+    return false
+}
+
+import DbManager from "../src/shared-classes/DbManager"
+import StorageClient from "../src/shared-classes/StorageClient"
+
+import audioProcess from "../src/services/post-process/audio"
+
+import { Track } from "../../music_server/src/models"
+
+import axios from "axios"
+import fs from "fs"
+import path from "path"
+import pMap from "p-map"
+
+const tmpPath = path.resolve(__dirname, ".tmp")
+
+let index = 0
+let tracksLength = 0
+
+async function recalculateMetadata(track) {
+    console.log(`\n`)
+    console.time(`recalculation ${track._id}`)
+    console.log(`Recalculating metadata for ${track._id}`)
+
+    // parse url https://domain/buket/file and fetch buket and file
+    const regex = new RegExp(`^https://(.*?)/(.*)$`)
+
+    const match = regex.exec(track.source)
+
+    const objectName = match[2].split("/").slice(1).join("/")
+
+    const filePath = path.resolve(tmpPath, objectName)
+
+    try {
+        if (!track.metadata) {
+            console.time(`fetching ${track._id}`)
+            // storage obj on memory 
+            let file = await axios.get(track.source, {
+                responseType: "stream",
+            })
+
+            if (!fs.existsSync(path.dirname(filePath))) {
+                fs.mkdirSync(path.dirname(filePath), { recursive: true })
+            }
+
+            // write file to tmpPath
+            const fileWriter = fs.createWriteStream(filePath)
+
+            file.data.pipe(fileWriter)
+
+            await new Promise((resolve, reject) => {
+                fileWriter.on("finish", resolve)
+                fileWriter.on("error", reject)
+            })
+            console.timeEnd(`fetching ${track._id}`)
+
+            console.time(`processing ${track._id}`)
+            file = await audioProcess({
+                filepath: filePath
+            })
+            track.metadata = file.metadata
+
+            if (!track.publisher) {
+                track.publisher = {}
+            }
+            await track.save()
+            await fs.promises.unlink(filePath)
+
+            console.timeEnd(`processing ${track._id}`)
+
+            console.log(`Updated metadata for ${track._id}`, track.metadata)
+        } else {
+            console.log(`Metadata already exists for ${track._id}, skipping...`)
+        }
+    } catch (error) {
+        console.error(error)
+
+        if (fs.existsSync(filePath)) {
+            await fs.promises.unlink(filePath)
+        }
+    }
+
+    index++
+
+    console.timeEnd(`recalculation ${track._id}`)
+    console.log(`Metadata done [${index}/${tracksLength}]`)
+}
+
+async function main() {
+    console.time("fix audio metadata")
+
+    const db = new DbManager()
+    const storage = StorageClient()
+
+    await db.initialize()
+    await storage.initialize()
+
+    console.log(`Finding tracks...`)
+
+    console.time("finding tracks")
+
+    let tracks = await Track.find({
+        source: {
+            $exists: true,
+        },
+    })
+
+    tracksLength = tracks.length
+
+    console.timeEnd("finding tracks")
+
+    console.log(`Found ${tracks.length} tracks.`)
+
+    console.log(`Starting fix...\n`)
+
+    if (!fs.existsSync(tmpPath)) {
+        fs.mkdirSync(tmpPath)
+    }
+
+    await pMap(tracks, recalculateMetadata, {
+        concurrency: 3,
+    })
+
+    console.timeEnd("fix audio metadata")
+    console.log("Done!")
+
+    process.on("exit", () => {
+        console.log("Exiting...")
+        fs.promises.rmdir(tmpPath, { recursive: true })
+    })
+
+    process.exit(0)
+}
+
+main().catch((error) => {
+    console.error(error)
+    process.exit(1)
+})
\ No newline at end of file
diff --git a/packages/music_server/fixments/00_sanitize_tracks_metadata.js b/packages/music_server/fixments/00_sanitize_tracks_metadata.js
new file mode 100644
index 00000000..a9f65f8f
--- /dev/null
+++ b/packages/music_server/fixments/00_sanitize_tracks_metadata.js
@@ -0,0 +1,40 @@
+import DbManager from "../src/shared-classes/DbManager"
+import { Track } from "../src/models"
+
+async function main() {
+    const db = new DbManager()
+
+    await db.initialize()
+
+    console.log(`Finding tracks...`)
+
+    console.time("finding tracks")
+    const tracksWithMetadata = await Track.find({
+        metadata: {
+            $exists: true,
+        },
+    })
+    console.timeEnd("finding tracks")
+
+    console.log(`Found ${tracksWithMetadata.length} tracks with metadata.`)
+
+    console.time("removing metadata")
+
+    for (const track of tracksWithMetadata) {
+        console.time(`removing metadata for ${track._id}`)
+        await Track.findByIdAndUpdate(track._id, {
+            $unset: {
+                metadata: "",
+            },
+        })
+        console.timeEnd(`removing metadata for ${track._id}`)
+    }
+
+    console.timeEnd("removing metadata")
+
+    console.log("Done!")
+
+    process.exit(0)
+}
+
+main().catch(console.error)
\ No newline at end of file