com.yworks.yshrink.ant.MethodSection Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yguard Show documentation
The open-source Java obfuscation tool working with Ant and Gradle by yWorks - the diagramming experts
The newest version!
package com.yworks.yshrink.ant;

import com.yworks.common.ant.TypePatternSet;
import com.yworks.common.ant.PatternMatchedSection;
import com.yworks.yshrink.util.Util;
import org.objectweb.asm.Type;

import java.util.EnumSet;

/**
 * Used by ant to handle the method element.
 */
public class MethodSection extends PatternMatchedSection {

  private String signature;
  private String name;
  private String className;
  private String returnType;
  private String args;
  private String throwsClause;


  {
    types = EnumSet.of(
        TypePatternSet.Type.NAME,
        TypePatternSet.Type.CLASS
    );
  }

  /**
   * Gets signature.
   *
   * @return the signature
   */
  public String getSignature() {
    return signature;
  }

  /**
   * Sets signature.
   *
   * @param signature the signature
   */
  public void setSignature( String signature ) {
    this.signature = signature;
    String[] methodArr = Util.toNativeMethod( signature );
    String methodName = methodArr[ 0 ];
    String methodDesc = methodArr[ 1 ];
    setName( methodName );
    setReturnType( Util.toJavaType( Type.getReturnType( methodDesc ).getDescriptor() ) );

    setArgs( Util.getArgumentString( Type.getArgumentTypes( methodDesc ) ) );
  }

  /**
   * Gets args.
   *
   * @return the args
   */
  public String getArgs() {
    return args;
  }

  /**
   * Sets args.
   *
   * @param args the args
   */
  public void setArgs( String args ) {
    this.args = args;
  }

  /**
   * Sets name.
   *
   * @param name the name
   */
  public void setName( String name ) {

    // in yGuard, the name-attribute is the signature.
    if ( name.trim().indexOf( ' ' ) != -1 ) {
      setSignature( name );
    } else {
      this.name = name;
    }
  }

  /**
   * Sets class.
   *
   * @param name the name
   */
  public void setClass( String name ) {
    this.className = Util.toInternalClass( name );
  }

  /**
   * Gets name.
   *
   * @return the name
   */
  public String getName() {
    return name;
  }

  /**
   * Gets class name.
   *
   * @return the class name
   */
  public String getClassName() {
    return className;
  }

  /**
   * Gets return type.
   *
   * @return the return type
   */
  public String getReturnType() {
    return returnType;
  }

  /**
   * Sets return type.
   *
   * @param returnType the return type
   */
  public void setReturnType( String returnType ) {
    this.returnType = Util.toInternalClass( returnType );
  }

  /**
   * Gets throws.
   *
   * @return the throws
   */
  public String getThrows() {
    return throwsClause;
  }

  /**
   * Sets throws.
   *
   * @param throwsClause the throws clause
   */
  public void setThrows( String throwsClause ) {
    this.throwsClause = throwsClause;
  }

  @Override
  public TypePatternSet createPatternSet() {
    System.out.println( "MethodSection.createPatternSet" );
    TypePatternSet typePatternSet = newTypePatternSet();
    typePatternSet.setType( "class" );
    addPatternSet( typePatternSet, typePatternSet.getType() );
    return typePatternSet;
  }

  /**
   * Instantiates a type pattern set,
   * subclasses may provide custom implementations.
   *
   * @return the new type pattern set
   */
  protected TypePatternSet newTypePatternSet() {
    return new TypePatternSet();
  }
}