#!/usr/bin/env bash
#
# Run real shell code with osh and bash, and compare the results.
#
# Limitation: If a script shells out to another bash script, osh won't be run.
# TODO: --hijack-shebang or just 'sed' all the scripts in a repo?
#
# Usage:
#   ./gold-test.sh <function name>

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

proc _compare {
  set +o errexit

  @Argv >_tmp/left.txt
  var left_status = $Status

  bin/osh @Argv >_tmp/right.txt
  var right_status = $Status

  set -o errexit

  if ! diff -u _tmp/left.txt _tmp/right.txt {
    echo FAIL
    return 1
  }

  if test $left_status != $right_status {
    echo "FAIL: Got status $right_status but expected $left_status"
    return 1
  }

  echo PASS
  return 0
}

# Uses
# - { busybox || true; } | head
# - $1
proc version-text {
  _compare test/spec.sh version-text
}

# Uses {core,osh}/*.py
proc count {
  _compare scripts/count.sh all
  _compare scripts/count.sh parser
  _compare scripts/count.sh parser-port
  _compare scripts/count.sh runtime
}

# Uses $(cd $(dirname $0) && pwd)
proc one-spec-test {
  _compare test/spec.sh builtins-special
}

# Uses redirect of functions.
proc html-summary {
  _compare test/spec-runner.sh html-summary
}

proc configure {
  _compare ./configure
}

proc no-op {
  _compare scripts/count.sh
}

proc gen-module-init {
  var modules = ''time datetime''
  _compare build/actions.sh gen-module-init $modules
}

proc wild {
  _compare test/wild.sh parse-usr-bin
}

# NOTE: zsh behaves differently under sh and bin/osh!  Looks like it is an
# inherited file descriptor issue.
#
# A bin/osh app bundle also behaves differently.  Maybe because of the internal
# environment variables.
proc startup-benchmark {
  _compare benchmarks/startup.sh compare-strace
}

proc glob {
  _compare gold/glob.sh
}

proc nix {
  _compare gold/nix.sh isElfSimpleWithStdin
}

proc all {
  version-text
  count
  one-spec-test
  html-summary
  configure
  no-op
  gen-module-init
  wild
  startup-benchmark
  glob
}

