From ff3832a6594cf50aeba6e2e43bd588ad6bd8fcdc Mon Sep 17 00:00:00 2001
From: Joshen Lim <joshenlimek@gmail.com>
Date: Thu, 8 May 2025 15:35:08 +0800
Subject: [PATCH] Fix query block read only catch (#35543)

* Fix query block read only catch

* Update apps/studio/components/ui/QueryBlock/QueryBlock.tsx

Co-authored-by: Alaister Young <alaister@users.noreply.github.com>

---------

Co-authored-by: Alaister Young <alaister@users.noreply.github.com>
---
 apps/studio/components/ui/QueryBlock/QueryBlock.tsx | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/apps/studio/components/ui/QueryBlock/QueryBlock.tsx b/apps/studio/components/ui/QueryBlock/QueryBlock.tsx
index b2d5fa639c4..99a885bec36 100644
--- a/apps/studio/components/ui/QueryBlock/QueryBlock.tsx
+++ b/apps/studio/components/ui/QueryBlock/QueryBlock.tsx
@@ -167,9 +167,14 @@ export const QueryBlock = ({
     onSuccess: (data) => {
       onResults?.(data.result)
       setQueryResult(data.result)
+
+      setReadOnlyError(false)
+      setQueryError(undefined)
     },
     onError: (error) => {
-      if (error?.message.includes('permission denied')) {
+      const permissionDenied = error.message.includes('permission denied')
+      const notOwner = error.message.includes('must be owner')
+      if (permissionDenied || notOwner) {
         setReadOnlyError(true)
         if (showRunButtonIfNotReadOnly) setShowWarning('hasWriteOperation')
       } else {