document signup

nkode_api.py

@@ -4,15 +4,24 @@ 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] = {}
+    signup_sessions: dict[UUID, UserSignupSession] = {}
+
+    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
 
     def generate_index_interface(self, customer_id: UUID) -> tuple[UUID, list[int]]:
         assert (customer_id in self.customers.keys())
@@ -23,9 +32,35 @@ class NKodeAPI(BaseModel):
             set_interface=set_interface.attr_indices,
             customer_id=customer_id,
         )
-        self.sessions[new_session.session_id] = new_session
+        self.signup_sessions[new_session.session_id] = new_session
         return new_session.session_id, new_session.set_interface
 
+
+    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.signup_sessions.keys())
+        self.signup_sessions[session_id].set_user_nkode(username, customer, key_selection)
+        return self.signup_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.signup_sessions.keys() and
+                customer_id == self.signup_sessions[session_id].customer_id and
+                username == self.signup_sessions[session_id].username
+        )
+        customer = self.customers[customer_id]
+        new_user = self.signup_sessions[session_id].create_user(username, customer, confirm_key_entry)
+        self.customers[customer_id].add_new_user(new_user)
+        del self.signup_sessions[session_id]
+        return True
+
     def get_login_index_interface(self, username: str, customer_id: UUID) -> list[int]:
         """
         TODO: how do we prevent a targeted denial-of-service attack?
@@ -37,31 +72,6 @@ class NKodeAPI(BaseModel):
         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]
@@ -71,13 +81,3 @@ class NKodeAPI(BaseModel):
         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