dev.sixpack.generator.GeneratorRecordBuilder Maven / Gradle / Ivy

Go to download

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

SDK to develop generators part of the Sixpack solution

The newest version!

package dev.sixpack.generator;

import dev.sixpack.api.data.Context;
import dev.sixpack.api.data.Template;
import dev.sixpack.api.exception.SixpackFatalException;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.List;

import static dev.sixpack.utils.FormatUtils.mFormat;

public class GeneratorRecordBuilder {
    private final String environment;
    private final String supplier;
    final private Generator generator;

    public GeneratorRecordBuilder(String environment,
                                  String supplier,
                                  Generator generator) {
        this.environment = environment;
        this.supplier = supplier;
        this.generator = generator;
    }

    public FactoryRecord build() {
        try {
            return new FactoryRecord(
                    environment,
                    supplier,
                    itemNameAnnotation(),
                    inputType(),
                    null, // TODO: implement
                    null, // TODO: implement
                    outputType(),
                    generator,
                    null,
                    templates());
        } catch (IllegalAccessException | InvocationTargetException e) {
            throw new RuntimeException("Error calling method to construct generator's manifest", e);
        }
    }

    private String itemNameAnnotation() {
        ItemName annotation = generator.getClass().getAnnotation(ItemName.class);
        if (annotation == null) {
            throw new IllegalArgumentException(mFormat(
                    "The generator class {} is not annotated with @ItemName. This annotation is used to uniquely identify item within the scope of this supplier on the given test environment. Please carefully chose the value and add the annotation @ItemName.",
                    generator.getClass().getSimpleName()));
        }
        return annotation.value();
    }

    private Class inputType() {
        return getMethodGenerate().getParameterTypes()[0];
    }

    private Class outputType() {
        return getMethodGenerate().getReturnType();
    }

    private List