From 22611acdaab0ac38e46dd8ef185a0c56831c63b8 Mon Sep 17 00:00:00 2001
From: Cristhian Escobar <escobarcristhian@utexas.edu>
Date: Wed, 25 Oct 2023 11:18:19 -0500
Subject: [PATCH] Parse Music Metadata V2

---
 .../repositories/CardanoWalletRepository.kt   | 77 +++++++++++++++++--
 .../usecases/WalletNFTSongsUseCase.kt         |  5 +-
 2 files changed, 74 insertions(+), 8 deletions(-)

diff --git a/shared/src/commonMain/kotlin/io.newm.shared/repositories/CardanoWalletRepository.kt b/shared/src/commonMain/kotlin/io.newm.shared/repositories/CardanoWalletRepository.kt
index 8d5d226a..27e48716 100644
--- a/shared/src/commonMain/kotlin/io.newm.shared/repositories/CardanoWalletRepository.kt
+++ b/shared/src/commonMain/kotlin/io.newm.shared/repositories/CardanoWalletRepository.kt
@@ -7,7 +7,7 @@ import org.koin.core.component.KoinComponent
 import org.koin.core.component.inject
 
 internal interface CardanoWalletRepository {
-    suspend fun getWalletNFTs(xpub: String): List<LedgerAssetMetadata>
+    suspend fun getWalletNFTs(xpub: String): List<NFTTrack>
 }
 
 internal class CardanoWalletRepositoryImpl : KoinComponent, CardanoWalletRepository {
@@ -15,11 +15,76 @@ internal class CardanoWalletRepositoryImpl : KoinComponent, CardanoWalletReposit
     private val service: CardanoWalletAPI by inject()
     private val logger = Logger.withTag("NewmKMM-CardanoWalletRepository")
 
-    override suspend fun getWalletNFTs(xpub: String): List<LedgerAssetMetadata> {
-        logger.d { "getWalletNFTs() w/ xpub key: $xpub" }
-        val nfts = service.getWalletNFTs(xpub)
-        logger.d { "Result : ${nfts.first()}" }
-        return nfts.first()
+    override suspend fun getWalletNFTs(xpub: String): List<NFTTrack> {
+        val walletNFTs = service.getWalletNFTs(xpub)
+        val tracks = walletNFTs.mapNotNull {
+            when (it.getMusicMetadataVersion()) {
+                2 -> it.getTrackFromMusicMetadataV2()
+                else -> {
+                    logger.d { "Unsupported music metadata version: ${it.getMusicMetadataVersion()}" }
+                    null
+                }
+            }
+        }
+        logger.d { "Result Size: ${tracks.size}" }
+        return tracks
     }
 }
 
+fun List<LedgerAssetMetadata>.getMusicMetadataVersion(): Int? {
+    return this.find { it.key == "music_metadata_version" }?.value?.toInt() ?: return null
+}
+
+data class NFTTrack(
+    val name: String,
+    val image: String,
+    val imageMimeType: String,
+    val files: List<File>,
+    val musicMetadataVersion: Int,
+    val releaseType: String?,
+    val releaseTitle: String?,
+    val distributor: String?,
+    val publicationDate: String?
+)
+
+data class File(
+    val name: String,
+    val mediaType: String,
+    val src: String
+)
+
+fun List<LedgerAssetMetadata>.getTrackFromMusicMetadataV2(): NFTTrack? {
+    val name = this.find { it.key == "name" }?.value ?: return null
+    val image = this.find { it.key == "image" }?.value ?: return null
+    val imageMimeType = this.find { it.key == "mediaType" }?.value ?: return null
+    val musicMetadataVersion =
+        this.find { it.key == "music_metadata_version" }?.value?.toInt() ?: return null
+
+    val fileList = this.find { it.key == "files" }?.children?.mapNotNull {
+        val fileName =
+            it.children.find { child -> child.key == "name" }?.value ?: return@mapNotNull null
+        val fileMediaType =
+            it.children.find { child -> child.key == "mediaType" }?.value ?: return@mapNotNull null
+        val fileSrc =
+            it.children.find { child -> child.key == "src" }?.value ?: return@mapNotNull null
+        File(fileName, fileMediaType, fileSrc)
+    }.orEmpty()
+
+    val release = this.find { it.key == "release" }?.children
+    val releaseType = release?.find { it.key == "release_type" }?.value
+    val releaseTitle = release?.find { it.key == "release_title" }?.value
+    val distributor = release?.find { it.key == "distributor" }?.value
+    val publicationDate = release?.find { it.key == "publication_date" }?.value
+
+    return NFTTrack(
+        name,
+        image,
+        imageMimeType,
+        fileList,
+        musicMetadataVersion,
+        releaseType,
+        releaseTitle,
+        distributor,
+        publicationDate
+    )
+}
diff --git a/shared/src/commonMain/kotlin/io.newm.shared/usecases/WalletNFTSongsUseCase.kt b/shared/src/commonMain/kotlin/io.newm.shared/usecases/WalletNFTSongsUseCase.kt
index 4bf86c46..6e6090f4 100644
--- a/shared/src/commonMain/kotlin/io.newm.shared/usecases/WalletNFTSongsUseCase.kt
+++ b/shared/src/commonMain/kotlin/io.newm.shared/usecases/WalletNFTSongsUseCase.kt
@@ -4,6 +4,7 @@ import co.touchlab.kermit.Logger
 import io.newm.shared.login.repository.LogInRepository
 import io.newm.shared.models.Song
 import io.newm.shared.repositories.CardanoWalletRepository
+import io.newm.shared.repositories.NFTTrack
 import io.newm.shared.repositories.testdata.MockSongs
 import io.newm.shared.services.LedgerAssetMetadata
 import io.newm.shared.services.UserAPI
@@ -13,13 +14,13 @@ import org.koin.core.component.inject
 interface WalletNFTSongsUseCase {
     fun getAllWalletNFTSongs(xPub: String): Flow<List<Song>>
 
-    suspend fun getWalletNFTs(xpub: String): List<LedgerAssetMetadata>
+    suspend fun getWalletNFTs(xpub: String): List<NFTTrack>
 }
 
 internal class WalletNFTSongsUseCaseImpl(private val repository: CardanoWalletRepository) : WalletNFTSongsUseCase {
 
     override fun getAllWalletNFTSongs(xPub: String): Flow<List<Song>> = MockSongs.getSongsFlow()
-    override suspend fun getWalletNFTs(xpub: String): List<LedgerAssetMetadata> {
+    override suspend fun getWalletNFTs(xpub: String): List<NFTTrack> {
         return repository.getWalletNFTs(xpub)
     }
 }