com.abubusoft.kripton.processor.sqlite.CodeBuilderUtility Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kripton-processor Show documentation
There is a newer version: 8.2.0-rc.4
/*******************************************************************************
 * Copyright 2015, 2017 Francesco Benincasa ([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.abubusoft.kripton.processor.sqlite;

import static com.abubusoft.kripton.processor.core.reflect.TypeUtility.typeName;

import java.lang.annotation.Annotation;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import javax.lang.model.util.Elements;

import com.abubusoft.kripton.processor.core.AssertKripton;
import com.abubusoft.kripton.processor.sqlite.grammars.jql.JQLChecker;
import com.abubusoft.kripton.processor.sqlite.model.SQLProperty;
import com.abubusoft.kripton.processor.sqlite.model.SQLiteEntity;
import com.abubusoft.kripton.processor.sqlite.model.SQLiteModelMethod;
import com.abubusoft.kripton.processor.sqlite.transform.SQLTransformer;
import com.squareup.javapoet.MethodSpec.Builder;
import com.squareup.javapoet.TypeName;

/**
 * The Class CodeBuilderUtility.
 */
public abstract class CodeBuilderUtility {

	/**
	 * Generate code necessary to put bean properties in content values map.
	 * Return primary key
	 *
	 * @param methodBuilder
	 *            used to code generation
	 * @param method
	 *            the method
	 * @param annotationClazz
	 *            the annotation clazz
	 * @return primary key.
	 */
	public static List extractUsedProperties(Builder methodBuilder, SQLiteModelMethod method, Class annotationClazz) {
		SQLiteEntity entity = method.getEntity();
		List listPropertyInContentValue = new ArrayList();

		Set foundColumns = JQLChecker.getInstance().extractColumnsToInsertOrUpdate(method, method.jql.value, entity);

		// for each property in entity except primaryKey and excluded properties
		for (SQLProperty item : entity.getCollection()) {
			if (foundColumns.size() > 0 && !foundColumns.contains(item.getName())) {
				continue;
			}

			// add property to list of used properties
			listPropertyInContentValue.add(item);
		}

		return listPropertyInContentValue;

	}
	
	/**
	 * Generate content values from entity.
	 *
	 * @param elementUtils
	 *            the element utils
	 * @param method
	 *            the method
	 * @param annotationClazz
	 *            the annotation clazz
	 * @param methodBuilder
	 *            the method builder
	 * @param alreadyUsedBeanPropertiesNames
	 *            the already used bean properties names
	 */
	public static void generateContentValuesFromEntity(Elements elementUtils, SQLiteModelMethod method, Class annotationClazz, Builder methodBuilder,
			List alreadyUsedBeanPropertiesNames) {
		String entityName = method.getParameters().get(0).value0;
		
		generateContentValuesFromEntity(elementUtils, method, entityName, annotationClazz, methodBuilder, alreadyUsedBeanPropertiesNames);
	}

	/**
	 * Generate content values from entity.
	 *
	 * @param elementUtils
	 *            the element utils
	 * @param method
	 *            the method
	 * @param annotationClazz
	 *            the annotation clazz
	 * @param methodBuilder
	 *            the method builder
	 * @param alreadyUsedBeanPropertiesNames
	 *            the already used bean properties names
	 */
	public static void generateContentValuesFromEntity(Elements elementUtils, SQLiteModelMethod method, String entityName, Class annotationClazz, Builder methodBuilder,
			List alreadyUsedBeanPropertiesNames) {
		// all check is already done

		SQLiteEntity entity = method.getEntity();

		
		TypeName entityClassName = typeName(entity.getElement());

		AssertKripton.assertTrueOrInvalidMethodSignException(!method.hasAdapterForParam(entityName), method, "method's parameter '%s' can not use a type adapter", entityName);

		Set updateColumns = JQLChecker.getInstance().extractColumnsToInsertOrUpdate(method, method.jql.value, entity);
		SQLProperty item;
		for (String columnName : updateColumns) {
			item = entity.get(columnName);
			AssertKripton.assertTrueOrUnknownPropertyInJQLException(item != null, method, columnName);

			// if (TypeUtility.isNullable(item) && !item.hasTypeAdapter()) {
			// methodBuilder.beginControlFlow("if ($L!=null)",
			// getter(entityName, entityClassName, item));
			// }

			// add property to list of used properties
			if (method.isLogEnabled()) {
				methodBuilder.addCode("_contentValues.put($S, ", item.columnName);
			} else {
				methodBuilder.addCode("_contentValues.put(");
			}
			SQLTransformer.javaProperty2ContentValues(methodBuilder, entityClassName, entityName, item);
			methodBuilder.addCode(");\n");

			// if (TypeUtility.isNullable(item) && !item.hasTypeAdapter()) {
			// methodBuilder.nextControlFlow("else");
			//
			// if (method.isLogEnabled()) {
			// methodBuilder.addCode("_contentValues.putNull($S);\n",
			// item.columnName);
			// } else {
			// methodBuilder.addCode("_contentValues.putNull();\n");
			// }
			//
			//
			// methodBuilder.endControlFlow();
			// }
		}

		methodBuilder.addCode("\n");

	}

}