# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

from ..samples.toyctf import tinytoy
from . import cyberbattle_env
from ..simulation.model import NodeInfo


class CyberBattleTiny(cyberbattle_env.CyberBattleEnv):
    """CyberBattle simulation on a tiny environment. (Useful for debugging purpose)"""
    _ori_nodes: dict[str, NodeInfo]
    def __init__(self, **kwargs):
        self._ori_nodes = tinytoy.nodes
        super().__init__(
            initial_environment=tinytoy.new_environment(),
            **kwargs)

    @property
    def ori_nodes(self):
        return self._ori_nodes
