tech.guyi.ipojo.compile.lib.configuration.Compile Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of compile-lib Show documentation
The newest version!
package tech.guyi.ipojo.compile.lib.configuration;

import com.google.gson.annotations.Expose;
import com.google.gson.annotations.SerializedName;
import javassist.CtClass;
import javassist.CtField;
import javassist.NotFoundException;
import lombok.Data;
import org.apache.commons.io.IOUtils;
import tech.guyi.ipojo.compile.lib.configuration.entry.CompileExclude;
import tech.guyi.ipojo.compile.lib.configuration.entry.Project;
import tech.guyi.ipojo.compile.lib.cons.AnnotationNames;
import tech.guyi.ipojo.compile.lib.enums.CompileType;
import tech.guyi.ipojo.compile.lib.enums.JdkVersion;
import tech.guyi.ipojo.compile.lib.expand.compile.defaults.configuration.entry.ConfigurationKeyEntry;
import tech.guyi.ipojo.compile.lib.utils.AnnotationUtils;
import tech.guyi.ipojo.compile.lib.utils.JavassistUtils;
import tech.guyi.ipojo.compile.lib.compile.entry.CompileClass;

import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URL;
import java.util.*;
import java.util.stream.Collectors;

@Data
public class Compile {

    @Expose
    @SerializedName("extends")
    private Set extendsName;

    @Expose(serialize = false)
    private String activator;

    @Expose
    private JdkVersion jdk;

    @Expose
    private String name;

    @Expose
    private String symbolicName;

    @Expose
    private CompileType type;

    @Expose
    @SerializedName("package")
    private String packageName;

    private Set modules = new HashSet<>();

    @Expose
    @SerializedName("manifest")
    private Map manifestTemplate = Collections.emptyMap();

    @Expose(serialize = false)
    private Project project = new Project();

    @Expose
    private CompileExclude exclude = new CompileExclude();

    @Expose
    private Map configuration = Collections.emptyMap();

    @Expose
    private Map env = Collections.emptyMap();

    private Set useComponents = new HashSet<>();
    private Set configurationKeys = new HashSet<>();

    public void addUseComponent(CtClass classes){
        this.getUseComponents().add(classes);
    }

    public void addUseComponent(CtField field){
        try {
            this.addUseComponent(field.getType());
        } catch (NotFoundException e) {
            e.printStackTrace();
        }
    }

    public String getSymbolicName(){
        return Optional.ofNullable(this.symbolicName).orElse(this.name);
    }

    /**
     * 设置JDK版本
     * @param url Class文件URL
     * @throws IOException
     */
    public void formatJavaVersion(URL url) throws IOException {
        if (this.jdk != null){
            byte[] arr = IOUtils.toByteArray(url.openStream());
            arr[7] = (byte) this.jdk.getTarget();
            OutputStream out = new FileOutputStream(url.getFile());
            out.write(arr);
            out.flush();
            out.close();
        }
    }

    public Set filterUseComponents(Set components){
        return components
                .stream()
                .filter(component ->
                        AnnotationUtils.getAnnotation(component.getClasses(), AnnotationNames.DynamicInject)
                                .map(annotation -> this.useComponents.stream().anyMatch(use ->
                                        use.getName().equals(component.getClasses().getName())
                                                || JavassistUtils.equalsType(component.getClasses(),use)))
                                .orElse(true)
                )
                .collect(Collectors.toSet());
    }

}