CMSgov/dpc-app

package gov.cms.dpc.macaroons.caveats;

import gov.cms.dpc.macaroons.config.TokenPolicy;
import gov.cms.dpc.macaroons.CaveatSupplier;
import gov.cms.dpc.macaroons.MacaroonCaveat;
import gov.cms.dpc.macaroons.MacaroonCondition;

/**
 * Implementation of {@link CaveatSupplier} which generates a version token based on the provided {@link TokenPolicy}
 */
public class VersionCaveatSupplier implements CaveatSupplier {

    static final String VERSION_KEY = "dpc_macaroon_version";

    private final String tokenVersion;

    public VersionCaveatSupplier(int tokenVersion) {
        this.tokenVersion = Integer.toString(tokenVersion);
    }

    @Override
    public MacaroonCaveat get() {
        return new MacaroonCaveat(new MacaroonCondition(VERSION_KEY, MacaroonCondition.Operator.EQ, this.tokenVersion));
    }
}