#!/usr/bin/env python3
import argparse
import glob
import os
import pwd
import subprocess
import sys
from typing import Any, Dict, List

TOOLS_DIR = os.path.dirname(os.path.abspath(__file__))
sys.path.insert(0, os.path.dirname(TOOLS_DIR))
ROOT_DIR = os.path.dirname(TOOLS_DIR)

# check for the venv
from tools.lib import sanity_check

sanity_check.check_venv(__file__)

# Import this after we do the sanity_check so it doesn't crash.
import orjson
from zulint.printer import BOLDRED, CYAN, ENDC, GREEN

INDEX_JS = 'frontend_tests/zjsunit/index.js'
NODE_COVERAGE_PATH = 'var/node-coverage/coverage-final.json'

USAGE = '''
    tools/test-js-with-node                      - to run all tests
    tools/test-js-with-node util.js activity.js  - to run just a couple tests
    tools/test-js-with-node --coverage           - to generate coverage report
    '''

# We do not yet require 100% line coverage for these files:
EXEMPT_FILES = {
    'static/js/admin.js',
    'static/js/archive.js',
    'static/js/attachments_ui.js',
    'static/js/avatar.js',
    'static/js/billing/helpers.js',
    'static/js/blueslip.js',
    'static/js/blueslip_stacktrace.ts',
    'static/js/channel.js',
    'static/js/click_handlers.js',
    'static/js/compose_actions.js',
    'static/js/composebox_typeahead.js',
    'static/js/compose_fade.js',
    'static/js/compose.js',
    'static/js/condense.js',
    'static/js/confirm_dialog.js',
    'static/js/copy_and_paste.js',
    'static/js/csrf.js',
    'static/js/debug.js',
    'static/js/desktop_integration.js',
    'static/js/drafts.js',
    'static/js/echo.js',
    'static/js/emoji_picker.js',
    'static/js/emojisets.js',
    'static/js/favicon.js',
    'static/js/feedback_widget.js',
    'static/js/floating_recipient_bar.js',
    'static/js/gear_menu.js',
    'static/js/global.d.ts',
    'static/js/hashchange.js',
    'static/js/hbs.d.ts',
    'static/js/hotkey.js',
    'static/js/hotspots.js',
    'static/js/info_overlay.js',
    'static/js/invite.js',
    'static/js/lightbox_canvas.js',
    'static/js/lightbox.js',
    'static/js/list_util.js',
    'static/js/loading.js',
    'static/js/local_message.js',
    'static/js/localstorage.js',
    'static/js/message_edit.js',
    'static/js/message_edit_history.js',
    'static/js/message_events.js',
    'static/js/message_fetch.js',
    'static/js/message_flags.js',
    'static/js/message_list_data.js',
    'static/js/message_list.js',
    'static/js/message_list_view.js',
    'static/js/message_live_update.js',
    'static/js/message_scroll.js',
    'static/js/message_util.js',
    'static/js/message_viewport.js',
    'static/js/muting_ui.js',
    'static/js/narrow.js',
    'static/js/navigate.js',
    'static/js/night_mode.js',
    'static/js/notifications.js',
    'static/js/overlays.js',
    'static/js/padded_widget.js',
    'static/js/page_params.js',
    'static/js/panels.js',
    'static/js/pm_list_dom.js',
    'static/js/poll_widget.js',
    'static/js/popovers.js',
    'static/js/ready.js',
    'static/js/realm_icon.js',
    'static/js/realm_logo.js',
    'static/js/recent_topics.js',
    'static/js/reload.js',
    'static/js/reload_state.js',
    'static/js/reminder.js',
    'static/js/rendered_markdown.js',
    'static/js/resize.js',
    'static/js/rows.js',
    'static/js/scroll_bar.js',
    'static/js/search_pill_widget.js',
    'static/js/sent_messages.js',
    'static/js/server_events.js',
    'static/js/dropdown_list_widget.js',
    'static/js/settings_account.js',
    'static/js/settings_bots.js',
    'static/js/settings_config.js',
    'static/js/settings_display.js',
    'static/js/settings_emoji.js',
    'static/js/settings_exports.js',
    'static/js/settings_invites.js',
    'static/js/settings.js',
    'static/js/settings_linkifiers.js',
    'static/js/settings_notifications.js',
    'static/js/settings_org.js',
    'static/js/settings_panel_menu.js',
    'static/js/settings_profile_fields.js',
    'static/js/settings_sections.js',
    'static/js/settings_streams.js',
    'static/js/settings_toggle.js',
    'static/js/settings_ui.js',
    'static/js/settings_users.js',
    'static/js/setup.js',
    'static/js/spoilers.js',
    'static/js/starred_messages.js',
    'static/js/stream_color.js',
    'static/js/stream_create.js',
    'static/js/stream_edit.js',
    'static/js/stream_list.js',
    'static/js/stream_muting.js',
    'static/js/stream_popover.js',
    'static/js/stream_ui_updates.js',
    'static/js/submessage.js',
    'static/js/subs.js',
    'static/js/message_view_header.js',
    'static/js/templates.js',
    'static/js/timerender.js',
    'static/js/todo_widget.js',
    'static/js/topic_list.js',
    'static/js/topic_zoom.js',
    'static/js/tutorial.js',
    'static/js/typing_events.js',
    'static/js/typing.js',
    'static/js/ui_init.js',
    'static/js/ui.js',
    'static/js/ui_report.js',
    'static/js/ui_util.js',
    'static/js/unread_ops.js',
    'static/js/unread_ui.js',
    'static/js/upload_widget.js',
    'static/js/user_status_ui.js',
    'static/js/zcommand.js',
    'static/js/zform.js',
    'static/js/zulip.js',
}

