mirror of https://git.sr.ht/~cadence/bibliogram
45 lines
1.1 KiB
JavaScript
45 lines
1.1 KiB
JavaScript
const constants = require("../constants")
|
|
const {request} = require("./request")
|
|
|
|
class TorSwitcher {
|
|
constructor() {
|
|
this.torManager = null
|
|
}
|
|
|
|
setManager(torManager) {
|
|
this.torManager = torManager
|
|
}
|
|
|
|
canUseTor() {
|
|
return !!this.torManager
|
|
}
|
|
|
|
/**
|
|
* Request from the URL.
|
|
* The test function will be called with the response object.
|
|
* If the test function succeeds, its return value will be returned here.
|
|
* If the test function fails, its error will be rejected here.
|
|
* Only include rate limit logic in the test function!
|
|
* @param {string} url
|
|
* @param {(res: import("./request_backends/reference").GrabResponse) => any} test
|
|
* @returns {Promise<import("./request_backends/reference")>}
|
|
*/
|
|
request(type, url, test) {
|
|
if (this.torManager && constants.tor.for[type]) {
|
|
return this.torManager.request(url, test)
|
|
} else {
|
|
return request(url).check(test)
|
|
}
|
|
}
|
|
}
|
|
|
|
const switcher = new TorSwitcher()
|
|
|
|
if (constants.tor.enabled) {
|
|
require("./tor").then(torManager => {
|
|
if (torManager) switcher.setManager(torManager)
|
|
})
|
|
}
|
|
|
|
module.exports = switcher
|