org.bitbucket.bradleysmithllc.etlunit.maven.util.JavaProperty Maven / Gradle / Ivy

Go to download
package org.bitbucket.bradleysmithllc.etlunit.maven.util;

/*
 * #%L
 * etlunit-feature-compiler Maven Mojo
 * %%
 * Copyright (C) 2010 - 2014 bradleysmithllc
 * %%
 * 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.
 * #L%
 */

import com.sun.codemodel.*;
import org.bitbucket.bradleysmithllc.json.validator.JsonSchemaObjectNode;

import java.util.List;
import java.util.Map;

public class JavaProperty
{
	private final String name;

	private final List enumValues;

	private final Class dataTypeClass;
	private final boolean required;

	public JavaProperty(String name, JsonSchemaObjectNode.valid_type type, List enumValues, boolean required)
	{
		this.name = name;
		this.required = required;

		this.enumValues = enumValues;

		if (enumValues == null)
		{
			switch (type)
			{
				case t_string:
					dataTypeClass = String.class;
					break;
				case t_number:
					dataTypeClass = String.class;
					break;
				case t_integer:
					dataTypeClass = int.class;
					break;
				case t_boolean:
					dataTypeClass = boolean.class;
					break;
				case t_object:
					dataTypeClass = Map.class;
					break;
				case t_array:
					dataTypeClass = List.class;
					break;
				case t_any:
					dataTypeClass = Object.class;
					break;
				case t_null:
					throw new IllegalArgumentException("Data type null not supported");
				default:
					throw new IllegalArgumentException("Data type not supported");
			}
		}
		else
		{
			this.dataTypeClass = null;
		}
	}

	public void append(JDefinedClass buffer) throws JClassAlreadyExistsException
	{
		JFieldVar field = buffer.field(JMod.FINAL | JMod.PUBLIC | JMod.STATIC, String.class, getProperPropertyName().toUpperCase() + "_JSON_NAME");

		field.init(JExpr.lit(name));

		JType dt = null;

		if (dataTypeClass == null)
		{
			// this is an enum
			JDefinedClass en = buffer._enum(getPropertyName() + "_enum");

			for (int i = 0; i < enumValues.size(); i++)
			{
				String e = enumValues.get(i);

				en.enumConstant(makePropertyName(e));
			}

			dt = en;
		}
		else
		{
			dt = buffer.getPackage().owner()._ref(dataTypeClass);
		}

		JFieldVar pfield = buffer.field(JMod.PRIVATE, dt, getPropertyName());

		JMethod meth = buffer.method(JMod.PUBLIC | JMod.FINAL, boolean.class, "has" + getProperPropertyName());
		JBlock bod = meth.body();
		bod.directStatement("return " + getPropertyName() + " != null;");

		JMethod getMth = buffer.method(JMod.PUBLIC | JMod.FINAL, dt, "get" + getProperPropertyName());

		if (required)
		{
			getMth.body().block().directStatement("if (!has" + getProperPropertyName() + "()) {throw new IllegalStateException(\"Property '" + getPropertyName() + "' never assigned\");}");
		}

		getMth.body().block()._return(pfield);
	}

	public String getProperPropertyName()
	{
		return makeProperPropertyName(name);
	}

	public static String makeProperPropertyName(String property)
	{
		String base = makePropertyName(property);

		base = Character.toUpperCase(base.charAt(0)) + base.substring(1);

		return base;
	}

	public String getPropertyName()
	{
		return makePropertyName(name);
	}

	public static String makePropertyName(String property)
	{
		StringBuffer buffer = new StringBuffer();

		char[] array = property.toCharArray();

		boolean capNext = false;

		for (char ch : array)
		{
			if (ch == '-' || ch == '_' || ch == '.')
			{
				capNext = true;
			}
			else
			{
				if (capNext)
				{
					buffer.append(Character.toUpperCase(ch));
					capNext = false;
				}
				else
				{
					buffer.append(ch);
				}
			}
		}

		return buffer.toString();
	}
}