[c78deb]: src / main / java / net / timbusproject / extractors / modules / tavernaextractor / TavernaExtractorOSGi.java History

Parent: [0acb23] (diff)

Download this file

TavernaExtractorOSGi.java    113 lines (91 with data), 5.1 kB 

  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
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
/**

 * Copyright (c) 2013/2014 Verein zur Foerderung der IT-Sicherheit in Oesterreich (SBA).

 * The work has been developed in the TIMBUS Project and the above-mentioned are Members of the TIMBUS Consortium.

 * TIMBUS is supported by the European Union under the 7th Framework Programme for research and technological

 * development and demonstration activities (FP7/2007-2013) under grant agreement no. 269940.

 *

 * 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, including without

 * limitation, any warranties or conditions of TITLE, NON-INFRINGEMENT, MERCHANTIBITLY, or FITNESS FOR A PARTICULAR

 * PURPOSE. In no event and under no legal theory, whether in tort (including negligence), contract, or otherwise,

 * unless required by applicable law or agreed to in writing, shall any Contributor be liable for damages, including

 * any direct, indirect, special, incidental, or consequential damages of any character arising as a result of this

 * License or out of the use or inability to use the Work.

 * See the License for the specific language governing permissions and limitation under the License.

 */



package net.timbusproject.extractors.modules.tavernaextractor;



import net.timbusproject.extractors.core.*;

import net.timbusproject.extractors.modules.tavernaextractor.utils.SSHManager;

import org.codehaus.jettison.json.JSONObject;

import org.osgi.framework.BundleContext;

import org.osgi.framework.Version;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;



import org.springframework.beans.factory.annotation.Autowired;



import java.nio.file.Paths;

import java.util.EnumSet;

import java.util.HashMap;



/**

 * munterberger@sba-research.org

 */

public class TavernaExtractorOSGi implements IExtractor{



    public TavernaExtractorOSGi() {}



    @Autowired

    private BundleContext bundleContext;



    @Override

    public String getName() {

        return bundleContext != null ? bundleContext.getBundle().getHeaders().get("Bundle-Name") : getClass().getSimpleName();

    }



    @Override

    public String getSymbolicName() {

        return bundleContext != null ? bundleContext.getBundle().getSymbolicName() : getClass().getCanonicalName();

    }



    @Override

    public Version getVersion() {

        return bundleContext != null ? bundleContext.getBundle().getVersion() : Version.emptyVersion;

    }



    @Override

    public HashMap<String, Parameter> getParameters() {

        HashMap<String, Parameter> parameters = new HashMap<String, Parameter>();



        Parameter defaultParameter = new Parameter(false, true, ParameterType.STRING);

        parameters.put("user", new Parameter(false));

        parameters.put("password", new Parameter(true));

        parameters.put("port", new Parameter(false, ParameterType.NUMBER));

        parameters.put("fqdn", new Parameter(false));

        parameters.put("tavernaHome", defaultParameter);

        parameters.put("archimateOutputPath", defaultParameter);

        parameters.put("inputFileName", defaultParameter);



        return parameters;

    }



    @Override

    public EnumSet<OperatingSystem> getSupportedOperatingSystems() {

        return EnumSet.of(OperatingSystem.LINUX);

    }



    @Override

    public String extract(Endpoint endpoint, boolean b) throws Exception {



        String tavernaHome = endpoint.getProperty("tavernaHome");

        String archimateOutputPath = endpoint.getProperty("archimateOutputPath");

        String inputFileName = endpoint.getProperty("inputFileName");



        SSHManager _sshManager = new SSHManager(endpoint.getProperty("fqdn")

                ,new Integer(endpoint.getProperty("port")),

                endpoint.getProperty("user"),

                endpoint.getProperty("password"));



        try {

            TavernaExtractor extractor = new TavernaExtractor(_sshManager);

            extractor.setTavernaHome(Paths.get(tavernaHome));

            extractor.setArchimateOutputPath(Paths.get(archimateOutputPath));

            extractor.setInputFileName(Paths.get(inputFileName));



            // starts the execution

            extractor.process(true); //true = remote execution



            // inform user about the result of the extraction

            String fileContent = extractor.getConvertedFileAsString();

            return new JSONObject().put("DONE", "Extraction was successfully!").put("Archimate file is available at ", endpoint.getProperty("fqdn") + archimateOutputPath).put("file content",fileContent).toString();

        }



        // !STUPID! But in the current TIMBUS testbed, error infos can not be broadcasted via user interface other than that.

        catch(Exception e){

            return new JSONObject().put("ERROR", e.getLocalizedMessage()).toString();

        }

    }

}