compressed web socket support

This commit is contained in:
MathMan05
2024-07-30 20:01:03 -05:00
parent b9f1bb5e53
commit 44294c6ea9
5 changed files with 241 additions and 162 deletions

View File

@@ -106,7 +106,7 @@ class Localuser {
async initwebsocket() { async initwebsocket() {
let returny = null; let returny = null;
const promise = new Promise((res) => { returny = res; }); const promise = new Promise((res) => { returny = res; });
this.ws = new WebSocket(this.serverurls.gateway.toString()); this.ws = new WebSocket(this.serverurls.gateway.toString() + "?encoding=json&v=9" + (DecompressionStream ? "&compress=zlib-stream" : ""));
this.ws.addEventListener('open', (event) => { this.ws.addEventListener('open', (event) => {
console.log('WebSocket connected'); console.log('WebSocket connected');
this.ws.send(JSON.stringify({ this.ws.send(JSON.stringify({
@@ -120,7 +120,7 @@ class Localuser {
"release_channel": "Custom", "release_channel": "Custom",
"browser_user_agent": navigator.userAgent "browser_user_agent": navigator.userAgent
}, },
"compress": false, "compress": !!DecompressionStream,
"presence": { "presence": {
"status": "online", "status": "online",
"since": new Date().getTime(), "since": new Date().getTime(),
@@ -130,9 +130,80 @@ class Localuser {
} }
})); }));
}); });
this.ws.addEventListener('message', (event) => { const ds = new DecompressionStream("deflate");
const temp = JSON.parse(event.data); const w = ds.writable.getWriter();
console.log(temp); const r = ds.readable.getReader();
let arr = new Uint8Array();
let build = "";
this.ws.addEventListener('message', async (event) => {
let temp;
if (event.data instanceof Blob) {
const buff = await event.data.arrayBuffer();
const array = new Uint8Array(buff);
const temparr = new Uint8Array(array.length + arr.length);
temparr.set(arr, 0);
temparr.set(array, arr.length);
arr = temparr;
const len = array.length;
if (!(array[len - 1] === 255 && array[len - 2] === 255 && array[len - 3] === 0 && array[len - 4] === 0)) {
return;
}
w.write(arr.buffer);
arr = new Uint8Array();
//console.log(data,test);
const read = (await r.read());
const data = new TextDecoder().decode(read.value);
build += data;
console.log("temp");
try {
temp = JSON.parse(build);
build = "";
}
catch {
return;
}
}
else {
temp = JSON.parse(event.data);
}
if (temp.op === 0 && temp.t === "READY") {
returny();
}
this.handleEvent(temp);
});
this.ws.addEventListener("close", event => {
console.log("WebSocket closed with code " + event.code);
if (this.wsinterval)
clearInterval(this.wsinterval);
this.unload();
document.getElementById("loading").classList.remove("doneloading");
document.getElementById("loading").classList.add("loading");
if (((event.code > 1000 && event.code < 1016) || wsCodesRetry.has(event.code))) {
if (this.connectionSucceed !== 0 && Date.now() > this.connectionSucceed + 20000)
this.errorBackoff = 0;
else
this.errorBackoff++;
this.connectionSucceed = 0;
document.getElementById("load-desc").innerHTML = "Unable to connect to the Spacebar server, retrying in <b>" + Math.round(0.2 + (this.errorBackoff * 2.8)) + "</b> seconds...";
setTimeout(() => {
document.getElementById("load-desc").textContent = "Retrying...";
this.initwebsocket().then(() => {
this.loaduser();
this.init();
document.getElementById("loading").classList.add("doneloading");
document.getElementById("loading").classList.remove("loading");
console.log("done loading");
});
}, 200 + (this.errorBackoff * 2800));
}
else
document.getElementById("load-desc").textContent = "Unable to connect to the Spacebar server. Please try logging out and back in.";
});
await promise;
return;
}
handleEvent(temp) {
console.debug(temp);
if (temp.s) if (temp.s)
this.lastSequence = temp.s; this.lastSequence = temp.s;
if (temp.op == 0) { if (temp.op == 0) {
@@ -149,7 +220,6 @@ class Localuser {
case "READY": case "READY":
this.gottenReady(temp); this.gottenReady(temp);
this.genusersettings(); this.genusersettings();
returny();
break; break;
case "MESSAGE_UPDATE": case "MESSAGE_UPDATE":
const message = SnowFlake.getSnowFlakeFromID(temp.d.id, Message).getObject(); const message = SnowFlake.getSnowFlakeFromID(temp.d.id, Message).getObject();
@@ -209,37 +279,6 @@ class Localuser {
this.ws.send(JSON.stringify({ op: 1, d: this.lastSequence })); this.ws.send(JSON.stringify({ op: 1, d: this.lastSequence }));
}, temp.d.heartbeat_interval); }, temp.d.heartbeat_interval);
} }
});
this.ws.addEventListener("close", event => {
console.log("WebSocket closed with code " + event.code);
if (this.wsinterval)
clearInterval(this.wsinterval);
this.unload();
document.getElementById("loading").classList.remove("doneloading");
document.getElementById("loading").classList.add("loading");
if (((event.code > 1000 && event.code < 1016) || wsCodesRetry.has(event.code))) {
if (this.connectionSucceed !== 0 && Date.now() > this.connectionSucceed + 20000)
this.errorBackoff = 0;
else
this.errorBackoff++;
this.connectionSucceed = 0;
document.getElementById("load-desc").innerHTML = "Unable to connect to the Spacebar server, retrying in <b>" + Math.round(0.2 + (this.errorBackoff * 2.8)) + "</b> seconds...";
setTimeout(() => {
document.getElementById("load-desc").textContent = "Retrying...";
this.initwebsocket().then(() => {
this.loaduser();
this.init();
document.getElementById("loading").classList.add("doneloading");
document.getElementById("loading").classList.remove("loading");
console.log("done loading");
});
}, 200 + (this.errorBackoff * 2800));
}
else
document.getElementById("load-desc").textContent = "Unable to connect to the Spacebar server. Please try logging out and back in.";
});
await promise;
return;
} }
resolveChannelFromID(ID) { resolveChannelFromID(ID) {
let resolve = this.guilds.find(guild => guild.channelids[ID]); let resolve = this.guilds.find(guild => guild.channelids[ID]);

View File

@@ -99,7 +99,9 @@ class Member {
} }
return memb; return memb;
} }
const promoise = fetch(guild.info.api.toString() + "/users/" + id + "/profile?with_mutual_guilds=true&with_mutual_friends_count=true&guild_id=" + guild.snowflake, { headers: guild.headers }).then(_ => _.json()).then(json => { const prom1 = fetch(guild.info.api.toString() + "/users/" + id + "/profile?with_mutual_guilds=true&with_mutual_friends_count=true&guild_id=" + guild.snowflake, { headers: guild.headers });
prom1.catch(_ => { console.log(_); });
const promoise = prom1.then(_ => _.json()).then(json => {
const memb = new Member(json, guild); const memb = new Member(json, guild);
Member.already[guild.id][id] = memb; Member.already[guild.id][id] = memb;
console.log("resolved"); console.log("resolved");

View File

@@ -112,7 +112,7 @@ class Localuser{
async initwebsocket():Promise<void>{ async initwebsocket():Promise<void>{
let returny=null let returny=null
const promise=new Promise((res)=>{returny=res}); const promise=new Promise((res)=>{returny=res});
this.ws = new WebSocket(this.serverurls.gateway.toString()); this.ws = new WebSocket(this.serverurls.gateway.toString()+"?encoding=json&v=9"+(DecompressionStream?"&compress=zlib-stream":""));
this.ws.addEventListener('open', (event) => { this.ws.addEventListener('open', (event) => {
console.log('WebSocket connected'); console.log('WebSocket connected');
this.ws.send(JSON.stringify({ this.ws.send(JSON.stringify({
@@ -126,7 +126,7 @@ class Localuser{
"release_channel": "Custom", "release_channel": "Custom",
"browser_user_agent": navigator.userAgent "browser_user_agent": navigator.userAgent
}, },
"compress": false, "compress": !!DecompressionStream,
"presence": { "presence": {
"status": "online", "status": "online",
"since": new Date().getTime(), "since": new Date().getTime(),
@@ -136,14 +136,81 @@ class Localuser{
} }
})) }))
}); });
const ds = new DecompressionStream("deflate");
const w= ds.writable.getWriter();
const r=ds.readable.getReader();
let arr=new Uint8Array();
let build="";
this.ws.addEventListener('message', async (event) => {
let temp:{op:number,t:string};
if(event.data instanceof Blob){
const buff=await event.data.arrayBuffer()
const array=new Uint8Array(buff);
this.ws.addEventListener('message', (event) => { const temparr=new Uint8Array(array.length+arr.length);
temparr.set(arr, 0);
temparr.set(array, arr.length);
arr=temparr;
const len=array.length;
if(!(array[len-1]===255&&array[len-2]===255&&array[len-3]===0&&array[len-4]===0)){
return;
}
w.write(arr.buffer);
arr=new Uint8Array();
//console.log(data,test);
const read=(await r.read());
const data=new TextDecoder().decode(read.value);
build+=data;
console.log("temp");
try{
temp=JSON.parse(build);
build="";
}catch{
return;
}
}else{
temp=JSON.parse(event.data);
}
if(temp.op===0&&temp.t==="READY"){
returny();
}
this.handleEvent(temp);
});
this.ws.addEventListener("close", event => {
console.log("WebSocket closed with code " + event.code);
if (this.wsinterval) clearInterval(this.wsinterval);
const temp=JSON.parse(event.data); this.unload();
console.log(temp) document.getElementById("loading").classList.remove("doneloading");
document.getElementById("loading").classList.add("loading");
if (((event.code>1000 && event.code<1016) || wsCodesRetry.has(event.code))) {
if (this.connectionSucceed!==0 && Date.now()>this.connectionSucceed+20000) this.errorBackoff=0;
else this.errorBackoff++;
this.connectionSucceed=0;
document.getElementById("load-desc").innerHTML="Unable to connect to the Spacebar server, retrying in <b>" + Math.round(0.2 + (this.errorBackoff*2.8)) + "</b> seconds...";
setTimeout(() => {
document.getElementById("load-desc").textContent="Retrying...";
this.initwebsocket().then(() => {
this.loaduser();
this.init();
document.getElementById("loading").classList.add("doneloading");
document.getElementById("loading").classList.remove("loading");
console.log("done loading");
});
}, 200 + (this.errorBackoff*2800));
} else document.getElementById("load-desc").textContent="Unable to connect to the Spacebar server. Please try logging out and back in.";
});
await promise;
return;
}
handleEvent(temp){
console.debug(temp);
if (temp.s) this.lastSequence=temp.s; if (temp.s) this.lastSequence=temp.s;
if(temp.op==0){ if(temp.op==0){
switch(temp.t){ switch(temp.t){
@@ -159,7 +226,6 @@ class Localuser{
case "READY": case "READY":
this.gottenReady(temp as readyjson); this.gottenReady(temp as readyjson);
this.genusersettings(); this.genusersettings();
returny();
break; break;
case "MESSAGE_UPDATE": case "MESSAGE_UPDATE":
const message=SnowFlake.getSnowFlakeFromID(temp.d.id,Message).getObject(); const message=SnowFlake.getSnowFlakeFromID(temp.d.id,Message).getObject();
@@ -219,39 +285,6 @@ class Localuser{
this.ws.send(JSON.stringify({op:1,d:this.lastSequence})) this.ws.send(JSON.stringify({op:1,d:this.lastSequence}))
},temp.d.heartbeat_interval) },temp.d.heartbeat_interval)
} }
});
this.ws.addEventListener("close", event => {
console.log("WebSocket closed with code " + event.code);
if (this.wsinterval) clearInterval(this.wsinterval);
this.unload();
document.getElementById("loading").classList.remove("doneloading");
document.getElementById("loading").classList.add("loading");
if (((event.code>1000 && event.code<1016) || wsCodesRetry.has(event.code))) {
if (this.connectionSucceed!==0 && Date.now()>this.connectionSucceed+20000) this.errorBackoff=0;
else this.errorBackoff++;
this.connectionSucceed=0;
document.getElementById("load-desc").innerHTML="Unable to connect to the Spacebar server, retrying in <b>" + Math.round(0.2 + (this.errorBackoff*2.8)) + "</b> seconds...";
setTimeout(() => {
document.getElementById("load-desc").textContent="Retrying...";
this.initwebsocket().then(() => {
this.loaduser();
this.init();
document.getElementById("loading").classList.add("doneloading");
document.getElementById("loading").classList.remove("loading");
console.log("done loading");
});
}, 200 + (this.errorBackoff*2800));
} else document.getElementById("load-desc").textContent="Unable to connect to the Spacebar server. Please try logging out and back in.";
});
await promise;
return;
} }
resolveChannelFromID(ID:string):Channel{ resolveChannelFromID(ID:string):Channel{
let resolve=this.guilds.find(guild => guild.channelids[ID]); let resolve=this.guilds.find(guild => guild.channelids[ID]);

View File

@@ -92,7 +92,9 @@ class Member{
} }
return memb; return memb;
} }
const promoise= fetch(guild.info.api.toString()+"/users/"+id+"/profile?with_mutual_guilds=true&with_mutual_friends_count=true&guild_id="+guild.snowflake,{headers:guild.headers}).then(_=>_.json()).then(json=>{ const prom1= fetch(guild.info.api.toString()+"/users/"+id+"/profile?with_mutual_guilds=true&with_mutual_friends_count=true&guild_id="+guild.snowflake,{headers:guild.headers})
prom1.catch(_=>{console.log(_)})
const promoise=prom1.then(_=>_.json()).then(json=>{
const memb=new Member(json,guild); const memb=new Member(json,guild);
Member.already[guild.id][id]=memb; Member.already[guild.id][id]=memb;
console.log("resolved") console.log("resolved")

View File

@@ -944,6 +944,7 @@ span {
text-align: center; text-align: center;
transition: transform .2s; transition: transform .2s;
vertical-align: middle; vertical-align: middle;
display: flex;
} }
#centerdiv { #centerdiv {
@@ -951,6 +952,8 @@ span {
top: 50%; top: 50%;
left: 50%; left: 50%;
transform: translate(-50%, -50%); transform: translate(-50%, -50%);
display: flex;
flex-direction: column;
} }
#loading.doneloading { #loading.doneloading {