aboutsummaryrefslogtreecommitdiffstats
path: root/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/services/heattotosca/impl/functiontranslation/FunctionTranslationGetFileImpl.java
blob: 253410e493c4678b61c3d68522e63e950b4b1701 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
/*
 * Copyright © 2016-2018 European Support Limited
 *
 * 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.openecomp.sdc.translator.services.heattotosca.impl.functiontranslation;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.onap.sdc.tosca.datatypes.model.ArtifactDefinition;
import org.onap.sdc.tosca.datatypes.model.NodeTemplate;
import org.openecomp.sdc.tosca.datatypes.ToscaArtifactType;
import org.openecomp.sdc.tosca.datatypes.ToscaFunctions;
import org.openecomp.sdc.tosca.services.ToscaConstants;
import org.openecomp.sdc.tosca.services.ToscaFileOutputService;
import org.openecomp.sdc.tosca.services.impl.ToscaFileOutputServiceCsarImpl;
import org.openecomp.sdc.translator.services.heattotosca.FunctionTranslation;

public class FunctionTranslationGetFileImpl implements FunctionTranslation {

    private static ArtifactDefinition createArtifactDefinition(Object function, ToscaFileOutputService toscaFileOutputService) {
        ArtifactDefinition artifactDefinition = new ArtifactDefinition();
        artifactDefinition.setType(ToscaArtifactType.NATIVE_DEPLOYMENT);
        artifactDefinition.setFile("../" + toscaFileOutputService.getArtifactsFolderName() + "/" + function);
        return artifactDefinition;
    }

    @Override
    public Object translateFunction(FunctionTranslator functionTranslator) {
        String file = ((String) functionTranslator.getFunctionValue()).replace("file:///", "");
        final String artifactId = file.split("\\.")[0];
        Map<String, Object> returnValue = new HashMap<>();
        List<String> artifactParameters = new ArrayList<>();
        artifactParameters.add(ToscaConstants.MODELABLE_ENTITY_NAME_SELF);
        returnValue.put(ToscaFunctions.GET_ARTIFACT.getFunctionName(), artifactParameters);
        artifactParameters.add(artifactId);
        ToscaFileOutputService toscaFileOutputService = new ToscaFileOutputServiceCsarImpl();
        if (functionTranslator.getToscaTemplate() instanceof NodeTemplate) {
            NodeTemplate nodeTemplate = (NodeTemplate) functionTranslator.getToscaTemplate();
            ArtifactDefinition artifactDefinition = createArtifactDefinition(file, toscaFileOutputService);
            if (nodeTemplate.getArtifacts() == null) {
                nodeTemplate.setArtifacts(new HashMap<>());
            }
            nodeTemplate.getArtifacts().put(artifactId, artifactDefinition);
        }
        return returnValue;
    }
}