homepage/src/widgets/glances/metrics/gpu.jsx

import { useTranslation } from "next-i18next/pages";
import dynamic from "next/dynamic";
import { useCallback } from "react";

import Block from "../components/block";
import Container from "../components/container";

import useDataPoints from "./use-data-points";

import { parseVersionForUrl } from "utils/proxy/api-helpers";
import useWidgetAPI from "utils/proxy/use-widget-api";

const ChartDual = dynamic(() => import("../components/chart_dual"), { ssr: false });

const defaultPointsLimit = 15;
const defaultInterval = 1000;

export default function Component({ service }) {
  const { t } = useTranslation();
  const { widget } = service;
  const { chart, refreshInterval = defaultInterval, pointsLimit = defaultPointsLimit, version = 3 } = widget;
  const apiVersion = parseVersionForUrl(version, 3);
  const [, gpuName] = widget.metric.split(":");

  const [dataPoints, addDataPoint] = useDataPoints(pointsLimit, { a: 0, b: 0 });

  const handleData = useCallback(
    (newData) => {
      if (!newData?.error) {
        const gpuData = newData.find((item) => item[item.key] == gpuName);
        if (gpuData) addDataPoint({ a: gpuData.mem, b: gpuData.proc });
      }
    },
    [addDataPoint, gpuName],
  );

  const { data, error } = useWidgetAPI(
    widget,
    `${apiVersion}/gpu`,
    {
      refreshInterval: Math.max(defaultInterval, refreshInterval),
    },
    { onSuccess: handleData },
  );

  if (error || (data && data.error)) {
    const finalError = error || data.error;
    return <Container error={finalError} widget={widget} />;
  }

  if (!data) {
    return (
      <Container chart={chart}>
        <Block position="bottom-3 left-3">-</Block>
      </Container>
    );
  }

  const gpuData = data.find((item) => item[item.key] == gpuName);

  if (!gpuData) {
    return (
      <Container chart={chart}>
        <Block position="bottom-3 left-3">-</Block>
      </Container>
    );
  }

  return (
    <Container chart={chart}>
      {chart && (
        <ChartDual
          dataPoints={dataPoints}
          label={[t("glances.mem"), t("glances.gpu")]}
          stack={["mem", "proc"]}
          formatter={(value) =>
            t("common.percent", {
              value,
              maximumFractionDigits: 1,
            })
          }
        />
      )}

      {chart && (
        <Block position="bottom-3 left-3">
          {gpuData && gpuData.name && <div className="text-xs opacity-50">{gpuData.name}</div>}

          <div className="text-xs opacity-50">
            {t("common.number", {
              value: gpuData.mem,
              maximumFractionDigits: 1,
            })}
            % {t("resources.mem")}
          </div>
        </Block>
      )}

      {!chart && (
        <Block position="bottom-3 left-3">
          <div className="text-xs opacity-50">
            {t("common.number", {
              value: gpuData.temperature,
              maximumFractionDigits: 1,
            })}
            &deg; C
          </div>
        </Block>
      )}

      <Block position="bottom-3 right-3">
        <div className="text-xs opacity-75">
          {!chart && (
            <div className="inline-block mr-1">
              {t("common.number", {
                value: gpuData.proc,
                maximumFractionDigits: 1,
              })}
              % {t("glances.gpu")}
            </div>
          )}
          {!chart && <>&bull;</>}
          <div className="inline-block ml-1">
            {t("common.number", {
              value: gpuData.proc,
              maximumFractionDigits: 1,
            })}
            % {t("glances.gpu")}
          </div>
        </div>
      </Block>

      <Block position="top-3 right-3">
        {chart && (
          <div className="text-xs opacity-50">
            {t("common.number", {
              value: gpuData.temperature,
              maximumFractionDigits: 1,
            })}
            &deg; C
          </div>
        )}

        {gpuData && gpuData.name && !chart && <div className="text-xs opacity-50">{gpuData.name}</div>}
      </Block>
    </Container>
  );
}