#!/bin/bash

set -e
cd $(dirname $0)

case "$1" in
    download_all)
        ./d download_mcufinder
        ./d download_svd
        ./d download_headers
        ./d download_cubedb
    ;;
    download_mcufinder)
        mkdir -p sources/mcufinder
        wget http://stmcufinder.com/API/getFiles.php -O sources/mcufinder/files.json
        wget http://stmcufinder.com/API/getMCUsForMCUFinderPC.php -O sources/mcufinder/mcus.json	
    ;;
    download_pdf)
	    jq -r .Files[].URL < sources/mcufinder/files.json  | wget -P sources/pdf/ -N -i -
    ;;
    download_svd)
    	rm -rf ./sources/git/stm32-rs
        git clone --depth 1 https://github.com/stm32-rs/stm32-rs.git ./sources/git/stm32-rs
        (cd ./sources/git/stm32-rs; make svdformat)
        mkdir -p sources/svd
        for f in ./sources/git/stm32-rs/svd/*.formatted; do
            base=$(basename $f | cut -f 1 -d .)
            cp $f sources/svd/$base.svd
        done
    ;;
    download_headers)
        for f in F0 F1 F2 F3 F4 F7 H7 L0 L1 L4 L5 G0 G4 WB WL; do
            rm -rf ./sources/git/STM32Cube$f
            git clone --depth 1 https://github.com/STMicroelectronics/STM32Cube$f sources/git/STM32Cube$f
        done
        rm -rf sources/headers
        mkdir -p sources/headers
        cp sources/git/STM32Cube*/Drivers/CMSIS/Device/ST/STM32*/Include/*.h sources/headers
        rm sources/headers/stm32??xx.h
        rm sources/headers/system_*.h
        rm sources/headers/partition_*.h
    ;;
    download_cubedb)
        rm -rf sources/cubedb
        git clone --depth 1 https://github.com/embassy-rs/stm32cube-database.git sources/cubedb
    ;;
    extract_all)
        peri=$2
        mkdir -p tmp/$peri

        cargo build --release --manifest-path ../svd2rust/Cargo.toml

        for f in `ls sources/svd`; do
            f=${f#"stm32"}
            f=${f%".svd"}
            echo -n processing $f ...
            RUST_LOG=info ../svd2rust/target/release/svd4rust extract-peripheral --svd sources/svd/stm32$f.svd --transform transform.yaml --peripheral $peri > tmp/$peri/$f.yaml 2> tmp/$peri/$f.yaml.out
            if [ $? -ne 0 ]; then 
                rm tmp/$peri/$f.yaml
                echo FAIL
            else
                rm tmp/$peri/$f.yaml.out
                echo OK
            fi
        done
    ;;
    *)
        echo "unknown command"
    ;;
esac