/*
 * Copyright (c) 2004 University of Murcia.  All rights reserved.
 * --------------------------------------------------------------
 * For more information, please see <http://www.umu.euro6ix.org/>.
 */

package org.umu.cops.prpdp;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.umu.cops.COPSStateMan;
import org.umu.cops.stack.*;

import java.net.Socket;
import java.util.HashMap;
import java.util.Map;

/**
 * State manager class for provisioning requests, at the PDP side.
 */
public class COPSPdpReqStateMan extends COPSStateMan {

    private final static Logger logger = LoggerFactory.getLogger(COPSPdpReqStateMan.class);

    /**
     * Object for performing policy data processing
     */
    private final COPSPdpDataProcess _process;

    /** COPS message transceiver used to send COPS messages */
    protected transient COPSPdpMsgSender _sender;

    /**
     * Creates a request state manager
     * @param clientType    Client-type
     * @param clientHandle  Client handle
     */
    public COPSPdpReqStateMan(final short clientType, final COPSHandle clientHandle, final COPSPdpDataProcess process,
                              final Socket socket) {
        super(clientType, clientHandle, socket);
        this._process = process;
        _sender = new COPSPdpMsgSender(_clientType, _handle, _socket);
        _status = Status.ST_INIT;
    }

    /**
     * Processes a COPS request
     * @param msg   COPS request received from the PEP
     * @throws COPSException
     */
    protected void processRequest(final COPSReqMsg msg) throws COPSException {
        // TODO - Implement me - see commented out code from history prior to May 4, 2015...
    }

    /**
     * Processes a report
     * @param msg   Report message from the PEP
     * @throws COPSException
     */
    protected void processReport(final COPSReportMsg msg) throws COPSException {
        if (msg.getClientSI() != null) {
            // Report Type
            final COPSReportType rtypemsg = msg.getReport();
            final Map<String, String> repSIs = new HashMap<>();
            String strobjprid = "";
            final COPSPrObjBase obj = new COPSPrObjBase(msg.getClientSI().getData().getData());
            switch (obj.getSNum()) {
                case COPSPrObjBase.PR_PRID:
                    strobjprid = obj.getData().str();
                    break;
                case COPSPrObjBase.PR_EPD:
                    // TODO FIXME - strobjprid is always empty here???
                    repSIs.put(strobjprid, obj.getData().str());
                    logger.info("PRID: " + strobjprid);
                    logger.info("EPD: " + obj.getData().str());
                    break;
                default:
                    break;
            }

            //** Here we must act in accordance with
            //** the report received
            switch (rtypemsg.getReportType()) {
                case SUCCESS:
                    _status = Status.ST_REPORT;
                    _process.successReport(this, repSIs);
                    break;
                case FAILURE:
                    _status = Status.ST_REPORT;
                    _process.failReport(this, repSIs);
                    break;
                case ACCOUNTING:
                    _status = Status.ST_ACCT;
                    _process.acctReport(this, repSIs);
                    break;
            }
        }
    }

    /**
     * Called when connection is closed
     * @param error Reason
     * @throws COPSPdpException
     */
    public void processClosedConnection(final COPSError error) throws COPSException {
        if (_process != null)
            _process.notifyClosedConnection(this, error);

        _status = Status.ST_CCONN;
    }

    /**
     * Called when no keep-alive is received
     * @throws COPSPdpException
     */
    public void processNoKAConnection() throws COPSException {
        if (_process != null)
            _process.notifyNoKAliveReceived(this);

        _status = Status.ST_NOKA;
    }

    /**
     * Deletes the request state
     * @throws COPSPdpException
     */
    protected void finalizeRequestState() throws COPSException {
        _sender.sendDeleteRequestState();
        _status = Status.ST_FINAL;
    }

    /**
     * Asks for a COPS sync
     * @throws COPSPdpException
     */
    public void syncRequestState() throws COPSException {
        _sender.sendSyncRequestState();
        _status = Status.ST_SYNC;
    }

    /**
     * Opens a new request state
     * @throws COPSPdpException
     */
    protected void openNewRequestState() throws COPSException {
        _sender.sendOpenNewRequestState();
        _status = Status.ST_NEW;
    }

    /**
     * Processes a COPS delete message
     * @param dMsg  <tt>COPSDeleteMsg</tt> received from the PEP
     * @throws COPSPdpException
     */
    public void processDeleteRequestState(final COPSDeleteMsg dMsg) throws COPSException {
        if (_process != null)
            _process.closeRequestState(this);

        _status = Status.ST_DEL;
    }

}