/** Investing section (DCA) */

import { Unit } from "../../utils/units.js";
import { satsBtcUsd } from "../shared.js";
import { periodIdToName } from "./utils.js";

/**
 * Build DCA classes data array
 * @param {Colors} colors
 * @param {MarketDca} dca
 */
export function buildDcaClasses(colors, dca) {
  return /** @type {const} */ ([
    [2025, "pink", true],
    [2024, "fuchsia", true],
    [2023, "purple", true],
    [2022, "blue", true],
    [2021, "sky", true],
    [2020, "teal", true],
    [2019, "green", true],
    [2018, "yellow", true],
    [2017, "orange", true],
    [2016, "red", false],
    [2015, "pink", false],
  ]).map(([year, colorKey, defaultActive]) => ({
    year,
    color: colors[colorKey],
    defaultActive,
    costBasis: dca.classAveragePrice[`_${year}`],
    returns: dca.classReturns[`_${year}`],
    stack: dca.classStack[`_${year}`],
  }));
}

/**
 * Create Investing section
 * @param {PartialContext} ctx
 * @param {Object} args
 * @param {Market["dca"]} args.dca
 * @param {Market["lookback"]} args.lookback
 * @param {Market["returns"]} args.returns
 */
export function createInvestingSection(ctx, { dca, lookback, returns }) {
  const { line, baseline, colors, createPriceLine } = ctx;
  const dcaClasses = buildDcaClasses(colors, dca);

  return {
    name: "Investing",
    tree: [
      // DCA vs Lump sum
      {
        name: "DCA vs Lump sum",
        tree: /** @type {const} */ ([
          ["1w", "_1w"],
          ["1m", "_1m"],
          ["3m", "_3m"],
          ["6m", "_6m"],
          ["1y", "_1y"],
          ["2y", "_2y"],
          ["3y", "_3y"],
          ["4y", "_4y"],
          ["5y", "_5y"],
          ["6y", "_6y"],
          ["8y", "_8y"],
          ["10y", "_10y"],
        ]).map(([id, key]) => {
          const name = periodIdToName(id, true);
          const priceAgo = lookback[key];
          const priceReturns = returns.priceReturns[key];
          const dcaCostBasis = dca.periodAveragePrice[key];
          const dcaReturns = dca.periodReturns[key];
          const dcaStack = dca.periodStack[key];
          const lumpSumStack = dca.periodLumpSumStack[key];
          return {
            name,
            tree: [
              {
                name: "Cost basis",
                title: `${name} DCA vs Lump Sum (Cost Basis)`,
                top: [
                  line({
                    metric: dcaCostBasis,
                    name: "DCA",
                    color: colors.green,
                    unit: Unit.usd,
                  }),
                  line({
                    metric: priceAgo,
                    name: "Lump sum",
                    color: colors.orange,
                    unit: Unit.usd,
                  }),
                ],
              },
              {
                name: "Returns",
                title: `${name} DCA vs Lump Sum (Returns)`,
                bottom: [
                  baseline({
                    metric: dcaReturns,
                    name: "DCA",
                    unit: Unit.percentage,
                  }),
                  baseline({
                    metric: priceReturns,
                    name: "Lump sum",
                    color: [colors.lime, colors.red],
                    unit: Unit.percentage,
                  }),
                  createPriceLine({ unit: Unit.percentage }),
                ],
              },
              {
                name: "Stack",
                title: `${name} DCA vs Lump Sum Stack ($100/day)`,
                bottom: [
                  ...satsBtcUsd(ctx, dcaStack, "DCA", colors.green),
                  ...satsBtcUsd(ctx, lumpSumStack, "Lump sum", colors.orange),
                ],
              },
            ],
          };
        }),
      },

      // DCA classes
      {
        name: "DCA classes",
        tree: [
          // Comparison charts (all years overlaid)
          {
            name: "Compare",
            tree: [
              {
                name: "Cost basis",
                title: "DCA Cost Basis by Year",
                top: dcaClasses.map(
                  ({ year, color, defaultActive, costBasis }) =>
                    line({
                      metric: costBasis,
                      name: `${year}`,
                      color,
                      defaultActive,
                      unit: Unit.usd,
                    }),
                ),
              },
              {
                name: "Returns",
                title: "DCA Returns by Year",
                bottom: dcaClasses.map(
                  ({ year, color, defaultActive, returns }) =>
                    baseline({
                      metric: returns,
                      name: `${year}`,
                      color,
                      defaultActive,
                      unit: Unit.percentage,
                    }),
                ),
              },
              {
                name: "Stack",
                title: "DCA Stack by Year ($100/day)",
                bottom: dcaClasses.flatMap(
                  ({ year, color, defaultActive, stack }) =>
                    satsBtcUsd(ctx, stack, `${year}`, color, { defaultActive }),
                ),
              },
            ],
          },
          // Individual year charts
          ...dcaClasses.map(({ year, color, costBasis, returns, stack }) => ({
            name: `${year}`,
            tree: [
              {
                name: "Cost basis",
                title: `DCA Class ${year} Cost Basis`,
                top: [
                  line({
                    metric: costBasis,
                    name: "Cost basis",
                    color,
                    unit: Unit.usd,
                  }),
                ],
              },
              {
                name: "Returns",
                title: `DCA Class ${year} Returns`,
                bottom: [
                  baseline({
                    metric: returns,
                    name: "Returns",
                    color,
                    unit: Unit.percentage,
                  }),
                ],
              },
              {
                name: "Stack",
                title: `DCA Class ${year} Stack ($100/day)`,
                bottom: satsBtcUsd(ctx, stack, "Stack", color),
              },
            ],
          })),
        ],
      },
    ],
  };
}