import {
  getKeyName,
  getTypeName,
  listServices,
  Model,
  navigateTypesInNamespace,
  Program,
} from "@typespec/compiler";
import { DuplicateTracker } from "@typespec/compiler/utils";
import { isSharedRoute } from "@typespec/http";
import { reportDiagnostic } from "./lib.js";
import { getParentResource, getResourceTypeKey, ResourceKey } from "./resource.js";
import { getActionDetails, getCollectionActionDetails } from "./rest.js";

function checkForDuplicateResourceKeyNames(program: Program) {
  const seenTypes = new Set<string>();

  function checkResourceModelKeys(model: Model) {
    if (model.name === "") {
      return;
    }
    let currentType: Model | undefined = model;
    const keyProperties = new DuplicateTracker<string, ResourceKey>();
    while (currentType) {
      const resourceKey = getResourceTypeKey(program, currentType);
      if (resourceKey) {
        const keyName = getKeyName(program, resourceKey.keyProperty)!;
        keyProperties.track(keyName, resourceKey);
      }

      currentType = getParentResource(program, currentType);
    }

    // Report a diagnostic for each duplicate key
    for (const [keyName, dupes] of keyProperties.entries()) {
      for (const key of dupes) {
        // Make sure we don't report a duplicate for a particular
        // resource type and key name more than once
        const fullName = `${getTypeName(key.resourceType)}.${keyName}`;
        if (!seenTypes.has(fullName)) {
          seenTypes.add(fullName);
          reportDiagnostic(program, {
            code: "duplicate-parent-key",
            format: {
              resourceName: key.resourceType.name,
              keyName,
            },
            target: key.keyProperty,
          });
        }
      }
    }
  }

  for (const service of listServices(program)) {
    // If the model type is defined under the service namespace, check that the
    // parent resource type(s) don't have the same key name as the
    // current resource type.
    navigateTypesInNamespace(service.type, {
      model: (model) => checkResourceModelKeys(model),
    });
  }
}

function checkForSharedRouteUnnamedActions(program: Program) {
  for (const service of listServices(program)) {
    // If the model type is defined under the service namespace, check that the
    // parent resource type(s) don't have the same key name as the
    // current resource type.
    navigateTypesInNamespace(service.type, {
      operation: (op) => {
        const actionDetails = getActionDetails(program, op);
        if (
          isSharedRoute(program, op) &&
          (actionDetails?.kind === "automatic" ||
            getCollectionActionDetails(program, op)?.kind === "automatic")
        ) {
          reportDiagnostic(program, {
            code: "shared-route-unspecified-action-name",
            target: op,
            format: {
              decoratorName: actionDetails ? "@action" : "@collectionAction",
            },
          });
        }
      },
    });
  }
}

export function $onValidate(program: Program) {
  // Make sure any defined resource types don't have any conflicts with parent
  // resource type key names
  checkForDuplicateResourceKeyNames(program);

  // Ensure that all shared routes with an `@action` decorator have an
  // explicitly named action
  checkForSharedRouteUnnamedActions(program);
}
