private static void processAccessors(
JFieldVar fieldVar, JDefinedClass packetClass, boolean fromClient) {
String name = WordUtils.capitalize(fieldVar.name());
if (fromClient) {
String methodName = "get" + name;
JMethod getter = packetClass.method(JMod.PUBLIC, fieldVar.type(), methodName);
getter.body()._return(fieldVar);
} else {
String methodName = "set" + name;
JMethod setter = packetClass.method(JMod.PUBLIC, Void.TYPE, methodName);
setter.param(fieldVar.type(), fieldVar.name());
setter.body().assign(JExpr._this().ref(fieldVar.name()), JExpr.ref(fieldVar.name()));
}
}
private static void processToString(JDefinedClass packetClass, JCodeModel codeModel) {
JClass string = codeModel.ref(String.class);
JClass stringBuilder = codeModel.ref(StringBuilder.class);
JClass arrays = codeModel.ref(Arrays.class);
JMethod toStringMeth = packetClass.method(JMod.PUBLIC, String.class, "toString");
toStringMeth.annotate(Override.class);
JBlock body = toStringMeth.body();
JVar stringBuilderVar = body.decl(stringBuilder, "sb");
stringBuilderVar =
stringBuilderVar.init(JExpr._new(stringBuilder).arg(packetClass.name() + "["));
JInvocation appendChain = null;
for (JFieldVar fieldVar : packetClass.fields().values()) {
if (appendChain != null) {
// a comma is needed
appendChain = appendChain.invoke("append").arg("," + fieldVar.name() + "=");
} else {
appendChain = stringBuilderVar.invoke("append").arg(fieldVar.name() + "=");
}
// now add the field to the toString output
JExpression expression =
fieldVar.type().isArray()
? arrays.staticInvoke("toString").arg(JExpr._this().ref(fieldVar.name()))
: fieldVar.type().isReference()
? JExpr._this().ref(fieldVar.name()).invoke("toString")
: JExpr._this().ref(fieldVar.name());
appendChain = appendChain.invoke("append").arg(expression);
}
if (appendChain != null) {
appendChain = appendChain.invoke("append").arg("]");
} else {
appendChain = stringBuilderVar.invoke("append").arg("]");
}
body.add(appendChain);
body._return(stringBuilderVar.invoke("toString"));
}
private static void processField(
FieldType field,
JDefinedClass packetClass,
JCodeModel codeModel,
AtomicInteger numberOfUnknowns,
boolean fromClient)
throws JClassAlreadyExistsException {
boolean isNested =
(field.getType() == PacketSimpleTypes.OPTIONAL)
|| (field.getType() == PacketSimpleTypes.ARRAY_STATIC)
|| (field.getType() == PacketSimpleTypes.ARRAY_VAR_SMALL)
|| (field.getType() == PacketSimpleTypes.ARRAY_VAR_BIG);
boolean isArray =
(field.getType() == PacketSimpleTypes.BUFFER_STATIC)
|| (field.getType() == PacketSimpleTypes.BUFFER_VAR_SMALL)
|| (field.getType() == PacketSimpleTypes.BUFFER_VAR_BIG)
|| (isNested && (field.getType() != PacketSimpleTypes.OPTIONAL));
boolean isStaticLength =
(isArray)
&& ((field.getType() == PacketSimpleTypes.ARRAY_STATIC)
|| (field.getType() == PacketSimpleTypes.BUFFER_STATIC));
// length is either not set for non arrays and static arrays,
// its 1byte for small stuff and 2bytes for big stuff.
int prefixLength =
(!isArray || isStaticLength)
? -1
: (field.getType() == PacketSimpleTypes.ARRAY_VAR_SMALL)
|| (field.getType() == PacketSimpleTypes.BUFFER_VAR_SMALL)
? 1
: 2;
String name = "";
if (field.getInfo() == null || field.getInfo().getName() == null) {
// check if we got special fields...
if (field.getType() == PacketSimpleTypes.AGENTID) {
name = "AgentID";
} else if (field.getType() == PacketSimpleTypes.OPTIONAL) {
name = "Optional";
} else {
name = "Unknown";
}
name += numberOfUnknowns.incrementAndGet();
} else {
name = field.getInfo().getName();
}
String fieldName = WordUtils.uncapitalize(name);
String fieldDescription =
(field.getInfo() == null
|| field.getInfo().getDescription() == null
|| field.getInfo().getDescription().isEmpty())
? ""
: "\n" + WordUtils.wrap(field.getInfo().getDescription(), /* maximumLength */ 50);
JType fieldType;
if (isNested) {
JDefinedClass nestedClass =
packetClass
._class(JMod.FINAL | JMod.STATIC | JMod.PUBLIC, "Nested" + name)
._implements(NestedMarker.class);
AtomicInteger numberOfUnknownsNested = new AtomicInteger();
for (FieldType nested : field.getField()) {
processField(nested, nestedClass, codeModel, numberOfUnknownsNested, fromClient);
}
// generate getters, setters
for (JFieldVar fieldVar : nestedClass.fields().values()) {
processAccessors(fieldVar, nestedClass, fromClient);
}
processToString(nestedClass, codeModel);
// nested classes are either arrays or optional...
// meaning we will later have to test if they are null before reading/writing
fieldType = isArray ? nestedClass.array() : nestedClass;
} else {
Class<?> fieldClass = convertFieldTypeToClass(field);
fieldType = codeModel._ref(fieldClass);
}
LOGGER.debug("|+-Processing field: {}, of type: {}", fieldName, fieldType);
// add the field
JFieldVar packetField = packetClass.field(JMod.PRIVATE, fieldType, fieldName);
if (fieldDescription != null && !fieldDescription.trim().equals("")) {
JDocComment jDocComment = packetField.javadoc();
jDocComment.add(fieldDescription);
}
// and dont forget array annotations if necessary
if (isArray) {
int size = (int) (field.getElements() == null ? -1 : field.getElements());
packetField
.annotate(IsArray.class)
.param("constant", isStaticLength)
.param("size", size)
.param("prefixLength", prefixLength);
}
// or any special annotations
if (field.getType() == PacketSimpleTypes.LONG) {
packetField.annotate(IsInt64.class);
}
if (field.getType() == PacketSimpleTypes.VARINT) {
packetField.annotate(IsVarInt.class);
}
if (field.getType() == PacketSimpleTypes.ASCII) {
packetField.annotate(IsASCII.class);
}
}
