jps/jps-builders/testSrc/org/jetbrains/jps/incremental/artifacts/LayoutElementTestUtil.java

/*
 * Copyright 2000-2012 JetBrains s.r.o.
 *
 * 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 org.jetbrains.jps.incremental.artifacts;

import com.intellij.util.PathUtil;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.jps.model.artifact.JpsArtifact;
import org.jetbrains.jps.model.artifact.elements.*;
import org.jetbrains.jps.model.java.JpsJavaExtensionService;
import org.jetbrains.jps.model.library.JpsLibrary;
import org.jetbrains.jps.model.module.JpsModule;

/**
 * @author nik
 */
public class LayoutElementTestUtil {
  public static LayoutElementCreator root() {
    return new LayoutElementCreator(JpsPackagingElementFactory.getInstance().createArtifactRoot(), null);
  }

  public static LayoutElementCreator archive(String name) {
    return new LayoutElementCreator(JpsPackagingElementFactory.getInstance().createArchive(name), null);
  }

  public static void addArtifactToLayout(JpsArtifact main, JpsArtifact included) {
    main.getRootElement().addChild(JpsPackagingElementFactory.getInstance().createArtifactOutput(included.createReference()));
  }

  public static class LayoutElementCreator {
    private JpsPackagingElementFactory myFactory;
    private JpsCompositePackagingElement myElement;
    private LayoutElementCreator myParent;
    
    public LayoutElementCreator(JpsCompositePackagingElement element, LayoutElementCreator parent) {
      myElement = element;
      myParent = parent;
      myFactory = JpsPackagingElementFactory.getInstance();
    }

    public LayoutElementCreator dir(String name) {
      JpsDirectoryPackagingElement dir = myFactory.createDirectory(name);
      myElement.addChild(dir);
      return new LayoutElementCreator(dir, this);
    }

    public LayoutElementCreator archive(String name) {
      JpsArchivePackagingElement archive = myFactory.createArchive(name);
      myElement.addChild(archive);
      return new LayoutElementCreator(archive, this);
    }

    public LayoutElementCreator fileCopy(String filePath) {
      return fileCopy(filePath, null);
    }

    public LayoutElementCreator fileCopy(String filePath, @Nullable String outputFileName) {
      return element(myFactory.createFileCopy(filePath, outputFileName));
    }

    public LayoutElementCreator dirCopy(String dirPath) {
      return element(myFactory.createDirectoryCopy(dirPath));
    }

    public LayoutElementCreator parentDirCopy(String filePath) {
      return dirCopy(PathUtil.getParentPath(filePath));
    }

    public LayoutElementCreator module(JpsModule module) {
      return element(JpsJavaExtensionService.getInstance().createProductionModuleOutput(module.createReference()));
    }

    public LayoutElementCreator element(JpsPackagingElement element) {
      myElement.addChild(element);
      return this;
    }

    public LayoutElementCreator lib(JpsLibrary library) {
      return element(myFactory.createLibraryElement(library.createReference()));
    }

    public LayoutElementCreator extractedDir(String jarPath, String pathInJar) {
      return element(myFactory.createExtractedDirectory(jarPath, pathInJar));
    }

    public LayoutElementCreator artifact(JpsArtifact included) {
      return element(myFactory.createArtifactOutput(included.createReference()));
    }

    public LayoutElementCreator end() {
      return myParent;
    }

    public JpsCompositePackagingElement buildElement() {
      if (myParent != null) {
        return myParent.buildElement();
      }
      return myElement;
    }
  }
}