from typing import List

from facefusion import rtc
from facefusion.types import RtcPeer, RtcStore, SessionId

RTC_STORE : RtcStore = {}


def init_peers(session_id : SessionId) -> None:
	RTC_STORE[session_id] = []


def has_peers(session_id : SessionId) -> bool:
	return bool(RTC_STORE.get(session_id))


def get_peers(session_id : SessionId) -> List[RtcPeer]:
	return RTC_STORE.get(session_id)


def delete_peers(session_id : SessionId) -> None:
	if session_id in RTC_STORE:
		rtc_peers = get_peers(session_id)

		if rtc_peers:
			rtc.delete_peers(rtc_peers)
			del RTC_STORE[session_id]

	return None


def clear() -> None:
	RTC_STORE.clear()