"""Secure Document Storage POC."""

from time import time
import boto3
from secdoc.config import KEY_ALIAS
from secdoc.config import TABLE_NAME
from secdoc.manager import SecureDocumentManager


def demo_save(item_data):
    """TODO."""
    table = boto3.resource('dynamodb').Table(TABLE_NAME)
    manager = SecureDocumentManager(table, KEY_ALIAS)
    manager.save_item(item_data)


def demo_read(item_key):
    """TODO."""
    table = boto3.resource('dynamodb').Table(TABLE_NAME)
    manager = SecureDocumentManager(table, KEY_ALIAS)
    return manager.get_item(item_key)


def main():
    """TODO."""
    item_data = {
        'service-name': 'ows-poc',
        'group-name': 'poc-data-{}'.format(time()),
        'ssn': '123-456-78901',
        'binary-field': b'\x00\x01\x02',
        'account-number': '#AB08',
        'favorite-color': 'Red'
    }
    demo_save(item_data)
    item_key = SecureDocumentManager.get_item_index(item_data)
    retrieved_item = demo_read(item_key)
    assert retrieved_item == item_data
    print('items matched')  # noqa T001


if __name__ == '__main__':
    main()