from .ban_list import BooruGachaBanList
from .combine import BooruGachaCombine
from .combine_bans import BooruGachaCombineBans
from .gacha_roll import BooruGachaRoll
from .tag_expand import BooruGachaTagExpand
from .tag_search import BooruGachaSearch

NODE_CLASS_MAPPINGS = {
    "BooruGachaSearch": BooruGachaSearch,
    "BooruGachaRoll": BooruGachaRoll,
    "BooruGachaBanList": BooruGachaBanList,
    "BooruGachaCombine": BooruGachaCombine,
    "BooruGachaCombineBans": BooruGachaCombineBans,
    "BooruGachaTagExpand": BooruGachaTagExpand,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "BooruGachaSearch": "Booru Gacha Search",
    "BooruGachaRoll": "Booru Gacha Roll",
    "BooruGachaBanList": "Booru Gacha Ban List",
    "BooruGachaCombine": "Booru Gacha Combine",
    "BooruGachaCombineBans": "Booru Gacha Combine Bans",
    "BooruGachaTagExpand": "Booru Gacha Tag Expand",
}
