import { graphql } from '~/lib/common/generated/gql/gql'
import { useQuery } from '@vue/apollo-composable'
import { workspaceLimitsQuery } from '~/lib/workspaces/graphql/queries'
import { WorkspacePlanConfigs } from '@speckle/shared'

graphql(`
  fragment WorkspacePlanLimits_Workspace on Workspace {
    id
    slug
    plan {
      name
    }
  }
`)

export const useWorkspaceLimits = (slug: string) => {
  const { result } = useQuery(
    workspaceLimitsQuery,
    () => ({
      slug
    }),
    () => ({
      enabled: !!slug
    })
  )

  // Plan limits
  const limits = computed(() => {
    const planName = result.value?.workspaceBySlug?.plan?.name
    if (!planName)
      return {
        projectCount: 0,
        modelCount: 0,
        versionsHistory: null,
        commentHistory: null
      }

    const planConfig = WorkspacePlanConfigs[planName]
    return planConfig?.limits
  })

  const versionLimitFormatted = computed(() => {
    const versionsHistory = limits.value?.versionsHistory
    if (!versionsHistory) return 'Unlimited'

    const { value, unit } = versionsHistory
    return `${value} ${unit}`
  })

  const commentLimitFormatted = computed(() => {
    const commentHistory = limits.value?.commentHistory
    if (!commentHistory) return 'Unlimited'

    const { value, unit } = commentHistory
    return `${value} ${unit}`
  })

  return {
    limits,
    versionLimitFormatted,
    commentLimitFormatted
  }
}