org.unlaxer.compiler.MemoryJavaFileManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tinyExpression Show documentation
TinyExpression implemented with Unlaxer
There is a newer version: 1.4.4
package org.unlaxer.compiler;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.tools.FileObject;
import javax.tools.ForwardingJavaFileManager;
import javax.tools.JavaFileManager;
import javax.tools.JavaFileObject;
import javax.tools.JavaFileObject.Kind;

public class MemoryJavaFileManager extends ForwardingJavaFileManager {
  private final Map outputFileByClassName = new HashMap<>();
  private final Map inputFileByClassName = new HashMap<>();
  
  private final JavaFileManagerContext javaFileManagerContext;

  public MemoryJavaFileManager(JavaFileManager fileManager , JavaFileManagerContext javaFileManagerContext) {
    super(fileManager);
    this.javaFileManagerContext = javaFileManagerContext;
  }

  @Override
  public JavaFileObject getJavaFileForOutput(Location location, String className, JavaFileObject.Kind kind,
      FileObject sibling) throws IOException {
    ByteArrayJavaFileObject fileObject = new ByteArrayJavaFileObject(className, kind);
    outputFileByClassName.put(className, fileObject);
    return fileObject;
  }

  @Override
  public JavaFileObject getJavaFileForInput(Location location, String className, Kind kind) throws IOException {
    if(kind == Kind.CLASS) {
      MemoryJavaFileObjectForClass memoryJavaFileObjectForClass = inputFileByClassName.get(className);
      if(memoryJavaFileObjectForClass != null) {
        return memoryJavaFileObjectForClass;
      }
    }
    return super.getJavaFileForInput(location, className, kind);
  }

  public Map getBytesByName() {
    Map classBytes = new HashMap<>();
    for (Map.Entry entry : outputFileByClassName.entrySet()) {
      classBytes.put(entry.getKey(), entry.getValue().getBytes());
    }
    return classBytes;
  }
  
  public void setJavaFileOBjectForClass(String className,MemoryJavaFileObjectForClass javaFileObject) {
    inputFileByClassName.put(className , javaFileObject);
  }

  @Override
  public Iterable list(Location location, String packageName, Set kinds, boolean recurse)
      throws IOException {
//    System.out.println("☆彡☆彡☆彡---- "+packageName+" ----☆彡☆彡☆彡"+location.getName());
    
    if(false == javaFileManagerContext.matchForOtherFileManager.test(location)) {
      return List.of();
    }
    
    List result = new ArrayList();
    
    inputFileByClassName.forEach((fullClassName,file)->{
      ClassName className = new ClassName(fullClassName);
      if(packageName.equals(className.packageName())){
        result.add(file);
      }
    });
    super.list(location, packageName, kinds, recurse)
      .forEach(result::add);
    
//    if(result.size()>0) {
//      System.out.println(result.size());
//    }
    
//    result.forEach(c->{
//      if(c.toUri().toASCIIString().contains("jdk")) {
//        
//        System.out.println("packageName:"+packageName);
//      }
//    });
//    System.out.println("◇◇◇◇◇◇;" + result.size() +"/"+ inputFileByClassName.size());
    return result;
  }
}