[vfc/nfvo/catalog.git] /

/**
 * Copyright 2016 [ZTE] and others.
 *
 * 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.openo.commontosca.catalog.model.plan.wso2;

import com.google.gson.Gson;

import com.eclipsesource.jaxrs.consumer.ConsumerFactory;

import org.apache.http.HttpEntity;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.mime.MultipartEntityBuilder;
import org.glassfish.jersey.client.ClientConfig;
import org.openo.commontosca.catalog.common.Config;
import org.openo.commontosca.catalog.db.exception.CatalogResourceException;
import org.openo.commontosca.catalog.model.plan.wso2.entity.DeletePackageResponse;
import org.openo.commontosca.catalog.model.plan.wso2.entity.DeployPackageResponse;
import org.openo.commontosca.catalog.model.plan.wso2.entity.StartProcessRequest;
import org.openo.commontosca.catalog.model.plan.wso2.entity.StartProcessResponse;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipInputStream;



public class Wso2ServiceConsumer {
  public static final String WSO2_APP_URL = "/openoapi/wso2bpel/v1/package";
  private static final Logger LOGGER = LoggerFactory.getLogger(Wso2ServiceConsumer.class);
  
  /**
   * deploy package.
   * @param zipFileLocation zip file location
   * @param planFilePath plan file path
   * @return DeployPackageResponse
   * @throws CatalogResourceException e1
   */
  public static DeployPackageResponse deployPackage(String zipFileLocation, String planFilePath)
      throws CatalogResourceException {
    InputStream ins = null;
    try {
      ins = getInputStream(zipFileLocation, planFilePath);
      RestResponse res = RestfulClient.post(Config.getConfigration().getWso2HostIp(),
          Integer.parseInt(Config.getConfigration().getWso2HostPort()), WSO2_APP_URL,
          buildRequest(ins, planFilePath));

      if (res.getStatusCode() == null || res.getResult() == null) {
        throw new CatalogResourceException(
            "Deploy Package return null. Response = " + res);
      }
      
      if (200 == res.getStatusCode() || 201 == res.getStatusCode()) {
        DeployPackageResponse response =
            new Gson().fromJson(res.getResult(), DeployPackageResponse.class);
        if (response.isSuccess()) {
          return response;
        }
      }

      throw new CatalogResourceException(
          "Deploy Package return fail. Response = " + res.getResult());
    } catch (FileNotFoundException e1) {
      throw new CatalogResourceException("Deploy Package failed.", e1);
    } finally {
      if (ins != null) {
        try {
          ins.close();
        } catch (IOException e1) {
          LOGGER.error("inputStream close failed !");
        }
      }
    }
  }

  private static HttpEntity buildRequest(InputStream inputStream, String filePath)
      throws FileNotFoundException {
    MultipartEntityBuilder builder = MultipartEntityBuilder.create();
    builder.seContentType(ContentType.MULTIPART_FORM_DATA);
    builder.addBinaryBody("file", inputStream, ContentType.APPLICATION_OCTET_STREAM,
        new File(filePath).getName());
    return builder.build();
  }

  @SuppressWarnings("resource")
  private static InputStream getInputStream(String zipFileLocation, String planFilePath)
      throws CatalogResourceException {
    ZipInputStream zin = null;
    try {
      InputStream in = new BufferedInputStream(new FileInputStream(zipFileLocation));
      zin = new ZipInputStream(in);
      ZipEntry ze;
      while ((ze = zin.getNextEntry()) != null) {
        if (planFilePath.equals(ze.getName())) {
          ZipFile zf = new ZipFile(zipFileLocation);
          return zf.getInputStream(ze);
        }
      }
    } catch (IOException e1) {
      throw new CatalogResourceException("Get InputStream failed. planFilePath = " + planFilePath,
          e1);
    } finally {
      closeStream(zin);
    }

    throw new CatalogResourceException("Get InputStream failed. planFilePath = " + planFilePath);
  }

  private static void closeStream(ZipInputStream zin) {

    if (zin != null) {
      try {
        zin.closeEntry();
      } catch (IOException e1) {
        LOGGER.error("zip inputStream close failed !");
      }
    }
  }

  /**
   * delet package.
   * @param packageName package to delete according packageName
   * @return DeletePackageResponse
   * @throws CatalogResourceException e1
   */
  public static DeletePackageResponse deletePackage(String packageName)
      throws CatalogResourceException {
    try {
      ClientConfig config = new ClientConfig();
      Iwso2RestService wso2Proxy = ConsumerFactory.createConsumer(
          Config.getConfigration().getWso2BaseUrl(), config, Iwso2RestService.class);
      DeletePackageResponse response = wso2Proxy.deletePackage(packageName);
      if (response.isSuccess()) {
        return response;
      }
      throw new CatalogResourceException(response.getException());
    } catch (Exception e1) {
      throw new CatalogResourceException(
          "Call Delete Package api failed. packageName = " + packageName, e1);
    }
  }


  /**
   * start process.
   * @param processId process id
   * @param params params
   * @return StartProcessResponse
   * @throws CatalogResourceException e1
   */
  public static StartProcessResponse startProcess(String processId, Map<String, Object> params)
      throws CatalogResourceException {
    try {
      ClientConfig config = new ClientConfig();
      Iwso2RestService wso2Proxy = ConsumerFactory.createConsumer(
          Config.getConfigration().getWso2BaseUrl(), config, Iwso2RestService.class);
      StartProcessResponse response =
          wso2Proxy.startProcess(new StartProcessRequest(processId, params));
      if (response.isSuccess()) {
        return response;
      }
      throw new CatalogResourceException(response.getException());
    } catch (Exception e1) {
      throw new CatalogResourceException("Call Start Process api failed.", e1);
    }
  }

}