/**
 * Layout tree model — the Dockview-style structure that replaces the
 * rails/bands grammar. Two node kinds:
 *
 *  - `split`: children laid out along an orientation with fractional weights.
 *  - `group`: a stack of panes (tabs) with one active; may be minimized to
 *    its header strip (DetailPane semantics).
 *
 * Everything the old grammar special-cased is just tree shape here: a "top row
 * spanning the right rail" is a column split; "a cell inside a column" is a
 * stacked group; spans fall out of tree position. All operations are pure and
 * return new trees; `normalize` keeps the structure canonical (no empty
 * groups, no single-child or same-orientation nested splits).
 */

export type Orientation = 'row' | 'column'

export interface SplitNode {
  type: 'split'
  id: string
  orientation: Orientation
  children: LayoutNode[]
  /** Parallel to children; relative flex weights. */
  weights: number[]
}

export interface GroupNode {
  type: 'group'
  id: string
  /** Pane ids stacked in this group (rendered as tabs when > 1). */
  panes: string[]
  /** The visible pane. */
  active: string
  /** Collapsed to header strip (chevron restores). */
  minimized?: boolean
  /**
   * Header hidden entirely (double-click the header to hide, double-click the
   * zone's top edge to bring it back). Minimize always shows the header —
   * a minimized group IS its header.
   */
  headerHidden?: boolean
}

export type LayoutNode = SplitNode | GroupNode

/** Where a dragged pane lands relative to a target group. */
export type DropPosition = 'center' | 'left' | 'right' | 'top' | 'bottom'

export type RootEdge = 'left' | 'right' | 'top' | 'bottom'

let seq = 0
export const nodeId = (kind: string) => `${kind}-${Date.now().toString(36)}-${(seq++).toString(36)}`

export const group = (panes: string[], options?: Partial<Omit<GroupNode, 'type' | 'panes'>>): GroupNode => ({
  type: 'group',
  id: options?.id ?? nodeId('g'),
  panes,
  active: options?.active ?? panes[0] ?? '',
  minimized: options?.minimized,
  headerHidden: options?.headerHidden
})

export const split = (
  orientation: Orientation,
  children: LayoutNode[],
  weights?: number[],
  id?: string
): SplitNode => ({
  type: 'split',
  id: id ?? nodeId('s'),
  orientation,
  children,
  weights: weights ?? children.map(() => 1)
})

// ---------------------------------------------------------------------------
// Queries
// ---------------------------------------------------------------------------

export function findGroup(node: LayoutNode, groupId: string): GroupNode | null {
  if (node.type === 'group') {
    return node.id === groupId ? node : null
  }

  for (const child of node.children) {
    const hit = findGroup(child, groupId)

    if (hit) {
      return hit
    }
  }

  return null
}

export function findGroupOfPane(node: LayoutNode, paneId: string): GroupNode | null {
  if (node.type === 'group') {
    return node.panes.includes(paneId) ? node : null
  }

  for (const child of node.children) {
    const hit = findGroupOfPane(child, paneId)

    if (hit) {
      return hit
    }
  }

  return null
}

export function allPaneIds(node: LayoutNode): string[] {
  return node.type === 'group' ? [...node.panes] : node.children.flatMap(allPaneIds)
}

// ---------------------------------------------------------------------------
// Structural edits (pure)
// ---------------------------------------------------------------------------

/**
 * Canonical form: unwrap single-child splits, flatten same-orientation
 * nesting (weights scaled into the parent's slot), and PRUNE EMPTY GROUPS —
 * dragging the last pane out of a zone closes the zone and its siblings
 * absorb the space (VS Code semantics). Keeping empties as "stable regions"
 * (the original FancyZones rule) let invisible residue accumulate into
 * corrupt-feeling structure (`row([]|[])` eating half a slot); authored
 * empty zones still exist inside the zone editor's own grid model, and an
 * editor-applied tree keeps them until the first structural op.
 */
export function normalize(node: LayoutNode): LayoutNode | null {
  if (node.type === 'group') {
    if (node.panes.length === 0) {
      return null
    }

    const active = node.panes.includes(node.active) ? node.active : node.panes[0]

    // NOTE: `headerHidden` is deliberately untouched here. A zone down to one
    // pane is headerless by default anyway, so a stored `true` is visually
    // redundant *while it's alone* — but normalize used to DROP it, which threw
    // away the user's standing choice: the bar came back the moment a pane
    // rejoined (close a stacked tool panel, toggle it back on). `false` is
    // sticky for the mirror reason — once a zone has had a tab bar, it keeps it.
    if (active === node.active) {
      return node
    }

    return { ...node, active }
  }

  const children: LayoutNode[] = []
  const weights: number[] = []

  node.children.forEach((child, i) => {
    const kept = normalize(child)

    if (!kept) {
      return
    }

    if (kept.type === 'split' && kept.orientation === node.orientation) {
      // Flatten: distribute this slot's weight across the flattened children
      // proportionally to their internal weights.
      const total = kept.weights.reduce((a, b) => a + b, 0) || 1
      kept.children.forEach((grandchild, j) => {
        children.push(grandchild)
        weights.push((node.weights[i] ?? 1) * ((kept.weights[j] ?? 1) / total))
      })

      return
    }

    children.push(kept)
    weights.push(node.weights[i] ?? 1)
  })

  if (children.length === 0) {
    return null
  }

  if (children.length === 1) {
    return children[0]
  }

  return { ...node, children, weights }
}

