com.github.DNAProject.sdk.claim.Claim Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of DNA-java-sdk Show documentation
The newest version!
/*
 * Copyright (C) 2018 The DNA Authors
 * This file is part of The DNA library.
 *
 *  The DNA is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  The DNA is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with The DNA.  If not, see .
 *
 */

package com.github.DNAProject.sdk.claim;

import com.alibaba.fastjson.annotation.JSONField;
import com.github.DNAProject.account.Account;
import com.github.DNAProject.common.Helper;
import com.github.DNAProject.core.DataSignature;
import com.github.DNAProject.crypto.Digest;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.github.DNAProject.crypto.SignatureScheme;

import java.text.SimpleDateFormat;
import java.util.*;

/**
 * Claim
 */
public class Claim {
    private String context = "";
    private String id = UUID.randomUUID().toString();
    private Map claim = new HashMap();
    String ClaimStr = "";

    public Claim(SignatureScheme scheme, Account acct, String ctx, Map claimMap, Map metadata,String publicKeyId) {
        context = ctx;
        claim.put("Context", context);
        if (claimMap != null) {
            claim.put("Content", claimMap);
        }
        claim.put("Metadata", new MetaData(metadata).getJson());
        id = Helper.toHexString(Digest.sha256(JSON.toJSONString(claim).getBytes()));
        claim.put("Id", id);
        claim.put("Version", "v1.0");
        DataSignature sign = new DataSignature(scheme, acct, getClaim());
        byte[] signature = sign.signature();
        SignatureInfo info = new SignatureInfo("", "",publicKeyId, signature);
        claim.put("Signature", info.getJson());

    }

    /**
     *
     * @param scheme
     * @param acct
     * @param ctx
     * @param clmMap
     * @param metadata
     * @param publicKeyId
     * @param expireTime
     * @throws Exception
     */
    public Claim(SignatureScheme scheme, Account acct, String ctx, Map clmMap, Map metadata,Map clmRevMap,String publicKeyId,long expireTime) throws Exception {
        String iss = metadata.get("Issuer");
        String sub = metadata.get("Subject");
        Header header = new Header("","",publicKeyId);
        Payload payload = new Payload("v1.0",iss,sub,System.currentTimeMillis()/1000,expireTime,ctx,clmMap,clmRevMap);
        String headerStr = JSONObject.toJSONString(header.getJson());
        String payloadStr = JSONObject.toJSONString(payload.getJson());
        byte[] headerBytes = Base64.getEncoder().encode(headerStr.getBytes());
        byte[] payloadBytes = Base64.getEncoder().encode(payloadStr.getBytes());
        DataSignature sign = new DataSignature(scheme, acct, new String(headerBytes) + "." + new String(payloadBytes));
        byte[] signature = sign.signature();
        ClaimStr += new String(headerBytes) + "." + new String(payloadBytes) + "." + new String(Base64.getEncoder().encode(signature));
    }


    public String getClaim() {
        Map tmp = new HashMap();
        for (Map.Entry e : claim.entrySet()) {
            tmp.put(e.getKey(), e.getValue());
        }
        return JSONObject.toJSONString(tmp);
    }
    public String getClaimStr() {
        return ClaimStr;
    }
}

class Header {
    public String Alg = "DNA-ES256";
    public String Typ = "JWT-X";
    public String Kid;
    public Header(String alg,String typ, String kid) {
//        Alg = alg;
//        Typ = typ;
        Kid = kid;
    }
    public Object getJson() {
        Map header = new HashMap();
        header.put("alg", Alg);
        header.put("typ", Typ);
        header.put("kid", Kid);
        return header;
    }
}
class Payload {
    public String Ver;
    public String Iss;
    public String Sub;
    public long Iat;
    public long Exp;
    public String Jti;
    @JSONField(name = "@context")
    public String Context;
    public Map ClmMap = new HashMap();
    public Map ClmRevMap = new HashMap();

    public Payload(String ver,String iss,String sub,long iat,long exp,String ctx,Map clmMap,Map clmRevMap) {
        Ver = ver;
        Iss = iss;
        Sub = sub;
        Iat = iat;
        Exp = exp;
        Context = ctx;
        ClmMap = clmMap;
        ClmRevMap = clmRevMap;
        Jti = Helper.toHexString(Digest.sha256(JSON.toJSONString(getJson()).getBytes()));
    }

    public Object getJson() {
        Map payload = new HashMap();
        payload.put("ver", Ver);
        payload.put("iss", Iss);
        payload.put("sub", Sub);
        payload.put("iat", Iat);
        payload.put("exp", Exp);
        payload.put("jti", Jti);
        payload.put("@context", Context);
        payload.put("clm",ClmMap);
        payload.put("clm-rev",ClmRevMap);
        return payload;
    }
}
class SignatureInfo {

    private String Format = "pgp";
    private String Algorithm = "ECDSAwithSHA256";
    private byte[] Value;
    private String PublicKeyId;

    public SignatureInfo(String format, String alg ,String publicKeyId,byte[] val) {
        Value = val;
        PublicKeyId = publicKeyId;
    }

    public Object getJson() {
        Map signature = new HashMap();
        signature.put("Format", Format);
        signature.put("Algorithm", Algorithm);
        signature.put("Value", Value);
        signature.put("PublicKeyId", PublicKeyId);
        return signature;
    }
}

class MetaData {

    private String CreateTime = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'").format(new Date());//"2017-08-25T10:03:04Z";
    private Map meta = new HashMap();

    public MetaData(Map map) {
        if (map != null) {
            meta = map;
        }
    }

    public Object getJson() {
        meta.put("CreateTime", CreateTime);
        Map tmp = new HashMap();
        for (Map.Entry e : meta.entrySet()) {
            tmp.put(e.getKey(), e.getValue());
        }
        return tmp;
    }
}