Skip to content

Commit 4ee1c81

Browse files
rbygraverbygrave
committed
Add support for @QueryParam, @FormParam, @default, @cookie, @Header 
1 parent e7a1ed3 commit 4ee1c81

File tree

7 files changed

+382
-132
lines changed

7 files changed

+382
-132
lines changed

‎src/main/java/io/dinject/javalin/generator/ElementReader.java‎

Lines changed: 215 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,215 @@
1+
package io.dinject.javalin.generator;
2+
3+
import io.dinject.controller.BeanParam;
4+
import io.dinject.controller.Cookie;
5+
import io.dinject.controller.Default;
6+
import io.dinject.controller.Form;
7+
import io.dinject.controller.FormParam;
8+
import io.dinject.controller.Header;
9+
import io.dinject.controller.QueryParam;
10+
11+
import javax.lang.model.element.Element;
12+
import javax.lang.model.element.TypeElement;
13+
import java.util.Collections;
14+
import java.util.Set;
15+
16+
class ElementReader {
17+
18+
private final ProcessingContext ctx;
19+
private final String rawType;
20+
private final TypeHandler typeHandler;
21+
private final String varName;
22+
private final String snakeName;
23+
24+
private String paramName;
25+
private ParamType paramType;
26+
private String paramDefault;
27+
private String docComment;
28+
29+
ElementReader(Element element, ProcessingContext ctx, ParamType defaultType) {
30+
this.ctx = ctx;
31+
this.rawType = element.asType().toString();
32+
this.typeHandler = TypeMap.get(rawType);
33+
34+
this.varName = element.getSimpleName().toString();
35+
this.snakeName = Util.snakeCase(varName);
36+
this.paramName = varName;
37+
this.docComment = ctx.docComment(element);
38+
39+
readAnnotations(element, defaultType);
40+
}
41+
42+
private void readAnnotations(Element element, ParamType defaultType) {
43+
Default defaultVal = element.getAnnotation(Default.class);
44+
if (defaultVal != null) {
45+
this.paramDefault = defaultVal.value();
46+
}
47+
Form form = element.getAnnotation(Form.class);
48+
if (form != null) {
49+
this.paramType = ParamType.FORM;
50+
return;
51+
}
52+
BeanParam beanParam = element.getAnnotation(BeanParam.class);
53+
if (beanParam != null) {
54+
this.paramType = ParamType.BEANPARAM;
55+
return;
56+
}
57+
QueryParam queryParam = element.getAnnotation(QueryParam.class);
58+
if (queryParam != null) {
59+
this.paramName = nameFrom(queryParam.value(), varName);
60+
this.paramType = ParamType.QUERYPARAM;
61+
return;
62+
}
63+
FormParam formParam = element.getAnnotation(FormParam.class);
64+
if (formParam != null) {
65+
this.paramName = nameFrom(formParam.value(), varName);
66+
this.paramType = ParamType.FORMPARAM;
67+
return;
68+
}
69+
Cookie cookieParam = element.getAnnotation(Cookie.class);
70+
if (cookieParam != null) {
71+
this.paramName = nameFrom(cookieParam.value(), varName);
72+
this.paramType = ParamType.COOKIE;
73+
this.paramDefault = null;
74+
return;
75+
}
76+
Header headerParam = element.getAnnotation(Header.class);
77+
if (headerParam != null) {
78+
this.paramName = nameFrom(headerParam.value(), Util.initcapSnake(snakeName));
79+
this.paramType = ParamType.HEADER;
80+
this.paramDefault = null;
81+
return;
82+
}
83+
if (paramType == null) {
84+
this.paramType = defaultType;
85+
}
86+
}
87+
88+
@Override
89+
public String toString() {
90+
return varName + " type:" + rawType + " paramType:" + paramType + " dft:" + paramDefault;
91+
}
92+
93+
private String nameFrom(String name, String defaultName) {
94+
if (name != null && !name.isEmpty()) {
95+
return name;
96+
}
97+
return defaultName;
98+
}
99+
100+
String getVarName() {
101+
return varName;
102+
}
103+
104+
private boolean hasParamDefault() {
105+
return paramDefault != null && !paramDefault.isEmpty();
106+
}
107+
108+
private boolean isJavalinContext() {
109+
return Constants.JAVALIN_CONTEXT.equals(rawType);
110+
}
111+
112+
private String shortType() {
113+
if (typeHandler != null) {
114+
return typeHandler.shortName();
115+
} else {
116+
return Util.shortName(rawType);
117+
}
118+
}
119+
120+
private String derivePathParam(Set<String> pathParams) {
121+
if (pathParams.contains(varName)) {
122+
return varName;
123+
}
124+
if (pathParams.contains(snakeName)) {
125+
return snakeName;
126+
}
127+
return null;
128+
}
129+
130+
void addImports(BeanReader bean) {
131+
if (typeHandler != null) {
132+
String importType = typeHandler.getImportType();
133+
if (importType != null) {
134+
bean.addImportType(rawType);
135+
}
136+
} else {
137+
bean.addImportType(rawType);
138+
}
139+
}
140+
141+
void writeParamName(Append writer) {
142+
if (isJavalinContext()) {
143+
writer.append("ctx");
144+
} else {
145+
writer.append(varName);
146+
}
147+
}
148+
149+
void writeCtxGet(Append writer, Set<String> pathParams) {
150+
151+
if (isJavalinContext()) {
152+
// no conversion for this parameter
153+
return;
154+
}
155+
156+
String shortType = shortType();
157+
writer.append(" %s %s = ", shortType, varName);
158+
if (setValue(writer, pathParams, shortType)) {
159+
writer.append(";").eol();
160+
}
161+
}
162+
163+
void setValue(Append writer) {
164+
String shortType = shortType();
165+
setValue(writer, Collections.emptySet(), shortType);
166+
}
167+
168+
private boolean setValue(Append writer, Set<String> pathParams, String shortType) {
169+
170+
if (ParamType.FORM == paramType) {
171+
writeForm(writer, shortType, varName, ParamType.FORMPARAM);
172+
return false;
173+
}
174+
if (ParamType.BEANPARAM == paramType) {
175+
writeForm(writer, shortType, varName, ParamType.QUERYPARAM);
176+
return false;
177+
}
178+
179+
// path parameters are expected to be not nullable
180+
// ... with query parameters nullable
181+
String pathParameter = derivePathParam(pathParams);
182+
String asMethod = (typeHandler == null) ? null : (pathParameter != null) ? typeHandler.asMethod() : typeHandler.toMethod();
183+
184+
if (asMethod != null) {
185+
writer.append(asMethod);
186+
}
187+
if (pathParameter != null) {
188+
writer.append("ctx.pathParam(\"%s\")", pathParameter);
189+
} else {
190+
if (typeHandler == null) {
191+
// assuming this is a body (POST, PATCH)
192+
writer.append("ctx.bodyAsClass(%s.class)", shortType);
193+
} else {
194+
if (hasParamDefault()) {
195+
writer.append("ctx.%s(\"%s\",\"%s\")", paramType, paramName, paramDefault);
196+
} else {
197+
writer.append("ctx.%s(\"%s\")", paramType, paramName);
198+
}
199+
}
200+
}
201+
202+
if (asMethod != null) {
203+
writer.append(")");
204+
}
205+
return true;
206+
}
207+
208+
private void writeForm(Append writer, String shortType, String varName, ParamType defaultParamType) {
209+
210+
TypeElement formBeanType = ctx.getTypeElement(rawType);
211+
212+
FormBeanReader form = new FormBeanReader(ctx, formBeanType, varName, shortType, defaultParamType);
213+
form.write(writer);
214+
}
215+
}

