Addressing reviews

pull/448/head
Beaudan Brown 6 years ago
parent 1f7787ecf7
commit fc8aa0211d

@ -50,7 +50,6 @@ class LokiPublicServerAPI {
this.chatAPI = chatAPI; this.chatAPI = chatAPI;
this.server = hostport; this.server = hostport;
this.channels = []; this.channels = [];
this.tokenPending = false;
this.tokenPromise = null; this.tokenPromise = null;
this.baseServerUrl = `https://${this.server}`; this.baseServerUrl = `https://${this.server}`;
} }
@ -80,10 +79,10 @@ class LokiPublicServerAPI {
thisChannel.stopPolling = true; thisChannel.stopPolling = true;
} }
async getServerToken() { async getOrRefreshServerToken() {
let token = await Signal.Data.getPublicServerTokenByServerName(this.server); let token = await Signal.Data.getPublicServerTokenByServerName(this.server);
if (!token) { if (!token) {
token = await this.getNewToken(); token = await this.refreshServerToken();
if (token) { if (token) {
await Signal.Data.savePublicServerToken({ await Signal.Data.savePublicServerToken({
server: this.server, server: this.server,
@ -94,9 +93,8 @@ class LokiPublicServerAPI {
return token; return token;
} }
async getNewToken() { async refreshServerToken() {
if (!this.tokenPending) { if (this.tokenPromise === null) {
this.tokenPending = true;
this.tokenPromise = new Promise(async res => { this.tokenPromise = new Promise(async res => {
const token = await this.requestToken(); const token = await this.requestToken();
if (!token) { if (!token) {
@ -112,7 +110,7 @@ class LokiPublicServerAPI {
}); });
} }
const token = await this.tokenPromise; const token = await this.tokenPromise;
this.tokenPending = false; this.tokenPromise = null;
return token; return token;
} }
@ -133,11 +131,7 @@ class LokiPublicServerAPI {
return null; return null;
} }
const body = await res.json(); const body = await res.json();
const { cipherText64, serverPubKey64 } = body; const token = await libloki.crypto.decryptToken(body);
const token = await libloki.crypto.decryptToken(
cipherText64,
serverPubKey64
);
return token; return token;
} }
@ -153,18 +147,15 @@ class LokiPublicServerAPI {
}), }),
}; };
let res;
let success = true;
try { try {
res = await nodeFetch( const res = await nodeFetch(
`${this.baseServerUrl}/loki/v1/submit_challenge`, `${this.baseServerUrl}/loki/v1/submit_challenge`,
options options
); );
success = res.ok; return res.ok;
} catch (e) { } catch (e) {
return false; return false;
} }
return success;
} }
} }

@ -158,7 +158,7 @@
} }
} }
async function decryptToken(ivAndCipherText64, serverPubKey64) { async function decryptToken({ ivAndCipherText64, serverPubKey64 }) {
const ivAndCipherText = new Uint8Array( const ivAndCipherText = new Uint8Array(
dcodeIO.ByteBuffer.fromBase64(ivAndCipherText64).toArrayBuffer() dcodeIO.ByteBuffer.fromBase64(ivAndCipherText64).toArrayBuffer()
); );

Loading…
Cancel
Save