br.com.objectos.way.sql.compiler.DefinesMethod Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of way-sql-compiler Show documentation
The newest version!
/*
 * Copyright 2015 Objectos, Fábrica de Software LTDA.
 *
 * 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 br.com.objectos.way.sql.compiler;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.util.List;

import br.com.objectos.way.code.CodeCanvasArtifact;
import br.com.objectos.way.code.CodeCanvasWriter;
import br.com.objectos.way.code.ImportInfo;
import br.com.objectos.way.code.ImportInfoSet;
import br.com.objectos.way.code.MethodInfo;
import br.com.objectos.way.code.PackageInfo;
import br.com.objectos.way.code.SimpleTypeInfo;
import br.com.objectos.way.code.TypeInfo;
import br.com.objectos.way.code.mustache.Writers;
import br.com.objectos.way.core.tmpl.mustache.IsMustacheSerializable;
import br.com.objectos.way.core.tmpl.mustache.MustacheObject;
import br.com.objectos.way.core.tmpl.mustache.Mustaches;
import br.com.objectos.way.sql.Column;
import br.com.objectos.way.sql.ColumnInfoMap;
import br.com.objectos.way.sql.ComparisonOperator;
import br.com.objectos.way.sql.HasTableInfo;
import br.com.objectos.way.sql.SortOrder;
import br.com.objectos.way.sql.Table;
import br.com.objectos.way.sql.TableInfo;

/**
 * @author [email protected] (Marcio Endo)
 */
class DefinesMethod implements IsMustacheSerializable {

  private final TypeInfo typeInfo;
  private final MethodInfo methodInfo;
  private final TableInfo tableInfo;

  public DefinesMethod(TypeInfo typeInfo, MethodInfo methodInfo, TableInfo tableInfo) {
    this.typeInfo = typeInfo;
    this.methodInfo = methodInfo;
    this.tableInfo = tableInfo;
  }

  public boolean isValid() {
    SimpleTypeInfo returnTypeInfo = methodInfo.returnTypeInfo();
    if (!returnTypeInfo.isInfoOf(TableInfo.class)) {
      methodInfo.compilationError("Method must return a TableInfo instance.");
      return false;
    }

    if (!methodInfo.hasParameterInfoListSize(0)) {
      methodInfo.compilationError("Method must have zero arguments.");
      return false;
    }

    return true;
  }

  public CodeCanvasArtifact toCodeCanvasArtifact() {
    PackageInfo packageInfo = typeInfo.packageInfo();
    return CodeCanvasWriter.forTemplate("/way-sql-compiler/Table.mustache")
        .name(packageInfo.toQualifiedName(tableInfo.getClassName()))
        .toCodeCanvasArtifact(Writers.INSTANCE, toMustache());
  }

  @Override
  public MustacheObject toMustache() {
    ImportInfoSet importInfoSet = importInfoSet();

    DefinesTable table = DefinesTable.of(tableInfo);
    table.addTo(importInfoSet);

    PackageInfo packageInfo = typeInfo.packageInfo();
    List imports = importInfoSet.toImportInfoList(packageInfo);

    return Mustaches.toMustacheHelper()
        .add("emptyImports", imports.isEmpty())
        .add("imports", imports)
        .add("method", methodInfo)
        .add("package", packageInfo)
        .add("table", table)
        .add("module", typeInfo)
        .toMustache();
  }

  public String toSchemaName() {
    return tableInfo.getSchemaName();
  }

  private ImportInfoSet importInfoSet() {
    ImportInfoSet importInfoSet = ImportInfoSet.setOf();

    importInfoSet.add(ElementType.class);
    importInfoSet.add(Retention.class);
    importInfoSet.add(RetentionPolicy.class);
    importInfoSet.add(Target.class);

    importInfoSet.add(Column.class);
    importInfoSet.add(ColumnInfoMap.class);
    importInfoSet.add(ComparisonOperator.class);
    importInfoSet.add(HasTableInfo.class);
    importInfoSet.add(SortOrder.class);
    importInfoSet.add(Table.class);
    importInfoSet.add(TableInfo.class);

    return importInfoSet;
  }
}