diff options
Diffstat (limited to 'server/models')
-rw-r--r-- | server/models/video/thumbnail.ts | 42 | ||||
-rw-r--r-- | server/models/video/video-playlist.ts | 1 |
2 files changed, 41 insertions, 2 deletions
diff --git a/server/models/video/thumbnail.ts b/server/models/video/thumbnail.ts index 3cad6c668..3d885f654 100644 --- a/server/models/video/thumbnail.ts +++ b/server/models/video/thumbnail.ts | |||
@@ -3,6 +3,8 @@ import { join } from 'path' | |||
3 | import { | 3 | import { |
4 | AfterDestroy, | 4 | AfterDestroy, |
5 | AllowNull, | 5 | AllowNull, |
6 | BeforeCreate, | ||
7 | BeforeUpdate, | ||
6 | BelongsTo, | 8 | BelongsTo, |
7 | Column, | 9 | Column, |
8 | CreatedAt, | 10 | CreatedAt, |
@@ -14,7 +16,8 @@ import { | |||
14 | UpdatedAt | 16 | UpdatedAt |
15 | } from 'sequelize-typescript' | 17 | } from 'sequelize-typescript' |
16 | import { buildRemoteVideoBaseUrl } from '@server/helpers/activitypub' | 18 | import { buildRemoteVideoBaseUrl } from '@server/helpers/activitypub' |
17 | import { MThumbnailVideo, MVideoAccountLight } from '@server/types/models' | 19 | import { afterCommitIfTransaction } from '@server/helpers/database-utils' |
20 | import { MThumbnail, MThumbnailVideo, MVideoAccountLight } from '@server/types/models' | ||
18 | import { ThumbnailType } from '../../../shared/models/videos/thumbnail.type' | 21 | import { ThumbnailType } from '../../../shared/models/videos/thumbnail.type' |
19 | import { logger } from '../../helpers/logger' | 22 | import { logger } from '../../helpers/logger' |
20 | import { CONFIG } from '../../initializers/config' | 23 | import { CONFIG } from '../../initializers/config' |
@@ -96,6 +99,9 @@ export class ThumbnailModel extends Model { | |||
96 | @UpdatedAt | 99 | @UpdatedAt |
97 | updatedAt: Date | 100 | updatedAt: Date |
98 | 101 | ||
102 | // If this thumbnail replaced existing one, track the old name | ||
103 | previousThumbnailFilename: string | ||
104 | |||
99 | private static readonly types: { [ id in ThumbnailType ]: { label: string, directory: string, staticPath: string } } = { | 105 | private static readonly types: { [ id in ThumbnailType ]: { label: string, directory: string, staticPath: string } } = { |
100 | [ThumbnailType.MINIATURE]: { | 106 | [ThumbnailType.MINIATURE]: { |
101 | label: 'miniature', | 107 | label: 'miniature', |
@@ -109,6 +115,12 @@ export class ThumbnailModel extends Model { | |||
109 | } | 115 | } |
110 | } | 116 | } |
111 | 117 | ||
118 | @BeforeCreate | ||
119 | @BeforeUpdate | ||
120 | static removeOldFile (instance: ThumbnailModel, options) { | ||
121 | return afterCommitIfTransaction(options.transaction, () => instance.removePreviousFilenameIfNeeded()) | ||
122 | } | ||
123 | |||
112 | @AfterDestroy | 124 | @AfterDestroy |
113 | static removeFiles (instance: ThumbnailModel) { | 125 | static removeFiles (instance: ThumbnailModel) { |
114 | logger.info('Removing %s file %s.', ThumbnailModel.types[instance.type].label, instance.filename) | 126 | logger.info('Removing %s file %s.', ThumbnailModel.types[instance.type].label, instance.filename) |
@@ -118,7 +130,18 @@ export class ThumbnailModel extends Model { | |||
118 | .catch(err => logger.error('Cannot remove thumbnail file %s.', instance.filename, err)) | 130 | .catch(err => logger.error('Cannot remove thumbnail file %s.', instance.filename, err)) |
119 | } | 131 | } |
120 | 132 | ||
121 | static loadWithVideoByName (filename: string, thumbnailType: ThumbnailType): Promise<MThumbnailVideo> { | 133 | static loadByFilename (filename: string, thumbnailType: ThumbnailType): Promise<MThumbnail> { |
134 | const query = { | ||
135 | where: { | ||
136 | filename, | ||
137 | type: thumbnailType | ||
138 | } | ||
139 | } | ||
140 | |||
141 | return ThumbnailModel.findOne(query) | ||
142 | } | ||
143 | |||
144 | static loadWithVideoByFilename (filename: string, thumbnailType: ThumbnailType): Promise<MThumbnailVideo> { | ||
122 | const query = { | 145 | const query = { |
123 | where: { | 146 | where: { |
124 | filename, | 147 | filename, |
@@ -150,7 +173,22 @@ export class ThumbnailModel extends Model { | |||
150 | return join(directory, this.filename) | 173 | return join(directory, this.filename) |
151 | } | 174 | } |
152 | 175 | ||
176 | getPreviousPath () { | ||
177 | const directory = ThumbnailModel.types[this.type].directory | ||
178 | return join(directory, this.previousThumbnailFilename) | ||
179 | } | ||
180 | |||
153 | removeThumbnail () { | 181 | removeThumbnail () { |
154 | return remove(this.getPath()) | 182 | return remove(this.getPath()) |
155 | } | 183 | } |
184 | |||
185 | removePreviousFilenameIfNeeded () { | ||
186 | if (!this.previousThumbnailFilename) return | ||
187 | |||
188 | const previousPath = this.getPreviousPath() | ||
189 | remove(previousPath) | ||
190 | .catch(err => logger.error('Cannot remove previous thumbnail file %s.', previousPath, { err })) | ||
191 | |||
192 | this.previousThumbnailFilename = undefined | ||
193 | } | ||
156 | } | 194 | } |
diff --git a/server/models/video/video-playlist.ts b/server/models/video/video-playlist.ts index 9e6ff1f81..49a406608 100644 --- a/server/models/video/video-playlist.ts +++ b/server/models/video/video-playlist.ts | |||
@@ -17,6 +17,7 @@ import { | |||
17 | Table, | 17 | Table, |
18 | UpdatedAt | 18 | UpdatedAt |
19 | } from 'sequelize-typescript' | 19 | } from 'sequelize-typescript' |
20 | import { v4 as uuidv4 } from 'uuid' | ||
20 | import { MAccountId, MChannelId } from '@server/types/models' | 21 | import { MAccountId, MChannelId } from '@server/types/models' |
21 | import { ActivityIconObject } from '../../../shared/models/activitypub/objects' | 22 | import { ActivityIconObject } from '../../../shared/models/activitypub/objects' |
22 | import { PlaylistObject } from '../../../shared/models/activitypub/objects/playlist-object' | 23 | import { PlaylistObject } from '../../../shared/models/activitypub/objects/playlist-object' |