# Mobile Verification Toolkit (MVT)
# Copyright (c) 2021-2023 The MVT Authors.
# Use of this software is governed by the MVT License 1.1 that can be found at
#   https://license.mvt.re/1.1/

import os
import sqlite3
from pathlib import Path


def get_artifact(fname):
    """
    Return the artifact path in the artifact folder
    """
    fpath = os.path.join(get_artifact_folder(), fname)
    if os.path.isfile(fpath):
        return fpath
    return


def get_artifact_folder():
    return os.path.join(os.path.dirname(__file__), "artifacts")


def get_ios_backup_folder():
    return os.path.join(os.path.dirname(__file__), "artifacts", "ios_backup")


def get_android_backup_folder():
    return os.path.join(os.path.dirname(__file__), "artifacts", "android_backup")


def get_android_androidqf():
    return os.path.join(os.path.dirname(__file__), "artifacts", "androidqf")


def get_indicator_file():
    print("PYTEST env", os.getenv("PYTEST_CURRENT_TEST"))


def add_backup_manifest_entry(backup_path, file_id, domain, relative_path):
    """
    Register an extra file in a test backup's Manifest.db
    """
    conn = sqlite3.connect(os.path.join(backup_path, "Manifest.db"))
    conn.execute(
        "INSERT INTO Files (fileID, domain, relativePath, flags, file) "
        "VALUES (?, ?, ?, 1, ?);",
        (file_id, domain, relative_path, b""),
    )
    conn.commit()
    # MVT opens backup databases with immutable=1, which ignores any -wal file,
    # so the new row has to be checkpointed into Manifest.db itself.
    conn.execute("PRAGMA journal_mode=DELETE;")
    conn.close()


def delete_tmp_db_files(file_path):
    """
    Remove Sqlite temporary files that appear on some platforms

    These can cause filesystem tests to fail depending on the OS.
    """
    for file_name in ["Manifest.db-wal", "Manifest.db-shm"]:
        path = os.path.join(file_path, file_name)
        if os.path.isfile(path):
            os.remove(path)


def list_files(path: str):
    files = []
    parent_path = Path(path).absolute().parent.as_posix()
    target_abs_path = os.path.abspath(path)
    for root, subdirs, subfiles in os.walk(target_abs_path):
        for fname in subfiles:
            file_path = os.path.relpath(os.path.join(root, fname), parent_path)
            files.append(file_path)

    return files