Compare commits
2 Commits
proposal-6
...
proposal-e
| Author | SHA1 | Date | |
|---|---|---|---|
|
c89c6f95a6 | ||
|
|
78d5352a48 |
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@@ -0,0 +1 @@
|
||||
node_modules/
|
||||
85
README.md
85
README.md
@@ -1,57 +1,28 @@
|
||||
# P2P Poll App
|
||||
|
||||
A peer-to-peer polling application where users can create polls, add options, and vote in real-time without a central server.
|
||||
|
||||
## Features
|
||||
|
||||
- **Real-time P2P voting** using WebRTC via PeerJS
|
||||
- **Dynamic option management** - add/remove options during polling
|
||||
- **Duplicate vote prevention** - one vote per user
|
||||
- **Automatic data synchronization** across all connected peers
|
||||
- **Local storage persistence** for poll recovery
|
||||
- **Responsive design** works on desktop and mobile
|
||||
- **No server required** - uses PeerJS free signaling service
|
||||
|
||||
## How to Use
|
||||
|
||||
1. **Open the app** in your browser (open `index.html`)
|
||||
2. **First user (Host)**: Leave the Peer ID field empty and click "Connect to Host"
|
||||
3. **Copy your Peer ID** using the "Copy Your Peer ID" button
|
||||
4. **Share your Peer ID** with other users (via chat, email, etc.)
|
||||
5. **Other users**: Paste the host's Peer ID and click "Connect to Host"
|
||||
6. **Create a poll** with question and options
|
||||
7. **Vote** by clicking on options (one vote per person)
|
||||
8. **Watch results** update in real-time across all devices
|
||||
|
||||
## Technical Details
|
||||
|
||||
- **P2P Library**: PeerJS (WebRTC-based)
|
||||
- **Frontend**: Vanilla JavaScript with modern CSS
|
||||
- **Data Sync**: Custom conflict resolution for concurrent operations
|
||||
- **Storage**: localStorage for basic persistence
|
||||
- **Network**: Full mesh topology where each peer connects to all others
|
||||
|
||||
## File Structure
|
||||
|
||||
```
|
||||
├── index.html # Main application
|
||||
├── css/
|
||||
│ └── styles.css # Application styling
|
||||
└── js/
|
||||
├── app.js # Main application logic
|
||||
├── peer-manager.js # P2P connection handling
|
||||
├── poll-manager.js # Poll data and sync logic
|
||||
└── ui-controller.js # UI interactions
|
||||
```
|
||||
|
||||
## Browser Support
|
||||
|
||||
Requires modern browsers with WebRTC support:
|
||||
- Chrome 23+
|
||||
- Firefox 22+
|
||||
- Safari 11+
|
||||
- Edge 79+
|
||||
|
||||
## Development
|
||||
|
||||
Simply open `index.html` in a browser - no build process required. For testing with multiple peers, open the app in multiple browser tabs or windows.
|
||||
# P2P Poll App
|
||||
|
||||
## Install npm packages
|
||||
```
|
||||
npm init -y
|
||||
npm install
|
||||
```
|
||||
|
||||
Note: the frontend obtains the packages `yjs` and `y-websocket` dynamically.
|
||||
|
||||
## Run backend
|
||||
```
|
||||
node backend.js
|
||||
```
|
||||
|
||||
### Note on Yjs suggestion
|
||||
Note that the following server setup is suggested in the Yjs docs (https://docs.yjs.dev/ecosystem/connection-provider/y-websocket):
|
||||
```
|
||||
npm install y-websocket-server
|
||||
HOST=localhost PORT=1000 npx y-websocket
|
||||
```
|
||||
|
||||
However, across a range of npm versions, this does not work.
|
||||
Nevertheless, the essential code from the `y-websocket-server` package is used here as provided in `utils.js`.
|
||||
|
||||
## Run frontend
|
||||
Open "frontend.html" in browser.
|
||||
|
||||
|
||||
31
backend.js
Normal file
31
backend.js
Normal file
@@ -0,0 +1,31 @@
|
||||
import { WebSocketServer } from 'ws';
|
||||
import { setupWSConnection } from './utils.js'
|
||||
|
||||
// Create a WebSocket server
|
||||
const WS_PORT = 8080;
|
||||
const wss = new WebSocketServer({ port: WS_PORT });
|
||||
|
||||
console.log('WebSocket server is running on ws://localhost:' + String(WS_PORT));
|
||||
|
||||
// Connection event handler
|
||||
wss.on('connection', setupWSConnection);
|
||||
/*
|
||||
wss.on('connection', (ws) => {
|
||||
console.log('Client connected');
|
||||
|
||||
// Message event handler
|
||||
ws.on('message', (message) => {
|
||||
let msg_str = String(message);
|
||||
console.log("Received: " + msg_str);
|
||||
|
||||
// If this is a text or state message (no Yjs logic) - echo the message back to the client
|
||||
if (msg_str.startsWith("TEXT_MESSAGE") | msg_str.startsWith("STATE_MESSAGE")) {
|
||||
ws.send(msg_str);
|
||||
}
|
||||
});
|
||||
|
||||
// Close event handler
|
||||
ws.on('close', () => {
|
||||
console.log('Client disconnected');
|
||||
});
|
||||
});*/
|
||||
665
css/styles.css
665
css/styles.css
@@ -1,665 +0,0 @@
|
||||
* {
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
box-sizing: border-box;
|
||||
}
|
||||
|
||||
body {
|
||||
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
|
||||
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
|
||||
min-height: 100vh;
|
||||
color: #333;
|
||||
}
|
||||
|
||||
.container {
|
||||
max-width: 1200px;
|
||||
margin: 0 auto;
|
||||
padding: 20px;
|
||||
}
|
||||
|
||||
.main-content {
|
||||
display: grid;
|
||||
grid-template-columns: 280px 1fr;
|
||||
gap: 20px;
|
||||
margin-top: 20px;
|
||||
}
|
||||
|
||||
.sidebar {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 20px;
|
||||
}
|
||||
|
||||
.sidebar-section {
|
||||
background: white;
|
||||
border-radius: 12px;
|
||||
padding: 20px;
|
||||
box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
|
||||
}
|
||||
|
||||
.sidebar-section h3 {
|
||||
margin: 0 0 15px 0;
|
||||
color: #333;
|
||||
font-size: 1.1rem;
|
||||
}
|
||||
|
||||
.sidebar-hint {
|
||||
margin-bottom: 10px;
|
||||
}
|
||||
|
||||
.hint-text {
|
||||
font-size: 0.8rem;
|
||||
color: #6b7280;
|
||||
font-style: italic;
|
||||
text-align: center;
|
||||
padding: 8px;
|
||||
background: #f9fafb;
|
||||
border-radius: 6px;
|
||||
border: 1px solid #e5e7eb;
|
||||
}
|
||||
|
||||
.polls-list {
|
||||
max-height: 300px;
|
||||
overflow-y: auto;
|
||||
}
|
||||
|
||||
.poll-item {
|
||||
padding: 12px;
|
||||
border: 1px solid #e5e7eb;
|
||||
border-radius: 8px;
|
||||
margin-bottom: 8px;
|
||||
cursor: pointer;
|
||||
transition: all 0.3s;
|
||||
position: relative;
|
||||
}
|
||||
|
||||
.poll-item:hover {
|
||||
border-color: #667eea;
|
||||
background: #f9fafb;
|
||||
transform: translateY(-1px);
|
||||
}
|
||||
|
||||
.poll-item.active {
|
||||
border-color: #667eea;
|
||||
background: #ede9fe;
|
||||
}
|
||||
|
||||
.poll-item-title {
|
||||
font-weight: 500;
|
||||
color: #333;
|
||||
margin-bottom: 4px;
|
||||
}
|
||||
|
||||
.poll-item-meta {
|
||||
font-size: 0.8rem;
|
||||
color: #6b7280;
|
||||
}
|
||||
|
||||
.poll-item::after {
|
||||
content: ">";
|
||||
position: absolute;
|
||||
right: 12px;
|
||||
top: 50%;
|
||||
transform: translateY(-50%);
|
||||
color: #9ca3af;
|
||||
font-size: 0.8rem;
|
||||
opacity: 0;
|
||||
transition: opacity 0.3s;
|
||||
}
|
||||
|
||||
.poll-item:hover::after {
|
||||
opacity: 1;
|
||||
}
|
||||
|
||||
.no-polls {
|
||||
color: #9ca3af;
|
||||
font-style: italic;
|
||||
text-align: center;
|
||||
padding: 20px 0;
|
||||
}
|
||||
|
||||
.peers-info {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 10px;
|
||||
}
|
||||
|
||||
.peers-status {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
font-size: 0.9rem;
|
||||
}
|
||||
|
||||
.status-text {
|
||||
padding: 2px 8px;
|
||||
border-radius: 12px;
|
||||
font-size: 0.8rem;
|
||||
font-weight: 500;
|
||||
}
|
||||
|
||||
.status-text.connected {
|
||||
background: #dcfce7;
|
||||
color: #16a34a;
|
||||
}
|
||||
|
||||
.status-text.disconnected {
|
||||
background: #fef2f2;
|
||||
color: #dc2626;
|
||||
}
|
||||
|
||||
.status-text.connecting {
|
||||
background: #fef3c7;
|
||||
color: #d97706;
|
||||
}
|
||||
|
||||
.peers-sidebar {
|
||||
list-style: none;
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
max-height: 150px;
|
||||
overflow-y: auto;
|
||||
}
|
||||
|
||||
.peers-sidebar li {
|
||||
padding: 6px 0;
|
||||
border-bottom: 1px solid #f3f4f6;
|
||||
font-size: 0.9rem;
|
||||
color: #6b7280;
|
||||
}
|
||||
|
||||
.peers-sidebar li:last-child {
|
||||
border-bottom: none;
|
||||
}
|
||||
|
||||
header {
|
||||
background: white;
|
||||
border-radius: 12px;
|
||||
padding: 20px;
|
||||
margin-bottom: 20px;
|
||||
box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
}
|
||||
|
||||
h1 {
|
||||
color: #667eea;
|
||||
font-size: 2rem;
|
||||
}
|
||||
|
||||
.connection-status {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: 10px;
|
||||
font-size: 0.9rem;
|
||||
}
|
||||
|
||||
#peer-id {
|
||||
background: #f0f0f0;
|
||||
padding: 5px 10px;
|
||||
border-radius: 6px;
|
||||
font-family: monospace;
|
||||
max-width: 150px;
|
||||
overflow: hidden;
|
||||
text-overflow: ellipsis;
|
||||
}
|
||||
|
||||
.status-connected {
|
||||
color: #10b981;
|
||||
}
|
||||
|
||||
.status-disconnected {
|
||||
color: #ef4444;
|
||||
}
|
||||
|
||||
.status-connecting {
|
||||
color: #f59e0b;
|
||||
}
|
||||
|
||||
.section {
|
||||
background: white;
|
||||
border-radius: 12px;
|
||||
padding: 20px;
|
||||
margin-bottom: 20px;
|
||||
box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
|
||||
}
|
||||
|
||||
.section.hidden {
|
||||
display: none;
|
||||
}
|
||||
|
||||
h2 {
|
||||
color: #333;
|
||||
margin-bottom: 20px;
|
||||
font-size: 1.5rem;
|
||||
}
|
||||
|
||||
h3 {
|
||||
color: #666;
|
||||
margin-bottom: 15px;
|
||||
font-size: 1.1rem;
|
||||
}
|
||||
|
||||
.connection-info {
|
||||
background: #f0f9ff;
|
||||
padding: 15px;
|
||||
border-radius: 8px;
|
||||
margin-bottom: 20px;
|
||||
border-left: 4px solid #3b82f6;
|
||||
}
|
||||
|
||||
.connection-info p {
|
||||
margin: 5px 0;
|
||||
font-size: 0.9rem;
|
||||
color: #1e40af;
|
||||
}
|
||||
|
||||
.connection-controls {
|
||||
display: flex;
|
||||
gap: 10px;
|
||||
margin-bottom: 20px;
|
||||
flex-wrap: wrap;
|
||||
}
|
||||
|
||||
input[type="text"] {
|
||||
flex: 1;
|
||||
min-width: 200px;
|
||||
padding: 12px;
|
||||
border: 2px solid #e5e7eb;
|
||||
border-radius: 8px;
|
||||
font-size: 1rem;
|
||||
transition: border-color 0.3s;
|
||||
}
|
||||
|
||||
input[type="text"]:focus {
|
||||
outline: none;
|
||||
border-color: #667eea;
|
||||
}
|
||||
|
||||
button {
|
||||
background: #667eea;
|
||||
color: white;
|
||||
border: none;
|
||||
padding: 12px 20px;
|
||||
border-radius: 8px;
|
||||
font-size: 1rem;
|
||||
cursor: pointer;
|
||||
transition: all 0.3s;
|
||||
}
|
||||
|
||||
button:hover {
|
||||
background: #5a67d8;
|
||||
transform: translateY(-1px);
|
||||
}
|
||||
|
||||
button:disabled {
|
||||
background: #9ca3af;
|
||||
cursor: not-allowed;
|
||||
transform: none;
|
||||
}
|
||||
|
||||
button.small-btn {
|
||||
padding: 8px 12px;
|
||||
font-size: 0.9rem;
|
||||
}
|
||||
|
||||
button.danger {
|
||||
background: #ef4444;
|
||||
}
|
||||
|
||||
button.danger:hover {
|
||||
background: #dc2626;
|
||||
}
|
||||
|
||||
|
||||
.poll-form {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 15px;
|
||||
}
|
||||
|
||||
#options-container {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 10px;
|
||||
}
|
||||
|
||||
.option-input {
|
||||
display: flex;
|
||||
gap: 10px;
|
||||
align-items: center;
|
||||
}
|
||||
|
||||
.option-text {
|
||||
flex: 1;
|
||||
}
|
||||
|
||||
.remove-option-btn {
|
||||
background: #ef4444;
|
||||
color: white;
|
||||
border: none;
|
||||
padding: 4px 8px;
|
||||
border-radius: 4px;
|
||||
cursor: pointer;
|
||||
font-size: 0.8rem;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
min-width: 60px;
|
||||
}
|
||||
|
||||
.remove-option-btn.hidden {
|
||||
display: none;
|
||||
}
|
||||
|
||||
.poll-header {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
margin-bottom: 20px;
|
||||
}
|
||||
|
||||
.vote-hint {
|
||||
margin-bottom: 20px;
|
||||
}
|
||||
|
||||
.vote-hint .hint-text {
|
||||
font-size: 0.9rem;
|
||||
color: #6b7280;
|
||||
text-align: center;
|
||||
padding: 10px;
|
||||
background: #f0f9ff;
|
||||
border-radius: 8px;
|
||||
border: 1px solid #bfdbfe;
|
||||
}
|
||||
|
||||
.options-list {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 15px;
|
||||
}
|
||||
|
||||
.poll-option {
|
||||
background: #f9fafb;
|
||||
padding: 15px;
|
||||
border-radius: 8px;
|
||||
border: 2px solid transparent;
|
||||
transition: all 0.3s;
|
||||
cursor: pointer;
|
||||
position: relative;
|
||||
}
|
||||
|
||||
.poll-option:hover {
|
||||
border-color: #667eea;
|
||||
transform: translateY(-1px);
|
||||
}
|
||||
|
||||
.poll-option.voted {
|
||||
border-color: #10b981;
|
||||
background: #ecfdf5;
|
||||
}
|
||||
|
||||
.poll-option.voted::before {
|
||||
content: "✓";
|
||||
position: absolute;
|
||||
top: 10px;
|
||||
right: 10px;
|
||||
background: #10b981;
|
||||
color: white;
|
||||
width: 20px;
|
||||
height: 20px;
|
||||
border-radius: 50%;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
font-size: 12px;
|
||||
font-weight: bold;
|
||||
}
|
||||
|
||||
.poll-option.change-vote {
|
||||
border-color: #f59e0b;
|
||||
background: #fffbeb;
|
||||
}
|
||||
|
||||
.poll-option.change-vote:hover {
|
||||
border-color: #d97706;
|
||||
}
|
||||
|
||||
.option-header {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
margin-bottom: 10px;
|
||||
}
|
||||
|
||||
.option-text {
|
||||
font-weight: 500;
|
||||
color: #333;
|
||||
}
|
||||
|
||||
.vote-count {
|
||||
background: #667eea;
|
||||
color: white;
|
||||
padding: 4px 8px;
|
||||
border-radius: 12px;
|
||||
font-size: 0.8rem;
|
||||
}
|
||||
|
||||
.vote-bar {
|
||||
background: #e5e7eb;
|
||||
height: 8px;
|
||||
border-radius: 4px;
|
||||
overflow: hidden;
|
||||
}
|
||||
|
||||
.vote-fill {
|
||||
background: #667eea;
|
||||
height: 100%;
|
||||
transition: width 0.3s ease;
|
||||
}
|
||||
|
||||
.poll-actions {
|
||||
display: flex;
|
||||
gap: 10px;
|
||||
margin-top: 20px;
|
||||
}
|
||||
|
||||
.overlay {
|
||||
position: fixed;
|
||||
top: 0;
|
||||
left: 0;
|
||||
right: 0;
|
||||
bottom: 0;
|
||||
background: rgba(0, 0, 0, 0.5);
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
z-index: 1000;
|
||||
}
|
||||
|
||||
.overlay.hidden {
|
||||
display: none;
|
||||
}
|
||||
|
||||
.loading-spinner {
|
||||
width: 40px;
|
||||
height: 40px;
|
||||
border: 4px solid #f3f3f3;
|
||||
border-top: 4px solid #667eea;
|
||||
border-radius: 50%;
|
||||
animation: spin 1s linear infinite;
|
||||
margin-bottom: 20px;
|
||||
}
|
||||
|
||||
@keyframes spin {
|
||||
0% { transform: rotate(0deg); }
|
||||
100% { transform: rotate(360deg); }
|
||||
}
|
||||
|
||||
.notification {
|
||||
position: fixed;
|
||||
top: 20px;
|
||||
right: 20px;
|
||||
background: #333;
|
||||
color: white;
|
||||
padding: 15px 20px;
|
||||
border-radius: 8px;
|
||||
box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
|
||||
z-index: 1001;
|
||||
max-width: 300px;
|
||||
}
|
||||
|
||||
.notification.success {
|
||||
background: #10b981;
|
||||
}
|
||||
|
||||
.notification.error {
|
||||
background: #ef4444;
|
||||
}
|
||||
|
||||
.notification.hidden {
|
||||
display: none;
|
||||
}
|
||||
|
||||
/* Modal Styles */
|
||||
.modal {
|
||||
position: fixed;
|
||||
top: 0;
|
||||
left: 0;
|
||||
right: 0;
|
||||
bottom: 0;
|
||||
background: rgba(0, 0, 0, 0.5);
|
||||
display: flex;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
z-index: 1002;
|
||||
}
|
||||
|
||||
.modal.hidden {
|
||||
display: none;
|
||||
}
|
||||
|
||||
.modal-content {
|
||||
background: white;
|
||||
border-radius: 12px;
|
||||
width: 90%;
|
||||
max-width: 400px;
|
||||
box-shadow: 0 10px 25px rgba(0, 0, 0, 0.2);
|
||||
overflow: hidden;
|
||||
}
|
||||
|
||||
.modal-header {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
padding: 20px;
|
||||
border-bottom: 1px solid #e5e7eb;
|
||||
}
|
||||
|
||||
.modal-header h3 {
|
||||
margin: 0;
|
||||
color: #333;
|
||||
font-size: 1.2rem;
|
||||
}
|
||||
|
||||
.modal-close {
|
||||
background: none;
|
||||
border: none;
|
||||
font-size: 1.5rem;
|
||||
color: #6b7280;
|
||||
cursor: pointer;
|
||||
padding: 0;
|
||||
width: 30px;
|
||||
height: 30px;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
border-radius: 50%;
|
||||
transition: all 0.3s;
|
||||
}
|
||||
|
||||
.modal-close:hover {
|
||||
background: #f3f4f6;
|
||||
color: #374151;
|
||||
}
|
||||
|
||||
.modal-body {
|
||||
padding: 20px;
|
||||
}
|
||||
|
||||
.modal-body input {
|
||||
width: 100%;
|
||||
padding: 12px;
|
||||
border: 2px solid #e5e7eb;
|
||||
border-radius: 8px;
|
||||
font-size: 1rem;
|
||||
transition: border-color 0.3s;
|
||||
}
|
||||
|
||||
.modal-body input:focus {
|
||||
outline: none;
|
||||
border-color: #667eea;
|
||||
}
|
||||
|
||||
.modal-footer {
|
||||
padding: 20px;
|
||||
border-top: 1px solid #e5e7eb;
|
||||
display: flex;
|
||||
gap: 10px;
|
||||
justify-content: flex-end;
|
||||
}
|
||||
|
||||
.btn-secondary {
|
||||
background: #6b7280;
|
||||
color: white;
|
||||
}
|
||||
|
||||
.btn-secondary:hover {
|
||||
background: #4b5563;
|
||||
}
|
||||
|
||||
.btn-primary {
|
||||
background: #667eea;
|
||||
color: white;
|
||||
}
|
||||
|
||||
.btn-primary:hover {
|
||||
background: #5a67d8;
|
||||
}
|
||||
|
||||
@media (max-width: 768px) {
|
||||
.container {
|
||||
padding: 10px;
|
||||
}
|
||||
|
||||
.main-content {
|
||||
grid-template-columns: 1fr;
|
||||
gap: 15px;
|
||||
}
|
||||
|
||||
.sidebar {
|
||||
order: 2;
|
||||
}
|
||||
|
||||
main {
|
||||
order: 1;
|
||||
}
|
||||
|
||||
header {
|
||||
flex-direction: column;
|
||||
gap: 15px;
|
||||
text-align: center;
|
||||
}
|
||||
|
||||
.connection-controls {
|
||||
flex-direction: column;
|
||||
}
|
||||
|
||||
.poll-header {
|
||||
flex-direction: column;
|
||||
gap: 10px;
|
||||
align-items: flex-start;
|
||||
}
|
||||
}
|
||||
196
frontend.html
Normal file
196
frontend.html
Normal file
@@ -0,0 +1,196 @@
|
||||
<!DOCTYPE html>
|
||||
<html>
|
||||
<head>
|
||||
<title>Poll Client</title>
|
||||
<style>
|
||||
body {
|
||||
font-family: Arial, sans-serif;
|
||||
max-width: 600px;
|
||||
margin: 0 auto;
|
||||
padding: 20px;
|
||||
}
|
||||
#messages {
|
||||
height: 300px;
|
||||
border: 1px solid #ccc;
|
||||
overflow-y: auto;
|
||||
padding: 10px;
|
||||
margin-bottom: 10px;
|
||||
}
|
||||
.message { margin: 5px 0; }
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<h1>Poll Client</h1>
|
||||
<div id="status">Connecting to server...</div>
|
||||
<div id="status2">Connecting to server...</div>
|
||||
<div id="messages"></div>
|
||||
<div>
|
||||
<input type="text" id="messageInput" placeholder="Type your message">
|
||||
<!--button onclick="sendMessage()">Send</button-->
|
||||
<button id="sendBtn">Send</button>
|
||||
<input id="optionInput" placeholder="Add option">
|
||||
<button id="addBtn">Add</button>
|
||||
<ul id="options"></ul>
|
||||
</div>
|
||||
|
||||
<script type="module">
|
||||
import * as Y from "https://esm.sh/yjs"
|
||||
import { WebsocketProvider } from "https://esm.sh/y-websocket"
|
||||
|
||||
const WS_PORT = 8080;
|
||||
const ydoc = new Y.Doc();
|
||||
const status = document.getElementById('status');
|
||||
const status2 = document.getElementById('status2');
|
||||
const messages = document.getElementById('messages');
|
||||
const messageInput = document.getElementById('messageInput');
|
||||
|
||||
// Connect to the backend (WebSocket server) via Yjs WebsocketProvider
|
||||
const wsp = new WebsocketProvider(
|
||||
'ws://localhost:' + String(WS_PORT), 'poll-room',
|
||||
ydoc
|
||||
)
|
||||
wsp.on('status', event => {
|
||||
//console.log("event.status =", event.status)
|
||||
if (event.status == "connected") {
|
||||
status.textContent = 'Yjs connected to WebSocket server';
|
||||
status.style.color = 'green';
|
||||
}
|
||||
else {
|
||||
status.textContent = 'Yjs disonnected from WebSocket server';
|
||||
status.style.color = 'red';
|
||||
}
|
||||
})
|
||||
wsp.onopen = () => {
|
||||
console.log('connected to y-websocket server');
|
||||
};
|
||||
ydoc.on('update', () => {
|
||||
console.log('Yjs document updated locally');
|
||||
});
|
||||
wsp.on('sync', isSynced => {
|
||||
console.log("isSynced =", isSynced)
|
||||
})
|
||||
|
||||
// Connect to the backend (WebSocket server) via common WebSocket (only for informative messages)
|
||||
const ws = new WebSocket('ws://localhost:' + String(WS_PORT));
|
||||
ws.onopen = () => {
|
||||
status2.textContent = 'WebSocket client connected to WebSocket server';
|
||||
status2.style.color = 'green';
|
||||
};
|
||||
ws.onmessage = (event) => {
|
||||
const message = document.createElement('div');
|
||||
message.className = 'message';
|
||||
message.textContent = event.data;
|
||||
messages.appendChild(message);
|
||||
messages.scrollTop = messages.scrollHeight;
|
||||
};
|
||||
ws.onerror = (error) => {
|
||||
status2.textContent = 'Error: ' + error.message;
|
||||
status2.style.color = 'red';
|
||||
};
|
||||
ws.onclose = () => {
|
||||
status2.textContent = 'WebSocket client disconnected from WebSocket server';
|
||||
status2.style.color = 'red';
|
||||
};
|
||||
|
||||
// Function to send a message
|
||||
function sendMessage(command, payload) {
|
||||
let message = "";
|
||||
if (command === "TEXT_MESSAGE") {
|
||||
message = command + ":" + messageInput.value.trim();
|
||||
messageInput.value = "";
|
||||
}
|
||||
else if (command.startsWith("STATE_MESSAGE--")) {
|
||||
message = command + ":" + payload;
|
||||
}
|
||||
else {
|
||||
console.log("Error: unknown command '" + command + "'")
|
||||
}
|
||||
|
||||
if (message) {
|
||||
/* To make compatible with y-websocket-server API:
|
||||
const encoder = encoding.createEncoder()
|
||||
encoding.writeVarUint(encoder, messageTextMessage)
|
||||
syncProtocol.writeUpdate(encoder, update)
|
||||
enc_message = encoding.toUint8Array(encoder)*/
|
||||
ws.send(message);
|
||||
}
|
||||
}
|
||||
|
||||
// Send message on button click
|
||||
document.getElementById("sendBtn").onclick = () => {
|
||||
sendMessage("TEXT_MESSAGE", "");
|
||||
};
|
||||
|
||||
// Send message on Enter key
|
||||
messageInput.addEventListener('keypress', (e) => {
|
||||
if (e.key === 'Enter') {
|
||||
sendMessage("TEXT_MESSAGE", "");
|
||||
}
|
||||
});
|
||||
|
||||
// Actual poll logic
|
||||
const optionsMap = ydoc.getMap("options");
|
||||
|
||||
const optionsList = document.getElementById("options");
|
||||
const input = document.getElementById("optionInput");
|
||||
|
||||
function render() {
|
||||
optionsList.innerHTML = ""
|
||||
|
||||
optionsMap.forEach((vote, name) => {
|
||||
// List element for this option
|
||||
const li = document.createElement("li")
|
||||
|
||||
// Label for this option
|
||||
const label = document.createElement("span")
|
||||
label.textContent = name + " — " + vote + " "
|
||||
|
||||
// "Vote"/"Unvote" button
|
||||
const voteBtn = document.createElement("button")
|
||||
if (vote == false) {
|
||||
voteBtn.textContent = "Vote"
|
||||
}
|
||||
else {
|
||||
voteBtn.textContent = "Unvote"
|
||||
}
|
||||
voteBtn.onclick = () => {
|
||||
vote = !vote;
|
||||
optionsMap.set(name, vote);
|
||||
sendMessage("STATE_MESSAGE--" + (vote ? "VOTE" : "UNVOTE"), name);
|
||||
}
|
||||
|
||||
// "Remove" button
|
||||
const removeBtn = document.createElement("button")
|
||||
removeBtn.textContent = "Remove"
|
||||
removeBtn.onclick = () => {
|
||||
optionsMap.delete(name);
|
||||
sendMessage("STATE_MESSAGE--REMOVE", name);
|
||||
}
|
||||
|
||||
li.appendChild(label)
|
||||
li.appendChild(voteBtn)
|
||||
li.appendChild(removeBtn)
|
||||
|
||||
optionsList.appendChild(li)
|
||||
})
|
||||
};
|
||||
|
||||
document.getElementById("addBtn").onclick = () => {
|
||||
const name = input.value.trim()
|
||||
|
||||
if (!name) return
|
||||
|
||||
if (!optionsMap.has(name)) {
|
||||
optionsMap.set(name, false);
|
||||
sendMessage("STATE_MESSAGE--ADD_OPTION", name);
|
||||
}
|
||||
|
||||
input.value = ""
|
||||
};
|
||||
|
||||
optionsMap.observe(render);
|
||||
|
||||
render();
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
133
index.html
133
index.html
@@ -1,133 +0,0 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>P2P Polling App</title>
|
||||
<link rel="stylesheet" href="css/styles.css">
|
||||
<script src="https://unpkg.com/peerjs@1.5.2/dist/peerjs.min.js"></script>
|
||||
</head>
|
||||
<body>
|
||||
<div class="container">
|
||||
<header>
|
||||
<h1>P2P Polling App</h1>
|
||||
<div class="connection-status">
|
||||
<span id="peer-id">Loading...</span>
|
||||
<span id="connection-indicator" class="status-disconnected">●</span>
|
||||
</div>
|
||||
</header>
|
||||
|
||||
<div class="main-content">
|
||||
<!-- Sidebar -->
|
||||
<aside class="sidebar">
|
||||
<div class="sidebar-section">
|
||||
<h3>Available Polls</h3>
|
||||
<div class="sidebar-hint">
|
||||
<p class="hint-text">Click a poll to view and vote</p>
|
||||
</div>
|
||||
<div id="polls-list" class="polls-list">
|
||||
<p class="no-polls">No polls created yet</p>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="sidebar-section">
|
||||
<h3>Connected Peers</h3>
|
||||
<div class="peers-info">
|
||||
<div class="peers-status">
|
||||
<span>Count: <span id="peer-count">0</span></span>
|
||||
<span id="connection-status-text" class="status-text">Disconnected</span>
|
||||
</div>
|
||||
<ul id="peers" class="peers-sidebar"></ul>
|
||||
</div>
|
||||
</div>
|
||||
</aside>
|
||||
|
||||
<main>
|
||||
<!-- Connection Setup -->
|
||||
<section id="connection-section" class="section">
|
||||
<h2>Connect to Peer</h2>
|
||||
<div class="connection-info">
|
||||
<p><strong>Host:</strong> Share your Peer ID below with others</p>
|
||||
<p><strong>Joiner:</strong> Enter the host's Peer ID to connect</p>
|
||||
</div>
|
||||
<div class="connection-controls">
|
||||
<input type="text" id="room-id" placeholder="Enter host's Peer ID to connect">
|
||||
<button id="join-btn">Connect to Host / Create New Poll</button>
|
||||
<button id="copy-id-btn">Copy Your Peer ID</button>
|
||||
</div>
|
||||
</section>
|
||||
|
||||
<!-- Poll Creation -->
|
||||
<section id="poll-creation" class="section hidden">
|
||||
<h2>Create Poll</h2>
|
||||
<div class="poll-form">
|
||||
<input type="text" id="poll-question" placeholder="Enter your poll question">
|
||||
<div id="options-container">
|
||||
<div class="option-input">
|
||||
<input type="text" class="option-text" placeholder="Option 1">
|
||||
<button class="remove-option-btn hidden">Remove</button>
|
||||
</div>
|
||||
<div class="option-input">
|
||||
<input type="text" class="option-text" placeholder="Option 2">
|
||||
<button class="remove-option-btn hidden">Remove</button>
|
||||
</div>
|
||||
</div>
|
||||
<button id="add-option-btn">+ Add Option</button>
|
||||
<button id="create-poll-btn">Create Poll</button>
|
||||
</div>
|
||||
</section>
|
||||
|
||||
<!-- Active Poll -->
|
||||
<section id="active-poll" class="section hidden">
|
||||
<div class="poll-header">
|
||||
<h2 id="poll-question-display"></h2>
|
||||
<button id="add-poll-option-btn" class="small-btn">+ Add Option</button>
|
||||
</div>
|
||||
|
||||
<div class="vote-hint">
|
||||
<p class="hint-text">Click any option to vote. Click your voted option again to remove your vote.</p>
|
||||
</div>
|
||||
|
||||
<div id="poll-options" class="options-list"></div>
|
||||
|
||||
<div class="poll-actions">
|
||||
<button id="reset-poll-btn">Reset Poll</button>
|
||||
<button id="new-poll-btn" class="small-btn">New Poll</button>
|
||||
</div>
|
||||
</section>
|
||||
</main>
|
||||
</div>
|
||||
|
||||
<!-- Loading Overlay -->
|
||||
<div id="loading-overlay" class="overlay hidden">
|
||||
<div class="loading-spinner"></div>
|
||||
<p>Connecting...</p>
|
||||
</div>
|
||||
|
||||
<!-- Notification Toast -->
|
||||
<div id="notification" class="notification hidden"></div>
|
||||
|
||||
<!-- Add Option Modal -->
|
||||
<div id="add-option-modal" class="modal hidden">
|
||||
<div class="modal-content">
|
||||
<div class="modal-header">
|
||||
<h3>Add New Option</h3>
|
||||
<button class="modal-close" id="close-modal-btn">×</button>
|
||||
</div>
|
||||
<div class="modal-body">
|
||||
<input type="text" id="new-option-input" placeholder="Enter new option text">
|
||||
</div>
|
||||
<div class="modal-footer">
|
||||
<button id="cancel-modal-btn" class="btn-secondary">Cancel</button>
|
||||
<button id="save-option-btn" class="btn-primary">Add Option</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<script src="js/peer-manager.js"></script>
|
||||
<script src="js/poll-manager.js"></script>
|
||||
<script src="js/ui-controller.js"></script>
|
||||
<script src="js/app.js"></script>
|
||||
</body>
|
||||
</html>
|
||||
210
js/app.js
210
js/app.js
@@ -1,210 +0,0 @@
|
||||
// Main application entry point
|
||||
class P2PPollApp {
|
||||
constructor() {
|
||||
this.peerManager = null;
|
||||
this.pollManager = null;
|
||||
this.uiController = null;
|
||||
this.isInitialized = false;
|
||||
}
|
||||
|
||||
async initialize() {
|
||||
try {
|
||||
// Initialize peer manager
|
||||
this.peerManager = new PeerManager();
|
||||
this.pollManager = new PollManager(this.peerManager);
|
||||
this.uiController = new UIController(this.peerManager, this.pollManager);
|
||||
|
||||
// Set up event handlers
|
||||
this.setupEventHandlers();
|
||||
|
||||
// Initialize peer connection
|
||||
await this.peerManager.initialize();
|
||||
|
||||
// Load saved data
|
||||
this.pollManager.loadFromLocalStorage();
|
||||
|
||||
// If there are saved polls, update the sidebar
|
||||
const savedPolls = this.pollManager.getAvailablePolls();
|
||||
if (savedPolls.length > 0) {
|
||||
this.uiController.updatePollsList(savedPolls);
|
||||
}
|
||||
|
||||
|
||||
this.isInitialized = true;
|
||||
console.log('P2P Poll App initialized successfully');
|
||||
|
||||
} catch (error) {
|
||||
console.error('Failed to initialize app:', error);
|
||||
this.uiController?.showNotification('Failed to initialize app: ' + error.message, 'error');
|
||||
}
|
||||
}
|
||||
|
||||
setupEventHandlers() {
|
||||
// Peer manager events
|
||||
this.peerManager.onConnectionStatusChange = (status, peerId) => {
|
||||
this.uiController.updatePeerId(peerId);
|
||||
|
||||
if (status === 'connected') {
|
||||
this.uiController.showNotification('Connected to P2P network', 'success');
|
||||
} else if (status === 'disconnected') {
|
||||
this.uiController.showNotification('Disconnected from P2P network', 'error');
|
||||
}
|
||||
};
|
||||
|
||||
this.peerManager.onPeerConnected = (peerId) => {
|
||||
this.uiController.showNotification(`Peer ${peerId} connected`, 'success');
|
||||
|
||||
// If we're the host and have polls, send them to the new peer
|
||||
if (this.peerManager.isHost) {
|
||||
const availablePolls = this.pollManager.getAvailablePolls();
|
||||
if (availablePolls.length > 0) {
|
||||
availablePolls.forEach(poll => {
|
||||
this.peerManager.sendMessage(peerId, {
|
||||
type: 'poll_update',
|
||||
poll: poll,
|
||||
senderId: this.peerManager.getPeerId()
|
||||
});
|
||||
});
|
||||
|
||||
// Also send current poll if there is one
|
||||
const currentPoll = this.pollManager.getCurrentPoll();
|
||||
if (currentPoll) {
|
||||
this.peerManager.sendMessage(peerId, {
|
||||
type: 'current_poll',
|
||||
poll: currentPoll,
|
||||
senderId: this.peerManager.getPeerId()
|
||||
});
|
||||
}
|
||||
}
|
||||
} else {
|
||||
this.peerManager.sendMessage(peerId, {
|
||||
type: 'sync_request',
|
||||
senderId: this.peerManager.getPeerId()
|
||||
});
|
||||
}
|
||||
};
|
||||
|
||||
this.peerManager.onPeerDisconnected = (peerId) => {
|
||||
this.uiController.showNotification(`Peer ${peerId} disconnected`, 'info');
|
||||
};
|
||||
|
||||
this.peerManager.onMessageReceived = (message, senderId) => {
|
||||
this.handleMessage(message, senderId);
|
||||
};
|
||||
|
||||
// Poll manager events
|
||||
this.pollManager.onPollCreated = (poll) => {
|
||||
this.uiController.renderPoll(poll);
|
||||
};
|
||||
|
||||
this.pollManager.onPollUpdated = (poll) => {
|
||||
this.uiController.renderPoll(poll);
|
||||
};
|
||||
|
||||
this.pollManager.onPollSelected = (poll) => {
|
||||
if (poll) {
|
||||
this.uiController.renderPoll(poll);
|
||||
this.uiController.showActivePoll();
|
||||
} else {
|
||||
this.uiController.showPollCreation();
|
||||
}
|
||||
};
|
||||
|
||||
this.pollManager.onPollsListUpdated = (polls) => {
|
||||
this.uiController.updatePollsList(polls);
|
||||
};
|
||||
}
|
||||
|
||||
handleMessage(message, senderId) {
|
||||
console.log('Processing message:', message, 'from:', senderId);
|
||||
|
||||
switch (message.type) {
|
||||
case 'poll_update':
|
||||
this.pollManager.syncPoll(message.poll);
|
||||
break;
|
||||
|
||||
case 'current_poll':
|
||||
this.pollManager.syncPoll(message.poll);
|
||||
if (this.pollManager.getCurrentPoll() && this.pollManager.getCurrentPoll().id === message.poll.id) {
|
||||
this.uiController.renderPoll(message.poll);
|
||||
this.uiController.showActivePoll();
|
||||
}
|
||||
break;
|
||||
|
||||
case 'vote':
|
||||
this.pollManager.handleVoteMessage(message.optionId, message.voterId);
|
||||
break;
|
||||
|
||||
case 'unvote':
|
||||
this.pollManager.handleUnvoteMessage(message.voterId);
|
||||
break;
|
||||
|
||||
case 'poll_reset':
|
||||
this.pollManager.handlePollReset();
|
||||
break;
|
||||
|
||||
case 'sync_request':
|
||||
const currentPoll = this.pollManager.getCurrentPoll();
|
||||
if (currentPoll) {
|
||||
this.peerManager.sendMessage(senderId, {
|
||||
type: 'current_poll',
|
||||
poll: currentPoll,
|
||||
senderId: this.peerManager.getPeerId()
|
||||
});
|
||||
}
|
||||
|
||||
const availablePolls = this.pollManager.getAvailablePolls();
|
||||
availablePolls.forEach(poll => {
|
||||
this.peerManager.sendMessage(senderId, {
|
||||
type: 'poll_update',
|
||||
poll: poll,
|
||||
senderId: this.peerManager.getPeerId()
|
||||
});
|
||||
});
|
||||
break;
|
||||
|
||||
default:
|
||||
console.warn('Unknown message type:', message.type);
|
||||
}
|
||||
}
|
||||
|
||||
destroy() {
|
||||
if (this.peerManager) {
|
||||
this.peerManager.destroy();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
document.addEventListener('DOMContentLoaded', async () => {
|
||||
const app = new P2PPollApp();
|
||||
|
||||
try {
|
||||
await app.initialize();
|
||||
} catch (error) {
|
||||
console.error('App initialization failed:', error);
|
||||
|
||||
// Show error message to user
|
||||
const errorDiv = document.createElement('div');
|
||||
errorDiv.className = 'notification error';
|
||||
errorDiv.textContent = 'Failed to initialize app. Please refresh the page.';
|
||||
errorDiv.style.position = 'fixed';
|
||||
errorDiv.style.top = '20px';
|
||||
errorDiv.style.right = '20px';
|
||||
errorDiv.style.zIndex = '1001';
|
||||
document.body.appendChild(errorDiv);
|
||||
}
|
||||
|
||||
// Cleanup on page unload
|
||||
window.addEventListener('beforeunload', () => {
|
||||
app.destroy();
|
||||
});
|
||||
});
|
||||
|
||||
// Handle page visibility changes
|
||||
document.addEventListener('visibilitychange', () => {
|
||||
if (document.hidden) {
|
||||
console.log('Page hidden - connection may become unstable');
|
||||
} else {
|
||||
console.log('Page visible - checking connection status');
|
||||
}
|
||||
});
|
||||
@@ -1,233 +0,0 @@
|
||||
class PeerManager {
|
||||
constructor() {
|
||||
this.peer = null;
|
||||
this.connections = new Map();
|
||||
this.roomId = null;
|
||||
this.isHost = false;
|
||||
this.onPeerConnected = null;
|
||||
this.onPeerDisconnected = null;
|
||||
this.onMessageReceived = null;
|
||||
this.onConnectionStatusChange = null;
|
||||
}
|
||||
|
||||
async initialize() {
|
||||
try {
|
||||
this.peer = new Peer({
|
||||
debug: 2,
|
||||
config: {
|
||||
iceServers: [
|
||||
{ urls: 'stun:stun.l.google.com:19302' },
|
||||
{ urls: 'stun:stun1.l.google.com:19302' }
|
||||
]
|
||||
}
|
||||
});
|
||||
|
||||
this.peer.on('open', (id) => {
|
||||
console.log('My peer ID is:', id);
|
||||
this.updateConnectionStatus('connected');
|
||||
this.updatePeersList();
|
||||
if (this.onConnectionStatusChange) {
|
||||
this.onConnectionStatusChange('connected', id);
|
||||
}
|
||||
});
|
||||
|
||||
this.peer.on('connection', (conn) => {
|
||||
this.handleIncomingConnection(conn);
|
||||
});
|
||||
|
||||
this.peer.on('disconnected', () => {
|
||||
this.updateConnectionStatus('disconnected');
|
||||
this.updatePeersList();
|
||||
});
|
||||
|
||||
this.peer.on('error', (err) => {
|
||||
console.error('Peer error:', err);
|
||||
this.updateConnectionStatus('error');
|
||||
});
|
||||
|
||||
} catch (error) {
|
||||
console.error('Failed to initialize peer:', error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
async createRoom() {
|
||||
if (!this.peer) {
|
||||
throw new Error('Peer not initialized');
|
||||
}
|
||||
|
||||
this.roomId = this.generateRoomId();
|
||||
this.isHost = true;
|
||||
|
||||
console.log('Created room:', this.roomId);
|
||||
return this.roomId;
|
||||
}
|
||||
|
||||
async joinRoom(roomId) {
|
||||
if (!this.peer) {
|
||||
throw new Error('Peer not initialized');
|
||||
}
|
||||
|
||||
this.roomId = roomId;
|
||||
this.isHost = false;
|
||||
|
||||
console.log('Attempting to join room:', roomId);
|
||||
|
||||
|
||||
try {
|
||||
const conn = this.peer.connect(roomId);
|
||||
await this.setupConnection(conn);
|
||||
console.log('Successfully joined room');
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('Failed to join room:', error);
|
||||
throw new Error('Could not connect to host. Make sure the host is online and you have the correct peer ID.');
|
||||
}
|
||||
}
|
||||
|
||||
handleIncomingConnection(conn) {
|
||||
this.setupConnection(conn);
|
||||
}
|
||||
|
||||
async setupConnection(conn) {
|
||||
return new Promise((resolve, reject) => {
|
||||
conn.on('open', () => {
|
||||
console.log('Connected to:', conn.peer);
|
||||
this.connections.set(conn.peer, conn);
|
||||
this.updatePeersList();
|
||||
|
||||
if (this.isHost) {
|
||||
if (this.onPeerConnected) {
|
||||
this.onPeerConnected(conn.peer);
|
||||
}
|
||||
} else {
|
||||
if (this.onPeerConnected) {
|
||||
this.onPeerConnected(conn.peer);
|
||||
}
|
||||
}
|
||||
|
||||
resolve(conn);
|
||||
});
|
||||
|
||||
conn.on('data', (data) => {
|
||||
this.handleMessage(data, conn.peer);
|
||||
});
|
||||
|
||||
conn.on('close', () => {
|
||||
console.log('Connection closed:', conn.peer);
|
||||
this.connections.delete(conn.peer);
|
||||
this.updatePeersList();
|
||||
|
||||
if (this.onPeerDisconnected) {
|
||||
this.onPeerDisconnected(conn.peer);
|
||||
}
|
||||
});
|
||||
|
||||
conn.on('error', (err) => {
|
||||
console.error('Connection error:', err);
|
||||
reject(err);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
handleMessage(data, senderId) {
|
||||
console.log('Received message:', data, 'from:', senderId);
|
||||
|
||||
if (this.onMessageReceived) {
|
||||
this.onMessageReceived(data, senderId);
|
||||
}
|
||||
}
|
||||
|
||||
sendMessage(peerId, message) {
|
||||
const conn = this.connections.get(peerId);
|
||||
if (conn && conn.open) {
|
||||
conn.send(message);
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
broadcastMessage(message) {
|
||||
let sentCount = 0;
|
||||
this.connections.forEach((conn, peerId) => {
|
||||
if (conn.open) {
|
||||
conn.send(message);
|
||||
sentCount++;
|
||||
}
|
||||
});
|
||||
return sentCount;
|
||||
}
|
||||
|
||||
getPeerId() {
|
||||
return this.peer ? this.peer.id : null;
|
||||
}
|
||||
|
||||
getRoomId() {
|
||||
return this.roomId;
|
||||
}
|
||||
|
||||
getConnectedPeers() {
|
||||
return Array.from(this.connections.keys());
|
||||
}
|
||||
|
||||
getPeerCount() {
|
||||
return this.connections.size;
|
||||
}
|
||||
|
||||
isConnected() {
|
||||
return this.peer && this.peer.disconnected !== true;
|
||||
}
|
||||
|
||||
generateRoomId() {
|
||||
return Math.random().toString(36).substr(2, 9).toUpperCase();
|
||||
}
|
||||
|
||||
updateConnectionStatus(status) {
|
||||
const indicator = document.getElementById('connection-indicator');
|
||||
const statusText = document.getElementById('connection-status-text');
|
||||
|
||||
if (indicator) {
|
||||
indicator.className = `status-${status}`;
|
||||
}
|
||||
|
||||
if (statusText) {
|
||||
statusText.textContent = status.charAt(0).toUpperCase() + status.slice(1);
|
||||
statusText.className = `status-text ${status}`;
|
||||
}
|
||||
}
|
||||
|
||||
updatePeersList() {
|
||||
const peersList = document.getElementById('peers');
|
||||
const peerCount = document.getElementById('peer-count');
|
||||
|
||||
if (peersList && peerCount) {
|
||||
peerCount.textContent = this.getPeerCount();
|
||||
peersList.innerHTML = '';
|
||||
|
||||
if (this.getPeerCount() === 0) {
|
||||
const li = document.createElement('li');
|
||||
li.textContent = 'No connected peers';
|
||||
li.style.fontStyle = 'italic';
|
||||
li.style.color = '#9ca3af';
|
||||
peersList.appendChild(li);
|
||||
} else {
|
||||
this.getConnectedPeers().forEach(peerId => {
|
||||
const li = document.createElement('li');
|
||||
li.textContent = `${peerId} ${peerId === this.peer.id ? '(You)' : ''}`;
|
||||
li.style.fontWeight = peerId === this.peer.id ? 'bold' : 'normal';
|
||||
li.style.color = peerId === this.peer.id ? '#667eea' : '#6b7280';
|
||||
peersList.appendChild(li);
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
destroy() {
|
||||
if (this.peer) {
|
||||
this.connections.forEach(conn => conn.close());
|
||||
this.peer.destroy();
|
||||
this.peer = null;
|
||||
this.connections.clear();
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1,428 +0,0 @@
|
||||
class PollManager {
|
||||
constructor(peerManager) {
|
||||
this.peerManager = peerManager;
|
||||
this.currentPoll = null;
|
||||
this.availablePolls = new Map(); // pollId -> poll
|
||||
this.myVotes = new Set();
|
||||
this.onPollUpdated = null;
|
||||
this.onPollCreated = null;
|
||||
this.onPollSelected = null;
|
||||
this.onPollsListUpdated = null;
|
||||
}
|
||||
|
||||
createPoll(question, options) {
|
||||
const poll = {
|
||||
id: this.generatePollId(),
|
||||
question: question.trim(),
|
||||
options: options.map((text, index) => ({
|
||||
id: `opt-${index}`,
|
||||
text: text.trim(),
|
||||
votes: 0,
|
||||
voters: []
|
||||
})),
|
||||
createdBy: this.peerManager.getPeerId(),
|
||||
createdAt: Date.now()
|
||||
};
|
||||
|
||||
// Clear any existing votes when creating a new poll
|
||||
this.myVotes.clear();
|
||||
|
||||
this.currentPoll = poll;
|
||||
this.availablePolls.set(poll.id, poll);
|
||||
this.saveToLocalStorage();
|
||||
|
||||
if (this.onPollCreated) {
|
||||
this.onPollCreated(poll);
|
||||
}
|
||||
|
||||
if (this.onPollsListUpdated) {
|
||||
this.onPollsListUpdated(Array.from(this.availablePolls.values()));
|
||||
}
|
||||
|
||||
// Broadcast new poll to all peers
|
||||
this.broadcastPollUpdate();
|
||||
|
||||
return poll;
|
||||
}
|
||||
|
||||
addOption(text) {
|
||||
if (!this.currentPoll) {
|
||||
throw new Error('No active poll');
|
||||
}
|
||||
|
||||
const newOption = {
|
||||
id: `opt-${Date.now()}`,
|
||||
text: text.trim(),
|
||||
votes: 0,
|
||||
voters: []
|
||||
};
|
||||
|
||||
this.currentPoll.options.push(newOption);
|
||||
this.saveToLocalStorage();
|
||||
this.notifyPollUpdated();
|
||||
this.broadcastPollUpdate();
|
||||
|
||||
return newOption;
|
||||
}
|
||||
|
||||
vote(optionId) {
|
||||
if (!this.currentPoll) {
|
||||
throw new Error('No active poll');
|
||||
}
|
||||
|
||||
const myPeerId = this.peerManager.getPeerId();
|
||||
|
||||
// Check if already voted for this option
|
||||
if (this.myVotes.has(optionId)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
this.removePreviousVote(myPeerId);
|
||||
|
||||
// Add new vote
|
||||
const option = this.currentPoll.options.find(opt => opt.id === optionId);
|
||||
if (!option) {
|
||||
throw new Error('Option not found');
|
||||
}
|
||||
|
||||
option.votes++;
|
||||
option.voters.push(myPeerId);
|
||||
this.myVotes.add(optionId);
|
||||
|
||||
this.saveToLocalStorage();
|
||||
this.notifyPollUpdated();
|
||||
this.broadcastVote(optionId, myPeerId);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
removePreviousVote(peerId) {
|
||||
if (!this.currentPoll) return;
|
||||
|
||||
this.currentPoll.options.forEach(option => {
|
||||
const voterIndex = option.voters.indexOf(peerId);
|
||||
if (voterIndex !== -1) {
|
||||
option.votes--;
|
||||
option.voters.splice(voterIndex, 1);
|
||||
|
||||
// Remove from my votes if it's my vote
|
||||
if (peerId === this.peerManager.getPeerId()) {
|
||||
this.myVotes.delete(option.id);
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
resetPoll() {
|
||||
if (!this.currentPoll) {
|
||||
throw new Error('No active poll');
|
||||
}
|
||||
|
||||
this.currentPoll.options.forEach(option => {
|
||||
option.votes = 0;
|
||||
option.voters = [];
|
||||
});
|
||||
|
||||
this.myVotes.clear();
|
||||
this.saveToLocalStorage();
|
||||
this.notifyPollUpdated();
|
||||
this.broadcastPollReset();
|
||||
}
|
||||
|
||||
syncPoll(pollData) {
|
||||
if (!this.availablePolls.has(pollData.id)) {
|
||||
this.availablePolls.set(pollData.id, pollData);
|
||||
this.saveToLocalStorage();
|
||||
|
||||
if (this.onPollsListUpdated) {
|
||||
this.onPollsListUpdated(Array.from(this.availablePolls.values()));
|
||||
}
|
||||
}
|
||||
|
||||
if (!this.currentPoll || this.currentPoll.id === pollData.id) {
|
||||
this.currentPoll = pollData;
|
||||
this.availablePolls.set(pollData.id, pollData);
|
||||
|
||||
this.validateMyVotes(pollData);
|
||||
|
||||
this.saveToLocalStorage();
|
||||
if (this.onPollCreated) {
|
||||
this.onPollCreated(pollData);
|
||||
}
|
||||
} else if (this.currentPoll.id === pollData.id) {
|
||||
this.mergePollData(pollData);
|
||||
this.availablePolls.set(pollData.id, this.currentPoll);
|
||||
|
||||
this.validateMyVotes(this.currentPoll);
|
||||
|
||||
this.saveToLocalStorage();
|
||||
this.notifyPollUpdated();
|
||||
}
|
||||
}
|
||||
|
||||
mergePollData(remotePoll) {
|
||||
const localOptions = new Map(this.currentPoll.options.map(opt => [opt.id, opt]));
|
||||
const remoteOptions = new Map(remotePoll.options.map(opt => [opt.id, opt]));
|
||||
|
||||
remoteOptions.forEach((remoteOpt, id) => {
|
||||
if (!localOptions.has(id)) {
|
||||
this.currentPoll.options.push(remoteOpt);
|
||||
localOptions.set(id, remoteOpt);
|
||||
}
|
||||
});
|
||||
|
||||
this.currentPoll.options.forEach(localOpt => {
|
||||
const remoteOpt = remoteOptions.get(localOpt.id);
|
||||
if (remoteOpt) {
|
||||
const allVoters = new Set([...localOpt.voters, ...remoteOpt.voters]);
|
||||
localOpt.voters = Array.from(allVoters);
|
||||
localOpt.votes = localOpt.voters.length;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
handleVoteMessage(optionId, voterId) {
|
||||
if (!this.currentPoll) return;
|
||||
|
||||
this.removePreviousVote(voterId);
|
||||
|
||||
const option = this.currentPoll.options.find(opt => opt.id === optionId);
|
||||
if (option && !option.voters.includes(voterId)) {
|
||||
option.votes++;
|
||||
option.voters.push(voterId);
|
||||
|
||||
// Update my votes if this is my vote
|
||||
if (voterId === this.peerManager.getPeerId()) {
|
||||
this.myVotes.add(optionId);
|
||||
}
|
||||
|
||||
this.saveToLocalStorage();
|
||||
this.notifyPollUpdated();
|
||||
}
|
||||
}
|
||||
|
||||
handleUnvoteMessage(voterId) {
|
||||
if (!this.currentPoll) return;
|
||||
|
||||
// Remove vote from this voter
|
||||
this.removePreviousVote(voterId);
|
||||
|
||||
// Update my votes if this is my vote
|
||||
if (voterId === this.peerManager.getPeerId()) {
|
||||
this.myVotes.clear();
|
||||
}
|
||||
|
||||
this.saveToLocalStorage();
|
||||
this.notifyPollUpdated();
|
||||
}
|
||||
|
||||
handlePollReset() {
|
||||
if (!this.currentPoll) return;
|
||||
|
||||
this.currentPoll.options.forEach(option => {
|
||||
option.votes = 0;
|
||||
option.voters = [];
|
||||
});
|
||||
|
||||
this.myVotes.clear();
|
||||
this.saveToLocalStorage();
|
||||
this.notifyPollUpdated();
|
||||
}
|
||||
|
||||
broadcastPollUpdate() {
|
||||
if (!this.currentPoll) return;
|
||||
|
||||
this.peerManager.broadcastMessage({
|
||||
type: 'poll_update',
|
||||
poll: this.currentPoll,
|
||||
senderId: this.peerManager.getPeerId()
|
||||
});
|
||||
}
|
||||
|
||||
broadcastVote(optionId, voterId) {
|
||||
this.peerManager.broadcastMessage({
|
||||
type: 'vote',
|
||||
optionId: optionId,
|
||||
voterId: voterId,
|
||||
senderId: this.peerManager.getPeerId()
|
||||
});
|
||||
}
|
||||
|
||||
broadcastPollReset() {
|
||||
this.peerManager.broadcastMessage({
|
||||
type: 'poll_reset',
|
||||
senderId: this.peerManager.getPeerId()
|
||||
});
|
||||
}
|
||||
|
||||
selectPoll(pollId) {
|
||||
const poll = this.availablePolls.get(pollId);
|
||||
if (poll) {
|
||||
this.currentPoll = poll;
|
||||
|
||||
// Clear votes when switching to a different poll
|
||||
// Only clear if this is a different poll than what we had voted in
|
||||
const currentVotedOption = this.getMyVotedOption();
|
||||
if (currentVotedOption && !poll.options.some(opt => opt.id === currentVotedOption)) {
|
||||
this.myVotes.clear();
|
||||
}
|
||||
|
||||
this.saveToLocalStorage();
|
||||
|
||||
if (this.onPollSelected) {
|
||||
this.onPollSelected(poll);
|
||||
}
|
||||
|
||||
if (this.onPollUpdated) {
|
||||
this.onPollUpdated(poll);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
createNewPoll() {
|
||||
this.currentPoll = null;
|
||||
// Clear current poll inputs but keep available polls
|
||||
if (this.onPollSelected) {
|
||||
this.onPollSelected(null);
|
||||
}
|
||||
}
|
||||
|
||||
getAvailablePolls() {
|
||||
return Array.from(this.availablePolls.values());
|
||||
}
|
||||
|
||||
unvote() {
|
||||
if (!this.currentPoll) {
|
||||
throw new Error('No active poll');
|
||||
}
|
||||
|
||||
const myPeerId = this.peerManager.getPeerId();
|
||||
const myVotedOption = this.getMyVotedOption();
|
||||
|
||||
if (!myVotedOption) {
|
||||
return false; // No vote to remove
|
||||
}
|
||||
|
||||
// Remove vote from current option
|
||||
this.removePreviousVote(myPeerId);
|
||||
|
||||
this.saveToLocalStorage();
|
||||
this.notifyPollUpdated();
|
||||
this.broadcastUnvote(myPeerId);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
validateMyVotes(poll) {
|
||||
const myPeerId = this.peerManager.getPeerId();
|
||||
const validVotes = new Set();
|
||||
|
||||
console.log('Validating votes for peer:', myPeerId);
|
||||
console.log('Current myVotes:', Array.from(this.myVotes));
|
||||
console.log('Poll voters:', poll.options.map(opt => ({ id: opt.id, voters: opt.voters })));
|
||||
|
||||
// Check each vote in myVotes to see if it's actually mine
|
||||
this.myVotes.forEach(optionId => {
|
||||
const option = poll.options.find(opt => opt.id === optionId);
|
||||
if (option && option.voters.includes(myPeerId)) {
|
||||
// This vote is actually mine
|
||||
validVotes.add(optionId);
|
||||
console.log(`Vote ${optionId} is valid for peer ${myPeerId}`);
|
||||
} else {
|
||||
console.log(`Vote ${optionId} is NOT valid for peer ${myPeerId}`);
|
||||
}
|
||||
});
|
||||
|
||||
// Replace myVotes with only the valid votes
|
||||
this.myVotes = validVotes;
|
||||
console.log('Final valid votes:', Array.from(this.myVotes));
|
||||
}
|
||||
|
||||
getCurrentPoll() {
|
||||
return this.currentPoll;
|
||||
}
|
||||
|
||||
broadcastUnvote(peerId) {
|
||||
this.peerManager.broadcastMessage({
|
||||
type: 'unvote',
|
||||
voterId: peerId,
|
||||
senderId: this.peerManager.getPeerId()
|
||||
});
|
||||
}
|
||||
|
||||
hasVoted(optionId) {
|
||||
return this.myVotes.has(optionId);
|
||||
}
|
||||
|
||||
getMyVotedOption() {
|
||||
return Array.from(this.myVotes)[0] || null;
|
||||
}
|
||||
|
||||
generatePollId() {
|
||||
return `poll-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
|
||||
}
|
||||
|
||||
saveToLocalStorage() {
|
||||
// Save all available polls
|
||||
const pollsData = Array.from(this.availablePolls.values());
|
||||
localStorage.setItem('p2p-polls-list', JSON.stringify(pollsData));
|
||||
|
||||
// Save current poll separately
|
||||
if (this.currentPoll) {
|
||||
localStorage.setItem('p2p-poll-current', JSON.stringify(this.currentPoll));
|
||||
} else {
|
||||
localStorage.removeItem('p2p-poll-current');
|
||||
}
|
||||
|
||||
localStorage.setItem('p2p-poll-my-votes', JSON.stringify(Array.from(this.myVotes)));
|
||||
}
|
||||
|
||||
loadFromLocalStorage() {
|
||||
try {
|
||||
const savedPollsList = localStorage.getItem('p2p-polls-list');
|
||||
const savedVotes = localStorage.getItem('p2p-poll-my-votes');
|
||||
|
||||
if (savedPollsList) {
|
||||
const polls = JSON.parse(savedPollsList);
|
||||
this.availablePolls.clear();
|
||||
polls.forEach(poll => {
|
||||
this.availablePolls.set(poll.id, poll);
|
||||
});
|
||||
|
||||
if (this.onPollsListUpdated) {
|
||||
this.onPollsListUpdated(polls);
|
||||
}
|
||||
}
|
||||
|
||||
if (savedVotes) {
|
||||
this.myVotes = new Set(JSON.parse(savedVotes));
|
||||
|
||||
// Validate votes against current poll
|
||||
const currentPoll = this.currentPoll || this.availablePolls.values().next().value;
|
||||
if (currentPoll) {
|
||||
this.validateMyVotes(currentPoll);
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('Failed to load from localStorage:', error);
|
||||
}
|
||||
}
|
||||
|
||||
clearData() {
|
||||
this.currentPoll = null;
|
||||
this.availablePolls.clear();
|
||||
this.myVotes.clear();
|
||||
localStorage.removeItem('p2p-poll-current');
|
||||
localStorage.removeItem('p2p-polls-list');
|
||||
localStorage.removeItem('p2p-poll-my-votes');
|
||||
}
|
||||
|
||||
notifyPollUpdated() {
|
||||
if (this.onPollUpdated) {
|
||||
this.onPollUpdated(this.currentPoll);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1,419 +0,0 @@
|
||||
class UIController {
|
||||
constructor(peerManager, pollManager) {
|
||||
this.peerManager = peerManager;
|
||||
this.pollManager = pollManager;
|
||||
this.initializeEventListeners();
|
||||
}
|
||||
|
||||
initializeEventListeners() {
|
||||
// Connection controls
|
||||
document.getElementById('join-btn').addEventListener('click', () => this.handleJoinRoom());
|
||||
document.getElementById('copy-id-btn').addEventListener('click', () => this.copyPeerId());
|
||||
|
||||
// Poll creation
|
||||
document.getElementById('create-poll-btn').addEventListener('click', () => this.handleCreatePoll());
|
||||
document.getElementById('add-option-btn').addEventListener('click', () => this.addOptionInput());
|
||||
|
||||
// Active poll
|
||||
document.getElementById('add-poll-option-btn').addEventListener('click', () => this.showAddOptionModal());
|
||||
document.getElementById('reset-poll-btn').addEventListener('click', () => this.handleResetPoll());
|
||||
document.getElementById('new-poll-btn').addEventListener('click', () => this.handleNewPoll());
|
||||
|
||||
// Modal controls
|
||||
document.getElementById('close-modal-btn').addEventListener('click', () => this.hideAddOptionModal());
|
||||
document.getElementById('cancel-modal-btn').addEventListener('click', () => this.hideAddOptionModal());
|
||||
document.getElementById('save-option-btn').addEventListener('click', () => this.handleAddOptionFromModal());
|
||||
|
||||
// Enter key handlers
|
||||
document.getElementById('room-id').addEventListener('keypress', (e) => {
|
||||
if (e.key === 'Enter') this.handleJoinRoom();
|
||||
});
|
||||
|
||||
document.getElementById('poll-question').addEventListener('keypress', (e) => {
|
||||
if (e.key === 'Enter') this.handleCreatePoll();
|
||||
});
|
||||
|
||||
document.getElementById('new-option-input').addEventListener('keypress', (e) => {
|
||||
if (e.key === 'Enter') this.handleAddOptionFromModal();
|
||||
});
|
||||
|
||||
// Close modal on background click
|
||||
document.getElementById('add-option-modal').addEventListener('click', (e) => {
|
||||
if (e.target.id === 'add-option-modal') {
|
||||
this.hideAddOptionModal();
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
handleJoinRoom() {
|
||||
const roomIdInput = document.getElementById('room-id');
|
||||
const peerId = roomIdInput.value.trim();
|
||||
|
||||
this.showLoading(true);
|
||||
|
||||
if (peerId) {
|
||||
// Connect to existing peer (host)
|
||||
this.peerManager.joinRoom(peerId)
|
||||
.then(() => {
|
||||
this.showNotification('Connected to host successfully!', 'success');
|
||||
this.showPollCreation();
|
||||
})
|
||||
.catch(error => {
|
||||
this.showNotification('Failed to connect: ' + error.message, 'error');
|
||||
console.error('Connect error:', error);
|
||||
})
|
||||
.finally(() => {
|
||||
this.showLoading(false);
|
||||
});
|
||||
} else {
|
||||
// Act as host - just show poll creation
|
||||
this.peerManager.createRoom();
|
||||
this.showNotification('You are the host. Share your Peer ID with others to connect.', 'success');
|
||||
this.showPollCreation();
|
||||
this.showLoading(false);
|
||||
}
|
||||
}
|
||||
|
||||
async copyPeerId() {
|
||||
const peerId = this.peerManager.getPeerId();
|
||||
if (peerId) {
|
||||
try {
|
||||
await navigator.clipboard.writeText(peerId);
|
||||
this.showNotification('Peer ID copied to clipboard!', 'success');
|
||||
} catch (error) {
|
||||
// Fallback for older browsers
|
||||
const textArea = document.createElement('textarea');
|
||||
textArea.value = peerId;
|
||||
document.body.appendChild(textArea);
|
||||
textArea.select();
|
||||
document.execCommand('copy');
|
||||
document.body.removeChild(textArea);
|
||||
this.showNotification('Peer ID copied to clipboard!', 'success');
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
handleCreatePoll() {
|
||||
const questionInput = document.getElementById('poll-question');
|
||||
const optionInputs = document.querySelectorAll('.option-text');
|
||||
const options = [];
|
||||
|
||||
console.log('Question input:', questionInput);
|
||||
console.log('Found option inputs:', optionInputs.length);
|
||||
|
||||
// Check if question input exists
|
||||
if (!questionInput) {
|
||||
console.error('Question input not found!');
|
||||
this.showNotification('Error: Question input not found', 'error');
|
||||
return;
|
||||
}
|
||||
|
||||
const question = questionInput.value ? questionInput.value.trim() : '';
|
||||
console.log('Question:', question);
|
||||
|
||||
optionInputs.forEach((input, index) => {
|
||||
console.log(`Input ${index}:`, input, 'value:', input?.value);
|
||||
if (input && typeof input.value !== 'undefined') {
|
||||
const text = input.value.trim();
|
||||
if (text) {
|
||||
options.push(text);
|
||||
}
|
||||
} else {
|
||||
console.warn(`Input ${index} is invalid or has no value property`);
|
||||
}
|
||||
});
|
||||
|
||||
console.log('Final options:', options);
|
||||
|
||||
if (!question) {
|
||||
this.showNotification('Please enter a poll question', 'error');
|
||||
return;
|
||||
}
|
||||
|
||||
if (options.length < 2) {
|
||||
this.showNotification('Please enter at least 2 options', 'error');
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
this.pollManager.createPoll(question, options);
|
||||
this.showNotification('Poll created successfully!', 'success');
|
||||
} catch (error) {
|
||||
console.error('Create poll error:', error);
|
||||
this.showNotification('Failed to create poll: ' + error.message, 'error');
|
||||
}
|
||||
}
|
||||
|
||||
showAddOptionModal() {
|
||||
const modal = document.getElementById('add-option-modal');
|
||||
const input = document.getElementById('new-option-input');
|
||||
modal.classList.remove('hidden');
|
||||
input.value = '';
|
||||
input.focus();
|
||||
}
|
||||
|
||||
hideAddOptionModal() {
|
||||
const modal = document.getElementById('add-option-modal');
|
||||
const input = document.getElementById('new-option-input');
|
||||
modal.classList.add('hidden');
|
||||
input.value = '';
|
||||
}
|
||||
|
||||
handleAddOptionFromModal() {
|
||||
const input = document.getElementById('new-option-input');
|
||||
const optionText = input.value.trim();
|
||||
|
||||
if (!optionText) {
|
||||
this.showNotification('Please enter an option text', 'error');
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
this.pollManager.addOption(optionText);
|
||||
this.hideAddOptionModal();
|
||||
this.showNotification('Option added successfully!', 'success');
|
||||
} catch (error) {
|
||||
this.showNotification('Failed to add option: ' + error.message, 'error');
|
||||
}
|
||||
}
|
||||
|
||||
handleAddPollOption() {
|
||||
this.showAddOptionModal();
|
||||
}
|
||||
|
||||
handleResetPoll() {
|
||||
if (confirm('Are you sure you want to reset all votes?')) {
|
||||
try {
|
||||
this.pollManager.resetPoll();
|
||||
this.showNotification('Poll reset successfully!', 'success');
|
||||
} catch (error) {
|
||||
this.showNotification('Failed to reset poll: ' + error.message, 'error');
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
handleNewPoll() {
|
||||
this.pollManager.createNewPoll();
|
||||
this.showPollCreation();
|
||||
this.clearPollForm();
|
||||
}
|
||||
|
||||
addOptionInput() {
|
||||
const container = document.getElementById('options-container');
|
||||
const optionCount = container.children.length;
|
||||
|
||||
const optionDiv = document.createElement('div');
|
||||
optionDiv.className = 'option-input';
|
||||
optionDiv.innerHTML = `
|
||||
<input type="text" class="option-text" placeholder="Option ${optionCount + 1}">
|
||||
<button class="remove-option-btn">Remove</button>
|
||||
`;
|
||||
|
||||
container.appendChild(optionDiv);
|
||||
this.updateOptionRemoveButtons();
|
||||
|
||||
// Add event listener to new input
|
||||
const newInput = optionDiv.querySelector('.option-text');
|
||||
newInput.addEventListener('keypress', (e) => {
|
||||
if (e.key === 'Enter') this.handleCreatePoll();
|
||||
});
|
||||
|
||||
// Add event listener to remove button
|
||||
const removeBtn = optionDiv.querySelector('.remove-option-btn');
|
||||
removeBtn.addEventListener('click', () => {
|
||||
optionDiv.remove();
|
||||
this.updateOptionRemoveButtons();
|
||||
this.updateOptionPlaceholders();
|
||||
});
|
||||
|
||||
// Focus on new input
|
||||
newInput.focus();
|
||||
}
|
||||
|
||||
updateOptionRemoveButtons() {
|
||||
const optionInputs = document.querySelectorAll('.option-input');
|
||||
const removeButtons = document.querySelectorAll('.remove-option-btn');
|
||||
|
||||
removeButtons.forEach((btn, index) => {
|
||||
if (optionInputs.length <= 2) {
|
||||
btn.classList.add('hidden');
|
||||
} else {
|
||||
btn.classList.remove('hidden');
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
updateOptionPlaceholders() {
|
||||
const optionInputs = document.querySelectorAll('.option-text');
|
||||
optionInputs.forEach((input, index) => {
|
||||
input.placeholder = `Option ${index + 1}`;
|
||||
});
|
||||
}
|
||||
|
||||
clearPollForm() {
|
||||
document.getElementById('poll-question').value = '';
|
||||
const container = document.getElementById('options-container');
|
||||
container.innerHTML = `
|
||||
<div class="option-input">
|
||||
<input type="text" class="option-text" placeholder="Option 1">
|
||||
<button class="remove-option-btn hidden">Remove</button>
|
||||
</div>
|
||||
<div class="option-input">
|
||||
<input type="text" class="option-text" placeholder="Option 2">
|
||||
<button class="remove-option-btn hidden">Remove</button>
|
||||
</div>
|
||||
`;
|
||||
this.updateOptionRemoveButtons();
|
||||
}
|
||||
|
||||
showPollCreation() {
|
||||
document.getElementById('connection-section').classList.add('hidden');
|
||||
document.getElementById('poll-creation').classList.remove('hidden');
|
||||
document.getElementById('active-poll').classList.add('hidden');
|
||||
}
|
||||
|
||||
showActivePoll() {
|
||||
document.getElementById('connection-section').classList.add('hidden');
|
||||
document.getElementById('poll-creation').classList.add('hidden');
|
||||
document.getElementById('active-poll').classList.remove('hidden');
|
||||
}
|
||||
|
||||
updatePeerId(peerId) {
|
||||
const element = document.getElementById('peer-id');
|
||||
if (element) {
|
||||
element.textContent = peerId || 'Loading...';
|
||||
}
|
||||
}
|
||||
|
||||
renderPoll(poll) {
|
||||
if (!poll) return;
|
||||
|
||||
this.showActivePoll();
|
||||
|
||||
// Update poll question
|
||||
document.getElementById('poll-question-display').textContent = poll.question;
|
||||
|
||||
// Render options
|
||||
const optionsContainer = document.getElementById('poll-options');
|
||||
optionsContainer.innerHTML = '';
|
||||
|
||||
const totalVotes = poll.options.reduce((sum, opt) => sum + opt.votes, 0);
|
||||
const myVotedOption = this.pollManager.getMyVotedOption();
|
||||
|
||||
poll.options.forEach(option => {
|
||||
const optionDiv = document.createElement('div');
|
||||
optionDiv.className = 'poll-option';
|
||||
|
||||
const hasVoted = this.pollManager.hasVoted(option.id);
|
||||
const myVotedOption = this.pollManager.getMyVotedOption();
|
||||
|
||||
console.log(`Option ${option.id}: hasVoted=${hasVoted}, myVotedOption=${myVotedOption}`);
|
||||
|
||||
if (hasVoted) {
|
||||
optionDiv.classList.add('voted');
|
||||
} else if (myVotedOption) {
|
||||
// User has voted but not for this option - this is a change vote option
|
||||
optionDiv.classList.add('change-vote');
|
||||
}
|
||||
|
||||
const percentage = totalVotes > 0 ? (option.votes / totalVotes * 100).toFixed(1) : 0;
|
||||
|
||||
optionDiv.innerHTML = `
|
||||
<div class="option-header">
|
||||
<span class="option-text">${option.text}</span>
|
||||
<span class="vote-count">${option.votes} votes</span>
|
||||
</div>
|
||||
<div class="vote-bar">
|
||||
<div class="vote-fill" style="width: ${percentage}%"></div>
|
||||
</div>
|
||||
`;
|
||||
|
||||
optionDiv.addEventListener('click', () => {
|
||||
const myVotedOption = this.pollManager.getMyVotedOption();
|
||||
|
||||
if (this.pollManager.hasVoted(option.id)) {
|
||||
// Clicking your current vote - unvote
|
||||
if (confirm('Remove your vote?')) {
|
||||
const success = this.pollManager.unvote();
|
||||
if (success) {
|
||||
this.showNotification('Vote removed!', 'success');
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// Either first vote or changing vote
|
||||
const success = this.pollManager.vote(option.id);
|
||||
if (success) {
|
||||
if (myVotedOption) {
|
||||
this.showNotification('Vote changed successfully!', 'success');
|
||||
} else {
|
||||
this.showNotification('Vote recorded!', 'success');
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
optionsContainer.appendChild(optionDiv);
|
||||
});
|
||||
}
|
||||
|
||||
showNotification(message, type = 'info') {
|
||||
const notification = document.getElementById('notification');
|
||||
notification.textContent = message;
|
||||
notification.className = `notification ${type}`;
|
||||
notification.classList.remove('hidden');
|
||||
|
||||
setTimeout(() => {
|
||||
notification.classList.add('hidden');
|
||||
}, 3000);
|
||||
}
|
||||
|
||||
showLoading(show) {
|
||||
const overlay = document.getElementById('loading-overlay');
|
||||
if (show) {
|
||||
overlay.classList.remove('hidden');
|
||||
} else {
|
||||
overlay.classList.add('hidden');
|
||||
}
|
||||
}
|
||||
|
||||
updatePeersList() {
|
||||
// Use the peer manager's updatePeersList method
|
||||
this.peerManager.updatePeersList();
|
||||
}
|
||||
|
||||
updatePollsList(polls) {
|
||||
const pollsList = document.getElementById('polls-list');
|
||||
|
||||
if (!pollsList) return;
|
||||
|
||||
if (polls.length === 0) {
|
||||
pollsList.innerHTML = '<p class="no-polls">No polls created yet</p>';
|
||||
return;
|
||||
}
|
||||
|
||||
pollsList.innerHTML = '';
|
||||
|
||||
polls.forEach(poll => {
|
||||
const pollDiv = document.createElement('div');
|
||||
pollDiv.className = 'poll-item';
|
||||
|
||||
if (this.pollManager.getCurrentPoll() && this.pollManager.getCurrentPoll().id === poll.id) {
|
||||
pollDiv.classList.add('active');
|
||||
}
|
||||
|
||||
const totalVotes = poll.options.reduce((sum, opt) => sum + opt.votes, 0);
|
||||
const createdDate = new Date(poll.createdAt).toLocaleString();
|
||||
|
||||
pollDiv.innerHTML = `
|
||||
<div class="poll-item-title">${poll.question}</div>
|
||||
<div class="poll-item-meta">${totalVotes} votes • ${createdDate}</div>
|
||||
`;
|
||||
|
||||
pollDiv.addEventListener('click', () => {
|
||||
this.pollManager.selectPoll(poll.id);
|
||||
});
|
||||
|
||||
pollsList.appendChild(pollDiv);
|
||||
});
|
||||
}
|
||||
}
|
||||
144
package-lock.json
generated
Normal file
144
package-lock.json
generated
Normal file
@@ -0,0 +1,144 @@
|
||||
{
|
||||
"name": "yjs-poll",
|
||||
"version": "1.0.0",
|
||||
"lockfileVersion": 3,
|
||||
"requires": true,
|
||||
"packages": {
|
||||
"": {
|
||||
"name": "yjs-poll",
|
||||
"version": "1.0.0",
|
||||
"license": "ISC",
|
||||
"dependencies": {
|
||||
"@y/protocols": "^1.0.6-1",
|
||||
"lib0": "^0.2.102",
|
||||
"ws": "^8.19.0",
|
||||
"yjs": "^14.0.0-7"
|
||||
}
|
||||
},
|
||||
"node_modules/@y/protocols": {
|
||||
"version": "1.0.6-rc.1",
|
||||
"resolved": "https://registry.npmjs.org/@y/protocols/-/protocols-1.0.6-rc.1.tgz",
|
||||
"integrity": "sha512-e/qs7hXcLk/SeNitxMXv2ymozyWFTULwbJEi7cAf/K/iXw9nGwGXHrR5TNluQ/bMwOX1cwuUT0hjEojkfH0gsA==",
|
||||
"license": "MIT",
|
||||
"dependencies": {
|
||||
"lib0": "^1.0.0-rc.1"
|
||||
},
|
||||
"engines": {
|
||||
"node": ">=16.0.0",
|
||||
"npm": ">=8.0.0"
|
||||
},
|
||||
"funding": {
|
||||
"type": "GitHub Sponsors ❤",
|
||||
"url": "https://github.com/sponsors/dmonad"
|
||||
},
|
||||
"peerDependencies": {
|
||||
"@y/y": "*"
|
||||
}
|
||||
},
|
||||
"node_modules/@y/protocols/node_modules/lib0": {
|
||||
"version": "1.0.0-rc.4",
|
||||
"resolved": "https://registry.npmjs.org/lib0/-/lib0-1.0.0-rc.4.tgz",
|
||||
"integrity": "sha512-mESL4089ji2HbS19DwAsYJRrtxZDjzMjEzjNR5kpzFBqlhvV0b7F9+SlA2SHOQU1Puu2A9xejKXtElK77swxLA==",
|
||||
"license": "MIT",
|
||||
"bin": {
|
||||
"0ecdsa-generate-keypair": "src/bin/0ecdsa-generate-keypair.js",
|
||||
"0gentesthtml": "src/bin/gentesthtml.js",
|
||||
"0serve": "src/bin/0serve.js"
|
||||
},
|
||||
"engines": {
|
||||
"node": ">=16"
|
||||
},
|
||||
"funding": {
|
||||
"type": "GitHub Sponsors ❤",
|
||||
"url": "https://github.com/sponsors/dmonad"
|
||||
}
|
||||
},
|
||||
"node_modules/@y/y": {
|
||||
"version": "14.0.0-16",
|
||||
"resolved": "https://registry.npmjs.org/@y/y/-/y-14.0.0-16.tgz",
|
||||
"integrity": "sha512-4zwbLnLannzUiEdJn9r3IS2FCVdHqADZHLqAivzAuKeoCGZ55JUFebp/YUtrYuTE2ZaBgctxpzulqAAHaahtIA==",
|
||||
"license": "MIT",
|
||||
"peer": true,
|
||||
"dependencies": {
|
||||
"lib0": "^0.2.115-6"
|
||||
},
|
||||
"engines": {
|
||||
"node": ">=16.0.0",
|
||||
"npm": ">=8.0.0"
|
||||
},
|
||||
"funding": {
|
||||
"type": "GitHub Sponsors ❤",
|
||||
"url": "https://github.com/sponsors/dmonad"
|
||||
}
|
||||
},
|
||||
"node_modules/isomorphic.js": {
|
||||
"version": "0.2.5",
|
||||
"resolved": "https://registry.npmjs.org/isomorphic.js/-/isomorphic.js-0.2.5.tgz",
|
||||
"integrity": "sha512-PIeMbHqMt4DnUP3MA/Flc0HElYjMXArsw1qwJZcm9sqR8mq3l8NYizFMty0pWwE/tzIGH3EKK5+jes5mAr85yw==",
|
||||
"license": "MIT",
|
||||
"funding": {
|
||||
"type": "GitHub Sponsors ❤",
|
||||
"url": "https://github.com/sponsors/dmonad"
|
||||
}
|
||||
},
|
||||
"node_modules/lib0": {
|
||||
"version": "0.2.117",
|
||||
"resolved": "https://registry.npmjs.org/lib0/-/lib0-0.2.117.tgz",
|
||||
"integrity": "sha512-DeXj9X5xDCjgKLU/7RR+/HQEVzuuEUiwldwOGsHK/sfAfELGWEyTcf0x+uOvCvK3O2zPmZePXWL85vtia6GyZw==",
|
||||
"license": "MIT",
|
||||
"dependencies": {
|
||||
"isomorphic.js": "^0.2.4"
|
||||
},
|
||||
"bin": {
|
||||
"0ecdsa-generate-keypair": "bin/0ecdsa-generate-keypair.js",
|
||||
"0gentesthtml": "bin/gentesthtml.js",
|
||||
"0serve": "bin/0serve.js"
|
||||
},
|
||||
"engines": {
|
||||
"node": ">=16"
|
||||
},
|
||||
"funding": {
|
||||
"type": "GitHub Sponsors ❤",
|
||||
"url": "https://github.com/sponsors/dmonad"
|
||||
}
|
||||
},
|
||||
"node_modules/ws": {
|
||||
"version": "8.19.0",
|
||||
"resolved": "https://registry.npmjs.org/ws/-/ws-8.19.0.tgz",
|
||||
"integrity": "sha512-blAT2mjOEIi0ZzruJfIhb3nps74PRWTCz1IjglWEEpQl5XS/UNama6u2/rjFkDDouqr4L67ry+1aGIALViWjDg==",
|
||||
"license": "MIT",
|
||||
"engines": {
|
||||
"node": ">=10.0.0"
|
||||
},
|
||||
"peerDependencies": {
|
||||
"bufferutil": "^4.0.1",
|
||||
"utf-8-validate": ">=5.0.2"
|
||||
},
|
||||
"peerDependenciesMeta": {
|
||||
"bufferutil": {
|
||||
"optional": true
|
||||
},
|
||||
"utf-8-validate": {
|
||||
"optional": true
|
||||
}
|
||||
}
|
||||
},
|
||||
"node_modules/yjs": {
|
||||
"version": "14.0.0-16",
|
||||
"resolved": "https://registry.npmjs.org/yjs/-/yjs-14.0.0-16.tgz",
|
||||
"integrity": "sha512-n7jMrQz4pgU/NFnf4qY53K2adR/fu6ViQ79qVIw6Og+BtuDs1hx3DjOi3iREVnA6tsxQXXVG3gvG0I2kpmAwoQ==",
|
||||
"license": "MIT",
|
||||
"dependencies": {
|
||||
"lib0": "^0.2.115-6"
|
||||
},
|
||||
"engines": {
|
||||
"node": ">=16.0.0",
|
||||
"npm": ">=8.0.0"
|
||||
},
|
||||
"funding": {
|
||||
"type": "GitHub Sponsors ❤",
|
||||
"url": "https://github.com/sponsors/dmonad"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
20
package.json
Normal file
20
package.json
Normal file
@@ -0,0 +1,20 @@
|
||||
{
|
||||
"name": "yjs-poll",
|
||||
"version": "1.0.0",
|
||||
"main": "index.js",
|
||||
"type": "module",
|
||||
"scripts": {
|
||||
"test": "echo \"Error: no test specified\" && exit 1",
|
||||
"y-websocket": "y-websocket"
|
||||
},
|
||||
"keywords": [],
|
||||
"author": "",
|
||||
"license": "ISC",
|
||||
"description": "",
|
||||
"dependencies": {
|
||||
"ws": "^8.19.0",
|
||||
"@y/protocols": "^1.0.6-1",
|
||||
"lib0": "^0.2.102",
|
||||
"yjs": "^14.0.0-7"
|
||||
}
|
||||
}
|
||||
291
utils.js
Normal file
291
utils.js
Normal file
@@ -0,0 +1,291 @@
|
||||
/* Adapted from https://github.com/yjs/y-websocket-server/blob/main/src/server.js, version of 2025-04-02
|
||||
(author: dmonad/Kevin Jahns, MIT license) */
|
||||
|
||||
import * as Y from 'yjs'
|
||||
import * as syncProtocol from '@y/protocols/sync'
|
||||
import * as awarenessProtocol from '@y/protocols/awareness'
|
||||
|
||||
import * as encoding from 'lib0/encoding'
|
||||
import * as decoding from 'lib0/decoding'
|
||||
import * as map from 'lib0/map'
|
||||
|
||||
import * as eventloop from 'lib0/eventloop'
|
||||
|
||||
/*import { callbackHandler, isCallbackSet } from './callback.js'
|
||||
|
||||
const CALLBACK_DEBOUNCE_WAIT = parseInt(process.env.CALLBACK_DEBOUNCE_WAIT || '2000')
|
||||
const CALLBACK_DEBOUNCE_MAXWAIT = parseInt(process.env.CALLBACK_DEBOUNCE_MAXWAIT || '10000')
|
||||
|
||||
const debouncer = eventloop.createDebouncer(CALLBACK_DEBOUNCE_WAIT, CALLBACK_DEBOUNCE_MAXWAIT)*/
|
||||
|
||||
const wsReadyStateConnecting = 0
|
||||
const wsReadyStateOpen = 1
|
||||
const wsReadyStateClosing = 2 // eslint-disable-line
|
||||
const wsReadyStateClosed = 3 // eslint-disable-line
|
||||
|
||||
// disable gc when using snapshots!
|
||||
const gcEnabled = process.env.GC !== 'false' && process.env.GC !== '0'
|
||||
// const persistenceDir = process.env.YPERSISTENCE
|
||||
/**
|
||||
* @type {{bindState: function(string,WSSharedDoc):void, writeState:function(string,WSSharedDoc):Promise<any>, provider: any}|null}
|
||||
*/
|
||||
let persistence = null
|
||||
|
||||
/**
|
||||
* @param {{bindState: function(string,WSSharedDoc):void,
|
||||
* writeState:function(string,WSSharedDoc):Promise<any>,provider:any}|null} persistence_
|
||||
*/
|
||||
export const setPersistence = persistence_ => {
|
||||
persistence = persistence_
|
||||
}
|
||||
|
||||
/**
|
||||
* @return {null|{bindState: function(string,WSSharedDoc):void,
|
||||
* writeState:function(string,WSSharedDoc):Promise<any>}|null} used persistence layer
|
||||
*/
|
||||
export const getPersistence = () => persistence
|
||||
|
||||
/**
|
||||
* @type {Map<string,WSSharedDoc>}
|
||||
*/
|
||||
export const docs = new Map()
|
||||
|
||||
const messageSync = 0
|
||||
const messageAwareness = 1
|
||||
// const messageAuth = 2
|
||||
//const messageTextMessage = 3
|
||||
//const messageStateMessage = 4
|
||||
|
||||
/**
|
||||
* @param {Uint8Array} update
|
||||
* @param {any} _origin
|
||||
* @param {WSSharedDoc} doc
|
||||
* @param {any} _tr
|
||||
*/
|
||||
const updateHandler = (update, _origin, doc, _tr) => {
|
||||
const encoder = encoding.createEncoder()
|
||||
encoding.writeVarUint(encoder, messageSync)
|
||||
syncProtocol.writeUpdate(encoder, update)
|
||||
const message = encoding.toUint8Array(encoder)
|
||||
doc.conns.forEach((_, conn) => send(doc, conn, message))
|
||||
}
|
||||
|
||||
/**
|
||||
* @type {(ydoc: Y.Doc) => Promise<void>}
|
||||
*/
|
||||
let contentInitializor = _ydoc => Promise.resolve()
|
||||
|
||||
/**
|
||||
* This function is called once every time a Yjs document is created. You can
|
||||
* use it to pull data from an external source or initialize content.
|
||||
*
|
||||
* @param {(ydoc: Y.Doc) => Promise<void>} f
|
||||
*/
|
||||
export const setContentInitializor = (f) => {
|
||||
contentInitializor = f
|
||||
}
|
||||
|
||||
export class WSSharedDoc extends Y.Doc {
|
||||
/**
|
||||
* @param {string} name
|
||||
*/
|
||||
constructor (name) {
|
||||
super({ gc: gcEnabled })
|
||||
this.name = name
|
||||
/**
|
||||
* Maps from conn to set of controlled user ids. Delete all user ids from awareness when this conn is closed
|
||||
* @type {Map<Object, Set<number>>}
|
||||
*/
|
||||
this.conns = new Map()
|
||||
/**
|
||||
* @type {awarenessProtocol.Awareness}
|
||||
*/
|
||||
this.awareness = new awarenessProtocol.Awareness(this)
|
||||
this.awareness.setLocalState(null)
|
||||
/**
|
||||
* @param {{ added: Array<number>, updated: Array<number>, removed: Array<number> }} changes
|
||||
* @param {Object | null} conn Origin is the connection that made the change
|
||||
*/
|
||||
const awarenessChangeHandler = ({ added, updated, removed }, conn) => {
|
||||
const changedClients = added.concat(updated, removed)
|
||||
if (conn !== null) {
|
||||
const connControlledIDs = /** @type {Set<number>} */ (this.conns.get(conn))
|
||||
if (connControlledIDs !== undefined) {
|
||||
added.forEach(clientID => { connControlledIDs.add(clientID) })
|
||||
removed.forEach(clientID => { connControlledIDs.delete(clientID) })
|
||||
}
|
||||
}
|
||||
// broadcast awareness update
|
||||
const encoder = encoding.createEncoder()
|
||||
encoding.writeVarUint(encoder, messageAwareness)
|
||||
encoding.writeVarUint8Array(encoder, awarenessProtocol.encodeAwarenessUpdate(this.awareness, changedClients))
|
||||
const buff = encoding.toUint8Array(encoder)
|
||||
this.conns.forEach((_, c) => {
|
||||
send(this, c, buff)
|
||||
})
|
||||
}
|
||||
this.awareness.on('update', awarenessChangeHandler)
|
||||
this.on('update', /** @type {any} */ (updateHandler))
|
||||
//if (isCallbackSet) {
|
||||
// this.on('update', (_update, _origin, doc) => {
|
||||
// debouncer(() => callbackHandler(/** @type {WSSharedDoc} */ (doc)))
|
||||
// })
|
||||
//}
|
||||
this.whenInitialized = contentInitializor(this)
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets a Y.Doc by name, whether in memory or on disk
|
||||
*
|
||||
* @param {string} docname - the name of the Y.Doc to find or create
|
||||
* @param {boolean} gc - whether to allow gc on the doc (applies only when created)
|
||||
* @return {WSSharedDoc}
|
||||
*/
|
||||
export const getYDoc = (docname, gc = true) => map.setIfUndefined(docs, docname, () => {
|
||||
const doc = new WSSharedDoc(docname)
|
||||
doc.gc = gc
|
||||
if (persistence !== null) {
|
||||
persistence.bindState(docname, doc)
|
||||
}
|
||||
docs.set(docname, doc)
|
||||
return doc
|
||||
})
|
||||
|
||||
/**
|
||||
* @param {any} conn
|
||||
* @param {WSSharedDoc} doc
|
||||
* @param {Uint8Array} message
|
||||
*/
|
||||
const messageListener = (conn, doc, message) => {
|
||||
try {
|
||||
const encoder = encoding.createEncoder()
|
||||
const decoder = decoding.createDecoder(new Uint8Array(message))
|
||||
const messageType = decoding.readVarUint(decoder)
|
||||
switch (messageType) {
|
||||
case messageSync:
|
||||
encoding.writeVarUint(encoder, messageSync)
|
||||
syncProtocol.readSyncMessage(decoder, encoder, doc, conn)
|
||||
|
||||
// If the `encoder` only contains the type of reply message and no
|
||||
// message, there is no need to send the message. When `encoder` only
|
||||
// contains the type of reply, its length is 1.
|
||||
if (encoding.length(encoder) > 1) {
|
||||
send(doc, conn, encoding.toUint8Array(encoder))
|
||||
}
|
||||
break
|
||||
case messageAwareness: {
|
||||
awarenessProtocol.applyAwarenessUpdate(doc.awareness, decoding.readVarUint8Array(decoder), conn)
|
||||
break
|
||||
}
|
||||
// If this is a text or a message (no Yjs logic) - echo the message back to the client
|
||||
default: {
|
||||
conn.send(String(message))
|
||||
console.log(String(message))
|
||||
break
|
||||
}
|
||||
}
|
||||
} catch (err) {
|
||||
console.error(err)
|
||||
// @ts-ignore
|
||||
doc.emit('error', [err])
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* @param {WSSharedDoc} doc
|
||||
* @param {any} conn
|
||||
*/
|
||||
const closeConn = (doc, conn) => {
|
||||
if (doc.conns.has(conn)) {
|
||||
/**
|
||||
* @type {Set<number>}
|
||||
*/
|
||||
// @ts-ignore
|
||||
const controlledIds = doc.conns.get(conn)
|
||||
doc.conns.delete(conn)
|
||||
awarenessProtocol.removeAwarenessStates(doc.awareness, Array.from(controlledIds), null)
|
||||
if (doc.conns.size === 0 && persistence !== null) {
|
||||
// if persisted, we store state and destroy ydocument
|
||||
persistence.writeState(doc.name, doc).then(() => {
|
||||
doc.destroy()
|
||||
})
|
||||
docs.delete(doc.name)
|
||||
}
|
||||
}
|
||||
conn.close()
|
||||
}
|
||||
|
||||
/**
|
||||
* @param {WSSharedDoc} doc
|
||||
* @param {import('ws').WebSocket} conn
|
||||
* @param {Uint8Array} m
|
||||
*/
|
||||
const send = (doc, conn, m) => {
|
||||
if (conn.readyState !== wsReadyStateConnecting && conn.readyState !== wsReadyStateOpen) {
|
||||
closeConn(doc, conn)
|
||||
}
|
||||
try {
|
||||
conn.send(m, {}, err => { err != null && closeConn(doc, conn) })
|
||||
} catch (e) {
|
||||
closeConn(doc, conn)
|
||||
}
|
||||
}
|
||||
|
||||
const pingTimeout = 30000
|
||||
|
||||
/**
|
||||
* @param {import('ws').WebSocket} conn
|
||||
* @param {import('http').IncomingMessage} req
|
||||
* @param {any} opts
|
||||
*/
|
||||
export const setupWSConnection = (conn, req, { docName = (req.url || '').slice(1).split('?')[0], gc = true } = {}) => {
|
||||
conn.binaryType = 'arraybuffer'
|
||||
// get doc, initialize if it does not exist yet
|
||||
const doc = getYDoc(docName, gc)
|
||||
doc.conns.set(conn, new Set())
|
||||
// listen and reply to events
|
||||
conn.on('message', /** @param {ArrayBuffer} message */ message => messageListener(conn, doc, message))
|
||||
|
||||
// Check if connection is still alive
|
||||
let pongReceived = true
|
||||
const pingInterval = setInterval(() => {
|
||||
if (!pongReceived) {
|
||||
if (doc.conns.has(conn)) {
|
||||
closeConn(doc, conn)
|
||||
}
|
||||
clearInterval(pingInterval)
|
||||
} else if (doc.conns.has(conn)) {
|
||||
pongReceived = false
|
||||
try {
|
||||
conn.ping()
|
||||
} catch (e) {
|
||||
closeConn(doc, conn)
|
||||
clearInterval(pingInterval)
|
||||
}
|
||||
}
|
||||
}, pingTimeout)
|
||||
conn.on('close', () => {
|
||||
closeConn(doc, conn)
|
||||
clearInterval(pingInterval)
|
||||
})
|
||||
conn.on('pong', () => {
|
||||
pongReceived = true
|
||||
})
|
||||
// put the following in a variables in a block so the interval handlers don't keep in in
|
||||
// scope
|
||||
{
|
||||
// send sync step 1
|
||||
const encoder = encoding.createEncoder()
|
||||
encoding.writeVarUint(encoder, messageSync)
|
||||
syncProtocol.writeSyncStep1(encoder, doc)
|
||||
send(doc, conn, encoding.toUint8Array(encoder))
|
||||
const awarenessStates = doc.awareness.getStates()
|
||||
if (awarenessStates.size > 0) {
|
||||
const encoder = encoding.createEncoder()
|
||||
encoding.writeVarUint(encoder, messageAwareness)
|
||||
encoding.writeVarUint8Array(encoder, awarenessProtocol.encodeAwarenessUpdate(doc.awareness, Array.from(awarenessStates.keys())))
|
||||
send(doc, conn, encoding.toUint8Array(encoder))
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user