jpexs-decompiler/trunk/test/com/jpexs/decompiler/flash/Generator.java

package com.jpexs.decompiler.flash;

import com.jpexs.decompiler.flash.abc.ABC;
import com.jpexs.decompiler.flash.abc.types.traits.Trait;
import com.jpexs.decompiler.flash.abc.types.traits.TraitMethodGetterSetter;
import com.jpexs.decompiler.flash.graph.GraphTargetItem;
import com.jpexs.decompiler.flash.tags.DoABCTag;
import com.jpexs.decompiler.flash.tags.Tag;
import java.io.FileInputStream;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Stack;

/**
 *
 * Generates stub for ActionScript3Test
 * 
 * @author JPEXS
 */
public class Generator {
   public static void main(String[] args) throws Exception {
      SWF swf=new SWF(new FileInputStream("testdata/as3/TestMovie.swf"));
      DoABCTag tag = null;
      for (Tag t : swf.tags) {
         if (t instanceof DoABCTag) {
            tag = (DoABCTag) t;
            break;
         }
      }
      ABC abc=tag.abc;
      int classId=abc.findClassByName("classes.Test");
      StringBuilder s=new StringBuilder();
      for(Trait t:abc.instance_info[classId].instance_traits.traits){
         if(t instanceof TraitMethodGetterSetter){
            String name=t.getName(abc).getName(abc.constants, new ArrayList<String>());
            if(name.startsWith("test")){
               s.append("@Test\r\npublic void ");
               s.append(name);
               s.append("(){\r\ndecompileMethod(\"");
               s.append(name);
               s.append("\", ");
               String src=abc.findBody(((TraitMethodGetterSetter)t).method_info).toString("", false, false, classId, abc, abc.constants, abc.method_info, new Stack<GraphTargetItem>(), false, false, new ArrayList<String>(), abc.instance_info[classId].instance_traits);
               String srcs[]=src.split("[\r\n]+");
               for(int i=0;i<srcs.length;i++){
                  String ss=srcs[i];
                  s.append("\"");
                  s.append(ss.trim().replace("\\", "\\\\").replace("\"", "\\\""));                 
                  s.append("\\r\\n\"");
                  if(i<srcs.length-1){
                     s.append("+");
                  }
                  s.append("\r\n");
               }
               s.append(", false);");
               s.append("}");
            }
         }
         
         PrintWriter pw=new PrintWriter("src.as");
         pw.println(s.toString());
         pw.close();
      }
   }
}