pyssb/ssb/feed.py
2017-08-05 18:21:14 +02:00

104 lines
2.9 KiB
Python

import time
from base64 import b64encode
from collections import namedtuple, OrderedDict
from hashlib import sha256
from simplejson import dumps, loads
from ssb.util import tag
OrderedMsg = namedtuple('OrderedMsg', ('previous', 'author', 'sequence', 'timestamp', 'hash', 'content'))
class NoPrivateKeyException(Exception):
pass
def to_ordered(data):
smsg = OrderedMsg(**data)
return OrderedDict((k, getattr(smsg, k)) for k in smsg._fields)
class Feed(object):
def __init__(self, public_key):
self.public_key = public_key
@property
def id(self):
return tag(self.public_key).decode('ascii')
def sign(self):
raise NoPrivateKeyException('Cannot use remote identity to sign (no private key!)')
class LocalFeed(Feed):
def __init__(self, private_key):
self.private_key = private_key
@property
def public_key(self):
return self.private_key.verify_key
def sign(self, msg):
return self.private_key.sign(msg).signature
class Message(object):
def __init__(self, feed, content, signature, sequence=1, timestamp=None, previous=None):
self.feed = feed
self.content = content
self.signature = signature
self.previous = previous
if self.previous:
self.sequence = self.previous.sequence + 1
else:
self.sequence = sequence
self.timestamp = int(time.time() * 1000) if timestamp is None else timestamp
@classmethod
def parse(cls, data, feed):
obj = loads(data, object_pairs_hook=OrderedDict)
msg = cls(feed, obj['content'], timestamp=obj['timestamp'])
return msg, obj['signature']
def to_dict(self, add_signature=True):
obj = to_ordered({
'previous': self.previous.key if self.previous else None,
'author': self.feed.id,
'sequence': self.sequence,
'timestamp': self.timestamp,
'hash': 'sha256',
'content': self.content
})
if add_signature:
obj['signature'] = self.signature
return obj
def verify(self, signature):
return self.signature == signature
@property
def hash(self):
hash = sha256(dumps(self.to_dict(), indent=2).encode('ascii')).digest()
return b64encode(hash).decode('ascii') + '.sha256'
@property
def key(self):
return '%' + self.hash
class LocalMessage(Message):
def __init__(self, feed, content, signature=None, sequence=1, timestamp=None, previous=None):
super(LocalMessage, self).__init__(feed, content, signature, sequence, timestamp, previous)
if signature is None:
self.signature = self.sign()
def sign(self):
# ensure ordering of keys and indentation of 2 characters, like ssb-keys
data = dumps(self.to_dict(add_signature=False), indent=2)
return (b64encode(bytes(self.feed.sign(data.encode('ascii')))) + b'.sig.ed25519').decode('ascii')