/*-
 * ============LICENSE_START=======================================================
 *  Copyright (C) 2020 Nordix Foundation.
 * ================================================================================
 * 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.
 *
 * SPDX-License-Identifier: Apache-2.0
 * ============LICENSE_END=========================================================
 */
package org.onap.so.etsi.nfvo.ns.lcm.bpmn.flows.tasks;

import static org.onap.so.etsi.nfvo.ns.lcm.bpmn.flows.CamundaVariableNameConstants.CREATE_NS_WORKFLOW_PROCESSING_EXCEPTION_PARAM_NAME;
import static org.onap.so.etsi.nfvo.ns.lcm.bpmn.flows.CamundaVariableNameConstants.JOB_ID_PARAM_NAME;
import java.time.LocalDateTime;
import java.util.Optional;
import org.camunda.bpm.engine.delegate.BpmnError;
import org.camunda.bpm.engine.delegate.DelegateExecution;
import org.onap.so.etsi.nfvo.ns.lcm.database.beans.JobStatusEnum;
import org.onap.so.etsi.nfvo.ns.lcm.database.beans.NfvoJob;
import org.onap.so.etsi.nfvo.ns.lcm.database.beans.NfvoJobStatus;
import org.onap.so.etsi.nfvo.ns.lcm.database.service.DatabaseServiceProvider;
import org.onap.so.etsi.nfvo.ns.lcm.model.InlineResponse400;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Waqas Ikram (waqas.ikram@est.tech)
 *
 */
public abstract class AbstractNetworkServiceTask {
    private final Logger logger = LoggerFactory.getLogger(getClass());
    protected final DatabaseServiceProvider databaseServiceProvider;

    public AbstractNetworkServiceTask(final DatabaseServiceProvider jobServiceProvider) {
        this.databaseServiceProvider = jobServiceProvider;
    }

    public void setJobStatus(final DelegateExecution execution, final JobStatusEnum jobStatus,
            final String description) {
        logger.info("Setting Job Status to {}", jobStatus);
        final NfvoJob nfvoJob = getNfvoJob(execution);
        nfvoJob.status(jobStatus);
        if (JobStatusEnum.STARTED.equals(jobStatus)) {
            nfvoJob.processInstanceId(execution.getProcessInstanceId());
        }

        if (JobStatusEnum.FINISHED.equals(jobStatus)) {
            nfvoJob.endTime(LocalDateTime.now());
        }

        nfvoJob.nfvoJobStatus(
                new NfvoJobStatus().status(jobStatus).description(description).updatedTime(LocalDateTime.now()));
        databaseServiceProvider.addJob(nfvoJob);

    }

    public void setJobStatusToError(final DelegateExecution execution, final String description) {
        logger.info("Setting Job Status to {}", JobStatusEnum.ERROR);

        final String jobId = (String) execution.getVariable(JOB_ID_PARAM_NAME);
        final Optional<NfvoJob> optional = databaseServiceProvider.getJob(jobId);
        if (optional.isPresent()) {
            final InlineResponse400 problemDetails =
                    (InlineResponse400) execution.getVariable(CREATE_NS_WORKFLOW_PROCESSING_EXCEPTION_PARAM_NAME);

            final NfvoJob nfvoJob = optional.get();
            nfvoJob.status(JobStatusEnum.ERROR).endTime(LocalDateTime.now());

            if (problemDetails != null) {
                logger.error("Found failed reason: {}", problemDetails);
                nfvoJob.nfvoJobStatus(new NfvoJobStatus().status(JobStatusEnum.ERROR)
                        .description(problemDetails.getDetail()).updatedTime(LocalDateTime.now()));
            }
            nfvoJob.nfvoJobStatus(new NfvoJobStatus().status(JobStatusEnum.ERROR).description(description)
                    .updatedTime(LocalDateTime.now()));

            databaseServiceProvider.addJob(nfvoJob);
        }
        logger.info("Finished setting Job Status to {}", JobStatusEnum.ERROR);

    }

    protected void abortOperation(final DelegateExecution execution, final String message) {
        abortOperation(execution, message, new InlineResponse400().detail(message));
    }

    protected void abortOperation(final DelegateExecution execution, final String message,
            final InlineResponse400 problemDetails) {
        logger.error(message);
        execution.setVariable(CREATE_NS_WORKFLOW_PROCESSING_EXCEPTION_PARAM_NAME, problemDetails);
        throw new BpmnError("WORKFLOW_FAILED");
    }

    private NfvoJob getNfvoJob(final DelegateExecution execution) {
        final String jobId = (String) execution.getVariable(JOB_ID_PARAM_NAME);
        final Optional<NfvoJob> optional = databaseServiceProvider.getJob(jobId);
        if (!optional.isPresent()) {
            final String message = "Unable to find job using job id: " + jobId;
            logger.error(message);
            execution.setVariable(CREATE_NS_WORKFLOW_PROCESSING_EXCEPTION_PARAM_NAME,
                    new InlineResponse400().detail(message));
            throw new BpmnError("WORKFLOW_FAILED");

        }
        return optional.get();
    }
}