com.android.build.gradle.internal.pipeline.TransformManager Java Examples

@Override
public void merge(TransformInvocation transformInvocation) {

    mainDexOut = getDexOutputLocation(transformInvocation.getOutputProvider(),"main", TransformManager.SCOPE_FULL_PROJECT);
    if (!mainDexOut.exists() || !mainDexOut.isDirectory()){
        return;
    }
    atomicInteger.set(org.apache.commons.io.FileUtils.listFiles(mainDexOut,new String[]{"dex"},true).size());
    for (AwbTransform awbTransform : variantOutputContext.getAwbTransformMap().values()) {
       merge(awbTransform.getAwbBundle());
    }

}
 

public void mergeAll(TransformInvocation transformInvocation) throws IOException {
    ProcessOutput output = outputHandler.createOutput();
    for (File file:allDexsArchives){
        logger.warning("mergeAll File:"+file.getAbsolutePath());
    }
    TransformOutputProvider outputProvider = transformInvocation.getOutputProvider();
    File outputDir =
            getDexOutputLocation(outputProvider, "main", TransformManager.SCOPE_FULL_PROJECT);

    File finalDexDir = new File(outputDir.getParentFile(),"temp");
    finalDexDir.mkdirs();
    List<ForkJoinTask<Void>> mergeTasks = new ArrayList();
    if (dexingType == DexingType.NATIVE_MULTIDEX) {
        mergeTasks.addAll(
                handleNativeMultiDex(
                        allDexsArchives,
                        output,
                        finalDexDir,
                         null));
    } else {
        mergeTasks.addAll(
                handleLegacyAndMonoDex(
                        allDexsArchives, output, finalDexDir, mainDexListFile == null ? null : mainDexListFile.getSingleFile()));
    }

    mergeTasks.forEach(voidForkJoinTask -> voidForkJoinTask.join());

    FileUtils.deleteDirectory(outputDir);
    FileUtils.moveDirectory(finalDexDir,outputDir);

}
 

@Override
public Set<ContentType> getOutputTypes() {
    if (appVariantContext.getAtlasExtension().getTBuildConfig().isFastProguard() && !appVariantContext.getAtlasExtension().getTBuildConfig().isKeepJavaResAfterProguard()) {
        return TransformManager.CONTENT_CLASS;
    }
    return super.getOutputTypes();
}
 

@NonNull
protected File getDexOutputLocation(
        @NonNull TransformOutputProvider outputProvider,
        @NonNull String name,
        @NonNull Set<? super QualifiedContent.Scope> scopes) {
    return outputProvider.getContentLocation(name, TransformManager.CONTENT_DEX, scopes, Format.DIRECTORY);
}
 

@Override
public void merge(TransformInvocation transformInvocation) {

    if (dexMerger == DexMergerTool.D8) {
        logger.info("D8 is used to merge dex.");
    }

    TransformOutputProvider outputProvider = transformInvocation.getOutputProvider();
    Collection<TransformInput> transformInputs = transformInvocation.getInputs();

    List<File> mainDexFiles = new ArrayList<>();

    final File[][] mergeDexs = {new File[0]};

    File outputDir =
            getDexOutputLocation(outputProvider, "main", TransformManager.SCOPE_FULL_PROJECT);
    // this deletes and creates the dir for the output
    try {
        FileUtils.cleanOutputDir(outputDir);
    } catch (IOException e) {
        e.printStackTrace();
    }

    variantOutputContext.setDexMergeFolder(outputDir);

    transformInputs.forEach((TransformInput transformInput) -> {
        File file = (File) ReflectUtils.getField(transformInput, "optionalRootLocation");
        if (file != null && file.exists()) {
            mainDexFiles.addAll(org.apache.commons.io.FileUtils.listFiles(file, new String[]{"jar", "dex"}, true));
            mergeDexs[0] = file.listFiles(pathname -> pathname.getName().endsWith(".jar") || pathname.isDirectory());
        }
    });


    merge(mainDexFiles,outputDir,mergeDexs);

}
 

@Override
public Set<QualifiedContent.Scope> getScopes() {
    return TransformManager.SCOPE_FULL_PROJECT;
}
 

public Set<QualifiedContent.Scope> getReferencedScopes() {
    return TransformManager.SCOPE_FULL_PROJECT;
}
 

@NonNull
@Override
public Set<QualifiedContent.ContentType> getInputTypes() {
    return TransformManager.DATA_BINDING_ARTIFACT;
}
 

@NonNull
@Override
public Set<QualifiedContent.Scope> getScopes() {
    return TransformManager.EMPTY_SCOPES;
}
 

@NonNull
@Override
public Set<QualifiedContent.ContentType> getInputTypes() {
    return TransformManager.CONTENT_CLASS;
}
 

@NonNull
@Override
public Set<? super QualifiedContent.Scope> getScopes() {
    return TransformManager.SCOPE_FULL_PROJECT;
}
 

@Override
public Set<QualifiedContent.ContentType> getInputTypes() {
    return TransformManager.CONTENT_CLASS;
}
 

@Override
public Set<QualifiedContent.ContentType> getInputTypes() {
    return TransformManager.CONTENT_JARS;
}
 

@Override
public Set<? super QualifiedContent.Scope> getScopes() {
    return TransformManager.SCOPE_FULL_PROJECT;
}
 

@Override
public Set<QualifiedContent.ContentType> getInputTypes() {
    return TransformManager.CONTENT_CLASS;
}
 

@Override
public Set<? super QualifiedContent.Scope> getScopes() {
    return TransformManager.SCOPE_FULL_PROJECT;
}
 

@Override
public Set<QualifiedContent.ContentType> getInputTypes() {
    return TransformManager.CONTENT_CLASS;
}
 

