49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from typing import Union, Tuple
|
|
import tarfile
|
|
import io
|
|
import os
|
|
|
|
from coredb import blockmetadb
|
|
from onionrblocks import onionrblockapi
|
|
from onionrblocks import insert
|
|
|
|
# Import types. Just for type hiting
|
|
from onionrtypes import UserID, DeterministicKeyPassphrase, BlockHash
|
|
|
|
from onionrcrypto import generate_deterministic
|
|
|
|
def find_site_gzip(user_id: str)->str:
|
|
sites = blockmetadb.get_blocks_by_type('osite')
|
|
user_site = None
|
|
for site in sites:
|
|
block = onionrblockapi.Block(site)
|
|
if block.isSigner(user_id):
|
|
user_site = block
|
|
if not user_site is None:
|
|
return tarfile.open(fileobj=io.BytesIO(user_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
|
|
|
|
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)
|
|
|
|
block_hash = insert(raw_tar.read(), signing_key=private_key)
|
|
|
|
return (public_key, block_hash)
|