/*
 *
 * Copyright (c) 2013 - 2017 Lijun Liao
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License version 3
 * as published by the Free Software Foundation with the addition of the
 * following permission added to Section 15 as permitted in Section 7(a):
 *
 * FOR ANY PART OF THE COVERED WORK IN WHICH THE COPYRIGHT IS OWNED BY
 * THE AUTHOR LIJUN LIAO. LIJUN LIAO DISCLAIMS THE WARRANTY OF NON INFRINGEMENT
 * OF THIRD PARTY RIGHTS.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 *
 * The interactive user interfaces in modified source and object code versions
 * of this program must display Appropriate Legal Notices, as required under
 * Section 5 of the GNU Affero General Public License.
 *
 * You can be released from the requirements of the license by purchasing
 * a commercial license. Buying such a license is mandatory as soon as you
 * develop commercial activities involving the XiPKI software without
 * disclosing the source code of your own applications.
 *
 * For more information, please contact Lijun Liao at this
 * address: lijun.liao@gmail.com
 */

package org.xipki.pki.ca.dbtool.port.ocsp;

import java.util.concurrent.atomic.AtomicBoolean;

import org.slf4j.Logger;
import org.xipki.commons.common.util.StringUtil;
import org.xipki.commons.datasource.DataSourceWrapper;
import org.xipki.commons.datasource.springframework.dao.DataAccessException;
import org.xipki.commons.security.HashAlgoType;
import org.xipki.pki.ca.dbtool.port.DbPorter;

/**
 * @author Lijun Liao
 * @since 2.0.0
 */

abstract class AbstractOcspCertStoreDbImporter extends DbPorter {

    protected static final String MSG_CERTS_FINISHED = "certs.finished";

    protected static final String SQL_ADD_ISSUER =
        "INSERT INTO ISSUER (ID,SUBJECT,NBEFORE,NAFTER,S1C,REV,RR,RT,RIT,CERT) "
        + "VALUES (?,?,?,?,?,?,?,?,?,?)";

    protected static final String SQL_ADD_CERT =
        "INSERT INTO CERT (ID,IID,SN,LUPDATE,NBEFORE,NAFTER,REV,RR,RT,RIT,PN)"
        + " VALUES (?,?,?,?,?,?,?,?,?,?,?)";

    protected static final String SQL_DEL_CERT =
        "DELETE FROM CERT WHERE ID>?";

    protected static final String SQL_ADD_CHASH =
        "INSERT INTO CHASH (CID,S1,S224,S256,S384,S512) VALUES (?,?,?,?,?,?)";

    protected static final String SQL_DEL_CHASH =
        "DELETE FROM CHASH WHERE ID>?";

    protected static final String SQL_ADD_CRAW =
        "INSERT INTO CRAW (CID,SUBJECT,CERT) VALUES (?,?,?)";

    protected static final String SQL_DEL_CRAW =
        "DELETE FROM CRAW WHERE ID>?";

    AbstractOcspCertStoreDbImporter(final DataSourceWrapper datasource, final String srcDir,
            final AtomicBoolean stopMe, final boolean evaluateOnly) throws Exception {
        super(datasource, srcDir, stopMe, evaluateOnly);
    }

    protected String sha1(final byte[] data) {
        return HashAlgoType.SHA1.base64Hash(data);
    }

    protected String sha224(final byte[] data) {
        return HashAlgoType.SHA224.base64Hash(data);
    }

    protected String sha256(final byte[] data) {
        return HashAlgoType.SHA256.base64Hash(data);
    }

    protected String sha384(final byte[] data) {
        return HashAlgoType.SHA384.base64Hash(data);
    }

    protected String sha512(final byte[] data) {
        return HashAlgoType.SHA512.base64Hash(data);
    }

    protected void deleteCertGreatherThan(final long id, final Logger log) {
        deleteFromTableWithLargerId("CRAW", "CID", id, log);
        deleteFromTableWithLargerId("CHASH", "CID", id, log);
        deleteFromTableWithLargerId("CERT", "ID", id, log);
    }

    protected void dropIndexes() throws DataAccessException {
        System.out.println("dropping indexes");
        long start = System.currentTimeMillis();

        datasource.dropForeignKeyConstraint(null, "FK_CERT_ISSUER1", "CERT");
        datasource.dropUniqueConstrain(null, "CONST_ISSUER_SN", "CERT");

        datasource.dropForeignKeyConstraint(null, "FK_CHASH_CERT1", "CHASH");
        datasource.dropForeignKeyConstraint(null, "FK_CRAW_CERT1", "CRAW");

        datasource.dropPrimaryKey(null, "PK_CERT", "CERT");
        datasource.dropPrimaryKey(null, "PK_CRAW", "CRAW");
        datasource.dropPrimaryKey(null, "PK_CHASH", "CHASH");

        long duration = (System.currentTimeMillis() - start) / 1000;
        System.out.println(" dropped indexes in " + StringUtil.formatTime(duration, false));
    }

    protected void recoverIndexes() throws DataAccessException {
        System.out.println("recovering indexes");
        long start = System.currentTimeMillis();

        datasource.addPrimaryKey(null, "PK_CERT", "CERT", "ID");
        datasource.addPrimaryKey(null, "PK_CRAW", "CRAW", "CID");
        datasource.addPrimaryKey(null, "PK_CHASH", "CHASH", "CID");

        datasource.addForeignKeyConstraint(null, "FK_CERT_ISSUER1", "CERT",
                "IID", "ISSUER", "ID", "CASCADE", "NO ACTION");
        datasource.addUniqueConstrain(null, "CONST_ISSUER_SN", "CERT", "IID", "SN");

        datasource.addForeignKeyConstraint(null, "FK_CRAW_CERT1", "CRAW", "CID", "CERT", "ID",
                "CASCADE", "NO ACTION");
        datasource.addForeignKeyConstraint(null, "FK_CHASH_CERT1", "CHASH", "CID", "CERT", "ID",
                "CASCADE", "NO ACTION");

        long duration = (System.currentTimeMillis() - start) / 1000;
        System.out.println(" recovered indexes in " + StringUtil.formatTime(duration, false));
    }

}