[ccsdk/cds.git] /

/*\r
 * Copyright © 2018 IBM Intellectual Property.\r
 * Modifications Copyright © 2018 IBM.\r
 *\r
 * Licensed under the Apache License, Version 2.0 (the "License");\r
 * you may not use this file except in compliance with the License.\r
 * You may obtain a copy of the License at\r
 *\r
 *     http://www.apache.org/licenses/LICENSE-2.0\r
 *\r
 * Unless required by applicable law or agreed to in writing, software\r
 * distributed under the License is distributed on an "AS IS" BASIS,\r
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
 * See the License for the specific language governing permissions and\r
 * limitations under the License.\r
 */\r
\r
package org.onap.ccsdk.apps.controllerblueprints.service;\r
\r
import com.att.eelf.configuration.EELFLogger;\r
import com.att.eelf.configuration.EELFManager;\r
import org.onap.ccsdk.apps.controllerblueprints.core.BluePrintException;\r
import org.onap.ccsdk.apps.controllerblueprints.core.BluePrintProcessorException;\r
import org.onap.ccsdk.apps.controllerblueprints.core.utils.BluePrintArchiveUtils;\r
import org.onap.ccsdk.apps.controllerblueprints.core.utils.BluePrintFileUtils;\r
import org.springframework.beans.factory.annotation.Value;\r
import org.springframework.http.codec.multipart.FilePart;\r
import org.springframework.stereotype.Service;\r
import org.springframework.util.FileSystemUtils;\r
import org.springframework.util.StringUtils;\r
import reactor.core.publisher.Mono;\r
\r
import java.io.File;\r
import java.io.IOException;\r
import java.nio.file.Files;\r
import java.nio.file.Path;\r
\r
/**\r
 * CbaFileManagementService.java Purpose: Provide Service processing CBA file management\r
 *\r
 * @author Steve Siani\r
 * @version 1.0\r
 */\r
@Service\r
public class CbaFileManagementService {\r
    private static EELFLogger log = EELFManager.getInstance().getLogger(CbaFileManagementService.class);\r
\r
    @Value("${controllerblueprints.loadCbaExtension}")\r
    private String cbaExtension;\r
\r
    private static final String CBA_FILE_NAME_PATTERN = "CBA_{0}_{1}";\r
\r
\r
    /**\r
     * cleanupSavedCBA: This method cleanup the Zip file and the unzip directory that was added.\r
     *\r
     * @param zipFileName zipFileName\r
     * @param cbaFileLocation cbaFileLocation\r
     * @return\r
     * @throws BluePrintException BluePrintException\r
     */\r
    public void cleanupSavedCBA(String zipFileName, Path cbaFileLocation) throws BluePrintException {\r
\r
        String fileNameWithoutExtension = BluePrintFileUtils.Companion.stripFileExtension(zipFileName);\r
\r
        //Delete the Zip file from the repository\r
        FileSystemUtils.deleteRecursively(BluePrintFileUtils.Companion.getBluePrintFile(zipFileName,cbaFileLocation));\r
\r
        //Delete the CBA directory from the repository\r
        FileSystemUtils.deleteRecursively(BluePrintFileUtils.Companion.getBluePrintFile(fileNameWithoutExtension,cbaFileLocation));\r
    }\r
\r
    /**\r
     * This is a saveCBAFile method\r
     * take a {@link FilePart}, transfer it to disk using a Flux of FilePart and return a {@link Mono} representing the CBA file name\r
     *\r
     * @param (filePart, targetDirectory) - the request part containing the file to be saved and the default directory where to save\r
     * @return a {@link Mono} String representing the result of the operation\r
     * @throws (BluePrintException, IOException) BluePrintException, IOException\r
     */\r
    public Mono<String> saveCBAFile(FilePart filePart, Path targetDirectory) throws BluePrintException, IOException {\r
\r
        // Normalize file name\r
        final String fileName = StringUtils.cleanPath(filePart.filename());\r
\r
        // Check if the file's extension is "CBA"\r
        if(!StringUtils.getFilenameExtension(fileName).equals(cbaExtension)) {\r
            throw new BluePrintException("Invalid file extension required " + cbaExtension);\r
        }\r
\r
        // Change file name to match a pattern\r
        String changedFileName = BluePrintFileUtils.Companion.getCBAGeneratedFileName(fileName, this.CBA_FILE_NAME_PATTERN);\r
\r
        // Copy file to the target location (Replacing existing file with the same name)\r
        Path targetLocation = targetDirectory.resolve(changedFileName);\r
\r
        // if a file with the same name already exists in a repository, delete and recreate it\r
        File file = new File(targetLocation.toString());\r
        if (file.exists())\r
            file.delete();\r
        file.createNewFile();\r
\r
        return filePart.transferTo(file).thenReturn(changedFileName);\r
    }\r
\r
    /**\r
     * Decompress the file into the cbaFileLocation parameter\r
     * @param zipFileName name of the zipped file\r
     * @param cbaFileLocation path in which the zipped file will get decompressed\r
     * @return String the path in which the file is decompressed\r
     * @throws BluePrintException Exception in the process\r
     */\r
    public String decompressCBAFile(final String zipFileName, Path cbaFileLocation) throws BluePrintException {\r
\r
        File file = BluePrintFileUtils.Companion.getBluePrintFile(zipFileName, cbaFileLocation);\r
        try {\r
            Path directoryPath = Files.createDirectories(cbaFileLocation.resolve(BluePrintFileUtils.Companion.stripFileExtension(zipFileName)));\r
            BluePrintArchiveUtils.Companion.deCompress(file, directoryPath.toString());\r
            return directoryPath.toString();\r
\r
        } catch (BluePrintProcessorException | IOException ex) {\r
            throw new BluePrintException(" Fail to decompress " + zipFileName, ex);\r
        }\r
\r
    }\r
\r
\r
}\r