package ui.gui.widgets;

import model.ca.CertificationAuthority;
import model.pki.cert.Certificate;
import model.pki.crl.RevokedCertificate;

import javax.swing.*;
import javax.swing.table.AbstractTableModel;
import java.time.format.DateTimeFormatter;
import java.util.List;

/**
 * Table model that displays issued certificates.
 */
public class CertTableModel extends AbstractTableModel {
    /**
     * Valid certificate icon.
     */
    private static final ImageIcon ICON_OK =
            new ImageIcon(CertTableModel.class.getResource("/verified.png"));

    /**
     * Revoked certificate icon (same as toolbar revoke icon).
     */
    private static final ImageIcon ICON_REVOKED =
            new ImageIcon(CertTableModel.class.getResource("/deletetest.png"));

    /**
     * Columns
     */
    private static final String[] COLS = new String[] {
            "", // Icon
            "Serial",
            "Subject",
            "Signed",
            "Expires"
    };

    /**
     * Pointer to {@link CertificationAuthority#getSigned()}
     */
    private List<Certificate> ptrData;

    /**
     * Pointer to {@link CertificationAuthority#getRevoked()}
     */
    private List<RevokedCertificate> ptrRevokedData;

    /**
     * EFFECTS: Set pointer to certs
     * MODIFIES: this
     */
    public void setPtrData(List<Certificate> ptrData) {
        this.ptrData = ptrData;
    }

    /**
     * EFFECTS: Set pointer to revoked
     * MODIFIES: this
     */
    public void setPtrRevokedData(List<RevokedCertificate> ptrRevokedData) {
        this.ptrRevokedData = ptrRevokedData;
    }

    /**
     * EFFECTS: Count rows.
     */
    @Override
    public int getRowCount() {
        return ptrData == null ? 0 : ptrData.size();
    }

    /**
     * EFFECTS: Count columns.
     */
    @Override
    public int getColumnCount() {
        return COLS.length;
    }

    /**
     * EFFECTS: Get column name.
     * REQUIRES: column in [9, getColumnCount())
     */
    @Override
    public String getColumnName(int column) {
        return COLS[column];
    }

    /**
     * EFFECTS: Return the value for a cell:
     *          ImageIcon (Valid / Revoked)
     *          String (Serial number)
     *          String (Subject)
     *          String (NotBefore)
     *          String (NotAfter)
     *          Throws {@link IllegalArgumentException} if columnIndex is not in 0 ~ 4
     * REQUIRES: rowIndex must in range.
     */
    @Override
    public Object getValueAt(int rowIndex, int columnIndex) {
        final Certificate e = ptrData.get(rowIndex);
        switch (columnIndex) {
            case 0: return ptrRevokedData.stream().anyMatch(r ->
                    r.getSerialNumber().getLong() == e.getCertificate().getSerialNumber().getLong()
            ) ? ICON_REVOKED : ICON_OK;
            case 1: return e.getCertificate().getSerialNumber().getLong();
            case 2: return e.getCertificate().getSubject().toString();
            case 3:
                return e.getCertificate().getValidity().getNotBefore().getTimestamp()
                        .format(DateTimeFormatter.ISO_LOCAL_DATE_TIME);
            case 4:
                return e.getCertificate().getValidity().getNotAfter().getTimestamp()
                        .format(DateTimeFormatter.ISO_LOCAL_DATE_TIME);
            default: throw new IllegalArgumentException();
        }
    }

    /**
     * EFFECTS: Return ImageIcon for 0, String otherwise.
     */
    @Override
    public Class<?> getColumnClass(int columnIndex) {
        switch (columnIndex) {
            case 0: return ImageIcon.class;
            default: return String.class;
        }
    }
}