from typing import List

from wizard.manager import SelectOption
from wizard.scriptrunner import WizardScriptRunner
from wizard.value import SubPackage, SubPackages


class MySubPackage(SubPackage):

    """
    Implement your own SubPackage.
    """

    _files: List[str]

    def __init__(self, name: str, files: List[str]):
        super().__init__(name)
        self._files = files

    @property
    def files(self):
        return iter(self._files)


class MyRunner(WizardScriptRunner):

    """
    Extends the runner and implement the missing methods.
    """

    # These are the methods you need to provide - See manager.ManagerModInterface
    # and manager.ManagerUserInterface for the documentation of each method.

    # The WizardRunner class extends both interfaces and already implements many
    # functions, but you can always override them.

    def selectOne(
        self, description: str, options: List[SelectOption], default: SelectOption
    ) -> SelectOption:
        ...

    def selectMany(
        self,
        description: str,
        options: List[SelectOption],
        default: List[SelectOption] = [],
    ) -> List[SelectOption]:
        ...

    def compareGameVersion(self, version: str) -> int:
        ...

    def compareSEVersion(self, version: str) -> int:
        ...

    def compareGEVersion(self, version: str) -> int:
        ...

    def compareWBVersion(self, version: str) -> int:
        ...

    def dataFileExists(self, *filepaths: str) -> bool:
        ...

    def getPluginLoadOrder(self, filename: str, fallback: int = -1) -> int:
        ...

    def getPluginStatus(self, filename) -> int:
        ...

    def getFilename(self, path: str) -> str:
        ...

    def getFolder(self, path: str) -> str:
        ...

# Create the sub-packages:
subpackages = SubPackages([MySubPackage(...) for ...])

# Create the runner:
runner = MyRunner(subpackages)

# Run a script:
status, result = runner.run("wizard.txt")

status  # Status of the execution.
result.subpackages  # List of selected subpackages.
result.plugins  # List of enabled plugins.
result.notes  # List of notes.
result.tweaks.disabled  # List of disabled INI settings.
result.tweaks.modified  # List of new or modified INI settings.

def error(self, exc: Exception):
    # Do whatever you want.
    ...

runner = WizardRunner(...)

# Abort the execution - Equivalent to a 'Cancel' keyword:
runner.abort()

# Retrieve the current context:
context = runner.context()

# Rewind to the given context:
runner.rewind(context)