import type { ObjectSourceKind, TreeNode, TreeNodeType } from "@/types/database";
import { matchesShortcut, type ShortcutLikeEvent } from "@/lib/keyboardShortcuts";

export type TreeNodeRowAction = "open-data" | "toggle" | "none";
export type TreeNodeRowDoubleClickAction =
  | "open-data"
  | "open-object-browser"
  | "open-object-browser-and-expand"
  | "open-source"
  | "open-saved-sql"
  | "toggle"
  | "none";
export type SidebarSelectionCopyAction = "copy-name" | "none";
export type SidebarActivation = "single" | "double";

const dataNodeTypes = new Set<TreeNodeType>(["table", "view"]);
const toggleLeafNodeTypes = new Set<TreeNodeType>(["redis-db", "mongo-collection"]);
const objectBrowserNodeTypes = new Set<TreeNodeType>(["database", "schema", "object-browser"]);
const sourceNodeTypes = new Set<TreeNodeType>(["procedure", "function", "package", "package-body"]);
const tableChildGroupNodeTypes = new Set<TreeNodeType>([
  "group-columns",
  "group-indexes",
  "group-fkeys",
  "group-triggers",
  "group-partitions",
]);
const databaseChildGroupNodeTypes = new Set<TreeNodeType>([
  "group-tables",
  "group-views",
  "group-procedures",
  "group-functions",
  "group-packages",
]);

export function objectSourceKindForTreeNode(type: TreeNodeType): ObjectSourceKind | null {
  if (type === "view") return "VIEW";
  if (type === "procedure") return "PROCEDURE";
  if (type === "function") return "FUNCTION";
  if (type === "package") return "PACKAGE";
  if (type === "package-body") return "PACKAGE_BODY";
  return null;
}

export function treeNodeRowAction(
  type: TreeNodeType,
  canExpand: boolean,
  activation: SidebarActivation = "single",
): TreeNodeRowAction {
  if (activation === "double") return "none";
  if (dataNodeTypes.has(type)) return "open-data";
  if (toggleLeafNodeTypes.has(type)) return "toggle";
  if (canExpand) return "toggle";
  return "none";
}

export function treeNodeRowDoubleClickAction(
  type: TreeNodeType,
  canOpenObjectBrowser: boolean,
  activation: SidebarActivation = "single",
  canExpand = false,
): TreeNodeRowDoubleClickAction {
  if (activation === "double") {
    if (dataNodeTypes.has(type)) return "open-data";
    if (sourceNodeTypes.has(type)) return "open-source";
    if (type === "saved-sql-file") return "open-saved-sql";
    if (toggleLeafNodeTypes.has(type)) return "toggle";
    if (canOpenObjectBrowser && objectBrowserNodeTypes.has(type) && canExpand) return "open-object-browser-and-expand";
    if (canOpenObjectBrowser && objectBrowserNodeTypes.has(type)) return "open-object-browser";
    if (canExpand) return "toggle";
  }
  if (canOpenObjectBrowser && objectBrowserNodeTypes.has(type)) return "open-object-browser";
  return "none";
}

export function sidebarSelectionCopyAction(event: ShortcutLikeEvent): SidebarSelectionCopyAction {
  return matchesShortcut(event, "Mod+C") ? "copy-name" : "none";
}

export function copyNameForTreeNode(node: TreeNode): string {
  if (tableChildGroupNodeTypes.has(node.type) && node.tableName) return node.tableName;
  if (databaseChildGroupNodeTypes.has(node.type)) return node.schema || node.database || node.label;
  return node.label;
}