Migrate "ms/controllerblueprints" from ccsdk/apps
[ccsdk/cds.git] / ms / blueprintsprocessor / functions / netconf-executor / src / main / kotlin / org / onap / ccsdk / apps / blueprintsprocessor / functions / netconf / executor / NetconfComponentFunction.kt
1 /*
2  *  Copyright © 2018-2019 IBM, Bell Canada.
3  *
4  *  Licensed under the Apache License, Version 2.0 (the "License");
5  *  you may not use this file except in compliance with the License.
6  *  You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  *  Unless required by applicable law or agreed to in writing, software
11  *  distributed under the License is distributed on an "AS IS" BASIS,
12  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  *  See the License for the specific language governing permissions and
14  *  limitations under the License.
15  */
16
17 package org.onap.ccsdk.apps.blueprintsprocessor.functions.netconf.executor
18
19 import com.fasterxml.jackson.databind.JsonNode
20 import org.onap.ccsdk.apps.blueprintsprocessor.functions.netconf.executor.api.DeviceInfo
21 import org.onap.ccsdk.apps.blueprintsprocessor.functions.resource.resolution.ResourceResolutionConstants
22 import org.onap.ccsdk.apps.blueprintsprocessor.functions.resource.resolution.ResourceResolutionService
23 import org.onap.ccsdk.apps.blueprintsprocessor.services.execution.AbstractScriptComponentFunction
24 import org.onap.ccsdk.apps.controllerblueprints.core.utils.JacksonUtils
25
26 abstract class NetconfComponentFunction : AbstractScriptComponentFunction() {
27
28     open fun resourceResolutionService(): ResourceResolutionService =
29             functionDependencyInstanceAsType(ResourceResolutionConstants.SERVICE_RESOURCE_RESOLUTION)
30
31     // Called from python script
32     fun initializeNetconfConnection(requirementName: String): NetconfDevice {
33         val deviceInfo = deviceProperties(requirementName)
34         return NetconfDevice(deviceInfo)
35     }
36
37     fun generateMessage(artifactName: String): String {
38         return bluePrintRuntimeService.resolveNodeTemplateArtifact(nodeTemplateName, artifactName)
39     }
40
41     fun resolveFromDatabase(resolutionKey: String, artifactName: String): String {
42         return resourceResolutionService().resolveFromDatabase(bluePrintRuntimeService, artifactName, resolutionKey)
43     }
44
45     fun resolveAndGenerateMessage(artifactMapping: String, artifactTemplate: String): String {
46         return resourceResolutionService().resolveResources(bluePrintRuntimeService, nodeTemplateName,
47                 artifactMapping, artifactTemplate)
48     }
49
50     fun resolveAndGenerateMessage(artifactPrefix: String): String {
51         return resourceResolutionService().resolveResources(bluePrintRuntimeService, nodeTemplateName,
52                 artifactPrefix, mapOf())
53     }
54
55     private fun deviceProperties(requirementName: String): DeviceInfo {
56
57         val blueprintContext = bluePrintRuntimeService.bluePrintContext()
58
59         val requirement = blueprintContext.nodeTemplateRequirement(nodeTemplateName, requirementName)
60
61         val capabilityProperties = bluePrintRuntimeService.resolveNodeTemplateCapabilityProperties(requirement
62                 .node!!, requirement.capability!!)
63
64         return deviceProperties(capabilityProperties)
65     }
66
67     private fun deviceProperties(capabilityProperty: MutableMap<String, JsonNode>): DeviceInfo {
68         return JacksonUtils.getInstanceFromMap(capabilityProperty, DeviceInfo::class.java)
69     }
70 }