Notebook Editor

// src/components/sso-spn-notebook-editor-iframe.tsx
import { redirect } from "next/navigation";
import { getDatabricksWorkspaceToken } from "@/lib/databricks-workspace-token";
import { generateProxyUrl } from "@/lib/token-encryption";

export default async function NotebookEditorIframe() {
  // Fetch workspace token using SPN credentials
  const tokenResult = await getDatabricksWorkspaceToken();

  if (!tokenResult.success) {
    redirect("/sso-spn"); // Redirect to re-authenticate
  }

  const { accessToken } = tokenResult.data;

  // Configuration from environment
  const appUrl = process.env.DATABRICKS_NOTEBOOK_APP_URL;
  const proxyBaseUrl = process.env.NEXT_PUBLIC_PROXY_URL;

  // Generate encrypted proxy URL
  const proxyPath = generateProxyUrl(accessToken, appUrl, "/");
  const fullProxyUrl = `${proxyBaseUrl}${proxyPath}`;

  return (
    <div className="h-full flex flex-col">
      <iframe
        src={fullProxyUrl}
        className="w-full h-full border-0"
        title="Notebook Editor"
        sandbox="allow-scripts allow-same-origin allow-forms allow-popups allow-modals allow-downloads"
        allow="clipboard-write; clipboard-read"
      />
    </div>
  );
}