parser = argparse.ArgumentParser(USAGE)
parser.add_argument('--coverage',
                    action="store_true",
                    help='Get coverage report')
parser.add_argument('--force',
                    action="store_true",
                    help='Run tests despite possible problems.')
parser.add_argument('args', nargs=argparse.REMAINDER)
options = parser.parse_args()
individual_files = options.args

from tools.lib.test_script import assert_provisioning_status_ok

assert_provisioning_status_ok(options.force)

def get_dev_host() -> str:
    # See similar code in dev_settings.py.  We only use
    # this to report where you can find coverage reports.
    # We duplicate the code here to avoid depending on
    # Django.

    host = os.getenv('EXTERNAL_HOST')
    if host is not None:
        return host

    user_id = os.getuid()
    user_name = pwd.getpwuid(user_id).pw_name
    if user_name == "zulipdev":
        hostname = os.uname()[1].lower()
        if '.zulipdev.org' not in hostname:
            hostname += '.zulipdev.org'
        return hostname + ':9991'
    else:
        # For local development environments, we use localhost by
        # default, via the "zulipdev.com" hostname.
        return 'zulipdev.com:9991'

def print_error(msg: str) -> None:
    print(BOLDRED + 'ERROR:' + ENDC + ' ' + msg)

def clean_file(orig_fn: str) -> str:
    fn = orig_fn
    if not fn.endswith(".js"):
        fn += ".js"
    if "frontend_tests/" not in fn:
        fn = os.path.join(ROOT_DIR, "frontend_tests", "node_tests", fn)
    fn = os.path.abspath(fn)
    if not os.path.exists(fn):
        print(f"Cannot find {orig_fn} ({fn})")
        sys.exit(1)
    return fn

def clean_files(fns: List[str]) -> List[str]:
    cleaned_files = [clean_file(fn) for fn in fns]
    return cleaned_files

def run_tests_via_node_js() -> int:
    os.environ['TZ'] = 'UTC'

    # The index.js test runner is the real "driver" here, and we launch
    # with either nyc or node, depending on whether we want coverage
    # reports.  Running under nyc is slower and creates funny
    # tracebacks, so you generally want to get coverage reports only
    # after making sure tests will pass.
    node_tests_cmd = ['node', '--stack-trace-limit=100', INDEX_JS]
    if individual_files:
        files = individual_files
    else:
        files = sorted(glob.glob("frontend_tests/node_tests/*.js"))

    node_tests_cmd += clean_files(files)
    if options.coverage:
        coverage_dir = os.path.join(ROOT_DIR, 'var/node-coverage')

        nyc = os.path.join(ROOT_DIR, "node_modules/.bin/nyc")
        command = [nyc, "--extension", ".hbs", "--extension", ".ts"]
        command += ["--report-dir", coverage_dir]
        command += ["--temp-directory", coverage_dir]
        command += ["-r=lcov", "-r=json", "-r=text-summary"]
        command += node_tests_cmd
    else:
        # Normal testing, no coverage analysis.
        # Run the index.js test runner, which runs all the other tests.
        command = node_tests_cmd

    print('Starting node tests...')

    # If we got this far, we can run the tests!
    try:
        ret = subprocess.check_call(command)
    except OSError:
        print(f'Bad command: {command}')
        raise
    except subprocess.CalledProcessError:
        print('\n** Tests failed, PLEASE FIX! **\n')
        sys.exit(1)
    return ret

