import json, os, shutil, subprocess

FF = r"C:\Users\homelab\ffbuild71\ffmpeg-n7.1-latest-win64-gpl-7.1\bin\ffmpeg.exe"

ROOT = r"C:\Users\homelab\stepping-promo"

REC = ROOT + r"\rec"

CLIPS = ROOT + r"\build\clips2"

os.makedirs(CLIPS, exist_ok=True)

os.makedirs(ROOT + r"\fonts", exist_ok=True)

os.chdir(ROOT)

def cpfont(cands, dst):

    for s in cands:

        if os.path.exists(s):

            shutil.copy(s, dst); return

    raise SystemExit("no font " + dst)

cpfont([r"C:\Windows\Fonts\consolab.ttf", r"C:\Windows\Fonts\consola.ttf"], ROOT + r"\fonts\mono.ttf")

cpfont([r"C:\Windows\Fonts\arialbd.ttf", r"C:\Windows\Fonts\arial.ttf"], ROOT + r"\fonts\sans.ttf")

FPS, W, H = 30, 1920, 1080

MONO, SANS = "fonts/mono.ttf", "fonts/sans.ttf"

GOLD, BONE, STAGE = "0xC8A24C", "0xF2ECE0", "0x0F0E0C"

VC = "-c:v h264_nvenc -pix_fmt yuv420p -preset p5 -rc vbr -cq 20 -b:v 0"

meta = json.load(open(ROOT + r"\voiceover\scene_times.json"))

ss, total, TAIL = meta["scene_start"], meta["total"], 1.6

order = ["s1","s2","s3","s4","s5","s6","s7","s8","s9","s10"]

cfg = {

  "s1": ("desk", "STEPPING  /  DIVINE NINE STEP CULTURE"),

  "s2": ("desk", "CHOOSE YOUR CITY"),

  "s3": ("desk", "THE LINE-UP  /  UPCOMING SHOWS"),

  "s4": ("desk", "EVERY SHOW, IN DETAIL"),

  "s5": ("desk", "THE YARD  /  THE COMMUNITY"),

  "s6": ("desk", "PROFILES  /  WHO'S WHO"),

  "s7": ("desk", "THE ROUTE  /  PLAN THE TRIP"),

  "s8": ("phone", "RIGHT IN YOUR PHONE"),

  "s9": ("desk", "MORE ON THE WAY"),

  "s10": ("end", ""),

}

def dur(i):

    s = order[i]

    return round((ss[order[i+1]] - ss[s]) if i < len(order)-1 else (total - ss[s] + TAIL), 3)

def esc(t): return t.replace(":", "\\:").replace("'", "’").replace(",", "\\,")

def cap(t):

    if not t: return ""

    return (f",drawtext=fontfile={MONO}:text='{esc(t)}':fontcolor={GOLD}:fontsize=30"

            f":x=88:y=h-96:box=1:boxcolor={STAGE}@0.6:boxborderw=22")

def q(p): return '"' + p + '"'

def run(cmd):

    r = subprocess.run(cmd, shell=True, capture_output=True, text=True)

    if r.returncode != 0:

        print("ERR:", r.stderr[-1400:]); raise SystemExit(1)

clips = []

for i, s in enumerate(order):

    kind, c = cfg[s]; d = dur(i); out = f"{CLIPS}\\{s}.mp4"

    fin = "fade=t=in:st=0:d=0.4,fade=t=out:st=%.3f:d=0.4" % max(d-0.4, 0)

    src = q(f"{REC}\\{s}.webm")

    if kind == "desk":

        vf = f"scale={W}:{H}:force_original_aspect_ratio=increase,crop={W}:{H},fps={FPS},{fin}{cap(c)}"

        run(f'{q(FF)} -y -sseof -{d} -t {d} -i {src} -vf "{vf}" -r {FPS} {VC} -an {q(out)}')

    elif kind == "phone":

        fc = (f"color={STAGE}:s={W}x{H}:d={d}:r={FPS}[bg];[0:v]scale=-1:1000,fps={FPS}[ph];"

              f"[bg][ph]overlay=(W-w)/2:(H-h)/2,{fin}{cap(c)}")

        run(f'{q(FF)} -y -sseof -{d} -t {d} -i {src} -filter_complex "{fc}" -r {FPS} {VC} -an {q(out)}')

    else:

        vf = (f"drawtext=fontfile={SANS}:text='STEPPING':fontcolor={BONE}:fontsize=120:x=(w-text_w)/2:y=(h-text_h)/2-40,"

              f"drawtext=fontfile={MONO}:text='BUILT FOR THE CULTURE':fontcolor={GOLD}:fontsize=30:x=(w-text_w)/2:y=(h/2)+70,"

              f"fade=t=in:st=0:d=0.5")

        run(f'{q(FF)} -y -f lavfi -i color={STAGE}:s={W}x{H}:d={d}:r={FPS} -vf "{vf}" {VC} -an {q(out)}')

    clips.append(out); print(f"  {s} {d}s ok")

with open(f"{CLIPS}\\list.txt", "w") as f:

    for c in clips: f.write("file '" + c.replace("\\", "/") + "'\n")

run(f'{q(FF)} -y -f concat -safe 0 -i {q(CLIPS + chr(92) + "list.txt")} -c copy {q(ROOT + chr(92) + "build" + chr(92) + "video2_silent.mp4")}')

vlen = round(total + TAIL, 3); fadest = round(total - 0.3, 3)

af = (f"[0:a]aresample=44100,aformat=channel_layouts=stereo,loudnorm=I=-16:TP=-1.5:LRA=11,apad=whole_dur={vlen},asplit=2[vo][key];"

      f"[1:a]atrim=0:{vlen},asetpts=N/SR/TB,aresample=44100,aformat=channel_layouts=stereo,volume=0.22,afade=t=in:st=0:d=2,afade=t=out:st={fadest}:d=2.0[m];"

      f"[m][key]sidechaincompress=threshold=0.035:ratio=9:attack=15:release=350:makeup=1[md];"

      f"[vo][md]amix=inputs=2:duration=longest:normalize=0,alimiter=limit=0.97[out]")

run(f'{q(FF)} -y -i {q(ROOT + chr(92) + "voiceover" + chr(92) + "stepping_vo.wav")} -i {q(ROOT + chr(92) + "assets" + chr(92) + "music_inspired.mp3")} '

    f'-filter_complex "{af}" -map "[out]" -ar 44100 -ac 2 -c:a aac -b:a 192k {q(ROOT + chr(92) + "build" + chr(92) + "audio2.m4a")}')

run(f'{q(FF)} -y -i {q(ROOT + chr(92) + "build" + chr(92) + "video2_silent.mp4")} -i {q(ROOT + chr(92) + "build" + chr(92) + "audio2.m4a")} '

    f'-map 0:v -map 1:a -c:v copy -c:a copy -shortest {q(ROOT + chr(92) + "build" + chr(92) + "stepping_walkthrough_v2.mp4")}')

print("DONE", vlen)