01: /*
02: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
03: *
04: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
05: *
06: * The contents of this file are subject to the terms of either the GNU
07: * General Public License Version 2 only ("GPL") or the Common Development
08: * and Distribution License("CDDL") (collectively, the "License"). You
09: * may not use this file except in compliance with the License. You can obtain
10: * a copy of the License at https://glassfish.dev.java.net/public/CDDL+GPL.html
11: * or glassfish/bootstrap/legal/LICENSE.txt. See the License for the specific
12: * language governing permissions and limitations under the License.
13: *
14: * When distributing the software, include this License Header Notice in each
15: * file and include the License file at glassfish/bootstrap/legal/LICENSE.txt.
16: * Sun designates this particular file as subject to the "Classpath" exception
17: * as provided by Sun in the GPL Version 2 section of the License file that
18: * accompanied this code. If applicable, add the following below the License
19: * Header, with the fields enclosed by brackets [] replaced by your own
20: * identifying information: "Portions Copyrighted [year]
21: * [name of copyright owner]"
22: *
23: * Contributor(s):
24: *
25: * If you wish your version of this file to be governed by only the CDDL or
26: * only the GPL Version 2, indicate your decision by adding "[Contributor]
27: * elects to include this software in this distribution under the [CDDL or GPL
28: * Version 2] license." If you don't indicate a single choice of license, a
29: * recipient has the option to distribute your version of this file under
30: * either the CDDL, the GPL Version 2 or to extend the choice of license to
31: * its licensees as provided above. However, if you add GPL Version 2 code
32: * and therefore, elected the GPL Version 2 license, then the option applies
33: * only if the new code is made subject to such option by the copyright
34: * holder.
35: */
36:
37: package com.sun.tools.xjc.addon.code_injector;
38:
39: import java.util.Collections;
40: import java.util.List;
41:
42: import com.sun.tools.xjc.Options;
43: import com.sun.tools.xjc.Plugin;
44: import com.sun.tools.xjc.model.CPluginCustomization;
45: import com.sun.tools.xjc.outline.ClassOutline;
46: import com.sun.tools.xjc.outline.Outline;
47: import com.sun.tools.xjc.util.DOMUtils;
48:
49: import org.xml.sax.ErrorHandler;
50:
51: /**
52: * Entry point of a plugin.
53: *
54: * See the javadoc of {@link Plugin} for what those methods mean.
55: *
56: * @author Kohsuke Kawaguchi
57: */
58: public class PluginImpl extends Plugin {
59: public String getOptionName() {
60: return "Xinject-code";
61: }
62:
63: public List<String> getCustomizationURIs() {
64: return Collections.singletonList(Const.NS);
65: }
66:
67: public boolean isCustomizationTagName(String nsUri, String localName) {
68: return nsUri.equals(Const.NS) && localName.equals("code");
69: }
70:
71: public String getUsage() {
72: return " -Xinject-code : inject specified Java code fragments into the generated code";
73: }
74:
75: // meat of the processing
76: public boolean run(Outline model, Options opt,
77: ErrorHandler errorHandler) {
78: for (ClassOutline co : model.getClasses()) {
79: CPluginCustomization c = co.target.getCustomizations()
80: .find(Const.NS, "code");
81: if (c == null)
82: continue; // no customization --- nothing to inject here
83:
84: c.markAsAcknowledged();
85: // TODO: ideally you should validate this DOM element to make sure
86: // that there's no typo/etc. JAXP 1.3 can do this very easily.
87: String codeFragment = DOMUtils.getElementText(c.element);
88:
89: // inject the specified code fragment into the implementation class.
90: co.implClass.direct(codeFragment);
91: }
92:
93: return true;
94: }
95: }
|