speckle-server/packages/server/test/graphql/comments.ts

import type {
  CreateCommentMutation,
  CreateCommentMutationVariables,
  CreateReplyMutation,
  CreateReplyMutationVariables,
  GetCommentQuery,
  GetCommentQueryVariables,
  GetCommentsQuery,
  GetCommentsQueryVariables
} from '@/modules/core/graph/generated/graphql'
import type { ExecuteOperationServer } from '@/test/graphqlHelper'
import { executeOperation } from '@/test/graphqlHelper'
import { gql } from 'graphql-tag'

const commentWithRepliesFragment = gql`
  fragment CommentWithReplies on Comment {
    id
    rawText
    text {
      doc
      attachments {
        id
        fileName
        streamId
      }
    }
    replies(limit: 10) {
      items {
        id
        text {
          doc
          attachments {
            id
            fileName
            streamId
          }
        }
      }
    }
  }
`

const createCommentMutation = gql`
  mutation CreateComment($input: CommentCreateInput!) {
    commentCreate(input: $input)
  }
`

const createReplyMutation = gql`
  mutation CreateReply($input: ReplyCreateInput!) {
    commentReply(input: $input)
  }
`

const getCommentQuery = gql`
  query GetComment($id: String!, $streamId: String!) {
    comment(id: $id, streamId: $streamId) {
      ...CommentWithReplies
    }
  }

  ${commentWithRepliesFragment}
`

const getCommentsQuery = gql`
  query GetComments($streamId: String!, $cursor: String) {
    comments(streamId: $streamId, limit: 10, cursor: $cursor) {
      totalCount
      cursor
      items {
        ...CommentWithReplies
      }
    }
  }

  ${commentWithRepliesFragment}
`

export const createComment = (
  apollo: ExecuteOperationServer,
  variables: CreateCommentMutationVariables
) =>
  executeOperation<CreateCommentMutation, CreateCommentMutationVariables>(
    apollo,
    createCommentMutation,
    variables
  )

export const createReply = (
  apollo: ExecuteOperationServer,
  variables: CreateReplyMutationVariables
) =>
  executeOperation<CreateReplyMutation, CreateReplyMutationVariables>(
    apollo,
    createReplyMutation,
    variables
  )

export const getComment = (
  apollo: ExecuteOperationServer,
  variables: GetCommentQueryVariables
) =>
  executeOperation<GetCommentQuery, GetCommentQueryVariables>(
    apollo,
    getCommentQuery,
    variables
  )

export const getComments = (
  apollo: ExecuteOperationServer,
  variables: GetCommentsQueryVariables
) =>
  executeOperation<GetCommentsQuery, GetCommentsQueryVariables>(
    apollo,
    getCommentsQuery,
    variables
  )