@Override
public Set<QualifiedContent.ContentType> getInputTypes() {
    return TransformManager.CONTENT_CLASS;
}
 

@Override
public Set<? super QualifiedContent.Scope> getScopes() {
    return TransformManager.SCOPE_FULL_PROJECT;
}
 

@Override
public Set<QualifiedContent.ScopeType> getScopes() {
    return TransformManager.SCOPE_FULL_WITH_IR_FOR_DEXING;
}
 

@Override
public Set<QualifiedContent.ContentType> getOutputTypes() {
    return TransformManager.CONTENT_JARS;
}
 

@Override
public Set<QualifiedContent.ContentType> getInputTypes() {
    return TransformManager.CONTENT_JARS;
}
 

@Override
public Set<? super QualifiedContent.Scope> getScopes() {
    return TransformManager.SCOPE_FULL_WITH_IR_FOR_DEXING;

}
 

@Override
public Set<QualifiedContent.ContentType> getOutputTypes() {
    return TransformManager.CONTENT_DEX;
}
 

@Override
public Set<QualifiedContent.Scope> getReferencedScopes() {
    return TransformManager.SCOPE_FULL_PROJECT;
}
 

@Override
public Set<QualifiedContent.Scope> getScopes() {
    return TransformManager.EMPTY_SCOPES;
}
 

private void doMainBundleProguard(TransformInvocation invocation) throws Exception {

        //apply bundle Inout
        Profiler.enter("bundleKeep");
        File bundleKeep = AtlasProguardHelper.generateBundleKeepCfg(appVariantContext);
        Profiler.release();

        Input input = new Input();
        AwbBundle awbBundle = new AwbBundle();
        awbBundle.getAndroidLibraries().addAll(AtlasBuildContext.androidDependencyTrees.get(appVariantContext.getVariantName()).getMainBundle().getAndroidLibraries());
        AwbTransform awbTransform = new AwbTransform(awbBundle);
        input.getAwbBundles().add(awbTransform);

        List<File> unProguardJars = new ArrayList<>();
        //Enter the input
        for (TransformInput transformInput : invocation.getInputs()) {
            for (JarInput jarInput : transformInput.getJarInputs()) {
                File file = jarInput.getFile();
                if (file.getName().startsWith("combined-rmerge")) {
                    unProguardJars.add(file);
                } else if (AtlasBuildContext.atlasMainDexHelperMap.get(appVariantContext.getVariantName()).inMainDex(jarInput)) {
                    awbTransform.getInputLibraries().add(file);
                }
            }
            for (DirectoryInput directoryInput : transformInput.getDirectoryInputs()) {
                if (AtlasBuildContext.atlasMainDexHelperMap.get(appVariantContext.getVariantName()).getInputDirs().contains(directoryInput.getFile())) {
                    awbTransform.getInputLibraries().add(directoryInput.getFile());
                }
            }
        }

        //inputting librarys
        input.getLibraries().addAll(
                appVariantContext.getScope().getGlobalScope().getAndroidBuilder().getBootClasspath(true));
        input.getLibraries().addAll(unProguardJars);

        //The default proguard configuration
        input.getDefaultProguardFiles().addAll(defaultProguardFiles);

        //bundle keeps
        input.getParentKeeps().add(bundleKeep);

        File outFile = invocation.getOutputProvider().getContentLocation("main", getOutputTypes(), getScopes(),
                Format.JAR);

        outFile.delete();
        input.proguardOutputDir = invocation.getOutputProvider().getContentLocation("main", getOutputTypes(), getScopes(),
                Format.DIRECTORY);
        input.printMapping = (File) ReflectUtils.getField(oldTransform, "printMapping");
        input.dump = (File) ReflectUtils.getField(oldTransform, "dump");
        input.printSeeds = (File) ReflectUtils.getField(oldTransform, "printSeeds");
        input.printUsage = (File) ReflectUtils.getField(oldTransform, "printUsage");
        input.printConfiguration = new File(appVariantContext.getProject().getBuildDir(), "outputs/proguard.cfg");

        Profiler.enter("executeproguard");
        BundleProguarder.execute(appVariantContext, input);
        transformInput(input);

        Profiler.release();

        for (File jar : unProguardJars) {

            File to = invocation.getOutputProvider().getContentLocation(FileNameUtils.getUniqueJarName(jar),
                    getOutputTypes(), getScopes(),
                    Format.JAR);
            FileUtils.copyFile(jar, to);

        }

        OriginalStream originalStream = OriginalStream.builder(appVariantContext.getProject(), "proguard-classes")
                .addContentTypes(com.android.build.gradle.internal.pipeline.TransformManager.CONTENT_CLASS)
                .addScope(QualifiedContent.Scope.PROJECT)
                .setFileCollection(appVariantContext.getProject().files(AtlasBuildContext.atlasMainDexHelperMap.get(appVariantContext.getVariantName()).getAllMainDexJars()))
                .build();

        if (nextTransformTask != null) {
            Collection consumedInputStreams = (Collection) ReflectUtils.getField(nextTransformTask, "consumedInputStreams");
            if (appVariantContext.getAtlasExtension().getTBuildConfig().isFastProguard() && consumedInputStreams != null) {
                consumedInputStreams.add(originalStream);
            }
        }

    }
 

@Override
public Set<? super QualifiedContent.Scope> getScopes() {
    return TransformManager.SCOPE_FULL_PROJECT;
}
 

@Override
public Set<QualifiedContent.ContentType> getInputTypes() {
    return TransformManager.CONTENT_CLASS;
}
 

@Override
public Set<? super QualifiedContent.Scope> getScopes() {
    return TransformManager.SCOPE_FULL_PROJECT;
}