Android Open Source - storm-gen Class Model

From Project

Back to project page storm-gen.

License

The source code is released under:

Apache License

If you think the Android project storm-gen listed in this page is inappropriate, such as containing malicious code/tools or violating the copyright, please email info at java2s dot com, thanks.

Java Source Code

/*******************************************************************************
 * Copyright 2012 Google, Inc.//  w ww. j  av a2s  .c  o m
 *
 * 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 com.turbomanage.storm.apt;

import java.util.ArrayList;
import java.util.List;

import com.turbomanage.storm.apt.entity.FieldModel;
import com.turbomanage.storm.types.EnumConverter;

public abstract class ClassModel {

  private String className;
  private String packageName;
  protected List<String> imports = new ArrayList<String>();
  protected List<FieldModel> fields = new ArrayList<FieldModel>();

  public List<String> getImports() {
    return imports;
  }

  public List<FieldModel> getFields() {
    return fields;
  }

  public void addField(FieldModel field) {
    fields.add(field);
    if (field.isEnum()) {
      addImport(EnumConverter.class.getCanonicalName());
    }
    // add import for converter if needed
    String converterType = field.getQualifiedConverterClass();
    addImport(converterType);
  }

  public void addImport(String importPath) {
    if (!imports.contains(importPath)) {
      imports.add(importPath);
    }
  }

  /**
   * @return Name of the generated class
   */
  public String getClassName() {
    return className;
  }

  /**
   * @return Package name of the generated class
   */
  public String getPackage() {
    return this.packageName;
  }

  /**
   * Returns the fully qualified class name of the generated class.
   *
   * @return Name of generated class with package prepended
   */
  public String getQualifiedClassName() {
    return this.getPackage() + "." + this.getClassName();
  }

  public void setClassName(String className) {
    this.className = className;
  }

  public void setPackageName(String packageName) {
    this.packageName = packageName;
  }

  protected String capFirst(String anyName) {
    String capFirstLetter = anyName.substring(0, 1).toUpperCase();
    return capFirstLetter + anyName.substring(1);
  }

  protected void parseQualifiedClassName(String helperClass) {
    int lastDot = helperClass.lastIndexOf('.');
    if (lastDot < 1) {
      throw new IllegalArgumentException("The default package is not allowed for type " + helperClass);
    }
    String pkg = helperClass.substring(0, lastDot);
    this.setPackageName(pkg);
    this.setClassName(helperClass.substring(lastDot + 1));
  }

}

Java Source Code List