com.stanfy.helium.gradle.UserConfig Maven / Gradle / Ivy

Go to download
package com.stanfy.helium.gradle;

import com.stanfy.helium.utils.Names;
import org.apache.commons.io.FilenameUtils;
import org.gradle.api.Project;

import java.io.File;
import java.util.HashMap;
import java.util.Map;

/**
 * Internal structure that stores user configuration.
 */
class UserConfig {

  /** Gradle project. */
  final Project project;

  /** Default source generation data. */
  SourceGenDslDelegate defaultSourceGeneration;

  /** Source generation rules per each spec. */
  private final Map specSourceGeneration = new HashMap();

  UserConfig(final Project project) {
    this.project = project;
  }

  public static String specName(final File specFile) {
    String name = FilenameUtils.getBaseName(specFile.getName());
    return Names.prettifiedName(name);
  }

  public void set(final File spec, final SourceGenDslDelegate delegate) {
    specSourceGeneration.put(spec, delegate);
  }

  public SourceGenDslDelegate getSourceGenFor(final File spec) {
    SourceGenDslDelegate res = specSourceGeneration.get(spec);
    if (res == null) {
      return defaultSourceGeneration;
    }
    if (defaultSourceGeneration != null) {
      if (res.getEntities() == null) {
        res.setEntities(defaultSourceGeneration.getEntities());
      }
      if (res.getConstants() == null) {
        res.setConstants(defaultSourceGeneration.getConstants());
      }
      if (res.getRetrofit() == null) {
        res.setRetrofit(defaultSourceGeneration.getRetrofit());
      }
    }
    return res;
  }

}