mirror of
https://github.com/overleaf/overleaf.git
synced 2024-12-07 12:37:49 -05:00
5520f9eed9
[real-time] migrate acceptance tests to auto-join GitOrigin-RevId: fc14a4e803bfab4d07f2773c6ecbdb5459d12684
154 lines
4 KiB
JavaScript
154 lines
4 KiB
JavaScript
/* eslint-disable
|
|
no-return-assign,
|
|
*/
|
|
// TODO: This file was created by bulk-decaffeinate.
|
|
// Fix any style issues and re-enable lint.
|
|
/*
|
|
* decaffeinate suggestions:
|
|
* DS102: Remove unnecessary code created because of implicit returns
|
|
* DS207: Consider shorter variations of null checks
|
|
* Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
|
|
*/
|
|
let Client
|
|
const { XMLHttpRequest } = require('../../libs/XMLHttpRequest')
|
|
const io = require('socket.io-client')
|
|
const async = require('async')
|
|
|
|
const request = require('request')
|
|
const Settings = require('@overleaf/settings')
|
|
const redis = require('@overleaf/redis-wrapper')
|
|
const rclient = redis.createClient(Settings.redis.websessions)
|
|
|
|
const uid = require('uid-safe').sync
|
|
const signature = require('cookie-signature')
|
|
|
|
io.util.request = function () {
|
|
const xhr = new XMLHttpRequest()
|
|
const _open = xhr.open
|
|
xhr.open = function () {
|
|
_open.apply(xhr, arguments)
|
|
if (Client.cookie != null) {
|
|
return xhr.setRequestHeader('Cookie', Client.cookie)
|
|
}
|
|
}
|
|
return xhr
|
|
}
|
|
|
|
module.exports = Client = {
|
|
cookie: null,
|
|
|
|
setSession(session, callback) {
|
|
if (callback == null) {
|
|
callback = function () {}
|
|
}
|
|
const sessionId = uid(24)
|
|
session.cookie = {}
|
|
return rclient.set('sess:' + sessionId, JSON.stringify(session), error => {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
const secret = Settings.security.sessionSecret
|
|
const cookieKey = 's:' + signature.sign(sessionId, secret)
|
|
Client.cookie = `${Settings.cookieName}=${cookieKey}`
|
|
return callback()
|
|
})
|
|
},
|
|
|
|
setAnonSession(projectId, anonymousAccessToken, callback) {
|
|
Client.setSession(
|
|
{
|
|
anonTokenAccess: {
|
|
[projectId]: anonymousAccessToken,
|
|
},
|
|
},
|
|
callback
|
|
)
|
|
},
|
|
|
|
unsetSession(callback) {
|
|
if (callback == null) {
|
|
callback = function () {}
|
|
}
|
|
Client.cookie = null
|
|
return callback()
|
|
},
|
|
|
|
connect(projectId, callback) {
|
|
const client = io.connect('http://localhost:3026', {
|
|
'force new connection': true,
|
|
query: new URLSearchParams({ projectId }).toString(),
|
|
})
|
|
let disconnected = false
|
|
client.on('disconnect', () => {
|
|
disconnected = true
|
|
})
|
|
client.on('connectionRejected', err => {
|
|
// Wait for disconnect ahead of continuing with the test sequence.
|
|
setTimeout(() => {
|
|
if (!disconnected) {
|
|
throw new Error('should disconnect after connectionRejected')
|
|
}
|
|
callback(err)
|
|
}, 10)
|
|
})
|
|
client.on('joinProjectResponse', resp => {
|
|
const { publicId, project, permissionsLevel, protocolVersion } = resp
|
|
client.publicId = publicId
|
|
callback(null, project, permissionsLevel, protocolVersion)
|
|
})
|
|
return client
|
|
},
|
|
|
|
getConnectedClients(callback) {
|
|
if (callback == null) {
|
|
callback = function () {}
|
|
}
|
|
return request.get(
|
|
{
|
|
url: 'http://localhost:3026/clients',
|
|
json: true,
|
|
},
|
|
(error, response, data) => callback(error, data)
|
|
)
|
|
},
|
|
|
|
getConnectedClient(clientId, callback) {
|
|
if (callback == null) {
|
|
callback = function () {}
|
|
}
|
|
return request.get(
|
|
{
|
|
url: `http://localhost:3026/clients/${clientId}`,
|
|
json: true,
|
|
},
|
|
(error, response, data) => {
|
|
if (response?.statusCode === 404) {
|
|
callback(new Error('not found'))
|
|
} else {
|
|
callback(error, data)
|
|
}
|
|
}
|
|
)
|
|
},
|
|
|
|
disconnectClient(clientId, callback) {
|
|
request.post(
|
|
{
|
|
url: `http://localhost:3026/client/${clientId}/disconnect`,
|
|
},
|
|
(error, response, data) => callback(error, data)
|
|
)
|
|
return null
|
|
},
|
|
|
|
disconnectAllClients(callback) {
|
|
return Client.getConnectedClients((error, clients) => {
|
|
if (error) return callback(error)
|
|
async.each(
|
|
clients,
|
|
(clientView, cb) => Client.disconnectClient(clientView.client_id, cb),
|
|
callback
|
|
)
|
|
})
|
|
},
|
|
}
|