From 76178ace79ad8d96ec8fbe8efe37445d3db8844b Mon Sep 17 00:00:00 2001
From: Zeu Capua <coding@zeu.dev>
Date: Wed, 21 Aug 2024 10:44:40 -0700
Subject: [PATCH] wrap OssfChart with CopyContainer, don't show copy button if
 error

---
 components/shared/CopyContainer.tsx |  3 ++
 pages/s/[org]/[repo]/index.tsx      | 48 +++++++++++++++++++++++------
 2 files changed, 42 insertions(+), 9 deletions(-)

diff --git a/components/shared/CopyContainer.tsx b/components/shared/CopyContainer.tsx
index 0c8524524..eb61fa3eb 100644
--- a/components/shared/CopyContainer.tsx
+++ b/components/shared/CopyContainer.tsx
@@ -9,6 +9,7 @@ interface CopyContainerProps {
   copySuccessMessage?: string;
   copyErrorMessage?: string;
   options?: Partial<Options>;
+  onCopyClick?: () => void;
   children: React.ReactNode;
 }
 
@@ -16,6 +17,7 @@ export const CopyContainer = ({
   copySuccessMessage = "Copied image to clipboard",
   copyErrorMessage = "Error copying image",
   options,
+  onCopyClick,
   children,
 }: CopyContainerProps) => {
   const { toast } = useToast();
@@ -27,6 +29,7 @@ export const CopyContainer = ({
       <button
         className="absolute top-2 right-2 p-2 bg-white rounded-md shadow-md opacity-0 group-hover:opacity-100 focus-within:opacity-100 scale-95 group-hover:scale-100 transition-all duration-200 ease-in-out transform"
         onClick={async () => {
+          onCopyClick?.();
           const node = copyRef.current;
 
           try {
diff --git a/pages/s/[org]/[repo]/index.tsx b/pages/s/[org]/[repo]/index.tsx
index 39556d7c2..8b3715500 100644
--- a/pages/s/[org]/[repo]/index.tsx
+++ b/pages/s/[org]/[repo]/index.tsx
@@ -370,15 +370,45 @@ export default function RepoPage({ repoData, ogImageUrl }: RepoPageProps) {
                         })
                       }
                     />
-                    <OssfChart
-                      repository={repoData.full_name}
-                      totalScore={repoData.ossf_scorecard_total_score}
-                      dependencyUpdateScore={repoData.ossf_scorecard_dependency_update_score}
-                      maintainedScore={repoData.ossf_scorecard_maintained_score}
-                      fuzzingScore={repoData.ossf_scorecard_fuzzing_score}
-                      isLoading={false}
-                      isError={!repoData.ossf_scorecard_total_score}
-                    />
+                    {!repoData.ossf_scorecard_total_score ? (
+                      <OssfChart
+                        repository={repoData.full_name}
+                        totalScore={repoData.ossf_scorecard_total_score}
+                        dependencyUpdateScore={repoData.ossf_scorecard_dependency_update_score}
+                        maintainedScore={repoData.ossf_scorecard_maintained_score}
+                        fuzzingScore={repoData.ossf_scorecard_fuzzing_score}
+                        isLoading={false}
+                        isError={!repoData.ossf_scorecard_total_score}
+                      />
+                    ) : (
+                      <CopyContainer
+                        onCopyClick={() => {
+                          posthog.capture("Repo Pages: copied OSSF Scorecard", {
+                            repository: repoData.full_name,
+                          });
+                        }}
+                        options={{
+                          windowWidth: 1700,
+                          allowTaint: true,
+                          onclone: (document, element) => {
+                            element.querySelectorAll("[data-copy-image-branding]").forEach((el) => {
+                              el.classList.remove("hidden");
+                            });
+                          },
+                        }}
+                      >
+                        <CopyImageBranding repository={repoData.full_name} />
+                        <OssfChart
+                          repository={repoData.full_name}
+                          totalScore={repoData.ossf_scorecard_total_score}
+                          dependencyUpdateScore={repoData.ossf_scorecard_dependency_update_score}
+                          maintainedScore={repoData.ossf_scorecard_maintained_score}
+                          fuzzingScore={repoData.ossf_scorecard_fuzzing_score}
+                          isLoading={false}
+                          isError={!repoData.ossf_scorecard_total_score}
+                        />
+                      </CopyContainer>
+                    )}
                   </div>
 
                   <RossChart