# Code generated by Lark OpenAPI.

from typing import Any, Optional, Union, Dict, List, Set, IO, Callable, Type
from lark_oapi.core.construct import init
from .move_result import MoveResult
from .delete_space_result import DeleteSpaceResult
from .simple_task_result import SimpleTaskResult


class TaskResult(object):
    _types = {
        "task_id": str,
        "move_result": List[MoveResult],
        "delete_space_result": DeleteSpaceResult,
        "simple_task_result": SimpleTaskResult,
    }

    def __init__(self, d=None):
        self.task_id: Optional[str] = None
        self.move_result: Optional[List[MoveResult]] = None
        self.delete_space_result: Optional[DeleteSpaceResult] = None
        self.simple_task_result: Optional[SimpleTaskResult] = None
        init(self, d, self._types)

    @staticmethod
    def builder() -> "TaskResultBuilder":
        return TaskResultBuilder()


class TaskResultBuilder(object):
    def __init__(self) -> None:
        self._task_result = TaskResult()

    def task_id(self, task_id: str) -> "TaskResultBuilder":
        self._task_result.task_id = task_id
        return self

    def move_result(self, move_result: List[MoveResult]) -> "TaskResultBuilder":
        self._task_result.move_result = move_result
        return self

    def delete_space_result(self, delete_space_result: DeleteSpaceResult) -> "TaskResultBuilder":
        self._task_result.delete_space_result = delete_space_result
        return self

    def simple_task_result(self, simple_task_result: SimpleTaskResult) -> "TaskResultBuilder":
        self._task_result.simple_task_result = simple_task_result
        return self

    def build(self) -> "TaskResult":
        return self._task_result