# 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 .entity_info import EntityInfo
from .entity_info import EntityInfo


class VisibleAssistRule(object):
    _types = {
        "id": str,
        "assist_rule_type": int,
        "subjects": List[EntityInfo],
        "objects": List[EntityInfo],
        "effect": int,
        "match_condition": int,
    }

    def __init__(self, d=None):
        self.id: Optional[str] = None
        self.assist_rule_type: Optional[int] = None
        self.subjects: Optional[List[EntityInfo]] = None
        self.objects: Optional[List[EntityInfo]] = None
        self.effect: Optional[int] = None
        self.match_condition: Optional[int] = None
        init(self, d, self._types)

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


class VisibleAssistRuleBuilder(object):
    def __init__(self) -> None:
        self._visible_assist_rule = VisibleAssistRule()

    def id(self, id: str) -> "VisibleAssistRuleBuilder":
        self._visible_assist_rule.id = id
        return self

    def assist_rule_type(self, assist_rule_type: int) -> "VisibleAssistRuleBuilder":
        self._visible_assist_rule.assist_rule_type = assist_rule_type
        return self

    def subjects(self, subjects: List[EntityInfo]) -> "VisibleAssistRuleBuilder":
        self._visible_assist_rule.subjects = subjects
        return self

    def objects(self, objects: List[EntityInfo]) -> "VisibleAssistRuleBuilder":
        self._visible_assist_rule.objects = objects
        return self

    def effect(self, effect: int) -> "VisibleAssistRuleBuilder":
        self._visible_assist_rule.effect = effect
        return self

    def match_condition(self, match_condition: int) -> "VisibleAssistRuleBuilder":
        self._visible_assist_rule.match_condition = match_condition
        return self

    def build(self) -> "VisibleAssistRule":
        return self._visible_assist_rule