Search in sources :

Example 1 with ParsedPackagingOptions

use of com.android.build.gradle.internal.packaging.ParsedPackagingOptions in project atlas by alibaba.

the class AtlasMergeJavaResourcesTransform method transform.

@Override
public void transform(TransformInvocation invocation) throws IOException {
    waitableExecutor.execute(new Callable<Void>() {

        @Override
        public Void call() throws Exception {
            cacheDir = new File(intermediateDir, "zip-cache");
            FileUtils.mkdirs(cacheDir);
            FileCacheByPath zipCache = new FileCacheByPath(cacheDir);
            TransformOutputProvider outputProvider = invocation.getOutputProvider();
            checkNotNull(outputProvider, "Missing output object for transform " + getName());
            ParsedPackagingOptions packagingOptions = new ParsedPackagingOptions(AtlasMergeJavaResourcesTransform.this.packagingOptions);
            boolean full = false;
            IncrementalFileMergerState state = loadMergeState();
            if (state == null || !invocation.isIncremental()) {
                /*
                     * This is a full build.
                     */
                state = new IncrementalFileMergerState();
                outputProvider.deleteAll();
                full = true;
            }
            List<Runnable> cacheUpdates = new ArrayList<>();
            Map<IncrementalFileMergerInput, QualifiedContent> contentMap = new HashMap<>();
            List<IncrementalFileMergerInput> inputs = new ArrayList<>(AtlasIncrementalFileMergeTransformUtils.toInput(invocation, zipCache, cacheUpdates, full, contentMap, null, appVariantOutputContext.getVariantContext().getVariantName()));
            /*
                 * In an ideal world, we could just send the inputs to the file merger. However, in the
                 * real world we live in, things are more complicated :)
                 *
                 * We need to:
                 *
                 * 1. We need to bring inputs that refer to the project scope before the other inputs.
                 * 2. Prefix libraries that come from directories with "lib/".
                 * 3. Filter all inputs to remove anything not accepted by acceptedPathsPredicate neither
                 * by packagingOptions.
                 */
            // Sort inputs to move project scopes to the start.
            inputs.sort((i0, i1) -> {
                int v0 = contentMap.get(i0).getScopes().contains(QualifiedContent.Scope.PROJECT) ? 0 : 1;
                int v1 = contentMap.get(i1).getScopes().contains(QualifiedContent.Scope.PROJECT) ? 0 : 1;
                return v0 - v1;
            });
            // Prefix libraries with "lib/" if we're doing libraries.
            assert mergedType.size() == 1;
            QualifiedContent.ContentType mergedType = AtlasMergeJavaResourcesTransform.this.mergedType.iterator().next();
            if (mergedType == ExtendedContentType.NATIVE_LIBS) {
                inputs = inputs.stream().map(i -> {
                    QualifiedContent qc = contentMap.get(i);
                    if (qc.getFile().isDirectory()) {
                        i = new RenameIncrementalFileMergerInput(i, s -> "lib/" + s, s -> s.substring("lib/".length()));
                        contentMap.put(i, qc);
                    }
                    return i;
                }).collect(Collectors.toList());
            }
            // Filter inputs.
            Predicate<String> inputFilter = acceptedPathsPredicate.and(path -> packagingOptions.getAction(path) != PackagingFileAction.EXCLUDE);
            inputs = inputs.stream().map(i -> {
                IncrementalFileMergerInput i2 = new FilterIncrementalFileMergerInput(i, inputFilter);
                contentMap.put(i2, contentMap.get(i));
                return i2;
            }).collect(Collectors.toList());
            /*
                 * Create the algorithm used by the merge transform. This algorithm decides on which
                 * algorithm to delegate to depending on the packaging option of the path. By default it
                 * requires just one file (no merging).
                 */
            StreamMergeAlgorithm mergeTransformAlgorithm = StreamMergeAlgorithms.select(path -> {
                PackagingFileAction packagingAction = packagingOptions.getAction(path);
                switch(packagingAction) {
                    case EXCLUDE:
                        // Should have been excluded from the input.
                        throw new AssertionError();
                    case PICK_FIRST:
                        return StreamMergeAlgorithms.pickFirst();
                    case MERGE:
                        return StreamMergeAlgorithms.concat();
                    case NONE:
                        return StreamMergeAlgorithms.acceptOnlyOne();
                    default:
                        throw new AssertionError();
                }
            });
            /*
                 * Create an output that uses the algorithm. This is not the final output because,
                 * unfortunately, we still have the complexity of the project scope overriding other scopes
                 * to solve.
                 *
                 * When resources inside a jar file are extracted to a directory, the results may not be
                 * expected on Windows if the file names end with "." (bug 65337573), or if there is an
                 * uppercase/lowercase conflict. To work around this issue, we copy these resources to a
                 * jar file.
                 */
            IncrementalFileMergerOutput baseOutput;
            if (mergedType == QualifiedContent.DefaultContentType.RESOURCES) {
                outputLocation = outputProvider.getContentLocation("resources", getOutputTypes(), getScopes(), Format.JAR);
                baseOutput = IncrementalFileMergerOutputs.fromAlgorithmAndWriter(mergeTransformAlgorithm, MergeOutputWriters.toZip(outputLocation));
                AtlasBuildContext.atlasMainDexHelperMap.get(appVariantOutputContext.getVariantContext().getVariantName()).addMainJavaRes(outputLocation);
            } else {
                outputLocation = outputProvider.getContentLocation("resources", getOutputTypes(), getScopes(), Format.DIRECTORY);
                baseOutput = IncrementalFileMergerOutputs.fromAlgorithmAndWriter(mergeTransformAlgorithm, MergeOutputWriters.toDirectory(outputLocation));
            }
            /*
                 * We need a custom output to handle the case in which the same path appears in multiple
                 * inputs and the action is NONE, but only one input is actually PROJECT. In this specific
                 * case we will ignore all other inputs.
                 */
            Set<IncrementalFileMergerInput> projectInputs = contentMap.keySet().stream().filter(i -> contentMap.get(i).getScopes().contains(QualifiedContent.Scope.PROJECT)).collect(Collectors.toSet());
            IncrementalFileMergerOutput output = new DelegateIncrementalFileMergerOutput(baseOutput) {

                @Override
                public void create(@NonNull String path, @NonNull List<IncrementalFileMergerInput> inputs) {
                    super.create(path, filter(path, inputs));
                }

                @Override
                public void update(@NonNull String path, @NonNull List<String> prevInputNames, @NonNull List<IncrementalFileMergerInput> inputs) {
                    super.update(path, prevInputNames, filter(path, inputs));
                }

                @Override
                public void remove(@NonNull String path) {
                    super.remove(path);
                }

                @NonNull
                private ImmutableList<IncrementalFileMergerInput> filter(@NonNull String path, @NonNull List<IncrementalFileMergerInput> inputs) {
                    PackagingFileAction packagingAction = packagingOptions.getAction(path);
                    if (packagingAction == PackagingFileAction.NONE && inputs.stream().anyMatch(projectInputs::contains)) {
                        inputs = inputs.stream().filter(projectInputs::contains).collect(ImmutableCollectors.toImmutableList());
                    }
                    return ImmutableList.copyOf(inputs);
                }
            };
            state = IncrementalFileMerger.merge(ImmutableList.copyOf(inputs), output, state);
            saveMergeState(state);
            cacheUpdates.forEach(Runnable::run);
            return null;
        }
    });
    for (AwbTransform awbTransform : appVariantOutputContext.getAwbTransformMap().values()) {
        File awbCacheDir = new File(intermediateDir, "awb-zip-cache" + File.separator + awbTransform.getAwbBundle().getName());
        waitableExecutor.execute(new Callable<Void>() {

            @Override
            public Void call() throws Exception {
                FileUtils.mkdirs(awbCacheDir);
                FileCacheByPath zipCache = new FileCacheByPath(awbCacheDir);
                ParsedPackagingOptions packagingOptions = new ParsedPackagingOptions(AtlasMergeJavaResourcesTransform.this.packagingOptions);
                boolean full = false;
                IncrementalFileMergerState state = loadAwbMergeState(awbTransform.getAwbBundle());
                if (state == null || !invocation.isIncremental()) {
                    /*
                         * This is a full build.
                         */
                    state = new IncrementalFileMergerState();
                    if (appVariantOutputContext.getAwbJniFolder(awbTransform.getAwbBundle()).exists() && mergedType.contains(ExtendedContentType.NATIVE_LIBS)) {
                        FileUtils.deleteDirectoryContents(appVariantOutputContext.getAwbJniFolder(awbTransform.getAwbBundle()));
                    }
                    if (appVariantOutputContext.getAwbJavaResFolder(awbTransform.getAwbBundle()).exists() && mergedType.contains(QualifiedContent.DefaultContentType.RESOURCES)) {
                        FileUtils.deleteDirectoryContents(appVariantOutputContext.getAwbJavaResFolder(awbTransform.getAwbBundle()));
                    }
                    full = true;
                }
                List<Runnable> cacheUpdates = new ArrayList<>();
                Map<IncrementalFileMergerInput, QualifiedContent> contentMap = new HashMap<>();
                List<IncrementalFileMergerInput> inputs = new ArrayList<>(AtlasIncrementalFileMergeTransformUtils.toInput(invocation, zipCache, cacheUpdates, full, contentMap, awbTransform, appVariantOutputContext.getVariantContext().getVariantName()));
                /*
                     * In an ideal world, we could just send the inputs to the file merger. However, in the
                     * real world we live in, things are more complicated :)
                     *
                     * We need to:
                     *
                     * 1. We need to bring inputs that refer to the project scope before the other inputs.
                     * 2. Prefix libraries that come from directories with "lib/".
                     * 3. Filter all inputs to remove anything not accepted by acceptedPathsPredicate neither
                     * by packagingOptions.
                     */
                // Sort inputs to move project scopes to the start.
                inputs.sort((i0, i1) -> {
                    int v0 = contentMap.get(i0).getScopes().contains(QualifiedContent.Scope.PROJECT) ? 0 : 1;
                    int v1 = contentMap.get(i1).getScopes().contains(QualifiedContent.Scope.PROJECT) ? 0 : 1;
                    return v0 - v1;
                });
                // Prefix libraries with "lib/" if we're doing libraries.
                assert mergedType.size() == 1;
                QualifiedContent.ContentType mergedType = AtlasMergeJavaResourcesTransform.this.mergedType.iterator().next();
                if (mergedType == ExtendedContentType.NATIVE_LIBS) {
                    inputs = inputs.stream().map(i -> {
                        QualifiedContent qc = contentMap.get(i);
                        if (qc.getFile().isDirectory()) {
                            i = new RenameIncrementalFileMergerInput(i, s -> "lib/" + s, s -> s.substring("lib/".length()));
                            contentMap.put(i, qc);
                        }
                        return i;
                    }).collect(Collectors.toList());
                }
                // Filter inputs.
                Predicate<String> inputFilter = acceptedPathsPredicate.and(path -> packagingOptions.getAction(path) != PackagingFileAction.EXCLUDE);
                inputs = inputs.stream().map(i -> {
                    IncrementalFileMergerInput i2 = new FilterIncrementalFileMergerInput(i, inputFilter);
                    contentMap.put(i2, contentMap.get(i));
                    return i2;
                }).collect(Collectors.toList());
                /*
                     * Create the algorithm used by the merge transform. This algorithm decides on which
                     * algorithm to delegate to depending on the packaging option of the path. By default it
                     * requires just one file (no merging).
                     */
                StreamMergeAlgorithm mergeTransformAlgorithm = StreamMergeAlgorithms.select(path -> {
                    PackagingFileAction packagingAction = packagingOptions.getAction(path);
                    switch(packagingAction) {
                        case EXCLUDE:
                            // Should have been excluded from the input.
                            throw new AssertionError();
                        case PICK_FIRST:
                            return StreamMergeAlgorithms.pickFirst();
                        case MERGE:
                            return StreamMergeAlgorithms.concat();
                        case NONE:
                            return StreamMergeAlgorithms.acceptOnlyOne();
                        default:
                            throw new AssertionError();
                    }
                });
                /*
                     * Create an output that uses the algorithm. This is not the final output because,
                     * unfortunately, we still have the complexity of the project scope overriding other scopes
                     * to solve.
                     *
                     * When resources inside a jar file are extracted to a directory, the results may not be
                     * expected on Windows if the file names end with "." (bug 65337573), or if there is an
                     * uppercase/lowercase conflict. To work around this issue, we copy these resources to a
                     * jar file.
                     */
                IncrementalFileMergerOutput baseOutput;
                if (mergedType == QualifiedContent.DefaultContentType.RESOURCES) {
                    File outputLocation = new File(appVariantOutputContext.getAwbJavaResFolder(awbTransform.getAwbBundle()), "res.jar");
                    if (!appVariantOutputContext.getAwbJavaResFolder(awbTransform.getAwbBundle()).exists()) {
                        appVariantOutputContext.getAwbJavaResFolder(awbTransform.getAwbBundle()).mkdirs();
                    }
                    createEmptyZipFile(outputLocation);
                    baseOutput = IncrementalFileMergerOutputs.fromAlgorithmAndWriter(mergeTransformAlgorithm, MergeOutputWriters.toZip(outputLocation));
                } else {
                    File outputLocation = appVariantOutputContext.getAwbJniFolder(awbTransform.getAwbBundle());
                    baseOutput = IncrementalFileMergerOutputs.fromAlgorithmAndWriter(mergeTransformAlgorithm, MergeOutputWriters.toDirectory(outputLocation));
                }
                /*
                     * We need a custom output to handle the case in which the same path appears in multiple
                     * inputs and the action is NONE, but only one input is actually PROJECT. In this specific
                     * case we will ignore all other inputs.
                     */
                Set<IncrementalFileMergerInput> projectInputs = contentMap.keySet().stream().filter(i -> contentMap.get(i).getScopes().contains(QualifiedContent.Scope.PROJECT)).collect(Collectors.toSet());
                IncrementalFileMergerOutput output = new DelegateIncrementalFileMergerOutput(baseOutput) {

                    @Override
                    public void create(@NonNull String path, @NonNull List<IncrementalFileMergerInput> inputs) {
                        super.create(path, filter(path, inputs));
                    }

                    @Override
                    public void update(@NonNull String path, @NonNull List<String> prevInputNames, @NonNull List<IncrementalFileMergerInput> inputs) {
                        super.update(path, prevInputNames, filter(path, inputs));
                    }

                    @Override
                    public void remove(@NonNull String path) {
                        super.remove(path);
                    }

                    @NonNull
                    private ImmutableList<IncrementalFileMergerInput> filter(@NonNull String path, @NonNull List<IncrementalFileMergerInput> inputs) {
                        PackagingFileAction packagingAction = packagingOptions.getAction(path);
                        if (packagingAction == PackagingFileAction.NONE && inputs.stream().anyMatch(projectInputs::contains)) {
                            inputs = inputs.stream().filter(projectInputs::contains).collect(ImmutableCollectors.toImmutableList());
                        }
                        return ImmutableList.copyOf(inputs);
                    }
                };
                state = IncrementalFileMerger.merge(ImmutableList.copyOf(inputs), output, state);
                saveAwbMergeState(state, awbTransform.getAwbBundle());
                cacheUpdates.forEach(Runnable::run);
                return null;
            }
        });
    }
    try {
        waitableExecutor.waitForTasksWithQuickFail(false);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    appVariantOutputContext.getAwbTransformMap().values().stream().forEach(awbTransform -> {
        if (awbTransform.getAwbBundle().isMBundle) {
            if (mergedType.contains(ExtendedContentType.NATIVE_LIBS)) {
                File bundleOutputLocation = appVariantOutputContext.getAwbJniFolder(awbTransform.getAwbBundle());
                if (bundleOutputLocation.exists()) {
                    try {
                        org.apache.commons.io.FileUtils.copyDirectory(bundleOutputLocation, outputLocation);
                        org.apache.commons.io.FileUtils.deleteDirectory(bundleOutputLocation);
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            } else {
                File bundleOutputLocation = new File(appVariantOutputContext.getAwbJavaResFolder(awbTransform.getAwbBundle()), "res.jar");
                File tempDir = new File(outputLocation.getParentFile(), "unzip");
                try {
                    if (bundleOutputLocation.exists() && ZipUtils.isZipFile(bundleOutputLocation)) {
                        BetterZip.unzipDirectory(bundleOutputLocation, tempDir);
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    });
    if (!mergedType.contains(ExtendedContentType.NATIVE_LIBS)) {
        File tempDir = new File(outputLocation.getParentFile(), "unzip");
        if (outputLocation != null && outputLocation.exists() && ZipUtils.isZipFile(outputLocation)) {
            BetterZip.unzipDirectory(outputLocation, tempDir);
        }
        if (tempDir.exists() && tempDir.listFiles() != null) {
            FileUtils.deleteIfExists(outputLocation);
            BetterZip.zipDirectory(tempDir, outputLocation);
        }
    }
    paths.parallelStream().forEach(s -> processAtlasNativeSo(s));
}
Also used : ZipOutputStream(java.util.zip.ZipOutputStream) com.android.builder.merge(com.android.builder.merge) java.util(java.util) SdkConstants(com.android.SdkConstants) BetterZip(com.taobao.android.builder.tools.zip.BetterZip) PackagingOptions(com.android.build.gradle.internal.dsl.PackagingOptions) Callable(java.util.concurrent.Callable) VariantScope(com.android.build.gradle.internal.scope.VariantScope) Matcher(java.util.regex.Matcher) ExtendedContentType(com.android.build.gradle.internal.pipeline.ExtendedContentType) ImmutableCollectors(com.android.utils.ImmutableCollectors) ImmutableList(com.google.common.collect.ImmutableList) FileUtils(com.android.utils.FileUtils) AtlasBuildContext(com.taobao.android.builder.AtlasBuildContext) MergeJavaResourcesTransform(com.android.build.gradle.internal.transforms.MergeJavaResourcesTransform) AwbBundle(com.taobao.android.builder.dependency.model.AwbBundle) FileCacheByPath(com.android.builder.files.FileCacheByPath) PackagingFileAction(com.android.build.gradle.internal.packaging.PackagingFileAction) ImmutableSet(com.google.common.collect.ImmutableSet) Files(java.nio.file.Files) Nullable(com.android.annotations.Nullable) WaitableExecutor(com.android.ide.common.internal.WaitableExecutor) Predicate(java.util.function.Predicate) ZipUtils(com.taobao.android.builder.tools.zip.ZipUtils) Preconditions.checkNotNull(com.google.common.base.Preconditions.checkNotNull) Collectors(java.util.stream.Collectors) ParsedPackagingOptions(com.android.build.gradle.internal.packaging.ParsedPackagingOptions) ZipUtil(org.gradle.internal.impldep.org.apache.tools.zip.ZipUtil) Consumer(java.util.function.Consumer) AppVariantOutputContext(com.android.build.gradle.internal.api.AppVariantOutputContext) java.io(java.io) NonNull(com.android.annotations.NonNull) MD5Util(com.taobao.android.builder.tools.MD5Util) com.android.build.api.transform(com.android.build.api.transform) Pattern(java.util.regex.Pattern) AwbTransform(com.android.build.gradle.internal.api.AwbTransform) AtlasIncrementalFileMergeTransformUtils(com.android.build.gradle.internal.pipeline.AtlasIncrementalFileMergeTransformUtils) ImmutableSet(com.google.common.collect.ImmutableSet) ExtendedContentType(com.android.build.gradle.internal.pipeline.ExtendedContentType) ImmutableList(com.google.common.collect.ImmutableList) Predicate(java.util.function.Predicate) NonNull(com.android.annotations.NonNull) ImmutableList(com.google.common.collect.ImmutableList) ParsedPackagingOptions(com.android.build.gradle.internal.packaging.ParsedPackagingOptions) PackagingFileAction(com.android.build.gradle.internal.packaging.PackagingFileAction) FileCacheByPath(com.android.builder.files.FileCacheByPath) AwbTransform(com.android.build.gradle.internal.api.AwbTransform)

Aggregations

SdkConstants (com.android.SdkConstants)1 NonNull (com.android.annotations.NonNull)1 Nullable (com.android.annotations.Nullable)1 com.android.build.api.transform (com.android.build.api.transform)1 AppVariantOutputContext (com.android.build.gradle.internal.api.AppVariantOutputContext)1 AwbTransform (com.android.build.gradle.internal.api.AwbTransform)1 PackagingOptions (com.android.build.gradle.internal.dsl.PackagingOptions)1 PackagingFileAction (com.android.build.gradle.internal.packaging.PackagingFileAction)1 ParsedPackagingOptions (com.android.build.gradle.internal.packaging.ParsedPackagingOptions)1 AtlasIncrementalFileMergeTransformUtils (com.android.build.gradle.internal.pipeline.AtlasIncrementalFileMergeTransformUtils)1 ExtendedContentType (com.android.build.gradle.internal.pipeline.ExtendedContentType)1 VariantScope (com.android.build.gradle.internal.scope.VariantScope)1 MergeJavaResourcesTransform (com.android.build.gradle.internal.transforms.MergeJavaResourcesTransform)1 FileCacheByPath (com.android.builder.files.FileCacheByPath)1 com.android.builder.merge (com.android.builder.merge)1 WaitableExecutor (com.android.ide.common.internal.WaitableExecutor)1 FileUtils (com.android.utils.FileUtils)1 ImmutableCollectors (com.android.utils.ImmutableCollectors)1 Preconditions.checkNotNull (com.google.common.base.Preconditions.checkNotNull)1 ImmutableList (com.google.common.collect.ImmutableList)1