from flask_admin import Admin

from src.db import db
from src.models import (CampaignPlacements, CampaignPlatformGroups, CampaignPlatforms, Label, MarketingAccount,
                        ProfitCenter)
from src.views import (CampaignPlacementsView, CampaignPlatformGroupsView, CampaignPlatformsView, HomeView, LabelView,
                       LoginMenuLink, LogoutMenuLink, MarketingAccountView, ProfitCenterView)


def init_app(app):
    admin = Admin(
        app,
        index_view=HomeView(name="Home", template="admin/index.html", url="/"),
        name="Decibel CMS",
        template_mode="bootstrap3",
    )
    admin.add_link(LoginMenuLink(name="Login", category="", url="/login"))
    admin.add_link(LogoutMenuLink(name="Logout", category="", url="/logout"))
    admin.add_view(
        CampaignPlacementsView(
            CampaignPlacements,
            db.session,
            name="CampaignPlacements",
            endpoint="/campaign-placements",
        )
    )
    admin.add_view(
        CampaignPlatformsView(
            CampaignPlatforms,
            db.session,
            name="CampaignPlatforms",
            endpoint="/campaign-platforms",
        )
    )
    admin.add_view(
        CampaignPlatformGroupsView(
            CampaignPlatformGroups,
            db.session,
            name="CampaignPlatformGroups",
            endpoint="/campaign-platforms-group",
        )
    )
    admin.add_view(
        LabelView(
            Label,
            db.session,
            name="Label",
            endpoint="/labels",
        )
    )
    admin.add_view(
        MarketingAccountView(
            MarketingAccount,
            db.session,
            name="MarketingAccount",
            endpoint="/marketing-accounts",
        )
    )

    admin.add_view(
        ProfitCenterView(
            ProfitCenter,
            db.session,
            name="ProfitCenter",
            endpoint="/profit-centers",
        )
    )