def check_line_coverage(fn: str, line_coverage: Dict[Any, Any], line_mapping: Dict[Any, Any], log: bool = True) -> bool:
    missing_lines = []
    for line in line_coverage:
        if line_coverage[line] == 0:
            actual_line = line_mapping[line]
            missing_lines.append(str(actual_line["start"]["line"]))
    if missing_lines:
        if log:
            print_error(f"{fn} no longer has complete node test coverage")
            print("  Lines missing coverage: {}".format(", ".join(sorted(missing_lines, key=int))))
            print()
        return False
    return True

def read_coverage() -> Any:
    coverage_json = None
    try:
        with open(NODE_COVERAGE_PATH, "rb") as f:
            coverage_json = orjson.loads(f.read())
    except OSError:
        print(NODE_COVERAGE_PATH + " doesn't exist. Cannot enforce fully covered files.")
        raise
    return coverage_json

def enforce_proper_coverage(coverage_json: Any) -> bool:
    all_js_files = set(
        glob.glob('static/js/*.js') +
        glob.glob('static/js/*.ts') +
        glob.glob('static/shared/js/*.js') +
        glob.glob('static/shared/js/*.ts') +
        glob.glob('static/js/billing/*.js'),
    )
    enforce_fully_covered = all_js_files - EXEMPT_FILES

    coverage_lost = False
    for relative_path in enforce_fully_covered:
        path = ROOT_DIR + "/" + relative_path
        if not (path in coverage_json):
            coverage_lost = True
            print_error(f"{relative_path} has no node test coverage")
            continue
        line_coverage = coverage_json[path]['s']
        line_mapping = coverage_json[path]['statementMap']
        if not check_line_coverage(relative_path, line_coverage, line_mapping):
            coverage_lost = True
    if coverage_lost:
        print()
        print("It looks like your changes lost 100% test coverage in one or more files.")
        print("Ideally, you should add some tests to restore coverage.")
        print("A worse option is to update EXEMPT_FILES in `tools/test-js-with-node`.")
        print("To run this check locally, use `test-js-with-node --coverage`.")
        print()

    coverage_not_enforced = False
    for path in coverage_json:
        relative_path = os.path.relpath(path, ROOT_DIR)
        if relative_path in EXEMPT_FILES:
            line_coverage = coverage_json[path]['s']
            line_mapping = coverage_json[path]['statementMap']
            if check_line_coverage(relative_path, line_coverage, line_mapping, log=False):
                coverage_not_enforced = True
                print_error(f"{relative_path} unexpectedly has 100% line coverage.")

    if coverage_not_enforced:
        print()
        print("One or more fully covered files are miscategorized.")
        print("Remove the file(s) from EXEMPT_FILES in `tools/test-js-with-node`.")

    problems_encountered = (coverage_lost or coverage_not_enforced)
    return problems_encountered

ret = run_tests_via_node_js()

if options.coverage and ret == 0:
    if not individual_files:
        coverage_json = read_coverage()
        problems_encountered = enforce_proper_coverage(coverage_json)
        if problems_encountered:
            ret = 1

    reports_location = f"http://{get_dev_host()}/node-coverage/index.html"
    print()
    print("View coverage reports at " + CYAN + reports_location + ENDC)

print()
if ret == 0:
    print(GREEN + "Test(s) passed. SUCCESS!" + ENDC)
else:
    print(BOLDRED + "FAIL - Test(s) failed" + ENDC)

sys.exit(ret)