//
// Copyright 2019 AT&T Intellectual Property
// Copyright 2019 Nokia
//
// 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.

//  This source code is part of the near-RT RIC (RAN Intelligent Controller)
//  platform project (RICP).


package httpmsghandlers

import "C"
import (
	"e2mgr/configuration"
	"e2mgr/e2managererrors"
	"e2mgr/logger"
	"e2mgr/models"
	"e2mgr/rmrCgo"
	"e2mgr/services"
	"e2mgr/services/rmrsender"
	"e2mgr/stateMachine"
	"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
	"time"
)

type DeleteAllRequestHandler struct {
	rnibDataService services.RNibDataService
	rmrSender       *rmrsender.RmrSender
	config          *configuration.Configuration
	logger          *logger.Logger
}

func NewDeleteAllRequestHandler(logger *logger.Logger, rmrSender *rmrsender.RmrSender, config *configuration.Configuration, rnibDataService services.RNibDataService) *DeleteAllRequestHandler {
	return &DeleteAllRequestHandler{
		logger:          logger,
		rnibDataService: rnibDataService,
		rmrSender:       rmrSender,
		config:          config,
	}
}

func (handler *DeleteAllRequestHandler) Handle(request models.Request) (models.IResponse, error) {

	err, continueFlow := handler.updateNodebStates(false)
	if err != nil {
		return nil, err
	}

	if continueFlow == false {
		return nil, nil
	}

	response := models.RmrMessage{MsgType: rmrCgo.RIC_SCTP_CLEAR_ALL}
	if err := handler.rmrSender.Send(&response); err != nil {
		handler.logger.Errorf("#DeleteAllRequestHandler.Handle - failed to send sctp clear all message to RMR: %s", err)
		return nil, e2managererrors.NewRmrError()
	}

	time.Sleep(time.Duration(handler.config.BigRedButtonTimeoutSec) * time.Second)
	handler.logger.Infof("#DeleteAllRequestHandler.Handle - timer expired")

	err, _ = handler.updateNodebStates(true)
	if err != nil {
		return nil, err
	}

	return nil, nil
}

func (handler *DeleteAllRequestHandler) updateNodebStates(timeoutExpired bool) (error, bool) {
	nbIdentityList, err := handler.rnibDataService.GetListNodebIds()

	if err != nil {
		handler.logger.Errorf("#DeleteAllRequestHandler.updateNodebStates - failed to get nodes list from RNIB. Error: %s", err.Error())
		return e2managererrors.NewRnibDbError(), false
	}

	if len(nbIdentityList) == 0 {
		return nil, false
	}

	numOfRanToShutDown := 0
	for _, nbIdentity := range nbIdentityList {

		node, err := handler.rnibDataService.GetNodeb((*nbIdentity).GetInventoryName())

		if err != nil {
			handler.logger.Errorf("#DeleteAllRequestHandler.updateNodebStates - failed to get nodeB entity for ran name: %v from RNIB. Error: %s",
				(*nbIdentity).GetInventoryName(), err.Error())
			continue
		}

		if timeoutExpired {

			if handler.saveNodebShutDownState(nbIdentity, node) {
				numOfRanToShutDown++
			}
			continue
		}
		if handler.saveNodebNextState(nbIdentity, node) {
			numOfRanToShutDown++
		}
	}

	if numOfRanToShutDown > 0 {
		handler.logger.Infof("#DeleteAllRequestHandler.updateNodebStates - update nodebs states in RNIB completed")
	} else {
		handler.logger.Infof("#DeleteAllRequestHandler.updateNodebStates - nodebs states are not updated ")
		return nil, false
	}

	return nil, true
}

func (handler *DeleteAllRequestHandler) saveNodebNextState(nbIdentity *entities.NbIdentity, node *entities.NodebInfo) bool {

	if node.ConnectionStatus == entities.ConnectionStatus_SHUTTING_DOWN {
		return true
	}

	nextStatus, res := stateMachine.NodeNextStateDeleteAll(node.ConnectionStatus)
	if res == false {
		return false
	}

	node.ConnectionStatus = nextStatus

	err := handler.rnibDataService.SaveNodeb(nbIdentity, node)

	if err != nil {
		handler.logger.Errorf("#DeleteAllRequestHandler.saveNodebNextState - failed to save nodeB entity for inventory name: %v to RNIB. Error: %s",
			(*nbIdentity).GetInventoryName(), err.Error())
		return false
	}

	if handler.logger.DebugEnabled() {
		handler.logger.Debugf("#DeleteAllRequestHandler.saveNodebNextState - connection status of inventory name: %v changed to %v",
			(*nbIdentity).GetInventoryName(), nextStatus.String())
	}
	return true
}

func (handler *DeleteAllRequestHandler) saveNodebShutDownState(nbIdentity *entities.NbIdentity, node *entities.NodebInfo) bool {

	if node.ConnectionStatus == entities.ConnectionStatus_SHUT_DOWN {
		return false
	}

	if node.ConnectionStatus != entities.ConnectionStatus_SHUTTING_DOWN {
		handler.logger.Errorf("#DeleteAllRequestHandler.saveNodebShutDownState - ignore, status is not Shutting Down, inventory name: %v ", (*nbIdentity).GetInventoryName())
		return false
	}

	node.ConnectionStatus = entities.ConnectionStatus_SHUT_DOWN

	err := handler.rnibDataService.SaveNodeb(nbIdentity, node)

	if err != nil {
		handler.logger.Errorf("#DeleteAllRequestHandler.saveNodebShutDownState - failed to save nodeB entity for inventory name: %v to RNIB. Error: %s",
			(*nbIdentity).GetInventoryName(), err.Error())
		return false
	}

	handler.logger.Errorf("#DeleteAllRequestHandler.saveNodebShutDownState - Shut Down , inventory name: %v ", (*nbIdentity).GetInventoryName())
	return true
}