/* * Copyright 2013-2017 consulo.io * * 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 consulo.csharp.ide.actions; import java.util.ArrayList; import java.util.List; import java.util.Map; import java.util.UUID; import javax.annotation.Nonnull; import javax.annotation.Nullable; import com.intellij.ide.actions.CreateFileFromTemplateAction; import com.intellij.ide.fileTemplates.DefaultCreateFromTemplateHandler; import com.intellij.ide.fileTemplates.FileTemplate; import com.intellij.openapi.fileTypes.FileType; import com.intellij.openapi.fileTypes.FileTypeRegistry; import com.intellij.openapi.module.Module; import com.intellij.openapi.module.ModuleUtilCore; import com.intellij.openapi.util.text.StringUtil; import com.intellij.openapi.vfs.LocalFileSystem; import com.intellij.openapi.vfs.VirtualFile; import com.intellij.openapi.vfs.VirtualFileSystem; import com.intellij.psi.PsiDirectory; import com.intellij.psi.util.QualifiedName; import com.intellij.testFramework.LightVirtualFile; import consulo.annotation.access.RequiredReadAction; import consulo.csharp.ide.refactoring.util.CSharpNameSuggesterUtil; import consulo.csharp.lang.CSharpFileType; import consulo.dotnet.module.extension.DotNetSimpleModuleExtension; /** * @author VISTALL * @since 24.03.2016 */ public class CSharpCreateFromTemplateHandler extends DefaultCreateFromTemplateHandler { @Nonnull public static CSharpCreateFromTemplateHandler getInstance() { return EP_NAME.findExtension(CSharpCreateFromTemplateHandler.class); } @Nullable @RequiredReadAction public static Module findModuleByPsiDirectory(final PsiDirectory directory) { LightVirtualFile l = new LightVirtualFile("test.cs", CSharpFileType.INSTANCE, "") { @Override public VirtualFile getParent() { return directory.getVirtualFile(); } @Nonnull @Override public VirtualFileSystem getFileSystem() { return LocalFileSystem.getInstance(); } }; return CreateFileFromTemplateAction.ModuleResolver.EP_NAME.composite().resolveModule(directory, CSharpFileType.INSTANCE); } @Override public boolean canCreate(PsiDirectory[] dirs) { return false; } @Override public boolean handlesTemplate(FileTemplate template) { FileType fileTypeByExtension = FileTypeRegistry.getInstance().getFileTypeByExtension(template.getExtension()); return fileTypeByExtension == CSharpFileType.INSTANCE; } @Override @RequiredReadAction public void prepareProperties(Map<String, Object> props) { PsiDirectory directory = (PsiDirectory) props.get("psiDirectory"); if(directory == null) { return; } Module module = ModuleUtilCore.findModuleForPsiElement(directory); if(module != null) { props.put("MODULE", module.getName()); } props.put("GUID", UUID.randomUUID().toString()); DotNetSimpleModuleExtension<?> extension = ModuleUtilCore.getExtension(directory, DotNetSimpleModuleExtension.class); if(extension == null) { Module moduleByPsiDirectory = findModuleByPsiDirectory(directory); if(moduleByPsiDirectory != null) { extension = ModuleUtilCore.getExtension(moduleByPsiDirectory, DotNetSimpleModuleExtension.class); } } String namespace = null; if(extension != null) { namespace = formatNamespace(extension.getNamespaceGeneratePolicy().calculateNamespace(directory)); } props.put("NAMESPACE_NAME", namespace); } @Nullable private static String formatNamespace(@Nullable String original) { if(StringUtil.isEmpty(original)) { return null; } QualifiedName qualifiedName = QualifiedName.fromDottedString(original); List<String> components = qualifiedName.getComponents(); List<String> newComponents = new ArrayList<>(components.size()); for(String component : components) { if(CSharpNameSuggesterUtil.isKeyword(component)) { newComponents.add("@" + component); } else { newComponents.add(component); } } return String.join(".", newComponents); } }