#!/bin/bash
#
# Test scripts found in the wild for both correctness and performance.
#
# Usage:
#   ./runtime.sh <function name>

set -o nounset
set -o pipefail
set -o errexit

source test/common.sh
source benchmarks/common.sh  # csv-concat

readonly BASE_DIR=_tmp/osh-runtime
readonly TAR_DIR=$PWD/_deps/osh-runtime  # Make it absolute

#
# Dependencies
#

readonly -a TAR_SUBDIRS=( ocaml-4.06.0 tcc-0.9.26 yash-2.46 )

# NOTE: Same list in oilshell.org/blob/run.sh.
proc tarballs {
  cat << """
tcc-0.9.26.tar.bz2
yash-2.46.tar.xz
ocaml-4.06.0.tar.xz
"""
}

proc download {
  mkdir -p $TAR_DIR
  tarballs | xargs -n 1 -I {} --verbose -- \
    wget --directory $TAR_DIR 'https://www.oilshell.org/blob/testdata/{}'
}

proc extract {
  time for f in []$TAR_DIR/*.{bz2,xz} {
    tar -x --directory $TAR_DIR --file $f 
  }
  ls -l $TAR_DIR
}

#
# Computation
#

readonly PY27_DIR=$PWD/Python-2.7.13

proc cpython-configure {
  local sh_path=$(1:-bash)
  local out_dir=$(2:-$BASE_DIR/${sh_path}-cpython-configure)
  mkdir -p $out_dir

  pushd $out_dir
  time $sh_path $PY27_DIR/configure || true
  popd

  tree $out_dir
}

# 18.9 seconds vs 11 seconds above.
proc osh-cpython-configure {
  cpython-configure $OSH_OVM $BASE_DIR/osh-cpython-configure
}

proc runtime-task {
  local raw_dir=$1  # output
  local job_id=$2
  local host=$3
  local host_hash=$4
  local sh_path=$5
  local shell_hash=$6
  local task_type=$7
  local task_arg=$8

  local shell_name=$[basename $sh_path]

  # NOTE: For abuild, this isn't a directory name.
  local x=$[basename $task_arg]
  local task_label="$(shell_name)-$(shell_hash)__$(x)"

  local times_out="$PWD/$raw_dir/$host.$job_id.times.csv"
  local vm_out_dir="$PWD/$raw_dir/$host.$job_id.virtual-memory"
  local files_out_dir="$PWD/$raw_dir/$host.$job_id.files/$task_label"
  mkdir -p $vm_out_dir $files_out_dir

  local -a TIME_PREFIX=(
    $PWD/benchmarks/time.py \
    --output $times_out \
    --field "$host" --field "$host_hash" \
    --field "$shell_name" --field "$shell_hash" \
    --field "$task_type" --field "$task_arg"
  )

  # Can't use array because of set -u bug!!!  Only fixed in bash 4.4.
  setglobal extra_args = ''''
  if test $shell_name = 'osh' {
    local pdump="$(vm_out_dir)/$(task_label)__parser.txt"
    local rdump="$(vm_out_dir)/$(task_label)__runtime.txt"
    setglobal extra_args = ""--parser-mem-dump $pdump --runtime-mem-dump $rdump""

    # Should we add a field here to say it has VM stats?
  }

  echo
  echo "--- $sh_path $task_type $task_arg ---"
  echo

  match $task_type {
    with abuild
      # NOTE: $task_arg unused.


      $(TIME_PREFIX[@]) -- \
        $sh_path $extra_args testdata/osh-runtime/abuild -h \
        > $files_out_dir/STDOUT.txt
      

    with cpython
      # NOTE: $task_arg unused.

      # This configure script defeats --runtime-mem-dump, I think because of:
      # exec $CONFIG_SHELL $as_opts "$as_myself" ${1+"$@"}

      # autoconf supports running configure from a different directory.
      pushd $files_out_dir >/dev/null

      $(TIME_PREFIX[@]) -- \
        $sh_path $extra_args $PY27_DIR/configure \
        > $files_out_dir/STDOUT.txt

      popd >/dev/null
      

    with configure
      local conf_dir=$task_arg

      pushd $conf_dir >/dev/null
      touch __TIMESTAMP

      $(TIME_PREFIX[@]) -- $sh_path $extra_args ./configure \
        > $files_out_dir/STDOUT.txt

      find . -type f -newer __TIMESTAMP \
        | xargs -I {} -- cp --verbose {} $files_out_dir
      popd >/dev/null
      

    with *
      die "Invalid task type $task_type"
      
  }
}

# TODO:
# - Add Python's configure -- same or different?
# - Unify abuild -h -- time it
# - --runtime-dump-mem and rename to --parser-dump-mem
#
# benchmark_name,shell,out_dir
# abuild-h
# X-configure
# config.status?
#
# Yeah need to come up with a name.  Not just conf-dirs.
# $dir-configure

# Do I add host/host_id?  Or just host_label and rely on provenance?

# Turn this into write-tasks?
# And then run-tasks?  run-all?

# Yeah it should be 
# osh-parser.sh write-tasks
# osh-runtime.sh write-tasks
# virtual-memory.sh write-tasks
#
# And then auto.sh run-tasks?  Then you can have consistent logging?

# For each configure file.
proc print-tasks {
  local provenance=$1

  # Add 1 field for each of 5 fields.
  cat $provenance | while read \
    job_id host_name host_hash sh_path shell_hash {

    match $sh_path {
      with mksh|zsh|bin/osh
        log "--- Skipping $sh_path"
        continue
        
    }

    # Need $PWD/$sh_path because we must change dirs to configure.
    match $sh_path {
      with */osh
        setglobal sh_path = "$PWD/$sh_path"
        
    }
    local prefix="$job_id $host_name $host_hash $sh_path $shell_hash"

    # NOTE: 'abuild-help' is a dummy label.
    echo $prefix abuild abuild-help
    echo $prefix cpython cpython-configure

    for dir in [$(TAR_SUBDIRS[@])] {
      echo $prefix configure $TAR_DIR/$dir
    }
  }
}

