From 96d74783e06ed5bedb79588f404421409392c5d2 Mon Sep 17 00:00:00 2001
From: Brian Cao <brian@umami.is>
Date: Sun, 13 Aug 2023 22:32:25 -0700
Subject: [PATCH] Edit button states.

---
 components/pages/reports/ReportsTable.js | 17 ++++++++++++-----
 queries/admin/report.ts                  |  1 +
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/components/pages/reports/ReportsTable.js b/components/pages/reports/ReportsTable.js
index e59e4069..39a35c96 100644
--- a/components/pages/reports/ReportsTable.js
+++ b/components/pages/reports/ReportsTable.js
@@ -1,9 +1,10 @@
-import { useState } from 'react';
-import { Flexbox, Icon, Icons, Text, Button, Modal } from 'react-basics';
+import ConfirmDeleteForm from 'components/common/ConfirmDeleteForm';
 import LinkButton from 'components/common/LinkButton';
 import SettingsTable from 'components/common/SettingsTable';
-import ConfirmDeleteForm from 'components/common/ConfirmDeleteForm';
 import { useMessages } from 'hooks';
+import useUser from 'hooks/useUser';
+import { useState } from 'react';
+import { Button, Flexbox, Icon, Icons, Modal, Text } from 'react-basics';
 
 export function ReportsTable({
   data = [],
@@ -16,6 +17,7 @@ export function ReportsTable({
 }) {
   const [report, setReport] = useState(null);
   const { formatMessage, labels } = useMessages();
+  const { user } = useUser();
 
   const domainColumn = [
     {
@@ -49,14 +51,19 @@ export function ReportsTable({
         filterValue={filterValue}
       >
         {row => {
-          const { id } = row;
+          const {
+            id,
+            userId: reportOwnerId,
+            website: { domain, userId: websiteOwnerId },
+          } = row;
           if (showDomain) {
-            row.domain = row.website.domain;
+            row.domain = domain;
           }
 
           return (
             <Flexbox gap={10}>
               <LinkButton href={`/reports/${id}`}>{formatMessage(labels.view)}</LinkButton>
+              {!showDomain || user.id === reportOwnerId || user.id === websiteOwnerId}
               <Button onClick={() => setReport(row)}>
                 <Icon>
                   <Icons.Trash />
diff --git a/queries/admin/report.ts b/queries/admin/report.ts
index 3c50c2cb..3c3e3e2d 100644
--- a/queries/admin/report.ts
+++ b/queries/admin/report.ts
@@ -162,6 +162,7 @@ export async function getReportsByUserId(
         website: {
           select: {
             domain: true,
+            userId: true,
           },
         },
       },