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.workers;
017
018import com.getkeepsafe.dexcount.Deobfuscator;
019import com.getkeepsafe.dexcount.PackageTree;
020import com.getkeepsafe.dexcount.source.SourceFile;
021import com.getkeepsafe.dexcount.source.SourceFiles;
022import org.gradle.api.file.RegularFileProperty;
023import org.slf4j.Logger;
024import org.slf4j.LoggerFactory;
025
026import java.io.File;
027import java.io.IOException;
028
029public abstract class JarWorker extends BaseWorker<JarWorker.Params> {
030    private static final Logger LOGGER = LoggerFactory.getLogger(JarWorker.class);
031
032    public interface Params extends BaseWorker.Params {
033        RegularFileProperty getJarFile();
034    }
035
036    @Override
037    protected PackageTree generatePackageTree() throws IOException {
038        PackageTree tree = new PackageTree(Deobfuscator.EMPTY);
039        File jarFile = getParameters().getJarFile().getAsFile().get();
040        try (SourceFile sf = SourceFiles.extractJarFromJar(jarFile)) {
041            sf.getMethodRefs().forEach(tree::addDeclaredMethodRef);
042            sf.getFieldRefs().forEach(tree::addDeclaredFieldRef);
043        }
044        return tree;
045    }
046
047    @Override
048    protected String getInputRepresentation() {
049        return getParameters().getJarFile().getAsFile().get().getName();
050    }
051
052    @Override
053    protected Logger getLogger() {
054        return LOGGER;
055    }
056}