Java tutorial
/* * Copyright 2000-2009 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 com.intellij.uiDesigner.propertyInspector.properties; import com.intellij.codeInsight.FileModificationService; import com.intellij.openapi.application.ApplicationManager; import com.intellij.openapi.command.CommandProcessor; import com.intellij.openapi.components.ServiceManager; import com.intellij.openapi.diagnostic.Logger; import com.intellij.openapi.fileEditor.OpenFileDescriptor; import com.intellij.openapi.project.Project; import com.intellij.openapi.ui.Messages; import com.intellij.openapi.util.Ref; import com.intellij.openapi.vfs.VirtualFile; import com.intellij.psi.*; import com.intellij.psi.codeStyle.CodeStyleManager; import com.intellij.psi.util.PsiTreeUtil; import com.intellij.uiDesigner.FormEditingUtil; import com.intellij.uiDesigner.UIDesignerBundle; import com.intellij.uiDesigner.compiler.AsmCodeGenerator; import com.intellij.uiDesigner.compiler.Utils; import com.intellij.uiDesigner.lw.IRootContainer; import com.intellij.uiDesigner.propertyInspector.InplaceContext; import com.intellij.uiDesigner.propertyInspector.Property; import com.intellij.uiDesigner.propertyInspector.PropertyEditor; import com.intellij.uiDesigner.propertyInspector.PropertyRenderer; import com.intellij.uiDesigner.propertyInspector.editors.BooleanEditor; import com.intellij.uiDesigner.propertyInspector.renderers.BooleanRenderer; import com.intellij.uiDesigner.radComponents.RadComponent; import com.intellij.util.IncorrectOperationException; import org.jetbrains.annotations.NotNull; import javax.swing.*; import java.util.List; /** * @author yole */ public class CustomCreateProperty extends Property<RadComponent, Boolean> { private static final Logger LOG = Logger .getInstance("#com.intellij.uiDesigner.propertyInspector.properties.CustomCreateProperty"); public static CustomCreateProperty getInstance(Project project) { return ServiceManager.getService(project, CustomCreateProperty.class); } private final BooleanRenderer myRenderer = new BooleanRenderer(); private final BooleanEditor myEditor = new BooleanEditor() { @Override public JComponent getComponent(final RadComponent component, final Boolean value, final InplaceContext inplaceContext) { JCheckBox result = (JCheckBox) super.getComponent(component, value, inplaceContext); final boolean customCreateRequired = component.isCustomCreateRequired(); if (customCreateRequired) { result.setEnabled(false); result.setSelected(true); } else { result.setEnabled(true); } return result; } }; public CustomCreateProperty() { super(null, "Custom Create"); } public Boolean getValue(final RadComponent component) { return component.isCustomCreate(); } @NotNull public PropertyRenderer<Boolean> getRenderer() { return myRenderer; } public PropertyEditor<Boolean> getEditor() { return myEditor; } @Override public boolean appliesToSelection(final List<RadComponent> selection) { if (selection.size() > 1) { // possible "enabled" state may be different for (RadComponent c : selection) { if (c.isCustomCreateRequired()) { return false; } } } return true; } protected void setValueImpl(final RadComponent component, final Boolean value) throws Exception { if (value.booleanValue() && component.getBinding() == null) { String initialBinding = BindingProperty.getDefaultBinding(component); String binding = Messages.showInputDialog(component.getProject(), UIDesignerBundle.message("custom.create.field.name.prompt"), UIDesignerBundle.message("custom.create.title"), Messages.getQuestionIcon(), initialBinding, new IdentifierValidator(component.getProject())); if (binding == null) { return; } try { new BindingProperty(component.getProject()).setValue(component, binding); } catch (Exception e1) { LOG.error(e1); } } component.setCustomCreate(value.booleanValue()); if (value.booleanValue()) { final IRootContainer root = FormEditingUtil.getRoot(component); if (root.getClassToBind() != null && Utils.getCustomCreateComponentCount(root) == 1) { final PsiClass aClass = FormEditingUtil.findClassToBind(component.getModule(), root.getClassToBind()); if (aClass != null && FormEditingUtil.findCreateComponentsMethod(aClass) == null) { generateCreateComponentsMethod(aClass); } } } } public static void generateCreateComponentsMethod(final PsiClass aClass) { final PsiFile psiFile = aClass.getContainingFile(); if (psiFile == null) return; final VirtualFile vFile = psiFile.getVirtualFile(); if (vFile == null) return; if (!FileModificationService.getInstance().prepareFileForWrite(psiFile)) return; final Ref<SmartPsiElementPointer> refMethod = new Ref<SmartPsiElementPointer>(); CommandProcessor.getInstance().executeCommand(aClass.getProject(), new Runnable() { public void run() { ApplicationManager.getApplication().runWriteAction(new Runnable() { public void run() { PsiElementFactory factory = JavaPsiFacade.getInstance(aClass.getProject()) .getElementFactory(); try { PsiMethod method = factory.createMethodFromText( "private void " + AsmCodeGenerator.CREATE_COMPONENTS_METHOD_NAME + "() { \n // TODO: place custom component creation code here \n }", aClass); final PsiMethod psiMethod = (PsiMethod) aClass.add(method); refMethod.set(SmartPointerManager.getInstance(aClass.getProject()) .createSmartPsiElementPointer(psiMethod)); CodeStyleManager.getInstance(aClass.getProject()).reformat(psiMethod); } catch (IncorrectOperationException e) { LOG.error(e); } } }); } }, null, null); if (!refMethod.isNull()) { SwingUtilities.invokeLater(new Runnable() { public void run() { final PsiMethod element = (PsiMethod) refMethod.get().getElement(); if (element != null) { final PsiCodeBlock body = element.getBody(); assert body != null; final PsiComment comment = PsiTreeUtil.getChildOfType(body, PsiComment.class); if (comment != null) { new OpenFileDescriptor(comment.getProject(), vFile, comment.getTextOffset()) .navigate(true); } } } }); } } }