/*-
 * ============LICENSE_START=======================================================
 * SDC
 * ================================================================================
 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * 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.
 * ============LICENSE_END=========================================================
 */

package org.openecomp.sdcrests.vsp.rest.services;

import org.apache.cxf.jaxrs.ext.multipart.Attachment;
import org.openecomp.sdc.logging.api.Logger;
import org.openecomp.sdc.logging.api.LoggerFactory;
import org.openecomp.sdc.logging.context.MdcUtil;
import org.openecomp.sdc.logging.messages.AuditMessages;
import org.openecomp.sdc.logging.types.LoggerServiceName;
import org.openecomp.sdc.vendorsoftwareproduct.ComponentManager;
import org.openecomp.sdc.vendorsoftwareproduct.ComponentManagerFactory;
import org.openecomp.sdc.vendorsoftwareproduct.ProcessManager;
import org.openecomp.sdc.vendorsoftwareproduct.ProcessManagerFactory;
import org.openecomp.sdc.vendorsoftwareproduct.VendorSoftwareProductConstants;
import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ProcessEntity;
import org.openecomp.sdc.versioning.dao.types.Version;
import org.openecomp.sdc.versioning.types.VersionableEntityAction;
import org.openecomp.sdcrests.vendorsoftwareproducts.types.ProcessEntityDto;
import org.openecomp.sdcrests.vendorsoftwareproducts.types.ProcessRequestDto;
import org.openecomp.sdcrests.vsp.rest.ComponentProcesses;
import org.openecomp.sdcrests.vsp.rest.mapping.MapProcessEntityToProcessEntityDto;
import org.openecomp.sdcrests.vsp.rest.mapping.MapProcessRequestDtoToProcessEntity;
import org.openecomp.sdcrests.wrappers.GenericCollectionWrapper;
import org.openecomp.sdcrests.wrappers.StringWrapperResponse;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Service;

import javax.inject.Named;
import javax.ws.rs.core.Response;
import java.io.File;
import java.io.InputStream;
import java.util.Collection;

import static org.openecomp.sdc.vendorsoftwareproduct.VendorSoftwareProductConstants.GENERAL_COMPONENT_ID;

@Named
@Service("componentProcesses")
@Scope(value = "prototype")
public class ComponentProcessesImpl implements ComponentProcesses {

  private ProcessManager processManager = ProcessManagerFactory.getInstance().createInterface();
  private ComponentManager componentManager =
      ComponentManagerFactory.getInstance().createInterface();
    private static final Logger logger =
            LoggerFactory.getLogger(ComponentProcessesImpl.class);

  @Override
  public Response list(String vspId, String versionId, String componentId, String user) {
    MdcUtil.initMdc(LoggerServiceName.List_Component_Processes.toString());
    Version vspVersion = resolveVspVersion(vspId, versionId, user, VersionableEntityAction.Read);
    validateComponentExistence(vspId, vspVersion, componentId, user);
    Collection<ProcessEntity> processes;
    if (componentId.equals(VendorSoftwareProductConstants.GENERAL_COMPONENT_ID)) {
      processes = processManager.listProcesses(vspId, vspVersion, null, user);
    } else {
      processes = processManager.listProcesses(vspId, vspVersion, componentId, user);
    }


    MapProcessEntityToProcessEntityDto mapper = new MapProcessEntityToProcessEntityDto();
    GenericCollectionWrapper<ProcessEntityDto> results = new GenericCollectionWrapper<>();
    for (ProcessEntity process : processes) {
      results.add(mapper.applyMapping(process, ProcessEntityDto.class));
    }

    return Response.ok(results).build();
  }

  @Override
  public Response deleteList(String vspId, String versionId, String componentId, String user) {
    MdcUtil.initMdc(LoggerServiceName.Delete_List_Component_Processes.toString());
    Version version = resolveVspVersion(vspId, null, user, VersionableEntityAction.Write);
    validateComponentExistence(vspId, version, componentId, user);
    processManager.deleteProcesses(vspId, version, componentId, user);

    return Response.ok().build();
  }

