#!/bin/sh
# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

set -e
SCRIPT_BASE="$(readlink -f "$(dirname "$0")")"
FACTORY_LOG_FILE=/var/log/factory.log

SUITE_FACTORY="$(dirname "$0")"/../../site_tests/suite_Factory
BOARD_SETUP="$(readlink -f "$SUITE_FACTORY")/board_setup_factory.sh"

# Default args for Goofy.
GOOFY_ARGS=""

# Default implementation for factory_setup (no-op).  May be overriden
# by board_setup_factory.sh.
factory_setup() {
    true
}

load_setup() {
  # Load board-specific parameters, if any.
  if [ -s $BOARD_SETUP ]; then
    echo "Loading board-specific parameters..." 1>&2
    . $BOARD_SETUP
  fi

  factory_setup
}

start_factory() {
  load_setup
  echo "
    Starting factory program...

    If you don't see factory window after more than one minute,
    try to switch to VT2 (Ctrl-Alt-F2), log in, and check the messages by:
      tail $FACTORY_LOG_FILE

    If it keeps failing, try to reset by:
      $SCRIPT_BASE/restart.sh -a
  "

  # Preload modules here
  modprobe i2c-dev 2>/dev/null || true

  # This script should be located in .../autotest/cros/factory.
  cd "$SCRIPT_BASE/../.."
  eval "$("$SCRIPT_BASE/startx.sh" 2>/var/log/startx.err)"
  "$SCRIPT_BASE/goofy" $GOOFY_ARGS >>"$FACTORY_LOG_FILE" 2>&1
}

stop_factory() {
  load_setup
  # Try to kill X, and any other Python scripts, five times.
  echo -n "Stopping factory."
  for i in $(seq 5); do
    pkill 'X|python' || break
    sleep 1
    echo -n "."
  done

  echo "

    Factory tests terminated. To check error messages, try
      tail $FACTORY_LOG_FILE

    To restart, press Ctrl-Alt-F2, log in, and type:
      $SCRIPT_BASE/restart.sh

    If restarting does not work, try to reset by:
      $SCRIPT_BASE/restart.sh -a
    "
}

case "$1" in
  "start" )
    start_factory "$@"
    ;;

  "stop" )
    stop_factory "$@"
    ;;

  * )
    echo "Usage: $0 [start|stop]"
    exit 1
    ;;
esac