#!/bin/bash

#

# Compile .md from .ipynb files for mkdocs

# Names of ipynb files to convert (from /notebooks) are specified in ipynb_pages.txt

set -e

function sed_replace {

  # https://stackoverflow.com/a/10467453/7529152

  # sed_replace <from> <to> <file>"

  sed -i "s/$(echo $1 | sed -e 's/\([[\/.*]\|\]\)/\\&/g')/$(echo $2 | sed -e 's/[\/&]/\\&/g')/g" $3

}

DIR=tutorials

for file in $(cat ipynb_pages.txt); do

    echo "converting: ${file}.ipynb"

    # .ipynb -> .md

    jupyter nbconvert --to markdown ../notebooks/${file}.ipynb --output-dir sources/${DIR}/

    file_out=sources/${DIR}/${file}.md

    dir_out=sources/${DIR}/${file}_files

    # fix the paths for the original images

    sed_replace '![img](../docs/theme_dir/img/' '![img](/bpnet/img/' $file_out

    # prepend the original ipython notebook link

    echo -e "Generated from [notebooks/${file}.ipynb](https://github.com/kundajelab/bpnet/blob/master/notebooks/${file}.ipynb)\n$(cat ${file_out})" > ${file_out}

    if [ -d "${dir_out}" ]; then

    # move the additional files <file>_files to theme_dir/img/ipynb/ (inline plots in ipynb)

    if [ -d "theme_dir/img/ipynb/${file}_files" ]; then

        rm -r theme_dir/img/ipynb/${file}_files

    fi

    mv -f ${dir_out} theme_dir/img/ipynb/

    # fix the path in the .md file

    sed_replace '![png]('${file}'_files' '![png](/bpnet/img/ipynb/'${file}'_files' $file_out

    sed_replace '![svg]('${file}'_files' '![svg](/bpnet/img/ipynb/'${file}'_files' $file_out

    fi

done

if [ -d "theme_dir/bpnet" ]; then

    rm -r theme_dir/bpnet

fi

mkdir -p theme_dir/bpnet

ln -sr theme_dir/img theme_dir/bpnet/img