|
13514
|
1 /*
|
|
|
2 * Copyright (c) 2013 Oracle and/or its affiliates. All rights reserved. This
|
|
|
3 * code is released under a tri EPL/GPL/LGPL license. You can use it,
|
|
|
4 * redistribute it and/or modify it under the terms of the:
|
|
|
5 *
|
|
|
6 * Eclipse Public License version 1.0
|
|
|
7 * GNU General Public License version 2
|
|
|
8 * GNU Lesser General Public License version 2.1
|
|
|
9 */
|
|
|
10 package com.oracle.truffle.ruby.nodes.objects;
|
|
|
11
|
|
|
12 import com.oracle.truffle.api.*;
|
|
|
13 import com.oracle.truffle.api.frame.*;
|
|
|
14 import com.oracle.truffle.ruby.nodes.*;
|
|
|
15 import com.oracle.truffle.ruby.runtime.*;
|
|
|
16 import com.oracle.truffle.ruby.runtime.control.*;
|
|
|
17 import com.oracle.truffle.ruby.runtime.core.*;
|
|
|
18
|
|
|
19 /**
|
|
|
20 * Define a new class, or get the existing one of the same name.
|
|
|
21 */
|
|
|
22 public class DefineOrGetClassNode extends RubyNode {
|
|
|
23
|
|
|
24 private final String name;
|
|
|
25 @Child protected RubyNode moduleDefinedIn;
|
|
|
26 @Child protected RubyNode superClass;
|
|
|
27
|
|
|
28 public DefineOrGetClassNode(RubyContext context, SourceSection sourceSection, String name, RubyNode moduleDefinedIn, RubyNode superClass) {
|
|
|
29 super(context, sourceSection);
|
|
|
30 this.name = name;
|
|
|
31 this.moduleDefinedIn = adoptChild(moduleDefinedIn);
|
|
|
32 this.superClass = adoptChild(superClass);
|
|
|
33 }
|
|
|
34
|
|
|
35 @Override
|
|
|
36 public Object execute(VirtualFrame frame) {
|
|
|
37 CompilerAsserts.neverPartOfCompilation();
|
|
|
38
|
|
|
39 final RubyContext context = getContext();
|
|
|
40
|
|
|
41 final RubyModule moduleDefinedInObject = (RubyModule) moduleDefinedIn.execute(frame);
|
|
|
42
|
|
|
43 // Look for a current definition of the class, or create a new one
|
|
|
44
|
|
|
45 final Object constantValue = moduleDefinedInObject.lookupConstant(name);
|
|
|
46
|
|
|
47 RubyClass definingClass;
|
|
|
48
|
|
|
49 if (constantValue == null) {
|
|
|
50 final Object self = frame.getArguments(RubyArguments.class).getSelf();
|
|
|
51
|
|
|
52 RubyModule parentModule;
|
|
|
53
|
|
|
54 if (self instanceof RubyModule) {
|
|
|
55 parentModule = (RubyModule) self;
|
|
|
56 } else {
|
|
|
57 // Because it's top level, and so self is the magic main object
|
|
|
58 parentModule = null;
|
|
|
59 }
|
|
|
60
|
|
|
61 final RubyClass superClassObject = (RubyClass) superClass.execute(frame);
|
|
|
62
|
|
|
63 if (superClassObject instanceof RubyException.RubyExceptionClass) {
|
|
|
64 definingClass = new RubyException.RubyExceptionClass(superClassObject, name);
|
|
|
65 } else {
|
|
|
66 definingClass = new RubyClass(parentModule, superClassObject, name);
|
|
|
67 }
|
|
|
68
|
|
|
69 moduleDefinedInObject.setConstant(name, definingClass);
|
|
|
70 moduleDefinedInObject.getSingletonClass().setConstant(name, definingClass);
|
|
|
71
|
|
|
72 definingClass.getRubyClass().include(moduleDefinedInObject);
|
|
|
73 } else {
|
|
|
74 if (constantValue instanceof RubyClass) {
|
|
|
75 definingClass = (RubyClass) constantValue;
|
|
|
76 } else {
|
|
|
77 throw new RaiseException(context.getCoreLibrary().typeErrorIsNotA(constantValue.toString(), "class"));
|
|
|
78 }
|
|
|
79 }
|
|
|
80
|
|
|
81 return definingClass;
|
|
|
82 }
|
|
|
83 }
|
