flexor-api/src/lib/collections.ts

import { CosmosClient } from '@azure/cosmos'
import { Logger } from 'fastify'
import uniq from 'lodash/uniq'
import { DatabaseItem } from '../types'

import { containerFor, createQuerySpec, queryItems, getItem } from './database'
import { User, UserSubscription, UserBlock, GroupMembership, UserItemType, GroupItemType } from '../types/collections'

export async function getUsers(client: CosmosClient, ids: string[], logger?: Logger): Promise<User[]> {
    return await queryItems<User>({
        container: containerFor(client, 'Users'),
        query: createQuerySpec(
            `SELECT u.id, u.name, u.imageUrl, u.coverImageUrl, u['group'], u.created FROM Users u WHERE ARRAY_CONTAINS(@ids, u.id)`,
            {
                ids: uniq(ids),
            }
        ),
        logger,
    })
}

export async function getUsersFromItems<T extends DatabaseItem>(client: CosmosClient, items: T[], logger?: Logger): Promise<User[]> {
    return await getUsers(client, items.map(i => i.id), logger)
}

export async function getApprovedSubscriptions(client: CosmosClient, from: string, to: string, logger?: Logger): Promise<UserSubscription[]> {
    return await queryItems<UserSubscription>({
        container: containerFor(client, 'Users'),
        query: createQuerySpec(
            `SELECT u.id FROM Users u WHERE
                u.subscriberId = @to
                u.pk = @from AND
                u.t = @type AND
                u.pending = false`,
            { from, to, type: UserItemType.Subscription }
        ),
        logger,
    })
}

export async function getUserBlocks(client: CosmosClient, from: string, to: string[], logger?: Logger): Promise<UserBlock[]> {
    return await queryItems<UserBlock>({
        container: containerFor(client, 'Users'),
        query: createQuerySpec(
            `SELECT u.id FROM Users u WHERE
                u.pk = @from
                AND u.t = @type
                AND ARRAY_CONTAINS(@to, u.blockedId)`,
            { from, to, type: UserItemType.Block }
        ),
        logger,
    })
}

export async function getUserMembership(client: CosmosClient, userId: string, logger?: Logger): Promise<GroupMembership | undefined> {
    const user = await getItem<User>({
        container: containerFor(client, 'Users'),
        id: userId,
    })

    if (!user) return
    if (!user.group) return

    const memberships = await queryItems<GroupMembership>({
        container: containerFor(client, 'Groups'),
        query: createQuerySpec(
            `SELECT * FROM Groups g WHERE
                g.pk = @pk AND
                g.t = @type AND
                g.userId = @user
            `,
            {
                pk: user.group.id,
                type: GroupItemType.Membership,
                user: user.id,
            }
        ),
        logger,
    })

    if (memberships.length > 0) return memberships[0]
}