readonly HEADER='status,elapsed_secs,host_name,host_hash,shell_name,shell_hash,task_type,task_arg'
readonly NUM_COLUMNS=7  # 5 from provenence, then task_type / task_arg

proc measure {
  local provenance=$1
  local raw_dir=$(2:-$BASE_DIR/raw)
  local pattern=$(3:-)

  #local base_dir=${2:-../benchmark-data/osh-parser}

  # Job ID is everything up to the first dot in the filename.
  local name=$[basename $provenance]
  local prefix=$(name%.provenance.txt)  # strip suffix

  local times_out="$raw_dir/$prefix.times.csv"
  mkdir -p $BASE_DIR/{raw,stage1} $raw_dir

  # Write Header of the CSV file that is appended to.
  echo $HEADER > $times_out

  local tasks=$BASE_DIR/tasks.txt
  print-tasks $provenance > $tasks

  # An empty pattern matches every line.
  time egrep $pattern $tasks |
    xargs -n $NUM_COLUMNS -- $0 runtime-task $raw_dir ||
    die "*** Some tasks failed. ***"

  cp -v $provenance $raw_dir
}

proc stage1 {
  local raw_dir=$(1:-$BASE_DIR/raw)
  local out_dir=$BASE_DIR/stage1

  mkdir -p $out_dir

  local times_csv=$out_dir/times.csv
  # Globs are in lexicographical order, which works for our dates.
  local -a a=($raw_dir/flanders.*.times.csv)
  local -a b=($raw_dir/lisa.*.times.csv)
  csv-concat $(a[-1]) $(b[-1]) > $times_csv

  local vm_csv=$out_dir/virtual-memory.csv

  local -a c=($raw_dir/flanders.*.virtual-memory)
  local -a d=($raw_dir/lisa.*.virtual-memory)
  benchmarks/virtual_memory.py osh-runtime $(c[-1]) $(d[-1]) > $vm_csv

  #local raw_dir=${1:-../benchmark-data/osh-parser}
}

proc print-report {
  local in_dir=$1
  local base_url='../../web'

  cat << """
<!DOCTYPE html>
<html>
  <head>
    <title>OSH Runtime Performance</title>
    <script type="text/javascript" src="$base_url/table/table-sort.js"></script>
    <link rel="stylesheet" type="text/css" href="$base_url/table/table-sort.css" />
    <link rel="stylesheet" type="text/css" href="$base_url/benchmarks.css" />

  </head>
  <body>
    <p id="home-link">
      <a href="/">oilshell.org</a>
    </p>
    <h2>OSH Runtime Performance</h2>

    <h3>Elapsed Time by Shell (milliseconds)</h3>

    <p>Some benchmarks call many external tools, while some exercise the shell
    interpreter itself.  Parse time is included.</p>
"""
  csv2html $in_dir/times.csv

  cat << """
    <h3>Memory Used to Run</h3>

    <p>Running under <code>osh-ovm</code>.  Memory usage is measured in MB
    (powers of 10), not MiB (powers of 2).</p>

"""
  csv2html $in_dir/virtual-memory.csv

  cat << """

    <h3>Shell and Host Details</h3>
"""
  csv2html $in_dir/shells.csv
  csv2html $in_dir/hosts.csv

  cat << """
  </body>
</html>
"""
}

#
# Non-configure scripts
#

proc abuild-h {
  local out_dir=_tmp/osh-runtime
  mkdir -p $out_dir

  # TODO: Should test the correctness too.
  local out=$out_dir/abuild-h-times.csv

  echo 'status,elapsed_secs,sh_path' > $out
  for sh_path in [bash dash mksh zsh $OSH_OVM] {
    benchmarks/time.py --output $out --field $sh_path -- \
      $sh_path benchmarks/testdata/abuild -h
  }
}

#
# Misc
#

# Same problem as tcc
proc qemu-old {
  local out_dir=$PWD/_tmp/qemu-old
  mkdir -p $out_dir
  configure-and-copy ~/src/qemu-1.6.0 $OSH_OVM $out_dir
}

# This doesn't work for ash either, because it uses the busybox pattern.  It
# says "exe: applet not found".  I guess yash doesn't configure under ash!
proc self-exe {
  set +o errexit
  dash << """
/proc/self/exe -V
"""
  echo

  _bin/osh << """
/proc/self/exe -V
"""

  _tmp/shells/ash << """
/proc/self/exe -V
"""
}

@Argv