/** Remove a pane wherever it lives. Closing the ACTIVE tab leaves selection on
 *  the neighbor that fills its slot (right; left when it was last) — same rule
 *  as terminals and the preview rail. */
export function removePane(node: LayoutNode, paneId: string): LayoutNode | null {
  const walk = (n: LayoutNode): LayoutNode => {
    if (n.type === 'group') {
      const at = n.panes.indexOf(paneId)

      if (at === -1) {
        return n
      }

      const panes = n.panes.filter(p => p !== paneId)

      // After splice, `at` indexes the old right neighbor (clamp left at end).
      return { ...n, panes, active: n.active === paneId ? (panes[Math.min(at, panes.length - 1)] ?? '') : n.active }
    }

    return { ...n, children: n.children.map(walk) }
  }

  return normalize(walk(node))
}

/**
 * Insert `paneId` at `target` group: `center` joins the stack (as a tab);
 * an edge splits the group in that direction. If the neighboring split
 * already runs in that orientation the new group is spliced in beside the
 * target instead of nesting (normalize would flatten it anyway).
 */
export function insertAtGroup(
  node: LayoutNode,
  targetGroupId: string,
  paneId: string,
  pos: DropPosition,
  /** Center drops only: stack BEFORE this pane id (`null`/omitted = append) —
   *  the tab-strip insertion divider's slot. */
  before?: null | string,
  /** Front the inserted pane — TRUE for a gesture (drop/reveal), FALSE for silent
   *  adoption (logs stacking into the terminal zone must not steal its tab). */
  activate: boolean = true
): LayoutNode | null {
  const walk = (n: LayoutNode): LayoutNode => {
    if (n.type === 'group') {
      if (n.id !== targetGroupId) {
        return n
      }

      if (pos === 'center') {
        const at = before ? n.panes.indexOf(before) : -1
        const panes = at >= 0 ? [...n.panes.slice(0, at), paneId, ...n.panes.slice(at)] : [...n.panes, paneId]

        // Gaining a pane pins the header EXPLICITLY shown (not just cleared):
        // a stack you can't see is a trap, and once a zone has ever stacked
        // the bar STAYS when it drops back to one tab — the auto-hide flicker
        // while dragging tabs around felt broken. Hiding is the user's call
        // (double-click / zone menu). Active moves only on a gesture; an empty
        // target has no prior tab, so the newcomer takes it regardless.
        const active = activate || n.panes.length === 0 ? paneId : n.active

        return { ...n, panes, active, headerHidden: false }
      }

      const orientation: Orientation = pos === 'left' || pos === 'right' ? 'row' : 'column'
      const leading = pos === 'left' || pos === 'top'
      const added = group([paneId])
      const children = leading ? [added, n] : [n, added]

      return split(orientation, children, [1, 1])
    }

    return { ...n, children: n.children.map(walk) }
  }

  return normalize(walk(node))
}

/**
 * The tree's VISIBLE shape: pane stacks + split orientations, with empty
 * groups skipped (editor-session trees may still hold them) and single-child
 * runs unwrapped. Two trees with equal signatures are indistinguishable on
 * screen regardless of node ids.
 */
function shapeSignature(node: LayoutNode): string {
  if (node.type === 'group') {
    return node.panes.length > 0 ? `[${node.panes.join(',')}]` : ''
  }

  const children = node.children.map(shapeSignature).filter(Boolean)

  if (children.length === 0) {
    return ''
  }

  return children.length === 1 ? children[0] : `${node.orientation}(${children.join('|')})`
}

/**
 * Move = remove + insert. If the target group vanished during removal (the
 * pane was its only occupant), the move is a no-op. A move whose result
 * LOOKS identical to the current layout is also a no-op — e.g. a "split
 * bottom" drop onto the zone the pane already sits alone below would only
 * rebuild the same arrangement under a fresh zone id.
 */
