#!/bin/bash
global TCID := '"zram.sh'"

source ./zram_lib.sh

proc run_zram {
echo "--------------------"
echo "running zram tests"
echo "--------------------"
./zram01.sh
echo ""
./zram02.sh
}

check_prereqs

# check zram module exists
global MODULE_PATH := "/lib/modules/$[uname -r]/kernel/drivers/block/zram/zram.ko"
if test -f $MODULE_PATH {
	run_zram
} elif test -b /dev/zram0 {
	run_zram
} else {
	echo "$TCID : No zram.ko module or /dev/zram0 device file not found"
	echo "$TCID : CONFIG_ZRAM is not set"
	exit 1
}
(CommandList
  children: [
    (Assignment
      keyword: Assign_None
      pairs: [(assign_pair lhs:(LhsName name:TCID) op:Equal rhs:{(DQ (zram.sh))} spids:[3])]
      spids: [3]
    )
    (C {(.)} {(./zram_lib.sh)})
    (FuncDef
      name: run_zram
      body: 
        (BraceGroup
          children: [
            (C {(echo)} {(DQ (--------------------))})
            (C {(echo)} {(DQ ("running zram tests"))})
            (C {(echo)} {(DQ (--------------------))})
            (C {(./zram01.sh)})
            (C {(echo)} {(DQ )})
            (C {(./zram02.sh)})
          ]
          spids: [19]
        )
      spids: [14 18]
    )
    (C {(check_prereqs)})
    (Assignment
      keyword: Assign_None
      pairs: [
        (assign_pair
          lhs: (LhsName name:MODULE_PATH)
          op: Equal
          rhs: 
            {(/lib/modules/) 
              (CommandSubPart
                command_list: (CommandList children:[(C {(uname)} {(-r)})])
                left_token: <Left_Backtick "`">
                spids: [59 63]
              ) (/kernel/drivers/block/zram/zram.ko)
            }
          spids: [57]
        )
      ]
      spids: [57]
    )
    (If
      arms: [
        (if_arm
          cond: [
            (Sentence
              child: (C {(Lit_Other "[")} {(-f)} {($ VSub_Name "$MODULE_PATH")} {(Lit_Other "]")})
              terminator: <Op_Semi ";">
            )
          ]
          action: [(C {(run_zram)})]
          spids: [-1 77]
        )
        (if_arm
          cond: [
            (Sentence
              child: (C {(Lit_Other "[")} {(-b)} {(/dev/zram0)} {(Lit_Other "]")})
              terminator: <Op_Semi ";">
            )
          ]
          action: [(C {(run_zram)})]
          spids: [82 93]
        )
      ]
      else_action: [
        (C {(echo)} 
          {(DQ ($ VSub_Name "$TCID") (" : No zram.ko module or /dev/zram0 device file not found"))}
        )
        (C {(echo)} {(DQ ($ VSub_Name "$TCID") (" : CONFIG_ZRAM is not set"))})
        (C {(exit)} {(1)})
      ]
      spids: [98 121]
    )
  ]
)