  @Override
  public Response create(ProcessRequestDto request, String vspId, String versionId,
                         String componentId, String user) {
    MdcUtil.initMdc(LoggerServiceName.Create_Component_Processes.toString());
    ProcessEntity process =
        new MapProcessRequestDtoToProcessEntity().applyMapping(request, ProcessEntity.class);
    process.setVspId(vspId);
    process.setVersion(resolveVspVersion(vspId, null, user, VersionableEntityAction.Write));
    if (!componentId.equals(VendorSoftwareProductConstants.GENERAL_COMPONENT_ID)) {
      process.setComponentId(componentId);
    }

    validateComponentExistence(vspId, process.getVersion(), componentId, user);
    ProcessEntity createdProcess = processManager.createProcess(process, user);

    return Response
        .ok(createdProcess != null ? new StringWrapperResponse(createdProcess.getId()) : null)
        .build();
  }

  @Override
  public Response get(String vspId, String versionId, String componentId, String processId,
                      String user) {
    MdcUtil.initMdc(LoggerServiceName.Get_Component_Processes.toString());
    Version vspVersion = resolveVspVersion(vspId, versionId, user, VersionableEntityAction.Read);
    validateComponentExistence(vspId, vspVersion, componentId, user);
    ProcessEntity process =
        processManager.getProcess(vspId, vspVersion, componentId, processId, user);
    ProcessEntityDto result =
        new MapProcessEntityToProcessEntityDto().applyMapping(process, ProcessEntityDto.class);
    return Response.ok(result).build();
  }

  @Override
  public Response delete(String vspId, String versionId, String componentId, String processId,
                         String user) {
    MdcUtil.initMdc(LoggerServiceName.Delete_Component_Processes.toString());
    Version version = resolveVspVersion(vspId, null, user, VersionableEntityAction.Write);
    validateComponentExistence(vspId, version, componentId, user);
    processManager.deleteProcess(vspId, version, componentId, processId, user);
    return Response.ok().build();
  }

  @Override
  public Response update(ProcessRequestDto request, String vspId, String versionId,
                         String componentId,
                         String processId, String user) {
    MdcUtil.initMdc(LoggerServiceName.Update_Component_Processes.toString());
    ProcessEntity process =
        new MapProcessRequestDtoToProcessEntity().applyMapping(request, ProcessEntity.class);
    process.setVspId(vspId);
    process.setVersion(resolveVspVersion(vspId, null, user, VersionableEntityAction.Write));
    process.setComponentId(componentId);
    process.setId(processId);
    validateComponentExistence(vspId, process.getVersion(), componentId, user);
    processManager.updateProcess(process, user);
    return Response.ok().build();
  }

  @Override
  public Response getUploadedFile(String vspId, String versionId, String componentId,
                                  String processId, String user) {
    MdcUtil.initMdc(LoggerServiceName.Get_Uploaded_File_Component_Processes.toString());
    Version vspVersion = resolveVspVersion(vspId, versionId, user, VersionableEntityAction.Read);
    validateComponentExistence(vspId, vspVersion, componentId, user);
    File file = processManager.getProcessArtifact(vspId, vspVersion, componentId, processId, user);

    Response.ResponseBuilder response = Response.ok(file);
    if (file == null) {
      return Response.status(Response.Status.NOT_FOUND).build();
    }
    response.header("Content-Disposition", "attachment; filename=" + file.getName());
    return response.build();
  }

  @Override
  public Response deleteUploadedFile(String vspId, String versionId, String componentId,
                                     String processId,
                                     String user) {
    MdcUtil.initMdc(LoggerServiceName.Delete_Uploaded_File_Component_Processes.toString());
    Version version = resolveVspVersion(vspId, null, user, VersionableEntityAction.Write);
    validateComponentExistence(vspId, version, componentId, user);
    processManager.deleteProcessArtifact(vspId, version, componentId, processId, user);
    return Response.ok().build();
  }

  @Override
  public Response uploadFile(Attachment attachment, String vspId, String versionId,
                             String componentId,
                             String processId, String user) {
    MdcUtil.initMdc(LoggerServiceName.Upload_File_Component_Processes.toString());
      logger.audit(AuditMessages.AUDIT_MSG + AuditMessages.UPLOAD_PROCESS_ARTIFACT + vspId);
    Version version = resolveVspVersion(vspId, null, user, VersionableEntityAction.Write);
    validateComponentExistence(vspId, version, componentId, user);
    processManager.uploadProcessArtifact(attachment.getObject(InputStream.class),
        attachment.getContentDisposition().getParameter("filename"), vspId, version, componentId,
        processId, user);
    return Response.ok().build();
  }


  private void validateComponentExistence(String vspId, Version version, String componentId,
                                          String user) {
    if (GENERAL_COMPONENT_ID.equals(componentId)) {
      return;
    }
    componentManager.validateComponentExistence(vspId, version, componentId, user);
  }
}