Merge pull request #15 from loverto:feat-upload

GitOrigin-RevId: 283741b1783a76867d93ea562cb8357d026217fa

src/main/java/org/sonatype/repository/helm/HelmUploadHandler.java

@@ -0,0 +1,125 @@
+/*
+ * Sonatype Nexus (TM) Open Source Version
+ * Copyright (c) 2018-present Sonatype, Inc.
+ * All rights reserved. Includes the third-party code listed at http://links.sonatype.com/products/nexus/oss/attributions.
+ *
+ * This program and the accompanying materials are made available under the terms of the Eclipse Public License Version 1.0,
+ * which accompanies this distribution and is available at http://www.eclipse.org/legal/epl-v10.html.
+ *
+ * Sonatype Nexus (TM) Professional Version is available from Sonatype, Inc. "Sonatype" and "Sonatype Nexus" are trademarks
+ * of Sonatype, Inc. Apache Maven is a trademark of the Apache Software Foundation. M2eclipse is a trademark of the
+ * Eclipse Foundation. All other trademarks are the property of their respective owners.
+ */
+package org.sonatype.repository.helm;
+
+import org.apache.commons.lang3.StringUtils;
+
+import org.sonatype.nexus.repository.Repository;
+import org.sonatype.nexus.repository.rest.UploadDefinitionExtension;
+import org.sonatype.nexus.repository.security.ContentPermissionChecker;
+import org.sonatype.nexus.repository.security.VariableResolverAdapter;
+import org.sonatype.nexus.repository.storage.StorageFacet;
+import org.sonatype.nexus.repository.storage.TempBlob;
+import org.sonatype.nexus.repository.transaction.TransactionalStoreBlob;
+import org.sonatype.nexus.repository.upload.ComponentUpload;
+import org.sonatype.nexus.repository.upload.UploadDefinition;
+import org.sonatype.nexus.repository.upload.UploadHandlerSupport;
+import org.sonatype.nexus.repository.upload.UploadResponse;
+import org.sonatype.nexus.repository.view.PartPayload;
+import org.sonatype.nexus.rest.ValidationErrorsException;
+import org.sonatype.repository.helm.internal.HelmFormat;
+import org.sonatype.repository.helm.internal.hosted.HelmHostedFacet;
+import org.sonatype.repository.helm.internal.metadata.HelmAttributes;
+import org.sonatype.repository.helm.internal.util.HelmAttributeParser;
+
+import javax.inject.Inject;
+import javax.inject.Named;
+import javax.inject.Singleton;
+
+import java.io.IOException;
+import java.util.Collections;
+import java.util.Set;
+
+import static org.sonatype.repository.helm.internal.util.HelmDataAccess.HASH_ALGORITHMS;
+
+/**
+ * Support helm upload for web page
+ *
+ * @author yinlongfei
+ */
+@Singleton
+@Named(HelmFormat.NAME)
+public class HelmUploadHandler
+    extends UploadHandlerSupport
+{
+  private UploadDefinition definition;
+
+  private final ContentPermissionChecker contentPermissionChecker;
+
+  private final VariableResolverAdapter variableResolverAdapter;
+
+  private final HelmAttributeParser helmPackageParser;
+
+
+  @Inject
+  public HelmUploadHandler(final ContentPermissionChecker contentPermissionChecker,
+                           final HelmAttributeParser helmPackageParser,
+                           @Named("simple") final VariableResolverAdapter variableResolverAdapter,
+                           final Set<UploadDefinitionExtension> uploadDefinitionExtensions)
+  {
+    super(uploadDefinitionExtensions);
+    this.contentPermissionChecker = contentPermissionChecker;
+    this.variableResolverAdapter = variableResolverAdapter;
+    this.helmPackageParser = helmPackageParser;
+  }
+
+
+  @Override
+  public UploadResponse handle(Repository repository, ComponentUpload upload) throws IOException {
+    HelmHostedFacet facet = repository.facet(HelmHostedFacet.class);
+    StorageFacet storageFacet = repository.facet(StorageFacet.class);
+
+    PartPayload payload = upload.getAssetUploads().get(0).getPayload();
+    try (TempBlob tempBlob = storageFacet.createTempBlob(payload, HASH_ALGORITHMS)) {
+      HelmAttributes attributesFromInputStream = helmPackageParser.getAttributesFromInputStream(tempBlob.get());
+
+      String name = attributesFromInputStream.getName();
+      String version = attributesFromInputStream.getVersion();
+
+      if (StringUtils.isBlank(name)) {
+        throw new ValidationErrorsException("Metadata is missing the name attribute");
+      }
+
+      if (StringUtils.isBlank(version)) {
+        throw new ValidationErrorsException("Metadata is missing the version attribute");
+      }
+
+      String extension = "tgz";
+
+      String path = name + "-" + version + "." + extension;
+
+      ensurePermitted(repository.getName(), HelmFormat.NAME, path, Collections.EMPTY_MAP);
+      return TransactionalStoreBlob.operation.withDb(storageFacet.txSupplier()).throwing(IOException.class)
+          .call(() -> new UploadResponse(facet.upload(path, tempBlob, payload)));
+    }
+  }
+
+
+  @Override
+  public UploadDefinition getDefinition() {
+    if (definition == null) {
+      definition = getDefinition(HelmFormat.NAME, false);
+    }
+    return definition;
+  }
+
+  @Override
+  public VariableResolverAdapter getVariableResolverAdapter() {
+    return variableResolverAdapter;
+  }
+
+  @Override
+  public ContentPermissionChecker contentPermissionChecker() {
+    return contentPermissionChecker;
+  }
+}

src/main/java/org/sonatype/repository/helm/internal/hosted/HelmHostedFacet.java

@@ -16,6 +16,8 @@
 
 import org.sonatype.nexus.repository.Facet;
 import org.sonatype.nexus.repository.Facet.Exposed;
+import org.sonatype.nexus.repository.storage.Asset;
+import org.sonatype.nexus.repository.storage.TempBlob;
 import org.sonatype.nexus.repository.view.Content;
 import org.sonatype.nexus.repository.view.Payload;
 import org.sonatype.repository.helm.internal.AssetKind;
@@ -33,5 +35,7 @@ public interface HelmHostedFacet
 
   void upload(String path, Payload payload, final AssetKind assetKind) throws IOException;
 
+  Asset upload(String path, TempBlob tempBlob,Payload payload) throws IOException;
+
   boolean delete(String path);
 }

src/main/java/org/sonatype/repository/helm/internal/hosted/HelmHostedFacetImpl.java

@@ -94,6 +94,8 @@ public Content get(final String path) {
     return helmDataAccess.toContent(asset, tx.requireBlob(asset.requireBlobRef()));
   }
 
+
+
   @Override
   public void upload(final String path,
                      final Payload payload,
@@ -111,6 +113,19 @@ public void upload(final String path,
     }
   }
 
+  @Override
+  public Asset upload(String path, TempBlob tempBlob,Payload payload) throws IOException {
+    checkNotNull(path);
+    checkNotNull(tempBlob);
+
+    StorageTx tx = UnitOfWork.currentTx();
+    Bucket bucket = tx.findBucket(getRepository());
+
+    Asset asset = createChartAsset(path, tx, bucket, tempBlob.get());
+    helmDataAccess.saveAsset(tx, asset, tempBlob, payload);
+    return asset;
+  }
+
   @Override
   @TransactionalDeleteBlob
   public boolean delete(final String path) {