/**
 *
 *     Copyright (c) 2017 Orange.  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.
 */
package org.onap.nbi.apis.serviceorder;

import java.util.Date;
import java.util.List;
import javax.validation.Valid;
import org.onap.nbi.apis.serviceorder.model.ServiceOrder;
import org.onap.nbi.apis.serviceorder.model.ServiceOrderItem;
import org.onap.nbi.apis.serviceorder.model.StateType;
import org.onap.nbi.apis.serviceorder.model.orchestrator.ServiceOrderInfo;
import org.onap.nbi.apis.serviceorder.repositories.ServiceOrderInfoRepository;
import org.onap.nbi.apis.serviceorder.repositories.ServiceOrderRepository;
import org.onap.nbi.apis.serviceorder.workflow.CheckOrderConsistenceManager;
import org.onap.nbi.apis.serviceorder.workflow.CreateAAICustomerManager;
import org.onap.nbi.apis.serviceorder.workflow.CreateAAIServiceTypeManager;
import org.onap.nbi.apis.serviceorder.workflow.SOTaskManager;
import org.onap.nbi.commons.JsonRepresentation;
import org.onap.nbi.commons.ResourceManagement;
import org.onap.nbi.exceptions.ValidationException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.util.MultiValueMap;
import org.springframework.validation.Errors;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/serviceOrder")
@EnableScheduling
public class ServiceOrderResource extends ResourceManagement<ServiceOrder> {

    @Autowired
    ServiceOrderRepository serviceOrderRepository;

    @Autowired
    CheckOrderConsistenceManager checkOrderConsistenceManager;

    @Autowired
    CreateAAICustomerManager createAAICustomer;

    @Autowired
    CreateAAIServiceTypeManager createAAIServiceType;

    @Autowired
    MongoTemplate mongoTemplate;

    @Autowired
    SOTaskManager serviceOrchestratorManager;

    @Autowired
    ServiceOrderInfoRepository serviceOrderInfoRepository;

    @Autowired
    MultiCriteriaRequestBuilder multiCriteriaRequestBuilder;


    @GetMapping(value = "/{serviceOrderId}", produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<Object> getServiceOrder(@PathVariable String serviceOrderId,
            @RequestParam MultiValueMap<String, String> params) {

        ServiceOrder serviceOrder = serviceOrderRepository.findOne(serviceOrderId);
        if (serviceOrder == null) {
            return ResponseEntity.notFound().build();
        }

        JsonRepresentation filter = new JsonRepresentation(params);
        return this.getResponse(serviceOrder, filter);
    }

    @GetMapping(value = "", produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<Object> findServiceOrder(@RequestParam MultiValueMap<String, String> params) {

        Query query = multiCriteriaRequestBuilder.buildRequest(params);
        List<ServiceOrder> serviceOrders = mongoTemplate.find(query, ServiceOrder.class);
        JsonRepresentation filter = new JsonRepresentation(params);
        long totalCount = serviceOrderRepository.count();
        HttpHeaders headers = new HttpHeaders();
        headers.add("X-Total-Count", String.valueOf(totalCount));
        headers.add("X-Result-Count", String.valueOf(serviceOrders.size()));

        ResponseEntity<Object> response = this.findResponse(serviceOrders, filter, headers);
        return response;

    }

    @DeleteMapping(value = "/{serviceOrderId}")
    public ResponseEntity<Object> deleteServiceOrder(@PathVariable String serviceOrderId) {

        serviceOrderRepository.delete(serviceOrderId);

        return this.deleteResponse();

    }

    @PostMapping(consumes = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<Object> createServiceOrder(@Valid @RequestBody ServiceOrder serviceOrder, Errors errors,
            @RequestParam MultiValueMap<String, String> params) {


        if (errors != null && errors.hasErrors()) {
            throw new ValidationException(errors.getAllErrors());
        }

        serviceOrder.setState(StateType.ACKNOWLEDGED);
        serviceOrder.setOrderDate(new Date());
        for (ServiceOrderItem serviceOrderItem : serviceOrder.getOrderItem()) {
            serviceOrderItem.setState(StateType.ACKNOWLEDGED);
        }

        ServiceOrder serviceOrderSaved = serviceOrderRepository.save(serviceOrder);
        serviceOrderSaved.setHref("serviceOrder/" + serviceOrderSaved.getId());
        serviceOrderRepository.save(serviceOrderSaved);
        JsonRepresentation filter = new JsonRepresentation(params);
        return this.createResponse(serviceOrderSaved, filter);

    }

    @Scheduled(fixedDelay = 5000)
    public void scheduleCheckServiceOrders() {
        List<ServiceOrder> acknowledgedOrders = serviceOrderRepository.findByState(StateType.ACKNOWLEDGED);
        for (ServiceOrder serviceOrder : acknowledgedOrders) {
            ServiceOrderInfo serviceOrderInfo = checkOrderConsistenceManager.checkServiceOrder(serviceOrder);
            if (serviceOrderInfo.isServiceOrderRejected()) {
                changeServiceOrderState(serviceOrder, StateType.REJECTED);
            } else if (serviceOrderInfo.isAllItemsCompleted()) {
                changeServiceOrderState(serviceOrder, StateType.COMPLETED);
            } else {
                serviceOrderRepository.save(serviceOrder);
                createAAICustomer.createAAICustomer(serviceOrderInfo);
                createAAIServiceType.createAAIServiceType(serviceOrder, serviceOrderInfo);
                serviceOrchestratorManager.registerServiceOrder(serviceOrder, serviceOrderInfo);
            }
        }
    }

    /**
     *
     * @param serviceOrder
     * @param stateType
     */
    private void changeServiceOrderState(ServiceOrder serviceOrder, StateType stateType) {
        serviceOrder.setState(stateType);
        serviceOrder.setCompletionDateTime(new Date());
        serviceOrderRepository.save(serviceOrder);
    }

}