from airflow.plugins_manager import AirflowPlugin
from flask_admin.base import MenuLink

github = MenuLink(
    category="Astronomer",
    name="Airflow Instance Github Repo",
    url="https://github.com/astronomerio/astronomer-dags",
)

astronomer_home = MenuLink(
    category="Astronomer", name="Astronomer Home", url="https://www.astronomer.io/"
)

aiflow_plugins = MenuLink(
    category="Astronomer",
    name="Airflow Plugins",
    url="https://github.com/airflow-plugins",
)

# Defining the plugin class
class AirflowTestPlugin(AirflowPlugin):
    name = "AstronomerMenuLinks"
    operators = []
    flask_blueprints = []
    hooks = []
    executors = []
    admin_views = []
    menu_links = [github, astronomer_home, aiflow_plugins]