from pathlib import Path
from typing import Any


class WorkspaceFile:
    name: str
    children: list['WorkspaceFile']

    def __init__(self, name: str, children: list['WorkspaceFile']):
        self.name = name
        self.children = children

    def to_dict(self) -> dict[str, Any]:
        """Converts the File object to a dictionary.

        Returns:
            The dictionary representation of the File object.
        """
        return {
            'name': self.name,
            'children': [child.to_dict() for child in self.children],
        }


def get_folder_structure(workdir: Path) -> WorkspaceFile:
    """Gets the folder structure of a directory.

    Args:
        workdir: The directory path.

    Returns:
        The folder structure.
    """
    root = WorkspaceFile(name=workdir.name, children=[])
    for item in workdir.iterdir():
        if item.is_dir():
            dir = get_folder_structure(item)
            if dir.children:
                root.children.append(dir)
        else:
            root.children.append(WorkspaceFile(name=item.name, children=[]))
    return root