from simple_logger_handler import setup_logger
import os
from typing import Set

logger = setup_logger(__name__)


def return_credentials(path: str) -> str:
	"""Read and return credentials from a file."""
	logger.debug(f"Opening credentials for path:{path}")
	try:
		with open(path) as file:
			return file.read().strip()
	except FileNotFoundError:
		logger.fatal(f"[FATAL] Secret file not found: {path}")
		raise
	except Exception as e:
		logger.fatal(f"[FATAL] Failed to read secret file {path}: {e}")
		raise


def cleanup_secret_files(secret_paths: Set[str]) -> None:
	"""Delete secret files after they've been loaded into memory.

	Args:
		secret_paths: Set of file paths to delete
	"""
	for path in secret_paths:
		try:
			if os.path.exists(path):
				os.remove(path)
				logger.debug(f"Deleted secret file: {path}")
			else:
				logger.debug(f"Secret file already removed: {path}")
		except Exception as e:
			logger.warning(f"Could not delete secret file {path}: {e}")