TypeDescriptor.java

/**
 * MVEL 2.0
 * Copyright (C) 2007 The Codehaus
 * Mike Brock, Dhanji Prasanna, John Graham, Mark Proctor
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.mvel2.ast;

import org.mvel2.CompileException;
import org.mvel2.ParserContext;
import org.mvel2.compiler.AbstractParser;
import org.mvel2.compiler.ExecutableStatement;
import org.mvel2.integration.VariableResolverFactory;
import org.mvel2.util.ParseTools;

import java.io.Serializable;
import java.util.Iterator;
import java.util.LinkedList;

import static java.lang.Character.isDigit;
import static org.mvel2.ast.ASTNode.COMPILE_IMMEDIATE;
import static org.mvel2.util.ArrayTools.findFirst;
import static org.mvel2.util.ParseTools.*;
import static org.mvel2.util.ReflectionUtil.toPrimitiveArrayType;

public class TypeDescriptor implements Serializable {
  private String className;
  private char[] expr;

  private int start;
  private int offset;

  private ArraySize[] arraySize;
  private ExecutableStatement[] compiledArraySize;
  int endRange;

  public TypeDescriptor(char[] name, int start, int offset, int fields) {
    updateClassName(this.expr = name, this.start = start, this.offset = offset, fields);
  }

  public void updateClassName(char[] name, int start, int offset, int fields) {
    this.expr = name;

    if (offset == 0 || !ParseTools.isIdentifierPart(name[start]) || isDigit(name[start])) return;

    if ((endRange = findFirst('(', start, offset, name)) == -1) {
      if ((endRange = findFirst('[', start, offset, name)) != -1) {
        className = new String(name, start, endRange - start).trim();
        int to;

        LinkedList<char[]> sizes = new LinkedList<char[]>();

        int end = start + offset;
        while (endRange < end) {
          while (endRange < end && isWhitespace(name[endRange])) endRange++;

          if (endRange == end || name[endRange] == '{') break;

          if (name[endRange] != '[') {
            throw new CompileException("unexpected token in constructor", name, endRange);
          }
          to = balancedCapture(name, endRange, start + offset, '[');
          sizes.add(subset(name, ++endRange, to - endRange));
          endRange = to + 1;
        }

        Iterator<char[]> iter = sizes.iterator();
        arraySize = new ArraySize[sizes.size()];

        for (int i = 0; i < arraySize.length; i++)
          arraySize[i] = new ArraySize(iter.next());

        if ((fields & COMPILE_IMMEDIATE) != 0) {
          compiledArraySize = new ExecutableStatement[arraySize.length];
          for (int i = 0; i < compiledArraySize.length; i++)
            compiledArraySize[i] = (ExecutableStatement) subCompileExpression(arraySize[i].value);
        }

        return;
      }

      className = new String(name, start, offset).trim();
    }
    else {
      className = new String(name, start, endRange - start).trim();
    }
  }

  public boolean isArray() {
    return arraySize != null;
  }

  public int getArrayLength() {
    return arraySize.length;
  }

  public ArraySize[] getArraySize() {
    return arraySize;
  }

  public ExecutableStatement[] getCompiledArraySize() {
    return compiledArraySize;
  }

  public String getClassName() {
    return className;
  }

  public void setClassName(String className) {
    this.className = className;
  }

  public boolean isClass() {
    return className != null && className.length() != 0;
  }

  public int getEndRange() {
    return endRange;
  }

  public void setEndRange(int endRange) {
    this.endRange = endRange;
  }
  
  public Class<?> getClassReference() throws ClassNotFoundException {
    return getClassReference(null, this);
  }
  
  public Class<?> getClassReference(ParserContext ctx) throws ClassNotFoundException {
    return getClassReference(ctx,this);
  }
  
  public static Class getClassReference(Class baseType,
                                        TypeDescriptor tDescr,
                                        VariableResolverFactory factory, ParserContext ctx) throws ClassNotFoundException {
    return findClass(factory, repeatChar('[', tDescr.arraySize.length) + "L" + baseType.getName() + ";", ctx);
  }

  public static Class getClassReference(ParserContext ctx, Class cls, TypeDescriptor tDescr) throws ClassNotFoundException {
    if (tDescr.isArray()) {
      cls = cls.isPrimitive() ?
              toPrimitiveArrayType(cls) :
              findClass(null, repeatChar('[', tDescr.arraySize.length) + "L" + cls.getName() + ";", ctx);
    }
    return cls;
  }


  public static Class getClassReference(ParserContext ctx, TypeDescriptor tDescr) throws ClassNotFoundException {
    Class cls;
    if (ctx != null && ctx.hasImport(tDescr.className)) {
      cls = ctx.getImport(tDescr.className);
      if (tDescr.isArray()) {
        cls = cls.isPrimitive() ?
                toPrimitiveArrayType(cls) :
                findClass(null, repeatChar('[', tDescr.arraySize.length) + "L" + cls.getName() + ";", ctx);
      }
    }
    else if (ctx == null && hasContextFreeImport(tDescr.className)) {
      cls = getContextFreeImport(tDescr.className);
      if (tDescr.isArray()) {
        cls = cls.isPrimitive() ?
                toPrimitiveArrayType(cls) :
                findClass(null, repeatChar('[', tDescr.arraySize.length) + "L" + cls.getName() + ";", ctx);
      }
    }
    else {
      cls = createClass(tDescr.getClassName(), ctx);
      if (tDescr.isArray()) {
        cls = cls.isPrimitive() ?
                toPrimitiveArrayType(cls) :
                findClass(null, repeatChar('[', tDescr.arraySize.length) + "L" + cls.getName() + ";", ctx);
      }
    }

    return cls;
  }

  public boolean isUndimensionedArray() {
    if (arraySize != null) {
      for (ArraySize anArraySize : arraySize) {
        if (anArraySize.value.length == 0) return true;
      }
    }

    return false;
  }

  public static boolean hasContextFreeImport(String name) {
    return AbstractParser.LITERALS.containsKey(name) && AbstractParser.LITERALS.get(name) instanceof Class;
  }

  public static Class getContextFreeImport(String name) {
    return (Class) AbstractParser.LITERALS.get(name);
  }

  public char[] getExpr() {
    return expr;
  }

  public int getStart() {
    return start;
  }

  public int getOffset() {
    return offset;
  }
}