SQL Editor

// Execute SQL statement
const executeQuery = async (sql: string, warehouseId: string) => {
  // Submit query (returns immediately with statement_id)
  const response = await fetch("/api/databricks/sql/execute", {
    method: "POST",
    headers: { "Content-Type": "application/json" },
    body: JSON.stringify({
      warehouse_id: warehouseId,
      statement: sql,
      wait_timeout: "0s", // Return immediately, poll for results
    }),
  });

  const { statement_id } = await response.json();

  // Poll for results
  const pollResults = async (): Promise<QueryResult> => {
    const status = await fetch(`/api/databricks/sql/status/${statement_id}`);
    const data = await status.json();

    switch (data.status.state) {
      case "SUCCEEDED":
        return {
          columns: data.manifest.schema.columns,
          rows: data.result.data_array,
          rowCount: data.manifest.total_row_count,
        };
      case "FAILED":
        throw new Error(data.status.error.message);
      case "RUNNING":
      case "PENDING":
        await new Promise(r => setTimeout(r, 1000));
        return pollResults();
      default:
        throw new Error(`Unknown state: ${data.status.state}`);
    }
  };

  return pollResults();
};

// Warehouse selector with status and controls
function WarehouseSelector({ onSelect }) {
  const { data: warehouses } = useQuery({
    queryKey: ["warehouses"],
    queryFn: () => fetch("/api/databricks/warehouses").then(r => r.json()),
    refetchInterval: 10000, // Poll every 10 seconds for status updates
  });

  const startMutation = useMutation({
    mutationFn: (id: string) =>
      fetch(`/api/databricks/warehouses/${id}/start`, { method: "POST" }),
    onSuccess: () => queryClient.invalidateQueries(["warehouses"]),
  });

  return (
    <Select onValueChange={onSelect}>
      {warehouses?.map(wh => (
        <SelectItem key={wh.id} value={wh.id}>
          <span>{wh.name}</span>
          <Badge variant={wh.state === "RUNNING" ? "success" : "secondary"}>
            {wh.state}
          </Badge>
          {wh.state === "STOPPED" && (
            <Button size="sm" onClick={() => startMutation.mutate(wh.id)}>
              Start
            </Button>
          )}
        </SelectItem>
      ))}
    </Select>
  );
}