io.reactiverse.es4x.codegen.generator.ModJS Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2018 Paulo Lopes.
 *
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  and Apache License v2.0 which accompanies this distribution.
 *
 *  The Eclipse Public License is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *
 *  The Apache License v2.0 is available at
 *  http://www.opensource.org/licenses/apache2.0.php
 *
 *  You may elect to redistribute this code under either of these licenses.
 */
package io.reactiverse.es4x.codegen.generator;

import io.vertx.codegen.*;

import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.HashSet;
import java.util.Map;

public class ModJS extends Generator {

  public ModJS() {
    kinds = new HashSet<>();
    kinds.add("class");
    kinds.add("enum");
    kinds.add("dataObject");
    kinds.add("module");

    name = "es4x-generator (mod.js)";
    incremental = true;
  }

  @Override
  public String filename(Model model) {
    return "npm/mod.js";
  }

  @Override
  public String render(Model model, int index, int size, Map session) {

    if (model instanceof EnumModel) {
      session.putIfAbsent("enum", "seen");
    }

    if (model instanceof ClassModel) {
      session.putIfAbsent("index", "seen");
    }

    if (model instanceof DataObjectModel) {
      session.putIfAbsent("options", "seen");
    }

    if (index != size - 1) {
      // wait for the last run
      return "";
    }

    StringWriter sw = new StringWriter();
    PrintWriter writer = new PrintWriter(sw);

    Util.generateLicense(writer);

    if (session.containsKey("enum")) {
      writer.println("export * from './enums.mjs';");
    }
    if (session.containsKey("options")) {
      writer.println("export * from './options.mjs';");
    }
    if (session.containsKey("index")) {
      writer.println("export * from './index.mjs';");
    }

    return sw.toString();
  }
}