Move probe to WS class for reuse in the badge

This commit is contained in:
Arlo Breault 2019-09-26 11:38:58 -04:00
parent 685c3bd262
commit aa107862c5
2 changed files with 32 additions and 26 deletions

View file

@ -1,4 +1,4 @@
/* global Util, chrome, Config, UI, Broker, Snowflake */
/* global Util, chrome, Config, UI, Broker, Snowflake, WS */
/* eslint no-unused-vars: 0 */
/*
@ -31,33 +31,25 @@ class WebExtUI extends UI {
this.setEnabled(false);
return;
}
(new Promise((resolve, reject) => {
const ws = WS.makeWebsocket(config.relayAddr);
ws.onopen = () => {
resolve();
ws.close();
};
ws.onerror = () => {
WS.probeWebsocket(config.relayAddr)
.then(
() => {
chrome.storage.local.get("snowflake-enabled", (result) => {
let enabled = this.enabled;
if (result['snowflake-enabled'] !== void 0) {
enabled = result['snowflake-enabled'];
} else {
log("Toggle state not yet saved");
}
this.setEnabled(enabled);
});
},
() => {
log('Could not connect to bridge.');
this.missingFeature = 'popupBridgeUnreachable';
this.setEnabled(false);
reject('Could not connect to bridge.');
ws.close();
};
}))
.then(() => {
chrome.storage.local.get("snowflake-enabled", (result) => {
let enabled = this.enabled;
if (result['snowflake-enabled'] !== void 0) {
enabled = result['snowflake-enabled'];
} else {
log("Toggle state not yet saved");
}
this.setEnabled(enabled);
});
})
.catch((e) => {
log(e);
});
}
);
}
postActive() {

View file

@ -54,6 +54,20 @@ class WS {
return ws;
}
static probeWebsocket(addr) {
return new Promise((resolve, reject) => {
const ws = WS.makeWebsocket(addr);
ws.onopen = () => {
resolve();
ws.close();
};
ws.onerror = () => {
reject();
ws.close();
};
});
}
}
WS.WSS_ENABLED = true;