xml/impl/src/com/intellij/xml/actions/xmlbeans/GenerateInstanceDocumentFromSchemaDialog.java

   1 package com.intellij.xml.actions.xmlbeans;
   2
   3
   4 import com.intellij.javaee.ExternalResourceManager;
   5 import com.intellij.openapi.project.Project;
   6 import com.intellij.openapi.ui.DialogWrapper;
   7 import com.intellij.openapi.ui.TextFieldWithBrowseButton;
   8 import com.intellij.openapi.vfs.VfsUtil;
   9 import com.intellij.openapi.vfs.VirtualFile;
  10 import com.intellij.psi.PsiFile;
  11 import com.intellij.psi.PsiManager;
  12 import com.intellij.psi.xml.XmlFile;
  13 import com.intellij.psi.xml.XmlTag;
  14 import com.intellij.xml.XmlBundle;
  15 import com.intellij.xml.XmlElementDescriptor;
  16 import org.jetbrains.annotations.Nullable;
  17
  18 import javax.swing.*;
  19 import javax.swing.event.DocumentEvent;
  20 import javax.swing.event.DocumentListener;
  21 import java.awt.*;
  22 import java.awt.event.ActionEvent;
  23 import java.awt.event.ActionListener;
  24 import java.awt.event.KeyAdapter;
  25 import java.awt.event.KeyEvent;
  26 import java.util.Collections;
  27 import java.util.List;
  28
  29 /**
  30  * @author Konstantin Bulenkov
  31  */
  32 public class GenerateInstanceDocumentFromSchemaDialog extends DialogWrapper {
  33   private JPanel panel;
  34   private TextFieldWithBrowseButton generateFromUrl;
  35   private JLabel status;
  36   private JLabel statusText;
  37   private JLabel generateFromUrlText;
  38   private JComboBox rootElementChooser;
  39   private JLabel rootElementChooserText;
  40   private JCheckBox enableRestrictionCheck;
  41   private JCheckBox enableUniqueCheck;
  42   private JTextField outputFileName;
  43   private JLabel outputFileNameText;
  44   private String previousUri;
  45   private Runnable myOkAction;
  46   private Project myProject;
  47
  48   public GenerateInstanceDocumentFromSchemaDialog(Project project, VirtualFile file) {
  49     super(project, true);
  50     myProject = project;
  51
  52     UIUtils.configureBrowseButton(project, generateFromUrl, new String[] {"xsd"}, XmlBundle.message("select.xsd.schema.dialog.title"), false);
  53
  54     doInitFor(rootElementChooserText, rootElementChooser);
  55     doInitFor(generateFromUrlText, generateFromUrl.getTextField());
  56     doInitFor(outputFileNameText, outputFileName);
  57     generateFromUrl.setText(file.getPresentableUrl());
  58     updateFile();
  59
  60     setTitle(XmlBundle.message("generate.instance.document.from.schema.dialog.title"));
  61
  62     init();
  63
  64     outputFileName.setText(file.getName()+ ".xml");
  65   }
  66
  67   public void doInitFor(JLabel textComponent, JComponent component) {
  68     textComponent.setLabelFor(component);
  69
  70     if (component instanceof JTextField) {
  71       ((JTextField)component).getDocument().addDocumentListener(new DocumentListener() {
  72         public void insertUpdate(DocumentEvent e) {
  73           validateData();
  74         }
  75
  76         public void removeUpdate(DocumentEvent e) {
  77           validateData();
  78         }
  79
  80         public void changedUpdate(DocumentEvent e) {
  81           validateData();
  82         }
  83       });
  84     } else if (component instanceof JComboBox) {
  85       JComboBox jComboBox = ((JComboBox) component);
  86
  87       jComboBox.addActionListener(new ActionListener() {
  88         public void actionPerformed(ActionEvent e) {
  89           validateData();
  90         }
  91       });
  92
  93       ((JTextField)jComboBox.getEditor().getEditorComponent()).getDocument().addDocumentListener(new DocumentListener() {
  94         public void insertUpdate(DocumentEvent e) {
  95           validateData();
  96           updateFile();
  97         }
  98
  99         public void removeUpdate(DocumentEvent e) {
 100           validateData();
 101           updateFile();
 102         }
 103
 104         public void changedUpdate(DocumentEvent e) {
 105           validateData();
 106           updateFile();
 107         }
 108       });
 109
 110       if (jComboBox.isEditable()) {
 111         jComboBox.getEditor().getEditorComponent().addKeyListener(new KeyAdapter() {
 112           public void keyTyped(KeyEvent e) {
 113             validateData();
 114           }
 115         });
 116       }
 117     }
 118   }
 119
 120   private void validateData() {
 121     String msg = doValidateWithData();
 122     setOKActionEnabled(msg == null);
 123     status.setText(msg == null ? "" : msg);
 124     status.setForeground(Color.RED);
 125   }
 126
 127   public void configureComboBox(JComboBox combo, List<String> lastValues) {  // without -editor.selectAll- no focus
 128     combo.setModel(new DefaultComboBoxModel(lastValues.toArray(new String[lastValues.size()])));
 129   }
 130
 131   private void updateFile() {
 132     String uri = generateFromUrl.getText();
 133     boolean hasPrevious = (previousUri != null && previousUri.equals(uri));
 134     final PsiFile psifile = findFile(uri);
 135     List<String> myRootValues;
 136
 137     if (psifile == null) {
 138       configureComboBox(rootElementChooser, Collections.EMPTY_LIST);
 139       return;
 140     }
 141
 142     final XmlTag rootTag = getRootTag(psifile);
 143
 144     if (rootTag == null) {
 145       configureComboBox(rootElementChooser, Collections.EMPTY_LIST);
 146       rootElementChooser.setSelectedIndex(-1);
 147       previousUri = uri;
 148       return;
 149     }
 150
 151     myRootValues = Xsd2InstanceUtils.addVariantsFromRootTag(rootTag);
 152
 153     Object selectedItem = rootElementChooser.getSelectedItem();
 154     configureComboBox(rootElementChooser, myRootValues);
 155
 156     if (hasPrevious) {
 157       rootElementChooser.setSelectedItem(selectedItem);
 158     } else {
 159       rootElementChooser.setSelectedIndex(myRootValues.size() > 0 ? 0:-1);
 160     }
 161     previousUri = uri;
 162   }
 163
 164   private XmlTag getRootTag(PsiFile psifile) {
 165     return ((XmlFile) psifile).getDocument().getRootTag();
 166   }
 167
 168   private PsiFile findFile(String uri) {
 169     final VirtualFile file = uri != null ? VfsUtil.findRelativeFile(ExternalResourceManager.getInstance().getResourceLocation(uri), null):null;
 170     return file != null ? PsiManager.getInstance(myProject).findFile(file):null;
 171   }
 172
 173   public String getOutputFileName() {
 174     return outputFileName.getText();
 175   }
 176
 177   public Boolean areCurrentParametersStillValid() {
 178     updateFile();
 179     return rootElementChooser.getSelectedItem() != null;
 180   }
 181
 182   @Nullable
 183   protected String doValidateWithData() {
 184     String rootElementName = getElementName();
 185     if (rootElementName == null || rootElementName.length() == 0) {
 186       return XmlBundle.message("schema2.instance.no.valid.root.element.name.validation.error");
 187     }
 188
 189     final PsiFile psiFile = findFile(getUrl().getText());
 190     if (psiFile instanceof XmlFile) {
 191       final XmlTag tag = getRootTag(psiFile);
 192       if (tag != null) {
 193         final XmlElementDescriptor descriptor = Xsd2InstanceUtils.getDescriptor(tag, rootElementName);
 194
 195         if (descriptor == null) {
 196           return XmlBundle.message("schema2.instance.no.valid.root.element.name.validation.error");
 197         }
 198       }
 199     }
 200
 201     final String fileName = getOutputFileName();
 202     if (fileName == null || fileName.length() == 0) {
 203       return XmlBundle.message("schema2.instance.output.file.name.is.empty.validation.problem");
 204     }
 205     return null;
 206
 207   }
 208
 209   protected boolean isAcceptableFile(VirtualFile virtualFile) {
 210     return GenerateInstanceDocumentFromSchemaAction.isAcceptableFileForGenerateSchemaFromInstanceDocument(virtualFile);
 211   }
 212
 213
 214   protected TextFieldWithBrowseButton getUrl() {
 215     return generateFromUrl;
 216   }
 217
 218   protected JLabel getUrlText() {
 219     return generateFromUrlText;
 220   }
 221
 222   protected JLabel getStatusTextField() {
 223     return statusText;
 224   }
 225
 226   protected JLabel getStatusField() {
 227     return status;
 228   }
 229
 230   protected JComponent createCenterPanel() {
 231     return panel;
 232   }
 233
 234   boolean enableUniquenessCheck() {
 235     return enableUniqueCheck.isSelected();
 236   }
 237
 238   boolean enableRestrictionCheck() {
 239     return enableRestrictionCheck.isSelected();
 240   }
 241
 242   String getElementName() {
 243     return (String)rootElementChooser.getSelectedItem();
 244   }
 245
 246   public void setOkAction(Runnable runnable) {
 247     myOkAction = runnable;
 248   }
 249
 250   @Override
 251   protected void doOKAction() {
 252     super.doOKAction();
 253     if (myOkAction != null) {
 254       myOkAction.run();
 255     }
 256   }
 257
 258 }