deploy/deployer/src/main/java/com/android/tools/deployer/CachedDexSplitter.java

/*
 * Copyright (C) 2018 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.android.tools.deployer;

import com.android.tools.deployer.model.ApkEntry;
import com.android.tools.deployer.model.DexClass;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Collection;
import java.util.function.Predicate;

public class CachedDexSplitter implements DexSplitter {

    private final SqlApkFileDatabase db;
    private final DexSplitter splitter;

    public CachedDexSplitter(SqlApkFileDatabase db, DexSplitter splitter) {
        this.db = db;
        this.splitter = splitter;
    }

    @Override
    public Collection<DexClass> split(ApkEntry dex, Predicate<DexClass> keepCode)
            throws DeployerException {
        // Try a cached version
        Collection<DexClass> classes = db.getClasses(dex);
        if (classes.isEmpty() || keepCode != null) {
            // TODO: Calling CacheDexSplitter.split() on a remote APK is fine but actually calling the real splitter on an remote
            // APK is going to crash. Checking path == null isn't going to work as that gives the remote path of the APK.
            // The File.exists() check is a temp fix. See b/135202430.
            if (dex.getApk().path == null || Files.notExists(Paths.get(dex.getApk().path))) {
                throw DeployerException.remoteApkNotFound();
            }
            classes = splitter.split(dex, keepCode);
            db.addClasses(classes);
        }
        return classes;
    }
}