From 91a4ae0364ef5d1c61c55fadbdf9190bc4a7c430 Mon Sep 17 00:00:00 2001
From: Luiz Ribeiro <luizribeiro@gmail.com>
Date: Sun, 22 Jan 2023 22:30:09 -0500
Subject: [PATCH] Add support for OCI Image Index

Fixes #243
---
 src/mediatypes.rs      | 5 +++++
 src/v2/config.rs       | 1 +
 src/v2/manifest/mod.rs | 2 +-
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/mediatypes.rs b/src/mediatypes.rs
index fb09c4d0..edc5aa03 100644
--- a/src/mediatypes.rs
+++ b/src/mediatypes.rs
@@ -32,6 +32,10 @@ pub enum MediaTypes {
     #[strum(serialize = "application/vnd.docker.container.image.v1+json")]
     #[strum(props(Sub = "vnd.docker.container.image.v1+json"))]
     ContainerConfigV1,
+    /// OCI iamge index
+    #[strum(serialize = "application/vnd.oci.image.index.v1+json")]
+    #[strum(props(Sub = "vnd.oci.image.index.v1+json"))]
+    OCIImageIndexV1,
     /// Generic JSON
     #[strum(serialize = "application/json")]
     #[strum(props(Sub = "json"))]
@@ -55,6 +59,7 @@ impl MediaTypes {
                     }
                     ("vnd.docker.image.rootfs.diff.tar.gzip", _) => Ok(MediaTypes::ImageLayerTgz),
                     ("vnd.docker.container.image.v1", "json") => Ok(MediaTypes::ContainerConfigV1),
+                    ("vnd.oci.image.index.v1", "json") => Ok(MediaTypes::OCIImageIndexV1),
                     _ => Err(crate::Error::UnknownMimeType(mtype.clone())),
                 }
             }
diff --git a/src/v2/config.rs b/src/v2/config.rs
index 19e709cd..7fbd0661 100644
--- a/src/v2/config.rs
+++ b/src/v2/config.rs
@@ -113,6 +113,7 @@ impl Config {
                     (MediaTypes::ManifestV2S2, Some(0.5)),
                     (MediaTypes::ManifestV2S1Signed, Some(0.4)),
                     (MediaTypes::ManifestList, Some(0.5)),
+                    (MediaTypes::OCIImageIndexV1, Some(0.5)),
                 ],
                 // GCR incorrectly parses `q` parameters, so we use special Accept for it.
                 // Bug: https://issuetracker.google.com/issues/159827510.
diff --git a/src/v2/manifest/mod.rs b/src/v2/manifest/mod.rs
index 2808403f..e6b89da5 100644
--- a/src/v2/manifest/mod.rs
+++ b/src/v2/manifest/mod.rs
@@ -85,7 +85,7 @@ impl Client {
                     content_digest,
                 ))
             }
-            mediatypes::MediaTypes::ManifestList => Ok((
+            mediatypes::MediaTypes::ManifestList | mediatypes::MediaTypes::OCIImageIndexV1 => Ok((
                 res.json::<ManifestList>().await.map(Manifest::ML)?,
                 content_digest,
             )),