use of org.ballerinalang.util.codegen.AnnAttachmentInfo in project ballerina by ballerina-lang.
the class HttpUtil method addHTTPSessionAndCorsHeaders.
public static void addHTTPSessionAndCorsHeaders(Context context, HTTPCarbonMessage requestMsg, HTTPCarbonMessage responseMsg) {
Session session = (Session) requestMsg.getProperty(HttpConstants.HTTP_SESSION);
if (session != null) {
boolean isSecureRequest = false;
AnnAttachmentInfo configAnn = context.getServiceInfo().getAnnotationAttachmentInfo(HttpConstants.PROTOCOL_PACKAGE_HTTP, HttpConstants.ANN_NAME_CONFIG);
if (configAnn != null) {
AnnAttributeValue httpsPortAttrVal = configAnn.getAttributeValue(HttpConstants.ANN_CONFIG_ATTR_HTTPS_PORT);
if (httpsPortAttrVal != null) {
Integer listenerPort = (Integer) requestMsg.getProperty(HttpConstants.LISTENER_PORT);
if (listenerPort != null && httpsPortAttrVal.getIntValue() == listenerPort) {
isSecureRequest = true;
}
}
}
session.generateSessionHeader(responseMsg, isSecureRequest);
}
// Process CORS if exists.
if (requestMsg.getHeader(HttpHeaderNames.ORIGIN.toString()) != null) {
CorsHeaderGenerator.process(requestMsg, responseMsg, true);
}
}
use of org.ballerinalang.util.codegen.AnnAttachmentInfo in project ballerina by ballerina-lang.
the class SwaggerResourceMapper method createHeadersModel.
/**
* Creates headers definitions for swagger response.
*
* @param annotationAttributeValue The annotation attribute value which has the headers.
* @param response The swagger response.
*/
private void createHeadersModel(AnnAttributeValue annotationAttributeValue, Response response) {
if (null != annotationAttributeValue) {
AnnAttributeValue[] headersValueArray = annotationAttributeValue.getAttributeValueArray();
for (AnnAttributeValue headersValue : headersValueArray) {
AnnAttachmentInfo headerAnnotationAttachment = headersValue.getAnnotationAttachmentValue();
Map<String, AnnAttributeValue> headerAnnAttributeValueMap = SwaggerUtils.convertToAttributeMap(headerAnnotationAttachment);
Map<String, Property> headers = new HashMap<>();
if (null != headerAnnAttributeValueMap.get("name") && null != headerAnnAttributeValueMap.get("headerType")) {
String headerName = headerAnnAttributeValueMap.get("name").getStringValue();
String type = headerAnnAttributeValueMap.get("headerType").getStringValue();
Property property = null;
if ("string".equals(type)) {
property = new StringProperty();
} else if ("number".equals(type) || "integer".equals(type)) {
property = new IntegerProperty();
} else if ("boolean".equals(type)) {
property = new BooleanProperty();
} else if ("array".equals(type)) {
property = new ArrayProperty();
}
if (null != property) {
if (null != headerAnnAttributeValueMap.get("description")) {
property.setDescription(headerAnnAttributeValueMap.get("description").getStringValue());
}
headers.put(headerName, property);
}
}
response.setHeaders(headers);
}
}
}
use of org.ballerinalang.util.codegen.AnnAttachmentInfo in project ballerina by ballerina-lang.
the class SwaggerResourceMapper method addResourceParameters.
/**
* Creates parameters in the swagger operation using the parameters in the ballerina resource definition.
*
* @param resource The ballerina resource definition.
* @param operationAdaptor The swagger operation.
*/
private void addResourceParameters(ResourceInfo resource, OperationAdaptor operationAdaptor) {
if (!"get".equalsIgnoreCase(operationAdaptor.getHttpOperation())) {
// Creating message body - required.
ModelImpl messageModel = new ModelImpl();
messageModel.setType("object");
Map<String, Model> definitions = new HashMap<>();
if (!definitions.containsKey("Message")) {
definitions.put("Message", messageModel);
this.swaggerDefinition.setDefinitions(definitions);
}
// Creating "Message m" parameter
BodyParameter messageParameter = new BodyParameter();
messageParameter.setName(resource.getParamNames()[0]);
RefModel refModel = new RefModel();
refModel.setReference("Message");
messageParameter.setSchema(refModel);
operationAdaptor.getOperation().addParameter(messageParameter);
}
// Creating query params and path params
AttributeInfo attributeInfo = resource.getAttributeInfo(AttributeInfo.Kind.PARAMETER_ANNOTATIONS_ATTRIBUTE);
if (attributeInfo instanceof ParamAnnotationAttributeInfo) {
ParamAnnotationAttributeInfo paramAttributeInfo = (ParamAnnotationAttributeInfo) resource.getAttributeInfo(AttributeInfo.Kind.PARAMETER_ANNOTATIONS_ATTRIBUTE);
ParamAnnAttachmentInfo[] attachmentInfoArray = paramAttributeInfo.getAttachmentInfoArray();
for (ParamAnnAttachmentInfo paramAnnAttachmentInfo : attachmentInfoArray) {
if (paramAnnAttachmentInfo.getAnnAttachmentInfos().length > 0) {
AnnAttachmentInfo annAttachmentInfo = paramAnnAttachmentInfo.getAnnAttachmentInfos()[0];
Map<String, AnnAttributeValue> paramAnnAttributeValueMap = SwaggerUtils.convertToAttributeMap(annAttachmentInfo);
if (paramAnnAttributeValueMap.size() == 1 && null != paramAnnAttributeValueMap.get("value")) {
// Add query parameter
if (annAttachmentInfo.getName().equalsIgnoreCase("QueryParam")) {
QueryParameter queryParameter = new QueryParameter();
// Set in value.
queryParameter.setIn("query");
// Set parameter name
String parameterName = paramAnnAttributeValueMap.get("value").getStringValue();
if ((parameterName == null) || parameterName.isEmpty()) {
parameterName = resource.getParamNames()[paramAnnAttachmentInfo.getParamIdex()];
}
queryParameter.setName(parameterName);
// Note: 'description' to be added using annotations, hence skipped here.
// Setting false to required(as per swagger spec). This can be overridden while parsing
// annotations.
queryParameter.required(false);
// Note: 'allowEmptyValue' to be added using annotations, hence skipped here.
// Set type
String paramType = resource.getParamTypes()[paramAnnAttachmentInfo.getParamIdex()].getName();
if ("int".equals(paramType)) {
queryParameter.setType("integer");
} else {
queryParameter.setType(paramType);
}
// Note: 'format' to be added using annotations, hence skipped here.
operationAdaptor.getOperation().addParameter(queryParameter);
}
if (annAttachmentInfo.getName().equalsIgnoreCase("PathParam")) {
PathParameter pathParameter = new PathParameter();
// Set in value
pathParameter.setIn("path");
// Set parameter name
String parameterName = paramAnnAttributeValueMap.get("value").getStringValue();
if ((parameterName == null) || parameterName.isEmpty()) {
parameterName = resource.getParamNames()[paramAnnAttachmentInfo.getParamIdex()];
}
pathParameter.setName(parameterName);
// Note: 'description' to be added using annotations, hence skipped here.
// Note: 'allowEmptyValue' to be added using annotations, hence skipped here.
// Set type
String paramType = resource.getParamTypes()[paramAnnAttachmentInfo.getParamIdex()].getName();
if ("int".equals(paramType)) {
pathParameter.setType("integer");
} else {
pathParameter.setType(paramType);
}
// Note: 'format' to be added using annotations, hence skipped here.
operationAdaptor.getOperation().addParameter(pathParameter);
}
}
}
}
}
}
use of org.ballerinalang.util.codegen.AnnAttachmentInfo in project ballerina by ballerina-lang.
the class SwaggerResourceMapper method createExternalDocsModel.
/**
* Creates external docs swagger definitions.
*
* @param annotationAttributeValue The annotation attribute value for external docs.
* @param operation The swagger operation.
*/
private void createExternalDocsModel(AnnAttributeValue annotationAttributeValue, Operation operation) {
if (null != annotationAttributeValue) {
AnnAttachmentInfo externalDocAnnotationAttachment = annotationAttributeValue.getAnnotationAttachmentValue();
ExternalDocs externalDocs = new ExternalDocs();
Map<String, AnnAttributeValue> externalDocsAnnAttributeValueMap = SwaggerUtils.convertToAttributeMap(externalDocAnnotationAttachment);
if (null != externalDocsAnnAttributeValueMap.get("description")) {
externalDocs.setDescription(externalDocsAnnAttributeValueMap.get("description").getStringValue());
}
if (null != externalDocsAnnAttributeValueMap.get("url")) {
externalDocs.setUrl(externalDocsAnnAttributeValueMap.get("url").getStringValue());
}
operation.setExternalDocs(externalDocs);
}
}
use of org.ballerinalang.util.codegen.AnnAttachmentInfo in project ballerina by ballerina-lang.
the class SwaggerResourceMapper method parseConsumesAnnotationAttachment.
/**
* Parses the consumes annotation attachments and updates the swagger operation.
*
* @param resource The ballerina resource definition.
* @param operation The swagger operation.
*/
private void parseConsumesAnnotationAttachment(ResourceInfo resource, Operation operation) {
AnnAttachmentInfo rConfigAnnAtchmnt = resource.getAnnotationAttachmentInfo(HttpConstants.HTTP_PACKAGE_PATH, HttpConstants.ANN_NAME_RESOURCE_CONFIG);
if (rConfigAnnAtchmnt == null) {
return;
}
AnnAttributeValue consumesAttrVal = rConfigAnnAtchmnt.getAttributeValue(HttpConstants.ANN_RESOURCE_ATTR_CONSUMES);
if (consumesAttrVal == null) {
return;
}
List<String> consumes = getStringList(consumesAttrVal.getAttributeValueArray());
operation.setConsumes(consumes);
}
Aggregations