# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Speedometer 3 Web Interaction Benchmark Pages
"""
import os
import re

from core import path_util
from page_sets import press_story
from telemetry import story

_SPEEDOMETER_SUITES = (
  'TodoMVC-JavaScript-ES5',
  'TodoMVC-JavaScript-ES6-Webpack',
  'TodoMVC-WebComponents',
  'TodoMVC-React',
  'TodoMVC-React-Complex-DOM',
  'TodoMVC-React-Redux',
  'TodoMVC-Backbone',
  'TodoMVC-Angular',
  'TodoMVC-Vue',
  'TodoMVC-jQuery',
  'TodoMVC-Preact',
  'TodoMVC-Svelte',
  'TodoMVC-Lit',
  'TodoMVC-JavaScript-ES5-Complex-DOM',
  'TodoMVC-JavaScript-ES6-Webpack-Complex-DOM',
  'TodoMVC-WebComponents-Complex-DOM',
  'TodoMVC-React-Redux-Complex-DOM',
  'TodoMVC-Backbone-Complex-DOM',
  'TodoMVC-Angular-Complex-DOM',
  'TodoMVC-Vue-Complex-DOM',
  'TodoMVC-jQuery-Complex-DOM',
  'TodoMVC-Preact-Complex-DOM',
  'TodoMVC-Svelte-Complex-DOM',
  'TodoMVC-Lit-Complex-DOM',
  'NewsSite-Next',
  'NewsSite-Nuxt',
  'Editor-CodeMirror',
  'Editor-TipTap',
  'Charts-observable-plot',
  'Charts-chartjs',
  'React-Stockcharts-SVG',
  'Perf-Dashboard',
)
_PAGE_SET_DIR = os.path.join(path_util.GetChromiumSrcDir(), 'tools', 'perf',
                             'page_sets')


class _Speedometer3Story(press_story.PressStory):
  URL = 'file://index.html'

  def __init__(self,
               page_set,
               should_filter_suites,
               filtered_suite_names=None,
               iterations=None,
               enable_details=False,
               take_memory_measurement=False):
    super(_Speedometer3Story, self).__init__(page_set)
    self._should_filter_suites = should_filter_suites
    self._filtered_suite_names = filtered_suite_names
    self._iterations = iterations
    self._enable_details = enable_details
    self._take_memory_measurement = take_memory_measurement

  @staticmethod
  def GetSuites(suite_regex):
    if not suite_regex:
      return []
    exp = re.compile(suite_regex)
    return [name for name in _SPEEDOMETER_SUITES if exp.search(name)]

  def RunNavigateSteps(self, action_runner):
    DEFAULT_ITERATIONS = 10

    url = self.file_path_url_with_scheme if self.is_file else self.url

    iterations = (self._iterations
                  if self._iterations is not None else DEFAULT_ITERATIONS)
    url = "%s?iterationCount=%s" % (url, iterations)
    action_runner.Navigate(
        url, script_to_evaluate_on_commit=self.script_to_evaluate_on_commit)

  def ExecuteTest(self, action_runner):
    action_runner.tab.WaitForDocumentReadyStateToBeComplete()

    if self._should_filter_suites:
      action_runner.ExecuteJavaScript(
          """
        Suites.forEach(function(suite) {
          suite.disabled = {{ filtered_suites }}.indexOf(suite.name) == -1;
        });
      """,
          filtered_suites=self._filtered_suite_names)

    action_runner.ExecuteJavaScript("""
        // Store all the results in the benchmarkClient
        window.testDone = false;
        window.metrics = Object.create(null);
        const client = window.benchmarkClient;
        const clientCopy = {
          didFinishLastIteration: client.didFinishLastIteration,
        };
        client.didFinishLastIteration = function(metrics) {
            clientCopy.didFinishLastIteration.call(this, metrics);
            window.metrics = metrics
            window.testDone = true;
        };
        """)
    action_runner.ExecuteJavaScript("""
        if (window.startTest) {
          window.startTest();
        } else {
          // Interactive Runner fallback / old 3.0 fallback.
          let startButton = document.getElementById("runSuites") ||
              document.querySelector("start-tests-button") ||
              document.querySelector(".buttons button");
          startButton.click();
        }
        """)
    action_runner.WaitForJavaScriptCondition('testDone', timeout=1800)
    if self._take_memory_measurement:
      action_runner.MeasureMemory(deterministic_mode=True)

  def ParseTestResults(self, action_runner):
    # Extract the timings for each suite
    metrics = action_runner.EvaluateJavaScript(
        "(function() { return window.metrics })()")
    assert metrics, "Expected metrics dict but got: %s" % metrics
    UNIT_LOOKUP = {
        "ms": "ms_smallerIsBetter",
        "score": "unitless_biggerIsBetter",
    }
    for name, metric in metrics.items():
      if not self._IsSpeedometerMetricEnabled(name):
        continue
      self.AddMeasurement(name, UNIT_LOOKUP[metric["unit"]], metric["values"])

  def _IsSpeedometerMetricEnabled(self, name):
    if self._enable_details:
      return True
    # Skip nested metrics:
    if "/" in name:
      return False
    # Skip top-level iteration metrics:
    if name.startswith("Iteration-"):
      return False
    if name == "Geomean":
      return False
    return True


class Speedometer30Story(_Speedometer3Story):
  NAME = 'Speedometer30'


class Speedometer31Story(_Speedometer3Story):
  NAME = 'Speedometer31'


class Speedometer3Story(Speedometer31Story):
  NAME = 'Speedometer3'


class Speedometer30CrossbenchStory(story.StorySet):
  NAME = 'speedometer3.crossbench'

  def __init__(self):
    super().__init__(
        base_dir=_PAGE_SET_DIR,
        archive_data_file='data/crossbench_android_speedometer_3.0.json',
        cloud_storage_bucket=story.PARTNER_BUCKET)