Onionr/onionr/httpapi/onionrsitesapi/sitefiles.py

45 lines
1.2 KiB
Python
Raw Normal View History

2019-11-04 10:52:38 +00:00
from typing import Union, Tuple
2019-10-31 09:29:03 +00:00
import tarfile
import io
import os
2019-10-31 09:29:03 +00:00
from coredb import blockmetadb
from onionrblocks import onionrblockapi
from onionrblocks import insert
2019-11-04 10:52:38 +00:00
# Import types. Just for type hiting
from onionrtypes import UserID, DeterministicKeyPassphrase, BlockHash
from onionrcrypto import generate_deterministic
2019-10-31 09:29:03 +00:00
def find_site_gzip(user_id: str)->str:
sites = blockmetadb.get_blocks_by_type('osite')
for site in sites:
if onionrblockapi.Block(site).isSigner(user_id):
return tarfile.open(fileobj=io.BytesIO(site.bcontent), mode='r')
return None
def get_file(user_id, file)->Union[bytes, None]:
ret_data = ""
site = find_site_gzip(user_id)
if site is None: return None
for file in site.getmembers():
if file.name == file:
return site.extractfile(file)
return None
2019-11-04 10:52:38 +00:00
def create_site(admin_pass: DeterministicKeyPassphrase, directory:str='.')->Tuple[UserID, BlockHash]:
public_key, private_key = generate_deterministic(admin_pass)
raw_tar = io.BytesIO()
tar = tarfile.open(mode='x:gz', fileobj=raw_tar)
tar.add(directory)
tar.close()
raw_tar.seek(0)
2019-11-04 10:52:38 +00:00
block_hash = insert(raw_tar.read(), signing_key=private_key)
2019-11-04 10:52:38 +00:00
return (public_key, block_hash)