/* ==========================================
 * Laverca Project
 * https://sourceforge.net/projects/laverca/
 * ==========================================
 * Copyright 2015 Laverca Project
 * 
 * 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 fi.laverca.mss;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import fi.laverca.CmsSignature;
import fi.laverca.Pkcs1;
import fi.laverca.Signature;
import fi.laverca.etsi.EtsiResponse;
import fi.laverca.jaxb.mss.MSSSignatureReq;
import fi.laverca.jaxb.mss.MSSSignatureResp;
import fi.laverca.jaxb.mss.MSSStatusResp;
import fi.laverca.jaxb.mssfi.PKCS1;

public abstract class MssResponse {

    private static final Log log = LogFactory.getLog(EtsiResponse.class);

    public MSSSignatureReq  originalSigReq;
    public MSSSignatureResp originalSigResp;
    public MSSStatusResp    finalStatusResp;

    public MssResponse(final MSSSignatureReq  originalSigReq,
                       final MSSSignatureResp originalSigResp,
                       final MSSStatusResp    finalStatusResp) { 
        this.originalSigReq  = originalSigReq;
        this.originalSigResp = originalSigResp;
        this.finalStatusResp = finalStatusResp;
    }
    
    /** 
     * Get the MSS signature
     * 
     * @return signature or null if the response contains no signature. 
     */
    public Signature getSignature() {
        
        if (this.finalStatusResp == null) return null;
        if (this.finalStatusResp.getMSSSignature() == null) return null;
        
        Signature s = null;

        // CMS
        byte[] sig = this.finalStatusResp.getMSSSignature().getBase64Signature();
        if (sig != null) {
            s = new CmsSignature(sig);
        }
        
        // FiCom PKCS1
        PKCS1 p1 = this.finalStatusResp.getMSSSignature().getPKCS1();
        if (p1 != null) {
            s = new Pkcs1(p1);
        }
        
        return s;
    }
    
    /**
     * Check if the response has signature available
     * 
     * @return true if signature is available
     */
    public boolean hasSignature() {
        if (this.finalStatusResp == null) return false;
        if (this.finalStatusResp.getMSSSignature() == null) return false;
        
        boolean hasSignature = false;
        
        if (this.finalStatusResp.getMSSSignature().getBase64Signature() != null) {
            hasSignature = true;
        }
        if (this.finalStatusResp.getMSSSignature().getPKCS1() != null) {
            hasSignature = true;
        }
        
        return hasSignature;
    }

    /** 
     * Get the PKCS7 signature from this response
     * @return PKCS7 signature or null if the signature is not in this format.
     * @deprecated Use {@link #getSignature()} instead
     */
    @Deprecated
    public CmsSignature getPkcs7Signature() {
        try {
            if (this.finalStatusResp == null)
                throw new RuntimeException("Illegal state. Null statusResp.");

            if (this.finalStatusResp.getMSSSignature() == null)
                throw new RuntimeException("Illegal state. Null statusResp.MSS_Signature");
            
            final CmsSignature p7 = new CmsSignature(this.finalStatusResp.getMSSSignature().getBase64Signature());
            return p7;
        } catch(IllegalArgumentException iae) {
            log.debug("not a pkcs7?", iae);
            return null;
        }

    }
    
    /** 
     * Get the PKCS1 signature from this response
     * @return PKCS1 signature or null if the signature is not in this format.
     * @deprecated Use {@link #getSignature()} instead
     */
    @Deprecated
    public Pkcs1 getPkcs1Signature() {
        try {
            if (this.finalStatusResp == null)
                throw new RuntimeException("illegal state. Null statusResp.");

            if (this.finalStatusResp.getMSSSignature() == null)
                throw new RuntimeException("illegal state. Null statusResp.MSS_Signature");
            
            final Pkcs1 p1 = new Pkcs1(this.finalStatusResp.getMSSSignature().getPKCS1());
            return p1;
        } catch(IllegalArgumentException iae) {
            log.debug("not a pkcs1?", iae);
            return null;
        }
    }

    /** 
     * Get the last MSS_StatusResp message
     * @return MSS_StatusResp
     */
    public MSSStatusResp getMSS_StatusResp() {
        return this.finalStatusResp;
    }

    /**
     * Get the latest status code of the transaction
     * @return latest status code or -1 if not available
     */
    public long getStatusCode() {
        
        try {
            if (this.finalStatusResp != null) {
                return this.finalStatusResp.getStatus().getStatusCode().getValue().longValue();
            }
            if (this.originalSigResp != null) {
                return this.originalSigResp.getStatus().getStatusCode().getValue().longValue();
            }
        } catch (NullPointerException npe) {
            // Got a faulty response... ignore
        }
        return -1;
    }
    
    /**
     * Get the latest status message of the transaction
     * @return latest status code or null if not available
     */
    public String getStatusMessage() {
        try {
            if (this.finalStatusResp != null) {
                return this.finalStatusResp.getStatus().getStatusMessage();
            }
            if (this.originalSigResp != null) {
                return this.originalSigResp.getStatus().getStatusMessage();
            }
        } catch (NullPointerException npe) {
            // Got a faulty response... ignore
        }
        return null;
    }
    
    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append(this.getClass().getSimpleName());
        sb.append(" [");
        sb.append("StatusCode=").append(this.getStatusCode());
        sb.append(", StatusMessage=").append(this.getStatusMessage());
        sb.append(", HasSignature=").append(this.hasSignature());
        sb.append("]");
        return sb.toString();
    }
    
}