adding comments

This commit is contained in:
raven 2023-01-15 22:59:58 -05:00
parent f9c2f44d2e
commit 7511ad9f11

View File

@ -1,12 +1,13 @@
const blessed = require('neo-blessed');
// Require the needed libs // Require the needed libs
const blessed = require('neo-blessed');
const Hyperswarm = require('hyperswarm') const Hyperswarm = require('hyperswarm')
const crypto = require('hypercore-crypto') const crypto = require('hypercore-crypto')
const b4a = require('b4a') const b4a = require('b4a')
const readline = require('readline') const readline = require('readline')
const fs = require("fs"); const fs = require("fs");
// Import our command // Import our commands - You must add this line.
// Otherwise there is no access to the command functions
const { login } = require('./commands/login'); const { login } = require('./commands/login');
const { execute } = require('./commands/exec'); const { execute } = require('./commands/exec');
const { stop } = require('./commands/stop'); const { stop } = require('./commands/stop');
@ -16,37 +17,38 @@ const { stats } = require('./commands/stats');
const { changeDir } = require('./commands/cd'); const { changeDir } = require('./commands/cd');
const { AIRequest } = require('./commands/AI'); const { AIRequest } = require('./commands/AI');
// Generate a random number, this is used when generating a anon name
let rand = Math.floor(Math.random() * 99999).toString(); let rand = Math.floor(Math.random() * 99999).toString();
// Storage for our clients information
let USERPWD = "/" let USERPWD = "/"
let DAPI_KEY let DAPI_KEY
let LOGGEDIN = false let LOGGEDIN = false
let MYKEY = [] let MYKEY = []
let conns = [] let conns = []
let connectedUsers = []; let connectedUsers = [];
let USERNAME = ["annon" + rand] let USERNAME = ["anon" + rand]
let DISCORD_USERID = [] let DISCORD_USERID = []
// Sleep function used when closing connections.
function sleep(ms) { function sleep(ms) {
return new Promise((resolve) => { return new Promise((resolve) => {
setTimeout(resolve, ms); setTimeout(resolve, ms);
}); });
} }
// Adding a user to a peer and user MAP to keep track of peers.
function addUser(user, peerId) { function addUser(user, peerId) {
connectedUsers.push({ name: user, peerId: peerId }); connectedUsers.push({ name: user, peerId: peerId });
sidebarBox.setContent(connectedUsers.map(user => `${user.peerId}`).join("\n")); sidebarBox.setContent(connectedUsers.map(user => `${user.peerId}`).join("\n"));
screen.render(); screen.render();
} }
// Removing a user to a peer and user MAP to keep track of peers.
// TODO: Get this to work properly
function removeUser(peerId) { function removeUser(peerId) {
connectedUsers = connectedUsers.filter(user => user.peerId !== peerId); connectedUsers = connectedUsers.filter(user => user.peerId !== peerId);
sidebarBox.setContent("Peers since connection: \n" + connectedUsers.map(user => `${user.peerId}`).join("\n")); sidebarBox.setContent("Peers since connection: \n" + connectedUsers.map(user => `${user.peerId}`).join("\n"));
screen.render(); screen.render();
} }
// Create the screen // Create the screen for blessed
const screen = blessed.screen({ const screen = blessed.screen({
smartCSR: true, smartCSR: true,
fastCSR: true fastCSR: true
@ -67,22 +69,24 @@ screen.key(['escape', 'q', 'C-c'], function (ch, key) {
}); });
// Return to chat // When users press ESC, you may press t or enter to go back
// to the chat input box
screen.key(['t', 'enter'], function (ch, key) { screen.key(['t', 'enter'], function (ch, key) {
stdinBox.focus(); stdinBox.focus();
}); });
// Return to chat // When users press ESC, you may press p to go to the mainbox
screen.key(['p', 'enter'], function (ch, key) { // This lets users scroll the chat feed.
screen.key(['p'], function (ch, key) {
sidebarBox.focus(); sidebarBox.focus();
}); });
// Debug ONLY - Use this to capture keypress events
// Debug ONLY
// screen.on('keypress', function(ch, key){ // screen.on('keypress', function(ch, key){
// console.log(JSON.stringify(key)); // console.log(JSON.stringify(key));
// }); // });
// Creating the mainbox
let mainBox = blessed.box({ let mainBox = blessed.box({
parent: screen, parent: screen,
top: 0, top: 0,
@ -107,6 +111,7 @@ let mainBox = blessed.box({
} }
}); });
// A function to update the mainbox scroll per message.
async function updateScroll() { async function updateScroll() {
mainBox.scrollTo(mainBox.getScrollHeight()); mainBox.scrollTo(mainBox.getScrollHeight());
@ -133,7 +138,7 @@ const stdinBox = blessed.textbox({
}); });
// Sidebar to display all peers that have been seen
let sidebarBox = blessed.box({ let sidebarBox = blessed.box({
parent: screen, parent: screen,
top: '0', top: '0',
@ -158,9 +163,11 @@ let sidebarBox = blessed.box({
} }
}); });
// Setting the sidebar label.
sidebarBox.setLabel("Peers since connection: 0") sidebarBox.setLabel("Peers since connection: 0")
// Replacing the console.log function with our own
// This sends all console.log events to the main window.
const originalLog = console.log; const originalLog = console.log;
console.log = (...args) => { console.log = (...args) => {
mainBox.setContent(mainBox.getContent() + `\n${args.join(' ')}`); mainBox.setContent(mainBox.getContent() + `\n${args.join(' ')}`);
@ -170,18 +177,19 @@ console.log = (...args) => {
} }
// Generate a random public key // Generate a random public key
const publicKey = crypto.randomBytes(32) const publicKey = crypto.randomBytes(32)
// Create the swarm and pass in the public key // Create the swarm and pass in the public key
const swarm = new Hyperswarm() const swarm = new Hyperswarm()
// Set up our commands
const commandDir = __dirname + '/commands/'; const commandDir = __dirname + '/commands/';
const commandFiles = fs.readdirSync(commandDir); const commandFiles = fs.readdirSync(commandDir);
const commands = {} const commands = {}
// For each command lets add it to a var
for (const file of commandFiles) { for (const file of commandFiles) {
const commandName = file.split(".")[0] const commandName = file.split(".")[0]
require(`${commandDir}/${file}`) require(`${commandDir}/${file}`)
@ -190,6 +198,7 @@ for (const file of commandFiles) {
} }
// The command handler
async function handleCommand(input) { async function handleCommand(input) {
if (input.startsWith("!") || input.startsWith(">")) { if (input.startsWith("!") || input.startsWith(">")) {
const command = input.split(" ") const command = input.split(" ")
@ -265,28 +274,39 @@ swarm.on('connection', conn => {
process.exit() process.exit()
}) })
// Remote Connection Name
const name = b4a.toString(conn.remotePublicKey, 'hex') const name = b4a.toString(conn.remotePublicKey, 'hex')
console.log(`* got a connection from ${name} (${USERNAME[0]}) *`) console.log(`* got a connection from ${name} (${USERNAME[0]}) *`)
// Add the user to the MAP
addUser(USERNAME[0], name) addUser(USERNAME[0], name)
// Update the sidebar
sidebarBox.setLabel("Peers since connection: " + connectedUsers.length) sidebarBox.setLabel("Peers since connection: " + connectedUsers.length)
// Render the changes
screen.render() screen.render()
// Add the connection to the conn list
conns.push(conn) conns.push(conn)
// IF the connection is closed, remove a connection
conn.once('close', () => conns.splice(conns.indexOf(conn), 1)) conn.once('close', () => conns.splice(conns.indexOf(conn), 1))
// Handle data as it comes in from the peer stream
conn.on('data', data => { conn.on('data', data => {
// If data shows that a peer has left, lets handle it and remove their key
if (data.toString().startsWith('CLOSED:')) { if (data.toString().startsWith('CLOSED:')) {
// Extract the key from the message string // Extract the key from the message string
const key = data.toString().split(':')[1].trim(); const key = data.toString().split(':')[1].trim();
removeUser(key) removeUser(key)
console.log(`Removing peer ${key}`); console.log(`Removing peer ${key}`);
(async () => {
// Wait 5 seconds, remove the peer from the connections
// This stops timeouts which crashes all peers.
(async () => {
await sleep(5000) await sleep(5000)
conns = conns.filter(c => c !== conn); conns = conns.filter(c => c !== conn);
conn.destroy(); conn.destroy();
})(); })();
} else { } else {
// If there is no actions detected, update chat.
console.log(`${data}`) console.log(`${data}`)
} }
// Use the USERNAME if it has been set, otherwise use the public key // Use the USERNAME if it has been set, otherwise use the public key
@ -301,7 +321,7 @@ swarm.on('error', (err) => {
// Join a common topic // Join a common topic
const topic = process.argv[2] ? b4a.from(process.argv[2], 'hex') : crypto.randomBytes(32) const topic = process.argv[2] ? b4a.from(process.argv[2], 'hex') : crypto.randomBytes(32)
// Join the topic with a timeout
setTimeout(() => { setTimeout(() => {
const discovery = swarm.join(topic, { const discovery = swarm.join(topic, {
lookup: true, lookup: true,
@ -343,5 +363,4 @@ stdinBox.on('submit', (input) => {
// }, 1000); // }, 1000);
stdinBox.focus(); stdinBox.focus();
// Render the screen // Render the screen
screen.render() screen.render()