]>
Commit | Line | Data |
---|---|---|
1735c825 | 1 | import { QueryTypes, Sequelize } from 'sequelize' |
48f07b4a | 2 | import { ServerInfo } from '../server/servers' |
09071200 | 3 | import { PluginType } from '../../models/plugins/plugin.type' |
df66d815 | 4 | |
1735c825 | 5 | let sequelizes: { [ id: number ]: Sequelize } = {} |
c8000975 | 6 | |
48f07b4a C |
7 | function getSequelize (internalServerNumber: number) { |
8 | if (sequelizes[internalServerNumber]) return sequelizes[internalServerNumber] | |
c8000975 | 9 | |
48f07b4a | 10 | const dbname = 'peertube_test' + internalServerNumber |
df66d815 C |
11 | const username = 'peertube' |
12 | const password = 'peertube' | |
13 | const host = 'localhost' | |
14 | const port = 5432 | |
15 | ||
c8000975 | 16 | const seq = new Sequelize(dbname, username, password, { |
df66d815 C |
17 | dialect: 'postgres', |
18 | host, | |
19 | port, | |
df66d815 C |
20 | logging: false |
21 | }) | |
c8000975 | 22 | |
48f07b4a | 23 | sequelizes[internalServerNumber] = seq |
c8000975 C |
24 | |
25 | return seq | |
df66d815 C |
26 | } |
27 | ||
48f07b4a C |
28 | function setActorField (internalServerNumber: number, to: string, field: string, value: string) { |
29 | const seq = getSequelize(internalServerNumber) | |
df66d815 | 30 | |
1735c825 | 31 | const options = { type: QueryTypes.UPDATE } |
df66d815 C |
32 | |
33 | return seq.query(`UPDATE actor SET "${field}" = '${value}' WHERE url = '${to}'`, options) | |
34 | } | |
35 | ||
48f07b4a C |
36 | function setVideoField (internalServerNumber: number, uuid: string, field: string, value: string) { |
37 | const seq = getSequelize(internalServerNumber) | |
a130f33c | 38 | |
1735c825 | 39 | const options = { type: QueryTypes.UPDATE } |
a130f33c C |
40 | |
41 | return seq.query(`UPDATE video SET "${field}" = '${value}' WHERE uuid = '${uuid}'`, options) | |
42 | } | |
43 | ||
48f07b4a C |
44 | function setPlaylistField (internalServerNumber: number, uuid: string, field: string, value: string) { |
45 | const seq = getSequelize(internalServerNumber) | |
9f79ade6 | 46 | |
1735c825 | 47 | const options = { type: QueryTypes.UPDATE } |
9f79ade6 C |
48 | |
49 | return seq.query(`UPDATE "videoPlaylist" SET "${field}" = '${value}' WHERE uuid = '${uuid}'`, options) | |
50 | } | |
51 | ||
48f07b4a C |
52 | async function countVideoViewsOf (internalServerNumber: number, uuid: string) { |
53 | const seq = getSequelize(internalServerNumber) | |
cda03765 C |
54 | |
55 | // tslint:disable | |
56 | const query = `SELECT SUM("videoView"."views") AS "total" FROM "videoView" INNER JOIN "video" ON "video"."id" = "videoView"."videoId" WHERE "video"."uuid" = '${uuid}'` | |
57 | ||
1735c825 C |
58 | const options = { type: QueryTypes.SELECT as QueryTypes.SELECT } |
59 | const [ { total } ] = await seq.query<{ total: number }>(query, options) | |
cda03765 C |
60 | |
61 | if (!total) return 0 | |
62 | ||
1735c825 C |
63 | // FIXME: check if we really need parseInt |
64 | return parseInt(total + '', 10) | |
cda03765 C |
65 | } |
66 | ||
48f07b4a C |
67 | async function closeAllSequelize (servers: ServerInfo[]) { |
68 | for (const server of servers) { | |
69 | if (sequelizes[ server.internalServerNumber ]) { | |
70 | await sequelizes[ server.internalServerNumber ].close() | |
71 | delete sequelizes[ server.internalServerNumber ] | |
c8000975 C |
72 | } |
73 | } | |
74 | } | |
75 | ||
09071200 C |
76 | function setPluginVersion (internalServerNumber: number, pluginName: string, newVersion: string) { |
77 | const seq = getSequelize(internalServerNumber) | |
78 | ||
79 | const options = { type: QueryTypes.UPDATE } | |
80 | ||
81 | return seq.query(`UPDATE "plugin" SET "version" = '${newVersion}' WHERE "name" = '${pluginName}'`, options) | |
82 | } | |
83 | ||
df66d815 | 84 | export { |
a130f33c | 85 | setVideoField, |
9f79ade6 | 86 | setPlaylistField, |
c8000975 | 87 | setActorField, |
cda03765 | 88 | countVideoViewsOf, |
09071200 | 89 | setPluginVersion, |
c8000975 | 90 | closeAllSequelize |
df66d815 | 91 | } |