001/*
002 * Copyright (C) 2015-2021 KeepSafe Software
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *      http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package com.getkeepsafe.dexcount.treegen;
017
018import com.getkeepsafe.dexcount.DexCountException;
019import com.getkeepsafe.dexcount.treegen.workers.ApkishWorker;
020import org.gradle.api.file.ConfigurableFileCollection;
021import org.gradle.api.tasks.CacheableTask;
022import org.gradle.api.tasks.InputFiles;
023import org.gradle.api.tasks.PathSensitive;
024import org.gradle.api.tasks.PathSensitivity;
025
026import java.io.File;
027
028@CacheableTask
029public abstract class Agp41LibraryPackageTreeTask extends ModernGeneratePackageTreeTask<ApkishWorker.Params, ApkishWorker> {
030    @InputFiles
031    @PathSensitive(PathSensitivity.RELATIVE)
032    public abstract ConfigurableFileCollection getAarBundleFileCollection();
033
034    @Override
035    protected Class<ApkishWorker> getWorkerClass() {
036        return ApkishWorker.class;
037    }
038
039    @Override
040    protected void configureParams(ApkishWorker.Params params) {
041        super.configureParams(params);
042
043        File aarFile = null;
044        for (File file : getAarBundleFileCollection()) {
045            if (file.getName().endsWith(".aar")) {
046                aarFile = file;
047                break;
048            }
049        }
050
051        if (aarFile == null) {
052            throw new DexCountException("No AAR file found");
053        }
054
055        params.getApkishFile().set(aarFile);
056    }
057}