net.minecraftforge.gradle.user.ReobfTaskFactory Maven / Gradle / Ivy

Go to download
/*
 * A Gradle plugin for the creation of Minecraft mods and MinecraftForge plugins.
 * Copyright (C) 2013-2019 Minecraft Forge
 * Copyright (C) 2020-2021 anatawa12 and other contributors
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301
 * USA
 */
package net.minecraftforge.gradle.user;

import java.io.File;
import java.util.List;

import net.minecraftforge.gradle.ArchiveTaskHelper;
import org.gradle.api.Action;
import org.gradle.api.NamedDomainObjectFactory;
import org.gradle.api.Project;
import org.gradle.api.Task;
import org.gradle.api.file.FileCollection;
import org.gradle.api.tasks.bundling.Jar;

import com.google.common.base.Preconditions;
import com.google.common.collect.Lists;

import groovy.lang.Closure;
import net.minecraftforge.gradle.common.Constants;
import net.minecraftforge.gradle.util.GradleConfigurationException;

public class ReobfTaskFactory implements NamedDomainObjectFactory
{
    private final UserBasePlugin plugin;

    public ReobfTaskFactory(UserBasePlugin plugin)
    {
        this.plugin = plugin;
    }

    @SuppressWarnings("serial")
    @Override
    public IReobfuscator create(final String jarName)
    {
        String name = Character.toUpperCase(jarName.charAt(0)) + jarName.substring(1);
        final TaskSingleReobf task = plugin.maybeMakeTask("reobf" + name, TaskSingleReobf.class);

        task.dependsOn(Constants.TASK_GENERATE_SRGS, jarName);
        task.mustRunAfter("test");

        Closure outputJar = new Closure(ReobfTaskFactory.class) {
            public File call()
            {
                return ArchiveTaskHelper.getArchivePath((Jar) plugin.project.getTasks().getByName(jarName));
            }
        };

        task.setJar(outputJar);

        final TaskExtractAnnotations annos = plugin.maybeMakeTask("extractAnnotations" + name, TaskExtractAnnotations.class);
        annos.setJar(outputJar);
        task.finalizedBy(annos);


        plugin.project.getTasks().getByName("assemble").dependsOn(task);

        ReobfTaskWrapper wrapper = new ReobfTaskWrapper(jarName, task);
        plugin.setupReobf(wrapper);

        // do after-Evaluate resolution, for the same of good error reporting
        plugin.project.afterEvaluate(new Action() {
            @Override
            public void execute(Project arg0)
            {
                Task jar = plugin.project.getTasks().getByName(jarName);
                if (!(jar instanceof Jar))
                {
                    throw new GradleConfigurationException(jarName + "  is not a jar task. Can only reobf jars!");
                }
            }
        });

        return wrapper;
    }

    public class ReobfTaskWrapper implements IReobfuscator
    {
        private final String name;
        private final TaskSingleReobf reobf;
        private ReobfMappingType mappingType;

        public ReobfTaskWrapper(String name, TaskSingleReobf reobf)
        {
            this.name = name;
            this.reobf = reobf;
        }

        public String getName()
        {
            return name;
        }

        /**
         * Returns the instance of {@link TaskSingleReobf} that this object
         * wraps.
         *
         * @return The task
         */
        public TaskSingleReobf getTask()
        {
            return reobf;
        }

        @Override
        public boolean equals(Object obj)
        {
            if (obj instanceof ReobfTaskWrapper)
            {
                return name.equals(((ReobfTaskWrapper) obj).name);
            }
            return false;
        }

        @Override
        public File getMappings()
        {
            return reobf.getPrimarySrg();
        }

        @Override
        public void setMappings(Object srg)
        {
            mappingType = ReobfMappingType.CUSTOM;
            reobf.setPrimarySrg(srg);
        }

        @Override
        public void setMappingType(ReobfMappingType type)
        {
            Preconditions.checkNotNull(type, "Mapping type cannot be null. Use setMappings() to use custom mappings.");
            Preconditions.checkArgument(type != ReobfMappingType.CUSTOM,
                    "Cannot set a Custom mapping type. Use setMappings() instead.");
            mappingType = type;
            reobf.setPrimarySrg(plugin.delayedFile(type.getPath()));
        }

        @Override
        public ReobfMappingType getMappingType()
        {
            return this.mappingType;
        }

        @Override
        public void setClasspath(FileCollection classpath)
        {
            reobf.setClasspath(classpath);
        }

        @Override
        public FileCollection getClasspath()
        {
            return reobf.getClasspath();
        }

        @Override
        public void setExtraLines(List