it.auties.whatsapp.model.signal.session.SessionChain Maven / Gradle / Ivy

Go to download
package it.auties.whatsapp.model.signal.session;


import it.auties.protobuf.annotation.ProtobufProperty;
import it.auties.protobuf.model.ProtobufMessage;
import it.auties.protobuf.model.ProtobufType;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReference;

public record SessionChain(
        @ProtobufProperty(index = 1, type = ProtobufType.INT32)
        AtomicInteger counter,
        @ProtobufProperty(index = 2, type = ProtobufType.BYTES)
        AtomicReference key,
        @ProtobufProperty(index = 3, type = ProtobufType.MAP, keyType = ProtobufType.INT32, valueType = ProtobufType.BYTES)
        ConcurrentHashMap messageKeys
) implements ProtobufMessage {
    public SessionChain(int counter, byte[] key) {
        this(new AtomicInteger(counter), new AtomicReference<>(key), new ConcurrentHashMap<>());
    }

    public boolean hasMessageKey(int counter) {
        return messageKeys.containsKey(counter);
    }
}