dragonflight/docker-compose.worker.yml

# Wild Dragon MAM — Worker Node Stack
# ─────────────────────────────────────
# Deploy on any machine you want to join the cluster as a worker.
# The primary stack (mam-api, db, redis) continues running on TrueNAS.
#
# Required env vars (set in .env.worker or export before running):
#   MAM_API_URL   URL of the primary MAM API  e.g. http://10.0.0.25:47432
#   NODE_TOKEN    Bearer token from the primary's Tokens page
#
# Optional hardware overrides (if Docker can't see /dev directly):
#   GPU_COUNT     Number of NVIDIA GPUs on this node  (default: auto-detect from /dev/nvidia*)
#   BMD_COUNT     Number of Blackmagic DeckLink cards  (default: auto-detect from /dev/blackmagic/)
#
# Optional env vars (needed only if starting the worker or capture profiles):
#   REDIS_URL, DATABASE_URL, S3_ENDPOINT, S3_BUCKET, S3_ACCESS_KEY, S3_SECRET_KEY
#   BMD_DEVICE_0  DeckLink device path (default: /dev/blackmagic/dv0)
#   BMD_DEVICE_1  DeckLink device path (default: /dev/blackmagic/dv1)
#
# Profiles:
#   (default)           node-agent only — cluster visibility + hardware heartbeat
#   --profile worker    + CPU job worker (proxy generation, transcoding)
#   --profile capture   + SDI capture service (requires Blackmagic DeckLink card)
#
# To enable GPU transcoding, also apply docker-compose.gpu.yml:
#   docker compose -f docker-compose.worker.yml -f docker-compose.gpu.yml --profile worker up -d
#
# Examples:
#   # Agent only (register node in cluster, report hardware):
#   docker compose -f docker-compose.worker.yml up -d
#
#   # SDI capture node (1 DeckLink, no GPU worker):
#   docker compose -f docker-compose.worker.yml --profile capture up -d
#
#   # Full node (GPU worker + SDI capture):
#   docker compose -f docker-compose.worker.yml -f docker-compose.gpu.yml \
#     --profile worker --profile capture up -d

services:

  node-agent:
    build: ./services/node-agent
    restart: unless-stopped
    environment:
      MAM_API_URL:   ${MAM_API_URL}
      NODE_TOKEN:    ${NODE_TOKEN:-}
      NODE_ROLE:     ${NODE_ROLE:-worker}
      AGENT_PORT:    3002
      HEARTBEAT_MS:  ${HEARTBEAT_MS:-30000}
      GPU_COUNT:     ${GPU_COUNT:--1}
      BMD_COUNT:     ${BMD_COUNT:--1}
    ports:
      - "${AGENT_PORT:-7436}:3002"
    networks:
      - wild-dragon-worker

  worker:
    build: ./services/worker
    profiles: [worker]
    restart: unless-stopped
    environment:
      REDIS_URL:      ${REDIS_URL}
      DATABASE_URL:   ${DATABASE_URL}
      S3_ENDPOINT:    ${S3_ENDPOINT}
      S3_BUCKET:      ${S3_BUCKET}
      S3_ACCESS_KEY:  ${S3_ACCESS_KEY}
      S3_SECRET_KEY:  ${S3_SECRET_KEY}
      S3_REGION:      ${S3_REGION:-us-east-1}
      NVENC_ENABLED:  ${NVENC_ENABLED:-false}
    networks:
      - wild-dragon-worker

  # SDI capture service — only start on nodes with Blackmagic DeckLink cards
  # Set BMD_DEVICE_0 in .env.worker to the actual device path, e.g. /dev/blackmagic/dv0
  capture:
    build: ./services/capture
    profiles: [capture]
    restart: unless-stopped
    environment:
      REDIS_URL:     ${REDIS_URL}
      DATABASE_URL:  ${DATABASE_URL}
      S3_ENDPOINT:   ${S3_ENDPOINT}
      S3_BUCKET:     ${S3_BUCKET}
      S3_ACCESS_KEY: ${S3_ACCESS_KEY}
      S3_SECRET_KEY: ${S3_SECRET_KEY}
      CAPTURE_PORT:  3001
    devices:
      - ${BMD_DEVICE_0:-/dev/blackmagic/dv0}:/dev/blackmagic/dv0
      - ${BMD_DEVICE_1:-/dev/blackmagic/dv1}:/dev/blackmagic/dv1
    ports:
      - "${CAPTURE_PORT:-7437}:3001"
    networks:
      - wild-dragon-worker

networks:
  wild-dragon-worker:
    driver: bridge