# Copyright 2022 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os
import sys
import tarfile
import tempfile

sys.path.append(
    os.path.abspath(os.path.join(os.path.dirname(__file__), 'test'))
)
from gs_util_wrapper import run_gsutil


def DownloadAndUnpackFromCloudStorage(url, output_dir):
    """Fetches a tarball from GCS and uncompresses it to |output_dir|."""

    tmp_file = 'image.tgz'
    with tempfile.TemporaryDirectory() as tmp_d:
        tmp_file_location = os.path.join(tmp_d, tmp_file)
        run_gsutil(['cp', url, tmp_file_location])
        tarfile.open(name=tmp_file_location, mode='r:gz').extractall(
            path=output_dir
        )