#!/bin/bash

defines=()
includes=()
externs=()

output_file=""
global_file=""
copy_folder=""
prepropressor=0

defines_str=""
includes_str=""

# parse command arguments
while getopts D:I:J:O:G:C:Ph flag
do
  case "${flag}" in
    D) defines+=( ${OPTARG} )
       defines_str+="-D${OPTARG} "
       ;;
    I) includes+=( ${OPTARG} )
       includes_str+="-I${OPTARG} "
       ;;
    J) externs+=( ${OPTARG} )
       includes_str+="-I${OPTARG} "
       ;;
    O) output_file=( ${OPTARG} );;
    G) global_file=( ${OPTARG} );;
    C) copy_folder=( ${OPTARG} );;
    P) prepropressor=1;;
    h) echo "Usage: [-D<macro>] [-I<include-path>] [-J<external-path>] [-O<output-file>] [-C<dest-folder>: copy to] [-G<global_header>] [-P: macro prepropressing] [-h help]"
       exit 0
    ;;
  \?)
    echo "Invalid option: -$OPTARG" 1>&2
    exit 1
    ;;
  esac
done

if [ "$global_file" != "" ]; then
    {
        # dump defines into a global header
        for value in ${defines[@]}; do
            arrNV=(${value//=/ })
            if (( ${#arrNV[@]} > 1 ));
            then
                echo "\`define ${arrNV[0]} ${arrNV[1]}"
            else
                echo "\`define $value"
            fi        
        done
    } > $global_file
fi

if [ "$copy_folder" != "" ]; then
    # copy source files   
    mkdir -p $copy_folder 
    for dir in ${includes[@]}; do
        find "$dir" -maxdepth 1 -type f | while read -r file; do
            if [ $prepropressor != 0 ]; then
                verilator $defines_str $includes_str -E -P $file > $copy_folder/$(basename -- $file)
            else
                cp $file $copy_folder
            fi
        done
    done
fi

if [ "$output_file" != "" ]; then
    {
        if [ "$global_file" == "" ]; then
            # dump defines
            for value in ${defines[@]}; do
                echo "+define+$value"
            done
        fi

        for dir in ${externs[@]}; do
            echo "+incdir+$(realpath $dir)"
        done

        for dir in ${externs[@]}; do
            find "$(realpath $dir)" -maxdepth 1 -type f \( -name "*.v" -o -name "*.sv" \) -print
        done

        if [ "$copy_folder" != "" ]; then
            # dump include directories
            echo "+incdir+$(realpath $copy_folder)"

            # dump source files
            find "$(realpath $copy_folder)" -maxdepth 1 -type f \( -name "*.v" -o -name "*.sv" \) -print     
        else
            # dump include directories
            for dir in ${includes[@]}; do
                echo "+incdir+$(realpath $dir)"
            done
            
            # dump source files
            for dir in ${includes[@]}; do
                find "$(realpath $dir)" -maxdepth 1 -type f \( -name "*.v" -o -name "*.sv" \) -print
            done
        fi
    } > $output_file
fi