001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common
008: * Development and Distribution License("CDDL") (collectively, the
009: * "License"). You may not use this file except in compliance with the
010: * License. You can obtain a copy of the License at
011: * http://www.netbeans.org/cddl-gplv2.html
012: * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
013: * specific language governing permissions and limitations under the
014: * License. When distributing the software, include this License Header
015: * Notice in each file and include the License file at
016: * nbbuild/licenses/CDDL-GPL-2-CP. Sun designates this
017: * particular file as subject to the "Classpath" exception as provided
018: * by Sun in the GPL Version 2 section of the License file that
019: * accompanied this code. If applicable, add the following below the
020: * License Header, with the fields enclosed by brackets [] replaced by
021: * your own identifying information:
022: * "Portions Copyrighted [year] [name of copyright owner]"
023: *
024: * Contributor(s):
025: *
026: * The Original Software is NetBeans. The Initial Developer of the Original
027: * Software is Sun Microsystems, Inc. Portions Copyright 2004-2005 Sun
028: * Microsystems, Inc. All Rights Reserved.
029: *
030: * If you wish your version of this file to be governed by only the CDDL
031: * or only the GPL Version 2, indicate your decision by adding
032: * "[Contributor] elects to include this software in this distribution
033: * under the [CDDL or GPL Version 2] license." If you do not indicate a
034: * single choice of license, a recipient has the option to distribute
035: * your version of this file under either the CDDL, the GPL Version 2 or
036: * to extend the choice of license to its licensees as provided above.
037: * However, if you add GPL Version 2 code and therefore, elected the GPL
038: * Version 2 license, then the option applies only if the new code is
039: * made subject to such option by the copyright holder.
040: */
041: package org.netbeans.modules.jmx.agentwizard;
042:
043: import java.util.HashSet;
044: import java.util.Set;
045:
046: import org.netbeans.modules.jmx.WizardConstants;
047:
048: import org.openide.WizardDescriptor;
049: import org.openide.filesystems.FileObject;
050: import org.openide.loaders.DataObject;
051: import org.openide.loaders.DataFolder;
052: import org.netbeans.api.java.source.JavaSource;
053:
054: import org.netbeans.spi.project.ui.templates.support.Templates;
055:
056: import org.netbeans.modules.jmx.JavaModelHelper;
057: import org.netbeans.modules.jmx.WizardHelpers;
058:
059: /**
060: *
061: * Wizard Agent code generator class
062: */
063: public class AgentGenerator {
064:
065: /**
066: * Entry point to generate agent code.
067: * @param wiz <CODE>WizardDescriptor</CODE> a wizard
068: * @throws java.io.IOException <CODE>IOException</CODE>
069: * @throws java.lang.Exception <CODE>Exception</CODE>
070: * @return <CODE>CreationResults</CODE> results of agent creation
071: */
072: public Set generateAgent(final WizardDescriptor wiz)
073: throws java.io.IOException, Exception {
074: FileObject createdFile = null;
075: final String agentName = Templates.getTargetName(wiz);
076: FileObject agentFolder = Templates.getTargetFolder(wiz);
077: DataFolder agentFolderDataObj = DataFolder
078: .findFolder(agentFolder);
079:
080: //==============================================
081: // agent generation
082: //==============================================
083: DataObject agentDObj = null;
084:
085: FileObject template = Templates.getTemplate(wiz);
086: DataObject dTemplate = DataObject.find(template);
087: agentDObj = dTemplate.createFromTemplate(agentFolderDataObj,
088: agentName);
089:
090: Boolean mainMethodSelected = (Boolean) wiz
091: .getProperty(WizardConstants.PROP_AGENT_MAIN_METHOD_SELECTED);
092: boolean removeMainMethod = ((mainMethodSelected == null) || (!mainMethodSelected));
093:
094: Boolean sampleCodeSelected = (Boolean) wiz
095: .getProperty(WizardConstants.PROP_AGENT_SAMPLE_CODE_SELECTED);
096:
097: boolean removeSampleCode = ((sampleCodeSelected != null) && (!sampleCodeSelected));
098:
099: //Obtain an JavaSource - represents a java file
100: JavaSource js = JavaModelHelper.getSource(agentDObj
101: .getPrimaryFile());
102:
103: JavaModelHelper.generateAgent(js, removeMainMethod,
104: removeSampleCode);
105:
106: WizardHelpers.save(agentDObj);
107: Set s = new HashSet();
108: s.add(agentDObj.getPrimaryFile());
109: return s;
110: }
111: }
|