run_tox.sh

#!/usr/bin/env bash

set -o xtrace

all_charms() {
    find charms -name "charmcraft.yaml" | sed 's|charms/\(.*\)/charmcraft.yaml|\1|'
}

# print checks to test based on the first arg
get_charms_to_test() {
    local charm=$1
    if [[ -z "$charm" ]]; then
        all_charms
    elif [[ "$charm" = "ops-sunbeam" ]]; then
        # ops-sunbeam is treated differently, so don't process it here
        false
    else
        local charms=($(all_charms))
        if [[ ! ${charms[@]} =~ $charm ]];
        then
            echo "Argument should be one of ${charms[@]}" >&2
            exit 1
        fi
        echo "$charm"
    fi
}

should_test_ops_sunbeam() {
    # ops-sunbeam should be tested
    # if no arguments (test everything)
    # or ops-sunbeam is specified.
    if [[ -z "$1" || "$1" = "ops-sunbeam" ]]; then
        true
    else
        false
    fi
}

if [[ $1 == "fmt" ]];
then
    src_path_array=$(ls -d -1 "charms/"**/src)
    tst_path_array=$(ls -d -1 "charms/"**/tests)
    lib_path_array=$(ls -d -1 "charms/"**/lib)

    src_path="${src_path_array[*]}"
    tst_path="${tst_path_array[*]}"
    lib_path="${lib_path_array[*]}"

    ops_sunbeam_src_path="ops-sunbeam/ops_sunbeam"
    ops_sunbeam_tst_path="ops-sunbeam/tests/unit_tests"

    isort ${src_path} ${tst_path} ${ops_sunbeam_src_path} ${ops_sunbeam_tst_path}
    black --config pyproject.toml ${src_path} ${tst_path} ${ops_sunbeam_src_path} ${ops_sunbeam_tst_path}
elif [[ $1 == "pep8" ]];
then
    src_path_array=$(ls -d -1 "charms/"**/src)
    tst_path_array=$(ls -d -1 "charms/"**/tests)

    src_path="${src_path_array[*]}"
    tst_path="${tst_path_array[*]}"

    ops_sunbeam_src_path="ops-sunbeam/ops_sunbeam"
    ops_sunbeam_tst_path="ops-sunbeam/tests/unit_tests"

    codespell ${src_path} ${tst_path} ${ops_sunbeam_src_path} ${ops_sunbeam_tst_path} || exit 1
    pflake8 --config pyproject.toml ${src_path} ${tst_path} ${ops_sunbeam_src_path} ${ops_sunbeam_tst_path} || exit 1
    isort --check-only --diff ${src_path} ${tst_path} ${ops_sunbeam_src_path} ${ops_sunbeam_tst_path} || exit 1
    black --config pyproject.toml --check --diff ${src_path} ${tst_path} ${ops_sunbeam_src_path} ${ops_sunbeam_tst_path} || exit 1
elif [[ $1 == "linters" ]]
then
    ops_sunbeam_src_path="ops-sunbeam/ops_sunbeam"

    PYTHONPATH=$(python3 ./repository.py pythonpath) mypy ${ops_sunbeam_src_path}
elif [[ $1 =~ ^(py3|py310|py312)$ ]];
then

    # Run py3 on ops-sunbeam
    if should_test_ops_sunbeam $2; then
        path_python=$(python3 ./repository.py pythonpath)
        pushd ops-sunbeam
        PYTHONPATH=$path_python python -m pytest tests/unit_tests/ -v --tb=short || exit 1
        popd
    fi

    # Run py3 on sunbeam charms
    for charm in $(get_charms_to_test $2); do
        python3 repository.py -v prepare --clean $charm || exit 1
        pushd charms/$charm
        PYTHONPATH=./src:./lib python -m pytest tests/unit/ -v --tb=short || exit 1
        popd
        python3 repository.py -v clean $charm || exit 1
    done

elif [[ $1 == "cover" ]];
then
    COV_CONFIG="$PWD/pyproject.toml"
    coverage erase

    # Run coverage on ops-sunbeam
    if should_test_ops_sunbeam $2; then
        path_python=$(python3 ./repository.py pythonpath)
        pushd ops-sunbeam
        coverage erase
        PYTHONPATH=$path_python python -m pytest tests/unit_tests/ --cov=ops_sunbeam --cov-branch --cov-config="$COV_CONFIG" --cov-report= --cov-append || exit 1
        popd
    fi

    # Run coverage on sunbeam charms
    for charm in $(get_charms_to_test $2); do
        python3 repository.py -v prepare --clean $charm || exit 1
        pushd charms/$charm
        coverage erase
        PYTHONPATH=./src:./lib:../../ops-sunbeam python -m pytest tests/unit/ --cov=src --cov-branch --cov-config="$COV_CONFIG" --cov-report= --cov-append || exit 1
        popd
    done

    coverage_files=( charms/*/.coverage ops-sunbeam/.coverage )
    existing_coverage_files=()
    for file in "${coverage_files[@]}"; do
        [[ -e $file ]] && existing_coverage_files+=( $file )
    done
    # Prepare coverage report
    coverage combine "${existing_coverage_files[@]}"
    coverage html -d cover
    coverage xml -o cover/coverage.xml
    coverage report

    # Common files should be deleted after coverage combine
    for charm in $(get_charms_to_test $2); do
        python3 repository.py -v clean $charm || exit 1
    done

elif [[ $1 == "build" ]];
then
    if [[ $# != 2 ]];
    then
        echo "Command format: tox -e build <charm>"
        exit 1
    fi

    charm=$2
    charms=($(all_charms))
    if [[ ! ${charms[@]} =~ $charm ]];
    then
        echo "Argument should be one of ${charms[@]}";
        exit 1
    fi

    python3 repository.py -v prepare --clean $charm || exit 1
    pushd charms/$charm || exit 1
    charmcraft -v pack || exit 1
    if [[ -e "${charm}.charm" ]];
    then
        echo "Removing bad downloaded charm maybe?"
        rm "${charm}.charm"
    fi
    echo "Renaming charm ${charm}_*.charm to ${pathed_charm}.charm"
    unpathed_charm=$(basename $charm)
    mv ${unpathed_charm}_*.charm ${unpathed_charm}.charm

    popd || exit 1
    cp charms/$charm/${unpathed_charm}.charm . || exit 1
    python3 repository.py -v clean $charm || exit 1

elif [[ $1 == "uv" ]];
then
    if [[ $# != 2 ]];
    then
        echo "Command format: tox -e uv <charm>"
        exit 1
    fi

    charm=$2
    charms=($(all_charms))
    if [[ ! ${charms[@]} =~ $charm ]];
    then
        echo "Argument should be one of ${charms[@]}";
        exit 1
    fi

    python3 repository.py -v prepare --clean $charm || exit 1
    pushd charms/$charm || exit 1
    JUJU_VERSION=3.6 PYTHONPATH=./lib uv lock --upgrade --no-cache
    popd || exit 1
    python3 repository.py -v clean $charm || exit 1

else
    echo "tox argument should be one of pep8, py3, py310, py312, cover";
    exit 1
fi