#!/bin/sh

BOARD=${1:-}
DASH=${2:-}
CURL="curl -fsSL"
API="https://api.github.com/repos/cesanta/mongoose"
RAW="https://raw.githubusercontent.com/cesanta/mongoose/HEAD/"

trap "rm -rf gen.mx" EXIT
fail() { echo $@ >&2 ; exit 1 ; }

# Find CubeMX CLI executable
MX=`find /Applications/STMicroelectronics/ -name STM32CubeMX -type f`

test -z "$MX" && fail "CubeMX is not installed. Install it first"
# test -d "$BOARD"  && fail "Directory $BOARD already exists, remove it first"

# Board name is not given. Show list of boards and exit
if test -z "$BOARD" ; then
  echo "Usage: $0 BOARD [full|minimal]"
  echo "Supported boards:"
  $CURL $API/contents/resources/downloads | perl -nle 'print "  $1" if /"name": "(.+?).ioc"/'
  exit 1
fi

# Create project directory
mkdir -p $BOARD
test -f "$BOARD/$BOARD.ioc" || $CURL $RAW/resources/downloads/$BOARD.ioc -o $BOARD/$BOARD.ioc 2>/dev/null
test -f "$BOARD/$BOARD.ioc" || fail echo "Usupported board $BOARD"

# Run CubeMX to generate project using a pre-configured .ioc file
echo config load `pwd`/$BOARD/$BOARD.ioc > gen.mx
echo project generate >> gen.mx
echo exit >> gen.mx
test -f "$BOARD/CMakeLists.txt" || $MX -q `pwd`/gen.mx

# Adopt generated code for Mongoose
MAIN_C=$BOARD/Core/Src/main.c
grep -q mongoose.h $MAIN_C || perl -i -ne 'print; print "#include \"mongoose.h\"\n" if /BEGIN Includes/' $MAIN_C
grep -q '^int _write' $MAIN_C || perl -i -ne 'print; print "int _write(int fd, unsigned char *buf, int len) {\n  HAL_UART_Transmit(&huart3, buf, len, HAL_MAX_DELAY);\n  return len;\n}\n" if /USER CODE BEGIN 0/' $MAIN_C
grep -q 'mg_mgr_init' $MAIN_C || perl -i -ne 'print; print "  struct mg_mgr mgr;\n  mg_mgr_init(&mgr);\n\n" if /USER CODE BEGIN WHILE/' $MAIN_C
grep -q 'mg_mgr_poll' $MAIN_C || perl -i -ne 'print "    mg_mgr_poll(&mgr, 0);\n" if /USER CODE END WHILE/; print;' $MAIN_C
test -f $BOARD/STM32H723XG_FLASH.ld && (grep -q eth_ram $BOARD/STM32H723XG_FLASH.ld || perl -i -ne 'print ; print "\n  /* Mongoose Ethernet driver */\n  .eth_ram : { *(.eth_ram .eth_ram*) } > RAM_D2 AT > FLASH\n" if /_sidata =/' $BOARD/STM32H723XG_FLASH.ld)
test -f $BOARD/LICENSE || $CURL $RAW/LICENSE -o $BOARD/LICENSE

# If dashboard is specified, copy it to the project
if test "$DASH" == "full" -o "$DASH" == "minimal" ; then
  test -d ${BOARD}_${DASH} || cp -r $BOARD ${BOARD}_${DASH}
  test -d ${BOARD}_${DASH}/Mongoose || mkdir -p ${BOARD}_${DASH}/Mongoose
  test -f ${BOARD}_${DASH}/Mongoose/dashboard.c || $CURL $RAW/tutorials/device-dashboard/$DASH/dashboard.c -o ${BOARD}_${DASH}/Mongoose/dashboard.c
  test -f ${BOARD}_${DASH}/Mongoose/dashboard.html || $CURL $RAW/tutorials/device-dashboard/$DASH/dashboard.html -o ${BOARD}_${DASH}/Mongoose/dashboard.html
  test -f ${BOARD}_${DASH}/Mongoose/html2c.js || $CURL $RAW/resources/html2c.js -o ${BOARD}_${DASH}/Mongoose/html2c.js
  MAIN2_C=${BOARD}_${DASH}/Core/Src/main.c
  grep -q 'mg_dash_init' $MAIN2_C || perl -i -ne 'print; print "  mg_dash_init(&mgr);\n" if /mg_mgr_init/' $MAIN2_C
  grep -q 'mg_dash_poll' $MAIN2_C || perl -i -ne 'print; print "    mg_dash_poll(&mgr);\n" if /mg_mgr_poll/' $MAIN2_C
  grep -q '^    Mongoose/file_data.c' ${BOARD}_${DASH}/CMakeLists.txt || perl -i -ne 'print; print "    Mongoose/dashboard.c\n    Mongoose/file_data.c\n" if /# Add user sources here/' ${BOARD}_${DASH}/CMakeLists.txt
  grep -q 'html2c.js' ${BOARD}_${DASH}/CMakeLists.txt || perl -i -ne 'print "# Generate file_data.c from dashboard.html\nadd_custom_command(\n    OUTPUT \${CMAKE_SOURCE_DIR}/Mongoose/file_data.c\n    COMMAND node \${CMAKE_SOURCE_DIR}/Mongoose/html2c.js \${CMAKE_SOURCE_DIR}/Mongoose/dashboard.html -o \${CMAKE_SOURCE_DIR}/Mongoose/file_data.c\n    DEPENDS \${CMAKE_SOURCE_DIR}/Mongoose/dashboard.html\n    VERBATIM\n)\n\n" if /^# Add sources to executable/; print;' ${BOARD}_${DASH}/CMakeLists.txt
fi