GenerateQueriesIfNotExistsPass.java

package joist.codegen.passes;

import joist.codegen.Codegen;
import joist.codegen.dtos.Entity;
import joist.sourcegen.GClass;

public class GenerateQueriesIfNotExistsPass implements Pass<Codegen> {

  public void pass(Codegen codegen) {
    for (Entity entity : codegen.getSchema().getEntities().values()) {
      if (entity.isCodeEntity()) {
        continue;
      }
      if (!codegen.getOutputSourceDirectory().exists(entity.getFullQueriesClassName())) {
        GClass queries = codegen.getOutputSourceDirectory().getClass(entity.getFullQueriesClassName());
        queries.baseClassName(entity.getFullQueriesCodegenClassName());
      } else {
        codegen.getOutputSourceDirectory().markTouched(entity.getFullQueriesClassName());
      }
    }
  }

}