1 files changed, 136 insertions, 0 deletions
diff --git a/plugins/gradle/jps-plugin/src/org/jetbrains/jps/gradle/model/impl/JpsGradleExtensionServiceImpl.java b/plugins/gradle/jps-plugin/src/org/jetbrains/jps/gradle/model/impl/JpsGradleExtensionServiceImpl.java
new file mode 100644
index 000000000000..81ed9076f13e
--- /dev/null
+++ b/plugins/gradle/jps-plugin/src/org/jetbrains/jps/gradle/model/impl/JpsGradleExtensionServiceImpl.java
@@ -0,0 +1,136 @@
+/*
+ * Copyright 2000-2014 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.gradle.model.impl;
+
+import com.intellij.openapi.diagnostic.Logger;
+import com.intellij.openapi.util.JDOMUtil;
+import com.intellij.openapi.util.io.FileUtil;
+import com.intellij.util.containers.ConcurrentFactoryMap;
+import com.intellij.util.containers.FactoryMap;
+import com.intellij.util.xmlb.XmlSerializer;
+import gnu.trove.THashMap;
+import org.jdom.Document;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+import org.jetbrains.jps.builders.storage.BuildDataPaths;
+import org.jetbrains.jps.gradle.model.JpsGradleExtensionService;
+import org.jetbrains.jps.gradle.model.JpsGradleModuleExtension;
+import org.jetbrains.jps.incremental.resources.ResourcesBuilder;
+import org.jetbrains.jps.incremental.resources.StandardResourceBuilderEnabler;
+import org.jetbrains.jps.model.JpsElementChildRole;
+import org.jetbrains.jps.model.JpsElementFactory;
+import org.jetbrains.jps.model.JpsSimpleElement;
+import org.jetbrains.jps.model.ex.JpsElementChildRoleBase;
+import org.jetbrains.jps.model.module.JpsDependencyElement;
+import org.jetbrains.jps.model.module.JpsModule;
+
+import java.io.File;
+import java.util.Map;
+
+/**
+ * @author Vladislav.Soroka
+ * @since 7/10/2014
+ */
+public class JpsGradleExtensionServiceImpl extends JpsGradleExtensionService {
+  private static final Logger LOG = Logger.getInstance(JpsGradleExtensionServiceImpl.class);
+  private static final JpsElementChildRole<JpsSimpleElement<Boolean>> PRODUCTION_ON_TEST_ROLE = JpsElementChildRoleBase.create("production on test");
+  private final Map<File, GradleProjectConfiguration> myLoadedConfigs =
+    new THashMap<File, GradleProjectConfiguration>(FileUtil.FILE_HASHING_STRATEGY);
+  private final FactoryMap<File, Boolean> myConfigFileExists = new ConcurrentFactoryMap<File, Boolean>() {
+    @Nullable
+    @Override
+    protected Boolean create(File key) {
+      return key.exists();
+    }
+  };
+
+  public JpsGradleExtensionServiceImpl() {
+    ResourcesBuilder.registerEnabler(new StandardResourceBuilderEnabler() {
+      @Override
+      public boolean isResourceProcessingEnabled(JpsModule module) {
+        // enable standard resource processing only if this is not a gradle module
+        // for gradle modules use gradle-aware resource builder
+        return getExtension(module) == null;
+      }
+    });
+  }
+
+  @Nullable
+  @Override
+  public JpsGradleModuleExtension getExtension(@NotNull JpsModule module) {
+    return module.getContainer().getChild(JpsGradleModuleExtensionImpl.ROLE);
+  }
+
+  @NotNull
+  @Override
+  public JpsGradleModuleExtension getOrCreateExtension(@NotNull JpsModule module) {
+    JpsGradleModuleExtension extension = module.getContainer().getChild(JpsGradleModuleExtensionImpl.ROLE);
+    if (extension == null) {
+      extension = new JpsGradleModuleExtensionImpl();
+      module.getContainer().setChild(JpsGradleModuleExtensionImpl.ROLE, extension);
+    }
+    return extension;
+  }
+
+  @Override
+  public void setProductionOnTestDependency(@NotNull JpsDependencyElement dependency, boolean value) {
+    if (value) {
+      dependency.getContainer().setChild(PRODUCTION_ON_TEST_ROLE, JpsElementFactory.getInstance().createSimpleElement(true));
+    }
+    else {
+      dependency.getContainer().removeChild(PRODUCTION_ON_TEST_ROLE);
+    }
+  }
+
+  @Override
+  public boolean isProductionOnTestDependency(@NotNull JpsDependencyElement dependency) {
+    JpsSimpleElement<Boolean> child = dependency.getContainer().getChild(PRODUCTION_ON_TEST_ROLE);
+    return child != null && child.getData();
+  }
+
+  @Override
+  public boolean hasGradleProjectConfiguration(@NotNull BuildDataPaths paths) {
+    return myConfigFileExists.get(new File(paths.getDataStorageRoot(), GradleProjectConfiguration.CONFIGURATION_FILE_RELATIVE_PATH));
+  }
+
+  @NotNull
+  @Override
+  public GradleProjectConfiguration getGradleProjectConfiguration(BuildDataPaths paths) {
+    final File dataStorageRoot = paths.getDataStorageRoot();
+    return getGradleProjectConfiguration(dataStorageRoot);
+  }
+
+  @NotNull
+  public GradleProjectConfiguration getGradleProjectConfiguration(@NotNull File dataStorageRoot) {
+    final File configFile = new File(dataStorageRoot, GradleProjectConfiguration.CONFIGURATION_FILE_RELATIVE_PATH);
+    GradleProjectConfiguration config;
+    synchronized (myLoadedConfigs) {
+      config = myLoadedConfigs.get(configFile);
+      if (config == null) {
+        config = new GradleProjectConfiguration();
+        try {
+          final Document document = JDOMUtil.loadDocument(configFile);
+          XmlSerializer.deserializeInto(config, document.getRootElement());
+        }
+        catch (Exception e) {
+          LOG.info(e);
+        }
+        myLoadedConfigs.put(configFile, config);
+      }
+    }
+    return config;
+  }
+}