com.uid2.shared.secure.gcp.InstanceDocument Maven / Gradle / Ivy

Go to download
package com.uid2.shared.secure.gcp;

import com.google.api.client.googleapis.auth.oauth2.GoogleIdToken;
import com.google.api.client.util.ArrayMap;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.math.BigDecimal;
import java.time.Instant;

public class InstanceDocument {
    private static final Logger LOGGER = LoggerFactory.getLogger(InstanceDocument.class);

    private String audience;
    private Instant issuedAt;
    private Instant expiredAt;
    private String subject;
    private boolean instanceConfidentiality;
    private Instant instanceCreatedAt;
    private String instanceId;
    private String projectId;
    private String zone;

    public InstanceDocument(GoogleIdToken vmInstanceDocument) {
        GoogleIdToken.Payload payload = vmInstanceDocument.getPayload();
        this.audience = (String)payload.getAudience();
        this.issuedAt = Instant.ofEpochSecond(payload.getIssuedAtTimeSeconds());
        this.expiredAt = Instant.ofEpochSecond(payload.getExpirationTimeSeconds());
        this.subject = payload.getSubject();
        ArrayMap googleMap = (ArrayMap)payload.get("google");
        ArrayMap computeEngineMap = (ArrayMap)googleMap.get("compute_engine");
        if (computeEngineMap.containsKey("instance_confidentiality")) {
            BigDecimal isConfidential = (BigDecimal) computeEngineMap.get("instance_confidentiality");
            this.instanceConfidentiality = isConfidential.equals(BigDecimal.ONE);
        } else {
            this.instanceConfidentiality = false;
        }
        long createdAt = ((BigDecimal)computeEngineMap.get("instance_creation_timestamp")).longValue();
        this.instanceCreatedAt = Instant.ofEpochSecond(createdAt);
        this.instanceId = (String)computeEngineMap.get("instance_id");
        this.projectId = (String)computeEngineMap.get("project_id");
        this.zone = (String)computeEngineMap.get("zone");

        LOGGER.debug("Received instance document { " + projectId + ", " + zone + ", " + instanceId + " }");
    }

    public String getAudience() {
        return audience;
    }

    public Instant getIssuedAt() {
        return issuedAt;
    }

    public Instant getExpiredAt() {
        return expiredAt;
    }

    public String getSubject() {
       return subject;
    }

    public boolean getInstanceConfidentiality() {
        return instanceConfidentiality;
    }

    public Instant getInstanceCreatedAt() {
        return instanceCreatedAt;
    }

    public String getInstanceId() {
        return instanceId;
    }

    public String getProjectId() {
        return projectId;
    }

    public String getZone() {
        return zone;
    }
}