Rework ChatBot #1

Merged
snxraven merged 7 commits from MiTask/LinkUp-P2P-Chat:main into main 2024-06-08 16:19:09 -04:00
Showing only changes of commit 114875cddb - Show all commits

38
app.js
View File

@ -6,6 +6,7 @@ import Localdrive from 'localdrive';
import fs from 'fs'; import fs from 'fs';
import Hyperdrive from 'hyperdrive'; import Hyperdrive from 'hyperdrive';
import Corestore from 'corestore'; import Corestore from 'corestore';
import { EventEmitter } from 'events';
const store = new Corestore('./storage'); const store = new Corestore('./storage');
const drive = new Hyperdrive(store); const drive = new Hyperdrive(store);
@ -18,6 +19,7 @@ let userAvatar = '';
let registeredUsers = JSON.parse(localStorage.getItem('registeredUsers')) || {}; let registeredUsers = JSON.parse(localStorage.getItem('registeredUsers')) || {};
let peerCount = 0; let peerCount = 0;
let currentRoom = null; let currentRoom = null;
const eventEmitter = new EventEmitter();
async function initialize() { async function initialize() {
swarm = new Hyperswarm(); swarm = new Hyperswarm();
@ -64,6 +66,17 @@ async function initialize() {
registerDiv.classList.remove('hidden'); registerDiv.classList.remove('hidden');
} }
eventEmitter.on('onMessage', async (messageObj) => {
if (messageObj.type === 'icon') {
const username = messageObj.username;
const avatarBuffer = Buffer.from(messageObj.avatar, 'base64');
await drive.put(`/icons/${username}.png`, avatarBuffer);
updateIcon(username, avatarBuffer);
} else {
onMessageAdded(messageObj.name, messageObj.message, messageObj.avatar);
}
});
swarm.on('connection', async (connection, info) => { swarm.on('connection', async (connection, info) => {
peerCount++; peerCount++;
updatePeerCount(); updatePeerCount();
@ -80,16 +93,9 @@ async function initialize() {
connection.write(iconMessage); connection.write(iconMessage);
} }
connection.on('data', async (data) => { connection.on('data', (data) => {
const messageObj = JSON.parse(data.toString()); const messageObj = JSON.parse(data.toString());
if (messageObj.type === 'icon') { eventEmitter.emit('onMessage', messageObj);
const username = messageObj.username;
const avatarBuffer = Buffer.from(messageObj.avatar, 'base64');
await drive.put(`/icons/${username}.png`, avatarBuffer);
updateIcon(username, avatarBuffer);
} else {
onMessageAdded(messageObj.name, messageObj.message, messageObj.avatar);
}
}); });
connection.on('close', () => { connection.on('close', () => {
@ -120,9 +126,9 @@ function registerUser(e) {
const avatarFile = document.querySelector('#avatar-file').files[0]; const avatarFile = document.querySelector('#avatar-file').files[0];
if (avatarFile) { if (avatarFile) {
const reader = new FileReader(); const reader = new FileReader();
reader.onload = (event) => { reader.onload = async (event) => {
const buffer = new Uint8Array(event.target.result); const buffer = new Uint8Array(event.target.result);
drive.put(`/icons/${regUsername}.png`, buffer); await drive.put(`/icons/${regUsername}.png`, buffer);
userAvatar = `http://localhost:1337/icons/${regUsername}.png`; // Set the correct URL userAvatar = `http://localhost:1337/icons/${regUsername}.png`; // Set the correct URL
registeredUsers[regUsername] = userAvatar; registeredUsers[regUsername] = userAvatar;
localStorage.setItem('registeredUsers', JSON.stringify(registeredUsers)); localStorage.setItem('registeredUsers', JSON.stringify(registeredUsers));
@ -237,9 +243,9 @@ function sendMessage(e) {
} }
function updatePeerCount() { function updatePeerCount() {
const peerCountElement = document.querySelector('#peer-count'); const peerCountElement = document.querySelector('#peers-count');
if (peerCountElement) { if (peerCountElement) {
peerCountElement.textContent = `Connected Peers: ${peerCount}`; peerCountElement.textContent = peerCount; // Display the actual peer count
} }
} }
@ -283,11 +289,11 @@ function truncateHash(hash) {
} }
async function updateIcon(username, avatarBuffer) { async function updateIcon(username, avatarBuffer) {
// Update the icon in the local HTML if necessary
// This can be adjusted as per your needs
const userIcon = document.querySelector(`img[src*="${username}.png"]`); const userIcon = document.querySelector(`img[src*="${username}.png"]`);
if (userIcon) { if (userIcon) {
userIcon.src = `http://localhost:1337/icons/${username}.png`; // Ensure the URL is correct const avatarBlob = new Blob([avatarBuffer], { type: 'image/png' });
const avatarUrl = URL.createObjectURL(avatarBlob);
userIcon.src = avatarUrl;
} }
} }