com.mockrunner.gen.proc.JavaClassGenerator Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mockrunner-core Show documentation
Show all versions of mockrunner-core Show documentation
Core classes common to all Mockrunner modules
package com.mockrunner.gen.proc;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
import com.mockrunner.util.common.ArrayUtil;
import com.mockrunner.util.common.ClassUtil;
import com.mockrunner.util.common.StringUtil;
public class JavaClassGenerator
{
private Package packageInfo;
private List imports;
private String className;
private boolean isAbstract;
private Class superClass;
private List interfaces;
private List memberTypes;
private List memberNames;
private String[] classCommentLines;
private boolean createJavaDocComments;
private List methods;
private List constructors;
public JavaClassGenerator()
{
reset();
}
public void reset()
{
imports = new ArrayList();
interfaces = new ArrayList();
memberTypes = new ArrayList();
memberNames = new ArrayList();
createJavaDocComments = true;
isAbstract = false;
methods = new ArrayList();
constructors = new ArrayList();
}
public void setCreateJavaDocComments(boolean createJavaDocComments)
{
this.createJavaDocComments = createJavaDocComments;
}
public void setPackage(Package packageInfo)
{
this.packageInfo = packageInfo;
}
public void setClassName(String className)
{
this.className = className;
}
public void setAbstract(boolean isAbstract)
{
this.isAbstract = isAbstract;
}
public void setSuperClass(Class superClass)
{
this.superClass = superClass;
}
public void addImport(Class importClass)
{
imports.add(importClass.getName());
}
public void addInterfaceImplementation(Class interfaceClass)
{
interfaces.add(interfaceClass);
}
public void setClassComment(String[] commentLines)
{
classCommentLines = (String[])ArrayUtil.copyArray(commentLines);
}
public void addMemberDeclaration(Class memberType, String name)
{
memberTypes.add(memberType);
memberNames.add(name);
}
public void addConstructorDeclaration()
{
constructors.add(new ConstructorDeclaration());
}
public void addConstructorDeclaration(ConstructorDeclaration constructor)
{
constructors.add(constructor);
}
public void addMethodDeclaration(MethodDeclaration method)
{
methods.add(method);
}
public String generate()
{
JavaLineAssembler assembler = new JavaLineAssembler();
assembler.appendPackageInfo(getPackageName());
appendImportBlocks(assembler);
appendCommentBlock(assembler, classCommentLines);
if(isAbstract)
{
assembler.appendClassDefintion(className, Modifier.toString(Modifier.ABSTRACT), getClassName(superClass), getClassNames(interfaces));
}
else
{
assembler.appendClassDefintion(className, "", getClassName(superClass), getClassNames(interfaces));
}
assembler.appendLeftBrace();
assembler.appendNewLine();
assembler.setIndentLevel(1);
appendMembers(assembler);
appendConstructors(assembler);
appendMethods(assembler);
assembler.setIndentLevel(0);
assembler.appendRightBrace();
return assembler.getResult();
}
private String getPackageName()
{
if(null != packageInfo)
{
return packageInfo.getName();
}
return null;
}
private String getClassName(Class clazz)
{
if(null == clazz) return null;
return ClassUtil.getClassName(clazz);
}
private String[] getClassNames(List classList)
{
if(null == classList || classList.size() <= 0) return null;
List nameList = new ArrayList();
for(int ii = 0; ii < classList.size(); ii++)
{
Class clazz = (Class)classList.get(ii);
if(null != clazz)
{
nameList.add(getClassName(clazz));
}
}
return (String[])nameList.toArray(new String[nameList.size()]);
}
private String[] getClassNames(Class[] arguments)
{
if(null == arguments || arguments.length <= 0) return null;
String[] names = new String[arguments.length];
for(int ii = 0; ii < arguments.length; ii++)
{
Class clazz = arguments[ii];
names[ii] = getClassName(clazz);
}
return names;
}
private String[] getArgumentNames(Class[] arguments, String[] argumentNames)
{
if(null == arguments || arguments.length <= 0) return null;
if(null != argumentNames && argumentNames.length >= arguments.length) return argumentNames;
if(null == argumentNames) argumentNames = new String[0];
String[] newNames = new String[arguments.length];
for(int ii = 0; ii < argumentNames.length; ii++)
{
newNames[ii] = argumentNames[ii];
}
for(int ii = argumentNames.length; ii < arguments.length; ii++)
{
newNames[ii] = ClassUtil.getArgumentName(arguments[ii]);
}
ArrayUtil.ensureUnique(newNames);
return newNames;
}
private void appendImportBlocks(JavaLineAssembler assembler)
{
List importBlocks = processImports();
for(int ii = 0; ii < importBlocks.size(); ii++)
{
Set currentBlock = (Set)importBlocks.get(ii);
assembler.appendImports(new ArrayList(currentBlock));
assembler.appendNewLine();
}
}
private void appendCommentBlock(JavaLineAssembler assembler, String[] commentLines)
{
if(createJavaDocComments)
{
assembler.appendJavaDocComment(commentLines);
}
else
{
assembler.appendBlockComment(commentLines);
}
}
private void appendMembers(JavaLineAssembler assembler)
{
String[] memberTypeNames = getClassNames(memberTypes);
if(null == memberTypeNames) return;
for(int ii = 0; ii < memberTypeNames.length; ii++)
{
assembler.appendMemberDeclaration(memberTypeNames[ii], (String)memberNames.get(ii));
}
}
private void appendMethods(JavaLineAssembler assembler)
{
for(int ii = 0; ii < methods.size(); ii++)
{
MethodDeclaration declaration = (MethodDeclaration)methods.get(ii);
appendMethodHeader(assembler, declaration);
String[] modifiers = prepareModifiers(declaration.getModifier());
String returnType = getClassName(declaration.getReturnType());
String[] argumentTypes = getClassNames(declaration.getArguments());
String[] exceptionTypes = getClassNames(declaration.getExceptions());
String[] argumentNames = getArgumentNames(declaration.getArguments(), declaration.getArgumentNames());
assembler.appendMethodDeclaration(modifiers, returnType, declaration.getName(), argumentTypes, argumentNames, exceptionTypes);
appendMethodBody(assembler, declaration);
}
}
private void appendConstructors(JavaLineAssembler assembler)
{
for(int ii = 0; ii < constructors.size(); ii++)
{
ConstructorDeclaration declaration = (ConstructorDeclaration)constructors.get(ii);
appendMethodHeader(assembler, declaration);
String[] argumentTypes = getClassNames(declaration.getArguments());
String[] exceptionTypes = getClassNames(declaration.getExceptions());
String[] argumentNames = getArgumentNames(declaration.getArguments(), declaration.getArgumentNames());
assembler.appendConstructorDeclaration(className, argumentTypes, argumentNames, exceptionTypes);
appendMethodBody(assembler, declaration);
}
}
private void appendMethodHeader(JavaLineAssembler assembler, ConstructorDeclaration declaration)
{
assembler.appendNewLine();
appendCommentBlock(assembler, declaration.getCommentLines());
}
private void appendMethodBody(JavaLineAssembler assembler, ConstructorDeclaration declaration)
{
assembler.appendIndent();
assembler.appendLeftBrace();
assembler.appendNewLine();
appendCodeLines(assembler, declaration.getCodeLines());
assembler.appendIndent();
assembler.appendRightBrace();
assembler.appendNewLine();
}
private void appendCodeLines(JavaLineAssembler assembler, String[] codeLines)
{
assembler.setIndentLevel(2);
assembler.appendCodeLines(codeLines);
assembler.setIndentLevel(1);
}
private String[] prepareModifiers(int modifier)
{
String modifierString = Modifier.toString(modifier);
if(null == modifierString || modifierString.trim().length() <= 0) return null;
return StringUtil.split(modifierString, " ", true);
}
private List processImports()
{
addMissingImports();
PackageImportSorter sorter = new PackageImportSorter();
return sorter.sortBlocks(imports);
}
private void addMissingImports()
{
addImportIfNecessary(superClass);
addImportsIfNecessary(interfaces);
addImportsIfNecessary(memberTypes);
for(int ii = 0; ii < constructors.size(); ii++)
{
ConstructorDeclaration declaration = (ConstructorDeclaration)constructors.get(ii);
addImportsForArguments(declaration);
addImportsForExceptions(declaration);
}
for(int ii = 0; ii < methods.size(); ii++)
{
MethodDeclaration declaration = (MethodDeclaration)methods.get(ii);
addImportForReturnType(declaration);
addImportsForArguments(declaration);
addImportsForExceptions(declaration);
}
}
private void addImportsForExceptions(ConstructorDeclaration declaration)
{
Class[] exceptions = declaration.getExceptions();
if(null == exceptions || exceptions.length <= 0) return;
for(int ii = 0; ii < exceptions.length; ii++)
{
addImportIfNecessary(exceptions[ii]);
}
}
private void addImportsForArguments(ConstructorDeclaration declaration)
{
Class[] arguments = declaration.getArguments();
if(null == arguments || arguments.length <= 0) return;
for(int ii = 0; ii < arguments.length; ii++)
{
addImportIfNecessary(arguments[ii]);
}
}
private void addImportForReturnType(MethodDeclaration declaration)
{
Class returnType = declaration.getReturnType();
if(null == returnType) return;
addImportIfNecessary(returnType);
}
private void addImportsIfNecessary(List classes)
{
if(null == classes) return;
for(int ii = 0; ii < classes.size(); ii++)
{
addImportIfNecessary((Class)classes.get(ii));
}
}
private void addImportIfNecessary(Class clazz)
{
if(null == clazz) return;
while(clazz.isArray()) clazz = clazz.getComponentType();
if(imports.contains(clazz.getName())) return;
if(clazz.getName().startsWith("java.lang")) return;
if(belongsToSamePackage(clazz)) return;
if(clazz.isPrimitive()) return;
addImport(clazz);
}
private boolean belongsToSamePackage(Class clazz)
{
String thisPackageName = getPackageName();
Package classPackage = clazz.getPackage();
String classPackageName = "";
if(null != classPackage)
{
classPackageName = classPackage.getName();
}
if(null == thisPackageName)
{
thisPackageName = "";
}
if(null == classPackageName)
{
classPackageName = "";
}
return thisPackageName.equals(classPackageName);
}
public static class ConstructorDeclaration
{
private Class[] arguments;
private String[] argumentNames;
private String[] codeLines;
private String[] commentLines;
private Class[] exceptions;
public String[] getCodeLines()
{
if(null == codeLines) return null;
return (String[])ArrayUtil.copyArray(codeLines);
}
public void setCodeLines(String[] codeLines)
{
this.codeLines = (String[])ArrayUtil.copyArray(codeLines);
}
public String[] getCommentLines()
{
if(null == commentLines) return null;
return (String[])ArrayUtil.copyArray(commentLines);
}
public void setCommentLines(String[] commentLines)
{
this.commentLines = (String[])ArrayUtil.copyArray(commentLines);
}
public String[] getArgumentNames()
{
if(null == argumentNames) return null;
return (String[])ArrayUtil.copyArray(argumentNames);
}
public void setArgumentNames(String[] argumentNames)
{
this.argumentNames = (String[])ArrayUtil.copyArray(argumentNames);
}
public Class[] getArguments()
{
if(null == arguments) return null;
return (Class[])ArrayUtil.copyArray(arguments);
}
public void setArguments(Class[] arguments)
{
this.arguments = (Class[])ArrayUtil.copyArray(arguments);
}
public Class[] getExceptions()
{
if(null == exceptions) return null;
return (Class[])ArrayUtil.copyArray(exceptions);
}
public void setExceptions(Class[] exceptions)
{
this.exceptions = (Class[])ArrayUtil.copyArray(exceptions);
}
}
public static class MethodDeclaration extends ConstructorDeclaration
{
private int modifier;
private Class returnType;
private String name;
public MethodDeclaration()
{
this("method");
}
public MethodDeclaration(String name)
{
this(Modifier.PUBLIC, name);
}
public MethodDeclaration(int modifier, String name)
{
this(modifier, name, Void.TYPE);
}
public MethodDeclaration(int modifier, String name, Class returnType)
{
setModifier(modifier);
setReturnType(returnType);
setName(name);
}
public int getModifier()
{
return modifier;
}
public void setModifier(int modifier)
{
this.modifier = modifier;
}
public String getName()
{
return name;
}
public void setName(String name)
{
this.name = name;
}
public Class getReturnType()
{
return returnType;
}
public void setReturnType(Class returnType)
{
this.returnType = returnType;
}
}
}