syncer.py

from datetime import datetime
from json import dumps as json_dumps
from logging import getLogger

from conf import CONFIG
from ib_hosted import ib
from notifier import Notifier
from util import Asset, State, get_all_live_assets, get_assets
from util.redis import REDIS

FADE_TIME = 0.5
SLIDE_TIME = 10
log = getLogger("Syncer")

log.info("Starting sync")


def asset_to_tiles(asset: Asset):
    log.debug("adding {} to Page".format(asset.id))

    tiles = []
    if asset.filetype == "video":
        tiles.append(
            {
                "type": "rawvideo",
                "asset": asset.id,
                "x1": 0,
                "y1": 0,
                "x2": 1920,
                "y2": 1080,
                "config": {
                    "fade_time": FADE_TIME,
                    "layer": -5,
                    "looped": True,
                },
            }
        )
    else:
        tiles.append(
            {
                "type": "image",
                "asset": asset.id,
                "x1": 0,
                "y1": 0,
                "x2": 1920,
                "y2": 1080,
                "config": {"fade_time": FADE_TIME},
            }
        )

    user_is_admin = REDIS.get(f"admin:{asset.userid}")
    if isinstance(user_is_admin, bytes):
        # somehow, writing "1" into redis makes it return bytes instead
        # of a string?
        user_is_admin = user_is_admin.decode()
    if user_is_admin != "1":
        tiles.append(
            {
                "type": "flat",
                "asset": "flat.png",
                "x1": 0,
                "y1": 1040,
                "x2": 1920,
                "y2": 1080,
                "config": {"color": "#000000", "alpha": 230, "fade_time": FADE_TIME},
            }
        )
        tiles.append(
            {
                "type": "markup",
                "asset": "default-font.ttf",
                "x1": 150,
                "y1": 1048,
                "x2": 1900,
                "y2": 1080,
                "config": {
                    "font_size": 25,
                    "fade_time": FADE_TIME,
                    "text": "Project by {user} - visit {url} to share your own.".format(
                        user=asset.username,
                        url=CONFIG["DOMAIN"],
                    ),
                    "color": "#dddddd",
                },
            }
        )
    if "EXTRA_ASSETS" in CONFIG:
        tiles.extend(CONFIG["EXTRA_ASSETS"])
    return tiles


if datetime.now().minute == int(CONFIG["NOTIFIER"].get("ALERT_MINUTE", 7)):
    n = Notifier()
    asset_states = {}
    for asset in get_assets():
        if asset.state not in (State.CONFIRMED, State.REJECTED, State.DELETED):
            if asset.state not in asset_states:
                asset_states[asset.state] = 0
            asset_states[asset.state] += 1
            log.info(
                f"asset {asset.id} is in state {asset.state}, uploaded by {asset.username}"
            )

    msg = []
    for state, count in sorted(asset_states.items()):
        msg.append(f"{count} assets in state {state}.")

    if msg:
        n.message(" ".join(msg), level="WARN")


pages = []
assets_visible = set()
for asset in get_all_live_assets():
    pages.append(
        {
            "auto_duration": SLIDE_TIME,
            "duration": SLIDE_TIME
            - (
                FADE_TIME * 2
            ),  # Because it seems like the fade time is exclusive of the 10 sec, so videos play for 11 secs.
            "interaction": {"key": ""},
            "layout_id": -1,  # Use first layout
            "overlap": 0,
            "tiles": asset_to_tiles(asset),
        }
    )
    assets_visible.add(asset.id)

log.info(
    "There are currently {} pages visible with asset ids: {}".format(
        len(pages), ", ".join([str(i) for i in sorted(assets_visible)])
    )
)

for setup_id in CONFIG["SETUP_IDS"]:
    slog = getLogger(f"Setup {setup_id}")
    slog.info("Getting old config")
    config = ib.get(f"setup/{setup_id}")["config"][""]
    setup_changed = False

    for schedule in config["schedules"]:
        if schedule["name"] == "User Content":
            slog.info('Found schedule "User Content"')
            assets_shown = set()

            for page in schedule["pages"]:
                for tile in page["tiles"]:
                    if tile["type"] in ("image", "rawvideo"):
                        assets_shown.add(tile["asset"])

            slog.info(
                "schedule shows assets: {}".format(
                    ", ".join([str(i) for i in sorted(assets_shown)])
                )
            )
            if assets_visible != assets_shown:
                schedule["pages"] = pages
                setup_changed = True

    if setup_changed:
        slog.warning("Config has changed, updating")
        ib.post(
            f"setup/{setup_id}",
            config=json_dumps({"": config}),
            mode="update",
        )
    else:
        slog.info("Config has not changed, skipping update")

log.info("updated everything")