Files
custom_discordjs/structures/client/client.js
T
2023-03-20 12:45:20 -04:00

165 lines
4.6 KiB
JavaScript

const opts = require('./clientOpts.js');
const gateWayIntents = require('../gateway/intents.js');
const gateWayEvents = require('../gateway/dispatch.js');
var WebSocketClient = require('websocket').client;
const WebSocketConnection = require('websocket').connection;
const handleResponses = require('./handleEvents.js');
const { EventEmitter } = require('events');
const axios = require('axios');
const { exit } = require('process');
class Client extends EventEmitter {
/** @type {WebSocketClient} */
ws;
/** @type {Number} */
heartBeatInterval;
/** @type {Array<opts.intents>} */
gwintents;
/** @type {String} */
#token;
/** @type {String} */
id;
/** @type {WebSocketConnection}*/
connection;
/** @type {Object} */
user_settings;
/** @type {Object} */
user_profile;
/**
* @param {opts} input
*/
constructor(input) {
super();
this.gwintents = input.intents;
}
async #heartbeat(hbInt, hbSequence) {
const toSend = JSON.stringify({ op: 1, d: 0 });
this.connection.send((toSend));
setInterval(() => {
this.connection.send(toSend);
}, hbInt);
}
/**
* @param {Number} hbint
*/
async #startHeartBeat(hbint)
{
this.heartBeatInterval = hbint;
console.log("INTERVAL SET TO: " + this.heartBeatInterval);
//Get the user intents
let iCount = 0;
for (let i of this.gwintents) {
iCount += (i) ? i : 0;
}
var idObj = {
op: 2,
d: {
token: this.#token.replace("Bot ", ""),
intents: iCount, //61440,
properties: {
os: "linux",
browser: "ion_",
device: "my_library"
}
}
};
this.connection.send(JSON.stringify(idObj));
this.#heartbeat(hbint);
}
messageRecieved(msg) {
this.emit("messageRecieved", msg);
}
ready() {
this.emit('ready');
}
customError(err) {
this.emit('error', err);
}
interactionRecieved(interaction) {
this.emit('interactionRecieved', interaction);
}
/**
* @param {String} token
*/
async login(token, isUser = false) {
if (!isUser) token = "Bot " + token;
return new Promise((resolve, reject) => {
this.ws = new WebSocketClient({maxReceivedFrameSize: Infinity});
this.#token = token;
this.ws.on('connect', async (connection) => {
connection.on('message', async (msg) => {
const data = JSON.parse(msg.utf8Data);
const response = await handleResponses(data, token, this.id);
if (response.op == 10) { this.#startHeartBeat(response.heartBeat, token); }
else if (response.op == 0) {
if (response.t == gateWayEvents.Ready) {
this.user_profile = response.profile;
this.user_settings = response.config;
this.id = response.profile.id;
this.ready();
}
else if (response.t == gateWayEvents.MessageCreate) {
if (data["d"]["author"]["id"] != this.user_profile.id){
this.messageRecieved(response.message);
}
}
else if (response.t == gateWayEvents.InteractionCreate) {
if (data["d"]["user"]["id"] != this.user_profile.id) {
this.interactionRecieved(response.interaction);
}
}
else console.log(response.t);
} else {
console.log(response.t);
}
});
connection.on('close', (code, desc) => {
console.log(`CONNECTION CLOSED WITH CODE ${code}\nREASON:\n ${desc}`);
exit(1);
});
connection.on('error', (err) => {
reject(err);
});
this.connection = connection;
});
this.ws.on('connectFailed', (err) => { reject(err); });
this.ws.connect("wss://gateway.discord.gg/?v=10&encoding=json");
});
}
}
//All client properties will be re-routed through this export
module.exports = { Client, gateWayIntents }