be7451aa9cbfa9ffcd438213616132fb5bee7df6
[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(artifactMapping: String, artifactTemplate: String): String = runBlocking {
48         resourceResolutionService().resolveResources(bluePrintRuntimeService, nodeTemplateName,
49                 artifactMapping, artifactTemplate)
50     }
51
52     fun resolveAndGenerateMessage(artifactPrefix: String): String = runBlocking {
53         resourceResolutionService().resolveResources(bluePrintRuntimeService, nodeTemplateName,
54                 artifactPrefix, mapOf())
55     }
56
57     private fun deviceProperties(requirementName: String): DeviceInfo {
58
59         val blueprintContext = bluePrintRuntimeService.bluePrintContext()
60
61         val requirement = blueprintContext.nodeTemplateRequirement(nodeTemplateName, requirementName)
62
63         val capabilityProperties = bluePrintRuntimeService.resolveNodeTemplateCapabilityProperties(requirement
64                 .node!!, requirement.capability!!)
65
66         return deviceProperties(capabilityProperties)
67     }
68
69     private fun deviceProperties(capabilityProperty: MutableMap<String, JsonNode>): DeviceInfo {
70         return JacksonUtils.getInstanceFromMap(capabilityProperty, DeviceInfo::class.java)
71     }
72 }