Compare commits
No commits in common. "main" and "0.0.9" have entirely different histories.
50 changed files with 1462 additions and 15493 deletions
10
.env.example
10
.env.example
|
|
@ -1,6 +1,5 @@
|
||||||
DATABASE_URL=postgres://postgres:postgres@localhost:5432/chattz
|
DATABASE_URL=postgres://postgres:postgres@localhost:5432/chattz
|
||||||
PORT=3000
|
PORT=3000
|
||||||
APP_BASE_URL=http://localhost:3000
|
|
||||||
|
|
||||||
# Authentik OIDC app values
|
# Authentik OIDC app values
|
||||||
OIDC_CLIENT_ID=replace-me
|
OIDC_CLIENT_ID=replace-me
|
||||||
|
|
@ -17,9 +16,6 @@ TURN_URLS=turn:turn.example.com:3478?transport=udp,turn:turn.example.com:3478?tr
|
||||||
TURN_USERNAME=replace-me
|
TURN_USERNAME=replace-me
|
||||||
TURN_PASSWORD=replace-me
|
TURN_PASSWORD=replace-me
|
||||||
|
|
||||||
# Cloudflare R2 media uploads
|
# 32+ random chars; used to sign session cookies
|
||||||
R2_ACCOUNT_ID=replace-me
|
SESSION_SECRET=replace-with-long-random-secret
|
||||||
R2_ACCESS_KEY_ID=replace-me
|
COOKIE_SECURE=false
|
||||||
R2_SECRET_ACCESS_KEY=replace-me
|
|
||||||
R2_BUCKET=chattz-media
|
|
||||||
MEDIA_BASE_URL=https://media.example.com
|
|
||||||
|
|
|
||||||
|
|
@ -11,51 +11,6 @@ jobs:
|
||||||
with:
|
with:
|
||||||
node-version: 24
|
node-version: 24
|
||||||
- uses: actions/checkout@v6
|
- uses: actions/checkout@v6
|
||||||
with:
|
|
||||||
fetch-depth: 0
|
|
||||||
- name: Update version from tag
|
|
||||||
run: |
|
|
||||||
# Extract version from tag (e.g. v1.2.3 -> 1.2.3)
|
|
||||||
VERSION=${GITHUB_REF_NAME#v}
|
|
||||||
echo "Bumping version to $VERSION"
|
|
||||||
npm version $VERSION --no-git-tag-version
|
|
||||||
- run: npm ci
|
|
||||||
- run: npm run dist:linux
|
|
||||||
- name: Build Windows installer with 32-bit Wine prefix
|
|
||||||
run: |
|
|
||||||
export WINEARCH=win32
|
|
||||||
export WINEPREFIX="$HOME/.wine32"
|
|
||||||
wineboot -u
|
|
||||||
npm run dist:win
|
|
||||||
- name: Copy installers into static folder
|
|
||||||
run: |
|
|
||||||
mkdir -p static/installers
|
|
||||||
find static/installers -mindepth 1 -maxdepth 1 -type f -delete
|
|
||||||
copy_first() {
|
|
||||||
pattern="$1"
|
|
||||||
out="$2"
|
|
||||||
file="$(find dist -maxdepth 1 -type f -name "$pattern" | head -n 1)"
|
|
||||||
if [ -n "$file" ]; then
|
|
||||||
cp "$file" "static/installers/$out"
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
# Copy versioned artifacts for electron-updater
|
|
||||||
cp dist/*.AppImage static/installers/ || true
|
|
||||||
cp dist/*.exe static/installers/ || true
|
|
||||||
cp dist/*.rpm static/installers/ || true
|
|
||||||
cp dist/*.deb static/installers/ || true
|
|
||||||
cp dist/*.msi static/installers/ || true
|
|
||||||
|
|
||||||
# Maintain generic names for stable website links
|
|
||||||
copy_first '*.rpm' 'chattz-linux.rpm'
|
|
||||||
copy_first '*.deb' 'chattz-linux.deb'
|
|
||||||
copy_first '*.AppImage' 'chattz-linux.AppImage'
|
|
||||||
copy_first '*.exe' 'chattz-windows.exe'
|
|
||||||
copy_first '*.msi' 'chattz-windows.msi'
|
|
||||||
|
|
||||||
# Metadata files for electron-updater
|
|
||||||
copy_first 'latest-linux.yml' 'latest-linux.yml'
|
|
||||||
copy_first 'latest.yml' 'latest.yml'
|
|
||||||
- name: Cache Cargo registry
|
- name: Cache Cargo registry
|
||||||
uses: actions/cache@v4
|
uses: actions/cache@v4
|
||||||
with:
|
with:
|
||||||
|
|
@ -82,7 +37,7 @@ jobs:
|
||||||
mkdir -p ~/discord
|
mkdir -p ~/discord
|
||||||
cp target/release/chattz ~/discord/discord
|
cp target/release/chattz ~/discord/discord
|
||||||
chmod +x ~/discord/discord
|
chmod +x ~/discord/discord
|
||||||
mkdir -p ~/discord/static/uploads
|
mkdir -p ~/discord/static
|
||||||
find ~/discord/static -mindepth 1 -maxdepth 1 ! -name 'uploads' -exec rm -rf {} +
|
rm -rf ~/discord/static/*
|
||||||
cp -r static/* ~/discord/static/
|
cp -r static/* ~/discord/static/
|
||||||
systemctl --user start discord
|
systemctl --user start discord
|
||||||
2
.gitignore
vendored
2
.gitignore
vendored
|
|
@ -1,4 +1,2 @@
|
||||||
/target
|
/target
|
||||||
.env
|
.env
|
||||||
node_modules
|
|
||||||
dist
|
|
||||||
|
|
|
||||||
891
Cargo.lock
generated
891
Cargo.lock
generated
File diff suppressed because it is too large
Load diff
11
Cargo.toml
11
Cargo.toml
|
|
@ -5,20 +5,17 @@ edition = "2024"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
anyhow = "1"
|
anyhow = "1"
|
||||||
aws-config = { version = "1", default-features = false, features = ["behavior-version-latest", "rt-tokio", "rustls"] }
|
axum = { version = "0.8", features = ["macros", "ws"] }
|
||||||
aws-sdk-s3 = { version = "1", default-features = false, features = ["rt-tokio", "rustls"] }
|
|
||||||
axum = { version = "0.8", features = ["macros", "ws", "multipart"] }
|
|
||||||
chrono = { version = "0.4", features = ["serde"] }
|
chrono = { version = "0.4", features = ["serde"] }
|
||||||
dotenvy = "0.15"
|
dotenvy = "0.15"
|
||||||
|
jsonwebtoken = "9"
|
||||||
reqwest = { version = "0.12", default-features = false, features = ["json", "rustls-tls"] }
|
reqwest = { version = "0.12", default-features = false, features = ["json", "rustls-tls"] }
|
||||||
sea-orm = { version = "1.1", default-features = false, features = ["sqlx-postgres", "runtime-tokio-rustls", "macros", "with-chrono", "with-uuid", "mock"] }
|
sea-orm = { version = "1.1", default-features = false, features = ["sqlx-postgres", "runtime-tokio-rustls", "macros", "with-chrono", "with-uuid"] }
|
||||||
sea-orm-migration = { version = "1.1", default-features = false, features = ["sqlx-postgres", "runtime-tokio-rustls"] }
|
sea-orm-migration = { version = "1.1", default-features = false, features = ["sqlx-postgres", "runtime-tokio-rustls"] }
|
||||||
serde = { version = "1", features = ["derive"] }
|
serde = { version = "1", features = ["derive"] }
|
||||||
serde_json = "1"
|
serde_json = "1"
|
||||||
sha2 = "0.10"
|
|
||||||
futures-util = "0.3"
|
futures-util = "0.3"
|
||||||
tokio = { version = "1", features = ["fs", "io-util", "macros", "rt-multi-thread"] }
|
tokio = { version = "1", features = ["macros", "rt-multi-thread"] }
|
||||||
tower = { version = "0.5", features = ["util"] }
|
|
||||||
tower-http = { version = "0.6", features = ["trace", "fs"] }
|
tower-http = { version = "0.6", features = ["trace", "fs"] }
|
||||||
tracing = "0.1"
|
tracing = "0.1"
|
||||||
tracing-subscriber = { version = "0.3", features = ["env-filter", "fmt"] }
|
tracing-subscriber = { version = "0.3", features = ["env-filter", "fmt"] }
|
||||||
|
|
|
||||||
11
README.md
11
README.md
|
|
@ -8,7 +8,7 @@ A simple single-instance Discord-style monolith in Rust using:
|
||||||
## What this includes
|
## What this includes
|
||||||
|
|
||||||
- OIDC login flow (`/auth/login`, `/auth/callback`, `/auth/logout`)
|
- OIDC login flow (`/auth/login`, `/auth/callback`, `/auth/logout`)
|
||||||
- HttpOnly session cookie auth
|
- Signed session cookie auth
|
||||||
- Channel voice chat over WebRTC (P2P mesh) with server WebSocket signaling
|
- Channel voice chat over WebRTC (P2P mesh) with server WebSocket signaling
|
||||||
- Guild invite codes (create + join)
|
- Guild invite codes (create + join)
|
||||||
- Direct messages (DM) between users
|
- Direct messages (DM) between users
|
||||||
|
|
@ -41,11 +41,6 @@ For voice reliability on restrictive networks, configure TURN in `.env`:
|
||||||
- `TURN_USERNAME`
|
- `TURN_USERNAME`
|
||||||
- `TURN_PASSWORD`
|
- `TURN_PASSWORD`
|
||||||
|
|
||||||
For production deployments:
|
|
||||||
- `APP_BASE_URL` must be your public app origin and should use `https`
|
|
||||||
- `MEDIA_BASE_URL` should be a separate media origin for user uploads
|
|
||||||
- uploads and soundboard require R2/object storage to be configured
|
|
||||||
|
|
||||||
3. Run app:
|
3. Run app:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
|
|
@ -60,7 +55,7 @@ Web UI is available at `http://localhost:${PORT}/`.
|
||||||
## Authentik setup notes
|
## Authentik setup notes
|
||||||
|
|
||||||
Create an Authentik OAuth2/OIDC provider + application and set:
|
Create an Authentik OAuth2/OIDC provider + application and set:
|
||||||
- Redirect URI: `${APP_BASE_URL}/auth/callback`
|
- Redirect URI: `http://localhost:3000/auth/callback`
|
||||||
- Scopes including at least: `openid profile email`
|
- Scopes including at least: `openid profile email`
|
||||||
|
|
||||||
If you change `PORT`, update `OIDC_REDIRECT_URL` and this redirect URI to match.
|
If you change `PORT`, update `OIDC_REDIRECT_URL` and this redirect URI to match.
|
||||||
|
|
@ -96,7 +91,6 @@ For Authentik these are commonly under `/application/o/...` for the app slug.
|
||||||
- `GET /channels/:channel_id/voice/ws` (WebSocket signaling)
|
- `GET /channels/:channel_id/voice/ws` (WebSocket signaling)
|
||||||
|
|
||||||
All endpoints except health and auth flow require the session cookie from successful login.
|
All endpoints except health and auth flow require the session cookie from successful login.
|
||||||
Authenticated WebSocket connections (`/ws`, `/channels/:channel_id/voice/ws`) also use the same cookie session.
|
|
||||||
|
|
||||||
## Notes
|
## Notes
|
||||||
|
|
||||||
|
|
@ -104,7 +98,6 @@ This is intentionally minimal and monolithic (single process, single Postgres in
|
||||||
Voice is implemented as browser-to-browser WebRTC audio with signaling in this server.
|
Voice is implemented as browser-to-browser WebRTC audio with signaling in this server.
|
||||||
For two users behind strict NAT/firewall, you may need TURN for reliable connectivity.
|
For two users behind strict NAT/firewall, you may need TURN for reliable connectivity.
|
||||||
The web UI remembers the last selected guild in browser local storage and auto-selects it on reload.
|
The web UI remembers the last selected guild in browser local storage and auto-selects it on reload.
|
||||||
User uploads are served from the configured media origin, not from `/static`.
|
|
||||||
|
|
||||||
Mic filter modes in the UI:
|
Mic filter modes in the UI:
|
||||||
- `NSNet2 (Compat)`: always-on denoising mode (implemented using DeepFilterNet3 with lighter suppression preset)
|
- `NSNet2 (Compat)`: always-on denoising mode (implemented using DeepFilterNet3 with lighter suppression preset)
|
||||||
|
|
|
||||||
BIN
build/icon.png
BIN
build/icon.png
Binary file not shown.
|
Before Width: | Height: | Size: 430 KiB |
|
|
@ -1 +0,0 @@
|
||||||
import "../static/shared/app-core.js";
|
|
||||||
|
|
@ -1,282 +0,0 @@
|
||||||
<!-- Generated from shared-html/index.template.html. Do not edit directly. -->
|
|
||||||
<!doctype html>
|
|
||||||
<html lang="en">
|
|
||||||
|
|
||||||
<head>
|
|
||||||
<meta charset="UTF-8" />
|
|
||||||
<meta name="viewport" content="width=device-width, initial-scale=1.0, viewport-fit=cover" />
|
|
||||||
<title>Chattz</title>
|
|
||||||
<link rel="stylesheet" href="../static/styles.css" />
|
|
||||||
<!-- Lucide Icons -->
|
|
||||||
<script src="../static/vendor/lucide.min.js"></script>
|
|
||||||
</head>
|
|
||||||
|
|
||||||
<body>
|
|
||||||
<div class="auth-screen" id="auth-screen">
|
|
||||||
<div class="auth-card">
|
|
||||||
<div class="brand-large">C</div>
|
|
||||||
<h1>Chattz</h1>
|
|
||||||
<p>Sign in to open your servers.</p>
|
|
||||||
<button id="login-btn">Login with Authentik</button>
|
|
||||||
<p id="status" class="status-line"></p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="shell hidden" id="main">
|
|
||||||
<aside class="server-rail">
|
|
||||||
<div class="brand" title="Home">
|
|
||||||
<i data-lucide="message-square"></i>
|
|
||||||
</div>
|
|
||||||
<div class="separator"></div>
|
|
||||||
<div id="guild-list" class="guild-list"></div>
|
|
||||||
<button id="add-guild-btn" class="guild-pill action-pill" title="Add a Server">
|
|
||||||
<i data-lucide="plus"></i>
|
|
||||||
</button>
|
|
||||||
</aside>
|
|
||||||
|
|
||||||
<aside class="channel-sidebar">
|
|
||||||
<header class="sidebar-header clickable" id="guild-header">
|
|
||||||
<h2 id="guild-title">No server selected</h2>
|
|
||||||
<div class="sidebar-header-actions">
|
|
||||||
<span id="invite-copied-badge" class="copied-badge hidden">Copied!</span>
|
|
||||||
<button id="create-invite-btn" class="header-action-btn" title="Create Invite Link">
|
|
||||||
<i data-lucide="link-2"></i>
|
|
||||||
</button>
|
|
||||||
<i data-lucide="chevron-down"></i>
|
|
||||||
</div>
|
|
||||||
</header>
|
|
||||||
|
|
||||||
<div class="sidebar-scroll">
|
|
||||||
<section class="sidebar-group">
|
|
||||||
<div class="group-title">
|
|
||||||
<i data-lucide="chevron-down" class="group-toggle"></i>
|
|
||||||
<span>Text Channels</span>
|
|
||||||
<button id="add-text-btn" class="add-btn" type="button" title="Create Text Channel">
|
|
||||||
<i data-lucide="plus"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
<div id="channel-list" class="channel-list"></div>
|
|
||||||
</section>
|
|
||||||
|
|
||||||
<section class="sidebar-group">
|
|
||||||
<div class="group-title">
|
|
||||||
<i data-lucide="chevron-down" class="group-toggle"></i>
|
|
||||||
<span>Voice Channels</span>
|
|
||||||
<button id="add-voice-btn" class="add-btn" type="button" title="Create Voice Channel">
|
|
||||||
<i data-lucide="plus"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
<div id="voice-channel-list" class="channel-list"></div>
|
|
||||||
</section>
|
|
||||||
|
|
||||||
<section class="sidebar-group">
|
|
||||||
<div class="group-title">
|
|
||||||
<i data-lucide="chevron-down" class="group-toggle"></i>
|
|
||||||
<span>Direct Messages</span>
|
|
||||||
</div>
|
|
||||||
<div id="dm-list" class="channel-list"></div>
|
|
||||||
</section>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="sidebar-footer">
|
|
||||||
<div id="voice-connection" class="voice-connection hidden">
|
|
||||||
<div class="vc-info">
|
|
||||||
<i data-lucide="signal-high" class="vc-icon"></i>
|
|
||||||
<div class="vc-text">
|
|
||||||
<span class="vc-status">Voice Connected</span>
|
|
||||||
<span id="vc-channel-name" class="vc-name">General</span>
|
|
||||||
</div>
|
|
||||||
<div class="vc-actions">
|
|
||||||
<button id="voice-video-btn" title="Turn on Camera"><i data-lucide="video-off"></i></button>
|
|
||||||
<button id="voice-screen-btn" title="Share Screen"><i data-lucide="monitor-off"></i></button>
|
|
||||||
<button id="voice-mute-btn" title="Mute"><i data-lucide="mic"></i></button>
|
|
||||||
<button id="voice-leave-btn" title="Disconnect"><i data-lucide="phone-off"></i></button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div id="soundboard" class="soundboard-container hidden">
|
|
||||||
<div class="soundboard-header">
|
|
||||||
<span>Sound Board</span>
|
|
||||||
<span id="add-sound-btn" class="btn-add-sound">+ Add Sound</span>
|
|
||||||
</div>
|
|
||||||
<div id="soundboard-grid" class="soundboard-grid"></div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="user-panel">
|
|
||||||
<div class="avatar-wrapper">
|
|
||||||
<div id="user-avatar" class="avatar">U</div>
|
|
||||||
<div class="status-dot online"></div>
|
|
||||||
</div>
|
|
||||||
<div class="user-info">
|
|
||||||
<div id="user-name" class="display-name">Username</div>
|
|
||||||
<div class="user-status">Online</div>
|
|
||||||
</div>
|
|
||||||
<div class="user-actions">
|
|
||||||
<div id="update-notifier" class="update-notifier hidden">
|
|
||||||
<button id="update-download-btn" title="Download Update"><i data-lucide="download"></i></button>
|
|
||||||
<button id="update-install-btn" title="Install Update" class="hidden"><i data-lucide="arrow-up-circle"></i></button>
|
|
||||||
</div>
|
|
||||||
<button id="user-settings-btn" title="Settings"><i data-lucide="settings"></i></button>
|
|
||||||
<button id="logout-btn" title="Logout"><i data-lucide="log-out"></i></button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</aside>
|
|
||||||
|
|
||||||
<main class="chat-pane">
|
|
||||||
<header class="chat-header">
|
|
||||||
<button id="mobile-menu-btn" class="mobile-toggle-btn" title="Menu">
|
|
||||||
<i data-lucide="menu"></i>
|
|
||||||
</button>
|
|
||||||
<i data-lucide="hash" class="header-icon"></i>
|
|
||||||
<h3 id="channel-title">Select a channel</h3>
|
|
||||||
<div style="flex:1"></div>
|
|
||||||
<button id="mobile-members-btn" class="mobile-toggle-btn" title="Members">
|
|
||||||
<i data-lucide="users"></i>
|
|
||||||
</button>
|
|
||||||
</header>
|
|
||||||
|
|
||||||
<div id="video-grid" class="video-grid hidden"></div>
|
|
||||||
<div id="message-list" class="message-list"></div>
|
|
||||||
|
|
||||||
<div class="chat-input-wrapper">
|
|
||||||
<form id="message-form" class="message-form">
|
|
||||||
<input id="media-file-input" type="file" class="hidden-file-input" accept="image/*,video/*,audio/*,.pdf,.txt,.zip,.rar,.7z,.tar,.gz,.json,.csv,.md" />
|
|
||||||
<button type="button" id="media-upload-btn" class="input-action-btn" title="Upload File">
|
|
||||||
<i data-lucide="paperclip"></i>
|
|
||||||
</button>
|
|
||||||
<button type="button" id="gif-btn" class="input-action-btn" title="Open GIF Picker">
|
|
||||||
<i data-lucide="image"></i>
|
|
||||||
</button>
|
|
||||||
<input id="message-body" autocomplete="off" placeholder="Message #channel" maxlength="4000" required />
|
|
||||||
<button type="submit" class="hidden"></button>
|
|
||||||
</form>
|
|
||||||
</div>
|
|
||||||
</main>
|
|
||||||
|
|
||||||
<aside class="utility-sidebar" id="utility-sidebar">
|
|
||||||
<header class="sidebar-header">
|
|
||||||
<h2>Members</h2>
|
|
||||||
</header>
|
|
||||||
<div class="member-list-wrapper">
|
|
||||||
<div id="member-list" class="member-list"></div>
|
|
||||||
</div>
|
|
||||||
</aside>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="overlay" class="overlay hidden"></div>
|
|
||||||
|
|
||||||
<!-- Modals -->
|
|
||||||
<div id="gif-picker" class="modal-container hidden">
|
|
||||||
<div class="modal gif-modal">
|
|
||||||
<div class="modal-header">
|
|
||||||
<h2>Select GIF</h2>
|
|
||||||
<button type="button" class="close-btn" id="gif-picker-close">
|
|
||||||
<i data-lucide="x"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
<div class="gif-search-wrapper">
|
|
||||||
<input type="text" id="gif-search-input" placeholder="Search for GIFs..." autocomplete="off">
|
|
||||||
<i data-lucide="search" class="search-icon"></i>
|
|
||||||
</div>
|
|
||||||
<div id="gif-results" class="gif-results-grid">
|
|
||||||
<!-- GIF results will be injected here -->
|
|
||||||
<div class="gif-loading hidden">Loading...</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div id="modal-container" class="modal-container hidden">
|
|
||||||
<div class="modal">
|
|
||||||
<h2 id="modal-title">Create Server</h2>
|
|
||||||
<form id="guild-form" class="modal-form">
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="guild-name">SERVER NAME</label>
|
|
||||||
<input id="guild-name" placeholder="My Awesome Server" maxlength="64" required />
|
|
||||||
</div>
|
|
||||||
<div class="modal-footer">
|
|
||||||
<button type="button" class="cancel-btn" id="modal-cancel">Cancel</button>
|
|
||||||
<button type="submit" class="submit-btn">Create</button>
|
|
||||||
</div>
|
|
||||||
</form>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="channel-modal" class="modal-container hidden">
|
|
||||||
<div class="modal">
|
|
||||||
<h2>Create Channel</h2>
|
|
||||||
<form id="channel-form" class="modal-form">
|
|
||||||
<div class="form-item">
|
|
||||||
<label>CHANNEL TYPE</label>
|
|
||||||
<div class="radio-group">
|
|
||||||
<label class="radio-item">
|
|
||||||
<input type="radio" name="channel-kind" value="text" checked>
|
|
||||||
<div class="radio-box">
|
|
||||||
<i data-lucide="hash"></i>
|
|
||||||
<div class="radio-text">
|
|
||||||
<strong>Text</strong>
|
|
||||||
<span>Send messages, images, and GIFs.</span>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</label>
|
|
||||||
<label class="radio-item">
|
|
||||||
<input type="radio" name="channel-kind" value="voice">
|
|
||||||
<div class="radio-box">
|
|
||||||
<i data-lucide="volume-2"></i>
|
|
||||||
<div class="radio-text">
|
|
||||||
<strong>Voice</strong>
|
|
||||||
<span>Hang out together with voice and video.</span>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</label>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="channel-name">CHANNEL NAME</label>
|
|
||||||
<input id="channel-name" placeholder="new-channel" maxlength="64" required />
|
|
||||||
</div>
|
|
||||||
<div class="modal-footer">
|
|
||||||
<button type="button" class="cancel-btn" id="channel-modal-cancel">Cancel</button>
|
|
||||||
<button type="submit" class="submit-btn">Create Channel</button>
|
|
||||||
</div>
|
|
||||||
</form>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="sound-modal" class="modal-container hidden">
|
|
||||||
<div class="modal">
|
|
||||||
<h2>Add Sound</h2>
|
|
||||||
<form id="sound-form" class="modal-form">
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="sound-name">NAME</label>
|
|
||||||
<input id="sound-name" placeholder="Quack" maxlength="32" required />
|
|
||||||
</div>
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="sound-icon">ICON (Emoji or Initials)</label>
|
|
||||||
<input id="sound-icon" placeholder="🦆" maxlength="4" required />
|
|
||||||
</div>
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="sound-file">AUDIO FILE (MP3/WAV)</label>
|
|
||||||
<input id="sound-file" type="file" accept="audio/*" required />
|
|
||||||
</div>
|
|
||||||
<div class="modal-footer">
|
|
||||||
<button type="button" class="cancel-btn" id="sound-modal-cancel">Cancel</button>
|
|
||||||
<button type="submit" class="submit-btn" id="sound-submit-btn">Add Sound</button>
|
|
||||||
</div>
|
|
||||||
</form>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="upload-limit-modal" class="modal-container hidden">
|
|
||||||
<div class="modal">
|
|
||||||
<h2>Upload Too Large</h2>
|
|
||||||
<p id="upload-limit-message" class="modal-copy">Uploads are limited to 50 MB per file.</p>
|
|
||||||
<div class="modal-footer">
|
|
||||||
<button type="button" class="submit-btn" id="upload-limit-ok">OK</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<script type="module" src="app.js?v=20260227-shared-core-1"></script>
|
|
||||||
</body>
|
|
||||||
|
|
||||||
</html>
|
|
||||||
|
|
@ -1,20 +0,0 @@
|
||||||
const { contextBridge, ipcRenderer } = require('electron');
|
|
||||||
|
|
||||||
function onIpc(channel, callback) {
|
|
||||||
const listener = (_event, payload) => callback(payload);
|
|
||||||
ipcRenderer.on(channel, listener);
|
|
||||||
return () => ipcRenderer.removeListener(channel, listener);
|
|
||||||
}
|
|
||||||
|
|
||||||
contextBridge.exposeInMainWorld('electronAPI', {
|
|
||||||
copyToClipboard: (text) => ipcRenderer.invoke('clipboard-write', text),
|
|
||||||
getUpdateState: () => ipcRenderer.invoke('get-update-state'),
|
|
||||||
checkForUpdatesNow: () => ipcRenderer.invoke('check-for-updates-now'),
|
|
||||||
checkForUpdates: () => ipcRenderer.send('check-for-updates'),
|
|
||||||
downloadUpdate: () => ipcRenderer.send('download-update'),
|
|
||||||
quitAndInstall: () => ipcRenderer.send('quit-and-install'),
|
|
||||||
onUpdateState: (callback) => onIpc('update-state', callback),
|
|
||||||
onUpdateAvailable: (callback) => onIpc('update-available', callback),
|
|
||||||
onUpdateDownloaded: (callback) => onIpc('update-downloaded', callback),
|
|
||||||
onUpdateError: (callback) => onIpc('update-error', callback)
|
|
||||||
});
|
|
||||||
249
desktop/vendor/deepfilternet3-noise-filter.esm.js
vendored
249
desktop/vendor/deepfilternet3-noise-filter.esm.js
vendored
File diff suppressed because one or more lines are too long
Binary file not shown.
BIN
desktop/vendor/deepfilternet3/v2/pkg/df_bg.wasm
vendored
BIN
desktop/vendor/deepfilternet3/v2/pkg/df_bg.wasm
vendored
Binary file not shown.
313
main.js
313
main.js
|
|
@ -1,313 +0,0 @@
|
||||||
const { app, BrowserWindow, session, desktopCapturer, ipcMain, clipboard } = require('electron');
|
|
||||||
const { autoUpdater } = require('electron-updater');
|
|
||||||
const path = require('path');
|
|
||||||
const packageJson = require('./package.json');
|
|
||||||
const PERIODIC_UPDATE_CHECK_INTERVAL_MS = 1 * 60 * 1000;
|
|
||||||
|
|
||||||
const updateState = {
|
|
||||||
status: 'idle', // idle | checking | available | downloading | downloaded | installing | not-available | error
|
|
||||||
info: null,
|
|
||||||
error: null,
|
|
||||||
};
|
|
||||||
|
|
||||||
function broadcastUpdateState() {
|
|
||||||
const wins = BrowserWindow.getAllWindows();
|
|
||||||
for (const win of wins) {
|
|
||||||
if (!win.isDestroyed()) {
|
|
||||||
win.webContents.send('update-state', updateState);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function isNewerVersionAvailable(info) {
|
|
||||||
const next = info && typeof info.version === 'string' ? info.version : '';
|
|
||||||
return Boolean(next) && next !== app.getVersion();
|
|
||||||
}
|
|
||||||
|
|
||||||
let updateCheckInProgress = false;
|
|
||||||
let installInProgress = false;
|
|
||||||
let periodicUpdateTimer = null;
|
|
||||||
|
|
||||||
function resolveBackendUrl() {
|
|
||||||
const configuredUrl = (process.env.CHATTZ_URL || process.env.APP_BASE_URL || '').trim();
|
|
||||||
if (configuredUrl) {
|
|
||||||
return configuredUrl.replace(/\/$/, '');
|
|
||||||
}
|
|
||||||
|
|
||||||
const packagedFallback = typeof packageJson.homepage === 'string'
|
|
||||||
? packageJson.homepage.trim()
|
|
||||||
: '';
|
|
||||||
if (app.isPackaged && packagedFallback) {
|
|
||||||
return packagedFallback.replace(/\/$/, '');
|
|
||||||
}
|
|
||||||
|
|
||||||
return 'http://localhost:3000';
|
|
||||||
}
|
|
||||||
|
|
||||||
function webContentsOrigin(webContents) {
|
|
||||||
if (!webContents || typeof webContents.getURL !== 'function') {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
return safeOrigin(webContents.getURL());
|
|
||||||
}
|
|
||||||
|
|
||||||
async function runUpdateCheck(reason = 'manual') {
|
|
||||||
if (installInProgress) {
|
|
||||||
console.log(`Update check skipped (${reason}): install already in progress`);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (updateCheckInProgress) {
|
|
||||||
console.log(`Update check skipped (${reason}): another check is already in progress`);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
updateCheckInProgress = true;
|
|
||||||
updateState.status = 'checking';
|
|
||||||
updateState.error = null;
|
|
||||||
try {
|
|
||||||
const result = await autoUpdater.checkForUpdates();
|
|
||||||
const info = result && result.updateInfo ? result.updateInfo : null;
|
|
||||||
if (isNewerVersionAvailable(info)) {
|
|
||||||
updateState.status = 'available';
|
|
||||||
updateState.info = info;
|
|
||||||
updateState.error = null;
|
|
||||||
} else {
|
|
||||||
updateState.status = 'not-available';
|
|
||||||
updateState.info = info;
|
|
||||||
updateState.error = null;
|
|
||||||
}
|
|
||||||
} catch (err) {
|
|
||||||
updateState.status = 'error';
|
|
||||||
updateState.error = err && err.message ? err.message : String(err);
|
|
||||||
console.error(`Update check failed (${reason})`, err);
|
|
||||||
} finally {
|
|
||||||
updateCheckInProgress = false;
|
|
||||||
broadcastUpdateState();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function installDownloadedUpdate() {
|
|
||||||
if (installInProgress) {
|
|
||||||
console.log('Update install already in progress');
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
installInProgress = true;
|
|
||||||
updateState.status = 'installing';
|
|
||||||
updateState.error = null;
|
|
||||||
broadcastUpdateState();
|
|
||||||
|
|
||||||
const wins = BrowserWindow.getAllWindows();
|
|
||||||
for (const win of wins) {
|
|
||||||
if (win.isDestroyed()) continue;
|
|
||||||
try {
|
|
||||||
win.removeAllListeners('close');
|
|
||||||
win.destroy();
|
|
||||||
} catch (err) {
|
|
||||||
console.warn('Failed to destroy window before update install', err);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
setImmediate(() => {
|
|
||||||
try {
|
|
||||||
autoUpdater.quitAndInstall(false, true);
|
|
||||||
} catch (err) {
|
|
||||||
installInProgress = false;
|
|
||||||
updateState.status = 'error';
|
|
||||||
updateState.error = err && err.message ? err.message : String(err);
|
|
||||||
broadcastUpdateState();
|
|
||||||
console.error('quitAndInstall failed', err);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
setTimeout(() => {
|
|
||||||
if (installInProgress) {
|
|
||||||
console.warn('Update install is still waiting for app shutdown');
|
|
||||||
}
|
|
||||||
}, 15000);
|
|
||||||
}
|
|
||||||
|
|
||||||
function startPeriodicUpdateChecks() {
|
|
||||||
if (periodicUpdateTimer || !app.isPackaged) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
periodicUpdateTimer = setInterval(() => {
|
|
||||||
void runUpdateCheck('periodic');
|
|
||||||
}, PERIODIC_UPDATE_CHECK_INTERVAL_MS);
|
|
||||||
}
|
|
||||||
|
|
||||||
function createWindow() {
|
|
||||||
const sess = session.fromPartition('persist:chattz');
|
|
||||||
const backendUrl = resolveBackendUrl();
|
|
||||||
const backendOrigin = new URL(backendUrl).origin;
|
|
||||||
console.log(`Desktop backend URL: ${backendUrl}`);
|
|
||||||
|
|
||||||
const win = new BrowserWindow({
|
|
||||||
width: 1200,
|
|
||||||
height: 800,
|
|
||||||
title: "Chattz Desktop",
|
|
||||||
webPreferences: {
|
|
||||||
nodeIntegration: false,
|
|
||||||
contextIsolation: true,
|
|
||||||
session: sess,
|
|
||||||
preload: path.join(__dirname, 'desktop', 'preload.js')
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
win.setAutoHideMenuBar(true);
|
|
||||||
win.setMenuBarVisibility(true);
|
|
||||||
|
|
||||||
sess.setPermissionCheckHandler((webContents, permission) => {
|
|
||||||
const origin = webContentsOrigin(webContents);
|
|
||||||
if (origin !== backendOrigin) return false;
|
|
||||||
return permission === 'media' || permission === 'clipboard-write';
|
|
||||||
});
|
|
||||||
|
|
||||||
sess.setPermissionRequestHandler((webContents, permission, callback) => {
|
|
||||||
const origin = webContentsOrigin(webContents);
|
|
||||||
if (origin === backendOrigin && (permission === 'media' || permission === 'clipboard-write')) {
|
|
||||||
callback(true);
|
|
||||||
} else {
|
|
||||||
callback(false);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
sess.setDisplayMediaRequestHandler((request, callback) => {
|
|
||||||
const origin = safeOrigin(request.frame?.url || win.webContents.getURL());
|
|
||||||
if (origin !== backendOrigin) {
|
|
||||||
callback(null);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
desktopCapturer.getSources({ types: ['screen', 'window'] }).then((sources) => {
|
|
||||||
if (sources && sources.length > 0) {
|
|
||||||
const screenSource = sources.find(s => s.id.startsWith('screen')) || sources[0];
|
|
||||||
callback({ video: screenSource, audio: 'loopback' });
|
|
||||||
} else {
|
|
||||||
callback(null);
|
|
||||||
}
|
|
||||||
}).catch(err => {
|
|
||||||
console.error("Failed to get desktop sources for screen share", err);
|
|
||||||
callback(null);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
win.loadURL(backendUrl).catch((err) => {
|
|
||||||
console.error(`Failed to load desktop app: ${err}`);
|
|
||||||
});
|
|
||||||
|
|
||||||
// Ensure renderer receives latest updater state after any (re)load.
|
|
||||||
// Delay broadcast by 300ms to give the renderer time to register its
|
|
||||||
// onUpdateState IPC listener before we push state.
|
|
||||||
win.webContents.on('did-finish-load', () => {
|
|
||||||
setTimeout(() => {
|
|
||||||
if (!win.isDestroyed()) broadcastUpdateState();
|
|
||||||
}, 300);
|
|
||||||
});
|
|
||||||
|
|
||||||
// Uncomment to debug
|
|
||||||
// win.webContents.openDevTools();
|
|
||||||
}
|
|
||||||
|
|
||||||
app.commandLine.appendSwitch('disable-webrtc-hw-encoding'); // Sometime helps resolve codec mismatch behavior
|
|
||||||
app.commandLine.appendSwitch('log-level', '3'); // Silences standard Chromium warning logs (like SRTP transport unprotect logs which are benign timing issues)
|
|
||||||
|
|
||||||
app.whenReady().then(() => {
|
|
||||||
ipcMain.handle('clipboard-write', (event, text) => {
|
|
||||||
clipboard.writeText(text);
|
|
||||||
return true;
|
|
||||||
});
|
|
||||||
|
|
||||||
ipcMain.handle('get-update-state', () => ({ ...updateState }));
|
|
||||||
ipcMain.handle('check-for-updates-now', async () => {
|
|
||||||
await runUpdateCheck('renderer-direct');
|
|
||||||
return { ...updateState };
|
|
||||||
});
|
|
||||||
|
|
||||||
createWindow();
|
|
||||||
|
|
||||||
// Configure Auto-Updater
|
|
||||||
autoUpdater.autoDownload = false;
|
|
||||||
autoUpdater.logger = console;
|
|
||||||
|
|
||||||
autoUpdater.on('update-available', (info) => {
|
|
||||||
updateState.status = 'available';
|
|
||||||
updateState.info = info;
|
|
||||||
updateState.error = null;
|
|
||||||
broadcastUpdateState();
|
|
||||||
const wins = BrowserWindow.getAllWindows();
|
|
||||||
if (wins.length > 0) wins[0].webContents.send('update-available', info);
|
|
||||||
});
|
|
||||||
|
|
||||||
autoUpdater.on('update-not-available', (info) => {
|
|
||||||
updateState.status = 'not-available';
|
|
||||||
updateState.info = info || null;
|
|
||||||
updateState.error = null;
|
|
||||||
broadcastUpdateState();
|
|
||||||
});
|
|
||||||
|
|
||||||
autoUpdater.on('update-downloaded', (info) => {
|
|
||||||
updateState.status = 'downloaded';
|
|
||||||
updateState.info = info;
|
|
||||||
updateState.error = null;
|
|
||||||
broadcastUpdateState();
|
|
||||||
const wins = BrowserWindow.getAllWindows();
|
|
||||||
if (wins.length > 0) wins[0].webContents.send('update-downloaded', info);
|
|
||||||
});
|
|
||||||
|
|
||||||
autoUpdater.on('error', (err) => {
|
|
||||||
updateState.status = 'error';
|
|
||||||
updateState.error = err.message;
|
|
||||||
broadcastUpdateState();
|
|
||||||
const wins = BrowserWindow.getAllWindows();
|
|
||||||
if (wins.length > 0) wins[0].webContents.send('update-error', err.message);
|
|
||||||
});
|
|
||||||
|
|
||||||
ipcMain.on('check-for-updates', () => {
|
|
||||||
void runUpdateCheck('manual');
|
|
||||||
});
|
|
||||||
|
|
||||||
ipcMain.on('download-update', () => {
|
|
||||||
if (installInProgress) return;
|
|
||||||
updateState.status = 'downloading';
|
|
||||||
updateState.error = null;
|
|
||||||
broadcastUpdateState();
|
|
||||||
autoUpdater.downloadUpdate();
|
|
||||||
});
|
|
||||||
|
|
||||||
ipcMain.on('quit-and-install', () => {
|
|
||||||
installDownloadedUpdate();
|
|
||||||
});
|
|
||||||
|
|
||||||
// Check once on startup
|
|
||||||
setTimeout(() => {
|
|
||||||
void runUpdateCheck('startup');
|
|
||||||
}, 5000);
|
|
||||||
startPeriodicUpdateChecks();
|
|
||||||
|
|
||||||
app.on('activate', () => {
|
|
||||||
if (BrowserWindow.getAllWindows().length === 0) {
|
|
||||||
createWindow();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
app.on('window-all-closed', () => {
|
|
||||||
if (process.platform !== 'darwin') {
|
|
||||||
app.quit();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
app.on('before-quit', () => {
|
|
||||||
if (periodicUpdateTimer) {
|
|
||||||
clearInterval(periodicUpdateTimer);
|
|
||||||
periodicUpdateTimer = null;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
function safeOrigin(value) {
|
|
||||||
try {
|
|
||||||
return new URL(value).origin;
|
|
||||||
} catch (_) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
5307
package-lock.json
generated
5307
package-lock.json
generated
File diff suppressed because it is too large
Load diff
61
package.json
61
package.json
|
|
@ -1,61 +0,0 @@
|
||||||
{
|
|
||||||
"name": "chattz-electron",
|
|
||||||
"version": "0.0.50",
|
|
||||||
"description": "Electron frontend for Chattz",
|
|
||||||
"author": "Pavel Flegr <pavelflegr@gmail.com>",
|
|
||||||
"homepage": "https://discord.flegr.me",
|
|
||||||
"repository": {
|
|
||||||
"type": "git",
|
|
||||||
"url": "https://git.flegr.me/pavel/discord.git"
|
|
||||||
},
|
|
||||||
"main": "main.js",
|
|
||||||
"scripts": {
|
|
||||||
"build:html": "node scripts/generate-html.js",
|
|
||||||
"start": "npm run build:html && electron .",
|
|
||||||
"dev": "npm run build:html && electron .",
|
|
||||||
"dist": "npm run build:html && electron-builder --publish never",
|
|
||||||
"dist:linux": "npm run build:html && electron-builder --linux --publish never",
|
|
||||||
"dist:win": "npm run build:html && electron-builder --win --publish never"
|
|
||||||
},
|
|
||||||
"build": {
|
|
||||||
"appId": "me.flegr.chattz",
|
|
||||||
"productName": "Chattz",
|
|
||||||
"linux": {
|
|
||||||
"target": [
|
|
||||||
"AppImage",
|
|
||||||
"rpm"
|
|
||||||
],
|
|
||||||
"category": "Chat",
|
|
||||||
"icon": "build/icon.png"
|
|
||||||
},
|
|
||||||
"win": {
|
|
||||||
"target": [
|
|
||||||
"nsis"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
"publish": [
|
|
||||||
{
|
|
||||||
"provider": "generic",
|
|
||||||
"url": "https://discord.flegr.me/static/installers",
|
|
||||||
"channel": "latest"
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"files": [
|
|
||||||
"main.js",
|
|
||||||
"package.json",
|
|
||||||
"desktop/",
|
|
||||||
"static/",
|
|
||||||
"!dist"
|
|
||||||
],
|
|
||||||
"directories": {
|
|
||||||
"output": "dist"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"devDependencies": {
|
|
||||||
"electron": "^34.5.8",
|
|
||||||
"electron-builder": "^25.1.8"
|
|
||||||
},
|
|
||||||
"dependencies": {
|
|
||||||
"electron-updater": "^6.8.3"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,57 +0,0 @@
|
||||||
const fs = require('fs');
|
|
||||||
const path = require('path');
|
|
||||||
|
|
||||||
const root = path.resolve(__dirname, '..');
|
|
||||||
const templatePath = path.join(root, 'shared-html', 'index.template.html');
|
|
||||||
const template = fs.readFileSync(templatePath, 'utf8');
|
|
||||||
|
|
||||||
const updaterMarkup = `<div id="update-notifier" class="update-notifier hidden">
|
|
||||||
<button id="update-download-btn" title="Download Update"><i data-lucide="download"></i></button>
|
|
||||||
<button id="update-install-btn" title="Install Update" class="hidden"><i data-lucide="arrow-up-circle"></i></button>
|
|
||||||
</div>`;
|
|
||||||
|
|
||||||
const variants = {
|
|
||||||
web: {
|
|
||||||
styles_href: '/static/styles.css',
|
|
||||||
lucide_src: '/static/vendor/lucide.min.js',
|
|
||||||
app_src: '/static/app.js?v=20260227-shared-core-1',
|
|
||||||
web_downloads: `<div class="auth-downloads">
|
|
||||||
<span>Desktop downloads:</span>
|
|
||||||
<a href="/static/installers/chattz-windows.exe">Windows</a>
|
|
||||||
<a href="/static/installers/chattz-linux.rpm">Linux (RPM)</a>
|
|
||||||
</div>`,
|
|
||||||
desktop_updater: updaterMarkup,
|
|
||||||
outPath: path.join(root, 'static', 'index.html'),
|
|
||||||
},
|
|
||||||
desktop: {
|
|
||||||
styles_href: '../static/styles.css',
|
|
||||||
lucide_src: '../static/vendor/lucide.min.js',
|
|
||||||
app_src: 'app.js?v=20260227-shared-core-1',
|
|
||||||
web_downloads: '',
|
|
||||||
desktop_updater: updaterMarkup,
|
|
||||||
outPath: path.join(root, 'desktop', 'index.html'),
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
for (const variant of Object.values(variants)) {
|
|
||||||
let output = template;
|
|
||||||
for (const [key, value] of Object.entries(variant)) {
|
|
||||||
if (key === 'outPath') continue;
|
|
||||||
const token = `{{${key}}}`;
|
|
||||||
if (value === '') {
|
|
||||||
output = output.replace(new RegExp(`^[ \\t]*\\{\\{${key}\\}\\}[ \\t]*\\n?`, 'm'), '');
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
output = output.replaceAll(token, value);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (/\{\{[a-z_]+\}\}/i.test(output)) {
|
|
||||||
throw new Error(`Unresolved template placeholders remain in ${variant.outPath}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
fs.writeFileSync(
|
|
||||||
variant.outPath,
|
|
||||||
`<!-- Generated from shared-html/index.template.html. Do not edit directly. -->\n${output}`,
|
|
||||||
'utf8'
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
@ -1,279 +0,0 @@
|
||||||
<!doctype html>
|
|
||||||
<html lang="en">
|
|
||||||
|
|
||||||
<head>
|
|
||||||
<meta charset="UTF-8" />
|
|
||||||
<meta name="viewport" content="width=device-width, initial-scale=1.0, viewport-fit=cover" />
|
|
||||||
<title>Chattz</title>
|
|
||||||
<link rel="stylesheet" href="{{styles_href}}" />
|
|
||||||
<!-- Lucide Icons -->
|
|
||||||
<script src="{{lucide_src}}"></script>
|
|
||||||
</head>
|
|
||||||
|
|
||||||
<body>
|
|
||||||
<div class="auth-screen" id="auth-screen">
|
|
||||||
<div class="auth-card">
|
|
||||||
<div class="brand-large">C</div>
|
|
||||||
<h1>Chattz</h1>
|
|
||||||
<p>Sign in to open your servers.</p>
|
|
||||||
<button id="login-btn">Login with Authentik</button>
|
|
||||||
{{web_downloads}}
|
|
||||||
<p id="status" class="status-line"></p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="shell hidden" id="main">
|
|
||||||
<aside class="server-rail">
|
|
||||||
<div class="brand" title="Home">
|
|
||||||
<i data-lucide="message-square"></i>
|
|
||||||
</div>
|
|
||||||
<div class="separator"></div>
|
|
||||||
<div id="guild-list" class="guild-list"></div>
|
|
||||||
<button id="add-guild-btn" class="guild-pill action-pill" title="Add a Server">
|
|
||||||
<i data-lucide="plus"></i>
|
|
||||||
</button>
|
|
||||||
</aside>
|
|
||||||
|
|
||||||
<aside class="channel-sidebar">
|
|
||||||
<header class="sidebar-header clickable" id="guild-header">
|
|
||||||
<h2 id="guild-title">No server selected</h2>
|
|
||||||
<div class="sidebar-header-actions">
|
|
||||||
<span id="invite-copied-badge" class="copied-badge hidden">Copied!</span>
|
|
||||||
<button id="create-invite-btn" class="header-action-btn" title="Create Invite Link">
|
|
||||||
<i data-lucide="link-2"></i>
|
|
||||||
</button>
|
|
||||||
<i data-lucide="chevron-down"></i>
|
|
||||||
</div>
|
|
||||||
</header>
|
|
||||||
|
|
||||||
<div class="sidebar-scroll">
|
|
||||||
<section class="sidebar-group">
|
|
||||||
<div class="group-title">
|
|
||||||
<i data-lucide="chevron-down" class="group-toggle"></i>
|
|
||||||
<span>Text Channels</span>
|
|
||||||
<button id="add-text-btn" class="add-btn" type="button" title="Create Text Channel">
|
|
||||||
<i data-lucide="plus"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
<div id="channel-list" class="channel-list"></div>
|
|
||||||
</section>
|
|
||||||
|
|
||||||
<section class="sidebar-group">
|
|
||||||
<div class="group-title">
|
|
||||||
<i data-lucide="chevron-down" class="group-toggle"></i>
|
|
||||||
<span>Voice Channels</span>
|
|
||||||
<button id="add-voice-btn" class="add-btn" type="button" title="Create Voice Channel">
|
|
||||||
<i data-lucide="plus"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
<div id="voice-channel-list" class="channel-list"></div>
|
|
||||||
</section>
|
|
||||||
|
|
||||||
<section class="sidebar-group">
|
|
||||||
<div class="group-title">
|
|
||||||
<i data-lucide="chevron-down" class="group-toggle"></i>
|
|
||||||
<span>Direct Messages</span>
|
|
||||||
</div>
|
|
||||||
<div id="dm-list" class="channel-list"></div>
|
|
||||||
</section>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="sidebar-footer">
|
|
||||||
<div id="voice-connection" class="voice-connection hidden">
|
|
||||||
<div class="vc-info">
|
|
||||||
<i data-lucide="signal-high" class="vc-icon"></i>
|
|
||||||
<div class="vc-text">
|
|
||||||
<span class="vc-status">Voice Connected</span>
|
|
||||||
<span id="vc-channel-name" class="vc-name">General</span>
|
|
||||||
</div>
|
|
||||||
<div class="vc-actions">
|
|
||||||
<button id="voice-video-btn" title="Turn on Camera"><i data-lucide="video-off"></i></button>
|
|
||||||
<button id="voice-screen-btn" title="Share Screen"><i data-lucide="monitor-off"></i></button>
|
|
||||||
<button id="voice-mute-btn" title="Mute"><i data-lucide="mic"></i></button>
|
|
||||||
<button id="voice-leave-btn" title="Disconnect"><i data-lucide="phone-off"></i></button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div id="soundboard" class="soundboard-container hidden">
|
|
||||||
<div class="soundboard-header">
|
|
||||||
<span>Sound Board</span>
|
|
||||||
<span id="add-sound-btn" class="btn-add-sound">+ Add Sound</span>
|
|
||||||
</div>
|
|
||||||
<div id="soundboard-grid" class="soundboard-grid"></div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="user-panel">
|
|
||||||
<div class="avatar-wrapper">
|
|
||||||
<div id="user-avatar" class="avatar">U</div>
|
|
||||||
<div class="status-dot online"></div>
|
|
||||||
</div>
|
|
||||||
<div class="user-info">
|
|
||||||
<div id="user-name" class="display-name">Username</div>
|
|
||||||
<div class="user-status">Online</div>
|
|
||||||
</div>
|
|
||||||
<div class="user-actions">
|
|
||||||
{{desktop_updater}}
|
|
||||||
<button id="user-settings-btn" title="Settings"><i data-lucide="settings"></i></button>
|
|
||||||
<button id="logout-btn" title="Logout"><i data-lucide="log-out"></i></button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</aside>
|
|
||||||
|
|
||||||
<main class="chat-pane">
|
|
||||||
<header class="chat-header">
|
|
||||||
<button id="mobile-menu-btn" class="mobile-toggle-btn" title="Menu">
|
|
||||||
<i data-lucide="menu"></i>
|
|
||||||
</button>
|
|
||||||
<i data-lucide="hash" class="header-icon"></i>
|
|
||||||
<h3 id="channel-title">Select a channel</h3>
|
|
||||||
<div style="flex:1"></div>
|
|
||||||
<button id="mobile-members-btn" class="mobile-toggle-btn" title="Members">
|
|
||||||
<i data-lucide="users"></i>
|
|
||||||
</button>
|
|
||||||
</header>
|
|
||||||
|
|
||||||
<div id="video-grid" class="video-grid hidden"></div>
|
|
||||||
<div id="message-list" class="message-list"></div>
|
|
||||||
|
|
||||||
<div class="chat-input-wrapper">
|
|
||||||
<form id="message-form" class="message-form">
|
|
||||||
<input id="media-file-input" type="file" class="hidden-file-input" accept="image/*,video/*,audio/*,.pdf,.txt,.zip,.rar,.7z,.tar,.gz,.json,.csv,.md" />
|
|
||||||
<button type="button" id="media-upload-btn" class="input-action-btn" title="Upload File">
|
|
||||||
<i data-lucide="paperclip"></i>
|
|
||||||
</button>
|
|
||||||
<button type="button" id="gif-btn" class="input-action-btn" title="Open GIF Picker">
|
|
||||||
<i data-lucide="image"></i>
|
|
||||||
</button>
|
|
||||||
<input id="message-body" autocomplete="off" placeholder="Message #channel" maxlength="4000" required />
|
|
||||||
<button type="submit" class="hidden"></button>
|
|
||||||
</form>
|
|
||||||
</div>
|
|
||||||
</main>
|
|
||||||
|
|
||||||
<aside class="utility-sidebar" id="utility-sidebar">
|
|
||||||
<header class="sidebar-header">
|
|
||||||
<h2>Members</h2>
|
|
||||||
</header>
|
|
||||||
<div class="member-list-wrapper">
|
|
||||||
<div id="member-list" class="member-list"></div>
|
|
||||||
</div>
|
|
||||||
</aside>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="overlay" class="overlay hidden"></div>
|
|
||||||
|
|
||||||
<!-- Modals -->
|
|
||||||
<div id="gif-picker" class="modal-container hidden">
|
|
||||||
<div class="modal gif-modal">
|
|
||||||
<div class="modal-header">
|
|
||||||
<h2>Select GIF</h2>
|
|
||||||
<button type="button" class="close-btn" id="gif-picker-close">
|
|
||||||
<i data-lucide="x"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
<div class="gif-search-wrapper">
|
|
||||||
<input type="text" id="gif-search-input" placeholder="Search for GIFs..." autocomplete="off">
|
|
||||||
<i data-lucide="search" class="search-icon"></i>
|
|
||||||
</div>
|
|
||||||
<div id="gif-results" class="gif-results-grid">
|
|
||||||
<!-- GIF results will be injected here -->
|
|
||||||
<div class="gif-loading hidden">Loading...</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div id="modal-container" class="modal-container hidden">
|
|
||||||
<div class="modal">
|
|
||||||
<h2 id="modal-title">Create Server</h2>
|
|
||||||
<form id="guild-form" class="modal-form">
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="guild-name">SERVER NAME</label>
|
|
||||||
<input id="guild-name" placeholder="My Awesome Server" maxlength="64" required />
|
|
||||||
</div>
|
|
||||||
<div class="modal-footer">
|
|
||||||
<button type="button" class="cancel-btn" id="modal-cancel">Cancel</button>
|
|
||||||
<button type="submit" class="submit-btn">Create</button>
|
|
||||||
</div>
|
|
||||||
</form>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="channel-modal" class="modal-container hidden">
|
|
||||||
<div class="modal">
|
|
||||||
<h2>Create Channel</h2>
|
|
||||||
<form id="channel-form" class="modal-form">
|
|
||||||
<div class="form-item">
|
|
||||||
<label>CHANNEL TYPE</label>
|
|
||||||
<div class="radio-group">
|
|
||||||
<label class="radio-item">
|
|
||||||
<input type="radio" name="channel-kind" value="text" checked>
|
|
||||||
<div class="radio-box">
|
|
||||||
<i data-lucide="hash"></i>
|
|
||||||
<div class="radio-text">
|
|
||||||
<strong>Text</strong>
|
|
||||||
<span>Send messages, images, and GIFs.</span>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</label>
|
|
||||||
<label class="radio-item">
|
|
||||||
<input type="radio" name="channel-kind" value="voice">
|
|
||||||
<div class="radio-box">
|
|
||||||
<i data-lucide="volume-2"></i>
|
|
||||||
<div class="radio-text">
|
|
||||||
<strong>Voice</strong>
|
|
||||||
<span>Hang out together with voice and video.</span>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</label>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="channel-name">CHANNEL NAME</label>
|
|
||||||
<input id="channel-name" placeholder="new-channel" maxlength="64" required />
|
|
||||||
</div>
|
|
||||||
<div class="modal-footer">
|
|
||||||
<button type="button" class="cancel-btn" id="channel-modal-cancel">Cancel</button>
|
|
||||||
<button type="submit" class="submit-btn">Create Channel</button>
|
|
||||||
</div>
|
|
||||||
</form>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="sound-modal" class="modal-container hidden">
|
|
||||||
<div class="modal">
|
|
||||||
<h2>Add Sound</h2>
|
|
||||||
<form id="sound-form" class="modal-form">
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="sound-name">NAME</label>
|
|
||||||
<input id="sound-name" placeholder="Quack" maxlength="32" required />
|
|
||||||
</div>
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="sound-icon">ICON (Emoji or Initials)</label>
|
|
||||||
<input id="sound-icon" placeholder="🦆" maxlength="4" required />
|
|
||||||
</div>
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="sound-file">AUDIO FILE (MP3/WAV)</label>
|
|
||||||
<input id="sound-file" type="file" accept="audio/*" required />
|
|
||||||
</div>
|
|
||||||
<div class="modal-footer">
|
|
||||||
<button type="button" class="cancel-btn" id="sound-modal-cancel">Cancel</button>
|
|
||||||
<button type="submit" class="submit-btn" id="sound-submit-btn">Add Sound</button>
|
|
||||||
</div>
|
|
||||||
</form>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="upload-limit-modal" class="modal-container hidden">
|
|
||||||
<div class="modal">
|
|
||||||
<h2>Upload Too Large</h2>
|
|
||||||
<p id="upload-limit-message" class="modal-copy">Uploads are limited to 50 MB per file.</p>
|
|
||||||
<div class="modal-footer">
|
|
||||||
<button type="button" class="submit-btn" id="upload-limit-ok">OK</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<script type="module" src="{{app_src}}"></script>
|
|
||||||
</body>
|
|
||||||
|
|
||||||
</html>
|
|
||||||
258
src/auth.rs
258
src/auth.rs
|
|
@ -1,25 +1,32 @@
|
||||||
use anyhow::{Result, anyhow};
|
use std::time::{Duration, SystemTime, UNIX_EPOCH};
|
||||||
|
|
||||||
|
use anyhow::{Context, Result, anyhow};
|
||||||
use axum::{
|
use axum::{
|
||||||
Json,
|
Json,
|
||||||
extract::{FromRef, FromRequestParts},
|
extract::{FromRef, FromRequestParts},
|
||||||
http::{HeaderMap, StatusCode, request::Parts},
|
http::{StatusCode, request::Parts},
|
||||||
response::{IntoResponse, Response},
|
response::{IntoResponse, Response},
|
||||||
};
|
};
|
||||||
use chrono::{Duration, Utc};
|
use jsonwebtoken::{DecodingKey, EncodingKey, Header, Validation, decode, encode};
|
||||||
use serde::Serialize;
|
use serde::{Deserialize, Serialize};
|
||||||
use sha2::{Digest, Sha256};
|
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
use crate::{AppState, db};
|
use crate::{AppState, db};
|
||||||
|
|
||||||
pub const OAUTH_STATE_COOKIE: &str = "chattz_oauth_state";
|
const SESSION_COOKIE: &str = "chattz_session";
|
||||||
pub const SESSION_COOKIE: &str = "chattz_session";
|
const OAUTH_STATE_COOKIE: &str = "chattz_oauth_state";
|
||||||
const SESSION_TTL_DAYS: i64 = 30;
|
const SESSION_TTL_SECS: u64 = 60 * 60 * 24 * 7;
|
||||||
|
|
||||||
|
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||||||
|
struct SessionClaims {
|
||||||
|
sub: String,
|
||||||
|
exp: usize,
|
||||||
|
iat: usize,
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct AuthUser {
|
pub struct AuthUser {
|
||||||
pub id: Uuid,
|
pub id: Uuid,
|
||||||
pub session_id: String,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
|
|
@ -36,13 +43,6 @@ impl ApiError {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn forbidden(msg: &str) -> Self {
|
|
||||||
Self {
|
|
||||||
status: StatusCode::FORBIDDEN,
|
|
||||||
message: msg.to_string(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn bad_request(msg: &str) -> Self {
|
pub fn bad_request(msg: &str) -> Self {
|
||||||
Self {
|
Self {
|
||||||
status: StatusCode::BAD_REQUEST,
|
status: StatusCode::BAD_REQUEST,
|
||||||
|
|
@ -56,13 +56,6 @@ impl ApiError {
|
||||||
message: msg.to_string(),
|
message: msg.to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn service_unavailable(msg: &str) -> Self {
|
|
||||||
Self {
|
|
||||||
status: StatusCode::SERVICE_UNAVAILABLE,
|
|
||||||
message: msg.to_string(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize)]
|
#[derive(Serialize)]
|
||||||
|
|
@ -72,13 +65,7 @@ struct ErrorBody<'a> {
|
||||||
|
|
||||||
impl IntoResponse for ApiError {
|
impl IntoResponse for ApiError {
|
||||||
fn into_response(self) -> Response {
|
fn into_response(self) -> Response {
|
||||||
(
|
(self.status, Json(ErrorBody { error: &self.message })).into_response()
|
||||||
self.status,
|
|
||||||
Json(ErrorBody {
|
|
||||||
error: &self.message,
|
|
||||||
}),
|
|
||||||
)
|
|
||||||
.into_response()
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -90,25 +77,25 @@ impl From<anyhow::Error> for ApiError {
|
||||||
|
|
||||||
impl<S> FromRequestParts<S> for AuthUser
|
impl<S> FromRequestParts<S> for AuthUser
|
||||||
where
|
where
|
||||||
AppState: FromRef<S>,
|
AppState: axum::extract::FromRef<S>,
|
||||||
S: Send + Sync,
|
S: Send + Sync,
|
||||||
{
|
{
|
||||||
type Rejection = ApiError;
|
type Rejection = ApiError;
|
||||||
|
|
||||||
async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self, Self::Rejection> {
|
async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self, Self::Rejection> {
|
||||||
let app = AppState::from_ref(state);
|
let app = AppState::from_ref(state);
|
||||||
let session_id = read_cookie_from_headers(&parts.headers, SESSION_COOKIE)
|
let token = read_cookie(parts, SESSION_COOKIE).ok_or_else(|| ApiError::unauthorized("missing session"))?;
|
||||||
.ok_or_else(|| ApiError::unauthorized("missing session cookie"))?;
|
let user_id = verify_session(&token, &app.settings.session_secret)
|
||||||
|
.map_err(|_| ApiError::unauthorized("invalid session"))?;
|
||||||
|
|
||||||
let user_id = db::touch_active_session(&app.db, &session_id)
|
let exists = db::user_exists(&app.db, user_id)
|
||||||
.await
|
.await
|
||||||
.map_err(|_| ApiError::unauthorized("invalid or expired session"))?
|
.map_err(|_| ApiError::unauthorized("session user not found"))?;
|
||||||
.ok_or_else(|| ApiError::unauthorized("invalid or expired session"))?;
|
if !exists {
|
||||||
|
return Err(ApiError::unauthorized("session user not found"));
|
||||||
|
}
|
||||||
|
|
||||||
Ok(Self {
|
Ok(Self { id: user_id })
|
||||||
id: user_id,
|
|
||||||
session_id,
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -116,14 +103,6 @@ pub fn new_oauth_state() -> String {
|
||||||
Uuid::new_v4().to_string()
|
Uuid::new_v4().to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn new_session_id() -> String {
|
|
||||||
Uuid::new_v4().simple().to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn session_expiry() -> chrono::DateTime<chrono::FixedOffset> {
|
|
||||||
(Utc::now() + Duration::days(SESSION_TTL_DAYS)).fixed_offset()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn make_oauth_state_cookie(value: &str, secure: bool) -> String {
|
pub fn make_oauth_state_cookie(value: &str, secure: bool) -> String {
|
||||||
format!(
|
format!(
|
||||||
"{name}={value}; Path=/; HttpOnly; SameSite=Lax; Max-Age=600{secure_flag}",
|
"{name}={value}; Path=/; HttpOnly; SameSite=Lax; Max-Age=600{secure_flag}",
|
||||||
|
|
@ -133,32 +112,69 @@ pub fn make_oauth_state_cookie(value: &str, secure: bool) -> String {
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn clear_oauth_state_cookie(secure: bool) -> String {
|
pub fn clear_oauth_state_cookie(secure: bool) -> String {
|
||||||
clear_cookie(OAUTH_STATE_COOKIE, secure, "Lax")
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn make_session_cookie(value: &str, secure: bool) -> String {
|
|
||||||
format!(
|
format!(
|
||||||
"{name}={value}; Path=/; HttpOnly; SameSite=Strict; Max-Age={ttl}{secure_flag}",
|
"{name}=; Path=/; HttpOnly; SameSite=Lax; Max-Age=0{secure_flag}",
|
||||||
name = SESSION_COOKIE,
|
name = OAUTH_STATE_COOKIE,
|
||||||
ttl = Duration::days(SESSION_TTL_DAYS).num_seconds(),
|
|
||||||
secure_flag = if secure { "; Secure" } else { "" }
|
secure_flag = if secure { "; Secure" } else { "" }
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn clear_session_cookie(secure: bool) -> String {
|
pub fn read_oauth_state_from_headers(headers: &axum::http::HeaderMap) -> Option<String> {
|
||||||
clear_cookie(SESSION_COOKIE, secure, "Strict")
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn read_cookie_from_headers(headers: &HeaderMap, cookie_name: &str) -> Option<String> {
|
|
||||||
let raw = headers.get(axum::http::header::COOKIE)?.to_str().ok()?;
|
let raw = headers.get(axum::http::header::COOKIE)?.to_str().ok()?;
|
||||||
raw.split(';').find_map(|pair| {
|
raw.split(';').find_map(|pair| {
|
||||||
let mut kv = pair.trim().splitn(2, '=');
|
let mut kv = pair.trim().splitn(2, '=');
|
||||||
let key = kv.next()?;
|
let key = kv.next()?;
|
||||||
let value = kv.next()?;
|
let value = kv.next()?;
|
||||||
(key == cookie_name).then(|| value.to_string())
|
(key == OAUTH_STATE_COOKIE).then(|| value.to_string())
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn new_session_cookie(user_id: Uuid, secret: &str, secure: bool) -> Result<String> {
|
||||||
|
let now = now_ts();
|
||||||
|
let claims = SessionClaims {
|
||||||
|
sub: user_id.to_string(),
|
||||||
|
iat: now as usize,
|
||||||
|
exp: (now + SESSION_TTL_SECS) as usize,
|
||||||
|
};
|
||||||
|
|
||||||
|
let token = encode(
|
||||||
|
&Header::default(),
|
||||||
|
&claims,
|
||||||
|
&EncodingKey::from_secret(secret.as_bytes()),
|
||||||
|
)
|
||||||
|
.context("failed to encode session token")?;
|
||||||
|
|
||||||
|
Ok(format!(
|
||||||
|
"{name}={token}; Path=/; HttpOnly; SameSite=Lax; Max-Age={ttl}{secure_flag}",
|
||||||
|
name = SESSION_COOKIE,
|
||||||
|
ttl = SESSION_TTL_SECS,
|
||||||
|
secure_flag = if secure { "; Secure" } else { "" }
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn clear_session_cookie(secure: bool) -> String {
|
||||||
|
format!(
|
||||||
|
"{name}=; Path=/; HttpOnly; SameSite=Lax; Max-Age=0{secure_flag}",
|
||||||
|
name = SESSION_COOKIE,
|
||||||
|
secure_flag = if secure { "; Secure" } else { "" }
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn verify_session(token: &str, secret: &str) -> Result<Uuid> {
|
||||||
|
let mut validation = Validation::default();
|
||||||
|
validation.validate_exp = true;
|
||||||
|
|
||||||
|
let data = decode::<SessionClaims>(
|
||||||
|
token,
|
||||||
|
&DecodingKey::from_secret(secret.as_bytes()),
|
||||||
|
&validation,
|
||||||
|
)
|
||||||
|
.context("failed to decode session token")?;
|
||||||
|
|
||||||
|
let user_id = Uuid::parse_str(&data.claims.sub).context("invalid sub in session token")?;
|
||||||
|
Ok(user_id)
|
||||||
|
}
|
||||||
|
|
||||||
pub fn validate_oauth_state(expected_cookie: Option<String>, query_state: &str) -> Result<()> {
|
pub fn validate_oauth_state(expected_cookie: Option<String>, query_state: &str) -> Result<()> {
|
||||||
let expected = expected_cookie.ok_or_else(|| anyhow!("missing oauth state cookie"))?;
|
let expected = expected_cookie.ok_or_else(|| anyhow!("missing oauth state cookie"))?;
|
||||||
if expected != query_state {
|
if expected != query_state {
|
||||||
|
|
@ -167,113 +183,19 @@ pub fn validate_oauth_state(expected_cookie: Option<String>, query_state: &str)
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn user_agent_hash(headers: &HeaderMap) -> Option<String> {
|
fn read_cookie(parts: &Parts, name: &str) -> Option<String> {
|
||||||
header_hash(headers, axum::http::header::USER_AGENT.as_str())
|
let raw = parts.headers.get(axum::http::header::COOKIE)?.to_str().ok()?;
|
||||||
|
raw.split(';').find_map(|pair| {
|
||||||
|
let mut kv = pair.trim().splitn(2, '=');
|
||||||
|
let key = kv.next()?;
|
||||||
|
let value = kv.next()?;
|
||||||
|
(key == name).then(|| value.to_string())
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn ip_hash(headers: &HeaderMap) -> Option<String> {
|
fn now_ts() -> u64 {
|
||||||
headers
|
SystemTime::now()
|
||||||
.get("x-forwarded-for")
|
.duration_since(UNIX_EPOCH)
|
||||||
.and_then(|v| v.to_str().ok())
|
.unwrap_or_else(|_| Duration::from_secs(0))
|
||||||
.and_then(|raw| raw.split(',').next().map(str::trim))
|
.as_secs()
|
||||||
.filter(|value| !value.is_empty())
|
|
||||||
.map(hash_string)
|
|
||||||
.or_else(|| header_hash(headers, "x-real-ip"))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn origin_matches(headers: &HeaderMap, expected_origin: &str) -> bool {
|
|
||||||
headers
|
|
||||||
.get(axum::http::header::ORIGIN)
|
|
||||||
.and_then(|value| value.to_str().ok())
|
|
||||||
.map(|origin| origin == expected_origin)
|
|
||||||
.unwrap_or(false)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn header_hash(headers: &HeaderMap, header_name: &str) -> Option<String> {
|
|
||||||
headers
|
|
||||||
.get(header_name)
|
|
||||||
.and_then(|v| v.to_str().ok())
|
|
||||||
.filter(|value| !value.trim().is_empty())
|
|
||||||
.map(hash_string)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn hash_string(value: &str) -> String {
|
|
||||||
let mut hasher = Sha256::new();
|
|
||||||
hasher.update(value.as_bytes());
|
|
||||||
let digest = hasher.finalize();
|
|
||||||
let mut out = String::with_capacity(digest.len() * 2);
|
|
||||||
for byte in digest {
|
|
||||||
out.push(nibble_to_hex(byte >> 4));
|
|
||||||
out.push(nibble_to_hex(byte & 0x0f));
|
|
||||||
}
|
|
||||||
out
|
|
||||||
}
|
|
||||||
|
|
||||||
fn nibble_to_hex(value: u8) -> char {
|
|
||||||
match value {
|
|
||||||
0..=9 => (b'0' + value) as char,
|
|
||||||
10..=15 => (b'a' + (value - 10)) as char,
|
|
||||||
_ => unreachable!(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn clear_cookie(name: &str, secure: bool, same_site: &str) -> String {
|
|
||||||
format!(
|
|
||||||
"{name}=; Path=/; HttpOnly; SameSite={same_site}; Max-Age=0{secure_flag}",
|
|
||||||
secure_flag = if secure { "; Secure" } else { "" }
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::{
|
|
||||||
SESSION_COOKIE, clear_session_cookie, hash_string, make_session_cookie, origin_matches,
|
|
||||||
read_cookie_from_headers,
|
|
||||||
};
|
|
||||||
use axum::http::{HeaderMap, header};
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn hash_string_is_stable() {
|
|
||||||
assert_eq!(
|
|
||||||
hash_string("example"),
|
|
||||||
"50d858e0985ecc7f60418aaf0cc5ab587f42c2570a884095a9e8ccacd0f6545c"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn reads_named_cookie_from_header() {
|
|
||||||
let mut headers = HeaderMap::new();
|
|
||||||
headers.insert(
|
|
||||||
header::COOKIE,
|
|
||||||
"other=value; chattz_session=session-123; another=ok"
|
|
||||||
.parse()
|
|
||||||
.unwrap(),
|
|
||||||
);
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
read_cookie_from_headers(&headers, SESSION_COOKIE),
|
|
||||||
Some("session-123".to_string())
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn origin_match_requires_exact_origin() {
|
|
||||||
let mut headers = HeaderMap::new();
|
|
||||||
headers.insert(header::ORIGIN, "http://localhost:3000".parse().unwrap());
|
|
||||||
|
|
||||||
assert!(origin_matches(&headers, "http://localhost:3000"));
|
|
||||||
assert!(!origin_matches(&headers, "https://localhost:3000"));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn session_cookie_has_strict_policy_and_clear_cookie_expires() {
|
|
||||||
let session_cookie = make_session_cookie("session-123", false);
|
|
||||||
let cleared_cookie = clear_session_cookie(false);
|
|
||||||
|
|
||||||
assert!(session_cookie.contains("HttpOnly"));
|
|
||||||
assert!(session_cookie.contains("SameSite=Strict"));
|
|
||||||
assert!(session_cookie.contains("Max-Age="));
|
|
||||||
assert!(cleared_cookie.contains("SameSite=Strict"));
|
|
||||||
assert!(cleared_cookie.contains("Max-Age=0"));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
312
src/chat.rs
312
src/chat.rs
|
|
@ -1,30 +1,15 @@
|
||||||
use std::collections::HashMap;
|
use crate::AppState;
|
||||||
|
|
||||||
use axum::extract::ws::{Message, WebSocket};
|
use axum::extract::ws::{Message, WebSocket};
|
||||||
use futures_util::{SinkExt, StreamExt};
|
use futures_util::{SinkExt, StreamExt};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
use std::collections::HashMap;
|
||||||
use tokio::sync::{RwLock, mpsc};
|
use tokio::sync::{RwLock, mpsc};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
use crate::{AppState, db};
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct ChatClient {
|
|
||||||
tx: mpsc::UnboundedSender<ServerEvent>,
|
|
||||||
is_idle: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Serialize, Clone, Debug, PartialEq, Eq)]
|
|
||||||
pub struct OnlineUser {
|
|
||||||
pub user_id: Uuid,
|
|
||||||
pub online: bool,
|
|
||||||
pub idle: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Default)]
|
||||||
pub struct ChatHub {
|
pub struct ChatHub {
|
||||||
// user_id -> connection_id -> client
|
// user_id -> sender
|
||||||
clients: RwLock<HashMap<Uuid, HashMap<Uuid, ChatClient>>>,
|
clients: RwLock<HashMap<Uuid, mpsc::UnboundedSender<ServerEvent>>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize, Clone)]
|
#[derive(Serialize, Clone)]
|
||||||
|
|
@ -41,120 +26,68 @@ pub enum ServerEvent {
|
||||||
UserPresence {
|
UserPresence {
|
||||||
user_id: Uuid,
|
user_id: Uuid,
|
||||||
online: bool,
|
online: bool,
|
||||||
idle: bool,
|
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
#[serde(tag = "type", rename_all = "snake_case")]
|
#[serde(tag = "type", rename_all = "snake_case")]
|
||||||
enum ClientEvent {
|
enum ClientEvent {
|
||||||
|
// Currently no interactive client events for the general chat WS
|
||||||
Ping,
|
Ping,
|
||||||
SetIdleStatus { is_idle: bool },
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ChatHub {
|
impl ChatHub {
|
||||||
pub async fn add_client(
|
pub async fn add_client(&self, user_id: Uuid, tx: mpsc::UnboundedSender<ServerEvent>) {
|
||||||
&self,
|
|
||||||
user_id: Uuid,
|
|
||||||
connection_id: Uuid,
|
|
||||||
tx: mpsc::UnboundedSender<ServerEvent>,
|
|
||||||
) -> Option<OnlineUser> {
|
|
||||||
let mut clients = self.clients.write().await;
|
let mut clients = self.clients.write().await;
|
||||||
let previous = aggregate_presence(clients.get(&user_id), user_id);
|
clients.insert(user_id, tx);
|
||||||
|
|
||||||
clients
|
|
||||||
.entry(user_id)
|
|
||||||
.or_default()
|
|
||||||
.insert(connection_id, ChatClient { tx, is_idle: false });
|
|
||||||
|
|
||||||
let current = aggregate_presence(clients.get(&user_id), user_id);
|
|
||||||
presence_delta(previous, current)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn remove_client(&self, user_id: Uuid, connection_id: Uuid) -> Option<OnlineUser> {
|
pub async fn remove_client(&self, user_id: Uuid) {
|
||||||
let mut clients = self.clients.write().await;
|
let mut clients = self.clients.write().await;
|
||||||
let previous = aggregate_presence(clients.get(&user_id), user_id);
|
clients.remove(&user_id);
|
||||||
|
|
||||||
if let Some(connections) = clients.get_mut(&user_id) {
|
|
||||||
connections.remove(&connection_id);
|
|
||||||
if connections.is_empty() {
|
|
||||||
clients.remove(&user_id);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let current = aggregate_presence(clients.get(&user_id), user_id);
|
|
||||||
presence_delta(previous, current)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn get_online_users_for(&self, visible_user_ids: &[Uuid]) -> Vec<OnlineUser> {
|
pub async fn get_online_users(&self) -> Vec<Uuid> {
|
||||||
let clients = self.clients.read().await;
|
let clients = self.clients.read().await;
|
||||||
visible_user_ids
|
clients.keys().cloned().collect()
|
||||||
.iter()
|
|
||||||
.filter_map(|user_id| aggregate_presence(clients.get(user_id), *user_id))
|
|
||||||
.filter(|presence| presence.online)
|
|
||||||
.collect()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn broadcast_to_many(&self, user_ids: Vec<Uuid>, event: ServerEvent) {
|
pub async fn broadcast_all(&self, event: ServerEvent) {
|
||||||
let clients = self.clients.read().await;
|
let clients = self.clients.read().await;
|
||||||
for user_id in user_ids {
|
for tx in clients.values() {
|
||||||
if let Some(connections) = clients.get(&user_id) {
|
let _ = tx.send(event.clone());
|
||||||
for client in connections.values() {
|
|
||||||
let _ = client.tx.send(event.clone());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn broadcast_to_user(&self, user_id: Uuid, event: ServerEvent) {
|
pub async fn broadcast_to_user(&self, user_id: Uuid, event: ServerEvent) {
|
||||||
let clients = self.clients.read().await;
|
let clients = self.clients.read().await;
|
||||||
if let Some(connections) = clients.get(&user_id) {
|
if let Some(tx) = clients.get(&user_id) {
|
||||||
for client in connections.values() {
|
let _ = tx.send(event);
|
||||||
let _ = client.tx.send(event.clone());
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn set_idle_status(
|
pub async fn broadcast_to_many(&self, user_ids: Vec<Uuid>, event: ServerEvent) {
|
||||||
&self,
|
let clients = self.clients.read().await;
|
||||||
user_id: Uuid,
|
for user_id in user_ids {
|
||||||
connection_id: Uuid,
|
if let Some(tx) = clients.get(&user_id) {
|
||||||
is_idle: bool,
|
let _ = tx.send(event.clone());
|
||||||
) -> Option<OnlineUser> {
|
}
|
||||||
let mut clients = self.clients.write().await;
|
|
||||||
let previous = aggregate_presence(clients.get(&user_id), user_id);
|
|
||||||
|
|
||||||
if let Some(connections) = clients.get_mut(&user_id)
|
|
||||||
&& let Some(client) = connections.get_mut(&connection_id)
|
|
||||||
{
|
|
||||||
client.is_idle = is_idle;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
let current = aggregate_presence(clients.get(&user_id), user_id);
|
|
||||||
presence_delta(previous, current)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn handle_socket(state: AppState, socket: WebSocket, user_id: Uuid) {
|
pub async fn handle_socket(state: AppState, socket: WebSocket, user_id: Uuid) {
|
||||||
let (mut ws_sender, mut ws_receiver) = socket.split();
|
let (mut ws_sender, mut ws_receiver) = socket.split();
|
||||||
let (tx, mut rx) = mpsc::unbounded_channel::<ServerEvent>();
|
let (tx, mut rx) = mpsc::unbounded_channel::<ServerEvent>();
|
||||||
let connection_id = Uuid::new_v4();
|
|
||||||
|
|
||||||
if let Some(presence) = state.chat.add_client(user_id, connection_id, tx).await {
|
state.chat.add_client(user_id, tx).await;
|
||||||
if let Ok(visible_user_ids) = db::list_visible_user_ids(&state.db, user_id).await {
|
state
|
||||||
state
|
.chat
|
||||||
.chat
|
.broadcast_all(ServerEvent::UserPresence {
|
||||||
.broadcast_to_many(
|
user_id,
|
||||||
visible_user_ids,
|
online: true,
|
||||||
ServerEvent::UserPresence {
|
})
|
||||||
user_id: presence.user_id,
|
.await;
|
||||||
online: presence.online,
|
|
||||||
idle: presence.idle,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let send_task = tokio::spawn(async move {
|
let send_task = tokio::spawn(async move {
|
||||||
while let Some(event) = rx.recv().await {
|
while let Some(event) = rx.recv().await {
|
||||||
|
|
@ -168,187 +101,18 @@ pub async fn handle_socket(state: AppState, socket: WebSocket, user_id: Uuid) {
|
||||||
});
|
});
|
||||||
|
|
||||||
while let Some(Ok(msg)) = ws_receiver.next().await {
|
while let Some(Ok(msg)) = ws_receiver.next().await {
|
||||||
match msg {
|
if let Message::Close(_) = msg {
|
||||||
Message::Close(_) => break,
|
break;
|
||||||
Message::Text(text) => {
|
|
||||||
if let Ok(ClientEvent::SetIdleStatus { is_idle }) =
|
|
||||||
serde_json::from_str::<ClientEvent>(&text)
|
|
||||||
&& let Some(presence) = state
|
|
||||||
.chat
|
|
||||||
.set_idle_status(user_id, connection_id, is_idle)
|
|
||||||
.await
|
|
||||||
&& let Ok(visible_user_ids) =
|
|
||||||
db::list_visible_user_ids(&state.db, user_id).await
|
|
||||||
{
|
|
||||||
state
|
|
||||||
.chat
|
|
||||||
.broadcast_to_many(
|
|
||||||
visible_user_ids,
|
|
||||||
ServerEvent::UserPresence {
|
|
||||||
user_id: presence.user_id,
|
|
||||||
online: presence.online,
|
|
||||||
idle: presence.idle,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
send_task.abort();
|
send_task.abort();
|
||||||
if let Some(presence) = state.chat.remove_client(user_id, connection_id).await
|
state.chat.remove_client(user_id).await;
|
||||||
&& let Ok(visible_user_ids) = db::list_visible_user_ids(&state.db, user_id).await
|
state
|
||||||
{
|
.chat
|
||||||
state
|
.broadcast_all(ServerEvent::UserPresence {
|
||||||
.chat
|
|
||||||
.broadcast_to_many(
|
|
||||||
visible_user_ids,
|
|
||||||
ServerEvent::UserPresence {
|
|
||||||
user_id: presence.user_id,
|
|
||||||
online: presence.online,
|
|
||||||
idle: presence.idle,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn aggregate_presence(
|
|
||||||
connections: Option<&HashMap<Uuid, ChatClient>>,
|
|
||||||
user_id: Uuid,
|
|
||||||
) -> Option<OnlineUser> {
|
|
||||||
let connections = connections?;
|
|
||||||
if connections.is_empty() {
|
|
||||||
return None;
|
|
||||||
}
|
|
||||||
|
|
||||||
let idle = connections.values().all(|client| client.is_idle);
|
|
||||||
Some(OnlineUser {
|
|
||||||
user_id,
|
|
||||||
online: true,
|
|
||||||
idle,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
fn presence_delta(previous: Option<OnlineUser>, current: Option<OnlineUser>) -> Option<OnlineUser> {
|
|
||||||
match (previous, current) {
|
|
||||||
(None, None) => None,
|
|
||||||
(Some(prev), Some(curr)) if prev == curr => None,
|
|
||||||
(Some(prev), None) => Some(OnlineUser {
|
|
||||||
user_id: prev.user_id,
|
|
||||||
online: false,
|
|
||||||
idle: false,
|
|
||||||
}),
|
|
||||||
(_, Some(curr)) => Some(curr),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::{ChatHub, OnlineUser, ServerEvent};
|
|
||||||
use serde_json::json;
|
|
||||||
use tokio::sync::mpsc;
|
|
||||||
use uuid::Uuid;
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn multiple_connections_keep_user_online_until_last_disconnect() {
|
|
||||||
let hub = ChatHub::default();
|
|
||||||
let user_id = Uuid::new_v4();
|
|
||||||
let first_connection = Uuid::new_v4();
|
|
||||||
let second_connection = Uuid::new_v4();
|
|
||||||
let (tx1, _rx1) = mpsc::unbounded_channel();
|
|
||||||
let (tx2, _rx2) = mpsc::unbounded_channel();
|
|
||||||
|
|
||||||
let first_presence = hub.add_client(user_id, first_connection, tx1).await;
|
|
||||||
let second_presence = hub.add_client(user_id, second_connection, tx2).await;
|
|
||||||
let after_first_disconnect = hub.remove_client(user_id, first_connection).await;
|
|
||||||
let after_last_disconnect = hub.remove_client(user_id, second_connection).await;
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
first_presence,
|
|
||||||
Some(OnlineUser {
|
|
||||||
user_id,
|
|
||||||
online: true,
|
|
||||||
idle: false,
|
|
||||||
})
|
|
||||||
);
|
|
||||||
assert_eq!(second_presence, None);
|
|
||||||
assert_eq!(after_first_disconnect, None);
|
|
||||||
assert_eq!(
|
|
||||||
after_last_disconnect,
|
|
||||||
Some(OnlineUser {
|
|
||||||
user_id,
|
|
||||||
online: false,
|
|
||||||
idle: false,
|
|
||||||
})
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn idle_status_only_flips_when_all_connections_are_idle() {
|
|
||||||
let hub = ChatHub::default();
|
|
||||||
let user_id = Uuid::new_v4();
|
|
||||||
let first_connection = Uuid::new_v4();
|
|
||||||
let second_connection = Uuid::new_v4();
|
|
||||||
let (tx1, _rx1) = mpsc::unbounded_channel();
|
|
||||||
let (tx2, _rx2) = mpsc::unbounded_channel();
|
|
||||||
|
|
||||||
hub.add_client(user_id, first_connection, tx1).await;
|
|
||||||
hub.add_client(user_id, second_connection, tx2).await;
|
|
||||||
|
|
||||||
let first_idle = hub.set_idle_status(user_id, first_connection, true).await;
|
|
||||||
let second_idle = hub.set_idle_status(user_id, second_connection, true).await;
|
|
||||||
let active_again = hub.set_idle_status(user_id, first_connection, false).await;
|
|
||||||
|
|
||||||
assert_eq!(first_idle, None);
|
|
||||||
assert_eq!(
|
|
||||||
second_idle,
|
|
||||||
Some(OnlineUser {
|
|
||||||
user_id,
|
|
||||||
online: true,
|
|
||||||
idle: true,
|
|
||||||
})
|
|
||||||
);
|
|
||||||
assert_eq!(
|
|
||||||
active_again,
|
|
||||||
Some(OnlineUser {
|
|
||||||
user_id,
|
|
||||||
online: true,
|
|
||||||
idle: false,
|
|
||||||
})
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn broadcast_to_user_reaches_all_active_connections() {
|
|
||||||
let hub = ChatHub::default();
|
|
||||||
let user_id = Uuid::new_v4();
|
|
||||||
let first_connection = Uuid::new_v4();
|
|
||||||
let second_connection = Uuid::new_v4();
|
|
||||||
let (tx1, mut rx1) = mpsc::unbounded_channel();
|
|
||||||
let (tx2, mut rx2) = mpsc::unbounded_channel();
|
|
||||||
|
|
||||||
hub.add_client(user_id, first_connection, tx1).await;
|
|
||||||
hub.add_client(user_id, second_connection, tx2).await;
|
|
||||||
|
|
||||||
hub.broadcast_to_user(
|
|
||||||
user_id,
|
user_id,
|
||||||
ServerEvent::DmCreated {
|
online: false,
|
||||||
other_user_id: Uuid::new_v4(),
|
})
|
||||||
message: json!({ "body": "hello" }),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await;
|
.await;
|
||||||
|
|
||||||
assert!(matches!(
|
|
||||||
rx1.try_recv(),
|
|
||||||
Ok(ServerEvent::DmCreated { message, .. }) if message["body"] == "hello"
|
|
||||||
));
|
|
||||||
assert!(matches!(
|
|
||||||
rx2.try_recv(),
|
|
||||||
Ok(ServerEvent::DmCreated { message, .. }) if message["body"] == "hello"
|
|
||||||
));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
169
src/config.rs
169
src/config.rs
|
|
@ -1,11 +1,8 @@
|
||||||
use anyhow::{Context, Result, anyhow};
|
use anyhow::{Context, Result};
|
||||||
use reqwest::Url;
|
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
pub struct Settings {
|
pub struct Settings {
|
||||||
pub port: u16,
|
pub port: u16,
|
||||||
pub app_base_url: String,
|
|
||||||
pub app_origin: String,
|
|
||||||
pub database_url: String,
|
pub database_url: String,
|
||||||
pub oidc_client_id: String,
|
pub oidc_client_id: String,
|
||||||
pub oidc_client_secret: String,
|
pub oidc_client_secret: String,
|
||||||
|
|
@ -14,36 +11,21 @@ pub struct Settings {
|
||||||
pub oidc_userinfo_url: String,
|
pub oidc_userinfo_url: String,
|
||||||
pub oidc_redirect_url: String,
|
pub oidc_redirect_url: String,
|
||||||
pub oidc_scopes: String,
|
pub oidc_scopes: String,
|
||||||
pub session_cookie_secure: bool,
|
pub session_secret: String,
|
||||||
|
pub cookie_secure: bool,
|
||||||
pub stun_urls: Vec<String>,
|
pub stun_urls: Vec<String>,
|
||||||
pub turn_urls: Vec<String>,
|
pub turn_urls: Vec<String>,
|
||||||
pub turn_username: Option<String>,
|
pub turn_username: Option<String>,
|
||||||
pub turn_password: Option<String>,
|
pub turn_password: Option<String>,
|
||||||
pub media: Option<MediaSettings>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
|
||||||
pub struct MediaSettings {
|
|
||||||
pub account_id: String,
|
|
||||||
pub access_key_id: String,
|
|
||||||
pub secret_access_key: String,
|
|
||||||
pub bucket: String,
|
|
||||||
pub public_base_url: String,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Settings {
|
impl Settings {
|
||||||
pub fn from_env() -> Result<Self> {
|
pub fn from_env() -> Result<Self> {
|
||||||
let app_base_url = required("APP_BASE_URL")?;
|
|
||||||
let app_url = Url::parse(&app_base_url).context("APP_BASE_URL must be a valid URL")?;
|
|
||||||
let app_origin = origin_from_url(&app_url)?;
|
|
||||||
|
|
||||||
Ok(Self {
|
Ok(Self {
|
||||||
port: std::env::var("PORT")
|
port: std::env::var("PORT")
|
||||||
.unwrap_or_else(|_| "3000".into())
|
.unwrap_or_else(|_| "3000".into())
|
||||||
.parse()
|
.parse()
|
||||||
.context("PORT must be a valid u16")?,
|
.context("PORT must be a valid u16")?,
|
||||||
app_base_url: trim_trailing_slash(&app_base_url),
|
|
||||||
app_origin,
|
|
||||||
database_url: required("DATABASE_URL")?,
|
database_url: required("DATABASE_URL")?,
|
||||||
oidc_client_id: required("OIDC_CLIENT_ID")?,
|
oidc_client_id: required("OIDC_CLIENT_ID")?,
|
||||||
oidc_client_secret: required("OIDC_CLIENT_SECRET")?,
|
oidc_client_secret: required("OIDC_CLIENT_SECRET")?,
|
||||||
|
|
@ -51,51 +33,20 @@ impl Settings {
|
||||||
oidc_token_url: required("OIDC_TOKEN_URL")?,
|
oidc_token_url: required("OIDC_TOKEN_URL")?,
|
||||||
oidc_userinfo_url: required("OIDC_USERINFO_URL")?,
|
oidc_userinfo_url: required("OIDC_USERINFO_URL")?,
|
||||||
oidc_redirect_url: required("OIDC_REDIRECT_URL")?,
|
oidc_redirect_url: required("OIDC_REDIRECT_URL")?,
|
||||||
oidc_scopes: std::env::var("OIDC_SCOPES")
|
oidc_scopes: std::env::var("OIDC_SCOPES").unwrap_or_else(|_| "openid profile email".to_string()),
|
||||||
.unwrap_or_else(|_| "openid profile email".to_string()),
|
session_secret: required("SESSION_SECRET")?,
|
||||||
session_cookie_secure: requires_secure_cookie(&app_url)?,
|
cookie_secure: std::env::var("COOKIE_SECURE")
|
||||||
|
.unwrap_or_else(|_| "false".into())
|
||||||
|
.parse()
|
||||||
|
.context("COOKIE_SECURE must be true/false")?,
|
||||||
stun_urls: parse_csv_env("STUN_URLS", "stun:stun.l.google.com:19302"),
|
stun_urls: parse_csv_env("STUN_URLS", "stun:stun.l.google.com:19302"),
|
||||||
turn_urls: parse_csv_env("TURN_URLS", ""),
|
turn_urls: parse_csv_env("TURN_URLS", ""),
|
||||||
turn_username: optional("TURN_USERNAME"),
|
turn_username: optional("TURN_USERNAME"),
|
||||||
turn_password: optional("TURN_PASSWORD"),
|
turn_password: optional("TURN_PASSWORD"),
|
||||||
media: MediaSettings::from_env()?,
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl MediaSettings {
|
|
||||||
fn from_env() -> Result<Option<Self>> {
|
|
||||||
let account_id = optional("R2_ACCOUNT_ID");
|
|
||||||
let access_key_id = optional("R2_ACCESS_KEY_ID");
|
|
||||||
let secret_access_key = optional("R2_SECRET_ACCESS_KEY");
|
|
||||||
let bucket = optional("R2_BUCKET");
|
|
||||||
let public_base_url = optional("MEDIA_BASE_URL").or_else(|| optional("R2_PUBLIC_BASE_URL"));
|
|
||||||
|
|
||||||
if account_id.is_none()
|
|
||||||
&& access_key_id.is_none()
|
|
||||||
&& secret_access_key.is_none()
|
|
||||||
&& bucket.is_none()
|
|
||||||
&& public_base_url.is_none()
|
|
||||||
{
|
|
||||||
return Ok(None);
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(Some(Self {
|
|
||||||
account_id: account_id.context("missing env var R2_ACCOUNT_ID")?,
|
|
||||||
access_key_id: access_key_id.context("missing env var R2_ACCESS_KEY_ID")?,
|
|
||||||
secret_access_key: secret_access_key.context("missing env var R2_SECRET_ACCESS_KEY")?,
|
|
||||||
bucket: bucket.context("missing env var R2_BUCKET")?,
|
|
||||||
public_base_url: trim_trailing_slash(
|
|
||||||
&public_base_url.context("missing env var MEDIA_BASE_URL")?,
|
|
||||||
),
|
|
||||||
}))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn endpoint_url(&self) -> String {
|
|
||||||
format!("https://{}.r2.cloudflarestorage.com", self.account_id)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn required(name: &str) -> Result<String> {
|
fn required(name: &str) -> Result<String> {
|
||||||
std::env::var(name).with_context(|| format!("missing env var {name}"))
|
std::env::var(name).with_context(|| format!("missing env var {name}"))
|
||||||
}
|
}
|
||||||
|
|
@ -115,105 +66,3 @@ fn parse_csv_env(name: &str, default_value: &str) -> Vec<String> {
|
||||||
.map(ToString::to_string)
|
.map(ToString::to_string)
|
||||||
.collect()
|
.collect()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn origin_from_url(url: &Url) -> Result<String> {
|
|
||||||
let host = url
|
|
||||||
.host_str()
|
|
||||||
.ok_or_else(|| anyhow!("APP_BASE_URL must include a host"))?;
|
|
||||||
let mut origin = format!("{}://{}", url.scheme(), host);
|
|
||||||
if let Some(port) = url.port() {
|
|
||||||
origin.push(':');
|
|
||||||
origin.push_str(&port.to_string());
|
|
||||||
}
|
|
||||||
Ok(origin)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn requires_secure_cookie(url: &Url) -> Result<bool> {
|
|
||||||
match url.scheme() {
|
|
||||||
"https" => Ok(true),
|
|
||||||
"http" => {
|
|
||||||
let host = url
|
|
||||||
.host_str()
|
|
||||||
.ok_or_else(|| anyhow!("APP_BASE_URL must include a host"))?;
|
|
||||||
if matches!(host, "localhost" | "127.0.0.1" | "::1") {
|
|
||||||
Ok(false)
|
|
||||||
} else {
|
|
||||||
Err(anyhow!(
|
|
||||||
"APP_BASE_URL must use https outside localhost when session cookies are enabled"
|
|
||||||
))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
other => Err(anyhow!("APP_BASE_URL scheme {other} is not supported")),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn trim_trailing_slash(value: &str) -> String {
|
|
||||||
value.trim_end_matches('/').to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::{MediaSettings, origin_from_url, requires_secure_cookie, trim_trailing_slash};
|
|
||||||
use reqwest::Url;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn origin_from_url_preserves_explicit_port() {
|
|
||||||
let url = Url::parse("https://chat.example.com:8443/app").unwrap();
|
|
||||||
let origin = origin_from_url(&url).unwrap();
|
|
||||||
|
|
||||||
assert_eq!(origin, "https://chat.example.com:8443");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn secure_cookie_is_required_for_https_origins() {
|
|
||||||
let url = Url::parse("https://chat.example.com").unwrap();
|
|
||||||
|
|
||||||
assert_eq!(requires_secure_cookie(&url).unwrap(), true);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn localhost_http_is_allowed_without_secure_cookie() {
|
|
||||||
let url = Url::parse("http://localhost:3000").unwrap();
|
|
||||||
|
|
||||||
assert_eq!(requires_secure_cookie(&url).unwrap(), false);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn non_localhost_http_is_rejected() {
|
|
||||||
let url = Url::parse("http://chat.example.com").unwrap();
|
|
||||||
let err = requires_secure_cookie(&url).unwrap_err();
|
|
||||||
|
|
||||||
assert!(
|
|
||||||
err.to_string()
|
|
||||||
.contains("APP_BASE_URL must use https outside localhost")
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn trim_trailing_slash_removes_only_suffix_slashes() {
|
|
||||||
assert_eq!(
|
|
||||||
trim_trailing_slash("https://chat.example.com///"),
|
|
||||||
"https://chat.example.com"
|
|
||||||
);
|
|
||||||
assert_eq!(
|
|
||||||
trim_trailing_slash("https://chat.example.com/app"),
|
|
||||||
"https://chat.example.com/app"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn media_endpoint_url_uses_account_id() {
|
|
||||||
let media = MediaSettings {
|
|
||||||
account_id: "acct123".to_string(),
|
|
||||||
access_key_id: "key".to_string(),
|
|
||||||
secret_access_key: "secret".to_string(),
|
|
||||||
bucket: "bucket".to_string(),
|
|
||||||
public_base_url: "https://media.example.com".to_string(),
|
|
||||||
};
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
media.endpoint_url(),
|
|
||||||
"https://acct123.r2.cloudflarestorage.com"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
|
||||||
726
src/db.rs
726
src/db.rs
|
|
@ -1,22 +1,17 @@
|
||||||
use std::collections::HashSet;
|
|
||||||
|
|
||||||
use anyhow::{Result, anyhow};
|
use anyhow::{Result, anyhow};
|
||||||
use chrono::{Duration, Utc};
|
use chrono::{Duration, Utc};
|
||||||
use sea_orm::{
|
use sea_orm::{
|
||||||
ActiveModelTrait, ActiveValue::Set, ColumnTrait, Condition, DatabaseConnection,
|
ActiveModelTrait, ActiveValue::Set, ColumnTrait, Condition, DatabaseConnection,
|
||||||
DatabaseTransaction, EntityTrait, PaginatorTrait, QueryFilter, QueryOrder, QuerySelect,
|
DatabaseTransaction, EntityTrait, PaginatorTrait, QueryFilter, QueryOrder, QuerySelect,
|
||||||
Statement, TransactionTrait, sea_query::OnConflict,
|
TransactionTrait, sea_query::OnConflict,
|
||||||
};
|
};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
entity::{
|
entity::{channels, direct_messages, guild_members, guilds, invites, messages, users},
|
||||||
attachments, channels, direct_messages, guild_members, guilds, invites, messages, sessions,
|
|
||||||
soundboard_sounds, users,
|
|
||||||
},
|
|
||||||
models::{
|
models::{
|
||||||
Attachment, BasicUser, Channel, DmConversation, DmMessageWithAuthor, Guild, Invite,
|
BasicUser, Channel, DmConversation, DmMessageWithAuthor, Guild, Invite, MessageWithAuthor,
|
||||||
MessageWithAuthor, SoundboardSound, User,
|
User,
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
@ -28,88 +23,6 @@ pub async fn user_exists(db: &DatabaseConnection, user_id: Uuid) -> Result<bool>
|
||||||
Ok(count > 0)
|
Ok(count > 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn create_session(
|
|
||||||
db: &DatabaseConnection,
|
|
||||||
session_id: &str,
|
|
||||||
user_id: Uuid,
|
|
||||||
expires_at: chrono::DateTime<chrono::FixedOffset>,
|
|
||||||
user_agent_hash: Option<String>,
|
|
||||||
ip_hash: Option<String>,
|
|
||||||
) -> Result<()> {
|
|
||||||
sessions::Entity::insert(sessions::ActiveModel {
|
|
||||||
id: Set(session_id.to_string()),
|
|
||||||
user_id: Set(user_id),
|
|
||||||
expires_at: Set(expires_at),
|
|
||||||
created_at: Set(Utc::now().fixed_offset()),
|
|
||||||
last_seen_at: Set(Utc::now().fixed_offset()),
|
|
||||||
revoked_at: Set(None),
|
|
||||||
user_agent_hash: Set(user_agent_hash),
|
|
||||||
ip_hash: Set(ip_hash),
|
|
||||||
})
|
|
||||||
.exec(db)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn touch_active_session(
|
|
||||||
db: &DatabaseConnection,
|
|
||||||
session_id: &str,
|
|
||||||
) -> Result<Option<Uuid>> {
|
|
||||||
let session = sessions::Entity::find_by_id(session_id.to_string())
|
|
||||||
.one(db)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let Some(session) = session else {
|
|
||||||
return Ok(None);
|
|
||||||
};
|
|
||||||
|
|
||||||
if session.revoked_at.is_some() || session.expires_at <= Utc::now().fixed_offset() {
|
|
||||||
return Ok(None);
|
|
||||||
}
|
|
||||||
|
|
||||||
let user_id = session.user_id;
|
|
||||||
sessions::Entity::update_many()
|
|
||||||
.col_expr(
|
|
||||||
sessions::Column::LastSeenAt,
|
|
||||||
sea_orm::sea_query::Expr::value(Utc::now().fixed_offset()),
|
|
||||||
)
|
|
||||||
.filter(sessions::Column::Id.eq(session_id.to_string()))
|
|
||||||
.exec(db)
|
|
||||||
.await?;
|
|
||||||
Ok(Some(user_id))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn revoke_session(db: &DatabaseConnection, session_id: &str) -> Result<()> {
|
|
||||||
if let Some(session) = sessions::Entity::find_by_id(session_id.to_string())
|
|
||||||
.one(db)
|
|
||||||
.await?
|
|
||||||
{
|
|
||||||
sessions::Entity::update_many()
|
|
||||||
.col_expr(
|
|
||||||
sessions::Column::RevokedAt,
|
|
||||||
sea_orm::sea_query::Expr::value(Some(Utc::now().fixed_offset())),
|
|
||||||
)
|
|
||||||
.filter(sessions::Column::Id.eq(session.id))
|
|
||||||
.exec(db)
|
|
||||||
.await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn cleanup_sessions(db: &DatabaseConnection) -> Result<()> {
|
|
||||||
sessions::Entity::delete_many()
|
|
||||||
.filter(
|
|
||||||
Condition::any()
|
|
||||||
.add(sessions::Column::ExpiresAt.lte(Utc::now().fixed_offset()))
|
|
||||||
.add(sessions::Column::RevokedAt.is_not_null()),
|
|
||||||
)
|
|
||||||
.exec(db)
|
|
||||||
.await?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn upsert_user_from_oidc(
|
pub async fn upsert_user_from_oidc(
|
||||||
db: &DatabaseConnection,
|
db: &DatabaseConnection,
|
||||||
oidc_sub: &str,
|
oidc_sub: &str,
|
||||||
|
|
@ -190,58 +103,18 @@ pub async fn list_guilds_for_user(db: &DatabaseConnection, user_id: Uuid) -> Res
|
||||||
.collect())
|
.collect())
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn list_visible_user_ids(db: &DatabaseConnection, user_id: Uuid) -> Result<Vec<Uuid>> {
|
|
||||||
let guild_ids: Vec<Uuid> = guild_members::Entity::find()
|
|
||||||
.filter(guild_members::Column::UserId.eq(user_id))
|
|
||||||
.select_only()
|
|
||||||
.column(guild_members::Column::GuildId)
|
|
||||||
.into_tuple()
|
|
||||||
.all(db)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let mut visible = HashSet::from([user_id]);
|
|
||||||
|
|
||||||
if !guild_ids.is_empty() {
|
|
||||||
let guild_users = guild_members::Entity::find()
|
|
||||||
.filter(guild_members::Column::GuildId.is_in(guild_ids))
|
|
||||||
.all(db)
|
|
||||||
.await?;
|
|
||||||
visible.extend(guild_users.into_iter().map(|membership| membership.user_id));
|
|
||||||
}
|
|
||||||
|
|
||||||
let dm_rows = direct_messages::Entity::find()
|
|
||||||
.filter(
|
|
||||||
Condition::any()
|
|
||||||
.add(direct_messages::Column::SenderUserId.eq(user_id))
|
|
||||||
.add(direct_messages::Column::RecipientUserId.eq(user_id)),
|
|
||||||
)
|
|
||||||
.all(db)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
for row in dm_rows {
|
|
||||||
if row.sender_user_id == user_id {
|
|
||||||
visible.insert(row.recipient_user_id);
|
|
||||||
} else {
|
|
||||||
visible.insert(row.sender_user_id);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(visible.into_iter().collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn create_guild(
|
pub async fn create_guild(
|
||||||
db: &DatabaseConnection,
|
db: &DatabaseConnection,
|
||||||
owner_user_id: Uuid,
|
owner_user_id: Uuid,
|
||||||
name: &str,
|
name: &str,
|
||||||
) -> Result<Guild> {
|
) -> Result<Guild> {
|
||||||
let txn = db.begin().await?;
|
|
||||||
let guild = guilds::Entity::insert(guilds::ActiveModel {
|
let guild = guilds::Entity::insert(guilds::ActiveModel {
|
||||||
id: Set(Uuid::new_v4()),
|
id: Set(Uuid::new_v4()),
|
||||||
name: Set(name.to_string()),
|
name: Set(name.to_string()),
|
||||||
owner_user_id: Set(owner_user_id),
|
owner_user_id: Set(owner_user_id),
|
||||||
..Default::default()
|
..Default::default()
|
||||||
})
|
})
|
||||||
.exec_with_returning(&txn)
|
.exec_with_returning(db)
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
guild_members::Entity::insert(guild_members::ActiveModel {
|
guild_members::Entity::insert(guild_members::ActiveModel {
|
||||||
|
|
@ -257,25 +130,12 @@ pub async fn create_guild(
|
||||||
.do_nothing()
|
.do_nothing()
|
||||||
.to_owned(),
|
.to_owned(),
|
||||||
)
|
)
|
||||||
.exec(&txn)
|
.exec(db)
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
txn.commit().await?;
|
|
||||||
Ok(map_guild(guild))
|
Ok(map_guild(guild))
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn is_guild_owner(
|
|
||||||
db: &DatabaseConnection,
|
|
||||||
guild_id: Uuid,
|
|
||||||
user_id: Uuid,
|
|
||||||
) -> Result<bool> {
|
|
||||||
let guild = guilds::Entity::find_by_id(guild_id).one(db).await?;
|
|
||||||
match guild {
|
|
||||||
Some(g) => Ok(g.owner_user_id == user_id),
|
|
||||||
None => Ok(false),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn create_invite(
|
pub async fn create_invite(
|
||||||
db: &DatabaseConnection,
|
db: &DatabaseConnection,
|
||||||
guild_id: Uuid,
|
guild_id: Uuid,
|
||||||
|
|
@ -317,12 +177,7 @@ pub async fn create_invite(
|
||||||
pub async fn join_invite(db: &DatabaseConnection, code: &str, user_id: Uuid) -> Result<Guild> {
|
pub async fn join_invite(db: &DatabaseConnection, code: &str, user_id: Uuid) -> Result<Guild> {
|
||||||
let txn = db.begin().await?;
|
let txn = db.begin().await?;
|
||||||
|
|
||||||
let invite = invites::Entity::find()
|
let invite = invites::Entity::find_by_id(code.to_string())
|
||||||
.from_raw_sql(Statement::from_sql_and_values(
|
|
||||||
sea_orm::DatabaseBackend::Postgres,
|
|
||||||
r#"SELECT * FROM invites WHERE code = $1 FOR UPDATE"#,
|
|
||||||
[code.into()],
|
|
||||||
))
|
|
||||||
.one(&txn)
|
.one(&txn)
|
||||||
.await?
|
.await?
|
||||||
.ok_or_else(|| anyhow!("invite not found"))?;
|
.ok_or_else(|| anyhow!("invite not found"))?;
|
||||||
|
|
@ -475,63 +330,6 @@ pub async fn create_message(
|
||||||
author_user_id: model.author_user_id,
|
author_user_id: model.author_user_id,
|
||||||
author_display_name: user.display_name,
|
author_display_name: user.display_name,
|
||||||
body: model.body,
|
body: model.body,
|
||||||
attachments: Vec::new(),
|
|
||||||
created_at: model.created_at,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn create_message_with_attachment(
|
|
||||||
db: &DatabaseConnection,
|
|
||||||
channel_id: Uuid,
|
|
||||||
author_user_id: Uuid,
|
|
||||||
body: &str,
|
|
||||||
object_key: &str,
|
|
||||||
media_url: &str,
|
|
||||||
mime_type: &str,
|
|
||||||
size_bytes: i64,
|
|
||||||
original_filename: &str,
|
|
||||||
) -> Result<MessageWithAuthor> {
|
|
||||||
let txn = db.begin().await?;
|
|
||||||
|
|
||||||
let model = messages::Entity::insert(messages::ActiveModel {
|
|
||||||
id: Set(Uuid::new_v4()),
|
|
||||||
channel_id: Set(channel_id),
|
|
||||||
author_user_id: Set(author_user_id),
|
|
||||||
body: Set(body.to_string()),
|
|
||||||
..Default::default()
|
|
||||||
})
|
|
||||||
.exec_with_returning(&txn)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let attachment = attachments::Entity::insert(attachments::ActiveModel {
|
|
||||||
id: Set(Uuid::new_v4()),
|
|
||||||
channel_message_id: Set(Some(model.id)),
|
|
||||||
direct_message_id: Set(None),
|
|
||||||
uploader_user_id: Set(author_user_id),
|
|
||||||
object_key: Set(object_key.to_string()),
|
|
||||||
media_url: Set(media_url.to_string()),
|
|
||||||
mime_type: Set(mime_type.to_string()),
|
|
||||||
size_bytes: Set(size_bytes),
|
|
||||||
original_filename: Set(original_filename.to_string()),
|
|
||||||
..Default::default()
|
|
||||||
})
|
|
||||||
.exec_with_returning(&txn)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let user = users::Entity::find_by_id(author_user_id)
|
|
||||||
.one(&txn)
|
|
||||||
.await?
|
|
||||||
.ok_or_else(|| anyhow!("author not found"))?;
|
|
||||||
|
|
||||||
txn.commit().await?;
|
|
||||||
|
|
||||||
Ok(MessageWithAuthor {
|
|
||||||
id: model.id,
|
|
||||||
channel_id: model.channel_id,
|
|
||||||
author_user_id: model.author_user_id,
|
|
||||||
author_display_name: user.display_name,
|
|
||||||
body: model.body,
|
|
||||||
attachments: vec![map_attachment(attachment)],
|
|
||||||
created_at: model.created_at,
|
created_at: model.created_at,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
@ -549,10 +347,6 @@ pub async fn list_messages(
|
||||||
.all(db)
|
.all(db)
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
let attachment_map =
|
|
||||||
list_attachments_for_channel_messages(db, rows.iter().map(|(msg, _)| msg.id).collect())
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(rows
|
Ok(rows
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|(msg, user)| {
|
.map(|(msg, user)| {
|
||||||
|
|
@ -565,7 +359,6 @@ pub async fn list_messages(
|
||||||
author_user_id: msg.author_user_id,
|
author_user_id: msg.author_user_id,
|
||||||
author_display_name,
|
author_display_name,
|
||||||
body: msg.body,
|
body: msg.body,
|
||||||
attachments: attachment_map.get(&msg.id).cloned().unwrap_or_default(),
|
|
||||||
created_at: msg.created_at,
|
created_at: msg.created_at,
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
@ -599,63 +392,6 @@ pub async fn create_direct_message(
|
||||||
recipient_user_id: model.recipient_user_id,
|
recipient_user_id: model.recipient_user_id,
|
||||||
author_display_name: user.display_name,
|
author_display_name: user.display_name,
|
||||||
body: model.body,
|
body: model.body,
|
||||||
attachments: Vec::new(),
|
|
||||||
created_at: model.created_at,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn create_direct_message_with_attachment(
|
|
||||||
db: &DatabaseConnection,
|
|
||||||
sender_user_id: Uuid,
|
|
||||||
recipient_user_id: Uuid,
|
|
||||||
body: &str,
|
|
||||||
object_key: &str,
|
|
||||||
media_url: &str,
|
|
||||||
mime_type: &str,
|
|
||||||
size_bytes: i64,
|
|
||||||
original_filename: &str,
|
|
||||||
) -> Result<DmMessageWithAuthor> {
|
|
||||||
let txn = db.begin().await?;
|
|
||||||
|
|
||||||
let model = direct_messages::Entity::insert(direct_messages::ActiveModel {
|
|
||||||
id: Set(Uuid::new_v4()),
|
|
||||||
sender_user_id: Set(sender_user_id),
|
|
||||||
recipient_user_id: Set(recipient_user_id),
|
|
||||||
body: Set(body.to_string()),
|
|
||||||
..Default::default()
|
|
||||||
})
|
|
||||||
.exec_with_returning(&txn)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let attachment = attachments::Entity::insert(attachments::ActiveModel {
|
|
||||||
id: Set(Uuid::new_v4()),
|
|
||||||
channel_message_id: Set(None),
|
|
||||||
direct_message_id: Set(Some(model.id)),
|
|
||||||
uploader_user_id: Set(sender_user_id),
|
|
||||||
object_key: Set(object_key.to_string()),
|
|
||||||
media_url: Set(media_url.to_string()),
|
|
||||||
mime_type: Set(mime_type.to_string()),
|
|
||||||
size_bytes: Set(size_bytes),
|
|
||||||
original_filename: Set(original_filename.to_string()),
|
|
||||||
..Default::default()
|
|
||||||
})
|
|
||||||
.exec_with_returning(&txn)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let user = users::Entity::find_by_id(sender_user_id)
|
|
||||||
.one(&txn)
|
|
||||||
.await?
|
|
||||||
.ok_or_else(|| anyhow!("sender not found"))?;
|
|
||||||
|
|
||||||
txn.commit().await?;
|
|
||||||
|
|
||||||
Ok(DmMessageWithAuthor {
|
|
||||||
id: model.id,
|
|
||||||
author_user_id: model.sender_user_id,
|
|
||||||
recipient_user_id: model.recipient_user_id,
|
|
||||||
author_display_name: user.display_name,
|
|
||||||
body: model.body,
|
|
||||||
attachments: vec![map_attachment(attachment)],
|
|
||||||
created_at: model.created_at,
|
created_at: model.created_at,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
@ -698,9 +434,6 @@ pub async fn list_direct_messages(
|
||||||
.map(|u| (u.id, u.display_name))
|
.map(|u| (u.id, u.display_name))
|
||||||
.collect();
|
.collect();
|
||||||
|
|
||||||
let attachment_map =
|
|
||||||
list_attachments_for_direct_messages(db, rows.iter().map(|msg| msg.id).collect()).await?;
|
|
||||||
|
|
||||||
Ok(rows
|
Ok(rows
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|msg| DmMessageWithAuthor {
|
.map(|msg| DmMessageWithAuthor {
|
||||||
|
|
@ -712,64 +445,11 @@ pub async fn list_direct_messages(
|
||||||
.cloned()
|
.cloned()
|
||||||
.unwrap_or_else(|| "Unknown User".to_string()),
|
.unwrap_or_else(|| "Unknown User".to_string()),
|
||||||
body: msg.body,
|
body: msg.body,
|
||||||
attachments: attachment_map.get(&msg.id).cloned().unwrap_or_default(),
|
|
||||||
created_at: msg.created_at,
|
created_at: msg.created_at,
|
||||||
})
|
})
|
||||||
.collect())
|
.collect())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn list_attachments_for_channel_messages(
|
|
||||||
db: &DatabaseConnection,
|
|
||||||
message_ids: Vec<Uuid>,
|
|
||||||
) -> Result<std::collections::HashMap<Uuid, Vec<Attachment>>> {
|
|
||||||
if message_ids.is_empty() {
|
|
||||||
return Ok(std::collections::HashMap::new());
|
|
||||||
}
|
|
||||||
|
|
||||||
let rows = attachments::Entity::find()
|
|
||||||
.filter(attachments::Column::ChannelMessageId.is_in(message_ids))
|
|
||||||
.order_by_asc(attachments::Column::CreatedAt)
|
|
||||||
.all(db)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let mut grouped = std::collections::HashMap::<Uuid, Vec<Attachment>>::new();
|
|
||||||
for row in rows {
|
|
||||||
if let Some(message_id) = row.channel_message_id {
|
|
||||||
grouped
|
|
||||||
.entry(message_id)
|
|
||||||
.or_default()
|
|
||||||
.push(map_attachment(row));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Ok(grouped)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn list_attachments_for_direct_messages(
|
|
||||||
db: &DatabaseConnection,
|
|
||||||
message_ids: Vec<Uuid>,
|
|
||||||
) -> Result<std::collections::HashMap<Uuid, Vec<Attachment>>> {
|
|
||||||
if message_ids.is_empty() {
|
|
||||||
return Ok(std::collections::HashMap::new());
|
|
||||||
}
|
|
||||||
|
|
||||||
let rows = attachments::Entity::find()
|
|
||||||
.filter(attachments::Column::DirectMessageId.is_in(message_ids))
|
|
||||||
.order_by_asc(attachments::Column::CreatedAt)
|
|
||||||
.all(db)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let mut grouped = std::collections::HashMap::<Uuid, Vec<Attachment>>::new();
|
|
||||||
for row in rows {
|
|
||||||
if let Some(message_id) = row.direct_message_id {
|
|
||||||
grouped
|
|
||||||
.entry(message_id)
|
|
||||||
.or_default()
|
|
||||||
.push(map_attachment(row));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Ok(grouped)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn list_dm_conversations(
|
pub async fn list_dm_conversations(
|
||||||
db: &DatabaseConnection,
|
db: &DatabaseConnection,
|
||||||
current_user_id: Uuid,
|
current_user_id: Uuid,
|
||||||
|
|
@ -843,16 +523,6 @@ fn map_guild(model: guilds::Model) -> Guild {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn map_attachment(model: attachments::Model) -> Attachment {
|
|
||||||
Attachment {
|
|
||||||
id: model.id,
|
|
||||||
media_url: model.media_url,
|
|
||||||
mime_type: model.mime_type,
|
|
||||||
size_bytes: model.size_bytes,
|
|
||||||
original_filename: model.original_filename,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn map_channel(model: channels::Model) -> Channel {
|
fn map_channel(model: channels::Model) -> Channel {
|
||||||
Channel {
|
Channel {
|
||||||
id: model.id,
|
id: model.id,
|
||||||
|
|
@ -905,385 +575,3 @@ async fn increment_invite_use_count(
|
||||||
active.update(txn).await?;
|
active.update(txn).await?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn list_sounds(db: &DatabaseConnection, guild_id: Uuid) -> Result<Vec<SoundboardSound>> {
|
|
||||||
let rows = soundboard_sounds::Entity::find()
|
|
||||||
.filter(soundboard_sounds::Column::GuildId.eq(guild_id))
|
|
||||||
.order_by_asc(soundboard_sounds::Column::CreatedAt)
|
|
||||||
.all(db)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(rows.into_iter().map(map_sound).collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn create_sound(
|
|
||||||
db: &DatabaseConnection,
|
|
||||||
guild_id: Uuid,
|
|
||||||
created_by_user_id: Uuid,
|
|
||||||
name: &str,
|
|
||||||
icon: &str,
|
|
||||||
object_key: &str,
|
|
||||||
media_url: &str,
|
|
||||||
mime_type: &str,
|
|
||||||
size_bytes: i64,
|
|
||||||
) -> Result<SoundboardSound> {
|
|
||||||
let model = soundboard_sounds::Entity::insert(soundboard_sounds::ActiveModel {
|
|
||||||
id: Set(Uuid::new_v4()),
|
|
||||||
guild_id: Set(guild_id),
|
|
||||||
created_by_user_id: Set(created_by_user_id),
|
|
||||||
name: Set(name.to_string()),
|
|
||||||
icon: Set(icon.to_string()),
|
|
||||||
object_key: Set(Some(object_key.to_string())),
|
|
||||||
media_url: Set(media_url.to_string()),
|
|
||||||
mime_type: Set(Some(mime_type.to_string())),
|
|
||||||
size_bytes: Set(Some(size_bytes)),
|
|
||||||
// Keep the legacy column populated until every deployment has applied
|
|
||||||
// the nullable migration and old fallback paths are fully removed.
|
|
||||||
file_path: Set(Some(media_url.to_string())),
|
|
||||||
..Default::default()
|
|
||||||
})
|
|
||||||
.exec_with_returning(db)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(map_sound(model))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_sound_by_id(db: &DatabaseConnection, id: Uuid) -> Result<Option<SoundboardSound>> {
|
|
||||||
let row = soundboard_sounds::Entity::find_by_id(id).one(db).await?;
|
|
||||||
Ok(row.map(map_sound))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_sound_object_key(db: &DatabaseConnection, id: Uuid) -> Result<Option<String>> {
|
|
||||||
let row = soundboard_sounds::Entity::find_by_id(id).one(db).await?;
|
|
||||||
Ok(row.and_then(|sound| sound.object_key))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn delete_sound(db: &DatabaseConnection, id: Uuid) -> Result<()> {
|
|
||||||
soundboard_sounds::Entity::delete_by_id(id).exec(db).await?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn map_sound(model: soundboard_sounds::Model) -> SoundboardSound {
|
|
||||||
SoundboardSound {
|
|
||||||
id: model.id,
|
|
||||||
guild_id: model.guild_id,
|
|
||||||
name: model.name,
|
|
||||||
icon: model.icon,
|
|
||||||
media_url: model.media_url,
|
|
||||||
mime_type: model.mime_type,
|
|
||||||
size_bytes: model.size_bytes,
|
|
||||||
created_by_user_id: model.created_by_user_id,
|
|
||||||
created_at: model.created_at,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::{create_guild, create_sound, join_invite, list_visible_user_ids, validate_invite};
|
|
||||||
use crate::entity::{direct_messages, guild_members, guilds, invites, soundboard_sounds};
|
|
||||||
use chrono::{Duration, Utc};
|
|
||||||
use sea_orm::{DatabaseBackend, MockDatabase, MockExecResult, Value};
|
|
||||||
use std::collections::{BTreeMap, BTreeSet};
|
|
||||||
use uuid::Uuid;
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn visible_users_include_self_shared_guilds_and_dm_partners() {
|
|
||||||
let current_user_id = Uuid::new_v4();
|
|
||||||
let guild_a = Uuid::new_v4();
|
|
||||||
let guild_b = Uuid::new_v4();
|
|
||||||
let guild_peer = Uuid::new_v4();
|
|
||||||
let shared_dm_peer = Uuid::new_v4();
|
|
||||||
|
|
||||||
let db = MockDatabase::new(DatabaseBackend::Postgres)
|
|
||||||
.append_query_results([vec![
|
|
||||||
BTreeMap::from([("guild_id".to_string(), Value::from(guild_a))]),
|
|
||||||
BTreeMap::from([("guild_id".to_string(), Value::from(guild_b))]),
|
|
||||||
]])
|
|
||||||
.append_query_results([vec![
|
|
||||||
guild_members::Model {
|
|
||||||
guild_id: guild_a,
|
|
||||||
user_id: current_user_id,
|
|
||||||
created_at: Utc::now().fixed_offset(),
|
|
||||||
},
|
|
||||||
guild_members::Model {
|
|
||||||
guild_id: guild_b,
|
|
||||||
user_id: current_user_id,
|
|
||||||
created_at: Utc::now().fixed_offset(),
|
|
||||||
},
|
|
||||||
guild_members::Model {
|
|
||||||
guild_id: guild_a,
|
|
||||||
user_id: guild_peer,
|
|
||||||
created_at: Utc::now().fixed_offset(),
|
|
||||||
},
|
|
||||||
]])
|
|
||||||
.append_query_results([vec![
|
|
||||||
direct_messages::Model {
|
|
||||||
id: Uuid::new_v4(),
|
|
||||||
sender_user_id: current_user_id,
|
|
||||||
recipient_user_id: shared_dm_peer,
|
|
||||||
body: "hello".to_string(),
|
|
||||||
created_at: Utc::now().fixed_offset(),
|
|
||||||
},
|
|
||||||
direct_messages::Model {
|
|
||||||
id: Uuid::new_v4(),
|
|
||||||
sender_user_id: shared_dm_peer,
|
|
||||||
recipient_user_id: current_user_id,
|
|
||||||
body: "hi".to_string(),
|
|
||||||
created_at: Utc::now().fixed_offset(),
|
|
||||||
},
|
|
||||||
]])
|
|
||||||
.into_connection();
|
|
||||||
|
|
||||||
let visible = list_visible_user_ids(&db, current_user_id).await.unwrap();
|
|
||||||
let visible: BTreeSet<_> = visible.into_iter().collect();
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
visible,
|
|
||||||
BTreeSet::from([current_user_id, guild_peer, shared_dm_peer])
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn visible_users_returns_self_when_no_relationships_exist() {
|
|
||||||
let current_user_id = Uuid::new_v4();
|
|
||||||
let db = MockDatabase::new(DatabaseBackend::Postgres)
|
|
||||||
.append_query_results([Vec::<BTreeMap<String, Value>>::new()])
|
|
||||||
.append_query_results([Vec::<direct_messages::Model>::new()])
|
|
||||||
.into_connection();
|
|
||||||
|
|
||||||
let visible = list_visible_user_ids(&db, current_user_id).await.unwrap();
|
|
||||||
|
|
||||||
assert_eq!(visible, vec![current_user_id]);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn validate_invite_rejects_expired_invites() {
|
|
||||||
let invite = invites::Model {
|
|
||||||
code: "expired".to_string(),
|
|
||||||
guild_id: Uuid::new_v4(),
|
|
||||||
created_by_user_id: Uuid::new_v4(),
|
|
||||||
created_at: Utc::now().fixed_offset(),
|
|
||||||
expires_at: Some((Utc::now() - Duration::minutes(1)).fixed_offset()),
|
|
||||||
max_uses: Some(5),
|
|
||||||
use_count: 0,
|
|
||||||
};
|
|
||||||
|
|
||||||
let err = validate_invite(&invite).unwrap_err();
|
|
||||||
assert!(err.to_string().contains("invite expired"));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn validate_invite_rejects_exhausted_invites() {
|
|
||||||
let invite = invites::Model {
|
|
||||||
code: "used".to_string(),
|
|
||||||
guild_id: Uuid::new_v4(),
|
|
||||||
created_by_user_id: Uuid::new_v4(),
|
|
||||||
created_at: Utc::now().fixed_offset(),
|
|
||||||
expires_at: Some((Utc::now() + Duration::minutes(5)).fixed_offset()),
|
|
||||||
max_uses: Some(1),
|
|
||||||
use_count: 1,
|
|
||||||
};
|
|
||||||
|
|
||||||
let err = validate_invite(&invite).unwrap_err();
|
|
||||||
assert!(err.to_string().contains("invite exhausted"));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn validate_invite_accepts_active_invites() {
|
|
||||||
let invite = invites::Model {
|
|
||||||
code: "active".to_string(),
|
|
||||||
guild_id: Uuid::new_v4(),
|
|
||||||
created_by_user_id: Uuid::new_v4(),
|
|
||||||
created_at: Utc::now().fixed_offset(),
|
|
||||||
expires_at: Some((Utc::now() + Duration::minutes(5)).fixed_offset()),
|
|
||||||
max_uses: Some(3),
|
|
||||||
use_count: 1,
|
|
||||||
};
|
|
||||||
|
|
||||||
validate_invite(&invite).unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn create_guild_runs_guild_and_membership_in_one_transaction() {
|
|
||||||
let owner_user_id = Uuid::new_v4();
|
|
||||||
let guild_id = Uuid::new_v4();
|
|
||||||
let created_at = Utc::now().fixed_offset();
|
|
||||||
let db = MockDatabase::new(DatabaseBackend::Postgres)
|
|
||||||
.append_query_results([vec![guilds::Model {
|
|
||||||
id: guild_id,
|
|
||||||
name: "Guild".to_string(),
|
|
||||||
owner_user_id,
|
|
||||||
created_at,
|
|
||||||
}]])
|
|
||||||
.append_exec_results([MockExecResult {
|
|
||||||
last_insert_id: 0,
|
|
||||||
rows_affected: 1,
|
|
||||||
}])
|
|
||||||
.into_connection();
|
|
||||||
|
|
||||||
let guild = create_guild(&db, owner_user_id, "Guild").await.unwrap();
|
|
||||||
let transaction_log = format!("{:?}", db.into_transaction_log());
|
|
||||||
|
|
||||||
assert_eq!(guild.id, guild_id);
|
|
||||||
assert!(transaction_log.contains("BEGIN"), "{transaction_log}");
|
|
||||||
assert!(transaction_log.contains("guilds"), "{transaction_log}");
|
|
||||||
assert!(
|
|
||||||
transaction_log.contains("guild_members"),
|
|
||||||
"{transaction_log}"
|
|
||||||
);
|
|
||||||
assert!(transaction_log.contains("COMMIT"), "{transaction_log}");
|
|
||||||
|
|
||||||
let guild_insert = transaction_log.find("guilds");
|
|
||||||
let membership_insert = transaction_log.find("guild_members");
|
|
||||||
assert!(guild_insert.is_some() && membership_insert.is_some());
|
|
||||||
assert!(guild_insert.unwrap() < membership_insert.unwrap());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn join_invite_locks_invite_and_updates_use_count_in_one_transaction() {
|
|
||||||
let user_id = Uuid::new_v4();
|
|
||||||
let guild_id = Uuid::new_v4();
|
|
||||||
let created_by_user_id = Uuid::new_v4();
|
|
||||||
let created_at = Utc::now().fixed_offset();
|
|
||||||
let invite = invites::Model {
|
|
||||||
code: "invite123".to_string(),
|
|
||||||
guild_id,
|
|
||||||
created_by_user_id,
|
|
||||||
created_at,
|
|
||||||
expires_at: Some((Utc::now() + Duration::minutes(5)).fixed_offset()),
|
|
||||||
max_uses: Some(5),
|
|
||||||
use_count: 0,
|
|
||||||
};
|
|
||||||
let updated_invite = invites::Model {
|
|
||||||
use_count: 1,
|
|
||||||
..invite.clone()
|
|
||||||
};
|
|
||||||
let guild = guilds::Model {
|
|
||||||
id: guild_id,
|
|
||||||
name: "Guild".to_string(),
|
|
||||||
owner_user_id: created_by_user_id,
|
|
||||||
created_at,
|
|
||||||
};
|
|
||||||
|
|
||||||
let db = MockDatabase::new(DatabaseBackend::Postgres)
|
|
||||||
.append_query_results([vec![invite]])
|
|
||||||
.append_query_results([Vec::<guild_members::Model>::new()])
|
|
||||||
.append_exec_results([MockExecResult {
|
|
||||||
last_insert_id: 0,
|
|
||||||
rows_affected: 1,
|
|
||||||
}])
|
|
||||||
.append_query_results([vec![updated_invite]])
|
|
||||||
.append_query_results([vec![guild]])
|
|
||||||
.into_connection();
|
|
||||||
|
|
||||||
let joined_guild = join_invite(&db, "invite123", user_id).await.unwrap();
|
|
||||||
let transaction_log = format!("{:?}", db.into_transaction_log());
|
|
||||||
|
|
||||||
assert_eq!(joined_guild.id, guild_id);
|
|
||||||
assert!(transaction_log.contains("BEGIN"), "{transaction_log}");
|
|
||||||
assert!(transaction_log.contains("FOR UPDATE"), "{transaction_log}");
|
|
||||||
assert!(
|
|
||||||
transaction_log.contains("guild_members"),
|
|
||||||
"{transaction_log}"
|
|
||||||
);
|
|
||||||
assert!(transaction_log.contains("UPDATE"), "{transaction_log}");
|
|
||||||
assert!(transaction_log.contains("invites"), "{transaction_log}");
|
|
||||||
assert!(transaction_log.contains("COMMIT"), "{transaction_log}");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn join_invite_does_not_increment_use_count_for_existing_member() {
|
|
||||||
let user_id = Uuid::new_v4();
|
|
||||||
let guild_id = Uuid::new_v4();
|
|
||||||
let created_by_user_id = Uuid::new_v4();
|
|
||||||
let created_at = Utc::now().fixed_offset();
|
|
||||||
let invite = invites::Model {
|
|
||||||
code: "invite123".to_string(),
|
|
||||||
guild_id,
|
|
||||||
created_by_user_id,
|
|
||||||
created_at,
|
|
||||||
expires_at: Some((Utc::now() + Duration::minutes(5)).fixed_offset()),
|
|
||||||
max_uses: Some(5),
|
|
||||||
use_count: 3,
|
|
||||||
};
|
|
||||||
let existing_member = guild_members::Model {
|
|
||||||
guild_id,
|
|
||||||
user_id,
|
|
||||||
created_at,
|
|
||||||
};
|
|
||||||
let guild = guilds::Model {
|
|
||||||
id: guild_id,
|
|
||||||
name: "Guild".to_string(),
|
|
||||||
owner_user_id: created_by_user_id,
|
|
||||||
created_at,
|
|
||||||
};
|
|
||||||
|
|
||||||
let db = MockDatabase::new(DatabaseBackend::Postgres)
|
|
||||||
.append_query_results([vec![invite]])
|
|
||||||
.append_query_results([vec![existing_member]])
|
|
||||||
.append_exec_results([MockExecResult {
|
|
||||||
last_insert_id: 0,
|
|
||||||
rows_affected: 1,
|
|
||||||
}])
|
|
||||||
.append_query_results([vec![guild]])
|
|
||||||
.into_connection();
|
|
||||||
|
|
||||||
let joined_guild = join_invite(&db, "invite123", user_id).await.unwrap();
|
|
||||||
let transaction_log = format!("{:?}", db.into_transaction_log());
|
|
||||||
|
|
||||||
assert_eq!(joined_guild.id, guild_id);
|
|
||||||
assert!(transaction_log.contains("FOR UPDATE"), "{transaction_log}");
|
|
||||||
assert!(
|
|
||||||
transaction_log.contains("guild_members"),
|
|
||||||
"{transaction_log}"
|
|
||||||
);
|
|
||||||
assert!(transaction_log.contains("COMMIT"), "{transaction_log}");
|
|
||||||
assert!(
|
|
||||||
!transaction_log.contains("UPDATE \"invites\""),
|
|
||||||
"{transaction_log}"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn create_sound_keeps_legacy_file_path_populated() {
|
|
||||||
let guild_id = Uuid::new_v4();
|
|
||||||
let sound_id = Uuid::new_v4();
|
|
||||||
let created_by_user_id = Uuid::new_v4();
|
|
||||||
let created_at = Utc::now().fixed_offset();
|
|
||||||
let media_url = "https://media.example.com/soundboard/test.mp3";
|
|
||||||
let db = MockDatabase::new(DatabaseBackend::Postgres)
|
|
||||||
.append_query_results([vec![soundboard_sounds::Model {
|
|
||||||
id: sound_id,
|
|
||||||
guild_id,
|
|
||||||
name: "Airhorn".to_string(),
|
|
||||||
icon: "AH".to_string(),
|
|
||||||
object_key: Some("soundboard/test.mp3".to_string()),
|
|
||||||
media_url: media_url.to_string(),
|
|
||||||
mime_type: Some("audio/mpeg".to_string()),
|
|
||||||
size_bytes: Some(1234),
|
|
||||||
file_path: Some(media_url.to_string()),
|
|
||||||
created_by_user_id,
|
|
||||||
created_at,
|
|
||||||
}]])
|
|
||||||
.into_connection();
|
|
||||||
|
|
||||||
let sound = create_sound(
|
|
||||||
&db,
|
|
||||||
guild_id,
|
|
||||||
created_by_user_id,
|
|
||||||
"Airhorn",
|
|
||||||
"AH",
|
|
||||||
"soundboard/test.mp3",
|
|
||||||
media_url,
|
|
||||||
"audio/mpeg",
|
|
||||||
1234,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
let transaction_log = format!("{:?}", db.into_transaction_log());
|
|
||||||
|
|
||||||
assert_eq!(sound.id, sound_id);
|
|
||||||
assert!(transaction_log.contains("file_path"), "{transaction_log}");
|
|
||||||
assert!(transaction_log.contains(media_url), "{transaction_log}");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -1,66 +0,0 @@
|
||||||
use sea_orm::entity::prelude::*;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
|
|
||||||
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
|
|
||||||
#[sea_orm(table_name = "attachments")]
|
|
||||||
pub struct Model {
|
|
||||||
#[sea_orm(primary_key, auto_increment = false)]
|
|
||||||
pub id: Uuid,
|
|
||||||
pub channel_message_id: Option<Uuid>,
|
|
||||||
pub direct_message_id: Option<Uuid>,
|
|
||||||
pub uploader_user_id: Uuid,
|
|
||||||
pub object_key: String,
|
|
||||||
pub media_url: String,
|
|
||||||
pub mime_type: String,
|
|
||||||
pub size_bytes: i64,
|
|
||||||
pub original_filename: String,
|
|
||||||
pub created_at: DateTimeWithTimeZone,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
|
|
||||||
pub enum Relation {
|
|
||||||
#[sea_orm(
|
|
||||||
belongs_to = "super::messages::Entity",
|
|
||||||
from = "Column::ChannelMessageId",
|
|
||||||
to = "super::messages::Column::Id",
|
|
||||||
on_update = "NoAction",
|
|
||||||
on_delete = "Cascade"
|
|
||||||
)]
|
|
||||||
Messages,
|
|
||||||
#[sea_orm(
|
|
||||||
belongs_to = "super::direct_messages::Entity",
|
|
||||||
from = "Column::DirectMessageId",
|
|
||||||
to = "super::direct_messages::Column::Id",
|
|
||||||
on_update = "NoAction",
|
|
||||||
on_delete = "Cascade"
|
|
||||||
)]
|
|
||||||
DirectMessages,
|
|
||||||
#[sea_orm(
|
|
||||||
belongs_to = "super::users::Entity",
|
|
||||||
from = "Column::UploaderUserId",
|
|
||||||
to = "super::users::Column::Id",
|
|
||||||
on_update = "NoAction",
|
|
||||||
on_delete = "NoAction"
|
|
||||||
)]
|
|
||||||
Users,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Related<super::messages::Entity> for Entity {
|
|
||||||
fn to() -> RelationDef {
|
|
||||||
Relation::Messages.def()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Related<super::direct_messages::Entity> for Entity {
|
|
||||||
fn to() -> RelationDef {
|
|
||||||
Relation::DirectMessages.def()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Related<super::users::Entity> for Entity {
|
|
||||||
fn to() -> RelationDef {
|
|
||||||
Relation::Users.def()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ActiveModelBehavior for ActiveModel {}
|
|
||||||
|
|
@ -1,10 +1,7 @@
|
||||||
pub mod attachments;
|
|
||||||
pub mod channels;
|
pub mod channels;
|
||||||
pub mod direct_messages;
|
pub mod direct_messages;
|
||||||
pub mod guild_members;
|
pub mod guild_members;
|
||||||
pub mod guilds;
|
pub mod guilds;
|
||||||
pub mod invites;
|
pub mod invites;
|
||||||
pub mod messages;
|
pub mod messages;
|
||||||
pub mod sessions;
|
|
||||||
pub mod soundboard_sounds;
|
|
||||||
pub mod users;
|
pub mod users;
|
||||||
|
|
|
||||||
|
|
@ -1,35 +0,0 @@
|
||||||
use sea_orm::entity::prelude::*;
|
|
||||||
|
|
||||||
#[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel)]
|
|
||||||
#[sea_orm(table_name = "sessions")]
|
|
||||||
pub struct Model {
|
|
||||||
#[sea_orm(primary_key, auto_increment = false)]
|
|
||||||
pub id: String,
|
|
||||||
pub user_id: Uuid,
|
|
||||||
pub expires_at: DateTimeWithTimeZone,
|
|
||||||
pub created_at: DateTimeWithTimeZone,
|
|
||||||
pub last_seen_at: DateTimeWithTimeZone,
|
|
||||||
pub revoked_at: Option<DateTimeWithTimeZone>,
|
|
||||||
pub user_agent_hash: Option<String>,
|
|
||||||
pub ip_hash: Option<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
|
|
||||||
pub enum Relation {
|
|
||||||
#[sea_orm(
|
|
||||||
belongs_to = "super::users::Entity",
|
|
||||||
from = "Column::UserId",
|
|
||||||
to = "super::users::Column::Id",
|
|
||||||
on_update = "NoAction",
|
|
||||||
on_delete = "Cascade"
|
|
||||||
)]
|
|
||||||
Users,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Related<super::users::Entity> for Entity {
|
|
||||||
fn to() -> RelationDef {
|
|
||||||
Relation::Users.def()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ActiveModelBehavior for ActiveModel {}
|
|
||||||
|
|
@ -1,53 +0,0 @@
|
||||||
use sea_orm::entity::prelude::*;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
|
|
||||||
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
|
|
||||||
#[sea_orm(table_name = "soundboard_sounds")]
|
|
||||||
pub struct Model {
|
|
||||||
#[sea_orm(primary_key, auto_increment = false)]
|
|
||||||
pub id: Uuid,
|
|
||||||
pub guild_id: Uuid,
|
|
||||||
pub name: String,
|
|
||||||
pub icon: String,
|
|
||||||
pub object_key: Option<String>,
|
|
||||||
pub media_url: String,
|
|
||||||
pub mime_type: Option<String>,
|
|
||||||
pub size_bytes: Option<i64>,
|
|
||||||
pub file_path: Option<String>,
|
|
||||||
pub created_by_user_id: Uuid,
|
|
||||||
pub created_at: DateTimeWithTimeZone,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
|
|
||||||
pub enum Relation {
|
|
||||||
#[sea_orm(
|
|
||||||
belongs_to = "super::guilds::Entity",
|
|
||||||
from = "Column::GuildId",
|
|
||||||
to = "super::guilds::Column::Id",
|
|
||||||
on_update = "NoAction",
|
|
||||||
on_delete = "Cascade"
|
|
||||||
)]
|
|
||||||
Guilds,
|
|
||||||
#[sea_orm(
|
|
||||||
belongs_to = "super::users::Entity",
|
|
||||||
from = "Column::CreatedByUserId",
|
|
||||||
to = "super::users::Column::Id",
|
|
||||||
on_update = "NoAction",
|
|
||||||
on_delete = "NoAction"
|
|
||||||
)]
|
|
||||||
Users,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Related<super::guilds::Entity> for Entity {
|
|
||||||
fn to() -> RelationDef {
|
|
||||||
Relation::Guilds.def()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Related<super::users::Entity> for Entity {
|
|
||||||
fn to() -> RelationDef {
|
|
||||||
Relation::Users.def()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ActiveModelBehavior for ActiveModel {}
|
|
||||||
735
src/handlers.rs
735
src/handlers.rs
|
|
@ -1,21 +1,17 @@
|
||||||
use axum::{
|
use axum::{
|
||||||
Json, Router,
|
Json, Router,
|
||||||
body::Bytes,
|
extract::{Path, Query, State, WebSocketUpgrade},
|
||||||
extract::{DefaultBodyLimit, Path, Query, State, WebSocketUpgrade},
|
|
||||||
http::{HeaderMap, StatusCode, header},
|
http::{HeaderMap, StatusCode, header},
|
||||||
response::{Html, IntoResponse, Redirect},
|
response::{Html, IntoResponse, Redirect},
|
||||||
routing::{get, post},
|
routing::{get, post},
|
||||||
};
|
};
|
||||||
use chrono::Datelike;
|
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
AppState,
|
AppState,
|
||||||
auth::{self, ApiError, AuthUser},
|
auth::{self, ApiError, AuthUser},
|
||||||
chat, db,
|
chat, db, voice,
|
||||||
models::{DmMessageWithAuthor, Guild, MessageWithAuthor, SoundboardSound},
|
|
||||||
voice,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
pub fn routes() -> Router<AppState> {
|
pub fn routes() -> Router<AppState> {
|
||||||
|
|
@ -30,11 +26,6 @@ pub fn routes() -> Router<AppState> {
|
||||||
"/dms/{other_user_id}/messages",
|
"/dms/{other_user_id}/messages",
|
||||||
get(list_dm_messages).post(send_dm_message),
|
get(list_dm_messages).post(send_dm_message),
|
||||||
)
|
)
|
||||||
.route(
|
|
||||||
"/dms/{other_user_id}/attachments",
|
|
||||||
post(upload_dm_attachment),
|
|
||||||
)
|
|
||||||
.layer(DefaultBodyLimit::max(MAX_MEDIA_UPLOAD_BYTES))
|
|
||||||
.route("/presence", get(presence_list))
|
.route("/presence", get(presence_list))
|
||||||
.route("/rtc-config", get(rtc_config))
|
.route("/rtc-config", get(rtc_config))
|
||||||
.route("/guilds", get(list_guilds).post(create_guild))
|
.route("/guilds", get(list_guilds).post(create_guild))
|
||||||
|
|
@ -46,25 +37,11 @@ pub fn routes() -> Router<AppState> {
|
||||||
)
|
)
|
||||||
.route("/guilds/{guild_id}/invites", post(create_invite))
|
.route("/guilds/{guild_id}/invites", post(create_invite))
|
||||||
.route("/invites/{code}/join", post(join_invite))
|
.route("/invites/{code}/join", post(join_invite))
|
||||||
.route(
|
|
||||||
"/guilds/{guild_id}/sounds",
|
|
||||||
get(list_sounds).post(upload_sound),
|
|
||||||
)
|
|
||||||
.layer(DefaultBodyLimit::max(MAX_MEDIA_UPLOAD_BYTES))
|
|
||||||
.route(
|
|
||||||
"/guilds/{guild_id}/sounds/{sound_id}",
|
|
||||||
post(delete_sound_post).delete(delete_sound),
|
|
||||||
)
|
|
||||||
.route("/channels", post(create_channel))
|
.route("/channels", post(create_channel))
|
||||||
.route(
|
.route(
|
||||||
"/channels/{channel_id}/messages",
|
"/channels/{channel_id}/messages",
|
||||||
get(list_messages).post(send_message),
|
get(list_messages).post(send_message),
|
||||||
)
|
)
|
||||||
.route(
|
|
||||||
"/channels/{channel_id}/attachments",
|
|
||||||
post(upload_channel_attachment),
|
|
||||||
)
|
|
||||||
.layer(DefaultBodyLimit::max(MAX_MEDIA_UPLOAD_BYTES))
|
|
||||||
.route("/channels/{channel_id}/voice/ws", get(voice_ws))
|
.route("/channels/{channel_id}/voice/ws", get(voice_ws))
|
||||||
.route("/ws", get(chat_ws))
|
.route("/ws", get(chat_ws))
|
||||||
}
|
}
|
||||||
|
|
@ -92,7 +69,7 @@ async fn auth_login(State(state): State<AppState>) -> Result<impl IntoResponse,
|
||||||
let mut headers = HeaderMap::new();
|
let mut headers = HeaderMap::new();
|
||||||
headers.insert(
|
headers.insert(
|
||||||
header::SET_COOKIE,
|
header::SET_COOKIE,
|
||||||
auth::make_oauth_state_cookie(&oauth_state, state.settings.session_cookie_secure)
|
auth::make_oauth_state_cookie(&oauth_state, state.settings.cookie_secure)
|
||||||
.parse()
|
.parse()
|
||||||
.map_err(|_| ApiError::internal("failed to set oauth cookie"))?,
|
.map_err(|_| ApiError::internal("failed to set oauth cookie"))?,
|
||||||
);
|
);
|
||||||
|
|
@ -158,11 +135,8 @@ async fn auth_callback(
|
||||||
Query(query): Query<AuthCallbackQuery>,
|
Query(query): Query<AuthCallbackQuery>,
|
||||||
headers: HeaderMap,
|
headers: HeaderMap,
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
auth::validate_oauth_state(
|
auth::validate_oauth_state(auth::read_oauth_state_from_headers(&headers), &query.state)
|
||||||
auth::read_cookie_from_headers(&headers, auth::OAUTH_STATE_COOKIE),
|
.map_err(|e| ApiError::bad_request(&e.to_string()))?;
|
||||||
&query.state,
|
|
||||||
)
|
|
||||||
.map_err(|e| ApiError::bad_request(&e.to_string()))?;
|
|
||||||
|
|
||||||
let token_res = state
|
let token_res = state
|
||||||
.http
|
.http
|
||||||
|
|
@ -228,51 +202,40 @@ async fn auth_callback(
|
||||||
.await
|
.await
|
||||||
.map_err(|e| ApiError::internal(&format!("failed to persist user: {e}")))?;
|
.map_err(|e| ApiError::internal(&format!("failed to persist user: {e}")))?;
|
||||||
|
|
||||||
let session_id = auth::new_session_id();
|
let session_cookie = auth::new_session_cookie(
|
||||||
db::create_session(
|
|
||||||
&state.db,
|
|
||||||
&session_id,
|
|
||||||
user.id,
|
user.id,
|
||||||
auth::session_expiry(),
|
&state.settings.session_secret,
|
||||||
auth::user_agent_hash(&headers),
|
state.settings.cookie_secure,
|
||||||
auth::ip_hash(&headers),
|
|
||||||
)
|
)
|
||||||
.await
|
.map_err(|e| ApiError::internal(&e.to_string()))?;
|
||||||
.map_err(|e| ApiError::internal(&format!("failed to create session: {e}")))?;
|
|
||||||
|
|
||||||
let mut headers = HeaderMap::new();
|
let mut headers = HeaderMap::new();
|
||||||
headers.append(
|
headers.append(
|
||||||
header::SET_COOKIE,
|
header::SET_COOKIE,
|
||||||
auth::clear_oauth_state_cookie(state.settings.session_cookie_secure)
|
session_cookie
|
||||||
.parse()
|
.parse()
|
||||||
.map_err(|_| ApiError::internal("failed to clear oauth state cookie"))?,
|
.map_err(|_| ApiError::internal("failed to set session cookie"))?,
|
||||||
);
|
);
|
||||||
headers.append(
|
headers.append(
|
||||||
header::SET_COOKIE,
|
header::SET_COOKIE,
|
||||||
auth::make_session_cookie(&session_id, state.settings.session_cookie_secure)
|
auth::clear_oauth_state_cookie(state.settings.cookie_secure)
|
||||||
.parse()
|
.parse()
|
||||||
.map_err(|_| ApiError::internal("failed to set session cookie"))?,
|
.map_err(|_| ApiError::internal("failed to clear oauth state cookie"))?,
|
||||||
);
|
);
|
||||||
|
|
||||||
Ok((headers, Redirect::to("/")))
|
Ok((headers, Redirect::to("/")))
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn auth_logout(
|
async fn auth_logout(State(state): State<AppState>) -> Result<impl IntoResponse, ApiError> {
|
||||||
State(state): State<AppState>,
|
|
||||||
user: AuthUser,
|
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
|
||||||
db::revoke_session(&state.db, &user.session_id)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&format!("failed to revoke session: {e}")))?;
|
|
||||||
|
|
||||||
let mut headers = HeaderMap::new();
|
let mut headers = HeaderMap::new();
|
||||||
headers.insert(
|
headers.append(
|
||||||
header::SET_COOKIE,
|
header::SET_COOKIE,
|
||||||
auth::clear_session_cookie(state.settings.session_cookie_secure)
|
auth::clear_session_cookie(state.settings.cookie_secure)
|
||||||
.parse()
|
.parse()
|
||||||
.map_err(|_| ApiError::internal("failed to clear session cookie"))?,
|
.map_err(|_| ApiError::internal("failed to clear session cookie"))?,
|
||||||
);
|
);
|
||||||
Ok((headers, StatusCode::NO_CONTENT))
|
|
||||||
|
Ok((StatusCode::NO_CONTENT, headers))
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize)]
|
#[derive(Serialize)]
|
||||||
|
|
@ -333,12 +296,9 @@ async fn me(State(state): State<AppState>, user: AuthUser) -> Result<impl IntoRe
|
||||||
|
|
||||||
async fn presence_list(
|
async fn presence_list(
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
user: AuthUser,
|
_user: AuthUser,
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
let visible_user_ids = db::list_visible_user_ids(&state.db, user.id)
|
let users = state.chat.get_online_users().await;
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&format!("failed to load visible users: {e}")))?;
|
|
||||||
let users = state.chat.get_online_users_for(&visible_user_ids).await;
|
|
||||||
Ok(Json(users))
|
Ok(Json(users))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -439,8 +399,11 @@ async fn join_invite(
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
user: AuthUser,
|
user: AuthUser,
|
||||||
Path(code): Path<String>,
|
Path(code): Path<String>,
|
||||||
) -> Result<Json<Guild>, ApiError> {
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
let guild = db::join_invite(&state.db, &code, user.id).await?;
|
let guild = db::join_invite(&state.db, &code.to_uppercase(), user.id)
|
||||||
|
.await
|
||||||
|
.map_err(|e| ApiError::bad_request(&format!("failed to join invite: {e}")))?;
|
||||||
|
|
||||||
Ok(Json(guild))
|
Ok(Json(guild))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -679,101 +642,11 @@ async fn send_dm_message(
|
||||||
Ok((StatusCode::CREATED, Json(SendMessageResponse { ok: true })))
|
Ok((StatusCode::CREATED, Json(SendMessageResponse { ok: true })))
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn upload_channel_attachment(
|
|
||||||
State(state): State<AppState>,
|
|
||||||
user: AuthUser,
|
|
||||||
Path(channel_id): Path<Uuid>,
|
|
||||||
headers: HeaderMap,
|
|
||||||
body: Bytes,
|
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
|
||||||
ensure_channel_member(&state, channel_id, user.id).await?;
|
|
||||||
let uploaded = upload_media_from_request(
|
|
||||||
&state,
|
|
||||||
&headers,
|
|
||||||
body,
|
|
||||||
channel_object_key_prefix(channel_id),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let message = db::create_message_with_attachment(
|
|
||||||
&state.db,
|
|
||||||
channel_id,
|
|
||||||
user.id,
|
|
||||||
"",
|
|
||||||
&uploaded.object_key,
|
|
||||||
&uploaded.media_url,
|
|
||||||
&uploaded.mime_type,
|
|
||||||
uploaded.size_bytes,
|
|
||||||
&uploaded.original_filename,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&format!("failed to create message attachment: {e}")))?;
|
|
||||||
|
|
||||||
let guild_id = db::guild_id_for_channel(&state.db, channel_id)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&format!("channel lookup failed: {e}")))?
|
|
||||||
.ok_or_else(|| ApiError::internal("channel not found after attachment upload"))?;
|
|
||||||
|
|
||||||
let members = db::list_guild_member_ids(&state.db, guild_id)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&format!("failed to list guild members: {e}")))?;
|
|
||||||
|
|
||||||
broadcast_channel_message(&state, members, channel_id, &message).await;
|
|
||||||
Ok((StatusCode::CREATED, Json(message)))
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn upload_dm_attachment(
|
|
||||||
State(state): State<AppState>,
|
|
||||||
user: AuthUser,
|
|
||||||
Path(other_user_id): Path<Uuid>,
|
|
||||||
headers: HeaderMap,
|
|
||||||
body: Bytes,
|
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
|
||||||
if other_user_id == user.id {
|
|
||||||
return Err(ApiError::bad_request("cannot send dm to yourself"));
|
|
||||||
}
|
|
||||||
|
|
||||||
let other_user_exists = db::user_exists(&state.db, other_user_id)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&format!("user check failed: {e}")))?;
|
|
||||||
if !other_user_exists {
|
|
||||||
return Err(ApiError {
|
|
||||||
status: StatusCode::NOT_FOUND,
|
|
||||||
message: "user not found".to_string(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
let uploaded = upload_media_from_request(
|
|
||||||
&state,
|
|
||||||
&headers,
|
|
||||||
body,
|
|
||||||
dm_object_key_prefix(user.id, other_user_id),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let message = db::create_direct_message_with_attachment(
|
|
||||||
&state.db,
|
|
||||||
user.id,
|
|
||||||
other_user_id,
|
|
||||||
"",
|
|
||||||
&uploaded.object_key,
|
|
||||||
&uploaded.media_url,
|
|
||||||
&uploaded.mime_type,
|
|
||||||
uploaded.size_bytes,
|
|
||||||
&uploaded.original_filename,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&format!("failed to create dm attachment: {e}")))?;
|
|
||||||
|
|
||||||
broadcast_dm_message(&state, user.id, other_user_id, &message).await;
|
|
||||||
Ok((StatusCode::CREATED, Json(message)))
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn voice_ws(
|
async fn voice_ws(
|
||||||
|
ws: WebSocketUpgrade,
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
user: AuthUser,
|
user: AuthUser,
|
||||||
Path(channel_id): Path<Uuid>,
|
Path(channel_id): Path<Uuid>,
|
||||||
ws: WebSocketUpgrade,
|
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
ensure_channel_member(&state, channel_id, user.id).await?;
|
ensure_channel_member(&state, channel_id, user.id).await?;
|
||||||
|
|
||||||
|
|
@ -795,470 +668,13 @@ async fn voice_ws(
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn chat_ws(
|
async fn chat_ws(
|
||||||
|
ws: WebSocketUpgrade,
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
user: AuthUser,
|
user: AuthUser,
|
||||||
ws: WebSocketUpgrade,
|
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
Ok(ws.on_upgrade(move |socket| chat::handle_socket(state, socket, user.id)))
|
Ok(ws.on_upgrade(move |socket| chat::handle_socket(state, socket, user.id)))
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn list_sounds(
|
|
||||||
State(state): State<AppState>,
|
|
||||||
user: AuthUser,
|
|
||||||
Path(guild_id): Path<Uuid>,
|
|
||||||
) -> Result<Json<Vec<SoundboardSound>>, ApiError> {
|
|
||||||
ensure_guild_member(&state, guild_id, user.id).await?;
|
|
||||||
let sounds = db::list_sounds(&state.db, guild_id).await?;
|
|
||||||
Ok(Json(sounds))
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn upload_sound(
|
|
||||||
State(state): State<AppState>,
|
|
||||||
user: AuthUser,
|
|
||||||
Path(guild_id): Path<Uuid>,
|
|
||||||
mut multipart: axum::extract::Multipart,
|
|
||||||
) -> Result<Json<SoundboardSound>, ApiError> {
|
|
||||||
ensure_guild_member(&state, guild_id, user.id).await?;
|
|
||||||
|
|
||||||
let mut name = None;
|
|
||||||
let mut icon = None;
|
|
||||||
let mut file_data = None;
|
|
||||||
let mut file_name = None;
|
|
||||||
|
|
||||||
while let Some(field) = multipart
|
|
||||||
.next_field()
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::bad_request(&e.to_string()))?
|
|
||||||
{
|
|
||||||
let field_name = field.name().unwrap_or_default().to_string();
|
|
||||||
if field_name == "name" {
|
|
||||||
name = Some(
|
|
||||||
field
|
|
||||||
.text()
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::bad_request(&e.to_string()))?,
|
|
||||||
);
|
|
||||||
} else if field_name == "icon" {
|
|
||||||
icon = Some(
|
|
||||||
field
|
|
||||||
.text()
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::bad_request(&e.to_string()))?,
|
|
||||||
);
|
|
||||||
} else if field_name == "file" {
|
|
||||||
file_name = Some(field.file_name().unwrap_or("sound.mp3").to_string());
|
|
||||||
file_data = Some(
|
|
||||||
field
|
|
||||||
.bytes()
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::bad_request(&e.to_string()))?,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let (name, icon, file_data, file_name) = match (name, icon, file_data, file_name) {
|
|
||||||
(Some(n), Some(i), Some(d), Some(f)) => (n, i, d, f),
|
|
||||||
_ => return Err(ApiError::bad_request("missing fields")),
|
|
||||||
};
|
|
||||||
|
|
||||||
let validated = validate_sound_upload(&file_name, &file_data)
|
|
||||||
.map_err(|e| ApiError::bad_request(&e.to_string()))?;
|
|
||||||
|
|
||||||
let storage = state
|
|
||||||
.media
|
|
||||||
.as_ref()
|
|
||||||
.ok_or_else(|| ApiError::service_unavailable("media storage is not configured"))?;
|
|
||||||
|
|
||||||
let object_key = format!(
|
|
||||||
"soundboard/{guild_id}/{}-{}",
|
|
||||||
Uuid::new_v4(),
|
|
||||||
sanitize_file_name(&file_name)
|
|
||||||
);
|
|
||||||
let media_url = storage
|
|
||||||
.upload_object(
|
|
||||||
&object_key,
|
|
||||||
file_data.to_vec(),
|
|
||||||
&validated.mime_type,
|
|
||||||
&file_name,
|
|
||||||
false,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&e.to_string()))?;
|
|
||||||
|
|
||||||
let sound = db::create_sound(
|
|
||||||
&state.db,
|
|
||||||
guild_id,
|
|
||||||
user.id,
|
|
||||||
name.trim(),
|
|
||||||
icon.trim(),
|
|
||||||
&object_key,
|
|
||||||
&media_url,
|
|
||||||
&validated.mime_type,
|
|
||||||
validated.size_bytes,
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
Ok(Json(sound))
|
|
||||||
}
|
|
||||||
|
|
||||||
const MAX_MEDIA_UPLOAD_BYTES: usize = 50 * 1024 * 1024;
|
|
||||||
|
|
||||||
struct UploadedMedia {
|
|
||||||
object_key: String,
|
|
||||||
media_url: String,
|
|
||||||
mime_type: String,
|
|
||||||
size_bytes: i64,
|
|
||||||
original_filename: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
struct ValidatedUpload {
|
|
||||||
mime_type: String,
|
|
||||||
inline: bool,
|
|
||||||
size_bytes: i64,
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn upload_media_from_request(
|
|
||||||
state: &AppState,
|
|
||||||
headers: &HeaderMap,
|
|
||||||
body: Bytes,
|
|
||||||
object_key_prefix: String,
|
|
||||||
) -> Result<UploadedMedia, ApiError> {
|
|
||||||
let storage = state
|
|
||||||
.media
|
|
||||||
.as_ref()
|
|
||||||
.ok_or_else(|| ApiError::service_unavailable("media storage is not configured"))?;
|
|
||||||
|
|
||||||
let original_filename = headers
|
|
||||||
.get("x-file-name")
|
|
||||||
.and_then(|v| v.to_str().ok())
|
|
||||||
.map(str::trim)
|
|
||||||
.filter(|v| !v.is_empty())
|
|
||||||
.map(ToString::to_string)
|
|
||||||
.ok_or_else(|| ApiError::bad_request("missing x-file-name header"))?;
|
|
||||||
|
|
||||||
let raw_mime_type = headers
|
|
||||||
.get(header::CONTENT_TYPE)
|
|
||||||
.and_then(|v| v.to_str().ok())
|
|
||||||
.map(str::trim)
|
|
||||||
.filter(|v| !v.is_empty())
|
|
||||||
.unwrap_or("application/octet-stream")
|
|
||||||
.to_string();
|
|
||||||
|
|
||||||
if body.is_empty() {
|
|
||||||
return Err(ApiError::bad_request("missing file upload"));
|
|
||||||
}
|
|
||||||
if body.len() > MAX_MEDIA_UPLOAD_BYTES {
|
|
||||||
return Err(ApiError::bad_request("file exceeds 50MB upload limit"));
|
|
||||||
}
|
|
||||||
|
|
||||||
let validated = validate_general_upload(&original_filename, &body, &raw_mime_type)
|
|
||||||
.map_err(|e| ApiError::bad_request(&e.to_string()))?;
|
|
||||||
let safe_name = sanitize_file_name(&original_filename);
|
|
||||||
let object_key = format!("{}/{}-{}", object_key_prefix, Uuid::new_v4(), safe_name);
|
|
||||||
let size_bytes = body.len() as i64;
|
|
||||||
let media_url = storage
|
|
||||||
.upload_object(
|
|
||||||
&object_key,
|
|
||||||
body.to_vec(),
|
|
||||||
&validated.mime_type,
|
|
||||||
&original_filename,
|
|
||||||
validated.inline,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&e.to_string()))?;
|
|
||||||
|
|
||||||
Ok(UploadedMedia {
|
|
||||||
object_key,
|
|
||||||
media_url,
|
|
||||||
mime_type: validated.mime_type,
|
|
||||||
size_bytes,
|
|
||||||
original_filename,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn broadcast_channel_message(
|
|
||||||
state: &AppState,
|
|
||||||
members: Vec<Uuid>,
|
|
||||||
channel_id: Uuid,
|
|
||||||
message: &MessageWithAuthor,
|
|
||||||
) {
|
|
||||||
state
|
|
||||||
.chat
|
|
||||||
.broadcast_to_many(
|
|
||||||
members,
|
|
||||||
chat::ServerEvent::MessageCreated {
|
|
||||||
channel_id,
|
|
||||||
message: serde_json::to_value(message).unwrap_or_default(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn broadcast_dm_message(
|
|
||||||
state: &AppState,
|
|
||||||
current_user_id: Uuid,
|
|
||||||
other_user_id: Uuid,
|
|
||||||
message: &DmMessageWithAuthor,
|
|
||||||
) {
|
|
||||||
state
|
|
||||||
.chat
|
|
||||||
.broadcast_to_user(
|
|
||||||
current_user_id,
|
|
||||||
chat::ServerEvent::DmCreated {
|
|
||||||
other_user_id,
|
|
||||||
message: serde_json::to_value(message).unwrap_or_default(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
state
|
|
||||||
.chat
|
|
||||||
.broadcast_to_user(
|
|
||||||
other_user_id,
|
|
||||||
chat::ServerEvent::DmCreated {
|
|
||||||
other_user_id: current_user_id,
|
|
||||||
message: serde_json::to_value(message).unwrap_or_default(),
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
}
|
|
||||||
|
|
||||||
fn sanitize_file_name(file_name: &str) -> String {
|
|
||||||
let sanitized: String = file_name
|
|
||||||
.chars()
|
|
||||||
.map(|ch| {
|
|
||||||
if ch.is_ascii_alphanumeric() || matches!(ch, '.' | '-' | '_') {
|
|
||||||
ch
|
|
||||||
} else {
|
|
||||||
'_'
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
let trimmed = sanitized.trim_matches(|ch| ch == '_' || ch == '.').trim();
|
|
||||||
if trimmed.is_empty() {
|
|
||||||
"upload.bin".to_string()
|
|
||||||
} else {
|
|
||||||
trimmed.to_string()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn validate_general_upload(
|
|
||||||
file_name: &str,
|
|
||||||
bytes: &[u8],
|
|
||||||
claimed_mime: &str,
|
|
||||||
) -> anyhow::Result<ValidatedUpload> {
|
|
||||||
let extension = lower_file_extension(file_name)
|
|
||||||
.ok_or_else(|| anyhow::anyhow!("file extension is required"))?;
|
|
||||||
let normalized_mime = sniff_upload_type(bytes, &extension, claimed_mime)
|
|
||||||
.ok_or_else(|| anyhow::anyhow!("unsupported upload type"))?;
|
|
||||||
let inline = is_inline_media_type(&normalized_mime);
|
|
||||||
|
|
||||||
Ok(ValidatedUpload {
|
|
||||||
mime_type: normalized_mime,
|
|
||||||
inline,
|
|
||||||
size_bytes: bytes.len() as i64,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
fn validate_sound_upload(file_name: &str, bytes: &[u8]) -> anyhow::Result<ValidatedUpload> {
|
|
||||||
let extension = lower_file_extension(file_name)
|
|
||||||
.ok_or_else(|| anyhow::anyhow!("sound file extension is required"))?;
|
|
||||||
let Some(mime_type) = sniff_audio_type(bytes, &extension) else {
|
|
||||||
return Err(anyhow::anyhow!(
|
|
||||||
"only mp3, ogg, and wav sounds are supported"
|
|
||||||
));
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(ValidatedUpload {
|
|
||||||
mime_type,
|
|
||||||
inline: false,
|
|
||||||
size_bytes: bytes.len() as i64,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
fn lower_file_extension(file_name: &str) -> Option<String> {
|
|
||||||
std::path::Path::new(file_name)
|
|
||||||
.extension()
|
|
||||||
.and_then(|ext| ext.to_str())
|
|
||||||
.map(|ext| ext.to_ascii_lowercase())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn sniff_upload_type(bytes: &[u8], extension: &str, claimed_mime: &str) -> Option<String> {
|
|
||||||
sniff_audio_type(bytes, extension)
|
|
||||||
.or_else(|| sniff_image_type(bytes, extension))
|
|
||||||
.or_else(|| sniff_video_type(bytes, extension))
|
|
||||||
.or_else(|| sniff_document_type(extension, claimed_mime))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn sniff_audio_type(bytes: &[u8], extension: &str) -> Option<String> {
|
|
||||||
match extension {
|
|
||||||
"mp3" if bytes.starts_with(b"ID3") || bytes.first().copied() == Some(0xff) => {
|
|
||||||
Some("audio/mpeg".to_string())
|
|
||||||
}
|
|
||||||
"ogg" if bytes.starts_with(b"OggS") => Some("audio/ogg".to_string()),
|
|
||||||
"wav" if bytes.starts_with(b"RIFF") && bytes.get(8..12) == Some(b"WAVE") => {
|
|
||||||
Some("audio/wav".to_string())
|
|
||||||
}
|
|
||||||
_ => None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn sniff_image_type(bytes: &[u8], extension: &str) -> Option<String> {
|
|
||||||
match extension {
|
|
||||||
"png" if bytes.starts_with(b"\x89PNG\r\n\x1a\n") => Some("image/png".to_string()),
|
|
||||||
"jpg" | "jpeg" if bytes.starts_with(&[0xff, 0xd8, 0xff]) => Some("image/jpeg".to_string()),
|
|
||||||
"gif" if bytes.starts_with(b"GIF87a") || bytes.starts_with(b"GIF89a") => {
|
|
||||||
Some("image/gif".to_string())
|
|
||||||
}
|
|
||||||
"webp" if bytes.starts_with(b"RIFF") && bytes.get(8..12) == Some(b"WEBP") => {
|
|
||||||
Some("image/webp".to_string())
|
|
||||||
}
|
|
||||||
_ => None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn sniff_video_type(bytes: &[u8], extension: &str) -> Option<String> {
|
|
||||||
match extension {
|
|
||||||
"mp4"
|
|
||||||
if bytes
|
|
||||||
.windows(8)
|
|
||||||
.any(|window| window == b"ftypisom" || window == b"ftypmp42") =>
|
|
||||||
{
|
|
||||||
Some("video/mp4".to_string())
|
|
||||||
}
|
|
||||||
"webm" if bytes.starts_with(&[0x1a, 0x45, 0xdf, 0xa3]) => Some("video/webm".to_string()),
|
|
||||||
_ => None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn sniff_document_type(extension: &str, claimed_mime: &str) -> Option<String> {
|
|
||||||
match extension {
|
|
||||||
"pdf" => Some("application/pdf".to_string()),
|
|
||||||
"txt" => Some("text/plain".to_string()),
|
|
||||||
"json" => Some("application/json".to_string()),
|
|
||||||
"csv" => Some("text/csv".to_string()),
|
|
||||||
"md" => Some("text/markdown".to_string()),
|
|
||||||
"zip" => Some("application/zip".to_string()),
|
|
||||||
"gz" => Some("application/gzip".to_string()),
|
|
||||||
_ if matches!(
|
|
||||||
claimed_mime,
|
|
||||||
"application/pdf"
|
|
||||||
| "text/plain"
|
|
||||||
| "application/json"
|
|
||||||
| "text/csv"
|
|
||||||
| "text/markdown"
|
|
||||||
| "application/zip"
|
|
||||||
| "application/gzip"
|
|
||||||
) =>
|
|
||||||
{
|
|
||||||
Some(claimed_mime.to_string())
|
|
||||||
}
|
|
||||||
_ => None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn is_inline_media_type(mime_type: &str) -> bool {
|
|
||||||
matches!(
|
|
||||||
mime_type,
|
|
||||||
"image/png"
|
|
||||||
| "image/jpeg"
|
|
||||||
| "image/gif"
|
|
||||||
| "image/webp"
|
|
||||||
| "video/mp4"
|
|
||||||
| "video/webm"
|
|
||||||
| "audio/mpeg"
|
|
||||||
| "audio/ogg"
|
|
||||||
| "audio/wav"
|
|
||||||
| "application/pdf"
|
|
||||||
| "text/plain"
|
|
||||||
| "application/json"
|
|
||||||
| "text/csv"
|
|
||||||
| "text/markdown"
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn channel_object_key_prefix(channel_id: Uuid) -> String {
|
|
||||||
let now = chrono::Utc::now();
|
|
||||||
format!(
|
|
||||||
"channels/{}/{:04}/{:02}",
|
|
||||||
channel_id,
|
|
||||||
now.year(),
|
|
||||||
now.month()
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn dm_object_key_prefix(user_a: Uuid, user_b: Uuid) -> String {
|
|
||||||
let now = chrono::Utc::now();
|
|
||||||
let (left, right) = if user_a <= user_b {
|
|
||||||
(user_a, user_b)
|
|
||||||
} else {
|
|
||||||
(user_b, user_a)
|
|
||||||
};
|
|
||||||
format!(
|
|
||||||
"dms/{}-{}/{:04}/{:02}",
|
|
||||||
left,
|
|
||||||
right,
|
|
||||||
now.year(),
|
|
||||||
now.month()
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn delete_sound_post(
|
|
||||||
state: State<AppState>,
|
|
||||||
user: AuthUser,
|
|
||||||
path: Path<(Uuid, Uuid)>,
|
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
|
||||||
delete_sound(state, user, path).await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn delete_sound(
|
|
||||||
State(state): State<AppState>,
|
|
||||||
user: AuthUser,
|
|
||||||
Path((guild_id, sound_id)): Path<(Uuid, Uuid)>,
|
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
|
||||||
ensure_guild_member(&state, guild_id, user.id).await?;
|
|
||||||
|
|
||||||
let sound = db::get_sound_by_id(&state.db, sound_id)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&e.to_string()))?
|
|
||||||
.ok_or_else(|| ApiError {
|
|
||||||
status: StatusCode::NOT_FOUND,
|
|
||||||
message: "sound not found".to_string(),
|
|
||||||
})?;
|
|
||||||
|
|
||||||
if sound.guild_id != guild_id {
|
|
||||||
return Err(ApiError::bad_request("sound does not belong to this guild"));
|
|
||||||
}
|
|
||||||
|
|
||||||
let is_owner = db::is_guild_owner(&state.db, guild_id, user.id)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&e.to_string()))?;
|
|
||||||
|
|
||||||
if sound.created_by_user_id != user.id && !is_owner {
|
|
||||||
return Err(ApiError {
|
|
||||||
status: StatusCode::FORBIDDEN,
|
|
||||||
message: "you do not have permission to delete this sound".to_string(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
let object_key = db::get_sound_object_key(&state.db, sound_id)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&format!("failed to load sound storage key: {e}")))?;
|
|
||||||
|
|
||||||
if let (Some(storage), Some(object_key)) = (state.media.as_ref(), object_key.as_deref()) {
|
|
||||||
storage
|
|
||||||
.delete_object(object_key)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&format!("failed to delete sound media: {e}")))?;
|
|
||||||
}
|
|
||||||
|
|
||||||
db::delete_sound(&state.db, sound_id)
|
|
||||||
.await
|
|
||||||
.map_err(|e| ApiError::internal(&format!("failed to delete sound record: {e}")))?;
|
|
||||||
|
|
||||||
Ok(StatusCode::NO_CONTENT)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn ensure_guild_member(
|
async fn ensure_guild_member(
|
||||||
state: &AppState,
|
state: &AppState,
|
||||||
guild_id: Uuid,
|
guild_id: Uuid,
|
||||||
|
|
@ -1293,100 +709,3 @@ async fn ensure_channel_member(
|
||||||
|
|
||||||
ensure_guild_member(state, guild_id, user_id).await
|
ensure_guild_member(state, guild_id, user_id).await
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::{
|
|
||||||
sanitize_file_name, upload_media_from_request, validate_general_upload,
|
|
||||||
validate_sound_upload,
|
|
||||||
};
|
|
||||||
use crate::{AppState, chat::ChatHub, config::Settings, voice::VoiceHub};
|
|
||||||
use axum::{
|
|
||||||
body::Bytes,
|
|
||||||
http::{HeaderMap, StatusCode, header},
|
|
||||||
};
|
|
||||||
use sea_orm::{DatabaseBackend, MockDatabase};
|
|
||||||
use std::sync::Arc;
|
|
||||||
|
|
||||||
fn test_state() -> AppState {
|
|
||||||
AppState {
|
|
||||||
db: Arc::new(MockDatabase::new(DatabaseBackend::Postgres).into_connection()),
|
|
||||||
settings: Arc::new(Settings {
|
|
||||||
port: 3000,
|
|
||||||
app_base_url: "http://localhost:3000".to_string(),
|
|
||||||
app_origin: "http://localhost:3000".to_string(),
|
|
||||||
database_url: "postgres://localhost/test".to_string(),
|
|
||||||
oidc_client_id: "client".to_string(),
|
|
||||||
oidc_client_secret: "secret".to_string(),
|
|
||||||
oidc_authorize_url: "http://localhost:3000/oidc/authorize".to_string(),
|
|
||||||
oidc_token_url: "http://localhost:3000/oidc/token".to_string(),
|
|
||||||
oidc_userinfo_url: "http://localhost:3000/oidc/userinfo".to_string(),
|
|
||||||
oidc_redirect_url: "http://localhost:3000/auth/callback".to_string(),
|
|
||||||
oidc_scopes: "openid profile email".to_string(),
|
|
||||||
session_cookie_secure: false,
|
|
||||||
stun_urls: vec!["stun:stun.l.google.com:19302".to_string()],
|
|
||||||
turn_urls: Vec::new(),
|
|
||||||
turn_username: None,
|
|
||||||
turn_password: None,
|
|
||||||
media: None,
|
|
||||||
}),
|
|
||||||
http: reqwest::Client::new(),
|
|
||||||
voice: Arc::new(VoiceHub::default()),
|
|
||||||
chat: Arc::new(ChatHub::default()),
|
|
||||||
media: None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn sanitize_file_name_replaces_unsafe_chars() {
|
|
||||||
assert_eq!(
|
|
||||||
sanitize_file_name("../../hello world?.mp3"),
|
|
||||||
"hello_world_.mp3"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn validates_png_uploads() {
|
|
||||||
let png = b"\x89PNG\r\n\x1a\nrest";
|
|
||||||
let upload = validate_general_upload("image.png", png, "image/png").unwrap();
|
|
||||||
assert_eq!(upload.mime_type, "image/png");
|
|
||||||
assert!(upload.inline);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn rejects_unknown_uploads() {
|
|
||||||
let err = validate_general_upload("payload.exe", b"MZ...", "application/octet-stream")
|
|
||||||
.unwrap_err();
|
|
||||||
assert!(err.to_string().contains("unsupported upload type"));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn validates_sound_uploads() {
|
|
||||||
let wav = b"RIFFdataWAVE";
|
|
||||||
let upload = validate_sound_upload("sound.wav", wav).unwrap();
|
|
||||||
assert_eq!(upload.mime_type, "audio/wav");
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn upload_helper_requires_configured_media_storage() {
|
|
||||||
let state = test_state();
|
|
||||||
let mut headers = HeaderMap::new();
|
|
||||||
headers.insert("x-file-name", "photo.png".parse().unwrap());
|
|
||||||
headers.insert(header::CONTENT_TYPE, "image/png".parse().unwrap());
|
|
||||||
|
|
||||||
let err = match upload_media_from_request(
|
|
||||||
&state,
|
|
||||||
&headers,
|
|
||||||
Bytes::from_static(b"\x89PNG\r\n\x1a\nrest"),
|
|
||||||
"channels/test".to_string(),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(_) => panic!("expected upload helper to reject missing media storage"),
|
|
||||||
Err(err) => err,
|
|
||||||
};
|
|
||||||
|
|
||||||
assert_eq!(err.status, StatusCode::SERVICE_UNAVAILABLE);
|
|
||||||
assert_eq!(err.message, "media storage is not configured");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
|
||||||
1126
src/main.rs
1126
src/main.rs
File diff suppressed because it is too large
Load diff
87
src/media.rs
87
src/media.rs
|
|
@ -1,87 +0,0 @@
|
||||||
use anyhow::{Result, anyhow};
|
|
||||||
use aws_config::BehaviorVersion;
|
|
||||||
use aws_sdk_s3::{
|
|
||||||
Client,
|
|
||||||
config::{Credentials, Region},
|
|
||||||
primitives::ByteStream,
|
|
||||||
};
|
|
||||||
|
|
||||||
use crate::config::MediaSettings;
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct MediaStorage {
|
|
||||||
client: Client,
|
|
||||||
bucket: String,
|
|
||||||
public_base_url: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl MediaStorage {
|
|
||||||
pub async fn new(settings: &MediaSettings) -> Result<Self> {
|
|
||||||
let shared_config = aws_config::defaults(BehaviorVersion::latest())
|
|
||||||
.region(Region::new("auto"))
|
|
||||||
.credentials_provider(Credentials::new(
|
|
||||||
settings.access_key_id.clone(),
|
|
||||||
settings.secret_access_key.clone(),
|
|
||||||
None,
|
|
||||||
None,
|
|
||||||
"chattz-r2",
|
|
||||||
))
|
|
||||||
.load()
|
|
||||||
.await;
|
|
||||||
|
|
||||||
let config = aws_sdk_s3::config::Builder::from(&shared_config)
|
|
||||||
.endpoint_url(settings.endpoint_url())
|
|
||||||
.force_path_style(true)
|
|
||||||
.build();
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
client: Client::from_conf(config),
|
|
||||||
bucket: settings.bucket.clone(),
|
|
||||||
public_base_url: settings.public_base_url.trim_end_matches('/').to_string(),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn upload_object(
|
|
||||||
&self,
|
|
||||||
object_key: &str,
|
|
||||||
bytes: Vec<u8>,
|
|
||||||
content_type: &str,
|
|
||||||
original_filename: &str,
|
|
||||||
inline: bool,
|
|
||||||
) -> Result<String> {
|
|
||||||
self.client
|
|
||||||
.put_object()
|
|
||||||
.bucket(&self.bucket)
|
|
||||||
.key(object_key)
|
|
||||||
.body(ByteStream::from(bytes))
|
|
||||||
.content_type(content_type)
|
|
||||||
.content_disposition(format!(
|
|
||||||
"{}; filename=\"{}\"",
|
|
||||||
if inline { "inline" } else { "attachment" },
|
|
||||||
sanitize_header_value(original_filename)
|
|
||||||
))
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.map_err(|e| anyhow!("failed to upload object to R2: {e}"))?;
|
|
||||||
|
|
||||||
Ok(format!("{}/{}", self.public_base_url, object_key))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn delete_object(&self, object_key: &str) -> Result<()> {
|
|
||||||
self.client
|
|
||||||
.delete_object()
|
|
||||||
.bucket(&self.bucket)
|
|
||||||
.key(object_key)
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.map_err(|e| anyhow!("failed to delete object from R2: {e}"))?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn sanitize_header_value(value: &str) -> String {
|
|
||||||
value
|
|
||||||
.chars()
|
|
||||||
.filter(|c| *c != '\\' && *c != '"' && !c.is_control())
|
|
||||||
.collect()
|
|
||||||
}
|
|
||||||
|
|
@ -11,13 +11,13 @@ impl MigrationTrait for Migration {
|
||||||
Table::create()
|
Table::create()
|
||||||
.table(Users::Table)
|
.table(Users::Table)
|
||||||
.if_not_exists()
|
.if_not_exists()
|
||||||
.col(ColumnDef::new(Users::Id).uuid().not_null().primary_key())
|
|
||||||
.col(
|
.col(
|
||||||
ColumnDef::new(Users::OidcSub)
|
ColumnDef::new(Users::Id)
|
||||||
.string()
|
.uuid()
|
||||||
.not_null()
|
.not_null()
|
||||||
.unique_key(),
|
.primary_key(),
|
||||||
)
|
)
|
||||||
|
.col(ColumnDef::new(Users::OidcSub).string().not_null().unique_key())
|
||||||
.col(ColumnDef::new(Users::Email).string())
|
.col(ColumnDef::new(Users::Email).string())
|
||||||
.col(ColumnDef::new(Users::DisplayName).string().not_null())
|
.col(ColumnDef::new(Users::DisplayName).string().not_null())
|
||||||
.col(ColumnDef::new(Users::AvatarUrl).string())
|
.col(ColumnDef::new(Users::AvatarUrl).string())
|
||||||
|
|
@ -42,7 +42,12 @@ impl MigrationTrait for Migration {
|
||||||
Table::create()
|
Table::create()
|
||||||
.table(Guilds::Table)
|
.table(Guilds::Table)
|
||||||
.if_not_exists()
|
.if_not_exists()
|
||||||
.col(ColumnDef::new(Guilds::Id).uuid().not_null().primary_key())
|
.col(
|
||||||
|
ColumnDef::new(Guilds::Id)
|
||||||
|
.uuid()
|
||||||
|
.not_null()
|
||||||
|
.primary_key(),
|
||||||
|
)
|
||||||
.col(ColumnDef::new(Guilds::Name).string().not_null())
|
.col(ColumnDef::new(Guilds::Name).string().not_null())
|
||||||
.col(ColumnDef::new(Guilds::OwnerUserId).uuid().not_null())
|
.col(ColumnDef::new(Guilds::OwnerUserId).uuid().not_null())
|
||||||
.col(
|
.col(
|
||||||
|
|
@ -103,7 +108,12 @@ impl MigrationTrait for Migration {
|
||||||
Table::create()
|
Table::create()
|
||||||
.table(Channels::Table)
|
.table(Channels::Table)
|
||||||
.if_not_exists()
|
.if_not_exists()
|
||||||
.col(ColumnDef::new(Channels::Id).uuid().not_null().primary_key())
|
.col(
|
||||||
|
ColumnDef::new(Channels::Id)
|
||||||
|
.uuid()
|
||||||
|
.not_null()
|
||||||
|
.primary_key(),
|
||||||
|
)
|
||||||
.col(ColumnDef::new(Channels::GuildId).uuid().not_null())
|
.col(ColumnDef::new(Channels::GuildId).uuid().not_null())
|
||||||
.col(ColumnDef::new(Channels::Name).string().not_null())
|
.col(ColumnDef::new(Channels::Name).string().not_null())
|
||||||
.col(
|
.col(
|
||||||
|
|
@ -128,7 +138,12 @@ impl MigrationTrait for Migration {
|
||||||
Table::create()
|
Table::create()
|
||||||
.table(Messages::Table)
|
.table(Messages::Table)
|
||||||
.if_not_exists()
|
.if_not_exists()
|
||||||
.col(ColumnDef::new(Messages::Id).uuid().not_null().primary_key())
|
.col(
|
||||||
|
ColumnDef::new(Messages::Id)
|
||||||
|
.uuid()
|
||||||
|
.not_null()
|
||||||
|
.primary_key(),
|
||||||
|
)
|
||||||
.col(ColumnDef::new(Messages::ChannelId).uuid().not_null())
|
.col(ColumnDef::new(Messages::ChannelId).uuid().not_null())
|
||||||
.col(ColumnDef::new(Messages::AuthorUserId).uuid().not_null())
|
.col(ColumnDef::new(Messages::AuthorUserId).uuid().not_null())
|
||||||
.col(ColumnDef::new(Messages::Body).text().not_null())
|
.col(ColumnDef::new(Messages::Body).text().not_null())
|
||||||
|
|
|
||||||
|
|
@ -11,12 +11,7 @@ impl MigrationTrait for Migration {
|
||||||
Table::create()
|
Table::create()
|
||||||
.table(Invites::Table)
|
.table(Invites::Table)
|
||||||
.if_not_exists()
|
.if_not_exists()
|
||||||
.col(
|
.col(ColumnDef::new(Invites::Code).string().not_null().primary_key())
|
||||||
ColumnDef::new(Invites::Code)
|
|
||||||
.string()
|
|
||||||
.not_null()
|
|
||||||
.primary_key(),
|
|
||||||
)
|
|
||||||
.col(ColumnDef::new(Invites::GuildId).uuid().not_null())
|
.col(ColumnDef::new(Invites::GuildId).uuid().not_null())
|
||||||
.col(ColumnDef::new(Invites::CreatedByUserId).uuid().not_null())
|
.col(ColumnDef::new(Invites::CreatedByUserId).uuid().not_null())
|
||||||
.col(
|
.col(
|
||||||
|
|
|
||||||
|
|
@ -17,16 +17,8 @@ impl MigrationTrait for Migration {
|
||||||
.not_null()
|
.not_null()
|
||||||
.primary_key(),
|
.primary_key(),
|
||||||
)
|
)
|
||||||
.col(
|
.col(ColumnDef::new(DirectMessages::SenderUserId).uuid().not_null())
|
||||||
ColumnDef::new(DirectMessages::SenderUserId)
|
.col(ColumnDef::new(DirectMessages::RecipientUserId).uuid().not_null())
|
||||||
.uuid()
|
|
||||||
.not_null(),
|
|
||||||
)
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(DirectMessages::RecipientUserId)
|
|
||||||
.uuid()
|
|
||||||
.not_null(),
|
|
||||||
)
|
|
||||||
.col(ColumnDef::new(DirectMessages::Body).text().not_null())
|
.col(ColumnDef::new(DirectMessages::Body).text().not_null())
|
||||||
.col(
|
.col(
|
||||||
ColumnDef::new(DirectMessages::CreatedAt)
|
ColumnDef::new(DirectMessages::CreatedAt)
|
||||||
|
|
|
||||||
|
|
@ -1,86 +0,0 @@
|
||||||
use sea_orm_migration::prelude::*;
|
|
||||||
|
|
||||||
#[derive(DeriveMigrationName)]
|
|
||||||
pub struct Migration;
|
|
||||||
|
|
||||||
#[async_trait::async_trait]
|
|
||||||
impl MigrationTrait for Migration {
|
|
||||||
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
||||||
manager
|
|
||||||
.create_table(
|
|
||||||
Table::create()
|
|
||||||
.table(SoundboardSounds::Table)
|
|
||||||
.if_not_exists()
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(SoundboardSounds::Id)
|
|
||||||
.uuid()
|
|
||||||
.not_null()
|
|
||||||
.primary_key(),
|
|
||||||
)
|
|
||||||
.col(ColumnDef::new(SoundboardSounds::GuildId).uuid().not_null())
|
|
||||||
.col(ColumnDef::new(SoundboardSounds::Name).string().not_null())
|
|
||||||
.col(ColumnDef::new(SoundboardSounds::Icon).string().not_null())
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(SoundboardSounds::FilePath)
|
|
||||||
.string()
|
|
||||||
.not_null(),
|
|
||||||
)
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(SoundboardSounds::CreatedByUserId)
|
|
||||||
.uuid()
|
|
||||||
.not_null(),
|
|
||||||
)
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(SoundboardSounds::CreatedAt)
|
|
||||||
.timestamp_with_time_zone()
|
|
||||||
.not_null()
|
|
||||||
.default(Expr::current_timestamp()),
|
|
||||||
)
|
|
||||||
.foreign_key(
|
|
||||||
ForeignKey::create()
|
|
||||||
.name("fk_soundboard_sounds_guild")
|
|
||||||
.from(SoundboardSounds::Table, SoundboardSounds::GuildId)
|
|
||||||
.to(Guilds::Table, Guilds::Id)
|
|
||||||
.on_delete(ForeignKeyAction::Cascade),
|
|
||||||
)
|
|
||||||
.foreign_key(
|
|
||||||
ForeignKey::create()
|
|
||||||
.name("fk_soundboard_sounds_user")
|
|
||||||
.from(SoundboardSounds::Table, SoundboardSounds::CreatedByUserId)
|
|
||||||
.to(Users::Table, Users::Id),
|
|
||||||
)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
||||||
manager
|
|
||||||
.drop_table(Table::drop().table(SoundboardSounds::Table).to_owned())
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum SoundboardSounds {
|
|
||||||
Table,
|
|
||||||
Id,
|
|
||||||
GuildId,
|
|
||||||
Name,
|
|
||||||
Icon,
|
|
||||||
FilePath,
|
|
||||||
CreatedByUserId,
|
|
||||||
CreatedAt,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum Users {
|
|
||||||
Table,
|
|
||||||
Id,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum Guilds {
|
|
||||||
Table,
|
|
||||||
Id,
|
|
||||||
}
|
|
||||||
|
|
@ -1,122 +0,0 @@
|
||||||
use sea_orm_migration::prelude::*;
|
|
||||||
|
|
||||||
#[derive(DeriveMigrationName)]
|
|
||||||
pub struct Migration;
|
|
||||||
|
|
||||||
#[async_trait::async_trait]
|
|
||||||
impl MigrationTrait for Migration {
|
|
||||||
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
||||||
manager
|
|
||||||
.create_table(
|
|
||||||
Table::create()
|
|
||||||
.table(Attachments::Table)
|
|
||||||
.if_not_exists()
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(Attachments::Id)
|
|
||||||
.uuid()
|
|
||||||
.not_null()
|
|
||||||
.primary_key(),
|
|
||||||
)
|
|
||||||
.col(ColumnDef::new(Attachments::ChannelMessageId).uuid().null())
|
|
||||||
.col(ColumnDef::new(Attachments::DirectMessageId).uuid().null())
|
|
||||||
.col(ColumnDef::new(Attachments::UploaderUserId).uuid().not_null())
|
|
||||||
.col(ColumnDef::new(Attachments::ObjectKey).string().not_null())
|
|
||||||
.col(ColumnDef::new(Attachments::MediaUrl).string().not_null())
|
|
||||||
.col(ColumnDef::new(Attachments::MimeType).string().not_null())
|
|
||||||
.col(ColumnDef::new(Attachments::SizeBytes).big_integer().not_null())
|
|
||||||
.col(ColumnDef::new(Attachments::OriginalFilename).string().not_null())
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(Attachments::CreatedAt)
|
|
||||||
.timestamp_with_time_zone()
|
|
||||||
.not_null()
|
|
||||||
.default(Expr::current_timestamp()),
|
|
||||||
)
|
|
||||||
.foreign_key(
|
|
||||||
ForeignKey::create()
|
|
||||||
.name("fk_attachments_channel_message")
|
|
||||||
.from(Attachments::Table, Attachments::ChannelMessageId)
|
|
||||||
.to(Messages::Table, Messages::Id)
|
|
||||||
.on_delete(ForeignKeyAction::Cascade),
|
|
||||||
)
|
|
||||||
.foreign_key(
|
|
||||||
ForeignKey::create()
|
|
||||||
.name("fk_attachments_direct_message")
|
|
||||||
.from(Attachments::Table, Attachments::DirectMessageId)
|
|
||||||
.to(DirectMessages::Table, DirectMessages::Id)
|
|
||||||
.on_delete(ForeignKeyAction::Cascade),
|
|
||||||
)
|
|
||||||
.foreign_key(
|
|
||||||
ForeignKey::create()
|
|
||||||
.name("fk_attachments_uploader")
|
|
||||||
.from(Attachments::Table, Attachments::UploaderUserId)
|
|
||||||
.to(Users::Table, Users::Id),
|
|
||||||
)
|
|
||||||
.check(
|
|
||||||
Expr::cust(
|
|
||||||
"(channel_message_id IS NOT NULL AND direct_message_id IS NULL) OR (channel_message_id IS NULL AND direct_message_id IS NOT NULL)",
|
|
||||||
),
|
|
||||||
)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
manager
|
|
||||||
.create_index(
|
|
||||||
Index::create()
|
|
||||||
.name("idx_attachments_channel_message")
|
|
||||||
.table(Attachments::Table)
|
|
||||||
.col(Attachments::ChannelMessageId)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
manager
|
|
||||||
.create_index(
|
|
||||||
Index::create()
|
|
||||||
.name("idx_attachments_direct_message")
|
|
||||||
.table(Attachments::Table)
|
|
||||||
.col(Attachments::DirectMessageId)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
||||||
manager
|
|
||||||
.drop_table(Table::drop().table(Attachments::Table).to_owned())
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum Attachments {
|
|
||||||
Table,
|
|
||||||
Id,
|
|
||||||
ChannelMessageId,
|
|
||||||
DirectMessageId,
|
|
||||||
UploaderUserId,
|
|
||||||
ObjectKey,
|
|
||||||
MediaUrl,
|
|
||||||
MimeType,
|
|
||||||
SizeBytes,
|
|
||||||
OriginalFilename,
|
|
||||||
CreatedAt,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum Messages {
|
|
||||||
Table,
|
|
||||||
Id,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum DirectMessages {
|
|
||||||
Table,
|
|
||||||
Id,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum Users {
|
|
||||||
Table,
|
|
||||||
Id,
|
|
||||||
}
|
|
||||||
|
|
@ -1,107 +0,0 @@
|
||||||
use sea_orm_migration::prelude::*;
|
|
||||||
|
|
||||||
#[derive(DeriveMigrationName)]
|
|
||||||
pub struct Migration;
|
|
||||||
|
|
||||||
#[async_trait::async_trait]
|
|
||||||
impl MigrationTrait for Migration {
|
|
||||||
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
||||||
manager
|
|
||||||
.create_table(
|
|
||||||
Table::create()
|
|
||||||
.table(Sessions::Table)
|
|
||||||
.if_not_exists()
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(Sessions::Id)
|
|
||||||
.string()
|
|
||||||
.not_null()
|
|
||||||
.primary_key(),
|
|
||||||
)
|
|
||||||
.col(ColumnDef::new(Sessions::UserId).uuid().not_null())
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(Sessions::ExpiresAt)
|
|
||||||
.timestamp_with_time_zone()
|
|
||||||
.not_null(),
|
|
||||||
)
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(Sessions::CreatedAt)
|
|
||||||
.timestamp_with_time_zone()
|
|
||||||
.not_null()
|
|
||||||
.default(Expr::current_timestamp()),
|
|
||||||
)
|
|
||||||
.col(
|
|
||||||
ColumnDef::new(Sessions::LastSeenAt)
|
|
||||||
.timestamp_with_time_zone()
|
|
||||||
.not_null()
|
|
||||||
.default(Expr::current_timestamp()),
|
|
||||||
)
|
|
||||||
.col(ColumnDef::new(Sessions::RevokedAt).timestamp_with_time_zone())
|
|
||||||
.col(ColumnDef::new(Sessions::UserAgentHash).string())
|
|
||||||
.col(ColumnDef::new(Sessions::IpHash).string())
|
|
||||||
.foreign_key(
|
|
||||||
ForeignKey::create()
|
|
||||||
.name("fk_sessions_user")
|
|
||||||
.from(Sessions::Table, Sessions::UserId)
|
|
||||||
.to(Users::Table, Users::Id)
|
|
||||||
.on_delete(ForeignKeyAction::Cascade),
|
|
||||||
)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
manager
|
|
||||||
.create_index(
|
|
||||||
Index::create()
|
|
||||||
.name("idx_sessions_user_id")
|
|
||||||
.table(Sessions::Table)
|
|
||||||
.col(Sessions::UserId)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
manager
|
|
||||||
.create_index(
|
|
||||||
Index::create()
|
|
||||||
.name("idx_sessions_expires_at")
|
|
||||||
.table(Sessions::Table)
|
|
||||||
.col(Sessions::ExpiresAt)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
manager
|
|
||||||
.create_index(
|
|
||||||
Index::create()
|
|
||||||
.name("idx_sessions_revoked_at")
|
|
||||||
.table(Sessions::Table)
|
|
||||||
.col(Sessions::RevokedAt)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
||||||
manager
|
|
||||||
.drop_table(Table::drop().table(Sessions::Table).to_owned())
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum Sessions {
|
|
||||||
Table,
|
|
||||||
Id,
|
|
||||||
UserId,
|
|
||||||
ExpiresAt,
|
|
||||||
CreatedAt,
|
|
||||||
LastSeenAt,
|
|
||||||
RevokedAt,
|
|
||||||
UserAgentHash,
|
|
||||||
IpHash,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum Users {
|
|
||||||
Table,
|
|
||||||
Id,
|
|
||||||
}
|
|
||||||
|
|
@ -1,72 +0,0 @@
|
||||||
use sea_orm_migration::prelude::*;
|
|
||||||
|
|
||||||
#[derive(DeriveMigrationName)]
|
|
||||||
pub struct Migration;
|
|
||||||
|
|
||||||
#[async_trait::async_trait]
|
|
||||||
impl MigrationTrait for Migration {
|
|
||||||
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
||||||
manager
|
|
||||||
.alter_table(
|
|
||||||
Table::alter()
|
|
||||||
.table(SoundboardSounds::Table)
|
|
||||||
.add_column(ColumnDef::new(SoundboardSounds::ObjectKey).string().null())
|
|
||||||
.add_column(ColumnDef::new(SoundboardSounds::MediaUrl).string().null())
|
|
||||||
.add_column(ColumnDef::new(SoundboardSounds::MimeType).string().null())
|
|
||||||
.add_column(
|
|
||||||
ColumnDef::new(SoundboardSounds::SizeBytes)
|
|
||||||
.big_integer()
|
|
||||||
.null(),
|
|
||||||
)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
manager
|
|
||||||
.get_connection()
|
|
||||||
.execute_unprepared(
|
|
||||||
r#"
|
|
||||||
UPDATE soundboard_sounds
|
|
||||||
SET media_url = file_path
|
|
||||||
WHERE media_url IS NULL
|
|
||||||
"#,
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
manager
|
|
||||||
.alter_table(
|
|
||||||
Table::alter()
|
|
||||||
.table(SoundboardSounds::Table)
|
|
||||||
.modify_column(
|
|
||||||
ColumnDef::new(SoundboardSounds::MediaUrl)
|
|
||||||
.string()
|
|
||||||
.not_null(),
|
|
||||||
)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
||||||
manager
|
|
||||||
.alter_table(
|
|
||||||
Table::alter()
|
|
||||||
.table(SoundboardSounds::Table)
|
|
||||||
.drop_column(SoundboardSounds::ObjectKey)
|
|
||||||
.drop_column(SoundboardSounds::MediaUrl)
|
|
||||||
.drop_column(SoundboardSounds::MimeType)
|
|
||||||
.drop_column(SoundboardSounds::SizeBytes)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum SoundboardSounds {
|
|
||||||
Table,
|
|
||||||
ObjectKey,
|
|
||||||
MediaUrl,
|
|
||||||
MimeType,
|
|
||||||
SizeBytes,
|
|
||||||
}
|
|
||||||
|
|
@ -1,50 +0,0 @@
|
||||||
use sea_orm_migration::prelude::*;
|
|
||||||
|
|
||||||
#[derive(DeriveMigrationName)]
|
|
||||||
pub struct Migration;
|
|
||||||
|
|
||||||
#[async_trait::async_trait]
|
|
||||||
impl MigrationTrait for Migration {
|
|
||||||
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
||||||
manager
|
|
||||||
.alter_table(
|
|
||||||
Table::alter()
|
|
||||||
.table(SoundboardSounds::Table)
|
|
||||||
.modify_column(ColumnDef::new(SoundboardSounds::FilePath).string().null())
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
||||||
manager
|
|
||||||
.get_connection()
|
|
||||||
.execute_unprepared(
|
|
||||||
r#"
|
|
||||||
UPDATE soundboard_sounds
|
|
||||||
SET file_path = media_url
|
|
||||||
WHERE file_path IS NULL
|
|
||||||
"#,
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
manager
|
|
||||||
.alter_table(
|
|
||||||
Table::alter()
|
|
||||||
.table(SoundboardSounds::Table)
|
|
||||||
.modify_column(
|
|
||||||
ColumnDef::new(SoundboardSounds::FilePath)
|
|
||||||
.string()
|
|
||||||
.not_null(),
|
|
||||||
)
|
|
||||||
.to_owned(),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(DeriveIden)]
|
|
||||||
enum SoundboardSounds {
|
|
||||||
Table,
|
|
||||||
FilePath,
|
|
||||||
}
|
|
||||||
|
|
@ -4,11 +4,6 @@ mod m20260213_000001_init;
|
||||||
mod m20260213_000002_invites;
|
mod m20260213_000002_invites;
|
||||||
mod m20260213_000003_channel_kind;
|
mod m20260213_000003_channel_kind;
|
||||||
mod m20260213_000004_direct_messages;
|
mod m20260213_000004_direct_messages;
|
||||||
mod m20260224_000005_soundboard;
|
|
||||||
mod m20260227_000006_attachments;
|
|
||||||
mod m20260227_000007_sessions;
|
|
||||||
mod m20260227_000008_soundboard_media;
|
|
||||||
mod m20260227_000009_soundboard_file_path_nullable;
|
|
||||||
|
|
||||||
pub struct Migrator;
|
pub struct Migrator;
|
||||||
|
|
||||||
|
|
@ -20,11 +15,6 @@ impl MigratorTrait for Migrator {
|
||||||
Box::new(m20260213_000002_invites::Migration),
|
Box::new(m20260213_000002_invites::Migration),
|
||||||
Box::new(m20260213_000003_channel_kind::Migration),
|
Box::new(m20260213_000003_channel_kind::Migration),
|
||||||
Box::new(m20260213_000004_direct_messages::Migration),
|
Box::new(m20260213_000004_direct_messages::Migration),
|
||||||
Box::new(m20260224_000005_soundboard::Migration),
|
|
||||||
Box::new(m20260227_000006_attachments::Migration),
|
|
||||||
Box::new(m20260227_000007_sessions::Migration),
|
|
||||||
Box::new(m20260227_000008_soundboard_media::Migration),
|
|
||||||
Box::new(m20260227_000009_soundboard_file_path_nullable::Migration),
|
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -31,12 +31,12 @@ pub struct Channel {
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize)]
|
#[derive(Debug, Clone, Serialize)]
|
||||||
pub struct Attachment {
|
pub struct Message {
|
||||||
pub id: Uuid,
|
pub id: Uuid,
|
||||||
pub media_url: String,
|
pub channel_id: Uuid,
|
||||||
pub mime_type: String,
|
pub author_user_id: Uuid,
|
||||||
pub size_bytes: i64,
|
pub body: String,
|
||||||
pub original_filename: String,
|
pub created_at: DateTimeWithTimeZone,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize)]
|
#[derive(Debug, Clone, Serialize)]
|
||||||
|
|
@ -46,14 +46,13 @@ pub struct BasicUser {
|
||||||
pub avatar_url: Option<String>,
|
pub avatar_url: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize)]
|
#[derive(Debug, Clone, Serialize, sea_orm::FromQueryResult)]
|
||||||
pub struct MessageWithAuthor {
|
pub struct MessageWithAuthor {
|
||||||
pub id: Uuid,
|
pub id: Uuid,
|
||||||
pub channel_id: Uuid,
|
pub channel_id: Uuid,
|
||||||
pub author_user_id: Uuid,
|
pub author_user_id: Uuid,
|
||||||
pub author_display_name: String,
|
pub author_display_name: String,
|
||||||
pub body: String,
|
pub body: String,
|
||||||
pub attachments: Vec<Attachment>,
|
|
||||||
pub created_at: DateTimeWithTimeZone,
|
pub created_at: DateTimeWithTimeZone,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -72,7 +71,6 @@ pub struct DmMessageWithAuthor {
|
||||||
pub recipient_user_id: Uuid,
|
pub recipient_user_id: Uuid,
|
||||||
pub author_display_name: String,
|
pub author_display_name: String,
|
||||||
pub body: String,
|
pub body: String,
|
||||||
pub attachments: Vec<Attachment>,
|
|
||||||
pub created_at: DateTimeWithTimeZone,
|
pub created_at: DateTimeWithTimeZone,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -86,16 +84,3 @@ pub struct Invite {
|
||||||
pub max_uses: Option<i32>,
|
pub max_uses: Option<i32>,
|
||||||
pub use_count: i32,
|
pub use_count: i32,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize)]
|
|
||||||
pub struct SoundboardSound {
|
|
||||||
pub id: Uuid,
|
|
||||||
pub guild_id: Uuid,
|
|
||||||
pub name: String,
|
|
||||||
pub icon: String,
|
|
||||||
pub media_url: String,
|
|
||||||
pub mime_type: Option<String>,
|
|
||||||
pub size_bytes: Option<i64>,
|
|
||||||
pub created_by_user_id: Uuid,
|
|
||||||
pub created_at: DateTimeWithTimeZone,
|
|
||||||
}
|
|
||||||
|
|
|
||||||
557
src/voice.rs
557
src/voice.rs
|
|
@ -10,16 +10,12 @@ use crate::{AppState, db};
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Default)]
|
||||||
pub struct VoiceHub {
|
pub struct VoiceHub {
|
||||||
rooms: RwLock<HashMap<Uuid, HashMap<Uuid, HashMap<Uuid, ClientHandle>>>>,
|
rooms: RwLock<HashMap<Uuid, HashMap<Uuid, ClientHandle>>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
struct ClientHandle {
|
struct ClientHandle {
|
||||||
display_name: String,
|
display_name: String,
|
||||||
is_sharing_video: bool,
|
|
||||||
is_sharing_screen: bool,
|
|
||||||
is_speaking: bool,
|
|
||||||
is_muted: bool,
|
|
||||||
tx: mpsc::UnboundedSender<ServerEvent>,
|
tx: mpsc::UnboundedSender<ServerEvent>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -27,53 +23,20 @@ struct ClientHandle {
|
||||||
pub struct VoiceParticipant {
|
pub struct VoiceParticipant {
|
||||||
pub user_id: Uuid,
|
pub user_id: Uuid,
|
||||||
pub display_name: String,
|
pub display_name: String,
|
||||||
pub is_sharing_video: bool,
|
|
||||||
pub is_sharing_screen: bool,
|
|
||||||
pub is_speaking: bool,
|
|
||||||
pub is_muted: bool,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize, Clone)]
|
#[derive(Serialize, Clone)]
|
||||||
#[serde(tag = "type", rename_all = "snake_case")]
|
#[serde(tag = "type", rename_all = "snake_case")]
|
||||||
enum ServerEvent {
|
enum ServerEvent {
|
||||||
Peers {
|
Peers { peers: Vec<VoiceParticipant> },
|
||||||
peers: Vec<VoiceParticipant>,
|
PeerJoined { user_id: Uuid, display_name: String },
|
||||||
},
|
PeerLeft { user_id: Uuid },
|
||||||
PeerJoined {
|
|
||||||
user_id: Uuid,
|
|
||||||
display_name: String,
|
|
||||||
},
|
|
||||||
PeerLeft {
|
|
||||||
user_id: Uuid,
|
|
||||||
},
|
|
||||||
VideoStatusChanged {
|
|
||||||
user_id: Uuid,
|
|
||||||
is_sharing_video: bool,
|
|
||||||
},
|
|
||||||
ScreenStatusChanged {
|
|
||||||
user_id: Uuid,
|
|
||||||
is_sharing_screen: bool,
|
|
||||||
},
|
|
||||||
SpeakingStatusChanged {
|
|
||||||
user_id: Uuid,
|
|
||||||
is_speaking: bool,
|
|
||||||
},
|
|
||||||
MuteStatusChanged {
|
|
||||||
user_id: Uuid,
|
|
||||||
is_muted: bool,
|
|
||||||
},
|
|
||||||
Signal {
|
Signal {
|
||||||
from_user_id: Uuid,
|
from_user_id: Uuid,
|
||||||
kind: String,
|
kind: String,
|
||||||
data: serde_json::Value,
|
data: serde_json::Value,
|
||||||
},
|
},
|
||||||
Error {
|
Error { message: String },
|
||||||
message: String,
|
|
||||||
},
|
|
||||||
PlaySound {
|
|
||||||
user_id: Uuid,
|
|
||||||
media_url: String,
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
|
|
@ -84,31 +47,6 @@ enum ClientEvent {
|
||||||
kind: String,
|
kind: String,
|
||||||
data: serde_json::Value,
|
data: serde_json::Value,
|
||||||
},
|
},
|
||||||
SetVideoStatus {
|
|
||||||
is_sharing_video: bool,
|
|
||||||
},
|
|
||||||
SetScreenStatus {
|
|
||||||
is_sharing_screen: bool,
|
|
||||||
},
|
|
||||||
SetSpeakingStatus {
|
|
||||||
is_speaking: bool,
|
|
||||||
},
|
|
||||||
SetMuteStatus {
|
|
||||||
is_muted: bool,
|
|
||||||
},
|
|
||||||
PlaySound {
|
|
||||||
sound_id: Uuid,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Default)]
|
|
||||||
struct VoiceStateChange {
|
|
||||||
joined: Option<VoiceParticipant>,
|
|
||||||
left_user_id: Option<Uuid>,
|
|
||||||
video_changed: Option<(Uuid, bool)>,
|
|
||||||
screen_changed: Option<(Uuid, bool)>,
|
|
||||||
speaking_changed: Option<(Uuid, bool)>,
|
|
||||||
mute_changed: Option<(Uuid, bool)>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl VoiceHub {
|
impl VoiceHub {
|
||||||
|
|
@ -119,7 +57,10 @@ impl VoiceHub {
|
||||||
};
|
};
|
||||||
|
|
||||||
room.iter()
|
room.iter()
|
||||||
.filter_map(|(user_id, connections)| aggregate_participant(Some(connections), *user_id))
|
.map(|(user_id, handle)| VoiceParticipant {
|
||||||
|
user_id: *user_id,
|
||||||
|
display_name: handle.display_name.clone(),
|
||||||
|
})
|
||||||
.collect()
|
.collect()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -127,58 +68,58 @@ impl VoiceHub {
|
||||||
&self,
|
&self,
|
||||||
room_id: Uuid,
|
room_id: Uuid,
|
||||||
user_id: Uuid,
|
user_id: Uuid,
|
||||||
connection_id: Uuid,
|
|
||||||
display_name: String,
|
display_name: String,
|
||||||
tx: mpsc::UnboundedSender<ServerEvent>,
|
tx: mpsc::UnboundedSender<ServerEvent>,
|
||||||
) -> (Vec<VoiceParticipant>, VoiceStateChange) {
|
) -> Vec<VoiceParticipant> {
|
||||||
let mut rooms = self.rooms.write().await;
|
let mut rooms = self.rooms.write().await;
|
||||||
let room = rooms.entry(room_id).or_default();
|
let room = rooms.entry(room_id).or_default();
|
||||||
|
|
||||||
let peers = room
|
let peers = room
|
||||||
.iter()
|
.iter()
|
||||||
.filter_map(|(peer_id, connections)| aggregate_participant(Some(connections), *peer_id))
|
.map(|(peer_id, peer)| VoiceParticipant {
|
||||||
|
user_id: *peer_id,
|
||||||
|
display_name: peer.display_name.clone(),
|
||||||
|
})
|
||||||
.collect::<Vec<_>>();
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
let previous = aggregate_participant(room.get(&user_id), user_id);
|
room.insert(
|
||||||
room.entry(user_id).or_default().insert(
|
user_id,
|
||||||
connection_id,
|
|
||||||
ClientHandle {
|
ClientHandle {
|
||||||
display_name,
|
display_name: display_name.clone(),
|
||||||
is_sharing_video: false,
|
|
||||||
is_sharing_screen: false,
|
|
||||||
is_speaking: false,
|
|
||||||
is_muted: false,
|
|
||||||
tx,
|
tx,
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
let current = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
|
|
||||||
(peers, diff_voice_state(previous, current))
|
for (peer_id, peer) in room.iter() {
|
||||||
}
|
if *peer_id != user_id {
|
||||||
|
let _ = peer.tx.send(ServerEvent::PeerJoined {
|
||||||
async fn leave(&self, room_id: Uuid, user_id: Uuid, connection_id: Uuid) -> VoiceStateChange {
|
user_id,
|
||||||
let mut rooms = self.rooms.write().await;
|
display_name: display_name.clone(),
|
||||||
let Some(room) = rooms.get_mut(&room_id) else {
|
});
|
||||||
return VoiceStateChange::default();
|
|
||||||
};
|
|
||||||
|
|
||||||
let previous = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
if let Some(connections) = room.get_mut(&user_id) {
|
|
||||||
connections.remove(&connection_id);
|
|
||||||
if connections.is_empty() {
|
|
||||||
room.remove(&user_id);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
let current = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
|
peers
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn leave(&self, room_id: Uuid, user_id: Uuid) {
|
||||||
|
let mut rooms = self.rooms.write().await;
|
||||||
|
let Some(room) = rooms.get_mut(&room_id) else {
|
||||||
|
return;
|
||||||
|
};
|
||||||
|
|
||||||
|
room.remove(&user_id);
|
||||||
|
|
||||||
|
for peer in room.values() {
|
||||||
|
let _ = peer.tx.send(ServerEvent::PeerLeft { user_id });
|
||||||
|
}
|
||||||
|
|
||||||
if room.is_empty() {
|
if room.is_empty() {
|
||||||
rooms.remove(&room_id);
|
rooms.remove(&room_id);
|
||||||
}
|
}
|
||||||
|
|
||||||
diff_voice_state(previous, current)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn relay_signal(
|
async fn relay_signal(
|
||||||
&self,
|
&self,
|
||||||
room_id: Uuid,
|
room_id: Uuid,
|
||||||
from_user_id: Uuid,
|
from_user_id: Uuid,
|
||||||
|
|
@ -191,10 +132,7 @@ impl VoiceHub {
|
||||||
return;
|
return;
|
||||||
};
|
};
|
||||||
|
|
||||||
if let Some(target) = room
|
if let Some(target) = room.get(&to_user_id) {
|
||||||
.get(&to_user_id)
|
|
||||||
.and_then(|connections| connections.values().next())
|
|
||||||
{
|
|
||||||
let _ = target.tx.send(ServerEvent::Signal {
|
let _ = target.tx.send(ServerEvent::Signal {
|
||||||
from_user_id,
|
from_user_id,
|
||||||
kind,
|
kind,
|
||||||
|
|
@ -202,214 +140,21 @@ impl VoiceHub {
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn set_video_status(
|
|
||||||
&self,
|
|
||||||
room_id: Uuid,
|
|
||||||
user_id: Uuid,
|
|
||||||
connection_id: Uuid,
|
|
||||||
is_sharing_video: bool,
|
|
||||||
) -> VoiceStateChange {
|
|
||||||
let mut rooms = self.rooms.write().await;
|
|
||||||
let Some(room) = rooms.get_mut(&room_id) else {
|
|
||||||
return VoiceStateChange::default();
|
|
||||||
};
|
|
||||||
|
|
||||||
let previous = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
if let Some(handle) = room
|
|
||||||
.get_mut(&user_id)
|
|
||||||
.and_then(|connections| connections.get_mut(&connection_id))
|
|
||||||
{
|
|
||||||
handle.is_sharing_video = is_sharing_video;
|
|
||||||
}
|
|
||||||
let current = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
diff_voice_state(previous, current)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn set_screen_status(
|
|
||||||
&self,
|
|
||||||
room_id: Uuid,
|
|
||||||
user_id: Uuid,
|
|
||||||
connection_id: Uuid,
|
|
||||||
is_sharing_screen: bool,
|
|
||||||
) -> VoiceStateChange {
|
|
||||||
let mut rooms = self.rooms.write().await;
|
|
||||||
let Some(room) = rooms.get_mut(&room_id) else {
|
|
||||||
return VoiceStateChange::default();
|
|
||||||
};
|
|
||||||
|
|
||||||
let previous = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
if let Some(handle) = room
|
|
||||||
.get_mut(&user_id)
|
|
||||||
.and_then(|connections| connections.get_mut(&connection_id))
|
|
||||||
{
|
|
||||||
handle.is_sharing_screen = is_sharing_screen;
|
|
||||||
}
|
|
||||||
let current = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
diff_voice_state(previous, current)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn set_speaking_status(
|
|
||||||
&self,
|
|
||||||
room_id: Uuid,
|
|
||||||
user_id: Uuid,
|
|
||||||
connection_id: Uuid,
|
|
||||||
is_speaking: bool,
|
|
||||||
) -> VoiceStateChange {
|
|
||||||
let mut rooms = self.rooms.write().await;
|
|
||||||
let Some(room) = rooms.get_mut(&room_id) else {
|
|
||||||
return VoiceStateChange::default();
|
|
||||||
};
|
|
||||||
|
|
||||||
let previous = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
if let Some(handle) = room
|
|
||||||
.get_mut(&user_id)
|
|
||||||
.and_then(|connections| connections.get_mut(&connection_id))
|
|
||||||
{
|
|
||||||
handle.is_speaking = is_speaking;
|
|
||||||
}
|
|
||||||
let current = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
diff_voice_state(previous, current)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn set_mute_status(
|
|
||||||
&self,
|
|
||||||
room_id: Uuid,
|
|
||||||
user_id: Uuid,
|
|
||||||
connection_id: Uuid,
|
|
||||||
is_muted: bool,
|
|
||||||
) -> VoiceStateChange {
|
|
||||||
let mut rooms = self.rooms.write().await;
|
|
||||||
let Some(room) = rooms.get_mut(&room_id) else {
|
|
||||||
return VoiceStateChange::default();
|
|
||||||
};
|
|
||||||
|
|
||||||
let previous = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
if let Some(handle) = room
|
|
||||||
.get_mut(&user_id)
|
|
||||||
.and_then(|connections| connections.get_mut(&connection_id))
|
|
||||||
{
|
|
||||||
handle.is_muted = is_muted;
|
|
||||||
}
|
|
||||||
let current = aggregate_participant(room.get(&user_id), user_id);
|
|
||||||
diff_voice_state(previous, current)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn play_sound(&self, room_id: Uuid, user_id: Uuid, media_url: String) {
|
|
||||||
let rooms = self.rooms.read().await;
|
|
||||||
let Some(room) = rooms.get(&room_id) else {
|
|
||||||
return;
|
|
||||||
};
|
|
||||||
|
|
||||||
for (peer_id, connections) in room.iter() {
|
|
||||||
if *peer_id == user_id {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
for peer in connections.values() {
|
|
||||||
let _ = peer.tx.send(ServerEvent::PlaySound {
|
|
||||||
user_id,
|
|
||||||
media_url: media_url.clone(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn emit_change(&self, room_id: Uuid, user_id: Uuid, change: VoiceStateChange) {
|
|
||||||
if is_voice_state_change_empty(&change) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
let rooms = self.rooms.read().await;
|
|
||||||
let Some(room) = rooms.get(&room_id) else {
|
|
||||||
return;
|
|
||||||
};
|
|
||||||
|
|
||||||
if let Some(participant) = change.joined {
|
|
||||||
broadcast_voice_event(
|
|
||||||
room,
|
|
||||||
user_id,
|
|
||||||
ServerEvent::PeerJoined {
|
|
||||||
user_id: participant.user_id,
|
|
||||||
display_name: participant.display_name,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
if let Some(left_user_id) = change.left_user_id {
|
|
||||||
broadcast_voice_event(
|
|
||||||
room,
|
|
||||||
user_id,
|
|
||||||
ServerEvent::PeerLeft {
|
|
||||||
user_id: left_user_id,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
if let Some((changed_user_id, is_sharing_video)) = change.video_changed {
|
|
||||||
broadcast_voice_event(
|
|
||||||
room,
|
|
||||||
user_id,
|
|
||||||
ServerEvent::VideoStatusChanged {
|
|
||||||
user_id: changed_user_id,
|
|
||||||
is_sharing_video,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
if let Some((changed_user_id, is_sharing_screen)) = change.screen_changed {
|
|
||||||
broadcast_voice_event(
|
|
||||||
room,
|
|
||||||
user_id,
|
|
||||||
ServerEvent::ScreenStatusChanged {
|
|
||||||
user_id: changed_user_id,
|
|
||||||
is_sharing_screen,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
if let Some((changed_user_id, is_speaking)) = change.speaking_changed {
|
|
||||||
broadcast_voice_event(
|
|
||||||
room,
|
|
||||||
user_id,
|
|
||||||
ServerEvent::SpeakingStatusChanged {
|
|
||||||
user_id: changed_user_id,
|
|
||||||
is_speaking,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
if let Some((changed_user_id, is_muted)) = change.mute_changed {
|
|
||||||
broadcast_voice_event(
|
|
||||||
room,
|
|
||||||
user_id,
|
|
||||||
ServerEvent::MuteStatusChanged {
|
|
||||||
user_id: changed_user_id,
|
|
||||||
is_muted,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn handle_socket(state: AppState, socket: WebSocket, room_id: Uuid, user_id: Uuid) {
|
pub async fn handle_socket(state: AppState, socket: WebSocket, room_id: Uuid, user_id: Uuid) {
|
||||||
let Some(user) = db::get_user_by_id(&state.db, user_id).await.ok().flatten() else {
|
let Some(user) = db::get_user_by_id(&state.db, user_id).await.ok().flatten() else {
|
||||||
return;
|
return;
|
||||||
};
|
};
|
||||||
let Ok(Some(guild_id)) = db::guild_id_for_channel(&state.db, room_id).await else {
|
|
||||||
return;
|
|
||||||
};
|
|
||||||
|
|
||||||
let (mut ws_sender, mut ws_receiver) = socket.split();
|
let (mut ws_sender, mut ws_receiver) = socket.split();
|
||||||
let (tx, mut rx) = mpsc::unbounded_channel::<ServerEvent>();
|
let (tx, mut rx) = mpsc::unbounded_channel::<ServerEvent>();
|
||||||
let connection_id = Uuid::new_v4();
|
|
||||||
|
|
||||||
let (peers, change) = state
|
let peers = state
|
||||||
.voice
|
.voice
|
||||||
.join(
|
.join(room_id, user_id, user.display_name.clone(), tx.clone())
|
||||||
room_id,
|
|
||||||
user_id,
|
|
||||||
connection_id,
|
|
||||||
user.display_name.clone(),
|
|
||||||
tx.clone(),
|
|
||||||
)
|
|
||||||
.await;
|
.await;
|
||||||
let _ = tx.send(ServerEvent::Peers { peers });
|
let _ = tx.send(ServerEvent::Peers { peers });
|
||||||
state.voice.emit_change(room_id, user_id, change).await;
|
|
||||||
|
|
||||||
let send_task = tokio::spawn(async move {
|
let send_task = tokio::spawn(async move {
|
||||||
while let Some(event) = rx.recv().await {
|
while let Some(event) = rx.recv().await {
|
||||||
|
|
@ -437,44 +182,6 @@ pub async fn handle_socket(state: AppState, socket: WebSocket, room_id: Uuid, us
|
||||||
.relay_signal(room_id, user_id, to_user_id, kind, data)
|
.relay_signal(room_id, user_id, to_user_id, kind, data)
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
Ok(ClientEvent::SetVideoStatus { is_sharing_video }) => {
|
|
||||||
let change = state
|
|
||||||
.voice
|
|
||||||
.set_video_status(room_id, user_id, connection_id, is_sharing_video)
|
|
||||||
.await;
|
|
||||||
state.voice.emit_change(room_id, user_id, change).await;
|
|
||||||
}
|
|
||||||
Ok(ClientEvent::SetScreenStatus { is_sharing_screen }) => {
|
|
||||||
let change = state
|
|
||||||
.voice
|
|
||||||
.set_screen_status(room_id, user_id, connection_id, is_sharing_screen)
|
|
||||||
.await;
|
|
||||||
state.voice.emit_change(room_id, user_id, change).await;
|
|
||||||
}
|
|
||||||
Ok(ClientEvent::SetSpeakingStatus { is_speaking }) => {
|
|
||||||
let change = state
|
|
||||||
.voice
|
|
||||||
.set_speaking_status(room_id, user_id, connection_id, is_speaking)
|
|
||||||
.await;
|
|
||||||
state.voice.emit_change(room_id, user_id, change).await;
|
|
||||||
}
|
|
||||||
Ok(ClientEvent::SetMuteStatus { is_muted }) => {
|
|
||||||
let change = state
|
|
||||||
.voice
|
|
||||||
.set_mute_status(room_id, user_id, connection_id, is_muted)
|
|
||||||
.await;
|
|
||||||
state.voice.emit_change(room_id, user_id, change).await;
|
|
||||||
}
|
|
||||||
Ok(ClientEvent::PlaySound { sound_id }) => {
|
|
||||||
if let Ok(Some(sound)) = db::get_sound_by_id(&state.db, sound_id).await
|
|
||||||
&& sound.guild_id == guild_id
|
|
||||||
{
|
|
||||||
state
|
|
||||||
.voice
|
|
||||||
.play_sound(room_id, user_id, sound.media_url)
|
|
||||||
.await;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
let _ = tx.send(ServerEvent::Error {
|
let _ = tx.send(ServerEvent::Error {
|
||||||
message: format!("invalid voice message: {err}"),
|
message: format!("invalid voice message: {err}"),
|
||||||
|
|
@ -488,183 +195,5 @@ pub async fn handle_socket(state: AppState, socket: WebSocket, room_id: Uuid, us
|
||||||
}
|
}
|
||||||
|
|
||||||
send_task.abort();
|
send_task.abort();
|
||||||
let change = state.voice.leave(room_id, user_id, connection_id).await;
|
state.voice.leave(room_id, user_id).await;
|
||||||
state.voice.emit_change(room_id, user_id, change).await;
|
|
||||||
}
|
|
||||||
|
|
||||||
fn aggregate_participant(
|
|
||||||
connections: Option<&HashMap<Uuid, ClientHandle>>,
|
|
||||||
user_id: Uuid,
|
|
||||||
) -> Option<VoiceParticipant> {
|
|
||||||
let connections = connections?;
|
|
||||||
if connections.is_empty() {
|
|
||||||
return None;
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut handles = connections.values();
|
|
||||||
let first = handles.next()?;
|
|
||||||
Some(VoiceParticipant {
|
|
||||||
user_id,
|
|
||||||
display_name: first.display_name.clone(),
|
|
||||||
is_sharing_video: connections.values().any(|handle| handle.is_sharing_video),
|
|
||||||
is_sharing_screen: connections.values().any(|handle| handle.is_sharing_screen),
|
|
||||||
is_speaking: connections.values().any(|handle| handle.is_speaking),
|
|
||||||
is_muted: connections.values().all(|handle| handle.is_muted),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
fn diff_voice_state(
|
|
||||||
previous: Option<VoiceParticipant>,
|
|
||||||
current: Option<VoiceParticipant>,
|
|
||||||
) -> VoiceStateChange {
|
|
||||||
match (previous, current) {
|
|
||||||
(None, None) => VoiceStateChange::default(),
|
|
||||||
(None, Some(current)) => VoiceStateChange {
|
|
||||||
joined: Some(current),
|
|
||||||
..Default::default()
|
|
||||||
},
|
|
||||||
(Some(previous), None) => VoiceStateChange {
|
|
||||||
left_user_id: Some(previous.user_id),
|
|
||||||
..Default::default()
|
|
||||||
},
|
|
||||||
(Some(previous), Some(current)) => VoiceStateChange {
|
|
||||||
video_changed: (previous.is_sharing_video != current.is_sharing_video)
|
|
||||||
.then_some((current.user_id, current.is_sharing_video)),
|
|
||||||
screen_changed: (previous.is_sharing_screen != current.is_sharing_screen)
|
|
||||||
.then_some((current.user_id, current.is_sharing_screen)),
|
|
||||||
speaking_changed: (previous.is_speaking != current.is_speaking)
|
|
||||||
.then_some((current.user_id, current.is_speaking)),
|
|
||||||
mute_changed: (previous.is_muted != current.is_muted)
|
|
||||||
.then_some((current.user_id, current.is_muted)),
|
|
||||||
..Default::default()
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn broadcast_voice_event(
|
|
||||||
room: &HashMap<Uuid, HashMap<Uuid, ClientHandle>>,
|
|
||||||
source_user_id: Uuid,
|
|
||||||
event: ServerEvent,
|
|
||||||
) {
|
|
||||||
for (peer_id, connections) in room {
|
|
||||||
if *peer_id == source_user_id {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
for peer in connections.values() {
|
|
||||||
let _ = peer.tx.send(event.clone());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn is_voice_state_change_empty(change: &VoiceStateChange) -> bool {
|
|
||||||
change.joined.is_none()
|
|
||||||
&& change.left_user_id.is_none()
|
|
||||||
&& change.video_changed.is_none()
|
|
||||||
&& change.screen_changed.is_none()
|
|
||||||
&& change.speaking_changed.is_none()
|
|
||||||
&& change.mute_changed.is_none()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::{VoiceHub, aggregate_participant};
|
|
||||||
use std::collections::HashMap;
|
|
||||||
use tokio::sync::mpsc;
|
|
||||||
use uuid::Uuid;
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn multiple_connections_keep_voice_participant_until_last_leave() {
|
|
||||||
let hub = VoiceHub::default();
|
|
||||||
let room_id = Uuid::new_v4();
|
|
||||||
let user_id = Uuid::new_v4();
|
|
||||||
let first_connection = Uuid::new_v4();
|
|
||||||
let second_connection = Uuid::new_v4();
|
|
||||||
let (tx1, _rx1) = mpsc::unbounded_channel();
|
|
||||||
let (tx2, _rx2) = mpsc::unbounded_channel();
|
|
||||||
|
|
||||||
let (_peers, first_join) = hub
|
|
||||||
.join(room_id, user_id, first_connection, "User".to_string(), tx1)
|
|
||||||
.await;
|
|
||||||
let (_peers, second_join) = hub
|
|
||||||
.join(room_id, user_id, second_connection, "User".to_string(), tx2)
|
|
||||||
.await;
|
|
||||||
let first_leave = hub.leave(room_id, user_id, first_connection).await;
|
|
||||||
let second_leave = hub.leave(room_id, user_id, second_connection).await;
|
|
||||||
|
|
||||||
assert!(first_join.joined.is_some());
|
|
||||||
assert!(second_join.joined.is_none());
|
|
||||||
assert!(first_leave.left_user_id.is_none());
|
|
||||||
assert_eq!(second_leave.left_user_id, Some(user_id));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn voice_mute_state_only_flips_when_all_connections_are_muted() {
|
|
||||||
let hub = VoiceHub::default();
|
|
||||||
let room_id = Uuid::new_v4();
|
|
||||||
let user_id = Uuid::new_v4();
|
|
||||||
let first_connection = Uuid::new_v4();
|
|
||||||
let second_connection = Uuid::new_v4();
|
|
||||||
let (tx1, _rx1) = mpsc::unbounded_channel();
|
|
||||||
let (tx2, _rx2) = mpsc::unbounded_channel();
|
|
||||||
|
|
||||||
let _ = hub
|
|
||||||
.join(room_id, user_id, first_connection, "User".to_string(), tx1)
|
|
||||||
.await;
|
|
||||||
let _ = hub
|
|
||||||
.join(room_id, user_id, second_connection, "User".to_string(), tx2)
|
|
||||||
.await;
|
|
||||||
|
|
||||||
let first_mute = hub
|
|
||||||
.set_mute_status(room_id, user_id, first_connection, true)
|
|
||||||
.await;
|
|
||||||
let second_mute = hub
|
|
||||||
.set_mute_status(room_id, user_id, second_connection, true)
|
|
||||||
.await;
|
|
||||||
let unmute = hub
|
|
||||||
.set_mute_status(room_id, user_id, first_connection, false)
|
|
||||||
.await;
|
|
||||||
|
|
||||||
assert!(first_mute.mute_changed.is_none());
|
|
||||||
assert_eq!(second_mute.mute_changed, Some((user_id, true)));
|
|
||||||
assert_eq!(unmute.mute_changed, Some((user_id, false)));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn aggregate_participant_combines_connection_state() {
|
|
||||||
let user_id = Uuid::new_v4();
|
|
||||||
let first_connection = Uuid::new_v4();
|
|
||||||
let second_connection = Uuid::new_v4();
|
|
||||||
let mut connections = HashMap::new();
|
|
||||||
let (tx1, _rx1) = mpsc::unbounded_channel();
|
|
||||||
let (tx2, _rx2) = mpsc::unbounded_channel();
|
|
||||||
|
|
||||||
connections.insert(
|
|
||||||
first_connection,
|
|
||||||
super::ClientHandle {
|
|
||||||
display_name: "User".to_string(),
|
|
||||||
is_sharing_video: true,
|
|
||||||
is_sharing_screen: false,
|
|
||||||
is_speaking: false,
|
|
||||||
is_muted: true,
|
|
||||||
tx: tx1,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
connections.insert(
|
|
||||||
second_connection,
|
|
||||||
super::ClientHandle {
|
|
||||||
display_name: "User".to_string(),
|
|
||||||
is_sharing_video: false,
|
|
||||||
is_sharing_screen: true,
|
|
||||||
is_speaking: true,
|
|
||||||
is_muted: false,
|
|
||||||
tx: tx2,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
|
|
||||||
let participant = aggregate_participant(Some(&connections), user_id).unwrap();
|
|
||||||
assert!(participant.is_sharing_video);
|
|
||||||
assert!(participant.is_sharing_screen);
|
|
||||||
assert!(participant.is_speaking);
|
|
||||||
assert!(!participant.is_muted);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
979
static/app.js
979
static/app.js
|
|
@ -1 +1,978 @@
|
||||||
import "./shared/app-core.js";
|
const state = {
|
||||||
|
me: null,
|
||||||
|
guilds: [],
|
||||||
|
channels: [],
|
||||||
|
dmConversations: [],
|
||||||
|
members: [],
|
||||||
|
voicePresence: new Map(),
|
||||||
|
selectedGuildId: null,
|
||||||
|
selectedTextChannelId: null,
|
||||||
|
selectedDmUserId: null,
|
||||||
|
selectedDmDisplayName: null,
|
||||||
|
selectedVoiceChannelId: null,
|
||||||
|
voice: {
|
||||||
|
ws: null,
|
||||||
|
joinedChannelId: null,
|
||||||
|
localStream: null,
|
||||||
|
rawStream: null,
|
||||||
|
audioContext: null,
|
||||||
|
peerConnections: new Map(),
|
||||||
|
muted: false,
|
||||||
|
iceServers: [{ urls: "stun:stun.l.google.com:19302" }],
|
||||||
|
},
|
||||||
|
voicePresencePollId: null,
|
||||||
|
chatWs: null,
|
||||||
|
lastMessageId: null,
|
||||||
|
onlineUsers: new Set(),
|
||||||
|
};
|
||||||
|
|
||||||
|
const el = {
|
||||||
|
authScreen: document.getElementById("auth-screen"),
|
||||||
|
loginBtn: document.getElementById("login-btn"),
|
||||||
|
main: document.getElementById("main"),
|
||||||
|
status: document.getElementById("status"),
|
||||||
|
|
||||||
|
// Guilds
|
||||||
|
guildList: document.getElementById("guild-list"),
|
||||||
|
addGuildBtn: document.getElementById("add-guild-btn"),
|
||||||
|
guildTitle: document.getElementById("guild-title"),
|
||||||
|
createInviteBtn: document.getElementById("create-invite-btn"),
|
||||||
|
|
||||||
|
// Channels
|
||||||
|
channelList: document.getElementById("channel-list"),
|
||||||
|
voiceChannelList: document.getElementById("voice-channel-list"),
|
||||||
|
addTextBtn: document.getElementById("add-text-btn"),
|
||||||
|
addVoiceBtn: document.getElementById("add-voice-btn"),
|
||||||
|
dmList: document.getElementById("dm-list"),
|
||||||
|
channelTitle: document.getElementById("channel-title"),
|
||||||
|
|
||||||
|
// Messages
|
||||||
|
messageList: document.getElementById("message-list"),
|
||||||
|
messageForm: document.getElementById("message-form"),
|
||||||
|
messageBody: document.getElementById("message-body"),
|
||||||
|
|
||||||
|
// User Panel
|
||||||
|
userName: document.getElementById("user-name"),
|
||||||
|
userAvatar: document.getElementById("user-avatar"),
|
||||||
|
logoutBtn: document.getElementById("logout-btn"),
|
||||||
|
|
||||||
|
// Voice Connection
|
||||||
|
voiceConnection: document.getElementById("voice-connection"),
|
||||||
|
vcChannelName: document.getElementById("vc-channel-name"),
|
||||||
|
voiceMuteBtn: document.getElementById("voice-mute-btn"),
|
||||||
|
voiceLeaveBtn: document.getElementById("voice-leave-btn"),
|
||||||
|
|
||||||
|
// Members
|
||||||
|
memberList: document.getElementById("member-list"),
|
||||||
|
|
||||||
|
// Modals
|
||||||
|
modalContainer: document.getElementById("modal-container"),
|
||||||
|
guildForm: document.getElementById("guild-form"),
|
||||||
|
guildName: document.getElementById("guild-name"),
|
||||||
|
modalCancel: document.getElementById("modal-cancel"),
|
||||||
|
|
||||||
|
channelModal: document.getElementById("channel-modal"),
|
||||||
|
channelForm: document.getElementById("channel-form"),
|
||||||
|
channelName: document.getElementById("channel-name"),
|
||||||
|
channelModalCancel: document.getElementById("channel-modal-cancel"),
|
||||||
|
|
||||||
|
// Mobile
|
||||||
|
mobileMenuBtn: document.getElementById("mobile-menu-btn"),
|
||||||
|
mobileMembersBtn: document.getElementById("mobile-members-btn"),
|
||||||
|
overlay: document.getElementById("overlay"),
|
||||||
|
utilitySidebar: document.getElementById("utility-sidebar"),
|
||||||
|
};
|
||||||
|
|
||||||
|
// --- API Helpers ---
|
||||||
|
|
||||||
|
async function api(path, options = {}) {
|
||||||
|
const res = await fetch(path, {
|
||||||
|
...options,
|
||||||
|
headers: {
|
||||||
|
"content-type": "application/json",
|
||||||
|
...(options.headers || {}),
|
||||||
|
},
|
||||||
|
credentials: "same-origin",
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!res.ok) {
|
||||||
|
let detail = "request failed";
|
||||||
|
try {
|
||||||
|
const body = await res.json();
|
||||||
|
detail = body.error || detail;
|
||||||
|
} catch { }
|
||||||
|
throw new Error(`${res.status}: ${detail}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (res.status === 204) return null;
|
||||||
|
return res.json();
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- Utils ---
|
||||||
|
|
||||||
|
function shortName(name) {
|
||||||
|
if (!name || typeof name !== 'string') return "?";
|
||||||
|
const trimmed = name.trim();
|
||||||
|
if (!trimmed) return "?";
|
||||||
|
const words = trimmed.split(/\s+/).filter(w => w.length > 0).slice(0, 2);
|
||||||
|
if (words.length === 0) return "?";
|
||||||
|
if (words.length === 1) return words[0].substring(0, 2).toUpperCase();
|
||||||
|
return words.map((w) => w[0]?.toUpperCase() || "").join("");
|
||||||
|
}
|
||||||
|
|
||||||
|
function escapeHtml(s) {
|
||||||
|
if (s === null || s === undefined) return "";
|
||||||
|
return String(s)
|
||||||
|
.replaceAll("&", "&")
|
||||||
|
.replaceAll("<", "<")
|
||||||
|
.replaceAll(">", ">")
|
||||||
|
.replaceAll('"', """)
|
||||||
|
.replaceAll("'", "'");
|
||||||
|
}
|
||||||
|
|
||||||
|
function formatDate(isoString) {
|
||||||
|
const d = new Date(isoString);
|
||||||
|
const now = new Date();
|
||||||
|
const isToday = d.toDateString() === now.toDateString();
|
||||||
|
|
||||||
|
if (isToday) {
|
||||||
|
return `Today at ${d.toLocaleTimeString([], { hour: '2-digit', minute: '2-digit' })}`;
|
||||||
|
}
|
||||||
|
return d.toLocaleDateString();
|
||||||
|
}
|
||||||
|
|
||||||
|
let audioCtx = null;
|
||||||
|
|
||||||
|
// Global interaction listener to unlock AudioContext (autoplay policy)
|
||||||
|
window.addEventListener('click', () => {
|
||||||
|
if (audioCtx && audioCtx.state === 'suspended') {
|
||||||
|
audioCtx.resume();
|
||||||
|
}
|
||||||
|
}, { once: true });
|
||||||
|
|
||||||
|
function playSound(type) {
|
||||||
|
try {
|
||||||
|
if (!audioCtx) audioCtx = new (window.AudioContext || window.webkitAudioContext)();
|
||||||
|
if (audioCtx.state === 'suspended') audioCtx.resume();
|
||||||
|
|
||||||
|
const sounds = {
|
||||||
|
message: { freq: 880, type: 'sine', duration: 0.1, volume: 0.1 },
|
||||||
|
dm: { freq: [660, 880], type: 'sine', duration: 0.15, volume: 0.1 },
|
||||||
|
join: { freq: [440, 880], type: 'sine', duration: 0.2, volume: 0.1 },
|
||||||
|
leave: { freq: [880, 440], type: 'sine', duration: 0.2, volume: 0.1 },
|
||||||
|
'peer-join': { freq: [660, 990], type: 'sine', duration: 0.15, volume: 0.05 },
|
||||||
|
'peer-leave': { freq: [990, 660], type: 'sine', duration: 0.15, volume: 0.05 },
|
||||||
|
};
|
||||||
|
|
||||||
|
const s = sounds[type];
|
||||||
|
if (!s) return;
|
||||||
|
|
||||||
|
const osc = audioCtx.createOscillator();
|
||||||
|
const gain = audioCtx.createGain();
|
||||||
|
|
||||||
|
osc.type = s.type;
|
||||||
|
if (Array.isArray(s.freq)) {
|
||||||
|
osc.frequency.setValueAtTime(s.freq[0], audioCtx.currentTime);
|
||||||
|
osc.frequency.exponentialRampToValueAtTime(s.freq[1], audioCtx.currentTime + s.duration);
|
||||||
|
} else {
|
||||||
|
osc.frequency.setValueAtTime(s.freq, audioCtx.currentTime);
|
||||||
|
}
|
||||||
|
|
||||||
|
gain.gain.setValueAtTime(s.volume, audioCtx.currentTime);
|
||||||
|
gain.gain.exponentialRampToValueAtTime(0.01, audioCtx.currentTime + s.duration);
|
||||||
|
|
||||||
|
osc.connect(gain);
|
||||||
|
gain.connect(audioCtx.destination);
|
||||||
|
|
||||||
|
osc.start();
|
||||||
|
osc.stop(audioCtx.currentTime + s.duration);
|
||||||
|
} catch (err) {
|
||||||
|
console.warn("playSound failed", err);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const LAST_GUILD_STORAGE_KEY = "chattz:lastGuildId";
|
||||||
|
|
||||||
|
// --- Renderers ---
|
||||||
|
|
||||||
|
function renderGuilds() {
|
||||||
|
el.guildList.innerHTML = "";
|
||||||
|
|
||||||
|
for (const guild of state.guilds) {
|
||||||
|
const btn = document.createElement("button");
|
||||||
|
btn.className = `guild-pill ${state.selectedGuildId === guild.id ? "active" : ""}`;
|
||||||
|
btn.title = guild.name;
|
||||||
|
btn.textContent = shortName(guild.name);
|
||||||
|
btn.onclick = async () => {
|
||||||
|
state.selectedGuildId = guild.id;
|
||||||
|
state.selectedTextChannelId = null;
|
||||||
|
state.selectedVoiceChannelId = null;
|
||||||
|
state.selectedDmUserId = null;
|
||||||
|
state.selectedDmDisplayName = null;
|
||||||
|
localStorage.setItem(LAST_GUILD_STORAGE_KEY, guild.id);
|
||||||
|
renderGuilds();
|
||||||
|
renderDMs();
|
||||||
|
await loadChannels();
|
||||||
|
await loadGuildMembers();
|
||||||
|
await refreshVoicePresence();
|
||||||
|
startVoicePresencePolling();
|
||||||
|
renderMessages([]);
|
||||||
|
updateHeaderLabels();
|
||||||
|
};
|
||||||
|
el.guildList.appendChild(btn);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function renderChannels() {
|
||||||
|
el.channelList.innerHTML = "";
|
||||||
|
el.voiceChannelList.innerHTML = "";
|
||||||
|
|
||||||
|
for (const channel of state.channels) {
|
||||||
|
const row = document.createElement("button");
|
||||||
|
row.className = `channel-row ${(channel.kind === 'text' && state.selectedTextChannelId === channel.id) ||
|
||||||
|
(channel.kind === 'voice' && state.selectedVoiceChannelId === channel.id) ? "active" : ""
|
||||||
|
}`;
|
||||||
|
|
||||||
|
const iconName = channel.kind === 'text' ? 'hash' : 'volume-2';
|
||||||
|
row.innerHTML = `<i data-lucide="${iconName}"></i> <span>${escapeHtml(channel.name)}</span>`;
|
||||||
|
|
||||||
|
row.onclick = async () => {
|
||||||
|
if (channel.kind === 'text') {
|
||||||
|
state.selectedDmUserId = null;
|
||||||
|
state.selectedDmDisplayName = null;
|
||||||
|
state.selectedTextChannelId = channel.id;
|
||||||
|
renderChannels();
|
||||||
|
renderDMs();
|
||||||
|
updateHeaderLabels();
|
||||||
|
const messages = await api(`/channels/${channel.id}/messages?limit=100`);
|
||||||
|
renderMessages(messages);
|
||||||
|
} else {
|
||||||
|
state.selectedDmUserId = null;
|
||||||
|
state.selectedDmDisplayName = null;
|
||||||
|
state.selectedVoiceChannelId = channel.id;
|
||||||
|
renderChannels();
|
||||||
|
renderDMs();
|
||||||
|
joinVoice();
|
||||||
|
}
|
||||||
|
if (window.innerWidth <= 768) closeMobileMenus();
|
||||||
|
};
|
||||||
|
|
||||||
|
if (channel.kind === 'text') {
|
||||||
|
el.channelList.appendChild(row);
|
||||||
|
} else {
|
||||||
|
el.voiceChannelList.appendChild(row);
|
||||||
|
|
||||||
|
const participants = state.voicePresence.get(channel.id) || [];
|
||||||
|
if (participants.length > 0) {
|
||||||
|
const pList = document.createElement("div");
|
||||||
|
pList.className = "voice-row-members channel-list";
|
||||||
|
pList.style.paddingLeft = "24px";
|
||||||
|
for (const p of participants) {
|
||||||
|
const pRow = document.createElement("div");
|
||||||
|
pRow.className = "channel-row";
|
||||||
|
pRow.style.padding = "2px 8px";
|
||||||
|
pRow.innerHTML = `<div class="avatar" style="width:20px;height:20px;font-size:10px">${shortName(p.display_name)}</div> <span>${escapeHtml(p.display_name)}</span>`;
|
||||||
|
pList.appendChild(pRow);
|
||||||
|
}
|
||||||
|
el.voiceChannelList.appendChild(pList);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
lucide.createIcons();
|
||||||
|
}
|
||||||
|
|
||||||
|
function renderDMs() {
|
||||||
|
el.dmList.innerHTML = "";
|
||||||
|
for (const dm of state.dmConversations) {
|
||||||
|
const isOnline = state.onlineUsers.has(dm.user_id);
|
||||||
|
const row = document.createElement("button");
|
||||||
|
row.className = `channel-row ${state.selectedDmUserId === dm.user_id ? "active" : ""}`;
|
||||||
|
row.innerHTML = `
|
||||||
|
<div class="avatar-wrapper">
|
||||||
|
<i data-lucide="message-circle"></i>
|
||||||
|
<div class="status-dot ${isOnline ? "online" : ""}"></div>
|
||||||
|
</div>
|
||||||
|
<span>${escapeHtml(dm.display_name)}</span>
|
||||||
|
`;
|
||||||
|
row.onclick = async () => {
|
||||||
|
state.selectedDmUserId = dm.user_id;
|
||||||
|
state.selectedDmDisplayName = dm.display_name;
|
||||||
|
state.selectedTextChannelId = null;
|
||||||
|
state.selectedVoiceChannelId = null;
|
||||||
|
renderChannels();
|
||||||
|
renderDMs();
|
||||||
|
updateHeaderLabels();
|
||||||
|
const messages = await api(`/dms/${dm.user_id}/messages?limit=100`);
|
||||||
|
renderMessages(messages);
|
||||||
|
};
|
||||||
|
el.dmList.appendChild(row);
|
||||||
|
}
|
||||||
|
|
||||||
|
lucide.createIcons();
|
||||||
|
}
|
||||||
|
|
||||||
|
function renderMessages(messages) {
|
||||||
|
el.messageList.innerHTML = "";
|
||||||
|
let lastAuthorId = null;
|
||||||
|
let lastTime = null;
|
||||||
|
|
||||||
|
for (const m of messages.slice().reverse()) {
|
||||||
|
const row = document.createElement("div");
|
||||||
|
const mDate = new Date(m.created_at);
|
||||||
|
const isGrouped = lastAuthorId === m.author_user_id &&
|
||||||
|
lastTime && (mDate - lastTime < 300000); // 5 minutes
|
||||||
|
|
||||||
|
row.className = `msg ${isGrouped ? "msg-grouped" : ""}`;
|
||||||
|
|
||||||
|
const displayName = m.author_display_name || "Unknown User";
|
||||||
|
|
||||||
|
if (isGrouped) {
|
||||||
|
row.innerHTML = `<div class="msg-content"><div class="msg-body">${escapeHtml(m.body)}</div></div>`;
|
||||||
|
} else {
|
||||||
|
row.innerHTML = `
|
||||||
|
<div class="msg-avatar">${shortName(displayName)}</div>
|
||||||
|
<div class="msg-content">
|
||||||
|
<div class="msg-header">
|
||||||
|
<span class="msg-author">${escapeHtml(displayName)}</span>
|
||||||
|
<span class="msg-time">${formatDate(m.created_at)}</span>
|
||||||
|
</div>
|
||||||
|
<div class="msg-body">${escapeHtml(m.body)}</div>
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
el.messageList.appendChild(row);
|
||||||
|
lastAuthorId = m.author_user_id;
|
||||||
|
lastTime = mDate;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (messages.length > 0) {
|
||||||
|
state.lastMessageId = messages[0].id;
|
||||||
|
}
|
||||||
|
|
||||||
|
el.messageList.scrollTop = el.messageList.scrollHeight;
|
||||||
|
}
|
||||||
|
|
||||||
|
function renderMembers() {
|
||||||
|
el.memberList.innerHTML = "";
|
||||||
|
for (const m of state.members) {
|
||||||
|
const isOnline = state.onlineUsers.has(m.id);
|
||||||
|
const row = document.createElement("div");
|
||||||
|
row.className = "member-row";
|
||||||
|
row.innerHTML = `
|
||||||
|
<div class="member-avatar">
|
||||||
|
${shortName(m.display_name)}
|
||||||
|
<div class="status-dot ${isOnline ? "online" : ""}"></div>
|
||||||
|
</div>
|
||||||
|
<div class="member-name">${escapeHtml(m.display_name)}</div>
|
||||||
|
`;
|
||||||
|
row.style.cursor = "pointer";
|
||||||
|
row.onclick = async () => {
|
||||||
|
if (state.me && m.id === state.me.id) return;
|
||||||
|
state.selectedDmUserId = m.id;
|
||||||
|
state.selectedDmDisplayName = m.display_name;
|
||||||
|
state.selectedTextChannelId = null;
|
||||||
|
state.selectedVoiceChannelId = null;
|
||||||
|
renderChannels();
|
||||||
|
renderDMs();
|
||||||
|
updateHeaderLabels();
|
||||||
|
const messages = await api(`/dms/${m.id}/messages?limit=100`);
|
||||||
|
renderMessages(messages);
|
||||||
|
};
|
||||||
|
el.memberList.appendChild(row);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function updateHeaderLabels() {
|
||||||
|
const guild = state.guilds.find((g) => g.id === state.selectedGuildId);
|
||||||
|
el.guildTitle.textContent = guild ? guild.name : "No server selected";
|
||||||
|
|
||||||
|
if (state.selectedDmUserId) {
|
||||||
|
const dmFromConversations = state.dmConversations.find((u) => u.user_id === state.selectedDmUserId);
|
||||||
|
const dmFromMembers = state.members.find((m) => m.id === state.selectedDmUserId);
|
||||||
|
const displayName = dmFromConversations?.display_name || dmFromMembers?.display_name || state.selectedDmDisplayName;
|
||||||
|
const dmName = displayName ? `@${displayName}` : "Direct Message";
|
||||||
|
el.channelTitle.textContent = dmName;
|
||||||
|
el.messageBody.placeholder = displayName
|
||||||
|
? `Message @${displayName}`
|
||||||
|
: "Message";
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const channel = state.channels.find((c) => c.id === state.selectedTextChannelId);
|
||||||
|
el.channelTitle.textContent = channel ? channel.name : "Select a channel";
|
||||||
|
el.messageBody.placeholder = channel ? `Message #${channel.name}` : "Select a channel";
|
||||||
|
}
|
||||||
|
|
||||||
|
async function createInviteLink() {
|
||||||
|
if (!state.selectedGuildId) {
|
||||||
|
alert("Select a server first.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
const invite = await api(`/guilds/${state.selectedGuildId}/invites`, {
|
||||||
|
method: "POST",
|
||||||
|
body: JSON.stringify({ max_uses: 50, expires_in_hours: 24 }),
|
||||||
|
});
|
||||||
|
const link = `${location.origin}/?invite=${encodeURIComponent(invite.code)}`;
|
||||||
|
try {
|
||||||
|
await navigator.clipboard.writeText(link);
|
||||||
|
alert(`Invite link copied:\n${link}`);
|
||||||
|
} catch {
|
||||||
|
prompt("Copy invite link:", link);
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
alert(err.message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- Logic ---
|
||||||
|
|
||||||
|
async function loadGuilds() {
|
||||||
|
state.guilds = await api("/guilds");
|
||||||
|
renderGuilds();
|
||||||
|
}
|
||||||
|
|
||||||
|
async function loadDMConversations() {
|
||||||
|
state.dmConversations = await api("/dms");
|
||||||
|
renderDMs();
|
||||||
|
}
|
||||||
|
|
||||||
|
async function loadGuildMembers() {
|
||||||
|
if (!state.selectedGuildId) {
|
||||||
|
state.members = [];
|
||||||
|
renderMembers();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
state.members = await api(`/guilds/${state.selectedGuildId}/members`);
|
||||||
|
renderMembers();
|
||||||
|
}
|
||||||
|
|
||||||
|
async function loadChannels() {
|
||||||
|
if (!state.selectedGuildId) {
|
||||||
|
state.channels = [];
|
||||||
|
renderChannels();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
state.channels = await api(`/guilds/${state.selectedGuildId}/channels`);
|
||||||
|
renderChannels();
|
||||||
|
}
|
||||||
|
|
||||||
|
async function refreshVoicePresence() {
|
||||||
|
if (!state.selectedGuildId) {
|
||||||
|
state.voicePresence.clear();
|
||||||
|
renderChannels();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const res = await api(`/guilds/${state.selectedGuildId}/voice-presence`);
|
||||||
|
state.voicePresence.clear();
|
||||||
|
for (const entry of res.channels || []) {
|
||||||
|
state.voicePresence.set(entry.channel_id, entry.participants || []);
|
||||||
|
}
|
||||||
|
renderChannels();
|
||||||
|
} catch (err) {
|
||||||
|
console.warn("voice presence failed", err);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function startVoicePresencePolling() {
|
||||||
|
if (state.voicePresencePollId) clearInterval(state.voicePresencePollId);
|
||||||
|
state.voicePresencePollId = setInterval(() => {
|
||||||
|
refreshVoicePresence().catch(() => { });
|
||||||
|
}, 3000);
|
||||||
|
}
|
||||||
|
|
||||||
|
function initChatWs() {
|
||||||
|
if (state.chatWs) state.chatWs.close();
|
||||||
|
const protocol = location.protocol === "https:" ? "wss:" : "ws:";
|
||||||
|
const wsUrl = `${protocol}//${location.host}/ws`;
|
||||||
|
const ws = new WebSocket(wsUrl);
|
||||||
|
state.chatWs = ws;
|
||||||
|
|
||||||
|
ws.onmessage = (event) => {
|
||||||
|
const msg = JSON.parse(event.data);
|
||||||
|
const authorId = msg.message?.author_user_id;
|
||||||
|
const isFromMe = authorId && state.me && authorId === state.me.id;
|
||||||
|
|
||||||
|
if (msg.type === "message_created") {
|
||||||
|
if (!isFromMe) playSound('message');
|
||||||
|
if (state.selectedTextChannelId === msg.channel_id) {
|
||||||
|
api(`/channels/${state.selectedTextChannelId}/messages?limit=100`).then(renderMessages);
|
||||||
|
}
|
||||||
|
} else if (msg.type === "dm_created") {
|
||||||
|
if (!isFromMe) playSound('dm');
|
||||||
|
if (state.selectedDmUserId === msg.other_user_id) {
|
||||||
|
api(`/dms/${state.selectedDmUserId}/messages?limit=100`).then(renderMessages);
|
||||||
|
playSound('dm');
|
||||||
|
loadDMConversations().then(renderDMs);
|
||||||
|
} else {
|
||||||
|
loadDMConversations().then(renderDMs);
|
||||||
|
}
|
||||||
|
} else if (msg.type === "user_presence") {
|
||||||
|
if (msg.online) {
|
||||||
|
state.onlineUsers.add(msg.user_id);
|
||||||
|
} else {
|
||||||
|
state.onlineUsers.delete(msg.user_id);
|
||||||
|
}
|
||||||
|
renderMembers();
|
||||||
|
renderDMs();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
ws.onclose = () => {
|
||||||
|
console.log("Chat WS closed, reconnecting...");
|
||||||
|
setTimeout(initChatWs, 3000);
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- Voice ---
|
||||||
|
|
||||||
|
function getVoiceWsUrl(channelId) {
|
||||||
|
const proto = location.protocol === "https:" ? "wss" : "ws";
|
||||||
|
return `${proto}://${location.host}/channels/${channelId}/voice/ws`;
|
||||||
|
}
|
||||||
|
|
||||||
|
function shouldInitiateOffer(peerId) {
|
||||||
|
if (!state.me || !state.me.id) return false;
|
||||||
|
return state.me.id > peerId;
|
||||||
|
}
|
||||||
|
|
||||||
|
function stopAndClearAudioPipeline() {
|
||||||
|
if (state.voice.localStream) {
|
||||||
|
for (const track of state.voice.localStream.getTracks()) track.stop();
|
||||||
|
}
|
||||||
|
if (state.voice.rawStream && state.voice.rawStream !== state.voice.localStream) {
|
||||||
|
for (const track of state.voice.rawStream.getTracks()) track.stop();
|
||||||
|
}
|
||||||
|
if (state.voice.audioContext) {
|
||||||
|
state.voice.audioContext.close().catch(() => { });
|
||||||
|
}
|
||||||
|
state.voice.localStream = null;
|
||||||
|
state.voice.rawStream = null;
|
||||||
|
state.voice.audioContext = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
async function buildAudioPipeline(rawStream) {
|
||||||
|
const audioContext = new AudioContext();
|
||||||
|
const source = audioContext.createMediaStreamSource(rawStream);
|
||||||
|
const destination = audioContext.createMediaStreamDestination();
|
||||||
|
state.voice.audioContext = audioContext;
|
||||||
|
|
||||||
|
let head = source;
|
||||||
|
|
||||||
|
head.connect(destination);
|
||||||
|
|
||||||
|
return destination.stream;
|
||||||
|
}
|
||||||
|
|
||||||
|
async function createLocalVoiceStream() {
|
||||||
|
const constraints = {
|
||||||
|
audio: {
|
||||||
|
channelCount: 1,
|
||||||
|
sampleRate: 48000,
|
||||||
|
echoCancellation: true,
|
||||||
|
noiseSuppression: true,
|
||||||
|
autoGainControl: false,
|
||||||
|
},
|
||||||
|
video: false,
|
||||||
|
};
|
||||||
|
const rawStream = await navigator.mediaDevices.getUserMedia(constraints);
|
||||||
|
const localStream = await buildAudioPipeline(rawStream);
|
||||||
|
state.voice.rawStream = rawStream;
|
||||||
|
state.voice.localStream = localStream;
|
||||||
|
if (state.voice.muted) {
|
||||||
|
state.voice.localStream.getAudioTracks().forEach((t) => {
|
||||||
|
t.enabled = false;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function ensurePeerConnection(peerId) {
|
||||||
|
if (state.voice.peerConnections.has(peerId)) {
|
||||||
|
return state.voice.peerConnections.get(peerId);
|
||||||
|
}
|
||||||
|
|
||||||
|
const pc = new RTCPeerConnection({ iceServers: state.voice.iceServers });
|
||||||
|
|
||||||
|
if (state.voice.localStream) {
|
||||||
|
for (const track of state.voice.localStream.getTracks()) {
|
||||||
|
pc.addTrack(track, state.voice.localStream);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pc.onicecandidate = (event) => {
|
||||||
|
if (!event.candidate || !state.voice.ws) return;
|
||||||
|
state.voice.ws.send(JSON.stringify({
|
||||||
|
type: "signal",
|
||||||
|
to_user_id: peerId,
|
||||||
|
kind: "ice",
|
||||||
|
data: event.candidate,
|
||||||
|
}));
|
||||||
|
};
|
||||||
|
|
||||||
|
pc.ontrack = (event) => {
|
||||||
|
let audio = document.getElementById(`audio-${peerId}`);
|
||||||
|
if (!audio) {
|
||||||
|
audio = document.createElement("audio");
|
||||||
|
audio.id = `audio-${peerId}`;
|
||||||
|
audio.autoplay = true;
|
||||||
|
audio.playsInline = true;
|
||||||
|
document.body.appendChild(audio);
|
||||||
|
}
|
||||||
|
audio.srcObject = event.streams[0];
|
||||||
|
};
|
||||||
|
|
||||||
|
state.voice.peerConnections.set(peerId, pc);
|
||||||
|
return pc;
|
||||||
|
}
|
||||||
|
|
||||||
|
async function sendOffer(peerId) {
|
||||||
|
const pc = ensurePeerConnection(peerId);
|
||||||
|
const offer = await pc.createOffer();
|
||||||
|
await pc.setLocalDescription(offer);
|
||||||
|
state.voice.ws.send(JSON.stringify({
|
||||||
|
type: "signal",
|
||||||
|
to_user_id: peerId,
|
||||||
|
kind: "offer",
|
||||||
|
data: offer,
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleSignal(fromPeerId, kind, data) {
|
||||||
|
const pc = ensurePeerConnection(fromPeerId);
|
||||||
|
|
||||||
|
if (kind === "offer") {
|
||||||
|
await pc.setRemoteDescription(new RTCSessionDescription(data));
|
||||||
|
const answer = await pc.createAnswer();
|
||||||
|
await pc.setLocalDescription(answer);
|
||||||
|
state.voice.ws.send(JSON.stringify({
|
||||||
|
type: "signal",
|
||||||
|
to_user_id: fromPeerId,
|
||||||
|
kind: "answer",
|
||||||
|
data: answer,
|
||||||
|
}));
|
||||||
|
} else if (kind === "answer") {
|
||||||
|
await pc.setRemoteDescription(new RTCSessionDescription(data));
|
||||||
|
} else if (kind === "ice") {
|
||||||
|
try {
|
||||||
|
await pc.addIceCandidate(data ? new RTCIceCandidate(data) : null);
|
||||||
|
} catch (err) {
|
||||||
|
console.warn("failed to add ice candidate", err);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function joinVoice() {
|
||||||
|
if (!state.selectedVoiceChannelId) return;
|
||||||
|
if (state.voice.joinedChannelId === state.selectedVoiceChannelId) return;
|
||||||
|
|
||||||
|
await leaveVoice();
|
||||||
|
|
||||||
|
try {
|
||||||
|
await createLocalVoiceStream();
|
||||||
|
} catch (err) {
|
||||||
|
console.error("microphone denied", err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const ws = new WebSocket(getVoiceWsUrl(state.selectedVoiceChannelId));
|
||||||
|
state.voice.ws = ws;
|
||||||
|
state.voice.joinedChannelId = state.selectedVoiceChannelId;
|
||||||
|
|
||||||
|
ws.onopen = () => {
|
||||||
|
const channel = state.channels.find(c => c.id === state.selectedVoiceChannelId);
|
||||||
|
el.vcChannelName.textContent = channel ? channel.name : "Voice";
|
||||||
|
el.voiceConnection.classList.remove("hidden");
|
||||||
|
playSound('join');
|
||||||
|
refreshVoicePresence().catch(() => { });
|
||||||
|
};
|
||||||
|
|
||||||
|
ws.onmessage = async (event) => {
|
||||||
|
const msg = JSON.parse(event.data);
|
||||||
|
if (msg.type === "peers") {
|
||||||
|
for (const peer of msg.peers) {
|
||||||
|
if (shouldInitiateOffer(peer.user_id)) await sendOffer(peer.user_id);
|
||||||
|
}
|
||||||
|
} else if (msg.type === "peer_joined") {
|
||||||
|
playSound('peer-join');
|
||||||
|
if (shouldInitiateOffer(msg.user_id)) await sendOffer(msg.user_id);
|
||||||
|
} else if (msg.type === "peer_left") {
|
||||||
|
playSound('peer-leave');
|
||||||
|
const pc = state.voice.peerConnections.get(msg.user_id);
|
||||||
|
if (pc) {
|
||||||
|
pc.close();
|
||||||
|
state.voice.peerConnections.delete(msg.user_id);
|
||||||
|
}
|
||||||
|
document.getElementById(`audio-${msg.user_id}`)?.remove();
|
||||||
|
} else if (msg.type === "signal") {
|
||||||
|
await handleSignal(msg.from_user_id, msg.kind, msg.data);
|
||||||
|
}
|
||||||
|
refreshVoicePresence().catch(() => { });
|
||||||
|
};
|
||||||
|
|
||||||
|
ws.onclose = () => {
|
||||||
|
el.voiceConnection.classList.add("hidden");
|
||||||
|
for (const pc of state.voice.peerConnections.values()) pc.close();
|
||||||
|
state.voice.peerConnections.clear();
|
||||||
|
stopAndClearAudioPipeline();
|
||||||
|
state.voice.joinedChannelId = null;
|
||||||
|
state.voice.ws = null;
|
||||||
|
playSound('leave');
|
||||||
|
refreshVoicePresence().catch(() => { });
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
async function leaveVoice() {
|
||||||
|
if (state.voice.ws) state.voice.ws.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
function toggleMute() {
|
||||||
|
if (!state.voice.localStream) return;
|
||||||
|
state.voice.muted = !state.voice.muted;
|
||||||
|
state.voice.localStream.getAudioTracks().forEach((t) => {
|
||||||
|
t.enabled = !state.voice.muted;
|
||||||
|
});
|
||||||
|
el.voiceMuteBtn.innerHTML = state.voice.muted ? '<i data-lucide="mic-off"></i>' : '<i data-lucide="mic"></i>';
|
||||||
|
el.voiceMuteBtn.style.color = state.voice.muted ? 'var(--danger)' : 'var(--text-muted)';
|
||||||
|
lucide.createIcons();
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- Mobile Logic ---
|
||||||
|
|
||||||
|
function toggleMobileMenu() {
|
||||||
|
const isOpen = el.main.classList.contains("menu-open");
|
||||||
|
if (isOpen) {
|
||||||
|
el.main.classList.remove("menu-open");
|
||||||
|
el.overlay.classList.add("hidden");
|
||||||
|
} else {
|
||||||
|
el.main.classList.add("menu-open");
|
||||||
|
el.utilitySidebar.classList.remove("active"); // close members if open
|
||||||
|
el.overlay.classList.remove("hidden");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function toggleMobileMembers() {
|
||||||
|
const isOpen = el.utilitySidebar.classList.contains("active");
|
||||||
|
if (isOpen) {
|
||||||
|
el.utilitySidebar.classList.remove("active");
|
||||||
|
el.overlay.classList.add("hidden");
|
||||||
|
} else {
|
||||||
|
el.utilitySidebar.classList.add("active");
|
||||||
|
el.main.classList.remove("menu-open"); // close menu if open
|
||||||
|
el.overlay.classList.remove("hidden");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function closeMobileMenus() {
|
||||||
|
el.main.classList.remove("menu-open");
|
||||||
|
el.utilitySidebar.classList.remove("active");
|
||||||
|
el.overlay.classList.add("hidden");
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- Initialization ---
|
||||||
|
|
||||||
|
async function init() {
|
||||||
|
lucide.createIcons();
|
||||||
|
|
||||||
|
el.loginBtn.onclick = () => { location.href = "/auth/login"; };
|
||||||
|
|
||||||
|
el.logoutBtn.onclick = async () => {
|
||||||
|
await leaveVoice();
|
||||||
|
await api("/auth/logout", { method: "POST" });
|
||||||
|
location.reload();
|
||||||
|
};
|
||||||
|
|
||||||
|
el.addGuildBtn.onclick = () => { el.modalContainer.classList.remove("hidden"); };
|
||||||
|
el.createInviteBtn.onclick = createInviteLink;
|
||||||
|
el.modalCancel.onclick = () => { el.modalContainer.classList.add("hidden"); };
|
||||||
|
|
||||||
|
el.addTextBtn.onclick = () => {
|
||||||
|
if (!state.selectedGuildId) {
|
||||||
|
alert("Create or select a server first.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
document.querySelector('input[name="channel-kind"][value="text"]').checked = true;
|
||||||
|
el.channelModal.classList.remove("hidden");
|
||||||
|
};
|
||||||
|
el.addVoiceBtn.onclick = () => {
|
||||||
|
if (!state.selectedGuildId) {
|
||||||
|
alert("Create or select a server first.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
document.querySelector('input[name="channel-kind"][value="voice"]').checked = true;
|
||||||
|
el.channelModal.classList.remove("hidden");
|
||||||
|
};
|
||||||
|
el.channelModalCancel.onclick = () => { el.channelModal.classList.add("hidden"); };
|
||||||
|
|
||||||
|
// Mobile Listeners
|
||||||
|
if (el.mobileMenuBtn) el.mobileMenuBtn.onclick = toggleMobileMenu;
|
||||||
|
if (el.mobileMembersBtn) el.mobileMembersBtn.onclick = toggleMobileMembers;
|
||||||
|
if (el.overlay) el.overlay.onclick = closeMobileMenus;
|
||||||
|
|
||||||
|
el.guildForm.onsubmit = async (e) => {
|
||||||
|
e.preventDefault();
|
||||||
|
try {
|
||||||
|
const guild = await api("/guilds", {
|
||||||
|
method: "POST",
|
||||||
|
body: JSON.stringify({ name: el.guildName.value }),
|
||||||
|
});
|
||||||
|
el.guildName.value = "";
|
||||||
|
el.modalContainer.classList.add("hidden");
|
||||||
|
state.guilds.push(guild);
|
||||||
|
state.selectedGuildId = guild.id;
|
||||||
|
state.selectedTextChannelId = null;
|
||||||
|
state.selectedVoiceChannelId = null;
|
||||||
|
state.selectedDmUserId = null;
|
||||||
|
state.selectedDmDisplayName = null;
|
||||||
|
localStorage.setItem(LAST_GUILD_STORAGE_KEY, guild.id);
|
||||||
|
renderGuilds();
|
||||||
|
await loadChannels();
|
||||||
|
await loadGuildMembers();
|
||||||
|
await refreshVoicePresence();
|
||||||
|
renderMessages([]);
|
||||||
|
updateHeaderLabels();
|
||||||
|
} catch (err) { alert(err.message); }
|
||||||
|
};
|
||||||
|
|
||||||
|
el.channelForm.onsubmit = async (e) => {
|
||||||
|
e.preventDefault();
|
||||||
|
if (!state.selectedGuildId) {
|
||||||
|
alert("Select a server first.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const kindInput = document.querySelector('input[name="channel-kind"]:checked');
|
||||||
|
const kind = kindInput ? kindInput.value : "text";
|
||||||
|
const channelName = el.channelName.value.trim();
|
||||||
|
if (!channelName) {
|
||||||
|
alert("Channel name is required.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
const created = await api("/channels", {
|
||||||
|
method: "POST",
|
||||||
|
body: JSON.stringify({
|
||||||
|
guild_id: state.selectedGuildId,
|
||||||
|
name: channelName,
|
||||||
|
kind: kind,
|
||||||
|
}),
|
||||||
|
});
|
||||||
|
el.channelName.value = "";
|
||||||
|
el.channelModal.classList.add("hidden");
|
||||||
|
await loadChannels();
|
||||||
|
if (created.kind === "text") {
|
||||||
|
state.selectedTextChannelId = created.id;
|
||||||
|
state.selectedVoiceChannelId = null;
|
||||||
|
state.selectedDmUserId = null;
|
||||||
|
state.selectedDmDisplayName = null;
|
||||||
|
renderChannels();
|
||||||
|
renderDMs();
|
||||||
|
updateHeaderLabels();
|
||||||
|
const messages = await api(`/channels/${created.id}/messages?limit=100`);
|
||||||
|
renderMessages(messages);
|
||||||
|
} else {
|
||||||
|
state.selectedVoiceChannelId = created.id;
|
||||||
|
state.selectedTextChannelId = null;
|
||||||
|
state.selectedDmUserId = null;
|
||||||
|
state.selectedDmDisplayName = null;
|
||||||
|
renderChannels();
|
||||||
|
renderDMs();
|
||||||
|
updateHeaderLabels();
|
||||||
|
}
|
||||||
|
} catch (err) { alert(err.message); }
|
||||||
|
};
|
||||||
|
|
||||||
|
el.messageForm.onsubmit = async (e) => {
|
||||||
|
e.preventDefault();
|
||||||
|
const body = el.messageBody.value.trim();
|
||||||
|
if (!body) return;
|
||||||
|
|
||||||
|
try {
|
||||||
|
if (state.selectedTextChannelId) {
|
||||||
|
await api(`/channels/${state.selectedTextChannelId}/messages`, {
|
||||||
|
method: "POST",
|
||||||
|
body: JSON.stringify({ body }),
|
||||||
|
});
|
||||||
|
} else if (state.selectedDmUserId) {
|
||||||
|
await api(`/dms/${state.selectedDmUserId}/messages`, {
|
||||||
|
method: "POST",
|
||||||
|
body: JSON.stringify({ body }),
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
el.messageBody.value = "";
|
||||||
|
const messages = state.selectedTextChannelId
|
||||||
|
? await api(`/channels/${state.selectedTextChannelId}/messages?limit=100`)
|
||||||
|
: await api(`/dms/${state.selectedDmUserId}/messages?limit=100`);
|
||||||
|
renderMessages(messages);
|
||||||
|
await loadDMConversations();
|
||||||
|
renderDMs();
|
||||||
|
} catch (err) { alert(err.message); }
|
||||||
|
};
|
||||||
|
|
||||||
|
el.voiceMuteBtn.onclick = toggleMute;
|
||||||
|
el.voiceLeaveBtn.onclick = leaveVoice;
|
||||||
|
|
||||||
|
try {
|
||||||
|
state.me = await api("/me");
|
||||||
|
if (state.me && state.me.display_name) {
|
||||||
|
el.userName.textContent = state.me.display_name;
|
||||||
|
el.userAvatar.textContent = shortName(state.me.display_name);
|
||||||
|
}
|
||||||
|
el.authScreen.classList.add("hidden");
|
||||||
|
el.main.classList.remove("hidden");
|
||||||
|
|
||||||
|
const params = new URLSearchParams(location.search);
|
||||||
|
const inviteCode = params.get("invite");
|
||||||
|
if (inviteCode) {
|
||||||
|
try {
|
||||||
|
const joinedGuild = await api(`/invites/${encodeURIComponent(inviteCode)}/join`, {
|
||||||
|
method: "POST",
|
||||||
|
});
|
||||||
|
localStorage.setItem(LAST_GUILD_STORAGE_KEY, joinedGuild.id);
|
||||||
|
} catch (err) {
|
||||||
|
alert(`Failed to join invite: ${err.message}`);
|
||||||
|
} finally {
|
||||||
|
params.delete("invite");
|
||||||
|
const nextQuery = params.toString();
|
||||||
|
const nextUrl = `${location.pathname}${nextQuery ? `?${nextQuery}` : ""}`;
|
||||||
|
history.replaceState(null, "", nextUrl);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
await loadGuilds();
|
||||||
|
await loadDMConversations();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const online = await api("/presence");
|
||||||
|
state.onlineUsers = new Set(online);
|
||||||
|
} catch (err) { console.warn("presence sync failed", err); }
|
||||||
|
|
||||||
|
if (state.guilds.length > 0) {
|
||||||
|
const lastGuildId = localStorage.getItem(LAST_GUILD_STORAGE_KEY);
|
||||||
|
const guild = state.guilds.find((g) => g.id === lastGuildId) || state.guilds[0];
|
||||||
|
state.selectedGuildId = guild.id;
|
||||||
|
localStorage.setItem(LAST_GUILD_STORAGE_KEY, guild.id);
|
||||||
|
renderGuilds();
|
||||||
|
await loadChannels();
|
||||||
|
await loadGuildMembers();
|
||||||
|
await refreshVoicePresence();
|
||||||
|
updateHeaderLabels();
|
||||||
|
} else {
|
||||||
|
state.members = [];
|
||||||
|
renderMembers();
|
||||||
|
updateHeaderLabels();
|
||||||
|
}
|
||||||
|
|
||||||
|
initChatWs();
|
||||||
|
startVoicePresencePolling();
|
||||||
|
lucide.createIcons();
|
||||||
|
} catch (err) {
|
||||||
|
console.error("init failed", err);
|
||||||
|
el.authScreen.classList.remove("hidden");
|
||||||
|
el.main.classList.add("hidden");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
init();
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,3 @@
|
||||||
<!-- Generated from shared-html/index.template.html. Do not edit directly. -->
|
|
||||||
<!doctype html>
|
<!doctype html>
|
||||||
<html lang="en">
|
<html lang="en">
|
||||||
|
|
||||||
|
|
@ -8,7 +7,7 @@
|
||||||
<title>Chattz</title>
|
<title>Chattz</title>
|
||||||
<link rel="stylesheet" href="/static/styles.css" />
|
<link rel="stylesheet" href="/static/styles.css" />
|
||||||
<!-- Lucide Icons -->
|
<!-- Lucide Icons -->
|
||||||
<script src="/static/vendor/lucide.min.js"></script>
|
<script src="https://unpkg.com/lucide@latest"></script>
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body>
|
<body>
|
||||||
|
|
@ -18,11 +17,6 @@
|
||||||
<h1>Chattz</h1>
|
<h1>Chattz</h1>
|
||||||
<p>Sign in to open your servers.</p>
|
<p>Sign in to open your servers.</p>
|
||||||
<button id="login-btn">Login with Authentik</button>
|
<button id="login-btn">Login with Authentik</button>
|
||||||
<div class="auth-downloads">
|
|
||||||
<span>Desktop downloads:</span>
|
|
||||||
<a href="/static/installers/chattz-windows.exe">Windows</a>
|
|
||||||
<a href="/static/installers/chattz-linux.rpm">Linux (RPM)</a>
|
|
||||||
</div>
|
|
||||||
<p id="status" class="status-line"></p>
|
<p id="status" class="status-line"></p>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
@ -43,7 +37,6 @@
|
||||||
<header class="sidebar-header clickable" id="guild-header">
|
<header class="sidebar-header clickable" id="guild-header">
|
||||||
<h2 id="guild-title">No server selected</h2>
|
<h2 id="guild-title">No server selected</h2>
|
||||||
<div class="sidebar-header-actions">
|
<div class="sidebar-header-actions">
|
||||||
<span id="invite-copied-badge" class="copied-badge hidden">Copied!</span>
|
|
||||||
<button id="create-invite-btn" class="header-action-btn" title="Create Invite Link">
|
<button id="create-invite-btn" class="header-action-btn" title="Create Invite Link">
|
||||||
<i data-lucide="link-2"></i>
|
<i data-lucide="link-2"></i>
|
||||||
</button>
|
</button>
|
||||||
|
|
@ -92,19 +85,10 @@
|
||||||
<span id="vc-channel-name" class="vc-name">General</span>
|
<span id="vc-channel-name" class="vc-name">General</span>
|
||||||
</div>
|
</div>
|
||||||
<div class="vc-actions">
|
<div class="vc-actions">
|
||||||
<button id="voice-video-btn" title="Turn on Camera"><i data-lucide="video-off"></i></button>
|
|
||||||
<button id="voice-screen-btn" title="Share Screen"><i data-lucide="monitor-off"></i></button>
|
|
||||||
<button id="voice-mute-btn" title="Mute"><i data-lucide="mic"></i></button>
|
<button id="voice-mute-btn" title="Mute"><i data-lucide="mic"></i></button>
|
||||||
<button id="voice-leave-btn" title="Disconnect"><i data-lucide="phone-off"></i></button>
|
<button id="voice-leave-btn" title="Disconnect"><i data-lucide="phone-off"></i></button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<div id="soundboard" class="soundboard-container hidden">
|
|
||||||
<div class="soundboard-header">
|
|
||||||
<span>Sound Board</span>
|
|
||||||
<span id="add-sound-btn" class="btn-add-sound">+ Add Sound</span>
|
|
||||||
</div>
|
|
||||||
<div id="soundboard-grid" class="soundboard-grid"></div>
|
|
||||||
</div>
|
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div class="user-panel">
|
<div class="user-panel">
|
||||||
|
|
@ -117,10 +101,6 @@
|
||||||
<div class="user-status">Online</div>
|
<div class="user-status">Online</div>
|
||||||
</div>
|
</div>
|
||||||
<div class="user-actions">
|
<div class="user-actions">
|
||||||
<div id="update-notifier" class="update-notifier hidden">
|
|
||||||
<button id="update-download-btn" title="Download Update"><i data-lucide="download"></i></button>
|
|
||||||
<button id="update-install-btn" title="Install Update" class="hidden"><i data-lucide="arrow-up-circle"></i></button>
|
|
||||||
</div>
|
|
||||||
<button id="user-settings-btn" title="Settings"><i data-lucide="settings"></i></button>
|
<button id="user-settings-btn" title="Settings"><i data-lucide="settings"></i></button>
|
||||||
<button id="logout-btn" title="Logout"><i data-lucide="log-out"></i></button>
|
<button id="logout-btn" title="Logout"><i data-lucide="log-out"></i></button>
|
||||||
</div>
|
</div>
|
||||||
|
|
@ -141,18 +121,10 @@
|
||||||
</button>
|
</button>
|
||||||
</header>
|
</header>
|
||||||
|
|
||||||
<div id="video-grid" class="video-grid hidden"></div>
|
|
||||||
<div id="message-list" class="message-list"></div>
|
<div id="message-list" class="message-list"></div>
|
||||||
|
|
||||||
<div class="chat-input-wrapper">
|
<div class="chat-input-wrapper">
|
||||||
<form id="message-form" class="message-form">
|
<form id="message-form" class="message-form">
|
||||||
<input id="media-file-input" type="file" class="hidden-file-input" accept="image/*,video/*,audio/*,.pdf,.txt,.zip,.rar,.7z,.tar,.gz,.json,.csv,.md" />
|
|
||||||
<button type="button" id="media-upload-btn" class="input-action-btn" title="Upload File">
|
|
||||||
<i data-lucide="paperclip"></i>
|
|
||||||
</button>
|
|
||||||
<button type="button" id="gif-btn" class="input-action-btn" title="Open GIF Picker">
|
|
||||||
<i data-lucide="image"></i>
|
|
||||||
</button>
|
|
||||||
<input id="message-body" autocomplete="off" placeholder="Message #channel" maxlength="4000" required />
|
<input id="message-body" autocomplete="off" placeholder="Message #channel" maxlength="4000" required />
|
||||||
<button type="submit" class="hidden"></button>
|
<button type="submit" class="hidden"></button>
|
||||||
</form>
|
</form>
|
||||||
|
|
@ -172,24 +144,6 @@
|
||||||
<div id="overlay" class="overlay hidden"></div>
|
<div id="overlay" class="overlay hidden"></div>
|
||||||
|
|
||||||
<!-- Modals -->
|
<!-- Modals -->
|
||||||
<div id="gif-picker" class="modal-container hidden">
|
|
||||||
<div class="modal gif-modal">
|
|
||||||
<div class="modal-header">
|
|
||||||
<h2>Select GIF</h2>
|
|
||||||
<button type="button" class="close-btn" id="gif-picker-close">
|
|
||||||
<i data-lucide="x"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
<div class="gif-search-wrapper">
|
|
||||||
<input type="text" id="gif-search-input" placeholder="Search for GIFs..." autocomplete="off">
|
|
||||||
<i data-lucide="search" class="search-icon"></i>
|
|
||||||
</div>
|
|
||||||
<div id="gif-results" class="gif-results-grid">
|
|
||||||
<!-- GIF results will be injected here -->
|
|
||||||
<div class="gif-loading hidden">Loading...</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div id="modal-container" class="modal-container hidden">
|
<div id="modal-container" class="modal-container hidden">
|
||||||
<div class="modal">
|
<div class="modal">
|
||||||
<h2 id="modal-title">Create Server</h2>
|
<h2 id="modal-title">Create Server</h2>
|
||||||
|
|
@ -247,41 +201,7 @@
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div id="sound-modal" class="modal-container hidden">
|
<script src="/static/app.js" defer></script>
|
||||||
<div class="modal">
|
|
||||||
<h2>Add Sound</h2>
|
|
||||||
<form id="sound-form" class="modal-form">
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="sound-name">NAME</label>
|
|
||||||
<input id="sound-name" placeholder="Quack" maxlength="32" required />
|
|
||||||
</div>
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="sound-icon">ICON (Emoji or Initials)</label>
|
|
||||||
<input id="sound-icon" placeholder="🦆" maxlength="4" required />
|
|
||||||
</div>
|
|
||||||
<div class="form-item">
|
|
||||||
<label for="sound-file">AUDIO FILE (MP3/WAV)</label>
|
|
||||||
<input id="sound-file" type="file" accept="audio/*" required />
|
|
||||||
</div>
|
|
||||||
<div class="modal-footer">
|
|
||||||
<button type="button" class="cancel-btn" id="sound-modal-cancel">Cancel</button>
|
|
||||||
<button type="submit" class="submit-btn" id="sound-submit-btn">Add Sound</button>
|
|
||||||
</div>
|
|
||||||
</form>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="upload-limit-modal" class="modal-container hidden">
|
|
||||||
<div class="modal">
|
|
||||||
<h2>Upload Too Large</h2>
|
|
||||||
<p id="upload-limit-message" class="modal-copy">Uploads are limited to 50 MB per file.</p>
|
|
||||||
<div class="modal-footer">
|
|
||||||
<button type="button" class="submit-btn" id="upload-limit-ok">OK</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<script type="module" src="/static/app.js?v=20260227-shared-core-1"></script>
|
|
||||||
</body>
|
</body>
|
||||||
|
|
||||||
</html>
|
</html>
|
||||||
File diff suppressed because it is too large
Load diff
1000
static/styles.css
1000
static/styles.css
File diff suppressed because it is too large
Load diff
249
static/vendor/deepfilternet3-noise-filter.esm.js
vendored
249
static/vendor/deepfilternet3-noise-filter.esm.js
vendored
File diff suppressed because one or more lines are too long
Binary file not shown.
BIN
static/vendor/deepfilternet3/v2/pkg/df_bg.wasm
vendored
BIN
static/vendor/deepfilternet3/v2/pkg/df_bg.wasm
vendored
Binary file not shown.
12
static/vendor/lucide.min.js
vendored
12
static/vendor/lucide.min.js
vendored
File diff suppressed because one or more lines are too long
33
test.html
33
test.html
|
|
@ -1,33 +0,0 @@
|
||||||
<!DOCTYPE html>
|
|
||||||
<html>
|
|
||||||
<head>
|
|
||||||
<link rel="stylesheet" href="static/styles.css" />
|
|
||||||
<style>
|
|
||||||
body { margin: 0; height: 100vh; display: flex; }
|
|
||||||
.chat-pane { flex: 1; position: relative; background: #333; }
|
|
||||||
.chat-header { height: 48px; background: #222; }
|
|
||||||
</style>
|
|
||||||
</head>
|
|
||||||
<body>
|
|
||||||
<div class="chat-pane">
|
|
||||||
<div class="chat-header">Header</div>
|
|
||||||
<div id="video-grid" class="video-grid">
|
|
||||||
<div class="video-item" onclick="toggle(this)" style="display:flex; justify-content:center; align-items:center; color:white;">Click me</div>
|
|
||||||
<div class="video-item" style="color:white;">Another</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<script>
|
|
||||||
function toggle(el) {
|
|
||||||
const grid = document.getElementById('video-grid');
|
|
||||||
const isFullscreen = el.classList.contains('fullscreen');
|
|
||||||
document.querySelectorAll('.video-item').forEach(e => e.classList.remove('fullscreen'));
|
|
||||||
if (isFullscreen) {
|
|
||||||
grid.classList.remove('has-fullscreen');
|
|
||||||
} else {
|
|
||||||
el.classList.add('fullscreen');
|
|
||||||
grid.classList.add('has-fullscreen');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
</script>
|
|
||||||
</body>
|
|
||||||
</html>
|
|
||||||
|
|
@ -1,5 +0,0 @@
|
||||||
|
|
||||||
> chattz-electron@0.1.0 start
|
|
||||||
> electron .
|
|
||||||
|
|
||||||
Loading desktop app from: file:///home/pavel/chattz/desktop/index.html?backend=https%3A%2F%2Fdiscord.flegr.me
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue