open-swe/apps/web/src/components/gen-ui/pull-request-opened.tsx

"use client";

import { useState } from "react";
import {
  GitPullRequest,
  Loader2,
  CheckCircle,
  ChevronDown,
  ChevronUp,
  ExternalLink,
} from "lucide-react";

type PullRequestOpenedProps = {
  status: "loading" | "generating" | "done";
  title?: string;
  description?: string;
  url?: string;
  prNumber?: number;
  branch?: string;
  targetBranch?: string;
};

export function PullRequestOpened({
  status,
  title,
  description,
  url,
  prNumber,
  branch,
  targetBranch = "main",
}: PullRequestOpenedProps) {
  const [expanded, setExpanded] = useState(false);

  const getStatusIcon = () => {
    switch (status) {
      case "loading":
        return (
          <div className="h-3.5 w-3.5 rounded-full border border-gray-300" />
        );
      case "generating":
        return <Loader2 className="h-3.5 w-3.5 animate-spin text-gray-500" />;
      case "done":
        return <CheckCircle className="h-3.5 w-3.5 text-green-500" />;
    }
  };

  const getStatusText = () => {
    switch (status) {
      case "loading":
        return "Preparing pull request...";
      case "generating":
        return "Opening pull request...";
      case "done":
        return prNumber
          ? `Pull request #${prNumber} opened`
          : "Pull request opened";
    }
  };

  const shouldShowToggle = () => {
    return status === "done" && description;
  };

  return (
    <div className="overflow-hidden rounded-md border border-gray-200">
      <div className="flex items-center border-b border-gray-200 bg-gray-50 p-2">
        <GitPullRequest className="mr-2 h-3.5 w-3.5 text-gray-500" />
        <div className="flex-1">
          {title && status === "done" && (
            <div className="mb-0.5 text-xs font-normal text-gray-800">
              {title}
            </div>
          )}
          {branch && status === "done" && (
            <div className="text-xs font-normal text-gray-500">
              {branch} → {targetBranch}
            </div>
          )}
          {!title && (
            <span className="text-xs font-normal text-gray-800">
              {getStatusText()}
            </span>
          )}
        </div>
        <div className="flex items-center gap-2">
          <span className="text-xs font-normal text-gray-500">
            {getStatusText()}
          </span>
          {getStatusIcon()}
          {url && status === "done" && (
            <a
              href={url}
              target="_blank"
              rel="noopener noreferrer"
              className="text-gray-500 hover:text-gray-700"
              title="Open pull request"
            >
              <ExternalLink className="h-3.5 w-3.5" />
            </a>
          )}
          {shouldShowToggle() && (
            <button
              onClick={() => setExpanded(!expanded)}
              className="text-gray-500 hover:text-gray-700"
            >
              {expanded ? (
                <ChevronUp className="h-3.5 w-3.5" />
              ) : (
                <ChevronDown className="h-3.5 w-3.5" />
              )}
            </button>
          )}
        </div>
      </div>

      {expanded && description && status === "done" && (
        <div className="border-t border-gray-200 p-2">
          <h3 className="mb-1 text-xs font-normal text-gray-500">
            Description
          </h3>
          <div className="text-xs font-normal whitespace-pre-wrap text-gray-800">
            {description}
          </div>
        </div>
      )}
    </div>
  );
}