#
# Copyright 2020 Ettus Research, a National Instruments Company
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
"""
Utilities for checking the filesystem status
"""

import hashlib
import pathlib
import subprocess
import time


def get_uhd_version(filesystem_root="/"):
    file = pathlib.Path(filesystem_root, "usr/bin/uhd_config_info")
    versionstring = subprocess.check_output([file, "--version"]).decode("utf-8").splitlines()[0]
    return versionstring


def get_mender_artifact(filesystem_root="/", parse_manually=False):
    def parse_artifact(output):
        for line in output.splitlines():
            if line.startswith("artifact_name="):
                return line[14:]
        return None

    if filesystem_root != "/":
        parse_manually = True
    if parse_manually:
        # parse mender artifact manually
        file = pathlib.Path(filesystem_root, "etc/mender/artifact_info")
        if not file.exists():
            return None
        return parse_artifact(file.read_text())
    try:
        output = subprocess.check_output(["/usr/bin/mender", "show-artifact"]).decode("utf-8")
        return output.splitlines()[0]
    except:
        return None


def get_fs_version(filesystem_root="/"):
    file = pathlib.Path(filesystem_root, "etc/version")
    if not file.exists():
        return None
    return file.read_text().splitlines()[0]


def get_opkg_status_date(date_only=False, filesystem_root="/"):
    if date_only:
        tformat = "%Y-%m-%d"
    else:
        tformat = "%Y-%m-%d %H:%M:%S"
    file = pathlib.Path(filesystem_root, "var/lib/opkg/status")
    if not file.exists():
        return None
    return time.strftime(tformat, time.gmtime(file.stat().st_mtime))


def get_opkg_status_md5sum(filesystem_root="/"):
    file = pathlib.Path(filesystem_root, "var/lib/opkg/status")
    if not file.exists():
        return None
    return hashlib.md5(file.read_bytes()).hexdigest()
