from modules import script_callbacks, shared  # pyright: ignore

try:
    from modules import ui_components  # pyright: ignore

    FormColorPicker = ui_components.FormColorPicker
except (ImportError, AttributeError):
    # for compatibility with old webui
    FormColorPicker = None


def on_ui_settings():
    section = ("enable_checker", "Enable Checker")

    shared.opts.add_option(
        "enable_checker_fix_forever_randomly_seed",
        shared.OptionInfo(
            True, "Set the value of seed to -1 when Generate forever buttons are clicked", section=section
        ),
    )

    shared.opts.add_option(
        "enable_checker_activate_dropdown_check",
        shared.OptionInfo(True, "Enable dropdown check", section=section),
    )
    shared.opts.add_option(
        "enable_checker_activate_weight_check",
        shared.OptionInfo(True, "Enable weight check", section=section),
    )
    shared.opts.add_option(
        "enable_checker_activate_extra_network_check",
        shared.OptionInfo(True, "Enable extra network check", section=section),
    )
    shared.opts.add_option(
        "enable_checker_check_version_compatibility",
        shared.OptionInfo(True, "Check version compatibility", section=section),
    )

    shared.opts.add_option(
        "enable_checker_custom_color",
        shared.OptionInfo(False, "Use custom colors", section=section),
    )
    shared.opts.add_option(
        "enable_checker_custom_color_enable",
        shared.OptionInfo(
            "#a0d8ef",
            "Custom color of enabled scripts",
            FormColorPicker,
            section=section,
        ),
    )
    shared.opts.add_option(
        "enable_checker_custom_color_disable",
        shared.OptionInfo(
            "#aeaeae",
            "Custom color of disabled scripts",
            FormColorPicker,
            section=section,
        ),
    )
    shared.opts.add_option(
        "enable_checker_custom_color_dropdown_enable",
        shared.OptionInfo(
            "#233873",
            "Custom color of enabled dropdown",
            FormColorPicker,
            section=section,
        ),
    )
    shared.opts.add_option(
        "enable_checker_custom_color_dropdown_disable",
        shared.OptionInfo(
            "#aeaeae",
            "Custom color of disabled dropdown",
            FormColorPicker,
            section=section,
        ),
    )
    shared.opts.add_option(
        "enable_checker_custom_color_zero_weihgt",
        shared.OptionInfo(
            "#aeaeae",
            "Custom color of 0 weight",
            FormColorPicker,
            section=section,
        ),
    )
    shared.opts.add_option(
        "enable_checker_custom_color_invalid_additional_networks",
        shared.OptionInfo(
            "#ed9797",
            "Custom color for invalid additional networks",
            FormColorPicker,
            section=section,
        ),
    )


script_callbacks.on_ui_settings(on_ui_settings)