chore: Migration checkInQuestion to tiptap format (#10542)

Signed-off-by: Matt Krick <[email protected]>

packages/server/postgres/migrations/2024-12-05T13:40:14.859Z_checkinToTipTap.ts

@@ -0,0 +1,138 @@
+import {mergeAttributes} from '@tiptap/core'
+import BaseLink from '@tiptap/extension-link'
+import Mention from '@tiptap/extension-mention'
+import {generateJSON} from '@tiptap/html'
+import StarterKit from '@tiptap/starter-kit'
+import {convertFromRaw, RawDraftContentState} from 'draft-js'
+import {Options, stateToHTML} from 'draft-js-export-html'
+import type {Kysely} from 'kysely'
+
+export const serverTipTapExtensions = [
+  StarterKit,
+  Mention.configure({
+    renderText({node}) {
+      return node.attrs.label
+    },
+    renderHTML({options, node}) {
+      return ['span', options.HTMLAttributes, `${node.attrs.label ?? node.attrs.id}`]
+    }
+  }),
+  Mention.extend({name: 'taskTag'}).configure({
+    renderHTML({options, node}) {
+      return ['span', options.HTMLAttributes, `#${node.attrs.id}`]
+    }
+  }),
+  BaseLink.extend({
+    parseHTML() {
+      return [{tag: 'a[href]:not([data-type="button"]):not([href *= "javascript:" i])'}]
+    },
+
+    renderHTML({HTMLAttributes}) {
+      return ['a', mergeAttributes(this.options.HTMLAttributes, HTMLAttributes, {class: 'link'}), 0]
+    }
+  })
+]
+
+const getNameFromEntity = (content: RawDraftContentState, userId: string) => {
+  const {blocks, entityMap} = content
+  const entityKey = Number(
+    Object.keys(entityMap).find((key) => entityMap[key]!.data?.userId === userId)
+  )
+  for (let i = 0; i < blocks.length; i++) {
+    const block = blocks[i]!
+    const {entityRanges, text} = block
+    const entityRange = entityRanges.find((range) => range.key === entityKey)
+    if (!entityRange) continue
+    const {length, offset} = entityRange
+    return text.slice(offset, offset + length)
+  }
+  console.log('found unknown for', userId, JSON.stringify(content))
+  return 'Unknown User'
+}
+
+export const convertKnownDraftToTipTap = (content: RawDraftContentState) => {
+  const contentState = convertFromRaw(content)
+  const options: Options = {
+    entityStyleFn: (entity) => {
+      const entityType = entity.getType().toLowerCase()
+      const data = entity.getData()
+      if (entityType === 'tag') {
+        return {
+          element: 'span',
+          attributes: {
+            'data-id': data.value,
+            'data-type': 'taskTag'
+          }
+        }
+      }
+      if (entityType === 'mention') {
+        const label = getNameFromEntity(content, data.userId)
+        return {
+          element: 'span',
+          attributes: {
+            'data-id': data.userId.toWellFormed(),
+            'data-label': label.toWellFormed(),
+            'data-type': 'mention'
+          }
+        }
+      }
+      return
+    }
+  }
+  const html = stateToHTML(contentState, options)
+  const json = generateJSON(html, serverTipTapExtensions)
+  return json
+}
+
+export async function up(db: Kysely<any>): Promise<void> {
+  let lastId = ''
+
+  for (let i = 0; i < 1e6; i++) {
+    const meetings = await db
+      .selectFrom('NewMeeting')
+      .select(['id', 'phases'])
+      .where('id', '>', lastId)
+      .orderBy('id asc')
+      .limit(1000)
+      .execute()
+    console.log('converting meetings', i * 1000)
+    if (meetings.length === 0) break
+    const updatePromises = [] as Promise<any>[]
+    for (const meeting of meetings) {
+      const {id, phases} = meeting
+      const [firstPhase] = phases
+      const checkInQuestion = firstPhase?.checkInQuestion
+      if (!checkInQuestion) continue
+      let content
+      try {
+        content = JSON.parse(checkInQuestion)
+      } catch (e) {
+        console.log('BAD QUESTION, SKIPPING', id, checkInQuestion)
+        continue
+      }
+      if (!('blocks' in content)) continue
+      // this is draftjs
+      const tipTapContent = convertKnownDraftToTipTap(content)
+      firstPhase.checkInQuestion = JSON.stringify(tipTapContent)
+      const doPromise = async () => {
+        try {
+          return await db
+            .updateTable('NewMeeting')
+            .set({phases: JSON.stringify(phases)})
+            .where('id', '=', id)
+            .execute()
+        } catch (e) {
+          console.log('GOT ERR', id, JSON.stringify(tipTapContent), e)
+          throw e
+        }
+      }
+      updatePromises.push(doPromise())
+    }
+    await Promise.all(updatePromises)
+    lastId = meetings.at(-1)!.id
+  }
+}
+
+export async function down(): Promise<void> {
+  // noop
+}