from typing import Any

from flet.controls.base_control import control
from flet.controls.services.service import Service

__all__ = ["SharedPreferences"]


@control("SharedPreferences")
class SharedPreferences(Service):
    async def set(self, key: str, value: Any) -> bool:
        assert value is not None
        return await self._invoke_method("set", {"key": key, "value": value})

    async def get(self, key: str):
        return await self._invoke_method("get", {"key": key})

    async def contains_key(self, key: str) -> bool:
        return await self._invoke_method("contains_key", {"key": key})

    async def remove(self, key: str) -> bool:
        return await self._invoke_method("remove", {"key": key})

    async def get_keys(self, key_prefix: str) -> list[str]:
        return await self._invoke_method("get_keys", {"key_prefix": key_prefix})

    async def clear(self) -> bool:
        return await self._invoke_method("clear")