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