‎src/main/java/io/dinject/javalin/generator/FormBeanReader.java‎

Lines changed: 118 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,118 @@
1+
package io.dinject.javalin.generator;
2+
3+
import javax.lang.model.element.Element;
4+
import javax.lang.model.element.TypeElement;
5+
import java.util.HashSet;
6+
import java.util.LinkedHashMap;
7+
import java.util.List;
8+
import java.util.Map;
9+
import java.util.Set;
10+
11+
class FormBeanReader {
12+
13+
private final ProcessingContext ctx;
14+
private final String beanVarName;
15+
private final String beanShortType;
16+
private final TypeElement beanType;
17+
18+
private final ParamType defaultParamType;
19+
20+
private final Set<String> setterMethods = new HashSet<>();
21+
22+
private final Map<String, FieldReader> fieldMap = new LinkedHashMap<>();
23+
24+
FormBeanReader(ProcessingContext ctx, TypeElement beanType, String beanVarName, String beanShortType, ParamType defaultParamType) {
25+
this.ctx = ctx;
26+
this.beanType = beanType;
27+
this.beanVarName = beanVarName;
28+
this.beanShortType = beanShortType;
29+
this.defaultParamType = defaultParamType;
30+
31+
read();
32+
}
33+
34+
private void read() {
35+
36+
final List<? extends Element> enclosedElements = beanType.getEnclosedElements();
37+
for (Element enclosedElement : enclosedElements) {
38+
switch (enclosedElement.getKind()) {
39+
case METHOD:
40+
readMethod(enclosedElement);
41+
break;
42+
case FIELD:
43+
readField(enclosedElement);
44+
break;
45+
}
46+
}
47+
}
48+
49+
private void readField(Element enclosedElement) {
50+
51+
FieldReader field = new FieldReader(ctx, enclosedElement, defaultParamType);
52+
fieldMap.put(field.getVarName(), field);
53+
}
54+
55+
private void readMethod(Element enclosedElement) {
56+
String simpleName = enclosedElement.getSimpleName().toString();
57+
if (simpleName.startsWith("set")) {
58+
setterMethods.add(simpleName);
59+
}
60+
}
61+
62+
void write(Append writer) {
63+
64+
writer.append(" new %s();", beanShortType).eol();
65+
66+
for (String setterMethod : setterMethods) {
67+
String propName = Util.propertyName(setterMethod);
68+
FieldReader field = fieldMap.get(propName);
69+
if (field != null) {
70+
field.setUseSetter(setterMethod);
71+
}
72+
}
73+
74+
for (FieldReader field : fieldMap.values()) {
75+
field.writeSet(writer, beanVarName);
76+
}
77+
writer.eol();
78+
}
79+
80+
static class FieldReader {
81+
82+
private final ElementReader element;
83+
84+
private String setterMethod;
85+
86+
FieldReader(ProcessingContext ctx, Element enclosedElement, ParamType defaultParamType) {
87+
this.element = new ElementReader(enclosedElement, ctx, defaultParamType);
88+
}
89+
90+
String getVarName() {
91+
return element.getVarName();
92+
}
93+
94+
@Override
95+
public String toString() {
96+
return element.toString();
97+
}
98+
99+
void writeSet(Append writer, String beanVarName) {
100+
if (setterMethod != null) {
101+
writer.append(" %s.%s(", beanVarName, setterMethod);
102+
element.setValue(writer);
103+
writer.append(");").eol();
104+
105+
} else {
106+
writer.append(" %s.%s = ", beanVarName, getVarName());
107+
element.setValue(writer);
108+
writer.append(";").eol();
109+
}
110+
}
111+
112+
void setUseSetter(String setterMethod) {
113+
this.setterMethod = setterMethod;
114+
}
115+
}
116+
117+
118+
}

0 commit comments

Comments
 (0)