Add a retrieveLicenseDocument method to LcpService (readium#519)

marshal1919 · May 21, 2024 · ac6392c · ac6392c
1 parent a91e76f
commit ac6392c
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/readium/lcp/src/main/java/org/readium/r2/lcp/LcpService.kt b/readium/lcp/src/main/java/org/readium/r2/lcp/LcpService.kt
@@ -28,6 +28,7 @@ import org.readium.r2.shared.publication.protection.ContentProtection
 import org.readium.r2.shared.util.Try
 import org.readium.r2.shared.util.asset.Asset
 import org.readium.r2.shared.util.asset.AssetRetriever
+import org.readium.r2.shared.util.asset.ContainerAsset
 import org.readium.r2.shared.util.format.Format
 
 /**
@@ -118,6 +119,13 @@ public interface LcpService {
         allowUserInteraction: Boolean
     ): Try<LcpLicense, LcpError>
 
+    /**
+     * Retrieves the license document from a LCP-protected publication asset.
+     */
+    public suspend fun retrieveLicenseDocument(
+        asset: ContainerAsset
+    ): Try<LicenseDocument, LcpError>
+
     /**
      * Injects a [licenseDocument] into the given [publicationFile] package.
      *

diff --git a/readium/lcp/src/main/java/org/readium/r2/lcp/service/LicensesService.kt b/readium/lcp/src/main/java/org/readium/r2/lcp/service/LicensesService.kt
@@ -42,6 +42,7 @@ import org.readium.r2.shared.util.FileExtension
 import org.readium.r2.shared.util.Try
 import org.readium.r2.shared.util.asset.Asset
 import org.readium.r2.shared.util.asset.AssetRetriever
+import org.readium.r2.shared.util.asset.ContainerAsset
 import org.readium.r2.shared.util.format.Format
 import org.readium.r2.shared.util.format.FormatHints
 import org.readium.r2.shared.util.format.FormatSpecification
@@ -243,6 +244,21 @@ internal class LicensesService(
             Try.failure(LcpError.wrap(e))
         }
 
+    override suspend fun retrieveLicenseDocument(
+        asset: ContainerAsset
+    ): Try<LicenseDocument, LcpError> =
+        withContext(Dispatchers.IO) {
+            try {
+                val licenseContainer = createLicenseContainer(context, asset)
+                val licenseData = licenseContainer.read()
+                Try.success(LicenseDocument(licenseData))
+            } catch (e: CancellationException) {
+                throw e
+            } catch (e: Exception) {
+                Try.failure(LcpError.wrap(e))
+            }
+        }
+
     private suspend fun retrieveLicense(
         container: LicenseContainer,
         authentication: LcpAuthenticating,