예제 #1
0
  /**
   * Sets the value of a complex reference (something like $foo.bar) Currently used by
   * ASTSetReference()
   *
   * @see ASTSetDirective
   * @param context context object containing this reference
   * @param value Object to set as value
   * @return true if successful, false otherwise
   * @throws MethodInvocationException
   */
  public boolean setValue(InternalContextAdapter context, Object value)
      throws MethodInvocationException {
    if (jjtGetNumChildren() == 0) {
      context.put(rootString, value);
      return true;
    }

    /*
     *  The rootOfIntrospection is the object we will
     *  retrieve from the Context. This is the base
     *  object we will apply reflection to.
     */

    Object result = getVariableValue(context, rootString);

    if (result == null) {
      String msg = "reference set is not a valid reference at " + Log.formatFileString(uberInfo);
      log.error(msg);
      return false;
    }

    /*
     * How many child nodes do we have?
     */

    for (int i = 0; i < numChildren - 1; i++) {
      result = jjtGetChild(i).execute(result, context);

      if (result == null) {
        if (strictRef) {
          String name = jjtGetChild(i + 1).getFirstToken().image;
          throw new MethodInvocationException(
              "Attempted to access '" + name + "' on a null value",
              null,
              name,
              uberInfo.getTemplateName(),
              jjtGetChild(i + 1).getLine(),
              jjtGetChild(i + 1).getColumn());
        }

        String msg = "reference set is not a valid reference at " + Log.formatFileString(uberInfo);
        log.error(msg);

        return false;
      }
    }

    if (astIndex != null) {
      // If astIndex is not null then we are actually setting an index reference,
      // something of the form $foo[1] =, or in general any reference that ends with
      // the brackets.  This means that we need to call a more general method
      // of the form set(Integer, <something>), or put(Object, <something), where
      // the first parameter is the index value and the second is the LHS of the set.

      Object argument = astIndex.jjtGetChild(0).value(context);
      // If negative, turn -1 into (size - 1)
      argument = ASTIndex.adjMinusIndexArg(argument, result, context, astIndex);
      Object[] params = {argument, value};
      Class[] paramClasses = {
        params[0] == null ? null : params[0].getClass(),
        params[1] == null ? null : params[1].getClass()
      };

      String methodName = "set";
      VelMethod method =
          ClassUtils.getMethod(methodName, params, paramClasses, result, context, astIndex, false);

      if (method == null) {
        // If we can't find a 'set' method, lets try 'put',  This warrents a little
        // investigation performance wise... if the user is using the hash
        // form $foo["blaa"], then it may be expensive to first try and fail on 'set'
        // then go to 'put'?  The problem is that getMethod will try the cache, then
        // perform introspection on 'result' for 'set'
        methodName = "put";
        method =
            ClassUtils.getMethod(
                methodName, params, paramClasses, result, context, astIndex, false);
      }

      if (method == null) {
        // couldn't find set or put method, so bail
        if (strictRef) {
          throw new VelocityException(
              "Found neither a 'set' or 'put' method with param types '("
                  + printClass(paramClasses[0])
                  + ","
                  + printClass(paramClasses[1])
                  + ")' on class '"
                  + result.getClass().getName()
                  + "' at "
                  + Log.formatFileString(astIndex));
        }
        return false;
      }

      try {
        method.invoke(result, params);
      } catch (RuntimeException e) {
        // Kludge since invoke throws Exception, pass up Runtimes
        throw e;
      } catch (Exception e) {
        throw new MethodInvocationException(
            "Exception calling method '"
                + methodName
                + "("
                + printClass(paramClasses[0])
                + ","
                + printClass(paramClasses[1])
                + ")' in  "
                + result.getClass(),
            e.getCause(),
            identifier,
            astIndex.getTemplateName(),
            astIndex.getLine(),
            astIndex.getColumn());
      }

      return true;
    }

    /*
     *  We support two ways of setting the value in a #set($ref.foo = $value ) :
     *  1) ref.setFoo( value )
     *  2) ref,put("foo", value ) to parallel the get() map introspection
     */

    try {
      VelPropertySet vs = rsvc.getUberspect().getPropertySet(result, identifier, value, uberInfo);

      if (vs == null) {
        if (strictRef) {
          throw new MethodInvocationException(
              "Object '"
                  + result.getClass().getName()
                  + "' does not contain property '"
                  + identifier
                  + "'",
              null,
              identifier,
              uberInfo.getTemplateName(),
              uberInfo.getLine(),
              uberInfo.getColumn());
        } else {
          return false;
        }
      }

      vs.invoke(result, value);
    } catch (InvocationTargetException ite) {
      /*
       *  this is possible
       */

      throw new MethodInvocationException(
          "ASTReference : Invocation of method '"
              + identifier
              + "' in  "
              + result.getClass()
              + " threw exception "
              + ite.getTargetException().toString(),
          ite.getTargetException(),
          identifier,
          getTemplateName(),
          this.getLine(),
          this.getColumn());
    }
    /** pass through application level runtime exceptions */
    catch (RuntimeException e) {
      throw e;
    } catch (Exception e) {
      /*
       *  maybe a security exception?
       */
      String msg =
          "ASTReference setValue() : exception : "
              + e
              + " template at "
              + Log.formatFileString(uberInfo);
      log.error(msg, e);
      throw new VelocityException(msg, e);
    }

    return true;
  }
예제 #2
0
  /**
   * This method helps to implement the "render literal if null" functionality.
   *
   * <p>VelocimacroProxy saves references to macro arguments (AST nodes) so that if we have a macro
   * #foobar($a $b) then there is key "$a.literal" which points to the literal presentation of the
   * argument provided to variable $a. If the value of $a is null, we render the string that was
   * provided as the argument.
   *
   * @param context
   * @return
   */
  private String getNullString(InternalContextAdapter context) {
    Object callingArgument = context.get(".literal." + nullString);

    if (callingArgument != null) return ((Node) callingArgument).literal();
    else return nullString;
  }