/* Copyright 2026 Marimo. All rights reserved. */

import { z } from "zod";

const DetailsSchema = z.object({
  detail: z.string(),
});

const ErrorSchema = z.object({
  error: z.string(),
});

export function prettyError(error: unknown): string {
  if (!error) {
    return "Unknown error";
  }
  if (error instanceof Error) {
    const details = DetailsSchema.safeParse(error.cause);
    if (details.success) {
      return details.data.detail;
    }
    return maybeExtractDetails(error.message);
  }
  if (typeof error === "object") {
    const details = DetailsSchema.safeParse(error);
    if (details.success) {
      return details.data.detail;
    }
    const parsed = ErrorSchema.safeParse(error);
    if (parsed.success) {
      return parsed.data.error;
    }
  }
  try {
    return JSON.stringify(error);
  } catch {
    return String(error);
  }
}

function maybeExtractDetails(message: string): string {
  const parsed = safeJSONParse(message);
  if (!parsed) {
    return message;
  }
  const details = DetailsSchema.safeParse(parsed);
  if (details.success) {
    return details.data.detail;
  }
  const error = ErrorSchema.safeParse(parsed);
  if (error.success) {
    return error.data.error;
  }
  return message;
}

function safeJSONParse(message: string): unknown {
  try {
    return JSON.parse(message);
  } catch {
    return message;
  }
}

export class CellNotInitializedError extends Error {
  constructor(
    message = "The cell containing this UI element has not been run yet. Please run the cell first.",
  ) {
    super(message);
    this.name = "CellNotInitializedError";
  }
}

export class NoKernelConnectedError extends Error {
  constructor(message = "Not yet connected to a kernel.") {
    super(message);
    this.name = "NoKernelConnectedError";
  }
}