key_manager.py

from __future__ import annotations

from db import DatabaseManager
from singleton import singleton
from typing import TYPE_CHECKING

from keys import Key

if TYPE_CHECKING:
    from keys import KeyType


@singleton
class KeysManager:
    def __init__(self) -> None:
        self.db = DatabaseManager()
        self._cached_keys: list[Key] = []

    def create_key(self, key_type: KeyType) -> Key:
        unalloted_keys = self.get_unalloted_keys(key_type)
        if unalloted_keys:
            return unalloted_keys[0]
        _key = Key(Key._generate_key(), key_type)
        self._cached_keys.append(_key)
        return _key


    def has_key(self, key_val: str) -> bool:
        self.db.crsr.execute(
            """
            SELECT value FROM keys
            WHERE value = ?
            """, [key_val]
        )
        return bool(self.db.crsr.fetchone())
    
    def get_unalloted_keys(self, of_type: KeyType):
        self.db.crsr.execute(
            """
            SELECT * FROM keys
            WHERE owner_name IS NULL and type = ?
            """, [of_type]
        )
        keys = self.db.crsr.fetchall()
        return [Key.from_tuple(key) for key in keys]

    def get_key(self, key_val: str) -> Key:
        self.db.crsr.execute(
            """
            SELECT * FROM keys
            WHERE value = ?
            """, [key_val]
        )
        key_tup = self.db.crsr.fetchone()
        return Key.from_tuple(key_tup)

    def list_all_keys(self):
        self.db.crsr.execute("""
        SELECT * FROM keys;
        """)
        keys = self.db.crsr.fetchall()
        return [Key.from_tuple(key_tup) for key_tup in keys]

    def _save(self) -> None:
        self.db.crsr.executemany(
            """
            INSERT INTO keys
            VALUES (?, ?, ?)
            ON CONFLICT(value) DO UPDATE SET owner_name=excluded.owner_name;
            """, [key.to_tuple() for key in self._cached_keys]
        )
        self.db.commit()
        self._cached_keys.clear()

    def _mark_dirty(self, key: Key):
        self._cached_keys.append(key)