com.netflix.spinnaker.halyard.core.registry.v1.GitProfileReader.java Source code

Introduction

Here is the source code for com.netflix.spinnaker.halyard.core.registry.v1.GitProfileReader.java

Source

/*
 * Copyright 2017 Google, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License")
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 *
 */

package com.netflix.spinnaker.halyard.core.registry.v1;

import com.netflix.spinnaker.halyard.core.FileModeUtils;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.FileVisitOption;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.stream.Collectors;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.compress.archivers.tar.TarArchiveEntry;
import org.apache.commons.compress.archivers.tar.TarArchiveOutputStream;
import org.apache.commons.io.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

@Component
@Slf4j
public class GitProfileReader implements ProfileReader {
    @Autowired
    String gitRoot;

    private final static String HALCONFIG_DIR = "halconfig";

    @Override
    public InputStream readProfile(String artifactName, String version, String profileName) throws IOException {
        return getContents(profilePath(artifactName, version, profileName));
    }

    @Override
    public BillOfMaterials readBom(String version) throws IOException {
        if (!Versions.isBranch(version)) {
            throw new IllegalArgumentException("Version must be a branch in the git profile reader");
        }
        String branch = Versions.fromBranch(version);

        BillOfMaterials.Artifact artifact = new BillOfMaterials.Artifact();
        artifact.setCommit(branch);
        artifact.setVersion(version);

        BillOfMaterials.Services services = new BillOfMaterials.Services();
        services.setDefaultArtifact(artifact);

        BillOfMaterials.Dependencies dependencies = new BillOfMaterials.Dependencies();
        dependencies.setDefaultArtifact(artifact);

        BillOfMaterials bom = new BillOfMaterials();
        bom.setDependencies(dependencies);
        bom.setVersion(version);
        bom.setServices(services);

        return bom;
    }

    @Override
    public Versions readVersions() throws IOException {
        throw new UnsupportedOperationException();
    }

    @Override
    public InputStream readArchiveProfile(String artifactName, String version, String profileName)
            throws IOException {
        Path profilePath = Paths.get(profilePath(artifactName, version, profileName));

        ByteArrayOutputStream os = new ByteArrayOutputStream();
        TarArchiveOutputStream tarArchive = new TarArchiveOutputStream(os);

        ArrayList<Path> filePathsToAdd = java.nio.file.Files
                .walk(profilePath, Integer.MAX_VALUE, FileVisitOption.FOLLOW_LINKS)
                .filter(path -> path.toFile().isFile()).collect(Collectors.toCollection(ArrayList::new));

        for (Path path : filePathsToAdd) {
            TarArchiveEntry tarEntry = new TarArchiveEntry(path.toFile(), profilePath.relativize(path).toString());
            int permissions = FileModeUtils.getFileMode(Files.getPosixFilePermissions(path));
            permissions = FileModeUtils.setFileBit(permissions);
            tarEntry.setMode(permissions);
            tarArchive.putArchiveEntry(tarEntry);
            IOUtils.copy(Files.newInputStream(path), tarArchive);
            tarArchive.closeArchiveEntry();
        }

        tarArchive.finish();
        tarArchive.close();

        return new ByteArrayInputStream(os.toByteArray());
    }

    private String profilePath(String artifactName, String version, String profileFileName) {
        return Paths.get(gitRoot, artifactName, HALCONFIG_DIR, profileFileName).toString();
    }

    private InputStream getContents(String objectName) throws IOException {
        log.info("Getting file contents of " + objectName);
        return new FileInputStream(objectName);
    }
}