Add bookmarks to archive takeout

pull/687/merge
Thibaut Girka 2018-08-29 13:06:37 +02:00 committed by ThibG
parent c20b27a9f4
commit 5d647197ec
1 changed files with 21 additions and 0 deletions

View File

@ -45,6 +45,7 @@ class BackupService < BaseService
dump_media_attachments!(tar) dump_media_attachments!(tar)
dump_outbox!(tar) dump_outbox!(tar)
dump_likes!(tar) dump_likes!(tar)
dump_bookmarks!(tar)
dump_actor!(tar) dump_actor!(tar)
end end
end end
@ -85,6 +86,7 @@ class BackupService < BaseService
actor[:image][:url] = 'header' + File.extname(actor[:image][:url]) if actor[:image] actor[:image][:url] = 'header' + File.extname(actor[:image][:url]) if actor[:image]
actor[:outbox] = 'outbox.json' actor[:outbox] = 'outbox.json'
actor[:likes] = 'likes.json' actor[:likes] = 'likes.json'
actor[:bookmarks] = 'bookmarks.json'
download_to_tar(tar, account.avatar, 'avatar' + File.extname(account.avatar.path)) if account.avatar.exists? download_to_tar(tar, account.avatar, 'avatar' + File.extname(account.avatar.path)) if account.avatar.exists?
download_to_tar(tar, account.header, 'header' + File.extname(account.header.path)) if account.header.exists? download_to_tar(tar, account.header, 'header' + File.extname(account.header.path)) if account.header.exists?
@ -115,6 +117,25 @@ class BackupService < BaseService
end end
end end
def dump_bookmarks!(tar)
collection = serialize(ActivityPub::CollectionPresenter.new(id: 'bookmarks.json', type: :ordered, size: 0, items: []), ActivityPub::CollectionSerializer)
Status.reorder(nil).joins(:bookmarks).includes(:account).merge(account.bookmarks).find_in_batches do |statuses|
statuses.each do |status|
collection[:totalItems] += 1
collection[:orderedItems] << ActivityPub::TagManager.instance.uri_for(status)
end
GC.start
end
json = Oj.dump(collection)
tar.add_file_simple('bookmarks.json', 0o444, json.bytesize) do |io|
io.write(json)
end
end
def collection_presenter def collection_presenter
ActivityPub::CollectionPresenter.new( ActivityPub::CollectionPresenter.new(
id: 'outbox.json', id: 'outbox.json',