// Copyright (C) 2024 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only

import * as vscode from 'vscode';
import * as path from 'path';
import * as fs from 'fs';

import { getSelectedKit, IsQtKit } from '@cmd/register-qt-path';
import { createLogger, telemetry } from 'qt-lib';
import { EXTENSION_ID } from '@/constants';
import { QtVersionFromKit } from '@/util/util';

const logger = createLogger('natvis');

export function registerNatvisCommand() {
  const getNatvis = (version: string) => {
    const extension = vscode.extensions.getExtension(
      `theqtcompany.${EXTENSION_ID}`
    );
    if (!extension) {
      const error = 'Could not find the extension';
      logger.error(error);
      throw new Error(error);
    }
    const extensionPath = extension.extensionPath;
    if (!extensionPath) {
      const error = 'Could not find the extension path';
      logger.error(error);
      throw new Error(error);
    }
    const natvisFile = path.join(
      extensionPath,
      'res',
      'natvis',
      `qt${version}.natvis.xml`
    );
    if (!fs.existsSync(natvisFile)) {
      const error = `Could not find the natvis file: ${natvisFile}`;
      logger.error(error);
      throw new Error(error);
    }
    return natvisFile;
  };

  const natvisDisposal = vscode.commands.registerCommand(
    `${EXTENSION_ID}.natvis`,
    async () => {
      telemetry.sendAction('natvis');
      const kit = await getSelectedKit();
      if (!kit || !IsQtKit(kit)) {
        const error = `${kit?.name} is not a Qt kit`;
        throw new Error(error);
      }
      const version = QtVersionFromKit(kit);
      if (version) {
        const majorVersion = version.split('.')[0];
        if (!majorVersion) {
          throw new Error('Could not determine the major version');
        }
        return getNatvis(majorVersion);
      }
      return undefined;
    }
  );
  const natvis5Disposal = vscode.commands.registerCommand(
    `${EXTENSION_ID}.natvis5`,
    () => {
      telemetry.sendAction('natvis5');
      return getNatvis('5');
    }
  );
  const natvis6Disposal = vscode.commands.registerCommand(
    `${EXTENSION_ID}.natvis6`,
    () => {
      telemetry.sendAction('natvis6');
      return getNatvis('6');
    }
  );

  return [natvisDisposal, natvis5Disposal, natvis6Disposal];
}