aboutsummaryrefslogtreecommitdiffstats
path: root/athenz-identity-provider-service/src/main/java/com/yahoo/vespa/hosted/athenz/instanceproviderservice/instanceconfirmation/InstanceConfirmation.java
blob: 0c6b5e67a800de972f0c6a86ff0875d816b21b17 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.athenz.instanceproviderservice.instanceconfirmation;

import com.fasterxml.jackson.annotation.JsonAnySetter;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.annotation.JsonUnwrapped;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.databind.DeserializationContext;
import com.fasterxml.jackson.databind.JsonDeserializer;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;
import com.yahoo.restapi.RestApi;
import com.yahoo.vespa.athenz.identityprovider.api.bindings.SignedIdentityDocumentEntity;
import com.yahoo.vespa.hosted.athenz.instanceproviderservice.impl.Utils;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

/**
 * InstanceConfirmation object as per Athenz InstanceConfirmation API.
 *
 * @author bjorncs
 */
public class InstanceConfirmation implements RestApi.JacksonRequestEntity {

    @JsonProperty("provider") public final String provider;
    @JsonProperty("domain") public final String domain;
    @JsonProperty("service") public final String service;

    @JsonProperty("attestationData") @JsonSerialize(using = SignedIdentitySerializer.class)
    public final SignedIdentityDocumentEntity signedIdentityDocument;
    @JsonUnwrapped public final Map<String, String> attributes = new HashMap<>(); // optional attributes that Athenz may provide

    @JsonCreator
    public InstanceConfirmation(@JsonProperty("provider") String provider,
                                @JsonProperty("domain") String domain,
                                @JsonProperty("service") String service,
                                @JsonProperty("attestationData") @JsonDeserialize(using = SignedIdentityDeserializer.class)
                                            SignedIdentityDocumentEntity signedIdentityDocument) {
        this.provider = provider;
        this.domain = domain;
        this.service = service;
        this.signedIdentityDocument = signedIdentityDocument;
    }

    @JsonAnySetter
    public void set(String name, String value) {
        attributes.put(name, value);
    }

    @Override
    public String toString() {
        return "InstanceConfirmation{" +
                "provider='" + provider + '\'' +
                ", domain='" + domain + '\'' +
                ", service='" + service + '\'' +
                ", signedIdentityDocument='" + signedIdentityDocument + '\'' +
                ", attributes=" + attributes +
                '}';
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        InstanceConfirmation that = (InstanceConfirmation) o;
        return Objects.equals(provider, that.provider) &&
                Objects.equals(domain, that.domain) &&
                Objects.equals(service, that.service) &&
                Objects.equals(signedIdentityDocument, that.signedIdentityDocument) &&
                Objects.equals(attributes, that.attributes);
    }

    @Override
    public int hashCode() {
        return Objects.hash(provider, domain, service, signedIdentityDocument, attributes);
    }

    public static class SignedIdentityDeserializer extends JsonDeserializer<SignedIdentityDocumentEntity> {
        @Override
        public SignedIdentityDocumentEntity deserialize(
                JsonParser jsonParser, DeserializationContext deserializationContext) throws IOException {
            String value = jsonParser.getValueAsString();
            return Utils.getMapper().readValue(value, SignedIdentityDocumentEntity.class);
        }
    }

    public static class SignedIdentitySerializer extends JsonSerializer<SignedIdentityDocumentEntity> {
        @Override
        public void serialize(
                SignedIdentityDocumentEntity document, JsonGenerator gen, SerializerProvider serializers) throws IOException {
            gen.writeString(Utils.getMapper().writeValueAsString(document));
        }
    }
}