com.malinskiy.sheldon.codegen.model.Deleter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of compiler Show documentation

Reactive preferences abstraction

There is a newer version: 1.0.1

package com.malinskiy.sheldon.codegen.model;

import com.google.common.base.Preconditions;

import com.malinskiy.sheldon.codegen.ProcessingException;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.TypeSpec;

import javax.annotation.Nonnull;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.Modifier;

public class Deleter implements Generatable {

    @Nonnull private final ExecutableElement method;
    @Nonnull private final String providerFieldName;

    public Deleter(@Nonnull ExecutableElement method,
                   @Nonnull String providerFieldName) {

        this.method = Preconditions.checkNotNull(method);
        this.providerFieldName = Preconditions.checkNotNull(providerFieldName);
    }

    @Override
    public void generateCode(TypeSpec.Builder builder) throws ProcessingException {
        builder.addMethod(
                MethodSpec.methodBuilder(method.getSimpleName().toString())
                          .addModifiers(Modifier.PUBLIC)
                          .addParameter(String.class, "key")
                          .addStatement(providerFieldName + ".remove($N)", "key")
                          .build());
    }
}