from sqlalchemy.dialects.postgresql import insert
from atlas_um.pgdb import InternalUserDomain, pgdb

from atlas_um.tasks import (
    task_orchestrate_update_token_length_for_dna_accounts,
)


INTERNAL_USER_DOMAINS = (
    "sonymusic.com",
    "rcarecords.com",
    "epicrecords.com",
    "pmgsonymusic.com",
    "aristarecordings.com",
    "ministryofsoundrecords.com",
    "recordsco.com",
    "redmusic.com",
    "centurymedia.de",
    "fourmusic.com",
    "sonymusicexternal.com",
    "liquidstate.co",
    "select.co.za",
    "nowmusic.com",
    "zeligwest.com",
    "workof-art.com",
    "stachemedia.com",
    "insideoutmusic.com",
)


def _set_internal_user_domains():
    pgdb.session.execute(
        insert(InternalUserDomain).on_conflict_do_nothing(),
        [{"domain": item} for item in INTERNAL_USER_DOMAINS],
    )
    pgdb.session.commit()


def _set_token_length_for_dna_accounts():
    task_orchestrate_update_token_length_for_dna_accounts.delay()


def init_app_presets():
    _set_internal_user_domains()
    _set_token_length_for_dna_accounts()