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.SQLiteDaoDefinition;
import com.abubusoft.kripton.processor.sqlite.model.SQLiteEntity;
import com.abubusoft.kripton.processor.sqlite.model.SQLProperty;
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;

// TODO: Auto-generated Javadoc
/**
 * 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) {
		SQLiteDaoDefinition daoDefinition = method.getParent();
		SQLiteEntity entity = daoDefinition.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) {
		// all check is already done

		SQLiteEntity entity = method.getParent().getEntity();

		String entityName = method.getParameters().get(0).value0;
		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");

	}

}