"adding the support for hugging face package manager"

Author: nitinp19

api/src/main/java/org/jfrog/artifactory/client/model/repository/settings/HuggingFaceRepositorySettings.java

@@ -0,0 +1,7 @@
+package org.jfrog.artifactory.client.model.repository.settings;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+
+@JsonIgnoreProperties(ignoreUnknown = true)
+public interface HuggingFaceRepositorySettings extends RepositorySettings {
+}

services/src/main/java/org/jfrog/artifactory/client/impl/jackson/RepositorySettingsMixIn.java

@@ -45,7 +45,8 @@
     @JsonSubTypes.Type(value = GoRepositorySettingsImpl.class, name = "go"),
     @JsonSubTypes.Type(value = CargoRepositorySettingsImpl.class, name = "cargo"),
     @JsonSubTypes.Type(value = TerraformRepositorySettingsImpl.class, name = "terraform"),
-    @JsonSubTypes.Type(value = OciRepositorySettingsImpl.class, name = "oci")
+    @JsonSubTypes.Type(value = OciRepositorySettingsImpl.class, name = "oci"),
+    @JsonSubTypes.Type(value = HuggingFaceRepositorySettingsImpl.class, name = "HuggingFaceML")
 })
 
 public abstract class RepositorySettingsMixIn {

services/src/main/java/org/jfrog/artifactory/client/model/impl/FederatedRepositoryBuilderImpl.java

@@ -16,7 +16,7 @@
  */
 public class FederatedRepositoryBuilderImpl extends NonVirtualRepositoryBuilderBase<FederatedRepositoryBuilder, FederatedRepository> implements FederatedRepositoryBuilder {
     private static Set<PackageType> federatedRepositorySupportedTypes = new HashSet<>(Arrays.asList(
-            bower, cocoapods, cran, conda, debian, docker, gems, generic, gitlfs, gradle, ivy, maven, npm, nuget, opkg, pypi, sbt, vagrant, yum, rpm, composer, conan, chef, puppet, helm, helmoci, go, cargo, terraform, oci
+            bower, cocoapods, cran, conda, debian, docker, gems, generic, gitlfs, gradle, ivy, maven, npm, nuget, opkg, pypi, sbt, vagrant, yum, rpm, composer, conan, chef, puppet, helm, helmoci, go, cargo, terraform, oci, HuggingFaceML
     ));
 
     protected List<FederatedMember> members = new ArrayList<>();

services/src/main/java/org/jfrog/artifactory/client/model/impl/LocalRepositoryBuilderImpl.java

@@ -17,7 +17,7 @@
  */
 public class LocalRepositoryBuilderImpl extends NonVirtualRepositoryBuilderBase<LocalRepositoryBuilder, LocalRepository> implements LocalRepositoryBuilder {
     private static Set<PackageType> localRepositorySupportedTypes = new HashSet<PackageType>(Arrays.asList(
-            bower, cocoapods, cran, conda, debian, docker, gems, generic, gitlfs, gradle, ivy, maven, npm, nuget, opkg, pypi, sbt, vagrant, yum, rpm, composer, conan, chef, puppet, helm, helmoci, go, cargo, terraform, oci
+            bower, cocoapods, cran, conda, debian, docker, gems, generic, gitlfs, gradle, ivy, maven, npm, nuget, opkg, pypi, sbt, vagrant, yum, rpm, composer, conan, chef, puppet, helm, helmoci, go, cargo, terraform, oci, HuggingFaceML
     ));
 
     protected LocalRepositoryBuilderImpl() {

services/src/main/java/org/jfrog/artifactory/client/model/impl/PackageTypeImpl.java

@@ -35,7 +35,8 @@ public enum PackageTypeImpl implements PackageType {
     go,
     cargo,
     terraform,
-    oci;
+    oci,
+    HuggingFaceML;
 
     @Override
     public boolean isCustom() {

services/src/main/java/org/jfrog/artifactory/client/model/impl/RemoteRepositoryBuilderImpl.java

@@ -17,7 +17,7 @@
  */
 public class RemoteRepositoryBuilderImpl extends NonVirtualRepositoryBuilderBase<RemoteRepositoryBuilder, RemoteRepository> implements RemoteRepositoryBuilder {
     private static Set<PackageType> remoteRepositorySupportedTypes = new HashSet<PackageType>(Arrays.asList(
-            bower, cocoapods, cran, conda, debian, docker, gems, generic, gitlfs, gradle, ivy, maven, npm, nuget, opkg, p2, pypi, sbt, vcs, yum, rpm, composer, conan, chef, puppet, helm, helmoci, go, cargo, terraform, oci
+            bower, cocoapods, cran, conda, debian, docker, gems, generic, gitlfs, gradle, ivy, maven, npm, nuget, opkg, p2, pypi, sbt, vcs, yum, rpm, composer, conan, chef, puppet, helm, helmoci, go, cargo, terraform, oci, HuggingFaceML
     ));
 
     private String url;

services/src/main/java/org/jfrog/artifactory/client/model/impl/VirtualRepositoryBuilderImpl.java

@@ -15,7 +15,7 @@
  */
 public class VirtualRepositoryBuilderImpl extends RepositoryBuilderBase<VirtualRepositoryBuilder, VirtualRepository> implements VirtualRepositoryBuilder {
     private static Set<PackageType> virtualRepositorySupportedTypes = new HashSet<PackageType>(Arrays.asList(
-            bower, cran, conda, docker, debian, gems, generic, gitlfs, gradle, ivy, maven, npm, nuget, p2, pypi, sbt, yum, rpm, composer, conan, chef, puppet, helm, helmoci, go, terraform, oci
+            bower, cran, conda, docker, debian, gems, generic, gitlfs, gradle, ivy, maven, npm, nuget, p2, pypi, sbt, yum, rpm, composer, conan, chef, puppet, helm, helmoci, go, terraform, oci, HuggingFaceML
     ));
 
     private Collection<String> repositories = Collections.emptyList();

services/src/main/java/org/jfrog/artifactory/client/model/repository/settings/impl/HuggingFaceRepositorySettingsImpl.java

@@ -0,0 +1,30 @@
+package org.jfrog.artifactory.client.model.repository.settings.impl;
+
+import org.jfrog.artifactory.client.model.PackageType;
+import org.jfrog.artifactory.client.model.impl.PackageTypeImpl;
+import org.jfrog.artifactory.client.model.repository.settings.AbstractRepositorySettings;
+import org.jfrog.artifactory.client.model.repository.settings.HuggingFaceRepositorySettings;
+
+public class HuggingFaceRepositorySettingsImpl extends AbstractRepositorySettings implements HuggingFaceRepositorySettings {
+    public static String defaultLayout = "simple-default";
+
+    public HuggingFaceRepositorySettingsImpl() {
+        super(defaultLayout);
+    }
+
+    @Override
+    public PackageType getPackageType() {
+        return PackageTypeImpl.HuggingFaceML;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        return o instanceof HuggingFaceRepositorySettingsImpl;
+    }
+
+    @Override
+    public int hashCode() {
+        return super.hashCode();
+    }
+}

services/src/test/java/org/jfrog/artifactory/client/HuggingFacePackageTypeRepositoryTest.java

@@ -0,0 +1,88 @@
+package org.jfrog.artifactory.client;
+
+import org.jfrog.artifactory.client.model.Repository;
+import org.jfrog.artifactory.client.model.builder.RepositoryBuilders;
+import org.jfrog.artifactory.client.model.impl.PackageTypeImpl;
+import org.jfrog.artifactory.client.model.repository.settings.HuggingFaceRepositorySettings;
+import org.jfrog.artifactory.client.model.repository.settings.impl.HuggingFaceRepositorySettingsImpl;
+import org.testng.annotations.AfterMethod;
+import org.testng.annotations.BeforeMethod;
+import org.testng.annotations.Test;
+
+import static org.testng.Assert.assertEquals;
+import static org.testng.Assert.assertNotNull;
+
+public class HuggingFacePackageTypeRepositoryTest extends ArtifactoryTestsBase {
+
+    private final String localRepo = "huggingface-local-1";
+    private final String federatedRepo = "huggingface-federated-1";
+    private final String virtualRepo = "huggingface-virtual-1";
+
+    @BeforeMethod
+    @AfterMethod
+    private void cleanup() {
+        deleteRepoIfExists(localRepo);
+        deleteRepoIfExists(federatedRepo);
+        deleteRepoIfExists(virtualRepo);
+    }
+
+    @Test
+    public void testHuggingFaceLocalRepo() {
+        HuggingFaceRepositorySettings settings = new HuggingFaceRepositorySettingsImpl();
+        RepositoryBuilders repositoryBuilders = artifactory.repositories().builders();
+        Repository localRepository = repositoryBuilders.localRepositoryBuilder()
+                .key(localRepo)
+                .description("new local huggingface repo")
+                .notes("some notes")
+                .repositorySettings(settings)
+                .build();
+
+        // Use position 0 for create, as required by the interface
+        artifactory.repositories().create(0, localRepository);
+        Repository localRepoFromServer = artifactory.repository(localRepo).get();
+        assertNotNull(localRepoFromServer);
+        assertEquals(localRepoFromServer.getKey(), localRepo);
+        assertEquals(localRepoFromServer.getDescription(), "new local huggingface repo");
+        assertEquals(localRepoFromServer.getNotes(), "some notes");
+        assertEquals(localRepoFromServer.getRepoLayoutRef(), "simple-default");
+        assertEquals(localRepoFromServer.getRepositorySettings().getPackageType(), PackageTypeImpl.HuggingFaceML);
+    }
+
+    @Test
+    public void testHuggingFaceFederatedRepo() {
+        HuggingFaceRepositorySettings settings = new HuggingFaceRepositorySettingsImpl();
+        RepositoryBuilders repositoryBuilders = artifactory.repositories().builders();
+        Repository federatedRepository = repositoryBuilders.federatedRepositoryBuilder()
+                .key(federatedRepo)
+                .description("new federated huggingface repo")
+                .notes("some notes")
+                .repositorySettings(settings)
+                .build();
+
+        artifactory.repositories().create(0, federatedRepository);
+        Repository federatedRepoFromServer = artifactory.repository(federatedRepo).get();
+        assertNotNull(federatedRepoFromServer);
+        assertEquals(federatedRepoFromServer.getKey(), federatedRepo);
+        assertEquals(federatedRepoFromServer.getDescription(), "new federated huggingface repo");
+        assertEquals(federatedRepoFromServer.getNotes(), "some notes");
+    }
+
+    @Test
+    public void testHuggingFaceVirtualRepo() {
+        HuggingFaceRepositorySettings settings = new HuggingFaceRepositorySettingsImpl();
+        RepositoryBuilders repositoryBuilders = artifactory.repositories().builders();
+        Repository virtualRepository = repositoryBuilders.virtualRepositoryBuilder()
+                .key(virtualRepo)
+                .description("new virtual huggingface repo")
+                .notes("some notes")
+                .repositorySettings(settings)
+                .build();
+
+        artifactory.repositories().create(0, virtualRepository);
+        Repository virtualRepoFromServer = artifactory.repository(virtualRepo).get();
+        assertNotNull(virtualRepoFromServer);
+        assertEquals(virtualRepoFromServer.getKey(), virtualRepo);
+        assertEquals(virtualRepoFromServer.getDescription(), "new virtual huggingface repo");
+        assertEquals(virtualRepoFromServer.getNotes(), "some notes");
+    }
+}

services/src/test/java/org/jfrog/artifactory/client/HuggingFaceRepositorySettingsTest.java

@@ -0,0 +1,45 @@
+package org.jfrog.artifactory.client;
+
+import org.jfrog.artifactory.client.model.PackageType;
+import org.jfrog.artifactory.client.model.impl.PackageTypeImpl;
+import org.jfrog.artifactory.client.model.repository.settings.impl.HuggingFaceRepositorySettingsImpl;
+import org.testng.annotations.Test;
+
+import static org.testng.Assert.*;
+
+/**
+ * Test for HuggingFace repository settings implementation
+ */
+public class HuggingFaceRepositorySettingsTest {
+
+    @Test
+    public void testHuggingFaceRepositorySettings() {
+        HuggingFaceRepositorySettingsImpl settings = new HuggingFaceRepositorySettingsImpl();
+
+        // Test package type
+        PackageType packageType = settings.getPackageType();
+        assertEquals(packageType, PackageTypeImpl.HuggingFaceML);
+
+        // Test default layout
+        assertEquals(settings.getRepoLayout(), "simple-default");
+
+        // Test layout setting
+        settings.setRepoLayout("custom-layout");
+        assertEquals(settings.getRepoLayout(), "custom-layout");
+
+        // Test that it's not a custom package type
+        assertFalse(packageType.isCustom());
+    }
+
+    @Test
+    public void testConsistencyWithOtherPackageManagers() {
+        HuggingFaceRepositorySettingsImpl huggingFaceSettings = new HuggingFaceRepositorySettingsImpl();
+
+        // Verify it follows the same interface pattern as other package managers
+        assertNotNull(huggingFaceSettings.getPackageType());
+        assertNotNull(huggingFaceSettings.getRepoLayout());
+
+        // Test that the default layout is properly set
+        assertEquals(HuggingFaceRepositorySettingsImpl.defaultLayout, "simple-default");
+    }
+}