export function movePane(
  root: LayoutNode,
  paneId: string,
  target: { groupId: string; pos: DropPosition; before?: null | string }
): LayoutNode {
  const from = findGroupOfPane(root, paneId)

  // No-op guards: dropping a pane onto its own single-pane group.
  if (from && from.id === target.groupId && from.panes.length === 1) {
    return root
  }

  const without = removePane(root, paneId)

  if (!without) {
    // The pane was the only thing in the tree.
    return root
  }

  if (!findGroup(without, target.groupId)) {
    return root
  }

  const next = insertAtGroup(without, target.groupId, paneId, target.pos, target.before) ?? root

  return shapeSignature(next) === shapeSignature(root) ? root : next
}

/**
 * Move a SELECTION of panes together (multi-tab drag), preserving their strip
 * order. The lead pane lands exactly like a single `movePane` (center joins at
 * `before`, an edge opens the split); the rest stack in behind it. `activeId`
 * (the pressed tab) fronts in the landing group. Same no-op guard as
 * `movePane`: a drop that rebuilds the visible arrangement returns `root`.
 */
export function movePanes(
  root: LayoutNode,
  paneIds: readonly string[],
  target: { groupId: string; pos: DropPosition; before?: null | string },
  activeId: string = paneIds[0] ?? ''
): LayoutNode {
  if (paneIds.length <= 1) {
    return paneIds.length === 1 ? movePane(root, paneIds[0], target) : root
  }

  let without: LayoutNode | null = root

  for (const id of paneIds) {
    without = without && removePane(without, id)
  }

  // The selection was the whole tree, or removal dissolved the target zone
  // (the selection was its only occupancy) — nowhere left to land.
  if (!without || !findGroup(without, target.groupId)) {
    return root
  }

  // The lead insert decides geometry; the rest stack into the lead's group at
  // the same slot (each lands before `before`, so the block keeps its order).
  // Only the lead activates — `insertAtGroup(activate)` would otherwise front
  // each follower in turn.
  const lead = paneIds[0]
  let next: LayoutNode | null = insertAtGroup(without, target.groupId, lead, target.pos, target.before)

  for (let i = 1; next && i < paneIds.length; i++) {
    const leadGroup = findGroupOfPane(next, lead)

    if (!leadGroup) {
      return root
    }

    const before = target.pos === 'center' ? (target.before ?? null) : null
    next = insertAtGroup(next, leadGroup.id, paneIds[i], 'center', before, false)
  }

  if (!next) {
    return root
  }

  const landed = findGroupOfPane(next, lead)

  if (landed && landed.panes.includes(activeId)) {
    next = setActivePane(next, landed.id, activeId)
  }

  return shapeSignature(next) === shapeSignature(root) ? root : next
}

/** Group ids of every leaf under a node, in tree order. */
export function groupLeafIds(node: LayoutNode): string[] {
  return node.type === 'group' ? [node.id] : node.children.flatMap(groupLeafIds)
}

function sameSet(ids: string[], set: Set<string>): boolean {
  return ids.length === set.size && ids.every(id => set.has(id))
}

/** The node whose complete leaf set equals `set` (a rectangular region in a
 *  guillotine tree is always exactly one subtree), or null. */
function findCover(node: LayoutNode, set: Set<string>): LayoutNode | null {
  if (sameSet(groupLeafIds(node), set)) {
    return node
  }

  if (node.type === 'split') {
    for (const child of node.children) {
      const hit = findCover(child, set)

      if (hit) {
        return hit
      }
    }
  }

  return null
}

/**
 * FancyZones span: merge the highlighted zones into ONE group holding the
 * dragged pane block (one pane, or a multi-tab selection in strip order),
 * absorbing any panes that lived in those zones as tabs. Only works when the
 * highlighted set forms a rectangular subtree (it always does for a combined
 * zone range on a guillotine tree); returns null otherwise so the caller can
 * fall back to a single-zone drop.
 */
export function mergeZonesWithPane(
  root: LayoutNode,
  groupIds: string[],
  paneId: string | readonly string[]
): LayoutNode | null {
  const paneIds = typeof paneId === 'string' ? [paneId] : [...paneId]
  const set = new Set(groupIds)

  if (set.size <= 1 || !findCover(root, set)) {
    return null
  }

  // Panes from the merged zones (tree order), minus the dragged block.
  const panesInSet: string[] = []

  const collect = (n: LayoutNode) => {
    if (n.type === 'group') {
      if (set.has(n.id)) {
        panesInSet.push(...n.panes.filter(p => !paneIds.includes(p)))
      }
    } else {
      n.children.forEach(collect)
    }
  }

  collect(root)

  // Any dragged pane living OUTSIDE the merged set is pulled from its origin
  // first (leaving that origin an empty zone). Inside the set it's absorbed.
  let working = root

  for (const id of paneIds) {
    const origin = findGroupOfPane(working, id)

    if (origin && !set.has(origin.id)) {
      working = removePane(working, id) ?? working
    }
  }

  const merged = group([...paneIds, ...panesInSet])

  const replace = (n: LayoutNode): LayoutNode => {
    if (sameSet(groupLeafIds(n), set)) {
      return merged
    }

    return n.type === 'split' ? { ...n, children: n.children.map(replace) } : n
  }

  return normalize(replace(working))
}

