/* * Copyright 2015, The Querydsl Team (http://www.querydsl.com/team) * * 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.querydsl.sql.codegen; import javax.inject.Inject; import javax.inject.Named; import com.mysema.codegen.model.SimpleType; import com.mysema.codegen.model.Type; import com.querydsl.codegen.QueryTypeFactory; /** * {@code SQLQueryTypeFactory} is a {@link QueryTypeFactory} implementation with configuration * data from this module * * @author tiwe * */ public final class SQLQueryTypeFactory implements QueryTypeFactory { private final String packageName, beanPackageName; private final int stripStart, stripEnd; private final String prefix, suffix; private final boolean replacePackage; @Inject public SQLQueryTypeFactory( @Named(SQLCodegenModule.PACKAGE_NAME) String packageName, @Named(SQLCodegenModule.BEAN_PACKAGE_NAME) String beanPackageName, @Named(SQLCodegenModule.BEAN_PREFIX) String beanPrefix, @Named(SQLCodegenModule.BEAN_SUFFIX) String beanSuffix, @Named(SQLCodegenModule.PREFIX) String prefix, @Named(SQLCodegenModule.SUFFIX) String suffix) { this.packageName = packageName; this.beanPackageName = beanPackageName; this.replacePackage = !packageName.equals(beanPackageName); this.stripStart = beanPrefix.length(); this.stripEnd = beanSuffix.length(); this.prefix = prefix; this.suffix = suffix; } @Override public Type create(Type type) { String packageName = type.getPackageName(); if (replacePackage) { packageName = this.packageName + packageName.substring(beanPackageName.length()); } String simpleName = type.getSimpleName(); simpleName = prefix + simpleName.substring(stripStart, simpleName.length() - stripEnd) + suffix; return new SimpleType(packageName + "." + simpleName, packageName, simpleName); } }