150 lines
4 KiB
TypeScript
150 lines
4 KiB
TypeScript
import type { mastodon } from 'masto'
|
|
import type { ComputedRef, Ref } from 'vue'
|
|
import { STORAGE_KEY_DRAFTS } from '~/constants'
|
|
import type { Draft, DraftMap } from '~/types'
|
|
import type { Mutable } from '~/types/utils'
|
|
|
|
export const currentUserDrafts = process.server || process.test ? computed<DraftMap>(() => ({})) : useUserLocalStorage<DraftMap>(STORAGE_KEY_DRAFTS, () => ({}))
|
|
|
|
export const builtinDraftKeys = [
|
|
'dialog',
|
|
'home',
|
|
]
|
|
|
|
export function getDefaultDraft(options: Partial<Mutable<mastodon.v1.CreateStatusParams> & Omit<Draft, 'params'>> = {}): Draft {
|
|
const {
|
|
attachments = [],
|
|
initialText = '',
|
|
status,
|
|
inReplyToId,
|
|
visibility,
|
|
sensitive,
|
|
spoilerText,
|
|
language,
|
|
mentions,
|
|
} = options
|
|
|
|
return {
|
|
attachments,
|
|
initialText,
|
|
params: {
|
|
status: status || '',
|
|
inReplyToId,
|
|
visibility: visibility || 'public',
|
|
sensitive: sensitive ?? false,
|
|
spoilerText: spoilerText || '',
|
|
language: language || '', // auto inferred from current language on posting
|
|
},
|
|
mentions,
|
|
lastUpdated: Date.now(),
|
|
}
|
|
}
|
|
|
|
export async function getDraftFromStatus(status: mastodon.v1.Status): Promise<Draft> {
|
|
return getDefaultDraft({
|
|
status: await convertMastodonHTML(status.content),
|
|
mediaIds: status.mediaAttachments.map(att => att.id),
|
|
visibility: status.visibility,
|
|
attachments: status.mediaAttachments,
|
|
sensitive: status.sensitive,
|
|
spoilerText: status.spoilerText,
|
|
language: status.language,
|
|
})
|
|
}
|
|
|
|
function getAccountsToMention(status: mastodon.v1.Status) {
|
|
const userId = currentUser.value?.account.id
|
|
const accountsToMention = new Set<string>()
|
|
if (status.account.id !== userId)
|
|
accountsToMention.add(status.account.acct)
|
|
status.mentions
|
|
.filter(mention => mention.id !== userId)
|
|
.map(mention => mention.acct)
|
|
.forEach(i => accountsToMention.add(i))
|
|
return Array.from(accountsToMention)
|
|
}
|
|
|
|
export function getReplyDraft(status: mastodon.v1.Status) {
|
|
const accountsToMention = getAccountsToMention(status)
|
|
return {
|
|
key: `reply-${status.id}`,
|
|
draft: () => {
|
|
return getDefaultDraft({
|
|
initialText: '',
|
|
inReplyToId: status!.id,
|
|
visibility: status.visibility,
|
|
mentions: accountsToMention,
|
|
})
|
|
},
|
|
}
|
|
}
|
|
|
|
export const isEmptyDraft = (draft: Draft | null | undefined) => {
|
|
if (!draft)
|
|
return true
|
|
const { params, attachments } = draft
|
|
const status = params.status || ''
|
|
const text = htmlToText(status).trim().replace(/^(@\S+\s?)+/, '').replaceAll(/```/g, '').trim()
|
|
|
|
return (text.length === 0)
|
|
&& attachments.length === 0
|
|
&& (params.spoilerText || '').length === 0
|
|
}
|
|
|
|
export interface UseDraft {
|
|
draft: Ref<Draft>
|
|
isEmpty: ComputedRef<boolean>
|
|
}
|
|
|
|
export function useDraft(
|
|
draftKey?: string,
|
|
initial: () => Draft = () => getDefaultDraft({}),
|
|
): UseDraft {
|
|
const draft = draftKey
|
|
? computed({
|
|
get() {
|
|
if (!currentUserDrafts.value[draftKey])
|
|
currentUserDrafts.value[draftKey] = initial()
|
|
return currentUserDrafts.value[draftKey]
|
|
},
|
|
set(val) {
|
|
currentUserDrafts.value[draftKey] = val
|
|
},
|
|
})
|
|
: ref(initial())
|
|
|
|
const isEmpty = computed(() => isEmptyDraft(draft.value))
|
|
|
|
onUnmounted(async () => {
|
|
// Remove draft if it's empty
|
|
if (isEmpty.value && draftKey) {
|
|
await nextTick()
|
|
delete currentUserDrafts.value[draftKey]
|
|
}
|
|
})
|
|
|
|
return { draft, isEmpty }
|
|
}
|
|
|
|
export function mentionUser(account: mastodon.v1.Account) {
|
|
openPublishDialog('dialog', getDefaultDraft({
|
|
status: `@${account.acct} `,
|
|
}), true)
|
|
}
|
|
|
|
export function directMessageUser(account: mastodon.v1.Account) {
|
|
openPublishDialog('dialog', getDefaultDraft({
|
|
status: `@${account.acct} `,
|
|
visibility: 'direct',
|
|
}), true)
|
|
}
|
|
|
|
export function clearEmptyDrafts() {
|
|
for (const key in currentUserDrafts.value) {
|
|
if (builtinDraftKeys.includes(key))
|
|
continue
|
|
if (!currentUserDrafts.value[key].params || isEmptyDraft(currentUserDrafts.value[key]))
|
|
delete currentUserDrafts.value[key]
|
|
}
|
|
}
|