PeerTube/server/helpers/requests.ts

92 lines
2.3 KiB
TypeScript
Raw Normal View History

2017-06-05 21:53:49 +02:00
import * as replay from 'request-replay'
import * as request from 'request'
import * as Promise from 'bluebird'
2016-02-05 18:03:20 +01:00
2017-05-15 22:22:03 +02:00
import {
RETRY_REQUESTS,
REMOTE_SCHEME,
CONFIG
} from '../initializers'
2017-06-10 22:15:25 +02:00
import { PodInstance } from '../models'
import { PodSignature } from '../../shared'
2017-05-15 22:22:03 +02:00
import { sign } from './peertube-crypto'
2016-02-05 18:03:20 +01:00
2017-06-10 22:15:25 +02:00
type MakeRetryRequestParams = {
url: string,
2017-10-31 16:31:24 +01:00
method: 'GET' | 'POST',
2017-06-10 22:15:25 +02:00
json: Object
}
function makeRetryRequest (params: MakeRetryRequestParams) {
return new Promise<{ response: request.RequestResponse, body: any }>((res, rej) => {
replay(
request(params, (err, response, body) => err ? rej(err) : res({ response, body })),
{
retries: RETRY_REQUESTS,
factor: 3,
maxTimeout: Infinity,
errorCodes: [ 'EADDRINFO', 'ETIMEDOUT', 'ECONNRESET', 'ESOCKETTIMEDOUT', 'ENOTFOUND', 'ECONNREFUSED' ]
}
)
})
}
2016-02-05 18:03:20 +01:00
2017-06-10 22:15:25 +02:00
type MakeSecureRequestParams = {
2017-10-31 16:31:24 +01:00
method: 'GET' | 'POST'
2017-06-10 22:15:25 +02:00
toPod: PodInstance
path: string
data?: Object
}
function makeSecureRequest (params: MakeSecureRequestParams) {
return new Promise<{ response: request.RequestResponse, body: any }>((res, rej) => {
const requestParams: {
url: string,
json: {
signature: PodSignature,
data: any
}
} = {
url: REMOTE_SCHEME.HTTP + '://' + params.toPod.host + params.path,
json: {
signature: null,
data: null
}
}
2016-02-05 18:03:20 +01:00
if (params.method !== 'POST') {
return rej(new Error('Cannot make a secure request with a non POST method.'))
}
2017-07-07 16:57:28 +02:00
const host = CONFIG.WEBSERVER.HOST
2017-07-07 16:57:28 +02:00
let dataToSign
if (params.data) {
dataToSign = params.data
} else {
// We do not have data to sign so we just take our host
// It is not ideal but the connection should be in HTTPS
dataToSign = host
}
2017-07-07 16:57:28 +02:00
sign(dataToSign).then(signature => {
requestParams.json.signature = {
host, // Which host we pretend to be
2017-07-07 16:57:28 +02:00
signature
}
// If there are data information
2017-07-07 16:57:28 +02:00
if (params.data) {
requestParams.json.data = params.data
2017-07-07 16:57:28 +02:00
}
2017-07-07 16:57:28 +02:00
request.post(requestParams, (err, response, body) => err ? rej(err) : res({ response, body }))
})
})
}
2016-02-05 18:03:20 +01:00
// ---------------------------------------------------------------------------
2016-02-05 18:03:20 +01:00
2017-05-15 22:22:03 +02:00
export {
makeRetryRequest,
makeSecureRequest
}