package org.asamk.signal.manager.storage.identities;

import org.signal.libsignal.protocol.IdentityKey;
import org.signal.libsignal.protocol.IdentityKeyPair;
import org.signal.libsignal.protocol.SignalProtocolAddress;

import java.util.function.Supplier;

public class SignalIdentityKeyStore implements org.signal.libsignal.protocol.state.IdentityKeyStore {

    private final Supplier<IdentityKeyPair> identityKeyPairSupplier;
    private final int localRegistrationId;
    private final IdentityKeyStore identityKeyStore;

    public SignalIdentityKeyStore(
            final Supplier<IdentityKeyPair> identityKeyPairSupplier,
            final int localRegistrationId,
            final IdentityKeyStore identityKeyStore
    ) {
        this.identityKeyPairSupplier = identityKeyPairSupplier;
        this.localRegistrationId = localRegistrationId;
        this.identityKeyStore = identityKeyStore;
    }

    @Override
    public IdentityKeyPair getIdentityKeyPair() {
        return identityKeyPairSupplier.get();
    }

    @Override
    public int getLocalRegistrationId() {
        return localRegistrationId;
    }

    @Override
    public IdentityChange saveIdentity(SignalProtocolAddress address, IdentityKey identityKey) {
        return identityKeyStore.saveIdentity(address.getName(), identityKey);
    }

    @Override
    public boolean isTrustedIdentity(SignalProtocolAddress address, IdentityKey identityKey, Direction direction) {
        return identityKeyStore.isTrustedIdentity(address.getName(), identityKey, direction);
    }

    @Override
    public IdentityKey getIdentity(SignalProtocolAddress address) {
        final var identityInfo = identityKeyStore.getIdentityInfo(address.getName());
        return identityInfo == null ? null : identityInfo.getIdentityKey();
    }
}