StripSuffixPostProcessor.java

package io.cattle.platform.schema.processor;

import io.cattle.platform.util.type.Priority;
import io.github.ibuildthecloud.gdapi.factory.SchemaFactory;
import io.github.ibuildthecloud.gdapi.factory.impl.AbstractSchemaPostProcessor;
import io.github.ibuildthecloud.gdapi.factory.impl.SchemaPostProcessor;
import io.github.ibuildthecloud.gdapi.model.impl.SchemaImpl;

import org.apache.commons.lang3.StringUtils;

public class StripSuffixPostProcessor extends AbstractSchemaPostProcessor implements SchemaPostProcessor, Priority {

    String suffix;

    @Override
    public SchemaImpl postProcessRegister(SchemaImpl schema, SchemaFactory factory) {
        if (!schema.getId().endsWith(suffix)) {
            return schema;
        }

        String newName = StringUtils.substringBeforeLast(schema.getId(), suffix);
        schema.setName(newName);

        return schema;
    }

    public String getSuffix() {
        return suffix;
    }

    public void setSuffix(String suffix) {
        this.suffix = suffix;
    }

    @Override
    public int getPriority() {
        return Priority.PRE;
    }

}