fauna.ts

import { Expr } from 'faunadb'
import faunadb from 'faunadb'
import { User } from 'src/pages/members'
const q = faunadb.query

function isFollowingUser({
  session,
  userRef,
}: {
  session: any
  userRef: Expr
}) {
  let isFollowing = false

  if (session) {
    const followerId = (session.user as User).id
    const ref = q.Match(q.Index('unique_user_and_follower'), [
      userRef,
      q.Ref(q.Collection('users'), followerId),
    ])
    isFollowing = q.If(
      q.Exists(ref),
      q.Select(['data', 'isFollowing'], q.Get(ref)),
      false
    ) as boolean
  }
  return isFollowing
}

export function getUserFromUserRef({
  ref,
  session,
}: {
  ref: Expr
  session: any
}) {
  const userDoc = q.Get(ref)
  const userId = q.Select(['ref', 'id'], userDoc)
  const isFollowing = isFollowingUser({ session, userRef: ref })

  return {
    id: userId,
    name: q.Select(['data', 'name'], userDoc, null),
    image: q.Select(['data', 'image'], userDoc, null),
    username: q.Select(['data', 'username'], userDoc, null),
    account: {
      firstName: q.Select(['data', 'account', 'firstName'], userDoc, null),
      bio: q.Select(['data', 'account', 'bio'], userDoc, null),
    },
    socials: q.Select(['data', 'socials'], userDoc, null),
    isFollowing,
  }
}

function hasLikedUpdate({
  session,
  goalUpdateRef,
}: {
  session: any
  goalUpdateRef: Expr
}) {
  let hasLiked = false
  if (session) {
    const userId = (session.user as User).id
    const ref = q.Match(q.Index('unique_update_user_like'), [
      goalUpdateRef,
      q.Ref(q.Collection('users'), userId),
    ])
    hasLiked = q.If(
      q.Exists(ref),
      q.Select(['data', 'liked'], q.Get(ref)),
      false
    ) as boolean
  }
  return hasLiked
}

function hasLikedComment({
  commentRef,
  session,
}: {
  commentRef: Expr
  session: any
}) {
  let hasLiked = false
  if (session) {
    const userId = (session.user as User).id
    const ref = q.Match(q.Index('unique_comment_user_like'), [
      commentRef,
      q.Ref(q.Collection('users'), userId),
    ])
    hasLiked = q.If(
      q.Exists(ref),
      q.Select(['data', 'liked'], q.Get(ref)),
      false
    ) as boolean
  }
  return hasLiked
}

function getAllUsersWhoLikedUpdate({
  goalUpdateRef,
  session,
}: {
  goalUpdateRef: Expr
  session: any
}) {
  return q.Map(
    q.Filter(
      q.Paginate(q.Match(q.Index('all_likes_by_update'), goalUpdateRef)),
      (updateLikeRef) => q.Select(['data', 'liked'], q.Get(updateLikeRef))
    ),
    (likeRef) =>
      getUserFromUserRef({
        ref: q.Select(['data', 'user'], q.Get(likeRef)),
        session,
      })
  )
}

function getAllUsersWhoLikedComment({
  commentRef,
  session,
}: {
  commentRef: Expr
  session
}) {
  return q.Map(
    q.Filter(
      q.Paginate(q.Match(q.Index('all_likes_by_comment'), commentRef)),
      (commentLikeRef) => {
        return q.Select(['data', 'liked'], q.Get(commentLikeRef))
      }
    ),
    (likeRef) =>
      getUserFromUserRef({
        ref: q.Select(['data', 'user'], q.Get(likeRef)),
        session,
      })
  )
}

function getAllCommentsOfUpdate({
  goalUpdateRef,
  session,
}: {
  goalUpdateRef: Expr
  session: any
}) {
  return q.Map(
    q.Paginate(q.Match(q.Index('all_comments_by_update'), goalUpdateRef)),
    (commentRef) => getCommentFromCommentRef({ ref: commentRef, session })
  )
}

export function getUpdateFromUpdateRef({
  ref: goalUpdateRef,
  session,
}: {
  ref: Expr
  session: any
}) {
  const goalUpdateDoc = q.Get(goalUpdateRef)
  const goalDoc = q.Get(q.Select(['data', 'goal'], goalUpdateDoc))
  const postedByDoc = q.Get(q.Select(['data', 'postedBy'], goalUpdateDoc))
  const description = q.Select(['data', 'description'], goalUpdateDoc)
  const createdAt = q.ToMillis(
    q.Select(['data', 'timestamps', 'createdAt'], goalUpdateDoc)
  )
  const updateId = q.Select(['ref', 'id'], goalUpdateDoc)
  const hasLiked = hasLikedUpdate({ goalUpdateRef, session })
  return {
    id: updateId,
    goal: {
      id: q.Select(['ref', 'id'], goalDoc),
      title: q.Select(['data', 'title'], goalDoc),
    },
    comments: getAllCommentsOfUpdate({ goalUpdateRef, session }),
    hasLiked,
    likes: getAllUsersWhoLikedUpdate({ goalUpdateRef, session }),
    description,
    createdAt,
    postedBy: getUserFromUserRef({
      ref: q.Select(['ref'], postedByDoc),
      session,
    }),
  }
}

export function getCommentFromCommentRef({
  ref: commentRef,
  session,
}: {
  ref: Expr
  session: any
}) {
  const commentDoc = q.Get(commentRef)
  const postedByDoc = q.Get(q.Select(['data', 'postedBy'], commentDoc))
  const hasLiked = hasLikedComment({ commentRef, session })
  return {
    id: q.Select(['ref', 'id'], commentDoc),
    updateId: q.Select(['data', 'update', 'id'], commentDoc),
    description: q.Select(['data', 'description'], commentDoc),
    createdAt: q.ToMillis(
      q.Select(['data', 'timestamps', 'createdAt'], commentDoc)
    ),
    hasLiked,
    likes: getAllUsersWhoLikedComment({ commentRef, session }),
    postedBy: getUserFromUserRef({
      ref: q.Select(['ref'], postedByDoc),
      session,
    }),
  }
}