Briven/apps/studio/components/layouts/SQLEditorLayout/SQLEditorNavV2/UnshareSnippetModal.tsx

import { useParams } from 'common'
import { EyeOffIcon } from 'lucide-react'
import { toast } from 'sonner'
import ConfirmationModal from 'ui-patterns/Dialogs/ConfirmationModal'

import { getContentById } from '@/data/content/content-id-query'
import { useContentUpsertMutation } from '@/data/content/content-upsert-mutation'
import { Snippet } from '@/data/content/sql-folders-query'
import { useSqlEditorV2StateSnapshot } from '@/state/sql-editor-v2'
import type { SqlSnippets } from '@/types'

export const UnshareSnippetModal = ({
  snippet,
  onClose,
  onSuccess,
}: {
  snippet?: Snippet
  onClose: () => void
  onSuccess?: () => void
}) => {
  const { ref: projectRef } = useParams()
  const snapV2 = useSqlEditorV2StateSnapshot()

  const { mutate: upsertContent } = useContentUpsertMutation({
    onError: (error) => {
      toast.error(`Failed to update query: ${error.message}`)
    },
  })

  const onUnshareSnippet = async () => {
    if (!projectRef) return console.error('Project ref is required')
    if (!snippet) return console.error('Snippet ID is required')

    const storeSnippet = snapV2.snippets[snippet.id]
    let snippetContent = storeSnippet?.snippet?.content

    if (snippetContent === undefined) {
      const { content } = await getContentById({ projectRef, id: snippet.id })
      snippetContent = content as unknown as SqlSnippets.Content
    }

    // [Joshen] Just as a final check - to ensure that the content is minimally there (empty string is fine)
    if (snippetContent === undefined) {
      return toast.error('Unable to update snippet visibility: Content is missing')
    }

    upsertContent(
      {
        projectRef,
        payload: {
          ...snippet,
          visibility: 'user',
          folder_id: null,
          content: snippetContent,
        },
      },
      {
        onSuccess: () => {
          snapV2.updateSnippet({
            id: snippet.id,
            snippet: { visibility: 'user', folder_id: null },
            skipSave: true,
          })
          toast.success('Snippet is now unshared from the project')
          onSuccess?.()
          onClose()
        },
      }
    )
  }

  return (
    <ConfirmationModal
      size="medium"
      title={`Confirm to unshare query: ${snippet?.name}`}
      confirmLabel="Unshare query"
      confirmLabelLoading="Unsharing query"
      visible={snippet !== undefined}
      onCancel={() => onClose()}
      onConfirm={() => onUnshareSnippet()}
      alert={{
        title: 'This SQL query will no longer be public to all team members',
        description: 'Only you will have access to this query',
      }}
    >
      <ul
        data-testid="confirm-unshare-snippet-modal"
        className="text-sm text-foreground-light space-y-5"
      >
        <li className="flex gap-3">
          <EyeOffIcon />
          <span>Project members will no longer be able to view this query.</span>
        </li>
      </ul>
    </ConfirmationModal>
  )
}