801c22a11639f3b4b2c86f35f466f062d8bc9e65
[ccsdk/cds.git] /
1 /*
2  * Copyright © 2017-2018 AT&T Intellectual Property.
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.controllerblueprints.service.utils
18
19 import org.onap.ccsdk.apps.controllerblueprints.core.BluePrintException
20 import org.onap.ccsdk.apps.controllerblueprints.core.data.ArtifactType
21 import org.onap.ccsdk.apps.controllerblueprints.core.data.DataType
22 import org.onap.ccsdk.apps.controllerblueprints.core.data.NodeType
23 import org.onap.ccsdk.apps.controllerblueprints.core.format
24 import org.onap.ccsdk.apps.controllerblueprints.core.interfaces.BluePrintRepoService
25 import org.onap.ccsdk.apps.controllerblueprints.core.service.BluePrintContext
26
27
28 class BluePrintEnhancerUtils {
29     companion object {
30
31         fun populateDataTypes(bluePrintContext: BluePrintContext, bluePrintRepoService: BluePrintRepoService,
32                               dataTypeName: String): DataType {
33             val dataType = bluePrintContext.serviceTemplate.dataTypes?.get(dataTypeName)
34                     ?: bluePrintRepoService.getDataType(dataTypeName)
35                     ?: throw BluePrintException("couldn't get DataType($dataTypeName) from repo.")
36             bluePrintContext.serviceTemplate.dataTypes?.put(dataTypeName, dataType)
37             return dataType
38         }
39
40
41         fun populateNodeType(bluePrintContext: BluePrintContext, bluePrintRepoService: BluePrintRepoService,
42                              nodeTypeName: String): NodeType {
43
44             val nodeType = bluePrintContext.serviceTemplate.nodeTypes?.get(nodeTypeName)
45                     ?: bluePrintRepoService.getNodeType(nodeTypeName)
46                     ?: throw BluePrintException(format("Couldn't get NodeType({}) from repo.", nodeTypeName))
47             bluePrintContext.serviceTemplate.nodeTypes?.put(nodeTypeName, nodeType)
48             return nodeType
49         }
50
51         fun populateArtifactType(bluePrintContext: BluePrintContext, bluePrintRepoService: BluePrintRepoService,
52                                  artifactTypeName: String): ArtifactType {
53
54             val artifactType = bluePrintContext.serviceTemplate.artifactTypes?.get(artifactTypeName)
55                     ?: bluePrintRepoService.getArtifactType(artifactTypeName)
56                     ?: throw BluePrintException("couldn't get ArtifactType($artifactTypeName) from repo.")
57             bluePrintContext.serviceTemplate.artifactTypes?.put(artifactTypeName, artifactType)
58             return artifactType
59         }
60     }
61 }