/* 
 * This file is part of cert-signer
 * Copyright (c) 2024 Paolo Lulli.
 * 
 * This program is free software: you can redistribute it and/or modify  
 * it under the terms of the GNU General Public License as published by  
 * the Free Software Foundation, version 3.
 *
 * 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 
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License 
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

package net.lulli.certsigner.strategy.vault;

import net.lulli.certsigner.network.VaultLocal;
import net.lulli.certsigner.strategy.CASetupStrategy;
import net.lulli.certsigner.ca.CaData;

import java.util.Base64;
import java.util.HashMap;
import java.util.Objects;

public class VaultCaSetup implements CASetupStrategy {
    private static String vaultEndpoint = System.getenv("VAULT_ENDPOINT");
    private static String vaultToken = System.getenv("VAULT_TOKEN");

    private static final String VAULT_PATH = "cert-auth";

    private final VaultLocal vaultLocal;
    private final String serviceName;

    public VaultCaSetup(String serviceName) {
        Objects.requireNonNull(serviceName);

        this.serviceName = serviceName;
        this.vaultLocal = new VaultLocal(vaultEndpoint, vaultToken);
    }

    @Override
    public void initialize() {
        var rootSubject = "CN=root-cert, O=" + serviceName;
        var data = new HashMap<String, String>();

        CaData caData = CaData.withSubject(rootSubject);
        data.put("certificate", Base64.getEncoder().encodeToString(caData.certificate().getBytes()));
        data.put("privateKey", Base64.getEncoder().encodeToString(caData.privateKey().getBytes()));

        vaultLocal.storeSecret(VAULT_PATH + "/" + serviceName, data);
    }
}