@Argv
(CommandList
  children: [
    (C {(set)} {(-o)} {(nounset)})
    (C {(set)} {(-o)} {(pipefail)})
    (C {(set)} {(-o)} {(errexit)})
    (FuncDef
      name: _compare
      body: 
        (BraceGroup
          children: [
            (C {(set)} {(Lit_Other "+") (o)} {(errexit)})
            (SimpleCommand
              words: [{(DQ ($ VSub_At "$@"))}]
              redirects: [(Redir op_id:Redir_Great fd:-1 arg_word:{(_tmp/left.txt)} spids:[67])]
            )
            (Assignment
              keyword: Assign_Local
              pairs: [
                (assign_pair
                  lhs: (LhsName name:left_status)
                  op: Equal
                  rhs: {($ VSub_QMark "$?")}
                  spids: [73]
                )
              ]
              spids: [71]
            )
            (SimpleCommand
              words: [{(bin/osh)} {(DQ ($ VSub_At "$@"))}]
              redirects: [(Redir op_id:Redir_Great fd:-1 arg_word:{(_tmp/right.txt)} spids:[84])]
            )
            (Assignment
              keyword: Assign_Local
              pairs: [
                (assign_pair
                  lhs: (LhsName name:right_status)
                  op: Equal
                  rhs: {($ VSub_QMark "$?")}
                  spids: [90]
                )
              ]
              spids: [88]
            )
            (C {(set)} {(-o)} {(errexit)})
            (If
              arms: [
                (if_arm
                  cond: [
                    (Sentence
                      child: 
                        (Pipeline
                          children: [(C {(diff)} {(-u)} {(_tmp/left.txt)} {(_tmp/right.txt)})]
                          negated: True
                        )
                      terminator: <Op_Semi ";">
                    )
                  ]
                  action: [
                    (C {(echo)} {(FAIL)})
                    (ControlFlow token:<ControlFlow_Return return> arg_word:{(1)})
                  ]
                  spids: [-1 116]
                )
              ]
              spids: [-1 129]
            )
            (If
              arms: [
                (if_arm
                  cond: [
                    (Sentence
                      child: 
                        (C {(test)} {($ VSub_Name "$left_status")} {(KW_Bang "!") (Lit_Other "=")} 
                          {($ VSub_Name "$right_status")}
                        )
                      terminator: <Op_Semi ";">
                    )
                  ]
                  action: [
                    (C {(echo)} 
                      {
                        (DQ ("FAIL: Got status ") ($ VSub_Name "$right_status") (" but expected ") 
                          ($ VSub_Name "$left_status")
                        )
                      }
                    )
                    (ControlFlow token:<ControlFlow_Return return> arg_word:{(1)})
                  ]
                  spids: [-1 145]
                )
              ]
              spids: [-1 163]
            )
            (C {(echo)} {(PASS)})
            (ControlFlow token:<ControlFlow_Return return> arg_word:{(0)})
          ]
          spids: [51]
        )
      spids: [47 50]
    )
    (FuncDef
      name: version-text
      body: (BraceGroup children:[(C {(_compare)} {(test/spec.sh)} {(version-text)})] spids:[192])
      spids: [188 191]
    )
    (FuncDef
      name: count
      body: 
        (BraceGroup
          children: [
            (C {(_compare)} {(scripts/count.sh)} {(all)})
            (C {(_compare)} {(scripts/count.sh)} {(parser)})
            (C {(_compare)} {(scripts/count.sh)} {(parser-port)})
            (C {(_compare)} {(scripts/count.sh)} {(runtime)})
          ]
          spids: [211]
        )
      spids: [207 210]
    )
    (FuncDef
      name: one-spec-test
      body: 
        (BraceGroup
          children: [(C {(_compare)} {(test/spec.sh)} {(builtins-special)})]
          spids: [251]
        )
      spids: [247 250]
    )
    (FuncDef
      name: html-summary
      body: 
        (BraceGroup
          children: [(C {(_compare)} {(test/spec-runner.sh)} {(html-summary)})]
          spids: [270]
        )
      spids: [266 269]
    )
    (FuncDef
      name: configure
      body: (BraceGroup children:[(C {(_compare)} {(./configure)})] spids:[286])
      spids: [282 285]
    )
    (FuncDef
      name: no-op
      body: (BraceGroup children:[(C {(_compare)} {(scripts/count.sh)})] spids:[300])
      spids: [296 299]
    )
    (FuncDef
      name: gen-module-init
      body: 
        (BraceGroup
          children: [
            (Assignment
              keyword: Assign_Local
              pairs: [
                (assign_pair
                  lhs: (LhsName name:modules)
                  op: Equal
                  rhs: {(SQ <"time datetime">)}
                  spids: [319]
                )
              ]
              spids: [317]
            )
            (C {(_compare)} {(build/actions.sh)} {(gen-module-init)} {($ VSub_Name "$modules")})
          ]
          spids: [314]
        )
      spids: [310 313]
    )
    (FuncDef
      name: wild
      body: (BraceGroup children:[(C {(_compare)} {(test/wild.sh)} {(parse-usr-bin)})] spids:[340])
      spids: [336 339]
    )
    (FuncDef
      name: startup-benchmark
      body: 
        (BraceGroup
          children: [(C {(_compare)} {(benchmarks/startup.sh)} {(compare-strace)})]
          spids: [371]
        )
      spids: [367 370]
    )
    (FuncDef
      name: glob
      body: (BraceGroup children:[(C {(_compare)} {(gold/glob.sh)})] spids:[387])
      spids: [383 386]
    )
    (FuncDef
      name: nix
      body: 
        (BraceGroup
          children: [(C {(_compare)} {(gold/nix.sh)} {(isElfSimpleWithStdin)})]
          spids: [401]
        )
      spids: [397 400]
    )
    (FuncDef
      name: all
      body: 
        (BraceGroup
          children: [
            (C {(version-text)})
            (C {(count)})
            (C {(one-spec-test)})
            (C {(html-summary)})
            (C {(configure)})
            (C {(no-op)})
            (C {(gen-module-init)})
            (C {(wild)})
            (C {(startup-benchmark)})
            (C {(glob)})
          ]
          spids: [417]
        )
      spids: [413 416]
    )
    (C {(DQ ($ VSub_At "$@"))})
  ]
)