84 lines
3.4 KiB
Python
84 lines
3.4 KiB
Python
from uuid import UUID, uuid4
|
|
from pydantic import BaseModel
|
|
|
|
from src.customer import Customer
|
|
from src.models import NKodePolicy
|
|
from src.user_signup_session import UserSignupSession
|
|
from src.user_cipher_keys import UserCipherKeys
|
|
from src.user import User
|
|
from src.user_interface import UserInterface
|
|
from src.customer_interface import CustomerInterface
|
|
|
|
|
|
class NKodeAPI(BaseModel):
|
|
customers: dict[UUID, Customer] = {}
|
|
sessions: dict[UUID, UserSignupSession] = {}
|
|
|
|
def generate_index_interface(self, customer_id: UUID) -> tuple[UUID, list[int]]:
|
|
assert (customer_id in self.customers.keys())
|
|
customer = self.customers[customer_id]
|
|
set_interface = UserInterface.new(customer.interface.attrs_per_key, customer.interface.numb_of_keys)
|
|
new_session = UserSignupSession(
|
|
session_id=uuid4(),
|
|
set_interface=set_interface.attr_indices,
|
|
customer_id=customer_id,
|
|
)
|
|
self.sessions[new_session.session_id] = new_session
|
|
return new_session.session_id, new_session.set_interface
|
|
|
|
def get_login_index_interface(self, username: str, customer_id: UUID) -> list[int]:
|
|
"""
|
|
TODO: how do we prevent a targeted denial-of-service attack?
|
|
"""
|
|
assert (customer_id in self.customers.keys())
|
|
customer = self.customers[customer_id]
|
|
assert (username in customer.users.keys())
|
|
user = customer.users[username]
|
|
user.user_interface.shuffle_interface()
|
|
return user.user_interface.attr_indices
|
|
|
|
def set_nkode(
|
|
self, username: str,
|
|
customer_id: UUID,
|
|
key_selection: list[int],
|
|
session_id: UUID
|
|
) -> list[int]:
|
|
assert (customer_id in self.customers.keys())
|
|
customer = self.customers[customer_id]
|
|
assert (username not in customer.users.keys())
|
|
assert (session_id in self.sessions.keys())
|
|
self.sessions[session_id].set_user_nkode(username, customer, key_selection)
|
|
return self.sessions[session_id].confirm_interface
|
|
|
|
def confirm_nkode(self, username: str, customer_id: UUID, confirm_key_entry: list[int], session_id: UUID) -> bool:
|
|
assert (
|
|
session_id in self.sessions.keys() and
|
|
customer_id == self.sessions[session_id].customer_id and
|
|
username == self.sessions[session_id].username
|
|
)
|
|
customer = self.customers[customer_id]
|
|
new_user = self.sessions[session_id].create_user(username, customer, confirm_key_entry)
|
|
self.customers[customer_id].add_new_user(new_user)
|
|
del self.sessions[session_id]
|
|
return True
|
|
|
|
def login(self, customer_id: UUID, username: str, key_selection: list[int]) -> bool:
|
|
assert (customer_id in self.customers.keys())
|
|
customer = self.customers[customer_id]
|
|
return customer.valid_key_entry(username, key_selection)
|
|
|
|
def renew_keys(self, customer_id: UUID) -> bool:
|
|
assert (customer_id in self.customers.keys())
|
|
return self.customers[customer_id].renew_keys()
|
|
|
|
def create_new_customer(self, numb_keys: int, numb_sets: int, nkode_policy: NKodePolicy) -> UUID:
|
|
new_customer = Customer(
|
|
customer_id=uuid4(),
|
|
interface=CustomerInterface.new(numb_keys, numb_sets),
|
|
users={},
|
|
nkode_policy=nkode_policy
|
|
)
|
|
self.customers[new_customer.customer_id] = new_customer
|
|
|
|
return new_customer.customer_id
|