153 lines
3.9 KiB
Python
153 lines
3.9 KiB
Python
# standard imports
|
|
import os
|
|
import json
|
|
import logging
|
|
import tempfile
|
|
import cbor2
|
|
|
|
# external imports
|
|
from hexathon import strip_0x
|
|
from cic_types import MetadataPointer
|
|
from cic_types.processor import generate_metadata_pointer
|
|
from cic_types.ext.metadata import MetadataRequestsHandler
|
|
|
|
# local imports
|
|
from .base import *
|
|
from cic.output import OutputWriter
|
|
|
|
logg = logging.getLogger(__name__)
|
|
|
|
|
|
class Proof(Data):
|
|
|
|
def __init__(self, path='.', attachments=None, writer=None):
|
|
super(Proof, self).__init__()
|
|
self.proofs = []
|
|
self.namespace = 'ge'
|
|
self.description = None
|
|
self.path = path
|
|
self.writer = writer
|
|
self.extra_attachments = attachments
|
|
self.attachments = {}
|
|
self.proof_path = os.path.join(self.path, 'proof.json')
|
|
self.temp_proof_path = tempfile.mkstemp()[1]
|
|
|
|
|
|
def load(self):
|
|
super(Proof, self).load()
|
|
|
|
f = open(self.proof_path, 'r')
|
|
o = json.load(f)
|
|
f.close()
|
|
|
|
self.set_version(o['version'])
|
|
self.description = o['description']
|
|
self.namespace = o['namespace']
|
|
self.issuer = o['issuer']
|
|
self.proofs = o['proofs']
|
|
|
|
if self.extra_attachments != None:
|
|
a = self.extra_attachments.asdict()
|
|
for k in a.keys():
|
|
self.attachments[k] = a[k]
|
|
|
|
hshs = self.__get_ordered_hashes()
|
|
self.proofs = list(map(strip_0x, hshs))
|
|
|
|
self.inited = True
|
|
|
|
|
|
def start(self):
|
|
super(Proof, self).start()
|
|
|
|
proof_template_file_path = os.path.join(data_dir, 'proof_template_v{}.json'.format(self.version()))
|
|
|
|
f = open(proof_template_file_path)
|
|
o = json.load(f)
|
|
f.close()
|
|
|
|
f = open(self.proof_path, 'w')
|
|
json.dump(o, f, sort_keys=True, indent="\t")
|
|
f.close()
|
|
|
|
|
|
def asdict(self):
|
|
return {
|
|
'version': self.version(),
|
|
'namespace': self.namespace,
|
|
'description': self.description,
|
|
'issuer': self.issuer,
|
|
'proofs': self.proofs,
|
|
}
|
|
|
|
|
|
def __get_ordered_hashes(self):
|
|
ks = list(self.attachments.keys())
|
|
ks.sort()
|
|
|
|
return ks
|
|
|
|
|
|
# def get(self):
|
|
# hsh = self.hash(b).hex()
|
|
# self.attachments[hsh] = self.temp_proof_path
|
|
# logg.debug('cbor of {} is {} hashes to {}'.format(v, b.hex(), hsh))
|
|
|
|
|
|
def root(self):
|
|
v = self.asdict()
|
|
#b = cbor2.dumps(v)
|
|
b = json.dumps(v)
|
|
|
|
f = open(self.temp_proof_path, 'w')
|
|
f.write(b)
|
|
f.close()
|
|
|
|
k = self.hash(b.encode('utf-8'))
|
|
|
|
return (k.hex(), b)
|
|
|
|
|
|
def process(self, token_address=None, token_symbol=None, writer=None):
|
|
if writer == None:
|
|
writer = self.writer
|
|
|
|
(k, v) = self.root()
|
|
writer.write(k, v)
|
|
|
|
token_symbol_bytes = token_symbol.encode('utf-8')
|
|
k = generate_metadata_pointer(token_symbol_bytes, MetadataPointer.TOKEN_PROOF_SYMBOL)
|
|
writer.write(k, v)
|
|
|
|
token_address_bytes = bytes.fromhex(strip_0x(token_address))
|
|
k = generate_metadata_pointer(token_address_bytes, MetadataPointer.TOKEN_PROOF)
|
|
writer.write(k, v)
|
|
|
|
# (hsh, hshs) = self.get()
|
|
#hshs = list(map(strip_0x, hshs))
|
|
# hshs_bin = list(map(bytes.fromhex, hshs))
|
|
# hshs_cat = b''.join(hshs_bin)
|
|
|
|
# f = open(self.temp_proof_path, 'rb')
|
|
# v = f.read()
|
|
# f.close()
|
|
# writer.write(hsh, v)
|
|
|
|
# r = self.hash(hshs_cat)
|
|
# r_hex = r.hex()
|
|
|
|
#logg.debug('generated proof {} for hashes {}'.format(r_hex, hshs))
|
|
|
|
#writer.write(r_hex, hshs_cat)
|
|
|
|
o = self.asdict()
|
|
f = open(self.proof_path, 'w')
|
|
json.dump(o, f, sort_keys=True, indent="\t")
|
|
f.close()
|
|
|
|
return k
|
|
|
|
|
|
def __str__(self):
|
|
return "description = {}\n".format(self.description)
|