#!/usr/bin/python
# Copyright (c) 2023 Nordic Semiconductor ASA
#
# SPDX-License-Identifier: Apache-2.0

"""
Simply mock for bash script to use with unit tests.
"""
import sys
import time
from argparse import ArgumentParser

from zen_of_python import zen_of_python


def main() -> int:
    parser = ArgumentParser()
    parser.add_argument('--sleep', action='store', default=0, type=float)
    parser.add_argument('--long-sleep', action='store_true')
    parser.add_argument('--return-code', action='store', default=0, type=int)
    parser.add_argument('--exception', action='store_true')

    args = parser.parse_args()

    if args.exception:
        # simulate crashing application
        raise Exception

    if args.long_sleep:
        # prints data and wait for certain time
        for line in zen_of_python:
            print(line, flush=True)
        time.sleep(args.sleep)
    else:
        # prints lines with delay
        for line in zen_of_python:
            print(line, flush=True)
            time.sleep(args.sleep)

    print('End of script', flush=True)
    print('Returns with code', args.return_code, flush=True)
    time.sleep(1)  # give a moment for external programs to collect all outputs
    return args.return_code


if __name__ == '__main__':
    sys.exit(main())