import enum

from pydantic import BaseModel


class Role(enum.StrEnum):
    VIEW_FAN_DATA = "fansifter_can_view_fan_data"
    CREATE_AD_REPORTS = "fansifter_can_create_ad_reports"
    SHARE_AD_CAMPAIGN_AUDIENCES = "fansifter_can_share_ad_campaign_audiences"
    CONNECT_AD_ACCOUNTS = "fansifter_can_connect_ad_accounts"
    VIEW_EMAIL_CAMPAIGNS = "fansifter_can_view_email_campaigns"
    CREATE_EMAIL_CAMPAIGNS = "fansifter_can_create_email_campaigns"
    VIEW_SMS_CAMPAIGNS = "fansifter_can_view_sms_campaigns"
    CREATE_SMS_CAMPAIGNS = "fansifter_can_create_sms_campaigns"


class TestUser(enum.StrEnum):
    NO_ACCESS = "no-access"
    BASIC = "basic"
    BASIC_SHOPIFY = "basic-shopify"
    AD_REPORTING = "ad-reporting"
    AD_AUDIENCES = "ad-audiences"
    APP_CONN_SHOPIFY = "app-conn-shopify"
    APP_CONN = "app-conn"
    EMAIL_VIEW = "email-view"
    EMAIL_CREATE = "email-create"
    SMS_VIEW = "sms-view"
    SMS_CREATE = "sms-create"
    AD_REP_AD_AUD_EMAIL_VIEW = "ad-rep-ad-aud-email-view"
    AD_REP_AD_AUD_EMAIL_CREATE_SMS_VIEW = "ad-rep-ad-aud-email-create-sms-view"
    CAPRICORN = "capricorn"
    AQUARIUS = "aquarius"
    PISCES = "pisces"
    ARIES = "aries"
    TAURUS = "taurus"
    GEMINI = "gemini"
    CANCER = "cancer"
    LEO = "leo"
    VIRGO = "virgo"
    LIBRA = "libra"
    SCORPIO = "scorpio"
    OPHIUCHUS = "ophiuchus"
    SAGITTARIUS = "sagittarius"


class IntegrationUser(BaseModel):
    email: str
    tenants: dict[str, list[Role]]