1 files changed, 148 insertions, 0 deletions
diff --git a/ms/blueprintsprocessor/modules/blueprints/blueprint-core/src/main/kotlin/org/onap/ccsdk/cds/controllerblueprints/core/BluePrintExtensionFunctions.kt b/ms/blueprintsprocessor/modules/blueprints/blueprint-core/src/main/kotlin/org/onap/ccsdk/cds/controllerblueprints/core/BluePrintExtensionFunctions.kt
new file mode 100644
index 000000000..daf833494
--- /dev/null
+++ b/ms/blueprintsprocessor/modules/blueprints/blueprint-core/src/main/kotlin/org/onap/ccsdk/cds/controllerblueprints/core/BluePrintExtensionFunctions.kt
@@ -0,0 +1,148 @@
+/*
+ *  Copyright © 2019 IBM.
+ *
+ *  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 org.onap.ccsdk.cds.controllerblueprints.core
+
+import org.onap.ccsdk.cds.controllerblueprints.core.annotations.ArtifactExpression
+import org.onap.ccsdk.cds.controllerblueprints.core.annotations.AttributeExpression
+import org.onap.ccsdk.cds.controllerblueprints.core.annotations.BluePrintsConstrain
+import org.onap.ccsdk.cds.controllerblueprints.core.annotations.BluePrintsDataType
+import org.onap.ccsdk.cds.controllerblueprints.core.annotations.BluePrintsProperty
+import org.onap.ccsdk.cds.controllerblueprints.core.annotations.DSLExpression
+import org.onap.ccsdk.cds.controllerblueprints.core.annotations.InputExpression
+import org.onap.ccsdk.cds.controllerblueprints.core.annotations.OperationOutputExpression
+import org.onap.ccsdk.cds.controllerblueprints.core.annotations.PropertyDefaultValue
+import org.onap.ccsdk.cds.controllerblueprints.core.annotations.PropertyExpression
+import org.onap.ccsdk.cds.controllerblueprints.core.data.ConstraintClause
+import org.onap.ccsdk.cds.controllerblueprints.core.data.DataType
+import org.onap.ccsdk.cds.controllerblueprints.core.data.EntrySchema
+import org.onap.ccsdk.cds.controllerblueprints.core.data.PropertyDefinition
+import org.onap.ccsdk.cds.controllerblueprints.core.dsl.dslExpression
+import org.onap.ccsdk.cds.controllerblueprints.core.dsl.getInput
+import org.onap.ccsdk.cds.controllerblueprints.core.dsl.getNodeTemplateArtifact
+import org.onap.ccsdk.cds.controllerblueprints.core.dsl.getNodeTemplateAttribute
+import org.onap.ccsdk.cds.controllerblueprints.core.dsl.getNodeTemplateOperationOutput
+import org.onap.ccsdk.cds.controllerblueprints.core.dsl.getNodeTemplateProperty
+import kotlin.reflect.KClass
+import kotlin.reflect.KProperty1
+import kotlin.reflect.KType
+import kotlin.reflect.full.declaredMemberProperties
+
+fun <T : KClass<*>> T.asBluePrintsDataTypes(): DataType {
+    val annotation = this.annotations.filter { it is BluePrintsDataType }.single() as BluePrintsDataType
+    checkNotNull(annotation) { "BluePrintsDataType annotation definition not found" }
+    val dataType = DataType().apply {
+        id = annotation.name
+        version = annotation.version
+        derivedFrom = annotation.derivedFrom
+        description = annotation.description
+    }
+    dataType.properties = this.asPropertyDefinitionMap()
+    return dataType
+}
+
+fun <T : KClass<*>> T.asPropertyDefinitionMap(): MutableMap<String, PropertyDefinition> {
+    val properties: MutableMap<String, PropertyDefinition> = hashMapOf()
+    this.declaredMemberProperties.forEach { member ->
+        properties[member.name] = member.asPropertyDefinition()
+    }
+    return properties
+}
+
+fun <T> KProperty1<T, *>.asPropertyDefinition(): PropertyDefinition {
+    val property = PropertyDefinition()
+    property.id = this.name
+    val getter = this.getter
+    property.required = !this.returnType.isMarkedNullable
+    property.type = this.returnType.asBluePrintsDataType(this.name)
+    if (this.returnType.arguments.isNotEmpty()) {
+        property.entrySchema = this.returnType.entitySchema()
+    }
+    this.annotations.forEach { fieldAnnotation ->
+        // println("Field : ${this.name} : Annotation : $fieldAnnotation")
+        when (fieldAnnotation) {
+            is BluePrintsProperty ->
+                property.description = fieldAnnotation.description
+            is PropertyDefaultValue ->
+                property.value = fieldAnnotation.value.asJsonType(property.type)
+            is BluePrintsConstrain -> {
+                if (property.constraints == null) property.constraints = arrayListOf()
+                property.constraints!!.add(fieldAnnotation.asBluePrintConstraintClause())
+            }
+            is InputExpression -> {
+                property.value = getInput(fieldAnnotation.propertyName)
+            }
+            is PropertyExpression -> {
+                property.value = getNodeTemplateProperty(
+                    fieldAnnotation.modelableEntityName,
+                    fieldAnnotation.propertyName, fieldAnnotation.subPropertyName
+                )
+            }
+            is AttributeExpression -> {
+                property.value = getNodeTemplateAttribute(
+                    fieldAnnotation.modelableEntityName,
+                    fieldAnnotation.attributeName, fieldAnnotation.subAttributeName
+                )
+            }
+            is ArtifactExpression -> {
+                property.value = getNodeTemplateArtifact(
+                    fieldAnnotation.modelableEntityName,
+                    fieldAnnotation.artifactName
+                )
+            }
+            is OperationOutputExpression -> {
+                property.value = getNodeTemplateOperationOutput(
+                    fieldAnnotation.modelableEntityName,
+                    fieldAnnotation.interfaceName, fieldAnnotation.propertyName, fieldAnnotation.subPropertyName
+                )
+            }
+            is DSLExpression -> {
+                property.value = dslExpression(fieldAnnotation.propertyName)
+            }
+        }
+    }
+    return property
+}
+
+internal fun BluePrintsConstrain.asBluePrintConstraintClause(): ConstraintClause {
+    TODO()
+}
+
+internal fun <T : KType> T.entitySchema(): EntrySchema {
+    val entrySchema = EntrySchema()
+    if (this.arguments.size == 1) {
+        entrySchema.type = this.arguments[0].type!!.asBluePrintsDataType("")
+    } else if (this.arguments.size == 2) {
+        entrySchema.type = this.arguments[1].type!!.asBluePrintsDataType("")
+    }
+    return entrySchema
+}
+
+internal fun <T : KType> T.asBluePrintsDataType(propertyName: String): String {
+    val simpleName = (this.classifier as? KClass<*>)?.java?.simpleName
+        ?: throw BluePrintException("filed to get simple name.")
+    return when (simpleName) {
+        "String", "Date" -> BluePrintConstants.DATA_TYPE_STRING
+        "int" -> BluePrintConstants.DATA_TYPE_INTEGER
+        "Boolean" -> BluePrintConstants.DATA_TYPE_BOOLEAN
+        "Float" -> BluePrintConstants.DATA_TYPE_FLOAT
+        "Double" -> BluePrintConstants.DATA_TYPE_DOUBLE
+        "List" -> BluePrintConstants.DATA_TYPE_LIST
+        "Map" -> BluePrintConstants.DATA_TYPE_MAP
+        "Object", "JsonNode", "ObjectNode", "ArrayNode" -> BluePrintConstants.DATA_TYPE_JSON
+        else -> simpleName
+    }
+}