#!/usr/bin/env python3

import glob
import os
import os.path
import shutil
import subprocess
import sys

PLUGIN_DIRS = ['audio', 'generic', 'net', 'text', 'utils', 'video']

command, meson_build_dir, meson_current_source_dir, meson_build_root, target, exclude, extra_env = sys.argv[
    1:8]

cargo_target_dir = os.path.join(meson_build_dir, 'target')

env = os.environ.copy()
env['CARGO_TARGET_DIR'] = cargo_target_dir

pkg_config_path = env.get('PKG_CONFIG_PATH', '').split(':')
pkg_config_path.append(os.path.join(meson_build_root, 'meson-uninstalled'))
env['PKG_CONFIG_PATH'] = ':'.join(pkg_config_path)

if len(extra_env) > 0:
    for e in extra_env.split(','):
        k, v = e.split(':')
        env[k] = v

if command == 'build':
    # cargo build
    ext = sys.argv[8]
    # when using --default-library=both 2 extensions are passed
    try:
        ext2 = sys.argv[9]
    except IndexError:
        ext2 = None

    # Build with the 'static' feature enforcing the minimal gst version required for static builds
    cargo_cmd = ['cargo', 'cbuild', '--features', 'static']
    if target == 'release':
        cargo_cmd.append('--release')
elif command == 'test':
    # cargo test
    cargo_cmd = ['cargo', 'ctest', '--no-fail-fast', '--color=always']
else:
    print("Unknown command:", command)
    sys.exit(1)


def run(cargo_cmd, env):
    try:
        subprocess.run(cargo_cmd, env=env, check=True)
    except subprocess.SubprocessError:
        sys.exit(1)


for d in PLUGIN_DIRS:
    for name in os.listdir(os.path.join(meson_current_source_dir, d)):
        if '{}/{}'.format(d, name) in exclude:
            continue

        cargo_toml = os.path.join(
            meson_current_source_dir, d, name, 'Cargo.toml')
        cmd = cargo_cmd + ['--manifest-path', cargo_toml]
        run(cmd, env)

if command == 'build':
    # Copy so files to build dir
    for f in glob.glob(os.path.join(cargo_target_dir, target, '*.' + ext)):
        shutil.copy(f, meson_build_dir)
    if ext2:
        for f in glob.glob(os.path.join(cargo_target_dir, target, '*.' + ext2)):
            shutil.copy(f, meson_build_dir)