ScriptClassInfo.java revision 2:da1e581c933b 
1/*
2 * Copyright (c) 2010, 2012, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.  Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25
26package jdk.nashorn.internal.tools.nasgen;
27
28import java.util.Collections;
29import java.util.HashMap;
30import java.util.LinkedList;
31import java.util.List;
32import java.util.Map;
33import jdk.internal.org.objectweb.asm.Type;
34import jdk.nashorn.internal.objects.annotations.Constructor;
35import jdk.nashorn.internal.objects.annotations.Function;
36import jdk.nashorn.internal.objects.annotations.Getter;
37import jdk.nashorn.internal.objects.annotations.Property;
38import jdk.nashorn.internal.objects.annotations.ScriptClass;
39import jdk.nashorn.internal.objects.annotations.Setter;
40import jdk.nashorn.internal.objects.annotations.SpecializedConstructor;
41import jdk.nashorn.internal.objects.annotations.SpecializedFunction;
42import jdk.nashorn.internal.objects.annotations.Where;
43import jdk.nashorn.internal.tools.nasgen.MemberInfo.Kind;
44
45/**
46 * All annotation information from a class that is annotated with
47 * the annotation com.sun.oracle.objects.annotations.ScriptClass.
48 *
49 */
50public final class ScriptClassInfo {
51    // descriptots for various annotations
52    static final String SCRIPT_CLASS_ANNO_DESC  = Type.getDescriptor(ScriptClass.class);
53    static final String CONSTRUCTOR_ANNO_DESC   = Type.getDescriptor(Constructor.class);
54    static final String FUNCTION_ANNO_DESC      = Type.getDescriptor(Function.class);
55    static final String GETTER_ANNO_DESC        = Type.getDescriptor(Getter.class);
56    static final String SETTER_ANNO_DESC        = Type.getDescriptor(Setter.class);
57    static final String PROPERTY_ANNO_DESC      = Type.getDescriptor(Property.class);
58    static final String WHERE_ENUM_DESC         = Type.getDescriptor(Where.class);
59    static final String SPECIALIZED_FUNCTION    = Type.getDescriptor(SpecializedFunction.class);
60    static final String SPECIALIZED_CONSTRUCTOR = Type.getDescriptor(SpecializedConstructor.class);
61
62    static final Map<String, Kind> annotations = new HashMap<>();
63
64    static {
65        annotations.put(SCRIPT_CLASS_ANNO_DESC, Kind.SCRIPT_CLASS);
66        annotations.put(FUNCTION_ANNO_DESC, Kind.FUNCTION);
67        annotations.put(CONSTRUCTOR_ANNO_DESC, Kind.CONSTRUCTOR);
68        annotations.put(GETTER_ANNO_DESC, Kind.GETTER);
69        annotations.put(SETTER_ANNO_DESC, Kind.SETTER);
70        annotations.put(PROPERTY_ANNO_DESC, Kind.PROPERTY);
71        annotations.put(SPECIALIZED_FUNCTION, Kind.SPECIALIZED_FUNCTION);
72        annotations.put(SPECIALIZED_CONSTRUCTOR, Kind.SPECIALIZED_CONSTRUCTOR);
73    }
74
75    // name of the script class
76    private String name;
77    // member info for script properties
78    private List<MemberInfo> members = Collections.emptyList();
79    // java class name that is annotated with @ScriptClass
80    private String javaName;
81
82    /**
83     * @return the name
84     */
85    public String getName() {
86        return name;
87    }
88
89    /**
90     * @param name the name to set
91     */
92    public void setName(final String name) {
93        this.name = name;
94    }
95
96    /**
97     * @return the members
98     */
99    public List<MemberInfo> getMembers() {
100        return Collections.unmodifiableList(members);
101    }
102
103    /**
104     * @param members the members to set
105     */
106    public void setMembers(final List<MemberInfo> members) {
107        this.members = members;
108    }
109
110    MemberInfo getConstructor() {
111        for (final MemberInfo memInfo : members) {
112            if (memInfo.getKind() == Kind.CONSTRUCTOR) {
113                return memInfo;
114            }
115        }
116        return null;
117    }
118
119    List<MemberInfo> getSpecializedConstructors() {
120        final List<MemberInfo> res = new LinkedList<>();
121        for (final MemberInfo memInfo : members) {
122            if (memInfo.getKind() == Kind.SPECIALIZED_CONSTRUCTOR) {
123                res.add(memInfo);
124            }
125        }
126        return res;
127    }
128
129    int getPrototypeMemberCount() {
130        int count = 0;
131        for (final MemberInfo memInfo : members) {
132            if (memInfo.getWhere() == Where.PROTOTYPE || memInfo.isConstructor()) {
133                count++;
134            }
135        }
136        return count;
137    }
138
139    int getConstructorMemberCount() {
140        int count = 0;
141        for (final MemberInfo memInfo : members) {
142            if (memInfo.getWhere() == Where.CONSTRUCTOR) {
143                count++;
144            }
145        }
146        return count;
147    }
148
149    int getInstancePropertyCount() {
150        int count = 0;
151        for (final MemberInfo memInfo : members) {
152            if (memInfo.getWhere() == Where.INSTANCE) {
153                count++;
154            }
155        }
156        return count;
157    }
158
159    MemberInfo find(final String findJavaName, final String findJavaDesc, final int findAccess) {
160        for (final MemberInfo memInfo : members) {
161            if (memInfo.getJavaName().equals(findJavaName) &&
162                memInfo.getJavaDesc().equals(findJavaDesc) &&
163                memInfo.getJavaAccess() == findAccess) {
164                return memInfo;
165            }
166        }
167        return null;
168    }
169
170    List<MemberInfo> findSpecializations(final String methodName) {
171        final List<MemberInfo> res = new LinkedList<>();
172        for (final MemberInfo memInfo : members) {
173            if (memInfo.getName().equals(methodName) &&
174                memInfo.getKind() == Kind.SPECIALIZED_FUNCTION) {
175                res.add(memInfo);
176            }
177        }
178        return res;
179    }
180
181    MemberInfo findSetter(final MemberInfo getter) {
182        assert getter.getKind() == Kind.GETTER : "getter expected";
183        final String getterName = getter.getName();
184        final Where getterWhere = getter.getWhere();
185        for (final MemberInfo memInfo : members) {
186            if (memInfo.getKind() == Kind.SETTER &&
187                getterName.equals(memInfo.getName()) &&
188                getterWhere == memInfo.getWhere()) {
189                return memInfo;
190            }
191        }
192        return null;
193    }
194
195    /**
196     * @return the javaName
197     */
198    public String getJavaName() {
199        return javaName;
200    }
201
202    /**
203     * @param javaName the javaName to set
204     */
205    void setJavaName(final String javaName) {
206        this.javaName = javaName;
207    }
208
209    String getConstructorClassName() {
210        return getJavaName() + StringConstants.CONSTRUCTOR_SUFFIX;
211    }
212
213    String getPrototypeClassName() {
214        return getJavaName() + StringConstants.PROTOTYPE_SUFFIX;
215    }
216
217    void verify() {
218        boolean constructorSeen = false;
219        for (final MemberInfo memInfo : getMembers()) {
220            if (memInfo.isConstructor()) {
221                if (constructorSeen) {
222                    error("more than @Constructor method");
223                }
224                constructorSeen = true;
225            }
226            try {
227                memInfo.verify();
228            } catch (final Exception e) {
229                error(e.getMessage());
230            }
231        }
232    }
233
234    private void error(final String msg) throws RuntimeException {
235        throw new RuntimeException(javaName + " : " + msg);
236    }
237}
238