com.greenfrvr.annyprefs.compiler.components.interfaces.RemoveInterfaceGenerator Maven / Gradle / Ivy

Go to download
package com.greenfrvr.annyprefs.compiler.components.interfaces;

import com.greenfrvr.annyprefs.compiler.DataSource;
import com.greenfrvr.annyprefs.compiler.prefs.PrefField;
import com.greenfrvr.annyprefs.compiler.utils.Utils;
import com.greenfrvr.annyprefs.compiler.utils.MethodsUtil;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;

import java.util.function.Consumer;

/**
 * Created by greenfrvr
 */
public class RemoveInterfaceGenerator extends InterfaceGenerator {

    RemoveInterfaceGenerator(DataSource dataSource) {
        super(dataSource);
    }

    public static RemoveInterfaceGenerator init(DataSource data) {
        return new RemoveInterfaceGenerator(data);
    }

    @Override
    String className() {
        return Utils.REMOVE.concat(data.name());
    }

    @Override
    TypeName superInterface() {
        return Utils.REMOVE_CLASS;
    }

    @Override
    Consumer constructMethod(TypeSpec.Builder builder) {
        return field -> builder.addMethod(MethodsUtil.removeMethodInstance(data.name(), field));
    }

}