MavenValueProviderFactory.java

/*******************************************************************************
 * Copyright (c) 2012-2016 Codenvy, S.A.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Codenvy, S.A. - initial API and implementation
 *******************************************************************************/
package org.eclipse.che.ide.extension.maven.server.projecttype;

import org.eclipse.che.api.core.ConflictException;
import org.eclipse.che.api.core.ForbiddenException;
import org.eclipse.che.api.core.ServerException;
import org.eclipse.che.api.project.server.FileEntry;
import org.eclipse.che.api.project.server.FolderEntry;
import org.eclipse.che.api.project.server.InvalidValueException;
import org.eclipse.che.api.project.server.ValueProvider;
import org.eclipse.che.api.project.server.ValueProviderFactory;
import org.eclipse.che.api.project.server.ValueStorageException;
import org.eclipse.che.api.project.server.VirtualFileEntry;
import org.eclipse.che.api.vfs.server.VirtualFile;
import org.eclipse.che.commons.annotation.Nullable;
import org.eclipse.che.commons.xml.XMLTreeException;
import org.eclipse.che.ide.maven.tools.Build;
import org.eclipse.che.ide.maven.tools.Model;
import org.eclipse.che.ide.maven.tools.Resource;

import java.io.IOException;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.ARTIFACT_ID;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.DEFAULT_RESOURCES_FOLDER;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.DEFAULT_SOURCE_FOLDER;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.DEFAULT_TEST_RESOURCES_FOLDER;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.DEFAULT_TEST_SOURCE_FOLDER;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.GROUP_ID;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.PACKAGING;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.PARENT_ARTIFACT_ID;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.PARENT_GROUP_ID;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.PARENT_VERSION;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.RESOURCE_FOLDER;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.SOURCE_FOLDER;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.TEST_SOURCE_FOLDER;
import static org.eclipse.che.ide.extension.maven.shared.MavenAttributes.VERSION;

/**
 * @author Evgen Vidolob
 */
public class MavenValueProviderFactory implements ValueProviderFactory {

    protected Model readModel(FolderEntry projectFolder) throws ValueStorageException, ServerException, ForbiddenException, IOException {
        FileEntry pomFile = (FileEntry)projectFolder.getChild("pom.xml");
        if (pomFile == null) {
            throw new ValueStorageException("pom.xml does not exist.");
        }
        return Model.readFrom(pomFile.getInputStream());
    }

    @Nullable
    protected VirtualFile getPom(FolderEntry projectFolder) {
        try {
            final VirtualFileEntry pomFile = projectFolder.getChild("pom.xml");
            if (pomFile != null) {
                return pomFile.getVirtualFile();
            }
            return null;
        } catch (ForbiddenException | ServerException e) {
            return null;
        }
    }

    protected void throwReadException(Exception e) throws ValueStorageException {
        throw new ValueStorageException("Can't read pom.xml : " + e.getMessage());
    }

    protected void throwWriteException(Exception e) throws ValueStorageException {
        throw new ValueStorageException("Can't write pom.xml : " + e.getMessage());
    }

    @Override
    public ValueProvider newInstance(FolderEntry projectFolder) {
        return new MavenValueProvider(projectFolder);
    }

    protected class MavenValueProvider implements ValueProvider {

        protected FolderEntry projectFolder;

        protected MavenValueProvider(FolderEntry projectFolder) {
            this.projectFolder = projectFolder;
        }

        @Override
        public List<String> getValues(String attributeName) throws ValueStorageException {
            try {
                String value = "";
                final Model model = readModel(projectFolder);
                if (attributeName.equals(ARTIFACT_ID)) {
                    value = model.getArtifactId();
                } else if (attributeName.equals(GROUP_ID)) {
                    value = model.getGroupId();
                } else if (attributeName.equals(PACKAGING)) {
                    final String packaging = model.getPackaging();
                    value = packaging == null ? "" : packaging;
                } else if (attributeName.equals(VERSION)) {
                    value = model.getVersion();
                } else if (attributeName.equals(PARENT_ARTIFACT_ID) && model.getParent() != null) {
                    value = model.getParent().getArtifactId();
                } else if (attributeName.equals(PARENT_GROUP_ID) && model.getParent() != null) {
                    value = model.getParent().getGroupId();
                } else if (attributeName.equals(PARENT_VERSION) && model.getParent() != null) {
                    value = model.getParent().getVersion();
                } else if (attributeName.equals(SOURCE_FOLDER)) {
                    Build build = model.getBuild();
                    if (build != null && build.getSourceDirectory() != null) {
                        value = build.getSourceDirectory();
                    } else {
                        value = DEFAULT_SOURCE_FOLDER;
                    }
                } else if (attributeName.equals(TEST_SOURCE_FOLDER)) {
                    Build build = model.getBuild();
                    if (build != null && build.getTestSourceDirectory() != null) {
                        value = build.getTestSourceDirectory();
                    } else {
                        value = DEFAULT_TEST_SOURCE_FOLDER;
                    }
                } else if (attributeName.equals(RESOURCE_FOLDER)) {
                    Build build = model.getBuild();
                    if (build != null && build.getResources() != null) {
                        return build.getResources().stream().map(Resource::getDirectory).collect(Collectors.toList());
                    } else {
                        return Arrays.asList(DEFAULT_RESOURCES_FOLDER, DEFAULT_TEST_RESOURCES_FOLDER);
                    }
                }

                return Collections.singletonList(value);
            } catch (ServerException | ForbiddenException | IOException e) {
                throwReadException(e);
            } catch (XMLTreeException e) {
                throw new ValueStorageException("Error parsing pom.xml : " + e.getMessage());
            }
            return null;
        }

        @Override
        public void setValues(String attributeName, List<String> value) throws ValueStorageException, InvalidValueException {
            try {
                VirtualFile pom = getPom(projectFolder);
                if (pom == null) {
                    Model model = Model.createModel();
                    model.setModelVersion("4.0.0");
                    pom = projectFolder.createFile("pom.xml", new byte[0]).getVirtualFile();
                    model.writeTo(pom);
                }

                switch (attributeName) {
                    case ARTIFACT_ID:
                        Model.readFrom(pom).setArtifactId(value.get(0)).writeTo(pom);
                        break;
                    case GROUP_ID:
                        Model.readFrom(pom).setGroupId(value.get(0)).writeTo(pom);
                        break;
                    case PACKAGING:
                        String packaging = value.get(0);
                        if (packaging.isEmpty()) {
                            packaging = null;
                        }
                        Model.readFrom(pom).setPackaging(packaging).writeTo(pom);
                        break;
                    case VERSION:
                        Model.readFrom(pom).setVersion(value.get(0)).writeTo(pom);
                        break;
                }
            } catch (ForbiddenException | ServerException | IOException | ConflictException e) {
                throwWriteException(e);
            } catch (XMLTreeException e) {
                throw new ValueStorageException("Error parsing pom.xml : " + e.getMessage());
            }
        }
    }
}