// ---------------------------------------------------------------------------
// Attribute edits
// ---------------------------------------------------------------------------

function mapGroups(node: LayoutNode, fn: (g: GroupNode) => GroupNode): LayoutNode {
  return node.type === 'group' ? fn(node) : { ...node, children: node.children.map(c => mapGroups(c, fn)) }
}

export function setActivePane(root: LayoutNode, groupId: string, paneId: string): LayoutNode {
  return mapGroups(root, g => (g.id === groupId && g.panes.includes(paneId) ? { ...g, active: paneId } : g))
}

/** Reorder a block of panes within a group as one unit (browser-tab drag
 *  semantics; a single-tab drag is a one-id block): the block lands at
 *  `toIndex` among the remaining tabs, keeping its own order. */
export function reorderPanesInGroup(
  root: LayoutNode,
  groupId: string,
  paneIds: readonly string[],
  toIndex: number
): LayoutNode {
  return mapGroups(root, g => {
    if (g.id !== groupId || !paneIds.every(p => g.panes.includes(p))) {
      return g
    }

    const without = g.panes.filter(p => !paneIds.includes(p))
    const index = Math.max(0, Math.min(without.length, toIndex))
    const panes = [...without.slice(0, index), ...paneIds, ...without.slice(index)]

    return { ...g, panes }
  })
}

export function setGroupMinimized(root: LayoutNode, groupId: string, minimized: boolean): LayoutNode {
  return mapGroups(root, g => (g.id === groupId ? { ...g, minimized } : g))
}

export function setGroupHeaderHidden(root: LayoutNode, groupId: string, headerHidden: boolean): LayoutNode {
  return mapGroups(root, g => (g.id === groupId ? { ...g, headerHidden } : g))
}

function replaceNode(node: LayoutNode, id: string, make: (g: GroupNode) => LayoutNode): LayoutNode {
  if (node.type === 'group') {
    return node.id === id ? make(node) : node
  }

  return { ...node, children: node.children.map(c => replaceNode(c, id, make)) }
}

/** Mirror the layout HORIZONTALLY (the titlebar flip toggle / ⌘\): reverse
 *  every ROW split's child order at EVERY depth, so left↔right flips
 *  everywhere. A right rail lands on the left with its OWN internal order
 *  mirrored too — so preview stays directly beside the file tree instead of
 *  jumping to the far edge (a shallow root-only reverse left nested rails in
 *  place). COLUMN splits keep their top↔bottom order (the terminal stays at
 *  the bottom). Its own involution: flipping twice is the identity. */
export function mirrorTreeHorizontal(root: LayoutNode): LayoutNode {
  if (root.type === 'group') {
    return root
  }

  const children = root.children.map(mirrorTreeHorizontal)

  return root.orientation === 'row'
    ? { ...root, children: children.reverse(), weights: [...root.weights].reverse() }
    : { ...root, children }
}

export function setSplitWeights(root: LayoutNode, splitId: string, weights: number[]): LayoutNode {
  if (root.type === 'split') {
    if (root.id === splitId) {
      return { ...root, weights }
    }

    return { ...root, children: root.children.map(c => setSplitWeights(c, splitId, weights)) }
  }

  return root
}

// ---------------------------------------------------------------------------
// Validation (persisted trees are untrusted)
// ---------------------------------------------------------------------------

export function isLayoutNode(value: unknown): value is LayoutNode {
  if (!value || typeof value !== 'object') {
    return false
  }

  const n = value as Record<string, unknown>

  if (n.type === 'group') {
    return (
      typeof n.id === 'string' &&
      Array.isArray(n.panes) &&
      n.panes.every(p => typeof p === 'string') &&
      typeof n.active === 'string'
    )
  }

  if (n.type === 'split') {
    return (
      typeof n.id === 'string' &&
      (n.orientation === 'row' || n.orientation === 'column') &&
      Array.isArray(n.children) &&
      n.children.length > 0 &&
      n.children.every(isLayoutNode) &&
      Array.isArray(n.weights) &&
      n.weights.length === n.children.length &&
      n.weights.every(w => typeof w === 'number' && Number.isFinite(w) && w > 0)
    )
  }

  return false
}
