X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fmodels%2Frequest-video-qadu.ts;h=27ce0ff29ced0a0f620140b6921ef13944f8ffc9;hb=127944aacb52721877449a1f202dc0add866ef39;hp=e914e06cd19f66ccf603767f5747c8e80cb99e83;hpb=e02643f32e4c97ca307f8fc5b69be79c40d70a3b;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/models/request-video-qadu.ts b/server/models/request-video-qadu.ts index e914e06cd..27ce0ff29 100644 --- a/server/models/request-video-qadu.ts +++ b/server/models/request-video-qadu.ts @@ -12,8 +12,8 @@ import { values } from 'lodash' import * as Sequelize from 'sequelize' +import { database as db } from '../initializers/database' import { REQUEST_VIDEO_QADU_TYPES } from '../initializers' - import { addMethodsToModel } from './utils' import { RequestVideoQaduClass, @@ -29,8 +29,8 @@ let listWithLimitAndRandom: RequestVideoQaduMethods.ListWithLimitAndRandom let removeByRequestIdsAndPod: RequestVideoQaduMethods.RemoveByRequestIdsAndPod let removeAll: RequestVideoQaduMethods.RemoveAll -export default function (sequelize, DataTypes) { - RequestVideoQadu = sequelize.define('RequestVideoQadu', +export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) { + RequestVideoQadu = sequelize.define('RequestVideoQadu', { type: { type: DataTypes.ENUM(values(REQUEST_VIDEO_QADU_TYPES)), @@ -83,15 +83,16 @@ function associate (models) { }) } -countTotalRequests = function (callback) { +countTotalRequests = function (callback: RequestVideoQaduMethods.CountTotalRequestsCallback) { const query = {} return RequestVideoQadu.count(query).asCallback(callback) } -listWithLimitAndRandom = function (limitPods, limitRequestsPerPod, callback) { - const Pod = RequestVideoQadu['sequelize'].models.Pod +listWithLimitAndRandom = function (limitPods: number, limitRequestsPerPod: number, callback: RequestVideoQaduMethods.ListWithLimitAndRandomCallback) { + const Pod = db.Pod + const tableJoin = '' - Pod.listRandomPodIdsWithRequest(limitPods, 'RequestVideoQadus', function (err, podIds) { + Pod.listRandomPodIdsWithRequest(limitPods, 'RequestVideoQadus', tableJoin, function (err, podIds) { if (err) return callback(err) // We don't have friends that have requests @@ -122,7 +123,7 @@ listWithLimitAndRandom = function (limitPods, limitRequestsPerPod, callback) { }) } -removeByRequestIdsAndPod = function (ids, podId, callback) { +removeByRequestIdsAndPod = function (ids: number[], podId: number, callback: RequestVideoQaduMethods.RemoveByRequestIdsAndPodCallback) { const query = { where: { id: { @@ -135,14 +136,14 @@ removeByRequestIdsAndPod = function (ids, podId, callback) { RequestVideoQadu.destroy(query).asCallback(callback) } -removeAll = function (callback) { +removeAll = function (callback: RequestVideoQaduMethods.RemoveAllCallback) { // Delete all requests RequestVideoQadu.truncate({ cascade: true }).asCallback(callback) } // --------------------------------------------------------------------------- -function groupAndTruncateRequests (requests, limitRequestsPerPod) { +function groupAndTruncateRequests (requests: RequestVideoQaduInstance[], limitRequestsPerPod: number) { const requestsGrouped = {} requests.forEach(function (request) {