1#!/usr/bin/env python3
2import argparse
3import json
4import subprocess
5from pathlib import Path
6 
7 
8def main() -> None:
9    parser = argparse.ArgumentParser()
10    parser.add_argument("--video", required=True)
11    parser.add_argument("--audio", required=True)
12    parser.add_argument("--output", required=True)
13    parser.add_argument("--audio-bitrate", default="192k")
14    args = parser.parse_args()
15 
16    video = Path(args.video).expanduser().resolve()
17    audio = Path(args.audio).expanduser().resolve()
18    output = Path(args.output).expanduser().resolve()
19    output.parent.mkdir(parents=True, exist_ok=True)
20 
21    subprocess.run(
22        [
23            "ffmpeg",
24            "-hide_banner",
25            "-loglevel",
26            "error",
27            "-i",
28            str(video),
29            "-i",
30            str(audio),
31            "-map",
32            "0:v:0",
33            "-map",
34            "1:a:0",
35            "-c:v",
36            "copy",
37            "-c:a",
38            "aac",
39            "-b:a",
40            args.audio_bitrate,
41            "-shortest",
42            str(output),
43            "-y"
44        ],
45        check=True
46    )
47 
48    probe = subprocess.run(
49        [
50            "ffprobe",
51            "-hide_banner",
52            "-show_entries",
53            "format=duration,size",
54            "-of",
55            "json",
56            str(output)
57        ],
58        check=True,
59        capture_output=True,
60        text=True
61    )
62 
63    print(json.dumps({"ok": True, "output": str(output), "probe": json.loads(probe.stdout)}))
64 
65 
66if __name__ == "__main__":
67    main()
68