logic.ts

import { kea } from 'kea'
import api from 'lib/api'
import { toParams, deleteWithUndo } from 'lib/utils'
import { annotationsModel } from '~/models/annotationsModel'
import { annotationsTableLogicType } from './logicType'
import { AnnotationType } from '~/types'
import { teamLogic } from '../teamLogic'

export const annotationsTableLogic = kea<annotationsTableLogicType>({
    path: ['scenes', 'annotations', 'logic'],
    loaders: ({ actions }) => ({
        annotations: {
            __default: [],
            loadAnnotations: async () => {
                const response = await api.get(
                    `api/projects/${teamLogic.values.currentTeamId}/annotations/?${toParams({ order: '-updated_at' })}`
                )
                actions.setNext(response.next)
                return response.results
            },
        },
    }),
    reducers: () => ({
        annotations: [
            [] as AnnotationType[],
            {
                appendAnnotations: (state, { annotations }) => [...state, ...annotations],
            },
        ],
        next: [
            null as string | null,
            {
                setNext: (_, { next }) => next,
            },
        ],
        loadingNext: [
            false,
            {
                loadAnnotationsNext: () => true,
                appendAnnotations: () => false,
            },
        ],
    }),
    actions: () => ({
        updateAnnotation: (id, content) => ({ id, content }),
        deleteAnnotation: (id) => ({ id }),
        restoreAnnotation: (id) => ({ id }),
        loadAnnotationsNext: () => true,
        setNext: (next) => ({ next }),
        appendAnnotations: (annotations: AnnotationType[]) => ({ annotations }),
    }),
    listeners: ({ actions, values }) => ({
        updateAnnotation: async ({ id, content }) => {
            await api.update(`api/projects/${teamLogic.values.currentTeamId}/annotations/${id}`, { content })
            actions.loadAnnotations()
        },
        restoreAnnotation: async ({ id }) => {
            await api.update(`api/projects/${teamLogic.values.currentTeamId}/annotations/${id}`, { deleted: false })
            actions.loadAnnotations()
        },
        deleteAnnotation: ({ id }) => {
            deleteWithUndo({
                endpoint: `projects/${teamLogic.values.currentTeamId}/annotations`,
                object: { name: 'Annotation', id },
                callback: () => actions.loadAnnotations(),
            })
        },
        loadAnnotationsNext: async () => {
            let results: AnnotationType[] = []
            if (values.next) {
                const response = await api.get(values.next)
                actions.setNext(response.next)
                results = response.results
            }
            actions.appendAnnotations(results)
        },
        [annotationsModel.actionTypes.createGlobalAnnotationSuccess]: () => {
            actions.loadAnnotations()
        },
    }),
    events: ({ actions }) => ({
        afterMount: () => actions.loadAnnotations(),
    }),
})