com.tmtron.enums.processor.CodeGenEnumConst Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of enum-mapper-processor Show documentation

Annotation processor that builds an enum mapper which causes a compile-time error when you forget an enum

There is a newer version: 1.0.4

Show newest version

/*
 * Copyright © 2017 Martin Trummer ([email protected])
 *
 * 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.tmtron.enums.processor;

import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.ParameterizedTypeName;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeVariableName;

import javax.lang.model.element.Element;

/**
 * Simple helper class to keep info about an enumeration constant which
 * is used during code-generation via javapoet
 */
class CodeGenEnumConst {

    // e.g. ONLINE
    final String identifier;
    // e.g. "setONLINE
    final String setterName;
    final ClassName interfaceClassName;
    // e.g. "IsetONLINE
    final TypeName interfaceTypeName;

    CodeGenEnumConst(TypeVariableName typeVariableName4Value, Element enumConstant) {
        identifier = enumConstant.toString();
        setterName = "set" + identifier;
        String interfaceName = "I" + setterName;
        interfaceClassName = ClassName.get("", interfaceName);
        // StagedBuilder
        interfaceTypeName = ParameterizedTypeName.get(interfaceClassName, typeVariableName4Value);
    }
}