import React, { useState } from "react";
import { createRoot } from "react-dom/client";
import { AgCharts } from "ag-charts-react";
import { AgChartLegendClickEvent, AgChartOptions } from "ag-charts-community";
import { getData } from "./data";

const ChartExample = () => {
  const [options, setOptions] = useState<AgChartOptions>({
    title: {
      text: `Renewable sources used to generate electricity for transport fuels`,
    },
    data: getData(),
    series: [
      {
        type: "line",
        xKey: "year",
        yKey: "Onshore wind",
        yName: "Onshore Wind",
      },
      {
        type: "line",
        xKey: "year",
        yKey: "Offshore wind",
        yName: "Offshore Wind",
      },
      {
        type: "line",
        xKey: "year",
        yKey: "Marine energy",
        yName: "Marine Energy",
      },
      {
        type: "line",
        xKey: "year",
        yKey: "Solar photovoltaics",
        yName: "Solar Photovoltaics",
      },
      {
        type: "line",
        xKey: "year",
        yKey: "Small scale Hydro",
        yName: "Small Scale Hydro",
      },
      {
        type: "line",
        xKey: "year",
        yKey: "Large scale Hydro",
        yName: "Large Scale Hydro",
      },
      {
        type: "line",
        xKey: "year",
        yKey: "Plant biomass",
        yName: "Plant Biomass",
      },
      {
        type: "line",
        xKey: "year",
        yKey: "Animal biomass",
        yName: "Animal Biomass",
      },
      {
        type: "line",
        xKey: "year",
        yKey: "Landfill gas",
        yName: "Landfill Gas",
      },
      {
        type: "line",
        xKey: "year",
        yKey: "Sewage gas",
        yName: "Sewage Gas",
      },
    ],
    axes: [
      {
        position: "bottom",
        type: "time",
        nice: false,
      },
      {
        position: "left",
        type: "number",
        title: {
          text: `kilotonnes of oil equivalent (ktoe)`,
        },
        label: {
          formatter: (params) => `${params.value / 1000}K`,
        },
      },
    ],
    legend: {
      listeners: {
        legendItemClick: ({ seriesId, itemId }: AgChartLegendClickEvent) => {
          console.log(`seriesId: ${seriesId}, itemId: ${itemId}`);
        },
      },
    },
  });

  return <AgCharts options={options} />;
};

const root = createRoot(document.getElementById("root")!);
root.render(<ChartExample />);

/** DARK MODE START **/
import { AgCharts as __chartAPI } from "ag-charts-community";

let darkmode =
  (localStorage["documentation:darkmode"] ||
    String(matchMedia("(prefers-color-scheme: dark)").matches)) === "true";

const isAgThemeOrUndefined = (theme) => {
  return (
    theme == null || (typeof theme === "string" && theme.startsWith("ag-"))
  );
};

const getDarkmodeTheme = (theme = "ag-default", preset) => {
  const baseTheme =
    preset === "price-volume" ? "ag-financial" : theme.replace(/-dark$/, "");
  return darkmode ? baseTheme + "-dark" : baseTheme;
};

__chartAPI.optionsMutationFn = function update(options, preset) {
  const nextOptions = { ...options };
  const theme = options.theme;
  if (isAgThemeOrUndefined(theme)) {
    nextOptions.theme = getDarkmodeTheme(theme, preset);
  } else if (
    typeof theme === "object" &&
    isAgThemeOrUndefined(theme.baseTheme)
  ) {
    nextOptions.theme = {
      ...options.theme,
      baseTheme: getDarkmodeTheme(theme.baseTheme, preset),
    };
  }
  return nextOptions;
};

const applyDarkmode = () => {
  document.documentElement.setAttribute("data-dark-mode", darkmode);
  const charts = document.querySelectorAll("[data-ag-charts]");
  charts.forEach((element) => {
    const chart = __chartAPI.getInstance(element.parentElement);
    if (chart == null) return;
    // This is just needed to trigger the theme update
    chart.update(chart.getOptions());
  });
  return charts.length !== 0;
};

if (!applyDarkmode()) {
  /* React defers updates. Rather than try and hook into the API, just wait until the darkmode is applied. */
  const observer = new MutationObserver(() => {
    if (applyDarkmode()) {
      observer.disconnect();
    }
  });
  observer.observe(document.body, {
    attributes: true,
    childList: true,
    subtree: true,
  });
}
window.addEventListener("message", (event) => {
  if (event.data && event.data.type === "color-scheme-change") {
    darkmode = event.data.darkmode;
    applyDarkmode();
  }
});
/** DARK MODE END **/
/** CONSOLE LOG START **/
const PRIMITIVE_TYPES = [
  "string",
  "number",
  "boolean",
  "undefined",
  "null",
  "NaN",
  "symbol",
];
const OBJECT_PROPERTIES_LIMIT = 3;

function getType(value) {
  if (value === null) return "null";
  if (Number.isNaN(value)) return "NaN";
  if (Array.isArray(value)) return "array";
  if (value instanceof Date) return "date";
  if (value instanceof RegExp) return "regexp";
  if (value instanceof Map) return "map";
  if (value instanceof Set) return "set";
  if (value instanceof WeakMap) return "weakmap";
  if (value instanceof WeakSet) return "weakset";
  if (value instanceof Promise) return "promise";
  if (value instanceof Error) return "error";
  if (typeof value === "object") return "object";
  return typeof value;
}

function safeStringify(obj, space = 2) {
  const seen = new WeakSet();
  const isLoggableArray = getType(obj) === "array" && obj.every(isLoggableType);
  const getObjectValue = (value) => {
    if (seen.has(value)) {
      return "[Circular]";
    } else if (value === undefined) {
      return "undefined";
    }
    seen.add(value);

    // Include custom class names if available
    if (
      value.constructor &&
      value.constructor.name &&
      getType(value) !== "object"
    ) {
      return `${value.constructor.name}Class { ... }`;
    }

    return value;
  };
  return isLoggableArray
    ? JSON.stringify(obj)
    : JSON.stringify(
        obj,
        (_, value) => {
          const valueType = getType(value);
          let newValue = value;
          if (valueType === "object") {
            newValue = getObjectValue(newValue);
          } else if (valueType === "array") {
            newValue = value.map((item) => {
              return getType(item === "object") ? getObjectValue(item) : item;
            });
          }

          return newValue;
        },
        space,
      );
}

function isPrimitiveType(value) {
  return PRIMITIVE_TYPES.includes(getType(value));
}

function isLoggableType(value) {
  const valueType = getType(value);

  return (
    isPrimitiveType(value) ||
    (valueType === "array" && value.every(isPrimitiveType)) ||
    (valueType === "object" &&
      Object.values(value).every(isPrimitiveType) &&
      Object.keys(value).length <= OBJECT_PROPERTIES_LIMIT)
  );
}

function getConsoleValue(value) {
  return isPrimitiveType(value)
    ? value
    : {
        __consoleLogObject: true,
        isLoggable: isLoggableType(value),
        argType: getType(value),
        safeString: safeStringify(value),
      };
}

function getConsoleLogData(args) {
  return args.map(getConsoleValue);
}

const originalConsoleLog = console.log;
console.log = (...args) => {
  try {
    window.parent.postMessage({
      type: "console-log",
      pageName: "legend",
      exampleName: "legend-click-series-toggle",
      data: getConsoleLogData(args),
    });
  } catch {
    // Posting is best-effort and shouldn't block normal console logging.
  }
  originalConsoleLog(...args);
};
/** CONSOLE LOG END **/
