br.com.objectos.way.sql.compiler.QueryColumn 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 br.com.objectos.way.code.ClassInfo;
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 com.google.common.base.Function;
import com.google.common.base.Optional;

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

  private final SqlColumn sqlColumn;
  private final ClassInfo referencedClassInfo;

  public QueryColumn(SqlColumn sqlColumn, ClassInfo referencedClassInfo) {
    this.sqlColumn = sqlColumn;
    this.referencedClassInfo = referencedClassInfo;
  }

  @Override
  public MustacheObject toMustache() {
    return Mustaches.toMustacheHelper()
        .addValue(sqlColumn)
        .add("foreignKey", foreignKey().orNull())
        .toMustache();
  }

  private Optional foreignKey() {
    return sqlColumn
        .toForeignKey(referencedClassInfo)
        .transform(new Function() {
          @Override
          public ForeignKey apply(SqlColumn input) {
            return new ForeignKey(input);
          }
        });
  }

  private class ForeignKey implements IsMustacheSerializable {

    private final SqlColumn fk;
    private final String parameter;

    public ForeignKey(SqlColumn fk) {
      this.fk = fk;
      parameter = fk.getSqlParameter();
    }

    @Override
    public MustacheObject toMustache() {
      return Mustaches.toMustacheHelper()
          .add("binder", binder())
          .add("fk", fk)
          .add("parameter", parameter)
          .toMustache();
    }

    private String binder() {
      String varName = referencedClassInfo.getVarName() + "." + fk.methodInfo().name() + "()";
      return fk.binder(varName);
    }

  }

}