#!/usr/bin/env bash
set -e
cd $(dirname $0)/..

ISORT_ARG=""
BLACK_ARG=""

while getopts 'dt:' OPT; do
    case $OPT in
        d)
            ISORT_ARG="--diff --check-only"
            BLACK_ARG="--diff --check"
            ;;
        t)
            TARGET=$OPTARG
            ;;
        ?)
            echo "Usage: `basename $0` [-d]"
    esac
done

if [[ $TARGET ]]; then
    directories=($TARGET)
else
    directories=(megengine test)
    if [[ -d examples ]]; then
        directories+=(examples)
    fi
fi

# do not isort megengine/__init__.py file, caused we must
# init library load path before load dependent lib in core
isort $ISORT_ARG -j $(nproc) -rc "${directories[@]}" -s megengine/__init__.py
black $BLACK_ARG --target-version=py35 --exclude='_imperative_rt/.*\.pyi$' -- "${directories[@]}"