org.eclipse.dirigible.tests.util.ProjectUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dirigible-tests-framework Show documentation
There is a newer version: 11.11.0
Show newest version
/*
 * Copyright (c) 2010-2024 Eclipse Dirigible contributors
 *
 * All rights reserved. This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v2.0 which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v20.html
 *
 * SPDX-FileCopyrightText: Eclipse Dirigible contributors SPDX-License-Identifier: EPL-2.0
 */
package org.eclipse.dirigible.tests.util;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.filefilter.TrueFileFilter;
import org.apache.commons.lang3.StringUtils;
import org.eclipse.dirigible.commons.config.DirigibleConfig;
import org.eclipse.dirigible.repository.api.IRepository;
import org.eclipse.dirigible.repository.api.IRepositoryStructure;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.util.Collection;
import java.util.Collections;
import java.util.Map;

@Component
public class ProjectUtil {

    private static final Logger LOGGER = LoggerFactory.getLogger(ProjectUtil.class);

    private final IRepository repository;

    ProjectUtil(IRepository repository) {
        this.repository = repository;
    }

    public void copyResourceProjectToDefaultUserWorkspace(String resourcesFolder) {
        copyResourceProjectToDefaultUserWorkspace(resourcesFolder, Collections.emptyMap());
    }

    public void copyResourceProjectToDefaultUserWorkspace(String resourcesFolder, Map placeholders) {
        copyResourceProjectToUserWorkspace(getDefaultUser(), resourcesFolder, placeholders);
    }

    public void copyResourceProjectToUserWorkspace(String user, String resourcesFolder, Map placeholders) {
        String targetProjectName = extractFolderName(resourcesFolder);

        copyResourceFolderContentToUserWorkspaceProject(user, resourcesFolder, targetProjectName, placeholders);
    }

    private String extractFolderName(String resourcesFolder) {
        boolean rootDir = resourcesFolder.lastIndexOf("/") == -1;
        return rootDir ? resourcesFolder : (resourcesFolder.substring(resourcesFolder.lastIndexOf("/") + 1));
    }

    public void copyResourceFolderContentToUserWorkspaceProject(String user, String resourcesFolder, String targetProjectName,
            Map placeholders) {
        String destinationDirPath = createUserWorkspaceFolderPath(user, targetProjectName);

        copyResourceFolder(resourcesFolder, destinationDirPath, placeholders);
    }

    private void copyResourceFolder(String resourcesFolder, String destinationDirPath, Map placeholders) {
        String sourceDirPath = getResourcePath(resourcesFolder);
        File sourceDir = new File(sourceDirPath);

        File destinationDir = new File(destinationDirPath);

        LOGGER.debug("Copy folder [{}] to folder [{}]", sourceDir, destinationDir);
        try {
            FileUtils.copyDirectory(sourceDir, destinationDir);
        } catch (IOException ex) {
            throw new IllegalStateException("Failed to copy [" + sourceDir + "] to " + destinationDir, ex);
        }

        Collection files = FileUtils.listFiles(destinationDir, TrueFileFilter.INSTANCE, TrueFileFilter.INSTANCE);
        for (File file : files) {
            replacePlaceholderInFile(file, placeholders);
        }
    }

    private String getResourcePath(String path) {
        URL resourceURL = ProjectUtil.class.getClassLoader()
                                           .getResource(path);
        if (null == resourceURL) {
            throw new IllegalStateException("Missing resource with path " + path);
        }
        return resourceURL.getPath();
    }

    private void replacePlaceholderInFile(File file, Map placeholders) {
        try {
            String content = FileUtils.readFileToString(file, StandardCharsets.UTF_8);

            for (Map.Entry entry : placeholders.entrySet()) {
                String placeholder = entry.getKey();
                String replacement = entry.getValue();

                LOGGER.debug("Replacing [{}] with [{}] in file [{}]", placeholder, replacement, file);
                content = StringUtils.replace(content, placeholder, replacement);
            }

            FileUtils.writeStringToFile(file, content, StandardCharsets.UTF_8);
        } catch (IOException ex) {
            throw new IllegalStateException("Failed to replace placeholders [" + placeholders + "] in file " + file, ex);
        }
    }

    private String createUserWorkspaceFolderPath(String user, String projectName) {
        return repository.getRepositoryPath() + File.separator + IRepositoryStructure.KEYWORD_USERS + File.separator + user + File.separator
                + IRepositoryStructure.KEYWORD_WORKSPACE + File.separator + projectName;
    }

    private String getDefaultUser() {
        return DirigibleConfig.BASIC_ADMIN_USERNAME.getFromBase64Value();
    }

}