PeerTube/server/initializers/migrations/0100-activitypub.ts

213 lines
5.9 KiB
TypeScript

import { values } from 'lodash'
import * as Sequelize from 'sequelize'
import { createPrivateAndPublicKeys } from '../../helpers/peertube-crypto'
import { shareVideoByServer } from '../../lib/activitypub/share'
import { getVideoActivityPubUrl, getVideoChannelActivityPubUrl } from '../../lib/activitypub/url'
import { createLocalAccountWithoutKeys } from '../../lib/user'
import { ApplicationModel } from '../../models/application/application'
import { JOB_CATEGORIES, SERVER_ACCOUNT_NAME } from '../constants'
async function up (utils: {
transaction: Sequelize.Transaction,
queryInterface: Sequelize.QueryInterface,
sequelize: Sequelize.Sequelize,
db: any
}): Promise<void> {
const q = utils.queryInterface
const db = utils.db
// Assert there are no friends
{
const query = 'SELECT COUNT(*) as total FROM "Pods"'
const options = {
type: Sequelize.QueryTypes.SELECT
}
const res = await utils.sequelize.query(query, options)
if (!res[0] || res[0].total !== 0) {
throw new Error('You need to quit friends.')
}
}
// Pods -> Servers
await utils.queryInterface.renameTable('Pods', 'Servers')
// Create Account table
await db.Account.sync()
// Create AccountFollows table
await db.AccountFollow.sync()
// Modify video abuse table
await db.VideoAbuse.destroy({ truncate: true })
await utils.queryInterface.removeColumn('VideoAbuses', 'reporterPodId')
await utils.queryInterface.removeColumn('VideoAbuses', 'reporterUsername')
// Create column link with Account table
{
const data = {
type: Sequelize.INTEGER,
allowNull: false,
references: {
model: 'Accounts',
key: 'id'
},
onDelete: 'CASCADE'
}
await q.addColumn('VideoAbuses', 'reporterAccountId', data)
}
// Drop request tables
await utils.queryInterface.dropTable('RequestToPods')
await utils.queryInterface.dropTable('RequestVideoEvents')
await utils.queryInterface.dropTable('RequestVideoQadus')
await utils.queryInterface.dropTable('Requests')
// Create application account
{
const applicationInstance = await ApplicationModel.findOne()
const accountCreated = await createLocalAccountWithoutKeys(SERVER_ACCOUNT_NAME, null, applicationInstance.id, undefined)
const { publicKey, privateKey } = await createPrivateAndPublicKeys()
accountCreated.set('publicKey', publicKey)
accountCreated.set('privateKey', privateKey)
await accountCreated.save()
}
// Drop old video channel foreign key (referencing Authors)
{
const query = 'ALTER TABLE "VideoChannels" DROP CONSTRAINT "VideoChannels_authorId_fkey"'
await utils.sequelize.query(query)
}
// Recreate accounts for each user
const users = await db.User.findAll()
for (const user of users) {
const account = await createLocalAccountWithoutKeys(user.username, user.id, null, undefined)
const { publicKey, privateKey } = await createPrivateAndPublicKeys()
account.set('publicKey', publicKey)
account.set('privateKey', privateKey)
await account.save()
}
{
const data = {
type: Sequelize.INTEGER,
allowNull: true,
onDelete: 'CASCADE',
reference: {
model: 'Account',
key: 'id'
}
}
await q.addColumn('VideoChannels', 'accountId', data)
{
const query = 'UPDATE "VideoChannels" SET "accountId" = ' +
'(SELECT "Accounts"."id" FROM "Accounts" INNER JOIN "Authors" ON "Authors"."userId" = "Accounts"."userId" ' +
'WHERE "VideoChannels"."authorId" = "Authors"."id")'
await utils.sequelize.query(query)
}
data.allowNull = false
await q.changeColumn('VideoChannels', 'accountId', data)
await q.removeColumn('VideoChannels', 'authorId')
}
// Add url column to "Videos"
{
const data = {
type: Sequelize.STRING,
defaultValue: null,
allowNull: true
}
await q.addColumn('Videos', 'url', data)
const videos = await db.Video.findAll()
for (const video of videos) {
video.url = getVideoActivityPubUrl(video)
await video.save()
}
data.allowNull = false
await q.changeColumn('Videos', 'url', data)
}
// Add url column to "VideoChannels"
{
const data = {
type: Sequelize.STRING,
defaultValue: null,
allowNull: true
}
await q.addColumn('VideoChannels', 'url', data)
const videoChannels = await db.VideoChannel.findAll()
for (const videoChannel of videoChannels) {
videoChannel.url = getVideoChannelActivityPubUrl(videoChannel)
await videoChannel.save()
}
data.allowNull = false
await q.changeColumn('VideoChannels', 'url', data)
}
// Loss old video rates, whatever
await utils.queryInterface.dropTable('UserVideoRates')
await db.AccountVideoRate.sync()
{
const data = {
type: Sequelize.ENUM(values(JOB_CATEGORIES)),
defaultValue: 'transcoding',
allowNull: false
}
await q.addColumn('Jobs', 'category', data)
}
await db.VideoShare.sync()
await db.VideoChannelShare.sync()
{
const videos = await db.Video.findAll({
include: [
{
model: db.Video['sequelize'].models.VideoChannel,
include: [
{
model: db.Video['sequelize'].models.Account,
include: [ { model: db.Video['sequelize'].models.Server, required: false } ]
}
]
},
{
model: db.Video['sequelize'].models.AccountVideoRate,
include: [ db.Video['sequelize'].models.Account ]
},
{
model: db.Video['sequelize'].models.VideoShare,
include: [ db.Video['sequelize'].models.Account ]
},
db.Video['sequelize'].models.Tag,
db.Video['sequelize'].models.VideoFile
]
})
for (const video of videos) {
await shareVideoByServer(video, undefined)
}
}
}
function down (options) {
throw new Error('Not implemented.')
}
export {
up,
down
}