pinafore/bin/restore-mastodon-data.js
2022-11-17 17:09:24 -08:00

52 lines
2.2 KiB
JavaScript

import { actions } from './mastodon-data.js'
import { users } from '../tests/users.js'
import { postStatus } from '../src/routes/_api/statuses.js'
import { followAccount } from '../src/routes/_api/follow.js'
import { favoriteStatus } from '../src/routes/_api/favorite.js'
import { reblogStatus } from '../src/routes/_api/reblog.js'
import fetch from 'node-fetch'
import FileApi from 'file-api'
import { pinStatus } from '../src/routes/_api/pin.js'
import { submitMedia } from '../tests/submitMedia.js'
global.File = FileApi.File
global.FormData = FileApi.FormData
global.fetch = fetch
export async function restoreMastodonData () {
console.log('Restoring mastodon data...')
const internalIdsToIds = {}
for (const action of actions) {
if (!action.post || /@/.test(action.post.text)) {
// If the action is a boost, favorite, mention, etc., then it needs to
// be delayed, otherwise it may appear in an unpredictable order and break the tests.
await new Promise(resolve => setTimeout(resolve, 1500))
}
console.log(JSON.stringify(action))
const accessToken = users[action.user].accessToken
if (action.post) {
const { text, media, sensitive, spoiler, privacy, inReplyTo, internalId } = action.post
const mediaIds = media && await Promise.all(media.map(async mediaItem => {
const mediaResponse = await submitMedia(accessToken, mediaItem, 'kitten')
return mediaResponse.id
}))
const inReplyToId = inReplyTo && internalIdsToIds[inReplyTo]
const status = await postStatus('localhost:3000', accessToken, text, inReplyToId, mediaIds,
sensitive, spoiler, privacy || 'public')
if (typeof internalId !== 'undefined') {
internalIdsToIds[internalId] = status.id
}
} else if (action.follow) {
await followAccount('localhost:3000', accessToken, users[action.follow].id)
} else if (action.favorite) {
await favoriteStatus('localhost:3000', accessToken, internalIdsToIds[action.favorite])
} else if (action.boost) {
await reblogStatus('localhost:3000', accessToken, internalIdsToIds[action.boost])
} else if (action.pin) {
await pinStatus('localhost:3000', accessToken, internalIdsToIds[action.pin])
}
}
console.log('Restored mastodon data')
}