mirror of
https://github.com/permissionlesstech/bitchat.git
synced 2026-07-25 19:05:20 +00:00
Compare commits
60
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
266827ceff | ||
|
|
97bc3f53bc | ||
|
|
9dc0ba6991 | ||
|
|
af954b05ea | ||
|
|
c8381737bb | ||
|
|
fa136d8973 | ||
|
|
7fbe6a4a7e | ||
|
|
4791114406 | ||
|
|
1a6a08f92a | ||
|
|
0b007eee1a | ||
|
|
1480b51c76 | ||
|
|
0e38ccfb3b | ||
|
|
e74f36927f | ||
|
|
cf3b85f65c | ||
|
|
8899cb7f9e | ||
|
|
22be3d6392 | ||
|
|
8a867a17a1 | ||
|
|
ed86ed1065 | ||
|
|
e74d9a7937 | ||
|
|
8289a6d05e | ||
|
|
38331e62f1 | ||
|
|
7fb1f4a219 | ||
|
|
99d1d1dccd | ||
|
|
879d8cba12 | ||
|
|
ac3a2f2d34 | ||
|
|
45650854e7 | ||
|
|
75dd83d9cc | ||
|
|
93d01b8fa6 | ||
|
|
6c0dbbbd0d | ||
|
|
09087b74cc | ||
|
|
cc76086615 | ||
|
|
638f3f5005 | ||
|
|
6091ee83ad | ||
|
|
82736c4991 | ||
|
|
707b22878d | ||
|
|
80bed1f395 | ||
|
|
4b287f7490 | ||
|
|
3caf2d7663 | ||
|
|
14d025cc2a | ||
|
|
19b28cf49d | ||
|
|
a2825ca288 | ||
|
|
3a995e20b6 | ||
|
|
6bda919dd4 | ||
|
|
4093ee6733 | ||
|
|
eb2c128cab | ||
|
|
3eb4f2bd72 | ||
|
|
193cfdc06a | ||
|
|
ffa0d7aa4f | ||
|
|
9e84f5e822 | ||
|
|
df36b19afe | ||
|
|
ab0da61533 | ||
|
|
3be8fbf1c4 | ||
|
|
764f016d17 | ||
|
|
a6cb8872fb | ||
|
|
3daf02732a | ||
|
|
0d1cdc7644 | ||
|
|
43eef0d49c | ||
|
|
bafa461f26 | ||
|
|
602d2316ef | ||
|
|
c60eff2c11 |
@@ -0,0 +1,85 @@
|
|||||||
|
name: Arti Binary Provenance
|
||||||
|
|
||||||
|
# The Arti xcframework is a vendored binary; these checks turn the policy in
|
||||||
|
# docs/ARTI-BINARY-PROVENANCE.md into an enforced gate:
|
||||||
|
# 1. The checked-in binary must match the hash manifest in the provenance doc.
|
||||||
|
# 2. A PR that changes the binary must also change at least one provenance
|
||||||
|
# input (Rust source, lockfile, build script, or the doc itself).
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches:
|
||||||
|
- main
|
||||||
|
paths:
|
||||||
|
- "localPackages/Arti/**"
|
||||||
|
- "docs/ARTI-BINARY-PROVENANCE.md"
|
||||||
|
pull_request:
|
||||||
|
paths:
|
||||||
|
- "localPackages/Arti/**"
|
||||||
|
- "docs/ARTI-BINARY-PROVENANCE.md"
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
verify-hashes:
|
||||||
|
name: Verify xcframework hashes against provenance doc
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v5
|
||||||
|
|
||||||
|
- name: Compare artifact hashes with manifest
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
doc="docs/ARTI-BINARY-PROVENANCE.md"
|
||||||
|
|
||||||
|
# Extract the manifest: lines of "<sha256> <path>" from the doc.
|
||||||
|
grep -E '^[0-9a-f]{64} localPackages/Arti/Frameworks/arti\.xcframework/' "$doc" \
|
||||||
|
| sort -k2 > expected.txt
|
||||||
|
|
||||||
|
if [ ! -s expected.txt ]; then
|
||||||
|
echo "::error::No hash manifest found in $doc"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Hash the same file set the doc documents.
|
||||||
|
find localPackages/Arti/Frameworks/arti.xcframework -maxdepth 3 -type f -print0 \
|
||||||
|
| sort -z | xargs -0 sha256sum | sed 's/ \.\// /' | sort -k2 > actual.txt
|
||||||
|
|
||||||
|
if ! diff -u expected.txt actual.txt; then
|
||||||
|
echo "::error::Checked-in arti.xcframework does not match the manifest in $doc. If the binary change is intentional, rebuild per the doc and update the manifest in the same PR."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
echo "All $(wc -l < actual.txt) artifact hashes match the provenance manifest."
|
||||||
|
|
||||||
|
require-provenance-evidence:
|
||||||
|
name: Binary changes must ship with provenance inputs
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
if: github.event_name == 'pull_request'
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v5
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Check changed files
|
||||||
|
run: |
|
||||||
|
set -euo pipefail
|
||||||
|
base="origin/${{ github.base_ref }}"
|
||||||
|
git fetch --no-tags --depth=1 origin "${{ github.base_ref }}"
|
||||||
|
changed=$(git diff --name-only "$base"...HEAD)
|
||||||
|
echo "Changed files:"
|
||||||
|
echo "$changed"
|
||||||
|
|
||||||
|
if ! echo "$changed" | grep -q '^localPackages/Arti/Frameworks/arti\.xcframework/'; then
|
||||||
|
echo "No binary artifact changes; nothing to verify."
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if echo "$changed" | grep -Eq '^(localPackages/Arti/(Cargo\.(toml|lock)|build-ios\.sh|arti-bitchat/)|docs/ARTI-BINARY-PROVENANCE\.md)'; then
|
||||||
|
echo "Binary change is accompanied by provenance inputs."
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "::error::arti.xcframework changed without matching source, lockfile, build-script, or provenance-doc changes. See docs/ARTI-BINARY-PROVENANCE.md (\"Do not accept an xcframework-only update\")."
|
||||||
|
exit 1
|
||||||
@@ -10,6 +10,9 @@ jobs:
|
|||||||
test:
|
test:
|
||||||
name: Run Swift Tests (${{ matrix.name }})
|
name: Run Swift Tests (${{ matrix.name }})
|
||||||
runs-on: macos-latest
|
runs-on: macos-latest
|
||||||
|
# A hung test must fail fast, not hold a runner for GitHub's 360-minute
|
||||||
|
# default (observed: intermittent app-suite hangs starving the queue).
|
||||||
|
timeout-minutes: 15
|
||||||
|
|
||||||
strategy:
|
strategy:
|
||||||
fail-fast: false # Don't cancel other matrix jobs when one fails
|
fail-fast: false # Don't cancel other matrix jobs when one fails
|
||||||
@@ -21,10 +24,6 @@ jobs:
|
|||||||
path: localPackages/BitLogger
|
path: localPackages/BitLogger
|
||||||
- name: BitFoundation
|
- name: BitFoundation
|
||||||
path: localPackages/BitFoundation
|
path: localPackages/BitFoundation
|
||||||
- name: Noise
|
|
||||||
path: localPackages/Noise
|
|
||||||
- name: Nostr
|
|
||||||
path: localPackages/Nostr
|
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout code
|
- name: Checkout code
|
||||||
@@ -42,5 +41,102 @@ jobs:
|
|||||||
${{ runner.os }}-${{ matrix.name }}-${{ hashFiles(format('{0}/**/Package.resolved', matrix.path)) }}
|
${{ runner.os }}-${{ matrix.name }}-${{ hashFiles(format('{0}/**/Package.resolved', matrix.path)) }}
|
||||||
${{ runner.os }}-${{ matrix.name }}-
|
${{ runner.os }}-${{ matrix.name }}-
|
||||||
|
|
||||||
|
- name: Build tests
|
||||||
|
# Built separately so the hang watchdog below times only test
|
||||||
|
# execution: a cold-cache coverage build on a slow runner can
|
||||||
|
# legitimately take several minutes, and is already bounded by the
|
||||||
|
# 15-minute job timeout.
|
||||||
|
run: swift build --build-tests --enable-code-coverage --package-path ${{ matrix.path }}
|
||||||
|
|
||||||
- name: Run Tests
|
- name: Run Tests
|
||||||
run: swift test --parallel --quiet --package-path ${{ matrix.path }}
|
# Perf benchmarks are excluded here and run in their own serial step
|
||||||
|
# below: measuring while parallel test processes contend for cores
|
||||||
|
# produces noisy numbers, and the XCTest measure machinery has hung
|
||||||
|
# intermittently under parallel workers on loaded runners. Excluded
|
||||||
|
# via --skip (not just the env guard): every app run since the
|
||||||
|
# baselines landed timed out at the 15-minute job limit with the
|
||||||
|
# baseline tests dispatched into the parallel phase.
|
||||||
|
#
|
||||||
|
# The watchdog samples any still-running test processes after 5
|
||||||
|
# minutes (the suite passes in seconds when healthy; the build is
|
||||||
|
# done by this step) and kills the run, so a hang fails fast with
|
||||||
|
# stacks in the log instead of a silent timeout.
|
||||||
|
env:
|
||||||
|
BITCHAT_SKIP_PERF_BASELINES: "1"
|
||||||
|
run: |
|
||||||
|
swift test --skip-build --parallel --quiet --enable-code-coverage \
|
||||||
|
--skip PerformanceBaselineTests \
|
||||||
|
--package-path ${{ matrix.path }} &
|
||||||
|
test_pid=$!
|
||||||
|
(
|
||||||
|
sleep 300
|
||||||
|
if kill -0 "$test_pid" 2>/dev/null; then
|
||||||
|
echo "::group::Tests still running after 5 minutes — sampling before kill"
|
||||||
|
for pid in $(pgrep -if 'swiftpm-testing|xctest|PackageTests' || true); do
|
||||||
|
echo "--- sample of pid $pid ---"
|
||||||
|
sample "$pid" 5 2>/dev/null || true
|
||||||
|
done
|
||||||
|
echo "::endgroup::"
|
||||||
|
pkill -KILL -P "$test_pid" 2>/dev/null || true
|
||||||
|
kill -KILL "$test_pid" 2>/dev/null || true
|
||||||
|
fi
|
||||||
|
) &
|
||||||
|
watchdog_pid=$!
|
||||||
|
wait "$test_pid" && status=0 || status=$?
|
||||||
|
kill "$watchdog_pid" 2>/dev/null || true
|
||||||
|
exit "$status"
|
||||||
|
|
||||||
|
# Benchmarks run serially on an otherwise idle runner for stable
|
||||||
|
# numbers; BITCHAT_PERF_LOG captures the PERF[...] lines for the gate.
|
||||||
|
- name: Run performance benchmarks (serial)
|
||||||
|
if: matrix.name == 'app'
|
||||||
|
timeout-minutes: 6
|
||||||
|
env:
|
||||||
|
BITCHAT_PERF_LOG: ${{ github.workspace }}/perf-output.log
|
||||||
|
run: swift test --quiet --filter PerformanceBaselineTests
|
||||||
|
|
||||||
|
# Order-of-magnitude performance regression gate. Floors are deliberately
|
||||||
|
# generous (see bitchatTests/Performance/perf-floors.json) so this
|
||||||
|
# catches algorithmic regressions, never runner variance.
|
||||||
|
- name: Performance floor gate
|
||||||
|
if: matrix.name == 'app'
|
||||||
|
run: ./scripts/check-perf-floors.sh perf-output.log
|
||||||
|
|
||||||
|
# Informational only: surfaces per-file and total line coverage in the
|
||||||
|
# job log so coverage trends are visible on every PR. No thresholds —
|
||||||
|
# this must never be the reason a build goes red.
|
||||||
|
- name: Coverage summary
|
||||||
|
run: |
|
||||||
|
BIN_PATH=$(swift build --show-bin-path --package-path ${{ matrix.path }})
|
||||||
|
PROF="$BIN_PATH/codecov/default.profdata"
|
||||||
|
XCTEST=$(find "$BIN_PATH" -maxdepth 1 -name '*.xctest' | head -1)
|
||||||
|
BINARY="$XCTEST/Contents/MacOS/$(basename "$XCTEST" .xctest)"
|
||||||
|
if [ -f "$PROF" ] && [ -f "$BINARY" ]; then
|
||||||
|
xcrun llvm-cov report "$BINARY" -instr-profile "$PROF" \
|
||||||
|
-ignore-filename-regex='(Tests|\.build|checkouts|Mocks|_PreviewHelpers)' || true
|
||||||
|
else
|
||||||
|
echo "No coverage data found; skipping summary."
|
||||||
|
fi
|
||||||
|
|
||||||
|
# SPM tests above only compile the macOS slice; this job covers the
|
||||||
|
# iOS-conditional code paths (UIKit, CoreBluetooth restoration, etc.).
|
||||||
|
ios-build:
|
||||||
|
name: Build iOS app (simulator)
|
||||||
|
runs-on: macos-latest
|
||||||
|
timeout-minutes: 15
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v5
|
||||||
|
|
||||||
|
- name: Build iOS (simulator, no signing)
|
||||||
|
# arm64 only: the vendored arti.xcframework has no x86_64 simulator slice.
|
||||||
|
run: |
|
||||||
|
set -o pipefail
|
||||||
|
xcodebuild -project bitchat.xcodeproj \
|
||||||
|
-scheme "bitchat (iOS)" \
|
||||||
|
-sdk iphonesimulator \
|
||||||
|
-destination 'generic/platform=iOS Simulator' \
|
||||||
|
ARCHS=arm64 \
|
||||||
|
CODE_SIGNING_ALLOWED=NO \
|
||||||
|
build
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
MARKETING_VERSION = 1.5.1
|
MARKETING_VERSION = 1.5.2
|
||||||
CURRENT_PROJECT_VERSION = 1
|
CURRENT_PROJECT_VERSION = 1
|
||||||
|
|
||||||
IPHONEOS_DEPLOYMENT_TARGET = 16.0
|
IPHONEOS_DEPLOYMENT_TARGET = 16.0
|
||||||
|
|||||||
+42
-16
@@ -1,6 +1,6 @@
|
|||||||
# bitchat Privacy Policy
|
# bitchat Privacy Policy
|
||||||
|
|
||||||
*Last updated: January 2025*
|
*Last updated: June 2026*
|
||||||
|
|
||||||
## Our Commitment
|
## Our Commitment
|
||||||
|
|
||||||
@@ -9,7 +9,7 @@ bitchat is designed with privacy as its foundation. We believe private communica
|
|||||||
## Summary
|
## Summary
|
||||||
|
|
||||||
- **No personal data collection** - We don't collect names, emails, or phone numbers
|
- **No personal data collection** - We don't collect names, emails, or phone numbers
|
||||||
- **No servers** - Everything happens on your device and through peer-to-peer connections
|
- **No accounts or company servers** - Mesh chat works peer-to-peer; optional Nostr features use public or user-selected relays
|
||||||
- **No tracking** - We have no analytics, telemetry, or user tracking
|
- **No tracking** - We have no analytics, telemetry, or user tracking
|
||||||
- **Open source** - You can verify these claims by reading our code
|
- **Open source** - You can verify these claims by reading our code
|
||||||
|
|
||||||
@@ -17,11 +17,11 @@ bitchat is designed with privacy as its foundation. We believe private communica
|
|||||||
|
|
||||||
### On Your Device Only
|
### On Your Device Only
|
||||||
|
|
||||||
1. **Identity Key**
|
1. **Identity Keys**
|
||||||
- A cryptographic key generated on first launch
|
- Cryptographic private keys generated on first launch or when optional Nostr identities are created
|
||||||
- Stored locally in your device's secure storage
|
- Stored locally in your device's secure storage
|
||||||
- Allows you to maintain "favorite" relationships across app restarts
|
- Allows you to maintain "favorite" relationships across app restarts
|
||||||
- Never leaves your device
|
- Private keys never leave your device; public keys are shared when needed for messaging
|
||||||
|
|
||||||
2. **Nickname**
|
2. **Nickname**
|
||||||
- The display name you choose (or auto-generated)
|
- The display name you choose (or auto-generated)
|
||||||
@@ -38,12 +38,19 @@ bitchat is designed with privacy as its foundation. We believe private communica
|
|||||||
- Stored only on your device
|
- Stored only on your device
|
||||||
- Allows you to recognize these peers in future sessions
|
- Allows you to recognize these peers in future sessions
|
||||||
|
|
||||||
|
5. **Optional Location Channel State**
|
||||||
|
- Your selected geohash channel, bookmarked geohashes, teleport flags, and bookmark display names
|
||||||
|
- Stored locally on your device so the location-channel UI can restore your choices
|
||||||
|
- Per-geohash Nostr identities are derived locally from a device seed stored in secure storage
|
||||||
|
- Exact latitude and longitude are not persisted by bitchat
|
||||||
|
|
||||||
### Temporary Session Data
|
### Temporary Session Data
|
||||||
|
|
||||||
During each session, bitchat temporarily maintains:
|
During each session, bitchat temporarily maintains:
|
||||||
- Active peer connections (forgotten when app closes)
|
- Active peer connections (forgotten when app closes)
|
||||||
- Routing information for message delivery
|
- Routing information for message delivery
|
||||||
- Cached messages for offline peers (12 hours max)
|
- Cached messages for offline peers (12 hours max)
|
||||||
|
- Your current location while optional location channels are enabled, used locally to compute geohash channels and friendly place names
|
||||||
|
|
||||||
## What Information is Shared
|
## What Information is Shared
|
||||||
|
|
||||||
@@ -62,13 +69,21 @@ When you join a password-protected room:
|
|||||||
- Your nickname appears in the member list
|
- Your nickname appears in the member list
|
||||||
- Room owners can see you've joined
|
- Room owners can see you've joined
|
||||||
|
|
||||||
|
### With Nostr Relays (Optional Features)
|
||||||
|
|
||||||
|
If you enable Nostr-backed features:
|
||||||
|
- Private fallback messages to mutual favorites are sent as encrypted NIP-17 gift wraps. Relays can see event metadata, but not message content.
|
||||||
|
- Public location-channel messages, location notes, and presence are scoped with geohash tags. Relays and other participants can see the geohash tag, event kind, timestamp, and public key used for that geohash.
|
||||||
|
- Exact GPS coordinates are not included in Nostr events by bitchat. The geohash precision you choose can still reveal an approximate area, from region-level to building-level.
|
||||||
|
- Automatic presence heartbeats are limited to low-precision geohashes (region, province, and city). More precise geohash posts happen only when you use those channels or location notes.
|
||||||
|
|
||||||
## What We DON'T Do
|
## What We DON'T Do
|
||||||
|
|
||||||
bitchat **never**:
|
bitchat **never**:
|
||||||
- Collects personal information
|
- Collects personal information
|
||||||
- Tracks your location
|
- Sells or shares your exact GPS location
|
||||||
- Stores data on servers
|
- Stores data on servers we operate
|
||||||
- Shares data with third parties
|
- Sells your data to advertisers or data brokers
|
||||||
- Uses analytics or telemetry
|
- Uses analytics or telemetry
|
||||||
- Creates user profiles
|
- Creates user profiles
|
||||||
- Requires registration
|
- Requires registration
|
||||||
@@ -84,19 +99,27 @@ All private messages use end-to-end encryption:
|
|||||||
## Your Rights
|
## Your Rights
|
||||||
|
|
||||||
You have complete control:
|
You have complete control:
|
||||||
- **Delete Everything**: Triple-tap the logo to instantly wipe all data
|
- **Delete Local State**: Triple-tap the logo to instantly wipe local keys, sessions, caches, and preferences
|
||||||
- **Leave Anytime**: Close the app and your presence disappears
|
- **Leave Anytime**: Close the app and local presence stops; relay-backed presence ages out
|
||||||
- **No Account**: Nothing to delete from servers because there are none
|
- **No Account**: No account record exists for you to delete from us
|
||||||
- **Portability**: Your data never leaves your device unless you export it
|
- **Portability**: Your local state stays on your device unless you send messages, use optional relay-backed features, or export it
|
||||||
|
|
||||||
## Bluetooth & Permissions
|
## Bluetooth & Permissions
|
||||||
|
|
||||||
bitchat requires Bluetooth permission to function:
|
bitchat requires Bluetooth permission to function:
|
||||||
- Used only for peer-to-peer communication
|
- Used only for peer-to-peer communication
|
||||||
- No location data is accessed or stored
|
|
||||||
- Bluetooth is not used for tracking
|
- Bluetooth is not used for tracking
|
||||||
- You can revoke this permission at any time in system settings
|
- You can revoke this permission at any time in system settings
|
||||||
|
|
||||||
|
## Location Permission
|
||||||
|
|
||||||
|
Location permission is optional and is used only for location channels:
|
||||||
|
- Used to compute local geohash channels and display names
|
||||||
|
- Requested as when-in-use permission
|
||||||
|
- Exact coordinates are not shared in messages or stored by bitchat
|
||||||
|
- Selected and bookmarked geohashes may persist locally until you remove them, use panic wipe, or delete the app
|
||||||
|
- You can revoke this permission at any time in system settings
|
||||||
|
|
||||||
## Children's Privacy
|
## Children's Privacy
|
||||||
|
|
||||||
bitchat does not knowingly collect information from children. The app has no age verification because it collects no personal information from anyone.
|
bitchat does not knowingly collect information from children. The app has no age verification because it collects no personal information from anyone.
|
||||||
@@ -106,12 +129,15 @@ bitchat does not knowingly collect information from children. The app has no age
|
|||||||
- **Messages**: Deleted from memory when app closes (unless room retention is enabled)
|
- **Messages**: Deleted from memory when app closes (unless room retention is enabled)
|
||||||
- **Identity Key**: Persists until you delete the app
|
- **Identity Key**: Persists until you delete the app
|
||||||
- **Favorites**: Persist until you remove them or delete the app
|
- **Favorites**: Persist until you remove them or delete the app
|
||||||
|
- **Location channel choices**: Selected/bookmarked geohashes persist locally until removed, panic-wiped, or the app is deleted
|
||||||
|
- **Nostr relay data**: Public geohash events and encrypted gift wraps may be retained by relays according to each relay's policy
|
||||||
- **Everything Else**: Exists only during active sessions
|
- **Everything Else**: Exists only during active sessions
|
||||||
|
|
||||||
## Security Measures
|
## Security Measures
|
||||||
|
|
||||||
- All communication is encrypted
|
- All communication is encrypted
|
||||||
- No data transmitted to servers (there are none)
|
- No accounts or company servers
|
||||||
|
- Optional Nostr relays receive only the events needed for Nostr-backed private fallback or public location channels
|
||||||
- Open source code for public audit
|
- Open source code for public audit
|
||||||
- Regular security updates
|
- Regular security updates
|
||||||
- Cryptographic signatures prevent tampering
|
- Cryptographic signatures prevent tampering
|
||||||
@@ -121,7 +147,7 @@ bitchat does not knowingly collect information from children. The app has no age
|
|||||||
If we update this policy:
|
If we update this policy:
|
||||||
- The "Last updated" date will change
|
- The "Last updated" date will change
|
||||||
- The updated policy will be included in the app
|
- The updated policy will be included in the app
|
||||||
- No retroactive changes can affect data (since we don't collect any)
|
- No retroactive changes can make us collect data already held only in your app
|
||||||
|
|
||||||
## Contact
|
## Contact
|
||||||
|
|
||||||
@@ -132,7 +158,7 @@ bitchat is an open source project. For privacy questions:
|
|||||||
|
|
||||||
## Philosophy
|
## Philosophy
|
||||||
|
|
||||||
Privacy isn't just a feature—it's the entire point. bitchat proves that modern communication doesn't require surrendering your privacy. No accounts, no servers, no surveillance. Just people talking freely.
|
Privacy isn't just a feature—it's the entire point. bitchat proves that modern communication doesn't require surrendering your privacy. No accounts, no company servers, no analytics. Just people talking freely.
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
|
|||||||
+10
-8
@@ -17,10 +17,8 @@ let package = Package(
|
|||||||
],
|
],
|
||||||
dependencies:[
|
dependencies:[
|
||||||
.package(path: "localPackages/Arti"),
|
.package(path: "localPackages/Arti"),
|
||||||
.package(path: "localPackages/Noise"),
|
|
||||||
.package(path: "localPackages/BitFoundation"),
|
.package(path: "localPackages/BitFoundation"),
|
||||||
.package(path: "localPackages/BitLogger"),
|
.package(path: "localPackages/BitLogger"),
|
||||||
.package(path: "localPackages/Nostr"),
|
|
||||||
.package(url: "https://github.com/21-DOT-DEV/swift-secp256k1", exact: "0.21.1")
|
.package(url: "https://github.com/21-DOT-DEV/swift-secp256k1", exact: "0.21.1")
|
||||||
],
|
],
|
||||||
targets: [
|
targets: [
|
||||||
@@ -30,8 +28,6 @@ let package = Package(
|
|||||||
.product(name: "P256K", package: "swift-secp256k1"),
|
.product(name: "P256K", package: "swift-secp256k1"),
|
||||||
.product(name: "BitFoundation", package: "BitFoundation"),
|
.product(name: "BitFoundation", package: "BitFoundation"),
|
||||||
.product(name: "BitLogger", package: "BitLogger"),
|
.product(name: "BitLogger", package: "BitLogger"),
|
||||||
.product(name: "Noise", package: "Noise"),
|
|
||||||
.product(name: "Nostr", package: "Nostr"),
|
|
||||||
.product(name: "Tor", package: "Arti")
|
.product(name: "Tor", package: "Arti")
|
||||||
],
|
],
|
||||||
path: "bitchat",
|
path: "bitchat",
|
||||||
@@ -52,16 +48,22 @@ let package = Package(
|
|||||||
name: "bitchatTests",
|
name: "bitchatTests",
|
||||||
dependencies: [
|
dependencies: [
|
||||||
"bitchat",
|
"bitchat",
|
||||||
.product(name: "BitFoundation", package: "BitFoundation"),
|
.product(name: "BitFoundation", package: "BitFoundation")
|
||||||
.product(name: "Nostr", package: "Nostr")
|
|
||||||
],
|
],
|
||||||
path: "bitchatTests",
|
path: "bitchatTests",
|
||||||
exclude: [
|
exclude: [
|
||||||
"Info.plist",
|
"Info.plist",
|
||||||
"README.md"
|
"README.md",
|
||||||
|
// CI perf gate data (read by scripts/check-perf-floors.sh),
|
||||||
|
// not a test resource.
|
||||||
|
"Performance/perf-floors.json"
|
||||||
],
|
],
|
||||||
resources: [
|
resources: [
|
||||||
.process("Localization")
|
.process("Localization"),
|
||||||
|
// Only the vector fixture: declaring the whole "Noise"
|
||||||
|
// directory would claim its .swift test files as resources
|
||||||
|
// and silently drop them from compilation.
|
||||||
|
.process("Noise/NoiseTestVectors.json")
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
]
|
]
|
||||||
|
|||||||
Generated
+11
-55
@@ -10,10 +10,6 @@
|
|||||||
17901751FD8010AFC8E750F2 /* bitchatShareExtension.appex in Embed Foundation Extensions */ = {isa = PBXBuildFile; fileRef = 61F92EBA29C47C0FCC482F1F /* bitchatShareExtension.appex */; settings = {ATTRIBUTES = (RemoveHeadersOnCopy, ); }; };
|
17901751FD8010AFC8E750F2 /* bitchatShareExtension.appex in Embed Foundation Extensions */ = {isa = PBXBuildFile; fileRef = 61F92EBA29C47C0FCC482F1F /* bitchatShareExtension.appex */; settings = {ATTRIBUTES = (RemoveHeadersOnCopy, ); }; };
|
||||||
3EE336D150427F736F32B56C /* P256K in Frameworks */ = {isa = PBXBuildFile; productRef = B1D9136AA0083366353BFA2F /* P256K */; };
|
3EE336D150427F736F32B56C /* P256K in Frameworks */ = {isa = PBXBuildFile; productRef = B1D9136AA0083366353BFA2F /* P256K */; };
|
||||||
885BBED78092484A5B069461 /* P256K in Frameworks */ = {isa = PBXBuildFile; productRef = 4EB6BA1B8464F1EA38F4E286 /* P256K */; };
|
885BBED78092484A5B069461 /* P256K in Frameworks */ = {isa = PBXBuildFile; productRef = 4EB6BA1B8464F1EA38F4E286 /* P256K */; };
|
||||||
A63163B62F80CB2500B8B128 /* Noise in Frameworks */ = {isa = PBXBuildFile; productRef = A63163B52F80CB2500B8B128 /* Noise */; };
|
|
||||||
A63163B82F80CB2D00B8B128 /* Noise in Frameworks */ = {isa = PBXBuildFile; productRef = A63163B72F80CB2D00B8B128 /* Noise */; };
|
|
||||||
A63180832F8103AB00B8B128 /* Nostr in Frameworks */ = {isa = PBXBuildFile; productRef = A63180822F8103AB00B8B128 /* Nostr */; };
|
|
||||||
A63180852F8103B600B8B128 /* Nostr in Frameworks */ = {isa = PBXBuildFile; productRef = A63180842F8103B600B8B128 /* Nostr */; };
|
|
||||||
A6BCF9482F80953E001CF9B9 /* BitFoundation in Frameworks */ = {isa = PBXBuildFile; productRef = A6BCF9472F80953E001CF9B9 /* BitFoundation */; };
|
A6BCF9482F80953E001CF9B9 /* BitFoundation in Frameworks */ = {isa = PBXBuildFile; productRef = A6BCF9472F80953E001CF9B9 /* BitFoundation */; };
|
||||||
A6BCF94A2F809550001CF9B9 /* BitFoundation in Frameworks */ = {isa = PBXBuildFile; productRef = A6BCF9492F809550001CF9B9 /* BitFoundation */; };
|
A6BCF94A2F809550001CF9B9 /* BitFoundation in Frameworks */ = {isa = PBXBuildFile; productRef = A6BCF9492F809550001CF9B9 /* BitFoundation */; };
|
||||||
A6E3E5702E77036A0032EA8A /* BitLogger in Frameworks */ = {isa = PBXBuildFile; productRef = A6E3E56F2E77036A0032EA8A /* BitLogger */; };
|
A6E3E5702E77036A0032EA8A /* BitLogger in Frameworks */ = {isa = PBXBuildFile; productRef = A6E3E56F2E77036A0032EA8A /* BitLogger */; };
|
||||||
@@ -160,9 +156,7 @@
|
|||||||
isa = PBXFrameworksBuildPhase;
|
isa = PBXFrameworksBuildPhase;
|
||||||
files = (
|
files = (
|
||||||
A6E3E5722E7703760032EA8A /* BitLogger in Frameworks */,
|
A6E3E5722E7703760032EA8A /* BitLogger in Frameworks */,
|
||||||
A63163B82F80CB2D00B8B128 /* Noise in Frameworks */,
|
|
||||||
3EE336D150427F736F32B56C /* P256K in Frameworks */,
|
3EE336D150427F736F32B56C /* P256K in Frameworks */,
|
||||||
A63180852F8103B600B8B128 /* Nostr in Frameworks */,
|
|
||||||
A6E3EA812E7706A80032EA8A /* Tor in Frameworks */,
|
A6E3EA812E7706A80032EA8A /* Tor in Frameworks */,
|
||||||
A6BCF94A2F809550001CF9B9 /* BitFoundation in Frameworks */,
|
A6BCF94A2F809550001CF9B9 /* BitFoundation in Frameworks */,
|
||||||
);
|
);
|
||||||
@@ -171,9 +165,7 @@
|
|||||||
isa = PBXFrameworksBuildPhase;
|
isa = PBXFrameworksBuildPhase;
|
||||||
files = (
|
files = (
|
||||||
A6E3E5702E77036A0032EA8A /* BitLogger in Frameworks */,
|
A6E3E5702E77036A0032EA8A /* BitLogger in Frameworks */,
|
||||||
A63163B62F80CB2500B8B128 /* Noise in Frameworks */,
|
|
||||||
885BBED78092484A5B069461 /* P256K in Frameworks */,
|
885BBED78092484A5B069461 /* P256K in Frameworks */,
|
||||||
A63180832F8103AB00B8B128 /* Nostr in Frameworks */,
|
|
||||||
A6E3EA7F2E7706720032EA8A /* Tor in Frameworks */,
|
A6E3EA7F2E7706720032EA8A /* Tor in Frameworks */,
|
||||||
A6BCF9482F80953E001CF9B9 /* BitFoundation in Frameworks */,
|
A6BCF9482F80953E001CF9B9 /* BitFoundation in Frameworks */,
|
||||||
);
|
);
|
||||||
@@ -236,8 +228,6 @@
|
|||||||
A6E3E5712E7703760032EA8A /* BitLogger */,
|
A6E3E5712E7703760032EA8A /* BitLogger */,
|
||||||
A6E3EA802E7706A80032EA8A /* Tor */,
|
A6E3EA802E7706A80032EA8A /* Tor */,
|
||||||
A6BCF9492F809550001CF9B9 /* BitFoundation */,
|
A6BCF9492F809550001CF9B9 /* BitFoundation */,
|
||||||
A63163B72F80CB2D00B8B128 /* Noise */,
|
|
||||||
A63180842F8103B600B8B128 /* Nostr */,
|
|
||||||
);
|
);
|
||||||
productName = bitchat_macOS;
|
productName = bitchat_macOS;
|
||||||
productReference = 8F3A7C058C2C8E1A06C8CF8B /* bitchat.app */;
|
productReference = 8F3A7C058C2C8E1A06C8CF8B /* bitchat.app */;
|
||||||
@@ -319,8 +309,6 @@
|
|||||||
A6E3E56F2E77036A0032EA8A /* BitLogger */,
|
A6E3E56F2E77036A0032EA8A /* BitLogger */,
|
||||||
A6E3EA7E2E7706720032EA8A /* Tor */,
|
A6E3EA7E2E7706720032EA8A /* Tor */,
|
||||||
A6BCF9472F80953E001CF9B9 /* BitFoundation */,
|
A6BCF9472F80953E001CF9B9 /* BitFoundation */,
|
||||||
A63163B52F80CB2500B8B128 /* Noise */,
|
|
||||||
A63180822F8103AB00B8B128 /* Nostr */,
|
|
||||||
);
|
);
|
||||||
productName = bitchat_iOS;
|
productName = bitchat_iOS;
|
||||||
productReference = 96D0D41CA19EE5A772AA8434 /* bitchat.app */;
|
productReference = 96D0D41CA19EE5A772AA8434 /* bitchat.app */;
|
||||||
@@ -333,7 +321,7 @@
|
|||||||
isa = PBXProject;
|
isa = PBXProject;
|
||||||
attributes = {
|
attributes = {
|
||||||
BuildIndependentTargetsInParallel = YES;
|
BuildIndependentTargetsInParallel = YES;
|
||||||
LastUpgradeCheck = 1640;
|
LastUpgradeCheck = 2650;
|
||||||
};
|
};
|
||||||
buildConfigurationList = 3EA424CBD51200895D361189 /* Build configuration list for PBXProject "bitchat" */;
|
buildConfigurationList = 3EA424CBD51200895D361189 /* Build configuration list for PBXProject "bitchat" */;
|
||||||
developmentRegion = en;
|
developmentRegion = en;
|
||||||
@@ -363,8 +351,6 @@
|
|||||||
A6E3E56E2E77036A0032EA8A /* XCLocalSwiftPackageReference "localPackages/BitLogger" */,
|
A6E3E56E2E77036A0032EA8A /* XCLocalSwiftPackageReference "localPackages/BitLogger" */,
|
||||||
A6E3EA7D2E7706720032EA8A /* XCLocalSwiftPackageReference "localPackages/Arti" */,
|
A6E3EA7D2E7706720032EA8A /* XCLocalSwiftPackageReference "localPackages/Arti" */,
|
||||||
A6BCF9462F80953E001CF9B9 /* XCLocalSwiftPackageReference "localPackages/BitFoundation" */,
|
A6BCF9462F80953E001CF9B9 /* XCLocalSwiftPackageReference "localPackages/BitFoundation" */,
|
||||||
A63163B42F80CB2500B8B128 /* XCLocalSwiftPackageReference "localPackages/Noise" */,
|
|
||||||
A63180812F8103AB00B8B128 /* XCLocalSwiftPackageReference "localPackages/Nostr" */,
|
|
||||||
);
|
);
|
||||||
preferredProjectObjectVersion = 90;
|
preferredProjectObjectVersion = 90;
|
||||||
projectDirPath = "";
|
projectDirPath = "";
|
||||||
@@ -460,7 +446,6 @@
|
|||||||
CODE_SIGNING_ALLOWED = YES;
|
CODE_SIGNING_ALLOWED = YES;
|
||||||
CODE_SIGNING_REQUIRED = YES;
|
CODE_SIGNING_REQUIRED = YES;
|
||||||
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
||||||
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
|
||||||
INFOPLIST_FILE = bitchatTests/Info.plist;
|
INFOPLIST_FILE = bitchatTests/Info.plist;
|
||||||
IPHONEOS_DEPLOYMENT_TARGET = "$(IPHONEOS_DEPLOYMENT_TARGET)";
|
IPHONEOS_DEPLOYMENT_TARGET = "$(IPHONEOS_DEPLOYMENT_TARGET)";
|
||||||
LD_RUNPATH_SEARCH_PATHS = (
|
LD_RUNPATH_SEARCH_PATHS = (
|
||||||
@@ -485,7 +470,6 @@
|
|||||||
CODE_SIGNING_ALLOWED = YES;
|
CODE_SIGNING_ALLOWED = YES;
|
||||||
CODE_SIGNING_REQUIRED = YES;
|
CODE_SIGNING_REQUIRED = YES;
|
||||||
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
||||||
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
|
||||||
INFOPLIST_FILE = bitchatTests/Info.plist;
|
INFOPLIST_FILE = bitchatTests/Info.plist;
|
||||||
IPHONEOS_DEPLOYMENT_TARGET = "$(IPHONEOS_DEPLOYMENT_TARGET)";
|
IPHONEOS_DEPLOYMENT_TARGET = "$(IPHONEOS_DEPLOYMENT_TARGET)";
|
||||||
LD_RUNPATH_SEARCH_PATHS = (
|
LD_RUNPATH_SEARCH_PATHS = (
|
||||||
@@ -512,7 +496,6 @@
|
|||||||
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
||||||
COMBINE_HIDPI_IMAGES = YES;
|
COMBINE_HIDPI_IMAGES = YES;
|
||||||
DEAD_CODE_STRIPPING = YES;
|
DEAD_CODE_STRIPPING = YES;
|
||||||
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
|
||||||
INFOPLIST_FILE = bitchatTests/Info.plist;
|
INFOPLIST_FILE = bitchatTests/Info.plist;
|
||||||
LD_RUNPATH_SEARCH_PATHS = (
|
LD_RUNPATH_SEARCH_PATHS = (
|
||||||
"$(inherited)",
|
"$(inherited)",
|
||||||
@@ -537,7 +520,6 @@
|
|||||||
CODE_SIGN_ALLOW_ENTITLEMENTS_MODIFICATION = YES;
|
CODE_SIGN_ALLOW_ENTITLEMENTS_MODIFICATION = YES;
|
||||||
CODE_SIGN_ENTITLEMENTS = bitchatShareExtension/bitchatShareExtension.entitlements;
|
CODE_SIGN_ENTITLEMENTS = bitchatShareExtension/bitchatShareExtension.entitlements;
|
||||||
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
||||||
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
|
||||||
INFOPLIST_FILE = bitchatShareExtension/Info.plist;
|
INFOPLIST_FILE = bitchatShareExtension/Info.plist;
|
||||||
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
||||||
IPHONEOS_DEPLOYMENT_TARGET = "$(IPHONEOS_DEPLOYMENT_TARGET)";
|
IPHONEOS_DEPLOYMENT_TARGET = "$(IPHONEOS_DEPLOYMENT_TARGET)";
|
||||||
@@ -570,7 +552,6 @@
|
|||||||
CODE_SIGN_ENTITLEMENTS = bitchat/bitchat.entitlements;
|
CODE_SIGN_ENTITLEMENTS = bitchat/bitchat.entitlements;
|
||||||
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
||||||
DEVELOPMENT_ASSET_PATHS = bitchat/_PreviewHelpers;
|
DEVELOPMENT_ASSET_PATHS = bitchat/_PreviewHelpers;
|
||||||
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
|
||||||
ENABLE_PREVIEWS = NO;
|
ENABLE_PREVIEWS = NO;
|
||||||
INFOPLIST_FILE = bitchat/Info.plist;
|
INFOPLIST_FILE = bitchat/Info.plist;
|
||||||
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
||||||
@@ -580,7 +561,7 @@
|
|||||||
"$(inherited)",
|
"$(inherited)",
|
||||||
"@executable_path/Frameworks",
|
"@executable_path/Frameworks",
|
||||||
);
|
);
|
||||||
MARKETING_VERSION = 1.5.1;
|
MARKETING_VERSION = 1.5.2;
|
||||||
PRODUCT_BUNDLE_IDENTIFIER = "$(PRODUCT_BUNDLE_IDENTIFIER)";
|
PRODUCT_BUNDLE_IDENTIFIER = "$(PRODUCT_BUNDLE_IDENTIFIER)";
|
||||||
PRODUCT_NAME = bitchat;
|
PRODUCT_NAME = bitchat;
|
||||||
SDKROOT = iphoneos;
|
SDKROOT = iphoneos;
|
||||||
@@ -604,7 +585,6 @@
|
|||||||
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
||||||
COMBINE_HIDPI_IMAGES = YES;
|
COMBINE_HIDPI_IMAGES = YES;
|
||||||
DEAD_CODE_STRIPPING = YES;
|
DEAD_CODE_STRIPPING = YES;
|
||||||
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
|
||||||
INFOPLIST_FILE = bitchatTests/Info.plist;
|
INFOPLIST_FILE = bitchatTests/Info.plist;
|
||||||
LD_RUNPATH_SEARCH_PATHS = (
|
LD_RUNPATH_SEARCH_PATHS = (
|
||||||
"$(inherited)",
|
"$(inherited)",
|
||||||
@@ -631,7 +611,6 @@
|
|||||||
CODE_SIGN_ENTITLEMENTS = bitchat/bitchat.entitlements;
|
CODE_SIGN_ENTITLEMENTS = bitchat/bitchat.entitlements;
|
||||||
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
||||||
DEVELOPMENT_ASSET_PATHS = bitchat/_PreviewHelpers;
|
DEVELOPMENT_ASSET_PATHS = bitchat/_PreviewHelpers;
|
||||||
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
|
||||||
ENABLE_PREVIEWS = YES;
|
ENABLE_PREVIEWS = YES;
|
||||||
INFOPLIST_FILE = bitchat/Info.plist;
|
INFOPLIST_FILE = bitchat/Info.plist;
|
||||||
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
||||||
@@ -641,7 +620,7 @@
|
|||||||
"$(inherited)",
|
"$(inherited)",
|
||||||
"@executable_path/Frameworks",
|
"@executable_path/Frameworks",
|
||||||
);
|
);
|
||||||
MARKETING_VERSION = 1.5.1;
|
MARKETING_VERSION = 1.5.2;
|
||||||
PRODUCT_BUNDLE_IDENTIFIER = "$(PRODUCT_BUNDLE_IDENTIFIER)";
|
PRODUCT_BUNDLE_IDENTIFIER = "$(PRODUCT_BUNDLE_IDENTIFIER)";
|
||||||
PRODUCT_NAME = bitchat;
|
PRODUCT_NAME = bitchat;
|
||||||
SDKROOT = iphoneos;
|
SDKROOT = iphoneos;
|
||||||
@@ -667,7 +646,6 @@
|
|||||||
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
||||||
COMBINE_HIDPI_IMAGES = YES;
|
COMBINE_HIDPI_IMAGES = YES;
|
||||||
DEAD_CODE_STRIPPING = YES;
|
DEAD_CODE_STRIPPING = YES;
|
||||||
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
|
||||||
ENABLE_PREVIEWS = YES;
|
ENABLE_PREVIEWS = YES;
|
||||||
INFOPLIST_FILE = bitchat/Info.plist;
|
INFOPLIST_FILE = bitchat/Info.plist;
|
||||||
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
||||||
@@ -677,7 +655,7 @@
|
|||||||
"@executable_path/../Frameworks",
|
"@executable_path/../Frameworks",
|
||||||
);
|
);
|
||||||
MACOSX_DEPLOYMENT_TARGET = "$(MACOSX_DEPLOYMENT_TARGET)";
|
MACOSX_DEPLOYMENT_TARGET = "$(MACOSX_DEPLOYMENT_TARGET)";
|
||||||
MARKETING_VERSION = 1.5.1;
|
MARKETING_VERSION = 1.5.2;
|
||||||
PRODUCT_BUNDLE_IDENTIFIER = "$(PRODUCT_BUNDLE_IDENTIFIER)";
|
PRODUCT_BUNDLE_IDENTIFIER = "$(PRODUCT_BUNDLE_IDENTIFIER)";
|
||||||
PRODUCT_NAME = bitchat;
|
PRODUCT_NAME = bitchat;
|
||||||
REGISTER_APP_GROUPS = YES;
|
REGISTER_APP_GROUPS = YES;
|
||||||
@@ -690,6 +668,7 @@
|
|||||||
isa = XCBuildConfiguration;
|
isa = XCBuildConfiguration;
|
||||||
buildSettings = {
|
buildSettings = {
|
||||||
ALWAYS_SEARCH_USER_PATHS = NO;
|
ALWAYS_SEARCH_USER_PATHS = NO;
|
||||||
|
CLANG_ANALYZER_LOCALIZABILITY_NONLOCALIZED = YES;
|
||||||
CLANG_ANALYZER_NONNULL = YES;
|
CLANG_ANALYZER_NONNULL = YES;
|
||||||
CLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;
|
CLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;
|
||||||
CLANG_CXX_LANGUAGE_STANDARD = "gnu++14";
|
CLANG_CXX_LANGUAGE_STANDARD = "gnu++14";
|
||||||
@@ -723,6 +702,7 @@
|
|||||||
CURRENT_PROJECT_VERSION = "$(CURRENT_PROJECT_VERSION)";
|
CURRENT_PROJECT_VERSION = "$(CURRENT_PROJECT_VERSION)";
|
||||||
DEAD_CODE_STRIPPING = YES;
|
DEAD_CODE_STRIPPING = YES;
|
||||||
DEBUG_INFORMATION_FORMAT = "dwarf-with-dsym";
|
DEBUG_INFORMATION_FORMAT = "dwarf-with-dsym";
|
||||||
|
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
||||||
ENABLE_NS_ASSERTIONS = NO;
|
ENABLE_NS_ASSERTIONS = NO;
|
||||||
ENABLE_STRICT_OBJC_MSGSEND = YES;
|
ENABLE_STRICT_OBJC_MSGSEND = YES;
|
||||||
ENABLE_USER_SCRIPT_SANDBOXING = YES;
|
ENABLE_USER_SCRIPT_SANDBOXING = YES;
|
||||||
@@ -740,6 +720,7 @@
|
|||||||
MTL_ENABLE_DEBUG_INFO = NO;
|
MTL_ENABLE_DEBUG_INFO = NO;
|
||||||
MTL_FAST_MATH = YES;
|
MTL_FAST_MATH = YES;
|
||||||
PRODUCT_NAME = "$(TARGET_NAME)";
|
PRODUCT_NAME = "$(TARGET_NAME)";
|
||||||
|
STRING_CATALOG_GENERATE_SYMBOLS = NO;
|
||||||
SWIFT_COMPILATION_MODE = wholemodule;
|
SWIFT_COMPILATION_MODE = wholemodule;
|
||||||
SWIFT_OPTIMIZATION_LEVEL = "-O";
|
SWIFT_OPTIMIZATION_LEVEL = "-O";
|
||||||
SWIFT_VERSION = "$(SWIFT_VERSION)";
|
SWIFT_VERSION = "$(SWIFT_VERSION)";
|
||||||
@@ -759,7 +740,6 @@
|
|||||||
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
||||||
COMBINE_HIDPI_IMAGES = YES;
|
COMBINE_HIDPI_IMAGES = YES;
|
||||||
DEAD_CODE_STRIPPING = YES;
|
DEAD_CODE_STRIPPING = YES;
|
||||||
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
|
||||||
ENABLE_PREVIEWS = NO;
|
ENABLE_PREVIEWS = NO;
|
||||||
INFOPLIST_FILE = bitchat/Info.plist;
|
INFOPLIST_FILE = bitchat/Info.plist;
|
||||||
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
||||||
@@ -769,7 +749,7 @@
|
|||||||
"@executable_path/../Frameworks",
|
"@executable_path/../Frameworks",
|
||||||
);
|
);
|
||||||
MACOSX_DEPLOYMENT_TARGET = "$(MACOSX_DEPLOYMENT_TARGET)";
|
MACOSX_DEPLOYMENT_TARGET = "$(MACOSX_DEPLOYMENT_TARGET)";
|
||||||
MARKETING_VERSION = 1.5.1;
|
MARKETING_VERSION = 1.5.2;
|
||||||
PRODUCT_BUNDLE_IDENTIFIER = "$(PRODUCT_BUNDLE_IDENTIFIER)";
|
PRODUCT_BUNDLE_IDENTIFIER = "$(PRODUCT_BUNDLE_IDENTIFIER)";
|
||||||
PRODUCT_NAME = bitchat;
|
PRODUCT_NAME = bitchat;
|
||||||
REGISTER_APP_GROUPS = YES;
|
REGISTER_APP_GROUPS = YES;
|
||||||
@@ -782,6 +762,7 @@
|
|||||||
isa = XCBuildConfiguration;
|
isa = XCBuildConfiguration;
|
||||||
buildSettings = {
|
buildSettings = {
|
||||||
ALWAYS_SEARCH_USER_PATHS = NO;
|
ALWAYS_SEARCH_USER_PATHS = NO;
|
||||||
|
CLANG_ANALYZER_LOCALIZABILITY_NONLOCALIZED = YES;
|
||||||
CLANG_ANALYZER_NONNULL = YES;
|
CLANG_ANALYZER_NONNULL = YES;
|
||||||
CLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;
|
CLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;
|
||||||
CLANG_CXX_LANGUAGE_STANDARD = "gnu++14";
|
CLANG_CXX_LANGUAGE_STANDARD = "gnu++14";
|
||||||
@@ -815,6 +796,7 @@
|
|||||||
CURRENT_PROJECT_VERSION = "$(CURRENT_PROJECT_VERSION)";
|
CURRENT_PROJECT_VERSION = "$(CURRENT_PROJECT_VERSION)";
|
||||||
DEAD_CODE_STRIPPING = YES;
|
DEAD_CODE_STRIPPING = YES;
|
||||||
DEBUG_INFORMATION_FORMAT = dwarf;
|
DEBUG_INFORMATION_FORMAT = dwarf;
|
||||||
|
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
||||||
ENABLE_STRICT_OBJC_MSGSEND = YES;
|
ENABLE_STRICT_OBJC_MSGSEND = YES;
|
||||||
ENABLE_TESTABILITY = YES;
|
ENABLE_TESTABILITY = YES;
|
||||||
ENABLE_USER_SCRIPT_SANDBOXING = YES;
|
ENABLE_USER_SCRIPT_SANDBOXING = YES;
|
||||||
@@ -839,6 +821,7 @@
|
|||||||
MTL_FAST_MATH = YES;
|
MTL_FAST_MATH = YES;
|
||||||
ONLY_ACTIVE_ARCH = YES;
|
ONLY_ACTIVE_ARCH = YES;
|
||||||
PRODUCT_NAME = "$(TARGET_NAME)";
|
PRODUCT_NAME = "$(TARGET_NAME)";
|
||||||
|
STRING_CATALOG_GENERATE_SYMBOLS = NO;
|
||||||
SWIFT_ACTIVE_COMPILATION_CONDITIONS = DEBUG;
|
SWIFT_ACTIVE_COMPILATION_CONDITIONS = DEBUG;
|
||||||
SWIFT_OPTIMIZATION_LEVEL = "-Onone";
|
SWIFT_OPTIMIZATION_LEVEL = "-Onone";
|
||||||
SWIFT_VERSION = "$(SWIFT_VERSION)";
|
SWIFT_VERSION = "$(SWIFT_VERSION)";
|
||||||
@@ -855,7 +838,6 @@
|
|||||||
CODE_SIGN_ALLOW_ENTITLEMENTS_MODIFICATION = YES;
|
CODE_SIGN_ALLOW_ENTITLEMENTS_MODIFICATION = YES;
|
||||||
CODE_SIGN_ENTITLEMENTS = bitchatShareExtension/bitchatShareExtension.entitlements;
|
CODE_SIGN_ENTITLEMENTS = bitchatShareExtension/bitchatShareExtension.entitlements;
|
||||||
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
CODE_SIGN_STYLE = "$(CODE_SIGN_STYLE)";
|
||||||
DEVELOPMENT_TEAM = "$(DEVELOPMENT_TEAM)";
|
|
||||||
INFOPLIST_FILE = bitchatShareExtension/Info.plist;
|
INFOPLIST_FILE = bitchatShareExtension/Info.plist;
|
||||||
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
INFOPLIST_KEY_CFBundleDisplayName = bitchat;
|
||||||
IPHONEOS_DEPLOYMENT_TARGET = "$(IPHONEOS_DEPLOYMENT_TARGET)";
|
IPHONEOS_DEPLOYMENT_TARGET = "$(IPHONEOS_DEPLOYMENT_TARGET)";
|
||||||
@@ -930,14 +912,6 @@
|
|||||||
/* End XCConfigurationList section */
|
/* End XCConfigurationList section */
|
||||||
|
|
||||||
/* Begin XCLocalSwiftPackageReference section */
|
/* Begin XCLocalSwiftPackageReference section */
|
||||||
A63163B42F80CB2500B8B128 /* XCLocalSwiftPackageReference "localPackages/Noise" */ = {
|
|
||||||
isa = XCLocalSwiftPackageReference;
|
|
||||||
relativePath = localPackages/Noise;
|
|
||||||
};
|
|
||||||
A63180812F8103AB00B8B128 /* XCLocalSwiftPackageReference "localPackages/Nostr" */ = {
|
|
||||||
isa = XCLocalSwiftPackageReference;
|
|
||||||
relativePath = localPackages/Nostr;
|
|
||||||
};
|
|
||||||
A6BCF9462F80953E001CF9B9 /* XCLocalSwiftPackageReference "localPackages/BitFoundation" */ = {
|
A6BCF9462F80953E001CF9B9 /* XCLocalSwiftPackageReference "localPackages/BitFoundation" */ = {
|
||||||
isa = XCLocalSwiftPackageReference;
|
isa = XCLocalSwiftPackageReference;
|
||||||
relativePath = localPackages/BitFoundation;
|
relativePath = localPackages/BitFoundation;
|
||||||
@@ -969,24 +943,6 @@
|
|||||||
package = B8C407587481BBB190741C93 /* XCRemoteSwiftPackageReference "swift-secp256k1" */;
|
package = B8C407587481BBB190741C93 /* XCRemoteSwiftPackageReference "swift-secp256k1" */;
|
||||||
productName = P256K;
|
productName = P256K;
|
||||||
};
|
};
|
||||||
A63163B52F80CB2500B8B128 /* Noise */ = {
|
|
||||||
isa = XCSwiftPackageProductDependency;
|
|
||||||
productName = Noise;
|
|
||||||
};
|
|
||||||
A63163B72F80CB2D00B8B128 /* Noise */ = {
|
|
||||||
isa = XCSwiftPackageProductDependency;
|
|
||||||
package = A63163B42F80CB2500B8B128 /* XCLocalSwiftPackageReference "localPackages/Noise" */;
|
|
||||||
productName = Noise;
|
|
||||||
};
|
|
||||||
A63180822F8103AB00B8B128 /* Nostr */ = {
|
|
||||||
isa = XCSwiftPackageProductDependency;
|
|
||||||
productName = Nostr;
|
|
||||||
};
|
|
||||||
A63180842F8103B600B8B128 /* Nostr */ = {
|
|
||||||
isa = XCSwiftPackageProductDependency;
|
|
||||||
package = A63180812F8103AB00B8B128 /* XCLocalSwiftPackageReference "localPackages/Nostr" */;
|
|
||||||
productName = Nostr;
|
|
||||||
};
|
|
||||||
A6BCF9472F80953E001CF9B9 /* BitFoundation */ = {
|
A6BCF9472F80953E001CF9B9 /* BitFoundation */ = {
|
||||||
isa = XCSwiftPackageProductDependency;
|
isa = XCSwiftPackageProductDependency;
|
||||||
productName = BitFoundation;
|
productName = BitFoundation;
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
<?xml version="1.0" encoding="UTF-8"?>
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
<Scheme
|
<Scheme
|
||||||
LastUpgradeVersion = "1640"
|
LastUpgradeVersion = "2650"
|
||||||
version = "1.3">
|
version = "1.3">
|
||||||
<BuildAction
|
<BuildAction
|
||||||
parallelizeBuildables = "YES"
|
parallelizeBuildables = "YES"
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
<?xml version="1.0" encoding="UTF-8"?>
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
<Scheme
|
<Scheme
|
||||||
LastUpgradeVersion = "1640"
|
LastUpgradeVersion = "2650"
|
||||||
version = "1.3">
|
version = "1.3">
|
||||||
<BuildAction
|
<BuildAction
|
||||||
parallelizeBuildables = "YES"
|
parallelizeBuildables = "YES"
|
||||||
|
|||||||
@@ -0,0 +1,102 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum SharedContentKind: String, Sendable, Equatable {
|
||||||
|
case text
|
||||||
|
case url
|
||||||
|
}
|
||||||
|
|
||||||
|
enum RuntimeScenePhase: String, Sendable, Equatable {
|
||||||
|
case active
|
||||||
|
case inactive
|
||||||
|
case background
|
||||||
|
}
|
||||||
|
|
||||||
|
enum TorLifecycleEvent: String, Sendable, Equatable {
|
||||||
|
case willStart
|
||||||
|
case willRestart
|
||||||
|
case didBecomeReady
|
||||||
|
case preferenceChanged
|
||||||
|
}
|
||||||
|
|
||||||
|
enum AppEvent: Sendable, Equatable {
|
||||||
|
case launched
|
||||||
|
case startupCompleted
|
||||||
|
case scenePhaseChanged(RuntimeScenePhase)
|
||||||
|
case openedURL(String)
|
||||||
|
case sharedContentAccepted(SharedContentKind)
|
||||||
|
case notificationOpened(peerID: PeerID?)
|
||||||
|
case deepLinkOpened(String)
|
||||||
|
case torLifecycleChanged(TorLifecycleEvent)
|
||||||
|
case nostrRelayConnectionChanged(Bool)
|
||||||
|
case terminationRequested
|
||||||
|
}
|
||||||
|
|
||||||
|
actor AppEventStream {
|
||||||
|
private var continuations: [UUID: AsyncStream<AppEvent>.Continuation] = [:]
|
||||||
|
|
||||||
|
func stream() -> AsyncStream<AppEvent> {
|
||||||
|
let id = UUID()
|
||||||
|
return AsyncStream { continuation in
|
||||||
|
continuations[id] = continuation
|
||||||
|
continuation.onTermination = { [id] _ in
|
||||||
|
Task {
|
||||||
|
await self.removeContinuation(id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func emit(_ event: AppEvent) {
|
||||||
|
for continuation in continuations.values {
|
||||||
|
continuation.yield(event)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func finish() {
|
||||||
|
for continuation in continuations.values {
|
||||||
|
continuation.finish()
|
||||||
|
}
|
||||||
|
continuations.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
private func removeContinuation(_ id: UUID) {
|
||||||
|
continuations.removeValue(forKey: id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Identity key for a direct conversation. Equality and hashing use the
|
||||||
|
/// canonical `id` only; `routingPeerID` carries the transport-level peer ID
|
||||||
|
/// the conversation is keyed under (see `ConversationID.directPeer`).
|
||||||
|
struct PeerHandle: Sendable, Identifiable {
|
||||||
|
let id: String
|
||||||
|
let routingPeerID: PeerID
|
||||||
|
}
|
||||||
|
|
||||||
|
extension PeerHandle: Equatable {
|
||||||
|
static func == (lhs: PeerHandle, rhs: PeerHandle) -> Bool {
|
||||||
|
lhs.id == rhs.id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
extension PeerHandle: Hashable {
|
||||||
|
func hash(into hasher: inout Hasher) {
|
||||||
|
hasher.combine(id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
enum ConversationID: Hashable, Sendable {
|
||||||
|
case mesh
|
||||||
|
case geohash(String)
|
||||||
|
case direct(PeerHandle)
|
||||||
|
|
||||||
|
init(channelID: ChannelID) {
|
||||||
|
switch channelID {
|
||||||
|
case .mesh:
|
||||||
|
self = .mesh
|
||||||
|
case .location(let channel):
|
||||||
|
self = .geohash(channel.geohash.lowercased())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,100 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
|
import CoreBluetooth
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class AppChromeModel: ObservableObject {
|
||||||
|
@Published private(set) var hasUnreadPrivateMessages = false
|
||||||
|
@Published var nickname: String
|
||||||
|
@Published var showingFingerprintFor: PeerID?
|
||||||
|
@Published var isAppInfoPresented = false
|
||||||
|
@Published var isLocationChannelsSheetPresented = false
|
||||||
|
@Published var showBluetoothAlert = false
|
||||||
|
@Published var bluetoothAlertMessage = ""
|
||||||
|
@Published var bluetoothState: CBManagerState = .unknown
|
||||||
|
@Published var showScreenshotPrivacyWarning = false
|
||||||
|
|
||||||
|
private let chatViewModel: ChatViewModel
|
||||||
|
private var cancellables = Set<AnyCancellable>()
|
||||||
|
|
||||||
|
init(chatViewModel: ChatViewModel, privateInboxModel: PrivateInboxModel) {
|
||||||
|
self.chatViewModel = chatViewModel
|
||||||
|
self.nickname = chatViewModel.nickname
|
||||||
|
|
||||||
|
bind(privateInboxModel: privateInboxModel)
|
||||||
|
}
|
||||||
|
|
||||||
|
var shouldSuppressScreenshotNotification: Bool {
|
||||||
|
isLocationChannelsSheetPresented || isAppInfoPresented
|
||||||
|
}
|
||||||
|
|
||||||
|
func setNickname(_ nickname: String) {
|
||||||
|
self.nickname = nickname
|
||||||
|
if chatViewModel.nickname != nickname {
|
||||||
|
chatViewModel.nickname = nickname
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func validateAndSaveNickname() {
|
||||||
|
chatViewModel.validateAndSaveNickname()
|
||||||
|
if nickname != chatViewModel.nickname {
|
||||||
|
nickname = chatViewModel.nickname
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func openMostRelevantPrivateChat() {
|
||||||
|
chatViewModel.openMostRelevantPrivateChat()
|
||||||
|
}
|
||||||
|
|
||||||
|
func showFingerprint(for peerID: PeerID) {
|
||||||
|
showingFingerprintFor = peerID
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearFingerprint() {
|
||||||
|
showingFingerprintFor = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func presentAppInfo() {
|
||||||
|
isAppInfoPresented = true
|
||||||
|
}
|
||||||
|
|
||||||
|
func triggerScreenshotPrivacyWarning() {
|
||||||
|
showScreenshotPrivacyWarning = true
|
||||||
|
}
|
||||||
|
|
||||||
|
func panicClearAllData() {
|
||||||
|
chatViewModel.panicClearAllData()
|
||||||
|
}
|
||||||
|
|
||||||
|
private func bind(privateInboxModel: PrivateInboxModel) {
|
||||||
|
privateInboxModel.$unreadPeerIDs
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] unreadPeerIDs in
|
||||||
|
self?.hasUnreadPrivateMessages = !unreadPeerIDs.isEmpty
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
chatViewModel.$nickname
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] nickname in
|
||||||
|
guard let self, self.nickname != nickname else { return }
|
||||||
|
self.nickname = nickname
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
chatViewModel.$showBluetoothAlert
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$showBluetoothAlert)
|
||||||
|
|
||||||
|
chatViewModel.$bluetoothAlertMessage
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$bluetoothAlertMessage)
|
||||||
|
|
||||||
|
chatViewModel.$bluetoothState
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$bluetoothState)
|
||||||
|
|
||||||
|
hasUnreadPrivateMessages = !privateInboxModel.unreadPeerIDs.isEmpty
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,388 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
|
import Foundation
|
||||||
|
import SwiftUI
|
||||||
|
import Tor
|
||||||
|
import UserNotifications
|
||||||
|
#if os(iOS)
|
||||||
|
import UIKit
|
||||||
|
#elseif os(macOS)
|
||||||
|
import AppKit
|
||||||
|
#endif
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class AppRuntime: ObservableObject {
|
||||||
|
let chatViewModel: ChatViewModel
|
||||||
|
let events = AppEventStream()
|
||||||
|
/// Single source of truth for conversation message state and selection
|
||||||
|
/// (docs/CONVERSATION-STORE-DESIGN.md). Owned here; the feature models
|
||||||
|
/// and `ChatViewModel` observe and mutate it through its intent API.
|
||||||
|
let conversations: ConversationStore
|
||||||
|
let peerIdentityStore: PeerIdentityStore
|
||||||
|
let locationPresenceStore: LocationPresenceStore
|
||||||
|
let publicChatModel: PublicChatModel
|
||||||
|
let privateInboxModel: PrivateInboxModel
|
||||||
|
let privateConversationModel: PrivateConversationModel
|
||||||
|
let verificationModel: VerificationModel
|
||||||
|
let conversationUIModel: ConversationUIModel
|
||||||
|
let locationChannelsModel: LocationChannelsModel
|
||||||
|
let peerListModel: PeerListModel
|
||||||
|
let appChromeModel: AppChromeModel
|
||||||
|
|
||||||
|
private let idBridge: NostrIdentityBridge
|
||||||
|
private var cancellables = Set<AnyCancellable>()
|
||||||
|
private var started = false
|
||||||
|
private var lastNostrRelayConnectedState = false
|
||||||
|
private var didHandleInitialNostrConnection = false
|
||||||
|
|
||||||
|
#if os(iOS)
|
||||||
|
private var didHandleInitialActive = false
|
||||||
|
private var didEnterBackground = false
|
||||||
|
#endif
|
||||||
|
|
||||||
|
init(
|
||||||
|
keychain: KeychainManagerProtocol = KeychainManager(),
|
||||||
|
idBridge: NostrIdentityBridge = NostrIdentityBridge()
|
||||||
|
) {
|
||||||
|
self.idBridge = idBridge
|
||||||
|
let conversations = ConversationStore()
|
||||||
|
let peerIdentityStore = PeerIdentityStore()
|
||||||
|
let locationPresenceStore = LocationPresenceStore()
|
||||||
|
let locationManager = LocationChannelManager.shared
|
||||||
|
self.conversations = conversations
|
||||||
|
self.peerIdentityStore = peerIdentityStore
|
||||||
|
self.locationPresenceStore = locationPresenceStore
|
||||||
|
self.chatViewModel = ChatViewModel(
|
||||||
|
keychain: keychain,
|
||||||
|
idBridge: idBridge,
|
||||||
|
identityManager: SecureIdentityStateManager(keychain),
|
||||||
|
conversations: conversations,
|
||||||
|
peerIdentityStore: peerIdentityStore,
|
||||||
|
locationPresenceStore: locationPresenceStore,
|
||||||
|
locationManager: locationManager
|
||||||
|
)
|
||||||
|
self.publicChatModel = PublicChatModel(conversations: conversations)
|
||||||
|
self.privateInboxModel = PrivateInboxModel(conversations: conversations)
|
||||||
|
self.locationChannelsModel = LocationChannelsModel(manager: locationManager)
|
||||||
|
self.privateConversationModel = PrivateConversationModel(
|
||||||
|
chatViewModel: self.chatViewModel,
|
||||||
|
conversations: conversations,
|
||||||
|
locationChannelsModel: self.locationChannelsModel,
|
||||||
|
peerIdentityStore: peerIdentityStore
|
||||||
|
)
|
||||||
|
self.verificationModel = VerificationModel(
|
||||||
|
chatViewModel: self.chatViewModel,
|
||||||
|
privateConversationModel: self.privateConversationModel,
|
||||||
|
peerIdentityStore: peerIdentityStore
|
||||||
|
)
|
||||||
|
self.conversationUIModel = ConversationUIModel(
|
||||||
|
chatViewModel: self.chatViewModel,
|
||||||
|
privateConversationModel: self.privateConversationModel,
|
||||||
|
conversations: conversations
|
||||||
|
)
|
||||||
|
self.peerListModel = PeerListModel(
|
||||||
|
chatViewModel: self.chatViewModel,
|
||||||
|
conversations: conversations,
|
||||||
|
locationChannelsModel: self.locationChannelsModel,
|
||||||
|
peerIdentityStore: peerIdentityStore,
|
||||||
|
locationPresenceStore: locationPresenceStore
|
||||||
|
)
|
||||||
|
self.appChromeModel = AppChromeModel(
|
||||||
|
chatViewModel: self.chatViewModel,
|
||||||
|
privateInboxModel: self.privateInboxModel
|
||||||
|
)
|
||||||
|
|
||||||
|
GeoRelayDirectory.shared.prefetchIfNeeded()
|
||||||
|
bindRuntimeObservers()
|
||||||
|
NotificationDelegate.shared.runtime = self
|
||||||
|
}
|
||||||
|
|
||||||
|
func start() {
|
||||||
|
guard !started else {
|
||||||
|
checkForSharedContent()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
started = true
|
||||||
|
NotificationDelegate.shared.runtime = self
|
||||||
|
VerificationService.shared.configure(with: chatViewModel.meshService)
|
||||||
|
announceInitialTorStatusIfNeeded()
|
||||||
|
|
||||||
|
Task(priority: .utility) { [weak self] in
|
||||||
|
guard let self else { return }
|
||||||
|
let nickname = await MainActor.run { self.chatViewModel.nickname }
|
||||||
|
let npub = await MainActor.run {
|
||||||
|
try? self.idBridge.getCurrentNostrIdentity()?.npub
|
||||||
|
}
|
||||||
|
await MainActor.run {
|
||||||
|
_ = VerificationService.shared.buildMyQRString(nickname: nickname, npub: npub)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
NetworkActivationService.shared.start()
|
||||||
|
GeohashPresenceService.shared.start()
|
||||||
|
checkForSharedContent()
|
||||||
|
|
||||||
|
record(.launched)
|
||||||
|
record(.startupCompleted)
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleOpenURL(_ url: URL) {
|
||||||
|
record(.openedURL(url.absoluteString))
|
||||||
|
|
||||||
|
if url.scheme == "bitchat", url.host == "share" {
|
||||||
|
checkForSharedContent()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleDidBecomeActiveNotification() {
|
||||||
|
chatViewModel.handleDidBecomeActive()
|
||||||
|
checkForSharedContent()
|
||||||
|
}
|
||||||
|
|
||||||
|
#if os(macOS)
|
||||||
|
func handleMacDidBecomeActiveNotification() {
|
||||||
|
record(.scenePhaseChanged(.active))
|
||||||
|
chatViewModel.handleDidBecomeActive()
|
||||||
|
checkForSharedContent()
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if os(iOS)
|
||||||
|
func handleScenePhaseChange(_ newPhase: ScenePhase) {
|
||||||
|
switch newPhase {
|
||||||
|
case .background:
|
||||||
|
record(.scenePhaseChanged(.background))
|
||||||
|
TorManager.shared.setAppForeground(false)
|
||||||
|
TorManager.shared.goDormantOnBackground()
|
||||||
|
chatViewModel.endGeohashSampling()
|
||||||
|
NostrRelayManager.shared.disconnect()
|
||||||
|
didEnterBackground = true
|
||||||
|
|
||||||
|
case .active:
|
||||||
|
record(.scenePhaseChanged(.active))
|
||||||
|
chatViewModel.meshService.startServices()
|
||||||
|
TorManager.shared.setAppForeground(true)
|
||||||
|
let shouldRefreshNostrConnections = didHandleInitialActive && didEnterBackground
|
||||||
|
|
||||||
|
if didHandleInitialActive && didEnterBackground {
|
||||||
|
if TorManager.shared.isAutoStartAllowed() && !TorManager.shared.isReady {
|
||||||
|
TorManager.shared.ensureRunningOnForeground()
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
didHandleInitialActive = true
|
||||||
|
}
|
||||||
|
|
||||||
|
didEnterBackground = false
|
||||||
|
|
||||||
|
if shouldRefreshNostrConnections && TorManager.shared.isAutoStartAllowed() {
|
||||||
|
Task.detached {
|
||||||
|
let _ = await TorManager.shared.awaitReady(timeout: 60)
|
||||||
|
await MainActor.run {
|
||||||
|
TorURLSession.shared.rebuild()
|
||||||
|
NostrRelayManager.shared.resetAllConnections()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
chatViewModel.handleDidBecomeActive()
|
||||||
|
checkForSharedContent()
|
||||||
|
|
||||||
|
case .inactive:
|
||||||
|
record(.scenePhaseChanged(.inactive))
|
||||||
|
|
||||||
|
@unknown default:
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
func applicationWillTerminate() {
|
||||||
|
record(.terminationRequested)
|
||||||
|
chatViewModel.applicationWillTerminate()
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleNotificationResponse(identifier: String, userInfo: [AnyHashable: Any]) {
|
||||||
|
if identifier.hasPrefix("private-"), let peerID = PeerID(str: userInfo["peerID"] as? String) {
|
||||||
|
record(.notificationOpened(peerID: peerID))
|
||||||
|
chatViewModel.startPrivateChat(with: peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
if let deepLink = userInfo["deeplink"] as? String, let url = URL(string: deepLink) {
|
||||||
|
record(.deepLinkOpened(deepLink))
|
||||||
|
openExternalURL(url)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func presentationOptions(
|
||||||
|
forNotificationIdentifier identifier: String,
|
||||||
|
userInfo: [AnyHashable: Any]
|
||||||
|
) async -> UNNotificationPresentationOptions {
|
||||||
|
if identifier.hasPrefix("private-"), let peerID = PeerID(str: userInfo["peerID"] as? String) {
|
||||||
|
if conversations.selectedPrivatePeerID == peerID {
|
||||||
|
return []
|
||||||
|
}
|
||||||
|
return [.banner, .sound]
|
||||||
|
}
|
||||||
|
|
||||||
|
if identifier.hasPrefix("geo-activity-"),
|
||||||
|
let deepLink = userInfo["deeplink"] as? String,
|
||||||
|
let geohash = deepLink.components(separatedBy: "/").last,
|
||||||
|
case .location(let channel) = locationChannelsModel.selectedChannel,
|
||||||
|
channel.geohash == geohash {
|
||||||
|
return []
|
||||||
|
}
|
||||||
|
|
||||||
|
return [.banner, .sound]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private extension AppRuntime {
|
||||||
|
func bindRuntimeObservers() {
|
||||||
|
NostrRelayManager.shared.$isConnected
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] isConnected in
|
||||||
|
self?.handleNostrRelayConnectionChanged(isConnected)
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
NotificationCenter.default.publisher(for: .TorWillRestart)
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.record(.torLifecycleChanged(.willRestart))
|
||||||
|
self?.chatViewModel.handleTorWillRestart()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
NotificationCenter.default.publisher(for: .TorDidBecomeReady)
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.record(.torLifecycleChanged(.didBecomeReady))
|
||||||
|
self?.chatViewModel.handleTorDidBecomeReady()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
NotificationCenter.default.publisher(for: .TorWillStart)
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.record(.torLifecycleChanged(.willStart))
|
||||||
|
self?.chatViewModel.handleTorWillStart()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
NotificationCenter.default.publisher(for: .TorUserPreferenceChanged)
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] notification in
|
||||||
|
self?.record(.torLifecycleChanged(.preferenceChanged))
|
||||||
|
self?.chatViewModel.handleTorPreferenceChanged(notification)
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
#if os(iOS)
|
||||||
|
NotificationCenter.default.publisher(for: UIApplication.userDidTakeScreenshotNotification)
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.handleScreenshotCaptured()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
func checkForSharedContent() {
|
||||||
|
guard let userDefaults = UserDefaults(suiteName: BitchatApp.groupID),
|
||||||
|
let sharedContent = userDefaults.string(forKey: "sharedContent"),
|
||||||
|
let sharedDate = userDefaults.object(forKey: "sharedContentDate") as? Date else {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
guard Date().timeIntervalSince(sharedDate) < TransportConfig.uiShareAcceptWindowSeconds else {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let contentKind = SharedContentKind(rawValue: userDefaults.string(forKey: "sharedContentType") ?? "") ?? .text
|
||||||
|
|
||||||
|
userDefaults.removeObject(forKey: "sharedContent")
|
||||||
|
userDefaults.removeObject(forKey: "sharedContentType")
|
||||||
|
userDefaults.removeObject(forKey: "sharedContentDate")
|
||||||
|
|
||||||
|
switch contentKind {
|
||||||
|
case .url:
|
||||||
|
if let data = sharedContent.data(using: .utf8),
|
||||||
|
let urlData = try? JSONSerialization.jsonObject(with: data) as? [String: String],
|
||||||
|
let url = urlData["url"] {
|
||||||
|
chatViewModel.sendMessage(url)
|
||||||
|
} else {
|
||||||
|
chatViewModel.sendMessage(sharedContent)
|
||||||
|
}
|
||||||
|
case .text:
|
||||||
|
chatViewModel.sendMessage(sharedContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
record(.sharedContentAccepted(contentKind))
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleNostrRelayConnectionChanged(_ isConnected: Bool) {
|
||||||
|
record(.nostrRelayConnectionChanged(isConnected))
|
||||||
|
|
||||||
|
let becameConnected = isConnected && !lastNostrRelayConnectedState
|
||||||
|
lastNostrRelayConnectedState = isConnected
|
||||||
|
|
||||||
|
guard started, becameConnected else { return }
|
||||||
|
|
||||||
|
let isInitialConnection = !didHandleInitialNostrConnection
|
||||||
|
didHandleInitialNostrConnection = true
|
||||||
|
|
||||||
|
if !chatViewModel.nostrHandlersSetup {
|
||||||
|
chatViewModel.setupNostrMessageHandling()
|
||||||
|
chatViewModel.nostrHandlersSetup = true
|
||||||
|
}
|
||||||
|
|
||||||
|
guard !isInitialConnection else { return }
|
||||||
|
|
||||||
|
chatViewModel.resubscribeCurrentGeohash()
|
||||||
|
chatViewModel.geoChannelCoordinator?.refreshSampling()
|
||||||
|
}
|
||||||
|
|
||||||
|
func announceInitialTorStatusIfNeeded() {
|
||||||
|
if TorManager.shared.torEnforced &&
|
||||||
|
!chatViewModel.torStatusAnnounced &&
|
||||||
|
TorManager.shared.isAutoStartAllowed() {
|
||||||
|
chatViewModel.torStatusAnnounced = true
|
||||||
|
chatViewModel.addGeohashOnlySystemMessage(
|
||||||
|
String(localized: "system.tor.starting", comment: "System message when Tor is starting")
|
||||||
|
)
|
||||||
|
} else if !TorManager.shared.torEnforced && !chatViewModel.torStatusAnnounced {
|
||||||
|
chatViewModel.torStatusAnnounced = true
|
||||||
|
chatViewModel.addGeohashOnlySystemMessage(
|
||||||
|
String(localized: "system.tor.dev_bypass", comment: "System message when Tor bypass is enabled in development")
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleScreenshotCaptured() {
|
||||||
|
if appChromeModel.isLocationChannelsSheetPresented {
|
||||||
|
appChromeModel.triggerScreenshotPrivacyWarning()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if appChromeModel.isAppInfoPresented {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
chatViewModel.handleScreenshotCaptured()
|
||||||
|
}
|
||||||
|
|
||||||
|
func openExternalURL(_ url: URL) {
|
||||||
|
#if os(iOS)
|
||||||
|
UIApplication.shared.open(url)
|
||||||
|
#else
|
||||||
|
NSWorkspace.shared.open(url)
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
func record(_ event: AppEvent) {
|
||||||
|
Task {
|
||||||
|
await events.emit(event)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,918 @@
|
|||||||
|
//
|
||||||
|
// ConversationStore.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Single source of truth for conversation message state (see
|
||||||
|
// docs/CONVERSATION-STORE-DESIGN.md). One `Conversation` object per
|
||||||
|
// `ConversationID`; all mutations flow through the store's intent API and
|
||||||
|
// every mutation emits a `ConversationChange` after state is consistent.
|
||||||
|
//
|
||||||
|
// The store also owns conversation selection: the active public channel and
|
||||||
|
// the selected private peer (the two UI selection axes) plus the derived
|
||||||
|
// `selectedConversationID`.
|
||||||
|
//
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
// For more information, see <https://unlicense.org>
|
||||||
|
//
|
||||||
|
|
||||||
|
import BitFoundation
|
||||||
|
import BitLogger
|
||||||
|
import Combine
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
// MARK: - Conversation
|
||||||
|
|
||||||
|
/// A single conversation timeline (`.mesh`, `.geohash`, or `.direct`).
|
||||||
|
///
|
||||||
|
/// Publishing granularity is per conversation: views observe ONE
|
||||||
|
/// `Conversation` object, so an append to chat A never invalidates observers
|
||||||
|
/// of chat B.
|
||||||
|
///
|
||||||
|
/// Mutations are `fileprivate` by design — only `ConversationStore`'s intent
|
||||||
|
/// API may mutate a conversation, keeping the store the sole writer.
|
||||||
|
@MainActor
|
||||||
|
final class Conversation: ObservableObject, Identifiable {
|
||||||
|
let id: ConversationID
|
||||||
|
/// Maximum retained messages; oldest are trimmed on overflow.
|
||||||
|
let cap: Int
|
||||||
|
|
||||||
|
@Published private(set) var messages: [BitchatMessage] = []
|
||||||
|
@Published private(set) var isUnread: Bool = false
|
||||||
|
|
||||||
|
/// Incrementally-maintained message-ID → index map for O(1) dedup and
|
||||||
|
/// delivery-status lookup. Kept in sync on every mutation:
|
||||||
|
/// - tail append: single insert
|
||||||
|
/// - out-of-order insert: suffix reindex from the insertion point
|
||||||
|
/// - trim: full rebuild — `removeFirst(k)` is already O(n), so the
|
||||||
|
/// rebuild does not change the asymptotics, and trim only happens once
|
||||||
|
/// the cap (1337) is reached. Simple and correct beats the
|
||||||
|
/// offset-tracking alternative here.
|
||||||
|
private var indexByMessageID: [String: Int] = [:]
|
||||||
|
|
||||||
|
fileprivate init(id: ConversationID, cap: Int) {
|
||||||
|
self.id = id
|
||||||
|
self.cap = max(1, cap)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: Reads
|
||||||
|
|
||||||
|
func containsMessage(withID messageID: String) -> Bool {
|
||||||
|
indexByMessageID[messageID] != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func message(withID messageID: String) -> BitchatMessage? {
|
||||||
|
guard let index = indexByMessageID[messageID] else { return nil }
|
||||||
|
return messages[index]
|
||||||
|
}
|
||||||
|
|
||||||
|
/// All message IDs currently in this conversation (unordered).
|
||||||
|
var messageIDs: Dictionary<String, Int>.Keys {
|
||||||
|
indexByMessageID.keys
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: Store-internal mutations
|
||||||
|
|
||||||
|
/// Result of an ordered insert. `trimmedMessageIDs` reports messages
|
||||||
|
/// evicted by the cap so the store can keep its message-ID →
|
||||||
|
/// conversation map exact.
|
||||||
|
fileprivate struct InsertResult {
|
||||||
|
let inserted: Bool
|
||||||
|
let trimmedMessageIDs: [String]
|
||||||
|
|
||||||
|
static let duplicate = InsertResult(inserted: false, trimmedMessageIDs: [])
|
||||||
|
}
|
||||||
|
|
||||||
|
fileprivate enum UpsertOutcome {
|
||||||
|
case appended(trimmedMessageIDs: [String])
|
||||||
|
case updated
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Inserts a message in timestamp order, deduplicating by message ID.
|
||||||
|
/// Fast path appends when the timestamp is >= the current tail;
|
||||||
|
/// otherwise a binary search finds the upper-bound insertion point so
|
||||||
|
/// arrival order is preserved among equal timestamps.
|
||||||
|
/// Reports `inserted: false` if a message with the same ID already exists.
|
||||||
|
fileprivate func insert(_ message: BitchatMessage) -> InsertResult {
|
||||||
|
guard indexByMessageID[message.id] == nil else { return .duplicate }
|
||||||
|
|
||||||
|
if let last = messages.last, message.timestamp < last.timestamp {
|
||||||
|
let index = insertionIndex(for: message.timestamp)
|
||||||
|
messages.insert(message, at: index)
|
||||||
|
reindex(from: index)
|
||||||
|
} else {
|
||||||
|
messages.append(message)
|
||||||
|
indexByMessageID[message.id] = messages.count - 1
|
||||||
|
}
|
||||||
|
|
||||||
|
return InsertResult(inserted: true, trimmedMessageIDs: trimIfNeeded())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Replace-or-append by message ID. An existing message keeps its
|
||||||
|
/// timeline position (in-place updates like media progress reuse the
|
||||||
|
/// original timestamp); a new message goes through ordered insertion.
|
||||||
|
fileprivate func upsert(_ message: BitchatMessage) -> UpsertOutcome {
|
||||||
|
if let index = indexByMessageID[message.id] {
|
||||||
|
messages[index] = message
|
||||||
|
return .updated
|
||||||
|
}
|
||||||
|
let result = insert(message)
|
||||||
|
return .appended(trimmedMessageIDs: result.trimmedMessageIDs)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Applies a delivery status keyed by message ID, honoring the
|
||||||
|
/// no-downgrade rule (the SOLE enforcement point — every delivery
|
||||||
|
/// update flows through the store): equal statuses are skipped, and
|
||||||
|
/// `.read` is never downgraded to `.delivered` or `.sent`.
|
||||||
|
/// Returns `true` when the status was applied.
|
||||||
|
fileprivate func applyDeliveryStatus(_ status: DeliveryStatus, forMessageID messageID: String) -> Bool {
|
||||||
|
guard let index = indexByMessageID[messageID] else { return false }
|
||||||
|
let message = messages[index]
|
||||||
|
guard !Self.shouldSkipStatusUpdate(current: message.deliveryStatus, new: status) else { return false }
|
||||||
|
|
||||||
|
message.deliveryStatus = status
|
||||||
|
// BitchatMessage is a reference type; write back through the
|
||||||
|
// subscript so the @Published wrapper emits.
|
||||||
|
messages[index] = message
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Republishes a message without changing state. Used for mirrored
|
||||||
|
/// copies that share a BitchatMessage instance: the first conversation's
|
||||||
|
/// status apply mutated the shared object, so this conversation's
|
||||||
|
/// observers still need an @Published emission to re-render.
|
||||||
|
@discardableResult
|
||||||
|
fileprivate func republishMessage(withID messageID: String) -> Bool {
|
||||||
|
guard let index = indexByMessageID[messageID] else { return false }
|
||||||
|
messages[index] = messages[index]
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
fileprivate func setUnread(_ unread: Bool) -> Bool {
|
||||||
|
guard isUnread != unread else { return false }
|
||||||
|
isUnread = unread
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Removes a single message by ID. Returns the removed message, or
|
||||||
|
/// `nil` when no message with that ID exists.
|
||||||
|
fileprivate func remove(messageID: String) -> BitchatMessage? {
|
||||||
|
guard let index = indexByMessageID[messageID] else { return nil }
|
||||||
|
let removed = messages.remove(at: index)
|
||||||
|
indexByMessageID.removeValue(forKey: messageID)
|
||||||
|
reindex(from: index)
|
||||||
|
return removed
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Removes every message matching `predicate`. Returns the removed
|
||||||
|
/// message IDs (empty when nothing matched).
|
||||||
|
fileprivate func removeAll(where predicate: (BitchatMessage) -> Bool) -> [String] {
|
||||||
|
var removedIDs: [String] = []
|
||||||
|
messages.removeAll { message in
|
||||||
|
guard predicate(message) else { return false }
|
||||||
|
removedIDs.append(message.id)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
guard !removedIDs.isEmpty else { return [] }
|
||||||
|
for id in removedIDs {
|
||||||
|
indexByMessageID.removeValue(forKey: id)
|
||||||
|
}
|
||||||
|
reindex(from: 0)
|
||||||
|
return removedIDs
|
||||||
|
}
|
||||||
|
|
||||||
|
fileprivate func clearMessages() {
|
||||||
|
messages.removeAll()
|
||||||
|
indexByMessageID.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: Diagnostics
|
||||||
|
|
||||||
|
/// Appends human-readable invariant violations for this conversation
|
||||||
|
/// (empty when healthy): the ID index must be the exact inverse of the
|
||||||
|
/// messages array, the cap must hold, and timestamps must be
|
||||||
|
/// non-decreasing (equal timestamps keep arrival order, so only strict
|
||||||
|
/// inversions are violations). O(messages); allocates only on violation.
|
||||||
|
fileprivate func collectInvariantViolations(into violations: inout [String], label: String) {
|
||||||
|
if indexByMessageID.count != messages.count {
|
||||||
|
violations.append("\(label): index has \(indexByMessageID.count) entries for \(messages.count) messages")
|
||||||
|
}
|
||||||
|
if messages.count > cap {
|
||||||
|
violations.append("\(label): \(messages.count) messages exceeds cap \(cap)")
|
||||||
|
}
|
||||||
|
var previousTimestamp: Date?
|
||||||
|
for position in messages.indices {
|
||||||
|
let message = messages[position]
|
||||||
|
// Count equality + every message resolving to its own position
|
||||||
|
// proves the index is exactly the inverse map (no stale extras).
|
||||||
|
if let index = indexByMessageID[message.id] {
|
||||||
|
if index != position {
|
||||||
|
violations.append("\(label): message \(message.id.prefix(8))… at \(position) indexed at \(index)")
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
violations.append("\(label): message \(message.id.prefix(8))… at \(position) missing from index")
|
||||||
|
}
|
||||||
|
if let previousTimestamp, message.timestamp < previousTimestamp {
|
||||||
|
violations.append("\(label): timestamp order violated at \(position)")
|
||||||
|
}
|
||||||
|
previousTimestamp = message.timestamp
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: Internals
|
||||||
|
|
||||||
|
static func shouldSkipStatusUpdate(current: DeliveryStatus?, new: DeliveryStatus) -> Bool {
|
||||||
|
guard let current else { return false }
|
||||||
|
if current == new { return true }
|
||||||
|
|
||||||
|
switch (current, new) {
|
||||||
|
case (.read, .delivered), (.read, .sent):
|
||||||
|
return true
|
||||||
|
default:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Upper-bound binary search: first index whose timestamp is strictly
|
||||||
|
/// greater than `timestamp`, so equal-timestamp messages keep arrival
|
||||||
|
/// order.
|
||||||
|
private func insertionIndex(for timestamp: Date) -> Int {
|
||||||
|
var low = 0
|
||||||
|
var high = messages.count
|
||||||
|
while low < high {
|
||||||
|
let mid = (low + high) / 2
|
||||||
|
if messages[mid].timestamp <= timestamp {
|
||||||
|
low = mid + 1
|
||||||
|
} else {
|
||||||
|
high = mid
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return low
|
||||||
|
}
|
||||||
|
|
||||||
|
private func reindex(from start: Int) {
|
||||||
|
for index in start..<messages.count {
|
||||||
|
indexByMessageID[messages[index].id] = index
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Trims oldest messages over the cap; returns the trimmed message IDs.
|
||||||
|
private func trimIfNeeded() -> [String] {
|
||||||
|
guard messages.count > cap else { return [] }
|
||||||
|
let overflow = messages.count - cap
|
||||||
|
let trimmedIDs = messages.prefix(overflow).map(\.id)
|
||||||
|
for id in trimmedIDs {
|
||||||
|
indexByMessageID.removeValue(forKey: id)
|
||||||
|
}
|
||||||
|
messages.removeFirst(overflow)
|
||||||
|
reindex(from: 0)
|
||||||
|
return trimmedIDs
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - ConversationChange
|
||||||
|
|
||||||
|
/// Typed mutation events for non-UI consumers (delivery tracking,
|
||||||
|
/// notifications, sync) that need "something changed in conversation X"
|
||||||
|
/// without subscribing to whole message arrays. Emitted on the store's
|
||||||
|
/// `changes` subject AFTER the corresponding state is consistent.
|
||||||
|
enum ConversationChange {
|
||||||
|
case appended(ConversationID, BitchatMessage)
|
||||||
|
case updated(ConversationID, messageID: String)
|
||||||
|
case statusChanged(ConversationID, messageID: String, DeliveryStatus)
|
||||||
|
case messageRemoved(ConversationID, messageID: String)
|
||||||
|
case cleared(ConversationID)
|
||||||
|
case removed(ConversationID)
|
||||||
|
case migrated(from: ConversationID, to: ConversationID)
|
||||||
|
case unreadChanged(ConversationID, isUnread: Bool)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - ConversationStore
|
||||||
|
|
||||||
|
/// Sole writer and sole holder of conversation message state. All mutations
|
||||||
|
/// go through the intent API below; backing collections are `private(set)`.
|
||||||
|
/// Reads are synchronous — writers and readers share the main actor, so
|
||||||
|
/// after an intent returns every observer sees the result.
|
||||||
|
@MainActor
|
||||||
|
final class ConversationStore: ObservableObject {
|
||||||
|
/// Conversation creation order; published so list-style consumers can
|
||||||
|
/// observe conversations appearing/disappearing without rebuilding from
|
||||||
|
/// the dictionary.
|
||||||
|
@Published private(set) var conversationIDs: [ConversationID] = []
|
||||||
|
@Published private(set) var selectedConversationID: ConversationID?
|
||||||
|
@Published private(set) var unreadConversations: Set<ConversationID> = []
|
||||||
|
|
||||||
|
// MARK: Selection state
|
||||||
|
// The two UI selection axes: which public channel is active, and which
|
||||||
|
// private chat (if any) is open on top of it. `selectedConversationID`
|
||||||
|
// is derived: the open private chat wins, otherwise the active public
|
||||||
|
// channel's conversation. Mutate via `setActiveChannel` /
|
||||||
|
// `setSelectedPrivatePeer` only.
|
||||||
|
|
||||||
|
@Published private(set) var activeChannel: ChannelID = .mesh
|
||||||
|
@Published private(set) var selectedPrivatePeerID: PeerID?
|
||||||
|
|
||||||
|
private(set) var conversationsByID: [ConversationID: Conversation] = [:]
|
||||||
|
|
||||||
|
/// Store-level message-ID → conversation-membership map for ID-only
|
||||||
|
/// lookups (delivery receipts arrive with a message ID, not a
|
||||||
|
/// conversation). Maintained incrementally at every mutation point —
|
||||||
|
/// all mutation is centralized in the intent API below, so the map is
|
||||||
|
/// exact, never scanned or rebuilt.
|
||||||
|
///
|
||||||
|
/// The value is a `Set` because a private message can legitimately live
|
||||||
|
/// in TWO direct conversations: step 2's raw per-peer keying mirrors a
|
||||||
|
/// message into both the stable-key and ephemeral-peer chats
|
||||||
|
/// (`mirrorToEphemeralIfNeeded`). A delivery update must reach both
|
||||||
|
/// copies.
|
||||||
|
private var conversationIDsByMessageID: [String: Set<ConversationID>] = [:]
|
||||||
|
|
||||||
|
/// Monotonic count of messages inserted into any conversation (appends,
|
||||||
|
/// upsert-appends, migration inserts). Field-observability only: the
|
||||||
|
/// periodic store audit folds the delta into its heartbeat line so logs
|
||||||
|
/// carry throughput context. Never read on a hot path.
|
||||||
|
private(set) var appendCount: Int = 0
|
||||||
|
|
||||||
|
/// Sample counter for the mirrored-republish debug log in the ID-only
|
||||||
|
/// `setDeliveryStatus` fan-out (first + every Nth occurrence).
|
||||||
|
private var mirroredRepublishLogCount = 0
|
||||||
|
|
||||||
|
let changes = PassthroughSubject<ConversationChange, Never>()
|
||||||
|
|
||||||
|
// MARK: Intent API
|
||||||
|
|
||||||
|
/// Returns the conversation for `id`, creating it (with the cap policy
|
||||||
|
/// for its kind) on first access.
|
||||||
|
@discardableResult
|
||||||
|
func conversation(for id: ConversationID) -> Conversation {
|
||||||
|
if let existing = conversationsByID[id] {
|
||||||
|
return existing
|
||||||
|
}
|
||||||
|
let conversation = Conversation(id: id, cap: Self.cap(for: id))
|
||||||
|
conversationsByID[id] = conversation
|
||||||
|
conversationIDs.append(id)
|
||||||
|
return conversation
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Appends a message in timestamp order. Returns `false` (and emits
|
||||||
|
/// nothing) if a message with the same ID is already present.
|
||||||
|
@discardableResult
|
||||||
|
func append(_ message: BitchatMessage, to id: ConversationID) -> Bool {
|
||||||
|
let conversation = conversation(for: id)
|
||||||
|
let result = conversation.insert(message)
|
||||||
|
guard result.inserted else { return false }
|
||||||
|
registerMessageID(message.id, in: id)
|
||||||
|
unregisterMessageIDs(result.trimmedMessageIDs, from: id)
|
||||||
|
changes.send(.appended(id, message))
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Replace-or-append by message ID (media progress, edits).
|
||||||
|
func upsertByID(_ message: BitchatMessage, in id: ConversationID) {
|
||||||
|
let conversation = conversation(for: id)
|
||||||
|
switch conversation.upsert(message) {
|
||||||
|
case .appended(let trimmedMessageIDs):
|
||||||
|
registerMessageID(message.id, in: id)
|
||||||
|
unregisterMessageIDs(trimmedMessageIDs, from: id)
|
||||||
|
changes.send(.appended(id, message))
|
||||||
|
case .updated:
|
||||||
|
changes.send(.updated(id, messageID: message.id))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Applies a delivery status keyed by message ID. Returns `false` when
|
||||||
|
/// the message is unknown or the update would downgrade the status
|
||||||
|
/// (read beats delivered beats sent).
|
||||||
|
@discardableResult
|
||||||
|
func setDeliveryStatus(_ status: DeliveryStatus, forMessageID messageID: String, in id: ConversationID) -> Bool {
|
||||||
|
guard let conversation = conversationsByID[id],
|
||||||
|
conversation.applyDeliveryStatus(status, forMessageID: messageID) else {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
changes.send(.statusChanged(id, messageID: messageID, status))
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Applies a delivery status to EVERY conversation containing
|
||||||
|
/// `messageID` (ID-only — delivery receipts don't know conversations;
|
||||||
|
/// mirrored private copies live in two direct chats). Returns `false`
|
||||||
|
/// when the message is unknown or no copy changed (equal status or
|
||||||
|
/// downgrade — read beats delivered beats sent).
|
||||||
|
///
|
||||||
|
/// `BitchatMessage` is a reference type, so mirrored copies sharing one
|
||||||
|
/// instance are mutated by the first conversation's apply. The skipped
|
||||||
|
/// conversations still hold the changed message, so they get an explicit
|
||||||
|
/// republish and `.statusChanged` event - otherwise a view observing the
|
||||||
|
/// mirrored conversation would render stale status. Distinct copies whose
|
||||||
|
/// update was genuinely rejected (downgrade) are left untouched, guarded
|
||||||
|
/// by status equality.
|
||||||
|
@discardableResult
|
||||||
|
func setDeliveryStatus(_ status: DeliveryStatus, forMessageID messageID: String) -> Bool {
|
||||||
|
guard let ids = conversationIDsByMessageID[messageID] else { return false }
|
||||||
|
var applied = false
|
||||||
|
var skipped: [ConversationID] = []
|
||||||
|
for id in ids {
|
||||||
|
if setDeliveryStatus(status, forMessageID: messageID, in: id) {
|
||||||
|
applied = true
|
||||||
|
} else {
|
||||||
|
skipped.append(id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
guard applied else { return false }
|
||||||
|
for id in skipped {
|
||||||
|
guard let conversation = conversationsByID[id],
|
||||||
|
conversation.message(withID: messageID)?.deliveryStatus == status,
|
||||||
|
conversation.republishMessage(withID: messageID) else { continue }
|
||||||
|
// Field proof the mirrored-copy republish path actually fires;
|
||||||
|
// sampled (first + every Nth) so mirrored chats can't spam logs.
|
||||||
|
mirroredRepublishLogCount += 1
|
||||||
|
if mirroredRepublishLogCount == 1
|
||||||
|
|| mirroredRepublishLogCount.isMultiple(of: TransportConfig.conversationStoreMirroredRepublishLogInterval) {
|
||||||
|
SecureLogger.debug(
|
||||||
|
"mirrored republish #\(mirroredRepublishLogCount) for \(messageID.prefix(8))… in \(id.auditDescription)",
|
||||||
|
category: .session
|
||||||
|
)
|
||||||
|
}
|
||||||
|
changes.send(.statusChanged(id, messageID: messageID, status))
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Current delivery status of `messageID` in whichever conversation
|
||||||
|
/// holds it (mirrored copies share status — see `setDeliveryStatus`).
|
||||||
|
func deliveryStatus(forMessageID messageID: String) -> DeliveryStatus? {
|
||||||
|
guard let ids = conversationIDsByMessageID[messageID] else { return nil }
|
||||||
|
for id in ids {
|
||||||
|
if let status = conversationsByID[id]?.message(withID: messageID)?.deliveryStatus {
|
||||||
|
return status
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Every conversation currently containing `messageID` (empty when the
|
||||||
|
/// message is unknown).
|
||||||
|
func conversationIDs(forMessageID messageID: String) -> Set<ConversationID> {
|
||||||
|
conversationIDsByMessageID[messageID] ?? []
|
||||||
|
}
|
||||||
|
|
||||||
|
func markRead(_ id: ConversationID) {
|
||||||
|
guard unreadConversations.contains(id) else { return }
|
||||||
|
unreadConversations.remove(id)
|
||||||
|
conversationsByID[id]?.setUnread(false)
|
||||||
|
changes.send(.unreadChanged(id, isUnread: false))
|
||||||
|
}
|
||||||
|
|
||||||
|
func markUnread(_ id: ConversationID) {
|
||||||
|
guard !unreadConversations.contains(id) else { return }
|
||||||
|
let conversation = conversation(for: id)
|
||||||
|
unreadConversations.insert(id)
|
||||||
|
conversation.setUnread(true)
|
||||||
|
changes.send(.unreadChanged(id, isUnread: true))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Selects a conversation (creating it if needed) or clears the
|
||||||
|
/// selection with `nil`.
|
||||||
|
func select(_ id: ConversationID?) {
|
||||||
|
if let id {
|
||||||
|
conversation(for: id)
|
||||||
|
}
|
||||||
|
guard selectedConversationID != id else { return }
|
||||||
|
selectedConversationID = id
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Switches the active public channel. While no private chat is open
|
||||||
|
/// the selection follows the channel.
|
||||||
|
func setActiveChannel(_ channelID: ChannelID) {
|
||||||
|
if activeChannel != channelID {
|
||||||
|
activeChannel = channelID
|
||||||
|
}
|
||||||
|
refreshDerivedSelection()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Opens a private chat (`nil` closes it, returning the selection to the
|
||||||
|
/// active public channel's conversation).
|
||||||
|
func setSelectedPrivatePeer(_ peerID: PeerID?) {
|
||||||
|
if selectedPrivatePeerID != peerID {
|
||||||
|
selectedPrivatePeerID = peerID
|
||||||
|
}
|
||||||
|
refreshDerivedSelection()
|
||||||
|
}
|
||||||
|
|
||||||
|
private func refreshDerivedSelection() {
|
||||||
|
if let peerID = selectedPrivatePeerID {
|
||||||
|
select(.directPeer(peerID))
|
||||||
|
} else {
|
||||||
|
select(ConversationID(channelID: activeChannel))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Moves all messages from `source` into `destination` (the
|
||||||
|
/// ephemeral↔stable peer-ID handoff): dedups by message ID, preserves
|
||||||
|
/// timestamp order, carries unread state over, and hands off the
|
||||||
|
/// selection — mirroring `ChatPrivateConversationCoordinator`'s
|
||||||
|
/// migration semantics. The source conversation is removed. Emits a
|
||||||
|
/// single `.migrated(from:to:)` once the whole move is consistent.
|
||||||
|
func migrateConversation(from source: ConversationID, to destination: ConversationID) {
|
||||||
|
guard source != destination, let sourceConversation = conversationsByID[source] else { return }
|
||||||
|
|
||||||
|
let destinationConversation = conversation(for: destination)
|
||||||
|
for message in sourceConversation.messages {
|
||||||
|
let result = destinationConversation.insert(message)
|
||||||
|
guard result.inserted else { continue }
|
||||||
|
registerMessageID(message.id, in: destination)
|
||||||
|
unregisterMessageIDs(result.trimmedMessageIDs, from: destination)
|
||||||
|
}
|
||||||
|
for messageID in sourceConversation.messageIDs {
|
||||||
|
unregisterMessageID(messageID, from: source)
|
||||||
|
}
|
||||||
|
|
||||||
|
let wasUnread = unreadConversations.contains(source)
|
||||||
|
let wasSelected = selectedConversationID == source
|
||||||
|
|
||||||
|
conversationsByID.removeValue(forKey: source)
|
||||||
|
conversationIDs.removeAll { $0 == source }
|
||||||
|
unreadConversations.remove(source)
|
||||||
|
|
||||||
|
if wasUnread, !unreadConversations.contains(destination) {
|
||||||
|
unreadConversations.insert(destination)
|
||||||
|
destinationConversation.setUnread(true)
|
||||||
|
}
|
||||||
|
if wasSelected {
|
||||||
|
selectedConversationID = destination
|
||||||
|
// Keep the private-peer selection axis consistent with the
|
||||||
|
// handed-off selection.
|
||||||
|
if let peerID = selectedPrivatePeerID,
|
||||||
|
source == .directPeer(peerID),
|
||||||
|
case .direct(let destinationHandle) = destination {
|
||||||
|
selectedPrivatePeerID = destinationHandle.routingPeerID
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
changes.send(.migrated(from: source, to: destination))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Removes a single message by ID from a conversation. Returns the
|
||||||
|
/// removed message, or `nil` (emitting nothing) when the conversation or
|
||||||
|
/// message is unknown.
|
||||||
|
@discardableResult
|
||||||
|
func removeMessage(withID messageID: String, from id: ConversationID) -> BitchatMessage? {
|
||||||
|
guard let conversation = conversationsByID[id],
|
||||||
|
let removed = conversation.remove(messageID: messageID) else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
unregisterMessageID(messageID, from: id)
|
||||||
|
changes.send(.messageRemoved(id, messageID: messageID))
|
||||||
|
return removed
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Removes every message matching `predicate` from a conversation,
|
||||||
|
/// emitting one `.messageRemoved` per removed message after the
|
||||||
|
/// conversation is consistent. No-op for unknown conversations.
|
||||||
|
func removeMessages(from id: ConversationID, where predicate: (BitchatMessage) -> Bool) {
|
||||||
|
guard let conversation = conversationsByID[id] else { return }
|
||||||
|
let removedIDs = conversation.removeAll(where: predicate)
|
||||||
|
unregisterMessageIDs(removedIDs, from: id)
|
||||||
|
for messageID in removedIDs {
|
||||||
|
changes.send(.messageRemoved(id, messageID: messageID))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Empties a conversation's timeline but keeps the conversation (and
|
||||||
|
/// its unread/selection state) alive.
|
||||||
|
func clear(_ id: ConversationID) {
|
||||||
|
guard let conversation = conversationsByID[id] else { return }
|
||||||
|
for messageID in conversation.messageIDs {
|
||||||
|
unregisterMessageID(messageID, from: id)
|
||||||
|
}
|
||||||
|
conversation.clearMessages()
|
||||||
|
changes.send(.cleared(id))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Removes a conversation entirely, including unread state; clears the
|
||||||
|
/// selection if it pointed at the removed conversation.
|
||||||
|
func removeConversation(_ id: ConversationID) {
|
||||||
|
guard let conversation = conversationsByID.removeValue(forKey: id) else { return }
|
||||||
|
for messageID in conversation.messageIDs {
|
||||||
|
unregisterMessageID(messageID, from: id)
|
||||||
|
}
|
||||||
|
conversationIDs.removeAll { $0 == id }
|
||||||
|
unreadConversations.remove(id)
|
||||||
|
if selectedConversationID == id {
|
||||||
|
selectedConversationID = nil
|
||||||
|
}
|
||||||
|
changes.send(.removed(id))
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearAll() {
|
||||||
|
let removedIDs = conversationIDs
|
||||||
|
guard !removedIDs.isEmpty || selectedConversationID != nil else { return }
|
||||||
|
|
||||||
|
conversationsByID.removeAll()
|
||||||
|
conversationIDs.removeAll()
|
||||||
|
unreadConversations.removeAll()
|
||||||
|
conversationIDsByMessageID.removeAll()
|
||||||
|
if selectedConversationID != nil {
|
||||||
|
selectedConversationID = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
for id in removedIDs {
|
||||||
|
changes.send(.removed(id))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: Diagnostics
|
||||||
|
|
||||||
|
/// Total messages across all conversations. O(#conversations) — heartbeat
|
||||||
|
/// logging only, never a hot path.
|
||||||
|
var totalMessageCount: Int {
|
||||||
|
conversationsByID.values.reduce(0) { $0 + $1.messages.count }
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Number of distinct message IDs in the store-level membership map.
|
||||||
|
var messageIDMapCount: Int {
|
||||||
|
conversationIDsByMessageID.count
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Verifies the store's correctness invariants and returns human-readable
|
||||||
|
/// violations (empty = healthy). Intended for a periodic field audit:
|
||||||
|
/// O(total messages) and allocation-free while healthy. Checks:
|
||||||
|
/// - the `conversationIDs` ordering array matches `conversationsByID`
|
||||||
|
/// - per conversation: ID index exact, cap held, timestamp order
|
||||||
|
/// (see `Conversation.collectInvariantViolations`)
|
||||||
|
/// - the message-ID → conversation map matches reality exactly: every
|
||||||
|
/// mapped membership points at a live conversation actually holding
|
||||||
|
/// the message, and total memberships equal total messages (with the
|
||||||
|
/// forward check, equality proves no conversation message is missing
|
||||||
|
/// from the map)
|
||||||
|
/// - `unreadConversations` only references existing conversations
|
||||||
|
/// - `selectedConversationID`, when set, references an existing
|
||||||
|
/// conversation (`select(_:)` creates on selection and
|
||||||
|
/// `removeConversation`/`clearAll` clear it, so existence is the
|
||||||
|
/// invariant for both the channel-derived and direct-peer cases)
|
||||||
|
func auditInvariants() -> [String] {
|
||||||
|
var violations: [String] = []
|
||||||
|
|
||||||
|
if conversationIDs.count != conversationsByID.count {
|
||||||
|
violations.append("conversationIDs lists \(conversationIDs.count) conversations but dictionary holds \(conversationsByID.count)")
|
||||||
|
}
|
||||||
|
for id in conversationIDs where conversationsByID[id] == nil {
|
||||||
|
violations.append("conversationIDs lists \(id.auditDescription) but no conversation exists")
|
||||||
|
}
|
||||||
|
|
||||||
|
var totalMessages = 0
|
||||||
|
for (id, conversation) in conversationsByID {
|
||||||
|
totalMessages += conversation.messages.count
|
||||||
|
conversation.collectInvariantViolations(into: &violations, label: id.auditDescription)
|
||||||
|
}
|
||||||
|
|
||||||
|
var totalMappedMemberships = 0
|
||||||
|
for (messageID, ids) in conversationIDsByMessageID {
|
||||||
|
totalMappedMemberships += ids.count
|
||||||
|
if ids.isEmpty {
|
||||||
|
violations.append("message map: \(messageID.prefix(8))… has an empty membership set")
|
||||||
|
}
|
||||||
|
for id in ids {
|
||||||
|
guard let conversation = conversationsByID[id] else {
|
||||||
|
violations.append("message map: \(messageID.prefix(8))… claims unknown conversation \(id.auditDescription)")
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if !conversation.containsMessage(withID: messageID) {
|
||||||
|
violations.append("message map: \(messageID.prefix(8))… not present in claimed conversation \(id.auditDescription)")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if totalMappedMemberships != totalMessages {
|
||||||
|
violations.append("message map holds \(totalMappedMemberships) memberships but conversations hold \(totalMessages) messages")
|
||||||
|
}
|
||||||
|
|
||||||
|
for id in unreadConversations where conversationsByID[id] == nil {
|
||||||
|
violations.append("unreadConversations contains unknown conversation \(id.auditDescription)")
|
||||||
|
}
|
||||||
|
|
||||||
|
if let selected = selectedConversationID, conversationsByID[selected] == nil {
|
||||||
|
violations.append("selectedConversationID \(selected.auditDescription) has no conversation")
|
||||||
|
}
|
||||||
|
|
||||||
|
return violations
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: Internals
|
||||||
|
|
||||||
|
private func registerMessageID(_ messageID: String, in id: ConversationID) {
|
||||||
|
conversationIDsByMessageID[messageID, default: []].insert(id)
|
||||||
|
// Single choke point for every successful insertion (append, upsert
|
||||||
|
// append, migration insert) — the audit heartbeat's throughput delta.
|
||||||
|
appendCount += 1
|
||||||
|
}
|
||||||
|
|
||||||
|
private func unregisterMessageID(_ messageID: String, from id: ConversationID) {
|
||||||
|
guard var ids = conversationIDsByMessageID[messageID] else { return }
|
||||||
|
ids.remove(id)
|
||||||
|
if ids.isEmpty {
|
||||||
|
conversationIDsByMessageID.removeValue(forKey: messageID)
|
||||||
|
} else {
|
||||||
|
conversationIDsByMessageID[messageID] = ids
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func unregisterMessageIDs(_ messageIDs: [String], from id: ConversationID) {
|
||||||
|
for messageID in messageIDs {
|
||||||
|
unregisterMessageID(messageID, from: id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func cap(for id: ConversationID) -> Int {
|
||||||
|
switch id {
|
||||||
|
case .mesh:
|
||||||
|
return TransportConfig.meshTimelineCap
|
||||||
|
case .geohash:
|
||||||
|
return TransportConfig.geoTimelineCap
|
||||||
|
case .direct:
|
||||||
|
return TransportConfig.privateChatCap
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Direct-conversation keying + derived views
|
||||||
|
|
||||||
|
extension ConversationID {
|
||||||
|
/// Direct-conversation ID keyed by the *raw* routing peer ID.
|
||||||
|
///
|
||||||
|
/// Direct conversations are deliberately keyed per `PeerID`, not per
|
||||||
|
/// resolved identity: the private-chat coordinators mirror messages into
|
||||||
|
/// both the ephemeral and stable peer's conversations
|
||||||
|
/// (`mirrorToEphemeralIfNeeded`) and consolidate/migrate between them
|
||||||
|
/// explicitly, so a raw lookup by whichever peer ID is selected always
|
||||||
|
/// finds the right timeline without an identity-resolution layer.
|
||||||
|
static func directPeer(_ peerID: PeerID) -> ConversationID {
|
||||||
|
.direct(PeerHandle(id: "peer:\(peerID.id)", routingPeerID: peerID))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
extension ConversationStore {
|
||||||
|
/// All direct conversations' messages keyed by routing peer ID — the
|
||||||
|
/// shape `ChatViewModel.privateChats` exposes to the coordinators.
|
||||||
|
/// Values are the conversations' backing arrays (COW), so building this
|
||||||
|
/// is O(#conversations), not O(#messages).
|
||||||
|
func directMessagesByRoutingPeerID() -> [PeerID: [BitchatMessage]] {
|
||||||
|
var messagesByPeerID: [PeerID: [BitchatMessage]] = [:]
|
||||||
|
messagesByPeerID.reserveCapacity(conversationsByID.count)
|
||||||
|
for (id, conversation) in conversationsByID {
|
||||||
|
guard case .direct(let handle) = id else { continue }
|
||||||
|
messagesByPeerID[handle.routingPeerID] = conversation.messages
|
||||||
|
}
|
||||||
|
return messagesByPeerID
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Unread direct conversations as routing peer IDs — the shape
|
||||||
|
/// `ChatViewModel.unreadPrivateMessages` exposes to the coordinators.
|
||||||
|
func unreadDirectRoutingPeerIDs() -> Set<PeerID> {
|
||||||
|
var peerIDs = Set<PeerID>()
|
||||||
|
for id in unreadConversations {
|
||||||
|
guard case .direct(let handle) = id else { continue }
|
||||||
|
peerIDs.insert(handle.routingPeerID)
|
||||||
|
}
|
||||||
|
return peerIDs
|
||||||
|
}
|
||||||
|
|
||||||
|
/// `true` when any direct conversation contains a message with `messageID`
|
||||||
|
/// (O(1) via the store-level message-ID → conversation map).
|
||||||
|
func directConversationsContainMessage(withID messageID: String) -> Bool {
|
||||||
|
conversationIDs(forMessageID: messageID).contains { id in
|
||||||
|
if case .direct = id { return true }
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Message IDs across all direct conversations (read-receipt pruning
|
||||||
|
/// keeps only receipts whose messages still exist).
|
||||||
|
func directMessageIDs() -> Set<String> {
|
||||||
|
var messageIDs = Set<String>()
|
||||||
|
for (id, conversation) in conversationsByID {
|
||||||
|
guard case .direct = id else { continue }
|
||||||
|
messageIDs.formUnion(conversation.messageIDs)
|
||||||
|
}
|
||||||
|
return messageIDs
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Removes every direct conversation (panic clear).
|
||||||
|
func removeAllDirectConversations() {
|
||||||
|
let directIDs = conversationIDs.filter { id in
|
||||||
|
if case .direct = id { return true }
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
for id in directIDs {
|
||||||
|
removeConversation(id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Diagnostics support
|
||||||
|
|
||||||
|
extension ConversationID {
|
||||||
|
/// Short, log-safe description for audit/diagnostic lines. Direct
|
||||||
|
/// conversations truncate the handle so full peer keys never hit logs.
|
||||||
|
fileprivate var auditDescription: String {
|
||||||
|
switch self {
|
||||||
|
case .mesh:
|
||||||
|
return "mesh"
|
||||||
|
case .geohash(let geohash):
|
||||||
|
return "geo:\(geohash)"
|
||||||
|
case .direct(let handle):
|
||||||
|
return "direct:\(handle.id.prefix(13))…"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#if DEBUG
|
||||||
|
// Test-only corruption hooks for `auditInvariants()` tests. The store is the
|
||||||
|
// sole writer by design — `Conversation`'s mutators are fileprivate and the
|
||||||
|
// store's backing collections are private — so the inconsistent states the
|
||||||
|
// audit exists to catch CANNOT be manufactured through the intent API. These
|
||||||
|
// DEBUG-only hooks deliberately bypass that lockdown to inject exactly those
|
||||||
|
// impossible states. Never call them outside tests.
|
||||||
|
extension Conversation {
|
||||||
|
/// Points an existing message's index entry at the wrong position
|
||||||
|
/// (positions 0 and 1 swap their index entries). Requires >= 2 messages.
|
||||||
|
func _testCorruptIndexEntries() {
|
||||||
|
guard messages.count >= 2 else { return }
|
||||||
|
indexByMessageID[messages[0].id] = 1
|
||||||
|
indexByMessageID[messages[1].id] = 0
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Drops a message's index entry entirely (count mismatch + missing).
|
||||||
|
func _testRemoveIndexEntry(forMessageID messageID: String) {
|
||||||
|
indexByMessageID.removeValue(forKey: messageID)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Swaps the first and last messages while keeping the index consistent,
|
||||||
|
/// so ONLY the timestamp-order invariant is violated (requires the two
|
||||||
|
/// messages to have distinct timestamps).
|
||||||
|
func _testCorruptOrderingPreservingIndex() {
|
||||||
|
guard messages.count >= 2 else { return }
|
||||||
|
messages.swapAt(0, messages.count - 1)
|
||||||
|
indexByMessageID[messages[0].id] = 0
|
||||||
|
indexByMessageID[messages[messages.count - 1].id] = messages.count - 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
extension ConversationStore {
|
||||||
|
/// Adds a map membership that the conversation does not actually hold.
|
||||||
|
func _testRegisterPhantomMessageID(_ messageID: String, in id: ConversationID) {
|
||||||
|
conversationIDsByMessageID[messageID, default: []].insert(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Drops a real map membership (conversation message missing from map).
|
||||||
|
func _testUnregisterMessageID(_ messageID: String, from id: ConversationID) {
|
||||||
|
conversationIDsByMessageID[messageID]?.remove(id)
|
||||||
|
if conversationIDsByMessageID[messageID]?.isEmpty == true {
|
||||||
|
conversationIDsByMessageID.removeValue(forKey: messageID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Appends past the conversation cap, bypassing trim (map kept exact so
|
||||||
|
/// only the cap invariant is violated).
|
||||||
|
func _testAppendBypassingCap(_ message: BitchatMessage, to id: ConversationID) {
|
||||||
|
let conversation = conversation(for: id)
|
||||||
|
conversation._testAppendBypassingTrim(message)
|
||||||
|
conversationIDsByMessageID[message.id, default: []].insert(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Marks a nonexistent conversation unread without creating it.
|
||||||
|
func _testInsertUnreadConversationID(_ id: ConversationID) {
|
||||||
|
unreadConversations.insert(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the selection directly, without `select(_:)`'s create-on-select.
|
||||||
|
func _testSetSelectedConversationID(_ id: ConversationID?) {
|
||||||
|
selectedConversationID = id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
extension Conversation {
|
||||||
|
fileprivate func _testAppendBypassingTrim(_ message: BitchatMessage) {
|
||||||
|
messages.append(message)
|
||||||
|
indexByMessageID[message.id] = messages.count - 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// MARK: - Public timeline derived views
|
||||||
|
|
||||||
|
extension ConversationStore {
|
||||||
|
/// Removes a message by ID from whichever public (mesh/geohash)
|
||||||
|
/// conversation contains it. Returns the removed message, if any.
|
||||||
|
@discardableResult
|
||||||
|
func removePublicMessage(withID messageID: String) -> BitchatMessage? {
|
||||||
|
for id in conversationIDs(forMessageID: messageID) {
|
||||||
|
switch id {
|
||||||
|
case .mesh, .geohash:
|
||||||
|
return removeMessage(withID: messageID, from: id)
|
||||||
|
case .direct:
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,187 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
|
import SwiftUI
|
||||||
|
#if os(iOS)
|
||||||
|
import UIKit
|
||||||
|
#endif
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class ConversationUIModel: ObservableObject {
|
||||||
|
@Published private(set) var showAutocomplete = false
|
||||||
|
@Published private(set) var autocompleteSuggestions: [String] = []
|
||||||
|
@Published private(set) var currentNickname: String
|
||||||
|
@Published private(set) var isBatchingPublic = false
|
||||||
|
@Published private(set) var canSendMediaInCurrentContext = true
|
||||||
|
|
||||||
|
private let chatViewModel: ChatViewModel
|
||||||
|
private let privateConversationModel: PrivateConversationModel
|
||||||
|
private let conversations: ConversationStore
|
||||||
|
private var activeChannel: ChannelID
|
||||||
|
private var cancellables = Set<AnyCancellable>()
|
||||||
|
|
||||||
|
init(
|
||||||
|
chatViewModel: ChatViewModel,
|
||||||
|
privateConversationModel: PrivateConversationModel,
|
||||||
|
conversations: ConversationStore
|
||||||
|
) {
|
||||||
|
self.chatViewModel = chatViewModel
|
||||||
|
self.privateConversationModel = privateConversationModel
|
||||||
|
self.conversations = conversations
|
||||||
|
self.activeChannel = conversations.activeChannel
|
||||||
|
self.currentNickname = chatViewModel.nickname
|
||||||
|
self.isBatchingPublic = chatViewModel.isBatchingPublic
|
||||||
|
self.showAutocomplete = chatViewModel.showAutocomplete
|
||||||
|
self.autocompleteSuggestions = chatViewModel.autocompleteSuggestions
|
||||||
|
self.canSendMediaInCurrentContext = chatViewModel.canSendMediaInCurrentContext
|
||||||
|
|
||||||
|
bind()
|
||||||
|
}
|
||||||
|
|
||||||
|
func setCurrentColorScheme(_ colorScheme: ColorScheme) {
|
||||||
|
chatViewModel.currentColorScheme = colorScheme
|
||||||
|
}
|
||||||
|
|
||||||
|
func setCurrentTheme(_ theme: AppTheme) {
|
||||||
|
chatViewModel.currentTheme = theme
|
||||||
|
}
|
||||||
|
|
||||||
|
func sendMessage(_ message: String) {
|
||||||
|
chatViewModel.sendMessage(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearCurrentConversation() {
|
||||||
|
chatViewModel.sendMessage("/clear")
|
||||||
|
}
|
||||||
|
|
||||||
|
func sendHug(to sender: String) {
|
||||||
|
chatViewModel.sendMessage("/hug @\(sender)")
|
||||||
|
}
|
||||||
|
|
||||||
|
func sendSlap(to sender: String) {
|
||||||
|
chatViewModel.sendMessage("/slap @\(sender)")
|
||||||
|
}
|
||||||
|
|
||||||
|
func block(peerID: PeerID?, displayName: String?) {
|
||||||
|
guard let displayName else { return }
|
||||||
|
|
||||||
|
if let peerID, peerID.isGeoChat,
|
||||||
|
let full = chatViewModel.fullNostrHex(forSenderPeerID: peerID) {
|
||||||
|
chatViewModel.blockGeohashUser(pubkeyHexLowercased: full, displayName: displayName)
|
||||||
|
} else {
|
||||||
|
chatViewModel.sendMessage("/block \(displayName)")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func updateAutocomplete(for text: String, cursorPosition: Int) {
|
||||||
|
chatViewModel.updateAutocomplete(for: text, cursorPosition: cursorPosition)
|
||||||
|
}
|
||||||
|
|
||||||
|
func completeNickname(_ nickname: String, in text: inout String) -> Int {
|
||||||
|
chatViewModel.completeNickname(nickname, in: &text)
|
||||||
|
}
|
||||||
|
|
||||||
|
func formatMessage(_ message: BitchatMessage, colorScheme: ColorScheme, theme: AppTheme? = nil) -> AttributedString {
|
||||||
|
chatViewModel.formatMessageAsText(message, colorScheme: colorScheme, theme: theme)
|
||||||
|
}
|
||||||
|
|
||||||
|
func formatMessageHeader(_ message: BitchatMessage, colorScheme: ColorScheme, theme: AppTheme? = nil) -> AttributedString {
|
||||||
|
chatViewModel.formatMessageHeader(message, colorScheme: colorScheme, theme: theme)
|
||||||
|
}
|
||||||
|
|
||||||
|
func mediaAttachment(for message: BitchatMessage) -> BitchatMessage.Media? {
|
||||||
|
message.mediaAttachment(for: currentNickname)
|
||||||
|
}
|
||||||
|
|
||||||
|
func isSelfSender(peerID: PeerID?, displayName: String?) -> Bool {
|
||||||
|
chatViewModel.isSelfSender(peerID: peerID, displayName: displayName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func isSentByCurrentUser(_ message: BitchatMessage) -> Bool {
|
||||||
|
message.sender == currentNickname || message.sender.hasPrefix(currentNickname + "#")
|
||||||
|
}
|
||||||
|
|
||||||
|
func isMediaMessageFromCurrentUser(_ message: BitchatMessage) -> Bool {
|
||||||
|
message.sender == currentNickname || message.senderPeerID == chatViewModel.meshService.myPeerID
|
||||||
|
}
|
||||||
|
|
||||||
|
func senderDisplayName(for peerID: PeerID, fallbackMessages: [BitchatMessage]) -> String? {
|
||||||
|
if peerID.isGeoDM || peerID.isGeoChat {
|
||||||
|
return chatViewModel.geohashDisplayName(for: peerID)
|
||||||
|
}
|
||||||
|
if let nickname = chatViewModel.meshService.peerNickname(peerID: peerID) {
|
||||||
|
return nickname
|
||||||
|
}
|
||||||
|
return fallbackMessages.last(where: { $0.senderPeerID == peerID && $0.sender != "system" })?.sender
|
||||||
|
}
|
||||||
|
|
||||||
|
#if os(iOS)
|
||||||
|
func processSelectedImage(_ image: UIImage?) {
|
||||||
|
chatViewModel.processThenSendImage(image)
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
func processSelectedImage(from url: URL?) {
|
||||||
|
#if os(macOS)
|
||||||
|
chatViewModel.processThenSendImage(from: url)
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
func sendVoiceNote(at url: URL) {
|
||||||
|
chatViewModel.sendVoiceNote(at: url)
|
||||||
|
}
|
||||||
|
|
||||||
|
func cancelMediaSend(messageID: String) {
|
||||||
|
chatViewModel.cancelMediaSend(messageID: messageID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func deleteMediaMessage(messageID: String) {
|
||||||
|
chatViewModel.deleteMediaMessage(messageID: messageID)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func bind() {
|
||||||
|
chatViewModel.$nickname
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$currentNickname)
|
||||||
|
|
||||||
|
chatViewModel.$showAutocomplete
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$showAutocomplete)
|
||||||
|
|
||||||
|
chatViewModel.$autocompleteSuggestions
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$autocompleteSuggestions)
|
||||||
|
|
||||||
|
chatViewModel.$isBatchingPublic
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$isBatchingPublic)
|
||||||
|
|
||||||
|
conversations.$activeChannel
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] channel in
|
||||||
|
self?.activeChannel = channel
|
||||||
|
self?.refreshComputedState()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
privateConversationModel.$selectedPeerID
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refreshComputedState()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func refreshComputedState() {
|
||||||
|
if let selectedPeerID = privateConversationModel.selectedPeerID {
|
||||||
|
canSendMediaInCurrentContext = !(selectedPeerID.isGeoDM || selectedPeerID.isGeoChat)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
switch activeChannel {
|
||||||
|
case .mesh:
|
||||||
|
canSendMediaInCurrentContext = true
|
||||||
|
case .location:
|
||||||
|
canSendMediaInCurrentContext = false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,176 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class LocationChannelsModel: ObservableObject {
|
||||||
|
@Published private(set) var permissionState: LocationChannelManager.PermissionState
|
||||||
|
@Published private(set) var availableChannels: [GeohashChannel]
|
||||||
|
@Published private(set) var selectedChannel: ChannelID
|
||||||
|
@Published private(set) var teleported: Bool
|
||||||
|
@Published private(set) var bookmarks: [String]
|
||||||
|
@Published private(set) var bookmarkNames: [String: String]
|
||||||
|
@Published private(set) var locationNames: [GeohashChannelLevel: String]
|
||||||
|
@Published private(set) var userTorEnabled: Bool
|
||||||
|
|
||||||
|
private let manager: LocationChannelManager
|
||||||
|
private let network: NetworkActivationService
|
||||||
|
private var cancellables = Set<AnyCancellable>()
|
||||||
|
|
||||||
|
init(
|
||||||
|
manager: LocationChannelManager? = nil,
|
||||||
|
network: NetworkActivationService? = nil
|
||||||
|
) {
|
||||||
|
let manager = manager ?? .shared
|
||||||
|
let network = network ?? .shared
|
||||||
|
|
||||||
|
self.manager = manager
|
||||||
|
self.network = network
|
||||||
|
self.permissionState = manager.permissionState
|
||||||
|
self.availableChannels = manager.availableChannels
|
||||||
|
self.selectedChannel = manager.selectedChannel
|
||||||
|
self.teleported = manager.teleported
|
||||||
|
self.bookmarks = manager.bookmarks
|
||||||
|
self.bookmarkNames = manager.bookmarkNames
|
||||||
|
self.locationNames = manager.locationNames
|
||||||
|
self.userTorEnabled = network.userTorEnabled
|
||||||
|
|
||||||
|
bind()
|
||||||
|
}
|
||||||
|
|
||||||
|
var currentBuildingGeohash: String? {
|
||||||
|
availableChannels.first(where: { $0.level == .building })?.geohash
|
||||||
|
}
|
||||||
|
|
||||||
|
func isSelected(_ channel: GeohashChannel) -> Bool {
|
||||||
|
guard case .location(let selected) = selectedChannel else { return false }
|
||||||
|
return selected == channel
|
||||||
|
}
|
||||||
|
|
||||||
|
func isBookmarked(_ geohash: String) -> Bool {
|
||||||
|
manager.isBookmarked(geohash)
|
||||||
|
}
|
||||||
|
|
||||||
|
func enableLocationChannels() {
|
||||||
|
manager.enableLocationChannels()
|
||||||
|
}
|
||||||
|
|
||||||
|
func refreshChannels() {
|
||||||
|
manager.refreshChannels()
|
||||||
|
}
|
||||||
|
|
||||||
|
func enableAndRefresh() {
|
||||||
|
manager.enableLocationChannels()
|
||||||
|
manager.refreshChannels()
|
||||||
|
}
|
||||||
|
|
||||||
|
func beginLiveRefresh() {
|
||||||
|
manager.beginLiveRefresh()
|
||||||
|
}
|
||||||
|
|
||||||
|
func endLiveRefresh() {
|
||||||
|
manager.endLiveRefresh()
|
||||||
|
}
|
||||||
|
|
||||||
|
func select(_ channel: ChannelID) {
|
||||||
|
manager.select(channel)
|
||||||
|
}
|
||||||
|
|
||||||
|
func markTeleported(for geohash: String, _ flag: Bool) {
|
||||||
|
manager.markTeleported(for: geohash, flag)
|
||||||
|
}
|
||||||
|
|
||||||
|
func toggleBookmark(_ geohash: String) {
|
||||||
|
manager.toggleBookmark(geohash)
|
||||||
|
}
|
||||||
|
|
||||||
|
func resolveBookmarkNameIfNeeded(for geohash: String) {
|
||||||
|
manager.resolveBookmarkNameIfNeeded(for: geohash)
|
||||||
|
}
|
||||||
|
|
||||||
|
func locationName(for level: GeohashChannelLevel) -> String? {
|
||||||
|
locationNames[level]
|
||||||
|
}
|
||||||
|
|
||||||
|
func setUserTorEnabled(_ enabled: Bool) {
|
||||||
|
network.setUserTorEnabled(enabled)
|
||||||
|
}
|
||||||
|
|
||||||
|
func refreshMeshChannelsIfNeeded() {
|
||||||
|
guard case .mesh = selectedChannel,
|
||||||
|
permissionState == .authorized,
|
||||||
|
availableChannels.isEmpty else {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
refreshChannels()
|
||||||
|
}
|
||||||
|
|
||||||
|
func openLocationChannel(for geohash: String) {
|
||||||
|
let normalized = geohash.trimmingCharacters(in: .whitespacesAndNewlines).lowercased()
|
||||||
|
let allowed = Set("0123456789bcdefghjkmnpqrstuvwxyz")
|
||||||
|
guard (2...12).contains(normalized.count),
|
||||||
|
normalized.allSatisfy({ allowed.contains($0) }) else {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let channel = GeohashChannel(level: level(forLength: normalized.count), geohash: normalized)
|
||||||
|
let isRegional = availableChannels.contains { $0.geohash == normalized }
|
||||||
|
if !isRegional && !availableChannels.isEmpty {
|
||||||
|
markTeleported(for: normalized, true)
|
||||||
|
}
|
||||||
|
select(.location(channel))
|
||||||
|
}
|
||||||
|
|
||||||
|
func teleport(to geohash: String) {
|
||||||
|
let normalized = geohash.trimmingCharacters(in: .whitespacesAndNewlines).lowercased()
|
||||||
|
let channel = GeohashChannel(level: level(forLength: normalized.count), geohash: normalized)
|
||||||
|
markTeleported(for: normalized, true)
|
||||||
|
select(.location(channel))
|
||||||
|
}
|
||||||
|
|
||||||
|
private func bind() {
|
||||||
|
manager.$permissionState
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$permissionState)
|
||||||
|
|
||||||
|
manager.$availableChannels
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$availableChannels)
|
||||||
|
|
||||||
|
manager.$selectedChannel
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$selectedChannel)
|
||||||
|
|
||||||
|
manager.$teleported
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$teleported)
|
||||||
|
|
||||||
|
manager.$bookmarks
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$bookmarks)
|
||||||
|
|
||||||
|
manager.$bookmarkNames
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$bookmarkNames)
|
||||||
|
|
||||||
|
manager.$locationNames
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$locationNames)
|
||||||
|
|
||||||
|
network.$userTorEnabled
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$userTorEnabled)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func level(forLength length: Int) -> GeohashChannelLevel {
|
||||||
|
switch length {
|
||||||
|
case 0...2: return .region
|
||||||
|
case 3...4: return .province
|
||||||
|
case 5: return .city
|
||||||
|
case 6: return .neighborhood
|
||||||
|
case 7: return .block
|
||||||
|
case 8...12: return .building
|
||||||
|
default: return .block
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
import Combine
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class LocationPresenceStore: ObservableObject {
|
||||||
|
@Published private(set) var currentGeohash: String?
|
||||||
|
@Published private(set) var geoNicknames: [String: String] = [:]
|
||||||
|
@Published private(set) var teleportedGeo: Set<String> = []
|
||||||
|
|
||||||
|
func setCurrentGeohash(_ geohash: String?) {
|
||||||
|
currentGeohash = geohash?.lowercased()
|
||||||
|
}
|
||||||
|
|
||||||
|
func setNickname(_ nickname: String, for pubkeyHex: String) {
|
||||||
|
geoNicknames[pubkeyHex.lowercased()] = nickname
|
||||||
|
}
|
||||||
|
|
||||||
|
func replaceGeoNicknames(_ nicknames: [String: String]) {
|
||||||
|
geoNicknames = Dictionary(
|
||||||
|
uniqueKeysWithValues: nicknames.map { key, value in
|
||||||
|
(key.lowercased(), value)
|
||||||
|
}
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearGeoNicknames() {
|
||||||
|
geoNicknames.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
func markTeleported(_ pubkeyHex: String) {
|
||||||
|
teleportedGeo.insert(pubkeyHex.lowercased())
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearTeleported(_ pubkeyHex: String) {
|
||||||
|
teleportedGeo.remove(pubkeyHex.lowercased())
|
||||||
|
}
|
||||||
|
|
||||||
|
func replaceTeleportedGeo(_ pubkeys: Set<String>) {
|
||||||
|
teleportedGeo = Set(pubkeys.map { $0.lowercased() })
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearTeleportedGeo() {
|
||||||
|
teleportedGeo.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
func reset() {
|
||||||
|
currentGeohash = nil
|
||||||
|
geoNicknames.removeAll()
|
||||||
|
teleportedGeo.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,125 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class PeerIdentityStore: ObservableObject {
|
||||||
|
@Published private(set) var encryptionStatuses: [PeerID: EncryptionStatus] = [:]
|
||||||
|
@Published private(set) var verifiedFingerprints: Set<String> = []
|
||||||
|
|
||||||
|
private(set) var peerFingerprintsByPeerID: [PeerID: String] = [:]
|
||||||
|
private(set) var selectedPrivateChatFingerprint: String?
|
||||||
|
|
||||||
|
private var stablePeerIDsByShortID: [PeerID: PeerID] = [:]
|
||||||
|
private var encryptionStatusCache: [PeerID: EncryptionStatus] = [:]
|
||||||
|
|
||||||
|
func stablePeerID(forShortID peerID: PeerID) -> PeerID? {
|
||||||
|
stablePeerIDsByShortID[peerID]
|
||||||
|
}
|
||||||
|
|
||||||
|
func shortPeerID(forStablePeerID stablePeerID: PeerID) -> PeerID? {
|
||||||
|
stablePeerIDsByShortID.first(where: { $0.value == stablePeerID })?.key
|
||||||
|
}
|
||||||
|
|
||||||
|
func setStablePeerID(_ stablePeerID: PeerID, forShortID peerID: PeerID) {
|
||||||
|
stablePeerIDsByShortID[peerID] = stablePeerID
|
||||||
|
}
|
||||||
|
|
||||||
|
func replaceStablePeerIDs(_ mappings: [PeerID: PeerID]) {
|
||||||
|
stablePeerIDsByShortID = mappings
|
||||||
|
}
|
||||||
|
|
||||||
|
func fingerprint(for peerID: PeerID) -> String? {
|
||||||
|
peerFingerprintsByPeerID[peerID]
|
||||||
|
}
|
||||||
|
|
||||||
|
func setFingerprint(_ fingerprint: String?, for peerID: PeerID) {
|
||||||
|
if let fingerprint {
|
||||||
|
peerFingerprintsByPeerID[peerID] = fingerprint
|
||||||
|
} else {
|
||||||
|
peerFingerprintsByPeerID.removeValue(forKey: peerID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func replaceFingerprintMappings(_ mappings: [PeerID: String]) {
|
||||||
|
peerFingerprintsByPeerID = mappings
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
func migrateFingerprintMapping(
|
||||||
|
from oldPeerID: PeerID,
|
||||||
|
to newPeerID: PeerID,
|
||||||
|
fallback: String? = nil
|
||||||
|
) -> String? {
|
||||||
|
let fingerprint = peerFingerprintsByPeerID.removeValue(forKey: oldPeerID) ?? fallback
|
||||||
|
if let fingerprint {
|
||||||
|
peerFingerprintsByPeerID[newPeerID] = fingerprint
|
||||||
|
if selectedPrivateChatFingerprint == nil {
|
||||||
|
selectedPrivateChatFingerprint = fingerprint
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return fingerprint
|
||||||
|
}
|
||||||
|
|
||||||
|
func setSelectedPrivateChatFingerprint(_ fingerprint: String?) {
|
||||||
|
selectedPrivateChatFingerprint = fingerprint
|
||||||
|
}
|
||||||
|
|
||||||
|
func cachedEncryptionStatus(for peerID: PeerID) -> EncryptionStatus? {
|
||||||
|
encryptionStatusCache[peerID]
|
||||||
|
}
|
||||||
|
|
||||||
|
func setCachedEncryptionStatus(_ status: EncryptionStatus, for peerID: PeerID) {
|
||||||
|
encryptionStatusCache[peerID] = status
|
||||||
|
}
|
||||||
|
|
||||||
|
func invalidateEncryptionCache(for peerID: PeerID? = nil) {
|
||||||
|
if let peerID {
|
||||||
|
encryptionStatusCache.removeValue(forKey: peerID)
|
||||||
|
} else {
|
||||||
|
encryptionStatusCache.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func encryptionStatus(for peerID: PeerID) -> EncryptionStatus? {
|
||||||
|
encryptionStatuses[peerID]
|
||||||
|
}
|
||||||
|
|
||||||
|
func setEncryptionStatus(_ status: EncryptionStatus?, for peerID: PeerID) {
|
||||||
|
if let status {
|
||||||
|
encryptionStatuses[peerID] = status
|
||||||
|
} else {
|
||||||
|
encryptionStatuses.removeValue(forKey: peerID)
|
||||||
|
}
|
||||||
|
invalidateEncryptionCache(for: peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func replaceEncryptionStatuses(_ statuses: [PeerID: EncryptionStatus]) {
|
||||||
|
encryptionStatuses = statuses
|
||||||
|
}
|
||||||
|
|
||||||
|
func setVerifiedFingerprints(_ fingerprints: Set<String>) {
|
||||||
|
verifiedFingerprints = fingerprints
|
||||||
|
}
|
||||||
|
|
||||||
|
func setVerified(_ fingerprint: String, verified: Bool) {
|
||||||
|
if verified {
|
||||||
|
verifiedFingerprints.insert(fingerprint)
|
||||||
|
} else {
|
||||||
|
verifiedFingerprints.remove(fingerprint)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func isVerified(_ fingerprint: String) -> Bool {
|
||||||
|
verifiedFingerprints.contains(fingerprint)
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearAll() {
|
||||||
|
encryptionStatuses.removeAll()
|
||||||
|
verifiedFingerprints.removeAll()
|
||||||
|
peerFingerprintsByPeerID.removeAll()
|
||||||
|
selectedPrivateChatFingerprint = nil
|
||||||
|
stablePeerIDsByShortID.removeAll()
|
||||||
|
encryptionStatusCache.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,260 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
|
import SwiftUI
|
||||||
|
|
||||||
|
struct MeshPeerRow: Identifiable, Equatable {
|
||||||
|
let peerID: PeerID
|
||||||
|
let displayName: String
|
||||||
|
let isMe: Bool
|
||||||
|
let hasUnread: Bool
|
||||||
|
let isBlocked: Bool
|
||||||
|
let isFavorite: Bool
|
||||||
|
let isConnected: Bool
|
||||||
|
let isReachable: Bool
|
||||||
|
let isMutualFavorite: Bool
|
||||||
|
let encryptionStatus: EncryptionStatus
|
||||||
|
let showsVerifiedBadgeWhenOffline: Bool
|
||||||
|
|
||||||
|
var id: String { peerID.id }
|
||||||
|
}
|
||||||
|
|
||||||
|
struct GeohashPersonRow: Identifiable, Equatable {
|
||||||
|
let id: String
|
||||||
|
let displayName: String
|
||||||
|
let isMe: Bool
|
||||||
|
let isTeleported: Bool
|
||||||
|
let isBlocked: Bool
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class PeerListModel: ObservableObject {
|
||||||
|
@Published private(set) var allPeers: [BitchatPeer] = []
|
||||||
|
@Published private(set) var meshRows: [MeshPeerRow] = []
|
||||||
|
@Published private(set) var geohashPeople: [GeohashPersonRow] = []
|
||||||
|
@Published private(set) var reachableMeshPeerCount = 0
|
||||||
|
@Published private(set) var connectedMeshPeerCount = 0
|
||||||
|
@Published private(set) var visibleGeohashPeerCount = 0
|
||||||
|
@Published private(set) var renderID = ""
|
||||||
|
|
||||||
|
private let chatViewModel: ChatViewModel
|
||||||
|
private let conversations: ConversationStore
|
||||||
|
private let locationChannelsModel: LocationChannelsModel
|
||||||
|
private let peerIdentityStore: PeerIdentityStore
|
||||||
|
private let locationPresenceStore: LocationPresenceStore
|
||||||
|
private var cancellables = Set<AnyCancellable>()
|
||||||
|
|
||||||
|
init(
|
||||||
|
chatViewModel: ChatViewModel,
|
||||||
|
conversations: ConversationStore,
|
||||||
|
locationChannelsModel: LocationChannelsModel? = nil,
|
||||||
|
peerIdentityStore: PeerIdentityStore? = nil,
|
||||||
|
locationPresenceStore: LocationPresenceStore? = nil
|
||||||
|
) {
|
||||||
|
self.chatViewModel = chatViewModel
|
||||||
|
self.conversations = conversations
|
||||||
|
self.locationChannelsModel = locationChannelsModel ?? LocationChannelsModel()
|
||||||
|
self.peerIdentityStore = peerIdentityStore ?? chatViewModel.peerIdentityStore
|
||||||
|
self.locationPresenceStore = locationPresenceStore ?? chatViewModel.locationPresenceStore
|
||||||
|
self.allPeers = chatViewModel.allPeers
|
||||||
|
|
||||||
|
bind()
|
||||||
|
refresh()
|
||||||
|
}
|
||||||
|
|
||||||
|
func colorForMeshPeer(id peerID: PeerID, isDark: Bool) -> Color {
|
||||||
|
chatViewModel.colorForMeshPeer(id: peerID, isDark: isDark)
|
||||||
|
}
|
||||||
|
|
||||||
|
func colorForGeohashPerson(id: String, isDark: Bool) -> Color {
|
||||||
|
chatViewModel.colorForNostrPubkey(id, isDark: isDark)
|
||||||
|
}
|
||||||
|
|
||||||
|
func participantCount(for geohash: String) -> Int {
|
||||||
|
chatViewModel.geohashParticipantCount(for: geohash)
|
||||||
|
}
|
||||||
|
|
||||||
|
func startConversation(with peerID: PeerID) {
|
||||||
|
chatViewModel.startPrivateChat(with: peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func toggleFavorite(peerID: PeerID) {
|
||||||
|
chatViewModel.toggleFavorite(peerID: peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func openGeohashDirectMessage(with pubkeyHex: String) {
|
||||||
|
chatViewModel.startGeohashDM(withPubkeyHex: pubkeyHex)
|
||||||
|
}
|
||||||
|
|
||||||
|
func blockGeohashUser(pubkeyHexLowercased: String, displayName: String) {
|
||||||
|
chatViewModel.blockGeohashUser(
|
||||||
|
pubkeyHexLowercased: pubkeyHexLowercased,
|
||||||
|
displayName: displayName
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
func unblockGeohashUser(pubkeyHexLowercased: String, displayName: String) {
|
||||||
|
chatViewModel.unblockGeohashUser(
|
||||||
|
pubkeyHexLowercased: pubkeyHexLowercased,
|
||||||
|
displayName: displayName
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func bind() {
|
||||||
|
chatViewModel.$allPeers
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] peers in
|
||||||
|
self?.allPeers = peers
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
chatViewModel.$nickname
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
locationPresenceStore.$teleportedGeo
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
conversations.$unreadConversations
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
peerIdentityStore.$encryptionStatuses
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
peerIdentityStore.$verifiedFingerprints
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
NotificationCenter.default.publisher(for: Notification.Name("peerStatusUpdated"))
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
chatViewModel.participantTracker.$visiblePeople
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
locationChannelsModel.$selectedChannel
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
locationChannelsModel.$teleported
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
locationChannelsModel.$availableChannels
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func refresh() {
|
||||||
|
let myPeerID = chatViewModel.meshService.myPeerID
|
||||||
|
let meshRows = allPeers.map { peer in
|
||||||
|
let isMe = peer.peerID == myPeerID
|
||||||
|
let verifiedBadge: Bool
|
||||||
|
if !isMe && !peer.isConnected,
|
||||||
|
let fingerprint = chatViewModel.getFingerprint(for: peer.peerID) {
|
||||||
|
verifiedBadge = peerIdentityStore.isVerified(fingerprint)
|
||||||
|
} else {
|
||||||
|
verifiedBadge = false
|
||||||
|
}
|
||||||
|
|
||||||
|
return MeshPeerRow(
|
||||||
|
peerID: peer.peerID,
|
||||||
|
displayName: isMe ? chatViewModel.nickname : peer.nickname,
|
||||||
|
isMe: isMe,
|
||||||
|
hasUnread: chatViewModel.hasUnreadMessages(for: peer.peerID),
|
||||||
|
isBlocked: !isMe && chatViewModel.isPeerBlocked(peer.peerID),
|
||||||
|
isFavorite: peer.favoriteStatus?.isFavorite ?? false,
|
||||||
|
isConnected: peer.isConnected,
|
||||||
|
isReachable: peer.isReachable,
|
||||||
|
isMutualFavorite: peer.isMutualFavorite,
|
||||||
|
encryptionStatus: chatViewModel.getEncryptionStatus(for: peer.peerID),
|
||||||
|
showsVerifiedBadgeWhenOffline: verifiedBadge
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
let meshCounts = meshRows.reduce(into: (reachable: 0, connected: 0)) { counts, row in
|
||||||
|
guard !row.isMe else { return }
|
||||||
|
if row.isConnected {
|
||||||
|
counts.connected += 1
|
||||||
|
counts.reachable += 1
|
||||||
|
} else if row.isReachable {
|
||||||
|
counts.reachable += 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let geohashPeople = buildGeohashPeople()
|
||||||
|
|
||||||
|
self.meshRows = meshRows
|
||||||
|
reachableMeshPeerCount = meshCounts.reachable
|
||||||
|
connectedMeshPeerCount = meshCounts.connected
|
||||||
|
self.geohashPeople = geohashPeople
|
||||||
|
visibleGeohashPeerCount = geohashPeople.count
|
||||||
|
renderID = (
|
||||||
|
meshRows.map {
|
||||||
|
"\($0.id)-\($0.isConnected)-\($0.isReachable)-\($0.hasUnread)-\($0.isFavorite)-\($0.isBlocked)"
|
||||||
|
} +
|
||||||
|
geohashPeople.map {
|
||||||
|
"geo:\($0.id)-\($0.isTeleported)-\($0.isBlocked)-\($0.displayName)"
|
||||||
|
}
|
||||||
|
).joined(separator: "|")
|
||||||
|
}
|
||||||
|
|
||||||
|
private func buildGeohashPeople() -> [GeohashPersonRow] {
|
||||||
|
let myHex = currentGeohashIdentityHex()
|
||||||
|
let teleportedSet = Set(locationPresenceStore.teleportedGeo.map { $0.lowercased() })
|
||||||
|
|
||||||
|
return chatViewModel.visibleGeohashPeople().map { person in
|
||||||
|
let isMe = person.id == myHex
|
||||||
|
return GeohashPersonRow(
|
||||||
|
id: person.id,
|
||||||
|
displayName: person.displayName,
|
||||||
|
isMe: isMe,
|
||||||
|
isTeleported: teleportedSet.contains(person.id.lowercased()) || (isMe && locationChannelsModel.teleported),
|
||||||
|
isBlocked: !isMe && chatViewModel.isGeohashUserBlocked(pubkeyHexLowercased: person.id)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func currentGeohashIdentityHex() -> String? {
|
||||||
|
guard case .location(let channel) = locationChannelsModel.selectedChannel,
|
||||||
|
let identity = try? chatViewModel.idBridge.deriveIdentity(forGeohash: channel.geohash) else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return identity.publicKeyHex.lowercased()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,323 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
/// Feature model for private (direct) conversations.
|
||||||
|
///
|
||||||
|
/// Reads the single-writer `ConversationStore` directly: `messages(for:)`
|
||||||
|
/// returns the peer's conversation backing array (no mirror dictionary), and
|
||||||
|
/// the store's typed `changes` subject drives invalidation — a change in the
|
||||||
|
/// SELECTED peer's conversation republishes this model, while appends to
|
||||||
|
/// other private chats only surface through the unread set. Direct
|
||||||
|
/// conversations are keyed by raw routing peer ID; the coordinators'
|
||||||
|
/// ephemeral/stable mirroring guarantees the selected peer's key always
|
||||||
|
/// holds the full timeline (see `ConversationID.directPeer`).
|
||||||
|
@MainActor
|
||||||
|
final class PrivateInboxModel: ObservableObject {
|
||||||
|
@Published private(set) var selectedPeerID: PeerID?
|
||||||
|
@Published private(set) var unreadPeerIDs: Set<PeerID> = []
|
||||||
|
|
||||||
|
private let conversations: ConversationStore
|
||||||
|
private var cancellables = Set<AnyCancellable>()
|
||||||
|
|
||||||
|
init(conversations: ConversationStore) {
|
||||||
|
self.conversations = conversations
|
||||||
|
self.selectedPeerID = conversations.selectedPrivatePeerID
|
||||||
|
self.unreadPeerIDs = conversations.unreadDirectRoutingPeerIDs()
|
||||||
|
|
||||||
|
bind()
|
||||||
|
}
|
||||||
|
|
||||||
|
func messages(for peerID: PeerID?) -> [BitchatMessage] {
|
||||||
|
guard let peerID else { return [] }
|
||||||
|
return conversations.conversationsByID[.directPeer(peerID)]?.messages ?? []
|
||||||
|
}
|
||||||
|
|
||||||
|
private func bind() {
|
||||||
|
conversations.$selectedPrivatePeerID
|
||||||
|
.dropFirst()
|
||||||
|
.sink { [weak self] peerID in
|
||||||
|
guard let self, self.selectedPeerID != peerID else { return }
|
||||||
|
self.selectedPeerID = peerID
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
conversations.changes
|
||||||
|
.sink { [weak self] change in
|
||||||
|
self?.apply(change)
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func apply(_ change: ConversationChange) {
|
||||||
|
switch change {
|
||||||
|
case .appended(let id, _),
|
||||||
|
.updated(let id, _),
|
||||||
|
.statusChanged(let id, _, _),
|
||||||
|
.messageRemoved(let id, _),
|
||||||
|
.cleared(let id):
|
||||||
|
republishIfSelected(id)
|
||||||
|
|
||||||
|
case .unreadChanged(let id, _):
|
||||||
|
guard isDirect(id) else { return }
|
||||||
|
refreshUnreadPeerIDs()
|
||||||
|
|
||||||
|
case .removed(let id):
|
||||||
|
guard isDirect(id) else { return }
|
||||||
|
refreshUnreadPeerIDs()
|
||||||
|
republishIfSelected(id)
|
||||||
|
|
||||||
|
case .migrated(let source, let destination):
|
||||||
|
guard isDirect(source) || isDirect(destination) else { return }
|
||||||
|
refreshUnreadPeerIDs()
|
||||||
|
republishIfSelected(source)
|
||||||
|
republishIfSelected(destination)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func republishIfSelected(_ id: ConversationID) {
|
||||||
|
guard let selectedPeerID, id == .directPeer(selectedPeerID) else { return }
|
||||||
|
objectWillChange.send()
|
||||||
|
}
|
||||||
|
|
||||||
|
private func refreshUnreadPeerIDs() {
|
||||||
|
let next = conversations.unreadDirectRoutingPeerIDs()
|
||||||
|
guard unreadPeerIDs != next else { return }
|
||||||
|
unreadPeerIDs = next
|
||||||
|
}
|
||||||
|
|
||||||
|
private func isDirect(_ id: ConversationID) -> Bool {
|
||||||
|
if case .direct = id { return true }
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
enum PrivateConversationAvailability: Equatable {
|
||||||
|
case bluetoothConnected
|
||||||
|
case meshReachable
|
||||||
|
case nostrAvailable
|
||||||
|
case offline
|
||||||
|
}
|
||||||
|
|
||||||
|
struct PrivateConversationHeaderState: Equatable {
|
||||||
|
let conversationPeerID: PeerID
|
||||||
|
let headerPeerID: PeerID
|
||||||
|
let displayName: String
|
||||||
|
let availability: PrivateConversationAvailability
|
||||||
|
let isFavorite: Bool
|
||||||
|
let encryptionStatus: EncryptionStatus?
|
||||||
|
|
||||||
|
var supportsFavoriteToggle: Bool {
|
||||||
|
!conversationPeerID.isGeoDM
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class PrivateConversationModel: ObservableObject {
|
||||||
|
@Published private(set) var selectedPeerID: PeerID?
|
||||||
|
@Published private(set) var selectedHeaderState: PrivateConversationHeaderState?
|
||||||
|
|
||||||
|
private let chatViewModel: ChatViewModel
|
||||||
|
private let conversations: ConversationStore
|
||||||
|
private let locationChannelsModel: LocationChannelsModel
|
||||||
|
private let peerIdentityStore: PeerIdentityStore
|
||||||
|
private var cancellables = Set<AnyCancellable>()
|
||||||
|
|
||||||
|
init(
|
||||||
|
chatViewModel: ChatViewModel,
|
||||||
|
conversations: ConversationStore,
|
||||||
|
locationChannelsModel: LocationChannelsModel? = nil,
|
||||||
|
peerIdentityStore: PeerIdentityStore? = nil
|
||||||
|
) {
|
||||||
|
self.chatViewModel = chatViewModel
|
||||||
|
self.conversations = conversations
|
||||||
|
self.locationChannelsModel = locationChannelsModel ?? LocationChannelsModel()
|
||||||
|
self.peerIdentityStore = peerIdentityStore ?? chatViewModel.peerIdentityStore
|
||||||
|
let initialPeerID = conversations.selectedPrivatePeerID
|
||||||
|
self.selectedPeerID = initialPeerID
|
||||||
|
self.selectedHeaderState = initialPeerID.flatMap { peerID in
|
||||||
|
makeHeaderState(for: peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
bind()
|
||||||
|
}
|
||||||
|
|
||||||
|
func startConversation(with peerID: PeerID) {
|
||||||
|
chatViewModel.startPrivateChat(with: peerID)
|
||||||
|
refreshSelectedConversation()
|
||||||
|
}
|
||||||
|
|
||||||
|
func openConversation(for peerID: PeerID) {
|
||||||
|
if peerID.isGeoChat {
|
||||||
|
guard let full = chatViewModel.fullNostrHex(forSenderPeerID: peerID) else { return }
|
||||||
|
chatViewModel.startGeohashDM(withPubkeyHex: full)
|
||||||
|
} else {
|
||||||
|
chatViewModel.startPrivateChat(with: peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
refreshSelectedConversation()
|
||||||
|
}
|
||||||
|
|
||||||
|
func endConversation() {
|
||||||
|
chatViewModel.endPrivateChat()
|
||||||
|
refreshSelectedConversation()
|
||||||
|
}
|
||||||
|
|
||||||
|
func toggleFavorite(peerID: PeerID) {
|
||||||
|
chatViewModel.toggleFavorite(peerID: peerID)
|
||||||
|
refreshSelectedConversation()
|
||||||
|
}
|
||||||
|
|
||||||
|
func toggleFavoriteForSelectedConversation() {
|
||||||
|
guard let headerPeerID = selectedHeaderState?.headerPeerID else { return }
|
||||||
|
toggleFavorite(peerID: headerPeerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func markMessagesAsRead(from peerID: PeerID) {
|
||||||
|
chatViewModel.markPrivateMessagesAsRead(from: peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func bind() {
|
||||||
|
conversations.$selectedPrivatePeerID
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refreshSelectedConversation()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
chatViewModel.$allPeers
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refreshSelectedConversation()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
peerIdentityStore.$encryptionStatuses
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refreshSelectedConversation()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
NotificationCenter.default.publisher(for: .favoriteStatusChanged)
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refreshSelectedConversation()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
NotificationCenter.default.publisher(for: Notification.Name("peerStatusUpdated"))
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refreshSelectedConversation()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
locationChannelsModel.$selectedChannel
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.refreshSelectedConversation()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func refreshSelectedConversation() {
|
||||||
|
selectedPeerID = conversations.selectedPrivatePeerID
|
||||||
|
selectedHeaderState = selectedPeerID.flatMap { peerID in
|
||||||
|
makeHeaderState(for: peerID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func makeHeaderState(for conversationPeerID: PeerID) -> PrivateConversationHeaderState {
|
||||||
|
let headerPeerID = chatViewModel.getShortIDForNoiseKey(conversationPeerID)
|
||||||
|
let peer = chatViewModel.getPeer(byID: headerPeerID)
|
||||||
|
let displayName = resolveDisplayName(for: conversationPeerID, headerPeerID: headerPeerID, peer: peer)
|
||||||
|
let availability = resolveAvailability(for: headerPeerID, peer: peer)
|
||||||
|
let encryptionStatus: EncryptionStatus? = conversationPeerID.isGeoDM
|
||||||
|
? nil
|
||||||
|
: chatViewModel.getEncryptionStatus(for: headerPeerID)
|
||||||
|
|
||||||
|
return PrivateConversationHeaderState(
|
||||||
|
conversationPeerID: conversationPeerID,
|
||||||
|
headerPeerID: headerPeerID,
|
||||||
|
displayName: displayName,
|
||||||
|
availability: availability,
|
||||||
|
isFavorite: chatViewModel.isFavorite(peerID: headerPeerID),
|
||||||
|
encryptionStatus: encryptionStatus
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func resolveDisplayName(
|
||||||
|
for conversationPeerID: PeerID,
|
||||||
|
headerPeerID: PeerID,
|
||||||
|
peer: BitchatPeer?
|
||||||
|
) -> String {
|
||||||
|
if conversationPeerID.isGeoDM, case .location(let channel) = locationChannelsModel.selectedChannel {
|
||||||
|
return "#\(channel.geohash)/@\(chatViewModel.geohashDisplayName(for: conversationPeerID))"
|
||||||
|
}
|
||||||
|
if let displayName = peer?.displayName {
|
||||||
|
return displayName
|
||||||
|
}
|
||||||
|
if let nickname = chatViewModel.meshService.peerNickname(peerID: headerPeerID) {
|
||||||
|
return nickname
|
||||||
|
}
|
||||||
|
if let favorite = FavoritesPersistenceService.shared.getFavoriteStatus(
|
||||||
|
for: Data(hexString: headerPeerID.id) ?? Data()
|
||||||
|
), !favorite.peerNickname.isEmpty {
|
||||||
|
return favorite.peerNickname
|
||||||
|
}
|
||||||
|
if headerPeerID.id.count == 16 {
|
||||||
|
let candidates = chatViewModel.identityManager.getCryptoIdentitiesByPeerIDPrefix(headerPeerID)
|
||||||
|
if let identity = candidates.first,
|
||||||
|
let social = chatViewModel.identityManager.getSocialIdentity(for: identity.fingerprint) {
|
||||||
|
if let pet = social.localPetname, !pet.isEmpty {
|
||||||
|
return pet
|
||||||
|
}
|
||||||
|
if !social.claimedNickname.isEmpty {
|
||||||
|
return social.claimedNickname
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if let noiseKey = headerPeerID.noiseKey {
|
||||||
|
let fingerprint = noiseKey.sha256Fingerprint()
|
||||||
|
if let social = chatViewModel.identityManager.getSocialIdentity(for: fingerprint) {
|
||||||
|
if let pet = social.localPetname, !pet.isEmpty {
|
||||||
|
return pet
|
||||||
|
}
|
||||||
|
if !social.claimedNickname.isEmpty {
|
||||||
|
return social.claimedNickname
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return String(localized: "common.unknown", comment: "Fallback label for unknown peer")
|
||||||
|
}
|
||||||
|
|
||||||
|
private func resolveAvailability(for headerPeerID: PeerID, peer: BitchatPeer?) -> PrivateConversationAvailability {
|
||||||
|
if let connectionState = peer?.connectionState {
|
||||||
|
switch connectionState {
|
||||||
|
case .bluetoothConnected:
|
||||||
|
return .bluetoothConnected
|
||||||
|
case .meshReachable:
|
||||||
|
return .meshReachable
|
||||||
|
case .nostrAvailable:
|
||||||
|
return .nostrAvailable
|
||||||
|
case .offline:
|
||||||
|
return .offline
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if chatViewModel.meshService.isPeerReachable(headerPeerID) {
|
||||||
|
return .meshReachable
|
||||||
|
}
|
||||||
|
if let noiseKey = Data(hexString: headerPeerID.id),
|
||||||
|
let favoriteStatus = FavoritesPersistenceService.shared.getFavoriteStatus(for: noiseKey),
|
||||||
|
favoriteStatus.isMutual {
|
||||||
|
return .nostrAvailable
|
||||||
|
}
|
||||||
|
if chatViewModel.meshService.isPeerConnected(headerPeerID) || chatViewModel.connectedPeers.contains(headerPeerID) {
|
||||||
|
return .bluetoothConnected
|
||||||
|
}
|
||||||
|
|
||||||
|
return .offline
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,77 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
|
import SwiftUI
|
||||||
|
|
||||||
|
/// Feature model for the active public (mesh/geohash) timeline.
|
||||||
|
///
|
||||||
|
/// Observes ONE `Conversation` object in the single-writer
|
||||||
|
/// `ConversationStore` — the active channel's — so appends to background
|
||||||
|
/// conversations (other geohashes, private chats) never invalidate it.
|
||||||
|
/// `messages` reads the observed conversation's backing array directly;
|
||||||
|
/// there is no mirror copy.
|
||||||
|
@MainActor
|
||||||
|
final class PublicChatModel: ObservableObject {
|
||||||
|
@Published private(set) var activeChannel: ChannelID
|
||||||
|
|
||||||
|
/// The active public conversation's timeline.
|
||||||
|
var messages: [BitchatMessage] { activeConversation.messages }
|
||||||
|
|
||||||
|
private let conversations: ConversationStore
|
||||||
|
private var activeConversation: Conversation
|
||||||
|
private var activeConversationCancellable: AnyCancellable?
|
||||||
|
private var cancellables = Set<AnyCancellable>()
|
||||||
|
|
||||||
|
init(conversations: ConversationStore) {
|
||||||
|
let channel = conversations.activeChannel
|
||||||
|
self.conversations = conversations
|
||||||
|
self.activeChannel = channel
|
||||||
|
self.activeConversation = conversations.conversation(for: ConversationID(channelID: channel))
|
||||||
|
|
||||||
|
observeActiveConversation()
|
||||||
|
bind()
|
||||||
|
}
|
||||||
|
|
||||||
|
private func bind() {
|
||||||
|
conversations.$activeChannel
|
||||||
|
.dropFirst()
|
||||||
|
.sink { [weak self] channel in
|
||||||
|
guard let self else { return }
|
||||||
|
self.activeChannel = channel
|
||||||
|
self.retargetActiveConversation(to: channel)
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
// The store replaces a conversation's object when it is removed
|
||||||
|
// (panic clear); retarget to the fresh instance so the observation
|
||||||
|
// never goes stale.
|
||||||
|
conversations.changes
|
||||||
|
.sink { [weak self] change in
|
||||||
|
guard let self,
|
||||||
|
case .removed(let id) = change,
|
||||||
|
id == self.activeConversation.id else { return }
|
||||||
|
self.retargetActiveConversation(to: self.activeChannel)
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func retargetActiveConversation(to channel: ChannelID) {
|
||||||
|
let conversation = conversations.conversation(for: ConversationID(channelID: channel))
|
||||||
|
guard conversation !== activeConversation else {
|
||||||
|
// Same object (e.g. re-selected channel): keep the existing
|
||||||
|
// observation, but `messages` may still differ from what views
|
||||||
|
// last rendered, so republish.
|
||||||
|
objectWillChange.send()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
objectWillChange.send()
|
||||||
|
activeConversation = conversation
|
||||||
|
observeActiveConversation()
|
||||||
|
}
|
||||||
|
|
||||||
|
private func observeActiveConversation() {
|
||||||
|
activeConversationCancellable = activeConversation.objectWillChange
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.objectWillChange.send()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,152 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct FingerprintPresentationState: Equatable {
|
||||||
|
let statusPeerID: PeerID
|
||||||
|
let peerNickname: String
|
||||||
|
let encryptionStatus: EncryptionStatus
|
||||||
|
let theirFingerprint: String?
|
||||||
|
let myFingerprint: String
|
||||||
|
let isVerified: Bool
|
||||||
|
|
||||||
|
var canToggleVerification: Bool {
|
||||||
|
encryptionStatus == .noiseSecured || encryptionStatus == .noiseVerified
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
enum VerificationScanOutcome: Equatable {
|
||||||
|
case requested(String)
|
||||||
|
case notFound
|
||||||
|
case invalid
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class VerificationModel: ObservableObject {
|
||||||
|
@Published private(set) var currentNickname: String
|
||||||
|
@Published private(set) var selectedPeerID: PeerID?
|
||||||
|
|
||||||
|
private let chatViewModel: ChatViewModel
|
||||||
|
private let peerIdentityStore: PeerIdentityStore
|
||||||
|
private var cancellables = Set<AnyCancellable>()
|
||||||
|
|
||||||
|
init(
|
||||||
|
chatViewModel: ChatViewModel,
|
||||||
|
privateConversationModel: PrivateConversationModel,
|
||||||
|
peerIdentityStore: PeerIdentityStore? = nil
|
||||||
|
) {
|
||||||
|
self.chatViewModel = chatViewModel
|
||||||
|
self.peerIdentityStore = peerIdentityStore ?? chatViewModel.peerIdentityStore
|
||||||
|
self.currentNickname = chatViewModel.nickname
|
||||||
|
self.selectedPeerID = privateConversationModel.selectedPeerID
|
||||||
|
|
||||||
|
bind(privateConversationModel: privateConversationModel)
|
||||||
|
}
|
||||||
|
|
||||||
|
func myQRString() -> String {
|
||||||
|
let npub = try? chatViewModel.idBridge.getCurrentNostrIdentity()?.npub
|
||||||
|
return VerificationService.shared.buildMyQRString(nickname: currentNickname, npub: npub) ?? ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func beginQRVerification(with qr: VerificationService.VerificationQR) -> Bool {
|
||||||
|
chatViewModel.beginQRVerification(with: qr)
|
||||||
|
}
|
||||||
|
|
||||||
|
func verifyScannedPayload(_ payload: String) -> VerificationScanOutcome {
|
||||||
|
guard let qr = VerificationService.shared.verifyScannedQR(payload) else {
|
||||||
|
return .invalid
|
||||||
|
}
|
||||||
|
|
||||||
|
guard chatViewModel.beginQRVerification(with: qr) else {
|
||||||
|
return .notFound
|
||||||
|
}
|
||||||
|
|
||||||
|
return .requested(qr.nickname)
|
||||||
|
}
|
||||||
|
|
||||||
|
func verifyFingerprint(for peerID: PeerID) {
|
||||||
|
chatViewModel.verifyFingerprint(for: peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func unverifyFingerprint(for peerID: PeerID) {
|
||||||
|
chatViewModel.unverifyFingerprint(for: peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func isVerified(peerID: PeerID) -> Bool {
|
||||||
|
guard let fingerprint = chatViewModel.getFingerprint(for: peerID) else { return false }
|
||||||
|
return peerIdentityStore.isVerified(fingerprint)
|
||||||
|
}
|
||||||
|
|
||||||
|
func fingerprintPresentation(for peerID: PeerID) -> FingerprintPresentationState {
|
||||||
|
let statusPeerID = chatViewModel.getShortIDForNoiseKey(peerID)
|
||||||
|
let encryptionStatus = chatViewModel.getEncryptionStatus(for: statusPeerID)
|
||||||
|
let theirFingerprint = chatViewModel.getFingerprint(for: statusPeerID)
|
||||||
|
let peerNickname = resolveDisplayName(for: peerID, statusPeerID: statusPeerID)
|
||||||
|
|
||||||
|
return FingerprintPresentationState(
|
||||||
|
statusPeerID: statusPeerID,
|
||||||
|
peerNickname: peerNickname,
|
||||||
|
encryptionStatus: encryptionStatus,
|
||||||
|
theirFingerprint: theirFingerprint,
|
||||||
|
myFingerprint: chatViewModel.getMyFingerprint(),
|
||||||
|
isVerified: theirFingerprint.map { peerIdentityStore.isVerified($0) } ?? false
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func bind(privateConversationModel: PrivateConversationModel) {
|
||||||
|
chatViewModel.$nickname
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$currentNickname)
|
||||||
|
|
||||||
|
privateConversationModel.$selectedPeerID
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.assign(to: &$selectedPeerID)
|
||||||
|
|
||||||
|
peerIdentityStore.$encryptionStatuses
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.objectWillChange.send()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
peerIdentityStore.$verifiedFingerprints
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.objectWillChange.send()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
chatViewModel.$allPeers
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.objectWillChange.send()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func resolveDisplayName(for peerID: PeerID, statusPeerID: PeerID) -> String {
|
||||||
|
if let peer = chatViewModel.getPeer(byID: statusPeerID) {
|
||||||
|
return peer.displayName
|
||||||
|
}
|
||||||
|
if let name = chatViewModel.meshService.peerNickname(peerID: statusPeerID) {
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
if let data = peerID.noiseKey {
|
||||||
|
if let favorite = FavoritesPersistenceService.shared.getFavoriteStatus(for: data),
|
||||||
|
!favorite.peerNickname.isEmpty {
|
||||||
|
return favorite.peerNickname
|
||||||
|
}
|
||||||
|
let fingerprint = data.sha256Fingerprint()
|
||||||
|
if let social = chatViewModel.identityManager.getSocialIdentity(for: fingerprint) {
|
||||||
|
if let pet = social.localPetname, !pet.isEmpty {
|
||||||
|
return pet
|
||||||
|
}
|
||||||
|
if !social.claimedNickname.isEmpty {
|
||||||
|
return social.claimedNickname
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return String(localized: "common.unknown", comment: "Label for an unknown peer")
|
||||||
|
}
|
||||||
|
}
|
||||||
+44
-203
@@ -6,131 +6,57 @@
|
|||||||
// For more information, see <https://unlicense.org>
|
// For more information, see <https://unlicense.org>
|
||||||
//
|
//
|
||||||
|
|
||||||
import Nostr
|
|
||||||
import Tor
|
|
||||||
import SwiftUI
|
import SwiftUI
|
||||||
import BitFoundation
|
|
||||||
import UserNotifications
|
import UserNotifications
|
||||||
|
|
||||||
@main
|
@main
|
||||||
struct BitchatApp: App {
|
struct BitchatApp: App {
|
||||||
static let bundleID = Bundle.main.bundleIdentifier ?? "chat.bitchat"
|
static let bundleID = Bundle.main.bundleIdentifier ?? "chat.bitchat"
|
||||||
static let groupID = "group.\(bundleID)"
|
static let groupID = "group.\(bundleID)"
|
||||||
|
|
||||||
@StateObject private var chatViewModel: ChatViewModel
|
@StateObject private var runtime: AppRuntime
|
||||||
|
@AppStorage(AppTheme.storageKey) private var appThemeRawValue = AppTheme.matrix.rawValue
|
||||||
#if os(iOS)
|
#if os(iOS)
|
||||||
@Environment(\.scenePhase) var scenePhase
|
@Environment(\.scenePhase) var scenePhase
|
||||||
@UIApplicationDelegateAdaptor(AppDelegate.self) var appDelegate
|
@UIApplicationDelegateAdaptor(AppDelegate.self) var appDelegate
|
||||||
// Skip the very first .active-triggered Tor restart on cold launch
|
|
||||||
@State private var didHandleInitialActive: Bool = false
|
|
||||||
@State private var didEnterBackground: Bool = false
|
|
||||||
#elseif os(macOS)
|
#elseif os(macOS)
|
||||||
@NSApplicationDelegateAdaptor(MacAppDelegate.self) var appDelegate
|
@NSApplicationDelegateAdaptor(MacAppDelegate.self) var appDelegate
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
private let idBridge = NostrIdentityBridge(keychain: KeychainManager())
|
|
||||||
|
|
||||||
init() {
|
init() {
|
||||||
GeoRelayDirectory.setupShared(dependencies: .live())
|
_runtime = StateObject(wrappedValue: AppRuntime())
|
||||||
NostrRelayManager.setupShared(dependencies: .live())
|
|
||||||
let keychain = KeychainManager()
|
|
||||||
let idBridge = self.idBridge
|
|
||||||
_chatViewModel = StateObject(
|
|
||||||
wrappedValue: ChatViewModel(
|
|
||||||
keychain: keychain,
|
|
||||||
idBridge: idBridge,
|
|
||||||
identityManager: SecureIdentityStateManager(keychain)
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
UNUserNotificationCenter.current().delegate = NotificationDelegate.shared
|
UNUserNotificationCenter.current().delegate = NotificationDelegate.shared
|
||||||
// Warm up georelay directory and refresh if stale (once/day)
|
|
||||||
GeoRelayDirectory.shared.prefetchIfNeeded()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var body: some Scene {
|
var body: some Scene {
|
||||||
WindowGroup {
|
WindowGroup {
|
||||||
ContentView()
|
ContentView()
|
||||||
.environmentObject(chatViewModel)
|
.environment(\.appTheme, AppTheme(rawValue: appThemeRawValue) ?? .matrix)
|
||||||
|
.environmentObject(runtime.publicChatModel)
|
||||||
|
.environmentObject(runtime.privateInboxModel)
|
||||||
|
.environmentObject(runtime.privateConversationModel)
|
||||||
|
.environmentObject(runtime.verificationModel)
|
||||||
|
.environmentObject(runtime.conversationUIModel)
|
||||||
|
.environmentObject(runtime.locationChannelsModel)
|
||||||
|
.environmentObject(runtime.peerListModel)
|
||||||
|
.environmentObject(runtime.appChromeModel)
|
||||||
.onAppear {
|
.onAppear {
|
||||||
NotificationDelegate.shared.chatViewModel = chatViewModel
|
appDelegate.runtime = runtime
|
||||||
// Inject live Noise service into VerificationService to avoid creating new BLE instances
|
runtime.start()
|
||||||
VerificationService.shared.configure(with: chatViewModel.meshService.getNoiseService())
|
|
||||||
// Prewarm Nostr identity and QR to make first VERIFY sheet fast
|
|
||||||
let nickname = chatViewModel.nickname
|
|
||||||
DispatchQueue.global(qos: .utility).async {
|
|
||||||
let npub = try? idBridge.getCurrentNostrIdentity()?.npub
|
|
||||||
_ = VerificationService.shared.buildMyQRString(nickname: nickname, npub: npub)
|
|
||||||
}
|
|
||||||
|
|
||||||
appDelegate.chatViewModel = chatViewModel
|
|
||||||
|
|
||||||
// Initialize network activation policy; will start Tor/Nostr only when allowed
|
|
||||||
NetworkActivationService.shared.start()
|
|
||||||
|
|
||||||
// Start presence service (will wait for Tor readiness)
|
|
||||||
GeohashPresenceService.shared.start()
|
|
||||||
|
|
||||||
// Check for shared content
|
|
||||||
checkForSharedContent()
|
|
||||||
}
|
}
|
||||||
.onOpenURL { url in
|
.onOpenURL { url in
|
||||||
handleURL(url)
|
runtime.handleOpenURL(url)
|
||||||
}
|
}
|
||||||
#if os(iOS)
|
#if os(iOS)
|
||||||
.onChange(of: scenePhase) { newPhase in
|
.onChange(of: scenePhase) { newPhase in
|
||||||
switch newPhase {
|
runtime.handleScenePhaseChange(newPhase)
|
||||||
case .background:
|
|
||||||
// Keep BLE mesh running in background; BLEService adapts scanning automatically
|
|
||||||
// Always send Tor to dormant on background for a clean restart later.
|
|
||||||
TorManager.shared.setAppForeground(false)
|
|
||||||
TorManager.shared.goDormantOnBackground()
|
|
||||||
// Stop geohash sampling while backgrounded
|
|
||||||
Task { @MainActor in
|
|
||||||
chatViewModel.endGeohashSampling()
|
|
||||||
}
|
|
||||||
// Proactively disconnect Nostr to avoid spurious socket errors while Tor is down
|
|
||||||
NostrRelayManager.shared.disconnect()
|
|
||||||
didEnterBackground = true
|
|
||||||
case .active:
|
|
||||||
// Restart services when becoming active
|
|
||||||
chatViewModel.meshService.startServices()
|
|
||||||
TorManager.shared.setAppForeground(true)
|
|
||||||
// On initial cold launch, Tor was just started in onAppear.
|
|
||||||
// Skip the deterministic restart the first time we become active.
|
|
||||||
if didHandleInitialActive && didEnterBackground {
|
|
||||||
if TorManager.shared.isAutoStartAllowed() && !TorManager.shared.isReady {
|
|
||||||
TorManager.shared.ensureRunningOnForeground()
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
didHandleInitialActive = true
|
|
||||||
}
|
|
||||||
didEnterBackground = false
|
|
||||||
if TorManager.shared.isAutoStartAllowed() {
|
|
||||||
Task.detached {
|
|
||||||
let _ = await TorManager.shared.awaitReady(timeout: 60)
|
|
||||||
await MainActor.run {
|
|
||||||
// Rebuild proxied sessions to bind to the live Tor after readiness
|
|
||||||
TorURLSession.shared.rebuild()
|
|
||||||
// Reconnect Nostr via fresh sessions; will gate until Tor 100%
|
|
||||||
NostrRelayManager.shared.resetAllConnections()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
checkForSharedContent()
|
|
||||||
case .inactive:
|
|
||||||
break
|
|
||||||
@unknown default:
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
.onReceive(NotificationCenter.default.publisher(for: UIApplication.didBecomeActiveNotification)) { _ in
|
.onReceive(NotificationCenter.default.publisher(for: UIApplication.didBecomeActiveNotification)) { _ in
|
||||||
// Check for shared content when app becomes active
|
runtime.handleDidBecomeActiveNotification()
|
||||||
checkForSharedContent()
|
|
||||||
}
|
}
|
||||||
#elseif os(macOS)
|
#elseif os(macOS)
|
||||||
.onReceive(NotificationCenter.default.publisher(for: NSApplication.didBecomeActiveNotification)) { _ in
|
.onReceive(NotificationCenter.default.publisher(for: NSApplication.didBecomeActiveNotification)) { _ in
|
||||||
// App became active
|
runtime.handleMacDidBecomeActiveNotification()
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
@@ -139,66 +65,18 @@ struct BitchatApp: App {
|
|||||||
.windowResizability(.contentSize)
|
.windowResizability(.contentSize)
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
private func handleURL(_ url: URL) {
|
|
||||||
if url.scheme == "bitchat" && url.host == "share" {
|
|
||||||
// Handle shared content
|
|
||||||
checkForSharedContent()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private func checkForSharedContent() {
|
|
||||||
// Check app group for shared content from extension
|
|
||||||
guard let userDefaults = UserDefaults(suiteName: BitchatApp.groupID) else {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
guard let sharedContent = userDefaults.string(forKey: "sharedContent"),
|
|
||||||
let sharedDate = userDefaults.object(forKey: "sharedContentDate") as? Date else {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Only process if shared within configured window
|
|
||||||
if Date().timeIntervalSince(sharedDate) < TransportConfig.uiShareAcceptWindowSeconds {
|
|
||||||
let contentType = userDefaults.string(forKey: "sharedContentType") ?? "text"
|
|
||||||
|
|
||||||
// Clear the shared content
|
|
||||||
userDefaults.removeObject(forKey: "sharedContent")
|
|
||||||
userDefaults.removeObject(forKey: "sharedContentType")
|
|
||||||
userDefaults.removeObject(forKey: "sharedContentDate")
|
|
||||||
// No need to force synchronize here
|
|
||||||
|
|
||||||
// Send the shared content immediately on the main queue
|
|
||||||
DispatchQueue.main.async {
|
|
||||||
if contentType == "url" {
|
|
||||||
// Try to parse as JSON first
|
|
||||||
if let data = sharedContent.data(using: .utf8),
|
|
||||||
let urlData = try? JSONSerialization.jsonObject(with: data) as? [String: String],
|
|
||||||
let url = urlData["url"] {
|
|
||||||
// Send plain URL
|
|
||||||
self.chatViewModel.sendMessage(url)
|
|
||||||
} else {
|
|
||||||
// Fallback to simple URL
|
|
||||||
self.chatViewModel.sendMessage(sharedContent)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
self.chatViewModel.sendMessage(sharedContent)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#if os(iOS)
|
#if os(iOS)
|
||||||
final class AppDelegate: NSObject, UIApplicationDelegate {
|
final class AppDelegate: NSObject, UIApplicationDelegate {
|
||||||
weak var chatViewModel: ChatViewModel?
|
weak var runtime: AppRuntime?
|
||||||
|
|
||||||
func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey : Any]? = nil) -> Bool {
|
func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey : Any]? = nil) -> Bool {
|
||||||
return true
|
true
|
||||||
}
|
}
|
||||||
|
|
||||||
func applicationWillTerminate(_ application: UIApplication) {
|
func applicationWillTerminate(_ application: UIApplication) {
|
||||||
chatViewModel?.applicationWillTerminate()
|
runtime?.applicationWillTerminate()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
@@ -207,79 +85,42 @@ final class AppDelegate: NSObject, UIApplicationDelegate {
|
|||||||
import AppKit
|
import AppKit
|
||||||
|
|
||||||
final class MacAppDelegate: NSObject, NSApplicationDelegate {
|
final class MacAppDelegate: NSObject, NSApplicationDelegate {
|
||||||
weak var chatViewModel: ChatViewModel?
|
weak var runtime: AppRuntime?
|
||||||
|
|
||||||
func applicationWillTerminate(_ notification: Notification) {
|
func applicationWillTerminate(_ notification: Notification) {
|
||||||
chatViewModel?.applicationWillTerminate()
|
runtime?.applicationWillTerminate()
|
||||||
}
|
}
|
||||||
|
|
||||||
func applicationShouldTerminateAfterLastWindowClosed(_ sender: NSApplication) -> Bool {
|
func applicationShouldTerminateAfterLastWindowClosed(_ sender: NSApplication) -> Bool {
|
||||||
return true
|
true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
final class NotificationDelegate: NSObject, UNUserNotificationCenterDelegate {
|
final class NotificationDelegate: NSObject, UNUserNotificationCenterDelegate {
|
||||||
static let shared = NotificationDelegate()
|
static let shared = NotificationDelegate()
|
||||||
weak var chatViewModel: ChatViewModel?
|
weak var runtime: AppRuntime?
|
||||||
|
|
||||||
func userNotificationCenter(_ center: UNUserNotificationCenter, didReceive response: UNNotificationResponse, withCompletionHandler completionHandler: @escaping () -> Void) {
|
func userNotificationCenter(_ center: UNUserNotificationCenter, didReceive response: UNNotificationResponse, withCompletionHandler completionHandler: @escaping () -> Void) {
|
||||||
let identifier = response.notification.request.identifier
|
let identifier = response.notification.request.identifier
|
||||||
let userInfo = response.notification.request.content.userInfo
|
let userInfo = response.notification.request.content.userInfo
|
||||||
|
|
||||||
// Check if this is a private message notification
|
Task { @MainActor in
|
||||||
if identifier.hasPrefix("private-") {
|
self.runtime?.handleNotificationResponse(identifier: identifier, userInfo: userInfo)
|
||||||
// Get peer ID from userInfo
|
|
||||||
if let peerID = userInfo["peerID"] as? String {
|
|
||||||
DispatchQueue.main.async {
|
|
||||||
self.chatViewModel?.startPrivateChat(with: PeerID(str: peerID))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
// Handle deeplink (e.g., geohash activity)
|
|
||||||
if let deep = userInfo["deeplink"] as? String, let url = URL(string: deep) {
|
|
||||||
#if os(iOS)
|
|
||||||
DispatchQueue.main.async { UIApplication.shared.open(url) }
|
|
||||||
#else
|
|
||||||
DispatchQueue.main.async { NSWorkspace.shared.open(url) }
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
completionHandler()
|
completionHandler()
|
||||||
}
|
}
|
||||||
|
|
||||||
func userNotificationCenter(_ center: UNUserNotificationCenter, willPresent notification: UNNotification, withCompletionHandler completionHandler: @escaping (UNNotificationPresentationOptions) -> Void) {
|
func userNotificationCenter(_ center: UNUserNotificationCenter, willPresent notification: UNNotification, withCompletionHandler completionHandler: @escaping (UNNotificationPresentationOptions) -> Void) {
|
||||||
let identifier = notification.request.identifier
|
let identifier = notification.request.identifier
|
||||||
let userInfo = notification.request.content.userInfo
|
let userInfo = notification.request.content.userInfo
|
||||||
|
|
||||||
// Check if this is a private message notification
|
Task {
|
||||||
if identifier.hasPrefix("private-") {
|
let options = await self.runtime?.presentationOptions(
|
||||||
// Get peer ID from userInfo
|
forNotificationIdentifier: identifier,
|
||||||
if let peerID = userInfo["peerID"] as? String {
|
userInfo: userInfo
|
||||||
// Don't show notification if the private chat is already open
|
) ?? [.banner, .sound]
|
||||||
// Access main-actor-isolated property via Task
|
completionHandler(options)
|
||||||
Task { @MainActor in
|
|
||||||
if self.chatViewModel?.selectedPrivateChatPeer == PeerID(str: peerID) {
|
|
||||||
completionHandler([])
|
|
||||||
} else {
|
|
||||||
completionHandler([.banner, .sound])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
// Suppress geohash activity notification if we're already in that geohash channel
|
|
||||||
if identifier.hasPrefix("geo-activity-"),
|
|
||||||
let deep = userInfo["deeplink"] as? String,
|
|
||||||
let gh = deep.components(separatedBy: "/").last {
|
|
||||||
if case .location(let ch) = LocationChannelManager.shared.selectedChannel, ch.geohash == gh {
|
|
||||||
completionHandler([])
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Show notification in all other cases
|
|
||||||
completionHandler([.banner, .sound])
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -15,30 +15,39 @@ enum ImageUtilsError: Error {
|
|||||||
enum ImageUtils {
|
enum ImageUtils {
|
||||||
private static let compressionQuality: CGFloat = 0.82
|
private static let compressionQuality: CGFloat = 0.82
|
||||||
private static let targetImageBytes: Int = 45_000
|
private static let targetImageBytes: Int = 45_000
|
||||||
|
private static let maxSourceImageBytes: Int = 10 * 1024 * 1024
|
||||||
|
|
||||||
static func processImage(at url: URL, maxDimension: CGFloat = 448) throws -> URL {
|
static func processImage(at url: URL, maxDimension: CGFloat = 448, outputDirectory: URL? = nil) throws -> URL {
|
||||||
// Security H1: Check file size BEFORE reading into memory
|
try validateImageSource(at: url)
|
||||||
let attrs = try FileManager.default.attributesOfItem(atPath: url.path)
|
|
||||||
guard let fileSize = attrs[.size] as? Int else {
|
|
||||||
throw ImageUtilsError.invalidImage
|
|
||||||
}
|
|
||||||
// Allow up to 10MB source images (will be scaled down)
|
|
||||||
guard fileSize <= 10 * 1024 * 1024 else {
|
|
||||||
throw ImageUtilsError.invalidImage
|
|
||||||
}
|
|
||||||
|
|
||||||
let data = try Data(contentsOf: url)
|
let data = try Data(contentsOf: url)
|
||||||
#if os(iOS)
|
#if os(iOS)
|
||||||
guard let image = UIImage(data: data) else { throw ImageUtilsError.invalidImage }
|
guard let image = UIImage(data: data) else { throw ImageUtilsError.invalidImage }
|
||||||
return try processImage(image, maxDimension: maxDimension)
|
return try processImage(image, maxDimension: maxDimension, outputDirectory: outputDirectory)
|
||||||
#else
|
#else
|
||||||
guard let image = NSImage(data: data) else { throw ImageUtilsError.invalidImage }
|
guard let image = NSImage(data: data) else { throw ImageUtilsError.invalidImage }
|
||||||
return try processImage(image, maxDimension: maxDimension)
|
return try processImage(image, maxDimension: maxDimension, outputDirectory: outputDirectory)
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static func validateImageSource(at url: URL) throws {
|
||||||
|
// Security H1: Check file size BEFORE reading into memory.
|
||||||
|
let attrs = try FileManager.default.attributesOfItem(atPath: url.path)
|
||||||
|
guard let fileSize = attrs[.size] as? Int,
|
||||||
|
fileSize > 0,
|
||||||
|
fileSize <= maxSourceImageBytes else {
|
||||||
|
throw ImageUtilsError.invalidImage
|
||||||
|
}
|
||||||
|
|
||||||
|
let options = [kCGImageSourceShouldCache: false] as CFDictionary
|
||||||
|
guard let source = CGImageSourceCreateWithURL(url as CFURL, options),
|
||||||
|
CGImageSourceGetType(source) != nil else {
|
||||||
|
throw ImageUtilsError.invalidImage
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#if os(iOS)
|
#if os(iOS)
|
||||||
static func processImage(_ image: UIImage, maxDimension: CGFloat = 448) throws -> URL {
|
static func processImage(_ image: UIImage, maxDimension: CGFloat = 448, outputDirectory: URL? = nil) throws -> URL {
|
||||||
return try autoreleasepool {
|
return try autoreleasepool {
|
||||||
// Scale the image first
|
// Scale the image first
|
||||||
let scaled = scaledImage(image, maxDimension: maxDimension)
|
let scaled = scaledImage(image, maxDimension: maxDimension)
|
||||||
@@ -64,7 +73,7 @@ enum ImageUtils {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let outputURL = try makeOutputURL()
|
let outputURL = try makeOutputURL(outputDirectory: outputDirectory)
|
||||||
try jpegData.write(to: outputURL, options: .atomic)
|
try jpegData.write(to: outputURL, options: .atomic)
|
||||||
return outputURL
|
return outputURL
|
||||||
}
|
}
|
||||||
@@ -106,7 +115,7 @@ enum ImageUtils {
|
|||||||
return data as Data
|
return data as Data
|
||||||
}
|
}
|
||||||
#else
|
#else
|
||||||
static func processImage(_ image: NSImage, maxDimension: CGFloat = 448) throws -> URL {
|
static func processImage(_ image: NSImage, maxDimension: CGFloat = 448, outputDirectory: URL? = nil) throws -> URL {
|
||||||
return try autoreleasepool {
|
return try autoreleasepool {
|
||||||
let scaled = scaledImage(image, maxDimension: maxDimension)
|
let scaled = scaledImage(image, maxDimension: maxDimension)
|
||||||
guard let inputCG = scaled.cgImage(forProposedRect: nil, context: nil, hints: nil) else {
|
guard let inputCG = scaled.cgImage(forProposedRect: nil, context: nil, hints: nil) else {
|
||||||
@@ -142,7 +151,7 @@ enum ImageUtils {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
let outputURL = try makeOutputURL()
|
let outputURL = try makeOutputURL(outputDirectory: outputDirectory)
|
||||||
try jpegData.write(to: outputURL, options: .atomic)
|
try jpegData.write(to: outputURL, options: .atomic)
|
||||||
return outputURL
|
return outputURL
|
||||||
}
|
}
|
||||||
@@ -186,12 +195,17 @@ enum ImageUtils {
|
|||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
private static func makeOutputURL() throws -> URL {
|
private static func makeOutputURL(outputDirectory: URL? = nil) throws -> URL {
|
||||||
let formatter = DateFormatter()
|
let formatter = DateFormatter()
|
||||||
formatter.dateFormat = "yyyyMMdd_HHmmss"
|
formatter.dateFormat = "yyyyMMdd_HHmmss"
|
||||||
let fileName = "img_\(formatter.string(from: Date())).jpg"
|
let fileName = "img_\(formatter.string(from: Date()))_\(UUID().uuidString).jpg"
|
||||||
|
|
||||||
let directory = try applicationFilesDirectory().appendingPathComponent("images/outgoing", isDirectory: true)
|
let directory: URL
|
||||||
|
if let outputDirectory {
|
||||||
|
directory = outputDirectory
|
||||||
|
} else {
|
||||||
|
directory = try applicationFilesDirectory().appendingPathComponent("images/outgoing", isDirectory: true)
|
||||||
|
}
|
||||||
try FileManager.default.createDirectory(at: directory, withIntermediateDirectories: true, attributes: nil)
|
try FileManager.default.createDirectory(at: directory, withIntermediateDirectories: true, attributes: nil)
|
||||||
return directory.appendingPathComponent(fileName)
|
return directory.appendingPathComponent(fileName)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -202,8 +202,12 @@ final class SecureIdentityStateManager: SecureIdentityStateManagerProtocol {
|
|||||||
let decryptedData = try AES.GCM.open(sealedBox, using: encryptionKey)
|
let decryptedData = try AES.GCM.open(sealedBox, using: encryptionKey)
|
||||||
cache = try JSONDecoder().decode(IdentityCache.self, from: decryptedData)
|
cache = try JSONDecoder().decode(IdentityCache.self, from: decryptedData)
|
||||||
} catch {
|
} catch {
|
||||||
// Log error but continue with empty cache
|
cache = IdentityCache()
|
||||||
SecureLogger.error(error, context: "Failed to load identity cache", category: .security)
|
let deleted = keychain.deleteIdentityKey(forKey: cacheKey)
|
||||||
|
SecureLogger.warning(
|
||||||
|
"Discarded unreadable identity cache; starting fresh (deleted=\(deleted), error=\(error.localizedDescription))",
|
||||||
|
category: .security
|
||||||
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -540,6 +540,543 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"app_info.appearance.liquid_glass" : {
|
||||||
|
"extractionState" : "manual",
|
||||||
|
"localizations" : {
|
||||||
|
"ar" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "زجاج سائل"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"bn" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "লিকুইড গ্লাস"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"de" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"en" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"es" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fil" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fr" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"he" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "זכוכית נוזלית"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"hi" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "लिक्विड ग्लास"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"id" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"it" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ja" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "リキッドガラス"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ko" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "리퀴드 글래스"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ms" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ne" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "लिक्विड ग्लास"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"nl" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"pl" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"pt" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"pt-BR" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ru" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "жидкое стекло"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"sv" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ta" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "லிக்விட் கிளாஸ்"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"th" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "ลิควิดกลาส"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"tr" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"uk" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "рідке скло"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ur" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "لیکویڈ گلاس"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"vi" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "liquid glass"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"zh-Hans" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "液态玻璃"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"zh-Hant" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "液態玻璃"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"app_info.appearance.matrix" : {
|
||||||
|
"extractionState" : "manual",
|
||||||
|
"localizations" : {
|
||||||
|
"ar" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "ماتريكس"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"bn" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "ম্যাট্রিক্স"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"de" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"en" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"es" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fil" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fr" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"he" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "מטריקס"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"hi" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "मैट्रिक्स"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"id" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"it" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ja" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "マトリックス"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ko" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "매트릭스"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ms" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ne" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "म्याट्रिक्स"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"nl" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"pl" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"pt" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"pt-BR" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ru" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "матрица"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"sv" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ta" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "மேட்ரிக்ஸ்"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"th" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "เมทริกซ์"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"tr" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"uk" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "матриця"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ur" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "میٹرکس"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"vi" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "matrix"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"zh-Hans" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "矩阵"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"zh-Hant" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "矩陣"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"app_info.appearance.title" : {
|
||||||
|
"extractionState" : "manual",
|
||||||
|
"localizations" : {
|
||||||
|
"ar" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "المظهر"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"bn" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "চেহারা"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"de" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "ERSCHEINUNGSBILD"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"en" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "APPEARANCE"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"es" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "APARIENCIA"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fil" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "HITSURA"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"fr" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "APPARENCE"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"he" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "מראה"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"hi" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "दिखावट"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"id" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "TAMPILAN"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"it" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "ASPETTO"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ja" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "外観"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ko" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "화면 모드"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ms" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "PENAMPILAN"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ne" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "रूप"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"nl" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "WEERGAVE"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"pl" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "WYGLĄD"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"pt" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "APARÊNCIA"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"pt-BR" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "APARÊNCIA"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ru" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "ОФОРМЛЕНИЕ"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"sv" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "UTSEENDE"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ta" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "தோற்றம்"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"th" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "ลักษณะที่ปรากฏ"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"tr" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "GÖRÜNÜM"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"uk" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "ОФОРМЛЕННЯ"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"ur" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "ظاہری شکل"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"vi" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "GIAO DIỆN"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"zh-Hans" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "外观"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"zh-Hant" : {
|
||||||
|
"stringUnit" : {
|
||||||
|
"state" : "translated",
|
||||||
|
"value" : "外觀"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
"app_info.close" : {
|
"app_info.close" : {
|
||||||
"extractionState" : "manual",
|
"extractionState" : "manual",
|
||||||
"localizations" : {
|
"localizations" : {
|
||||||
@@ -24397,7 +24934,7 @@
|
|||||||
"en" : {
|
"en" : {
|
||||||
"stringUnit" : {
|
"stringUnit" : {
|
||||||
"state" : "translated",
|
"state" : "translated",
|
||||||
"value" : "chat with people near you using geohash channels. only a coarse geohash is shared, never exact GPS. your IP address is hidden by routing all traffic over tor."
|
"value" : "chat with people near you using geohash channels. the selected geohash is public and may reveal an approximate area; exact GPS is never shared. your IP address is hidden by routing all traffic over tor."
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"es" : {
|
"es" : {
|
||||||
|
|||||||
@@ -6,6 +6,7 @@
|
|||||||
// For more information, see <https://unlicense.org>
|
// For more information, see <https://unlicense.org>
|
||||||
//
|
//
|
||||||
|
|
||||||
|
import BitFoundation
|
||||||
import Foundation
|
import Foundation
|
||||||
|
|
||||||
extension BitchatMessage {
|
extension BitchatMessage {
|
||||||
|
|||||||
@@ -96,7 +96,7 @@ struct RequestSyncPacket {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
guard let pp = p, let mm = m, let dd = payload, pp >= 1, mm > 0 else { return nil }
|
guard let pp = p, let mm = m, let dd = payload, pp >= 1, pp <= GCSFilter.maxP, mm > 0 else { return nil }
|
||||||
return RequestSyncPacket(p: pp, m: mm, data: dd, types: types, sinceTimestamp: sinceTimestamp, fragmentIdFilter: fragmentIdFilter)
|
return RequestSyncPacket(p: pp, m: mm, data: dd, types: types, sinceTimestamp: sinceTimestamp, fragmentIdFilter: fragmentIdFilter)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+3
-3
@@ -533,7 +533,7 @@ final class NoiseSymmetricState {
|
|||||||
final class NoiseHandshakeState {
|
final class NoiseHandshakeState {
|
||||||
private let role: NoiseRole
|
private let role: NoiseRole
|
||||||
private let pattern: NoisePattern
|
private let pattern: NoisePattern
|
||||||
private let keychain: SecureMemoryCleaner
|
private let keychain: KeychainManagerProtocol
|
||||||
private var symmetricState: NoiseSymmetricState
|
private var symmetricState: NoiseSymmetricState
|
||||||
|
|
||||||
// Keys
|
// Keys
|
||||||
@@ -556,7 +556,7 @@ final class NoiseHandshakeState {
|
|||||||
init(
|
init(
|
||||||
role: NoiseRole,
|
role: NoiseRole,
|
||||||
pattern: NoisePattern,
|
pattern: NoisePattern,
|
||||||
keychain: SecureMemoryCleaner,
|
keychain: KeychainManagerProtocol,
|
||||||
localStaticKey: Curve25519.KeyAgreement.PrivateKey? = nil,
|
localStaticKey: Curve25519.KeyAgreement.PrivateKey? = nil,
|
||||||
remoteStaticKey: Curve25519.KeyAgreement.PublicKey? = nil,
|
remoteStaticKey: Curve25519.KeyAgreement.PublicKey? = nil,
|
||||||
prologue: Data = Data(),
|
prologue: Data = Data(),
|
||||||
@@ -924,7 +924,7 @@ extension NoisePattern {
|
|||||||
|
|
||||||
// MARK: - Errors
|
// MARK: - Errors
|
||||||
|
|
||||||
public enum NoiseError: Error {
|
enum NoiseError: Error {
|
||||||
case uninitializedCipher
|
case uninitializedCipher
|
||||||
case invalidCiphertext
|
case invalidCiphertext
|
||||||
case handshakeComplete
|
case handshakeComplete
|
||||||
+5
-7
@@ -10,7 +10,7 @@ import BitLogger
|
|||||||
import BitFoundation
|
import BitFoundation
|
||||||
import Foundation
|
import Foundation
|
||||||
|
|
||||||
public final class NoiseRateLimiter {
|
final class NoiseRateLimiter {
|
||||||
private var handshakeTimestamps: [PeerID: [Date]] = [:]
|
private var handshakeTimestamps: [PeerID: [Date]] = [:]
|
||||||
private var messageTimestamps: [PeerID: [Date]] = [:]
|
private var messageTimestamps: [PeerID: [Date]] = [:]
|
||||||
|
|
||||||
@@ -19,10 +19,8 @@ public final class NoiseRateLimiter {
|
|||||||
private var globalMessageTimestamps: [Date] = []
|
private var globalMessageTimestamps: [Date] = []
|
||||||
|
|
||||||
private let queue = DispatchQueue(label: "chat.bitchat.noise.ratelimit", attributes: .concurrent)
|
private let queue = DispatchQueue(label: "chat.bitchat.noise.ratelimit", attributes: .concurrent)
|
||||||
|
|
||||||
public init() {}
|
func allowHandshake(from peerID: PeerID) -> Bool {
|
||||||
|
|
||||||
public func allowHandshake(from peerID: PeerID) -> Bool {
|
|
||||||
return queue.sync(flags: .barrier) {
|
return queue.sync(flags: .barrier) {
|
||||||
let now = Date()
|
let now = Date()
|
||||||
let oneMinuteAgo = now.addingTimeInterval(-60)
|
let oneMinuteAgo = now.addingTimeInterval(-60)
|
||||||
@@ -51,7 +49,7 @@ public final class NoiseRateLimiter {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public func allowMessage(from peerID: PeerID) -> Bool {
|
func allowMessage(from peerID: PeerID) -> Bool {
|
||||||
return queue.sync(flags: .barrier) {
|
return queue.sync(flags: .barrier) {
|
||||||
let now = Date()
|
let now = Date()
|
||||||
let oneSecondAgo = now.addingTimeInterval(-1)
|
let oneSecondAgo = now.addingTimeInterval(-1)
|
||||||
@@ -87,7 +85,7 @@ public final class NoiseRateLimiter {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public func resetAll() {
|
func resetAll() {
|
||||||
queue.async(flags: .barrier) {
|
queue.async(flags: .barrier) {
|
||||||
self.handshakeTimestamps.removeAll()
|
self.handshakeTimestamps.removeAll()
|
||||||
self.messageTimestamps.removeAll()
|
self.messageTimestamps.removeAll()
|
||||||
+5
-5
@@ -8,7 +8,7 @@
|
|||||||
|
|
||||||
import Foundation
|
import Foundation
|
||||||
|
|
||||||
public enum NoiseSecurityConstants {
|
enum NoiseSecurityConstants {
|
||||||
// Maximum message size to prevent memory exhaustion
|
// Maximum message size to prevent memory exhaustion
|
||||||
static let maxMessageSize = 65535 // 64KB as per Noise spec
|
static let maxMessageSize = 65535 // 64KB as per Noise spec
|
||||||
|
|
||||||
@@ -28,10 +28,10 @@ public enum NoiseSecurityConstants {
|
|||||||
static let maxSessionsPerPeer = 3
|
static let maxSessionsPerPeer = 3
|
||||||
|
|
||||||
// Rate limiting
|
// Rate limiting
|
||||||
public static let maxHandshakesPerMinute = 10
|
static let maxHandshakesPerMinute = 10
|
||||||
public static let maxMessagesPerSecond = 100
|
static let maxMessagesPerSecond = 100
|
||||||
|
|
||||||
// Global rate limiting (across all peers)
|
// Global rate limiting (across all peers)
|
||||||
public static let maxGlobalHandshakesPerMinute = 30
|
static let maxGlobalHandshakesPerMinute = 30
|
||||||
public static let maxGlobalMessagesPerSecond = 500
|
static let maxGlobalMessagesPerSecond = 500
|
||||||
}
|
}
|
||||||
+1
-1
@@ -8,7 +8,7 @@
|
|||||||
|
|
||||||
import Foundation
|
import Foundation
|
||||||
|
|
||||||
public enum NoiseSecurityError: Error {
|
enum NoiseSecurityError: Error {
|
||||||
case sessionExpired
|
case sessionExpired
|
||||||
case sessionExhausted
|
case sessionExhausted
|
||||||
case messageTooLarge
|
case messageTooLarge
|
||||||
+3
-3
@@ -8,15 +8,15 @@
|
|||||||
|
|
||||||
import Foundation
|
import Foundation
|
||||||
|
|
||||||
public struct NoiseSecurityValidator {
|
struct NoiseSecurityValidator {
|
||||||
|
|
||||||
/// Validate message size
|
/// Validate message size
|
||||||
public static func validateMessageSize(_ data: Data) -> Bool {
|
static func validateMessageSize(_ data: Data) -> Bool {
|
||||||
return data.count <= NoiseSecurityConstants.maxMessageSize
|
return data.count <= NoiseSecurityConstants.maxMessageSize
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Validate handshake message size
|
/// Validate handshake message size
|
||||||
public static func validateHandshakeMessageSize(_ data: Data) -> Bool {
|
static func validateHandshakeMessageSize(_ data: Data) -> Bool {
|
||||||
return data.count <= NoiseSecurityConstants.maxHandshakeMessageSize
|
return data.count <= NoiseSecurityConstants.maxHandshakeMessageSize
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
+6
-6
@@ -11,10 +11,10 @@ import Foundation
|
|||||||
import CryptoKit
|
import CryptoKit
|
||||||
import BitFoundation
|
import BitFoundation
|
||||||
|
|
||||||
public class NoiseSession {
|
class NoiseSession {
|
||||||
let peerID: PeerID
|
let peerID: PeerID
|
||||||
let role: NoiseRole
|
let role: NoiseRole
|
||||||
private let keychain: SecureMemoryCleaner
|
private let keychain: KeychainManagerProtocol
|
||||||
private var state: NoiseSessionState = .uninitialized
|
private var state: NoiseSessionState = .uninitialized
|
||||||
private var handshakeState: NoiseHandshakeState?
|
private var handshakeState: NoiseHandshakeState?
|
||||||
private var sendCipher: NoiseCipherState?
|
private var sendCipher: NoiseCipherState?
|
||||||
@@ -34,7 +34,7 @@ public class NoiseSession {
|
|||||||
init(
|
init(
|
||||||
peerID: PeerID,
|
peerID: PeerID,
|
||||||
role: NoiseRole,
|
role: NoiseRole,
|
||||||
keychain: SecureMemoryCleaner,
|
keychain: KeychainManagerProtocol,
|
||||||
localStaticKey: Curve25519.KeyAgreement.PrivateKey,
|
localStaticKey: Curve25519.KeyAgreement.PrivateKey,
|
||||||
remoteStaticKey: Curve25519.KeyAgreement.PublicKey? = nil
|
remoteStaticKey: Curve25519.KeyAgreement.PublicKey? = nil
|
||||||
) {
|
) {
|
||||||
@@ -182,7 +182,7 @@ public class NoiseSession {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public func isEstablished() -> Bool {
|
func isEstablished() -> Bool {
|
||||||
return sessionQueue.sync {
|
return sessionQueue.sync {
|
||||||
if case .established = state {
|
if case .established = state {
|
||||||
return true
|
return true
|
||||||
@@ -217,8 +217,8 @@ public class NoiseSession {
|
|||||||
sentHandshakeMessages.removeAll()
|
sentHandshakeMessages.removeAll()
|
||||||
|
|
||||||
// Clear handshake hash
|
// Clear handshake hash
|
||||||
if handshakeHash != nil {
|
if var hash = handshakeHash {
|
||||||
keychain.secureClear(&handshakeHash!)
|
keychain.secureClear(&hash)
|
||||||
}
|
}
|
||||||
handshakeHash = nil
|
handshakeHash = nil
|
||||||
|
|
||||||
+1
-1
@@ -6,7 +6,7 @@
|
|||||||
// For more information, see <https://unlicense.org>
|
// For more information, see <https://unlicense.org>
|
||||||
//
|
//
|
||||||
|
|
||||||
public enum NoiseSessionError: Error, Equatable {
|
enum NoiseSessionError: Error, Equatable {
|
||||||
case invalidState
|
case invalidState
|
||||||
case notEstablished
|
case notEstablished
|
||||||
case sessionNotFound
|
case sessionNotFound
|
||||||
+15
-15
@@ -11,18 +11,18 @@ import CryptoKit
|
|||||||
import Foundation
|
import Foundation
|
||||||
import BitFoundation
|
import BitFoundation
|
||||||
|
|
||||||
public final class NoiseSessionManager {
|
final class NoiseSessionManager {
|
||||||
private var sessions: [PeerID: NoiseSession] = [:]
|
private var sessions: [PeerID: NoiseSession] = [:]
|
||||||
private let localStaticKey: Curve25519.KeyAgreement.PrivateKey
|
private let localStaticKey: Curve25519.KeyAgreement.PrivateKey
|
||||||
private let keychain: SecureMemoryCleaner
|
private let keychain: KeychainManagerProtocol
|
||||||
private let sessionFactory: (PeerID, NoiseRole) -> NoiseSession
|
private let sessionFactory: (PeerID, NoiseRole) -> NoiseSession
|
||||||
private let managerQueue = DispatchQueue(label: "chat.bitchat.noise.manager", attributes: .concurrent)
|
private let managerQueue = DispatchQueue(label: "chat.bitchat.noise.manager", attributes: .concurrent)
|
||||||
|
|
||||||
// Callbacks
|
// Callbacks
|
||||||
public var onSessionEstablished: ((PeerID, Curve25519.KeyAgreement.PublicKey) -> Void)?
|
var onSessionEstablished: ((PeerID, Curve25519.KeyAgreement.PublicKey) -> Void)?
|
||||||
var onSessionFailed: ((PeerID, Error) -> Void)?
|
var onSessionFailed: ((PeerID, Error) -> Void)?
|
||||||
|
|
||||||
public init(localStaticKey: Curve25519.KeyAgreement.PrivateKey, keychain: SecureMemoryCleaner) {
|
init(localStaticKey: Curve25519.KeyAgreement.PrivateKey, keychain: KeychainManagerProtocol) {
|
||||||
self.localStaticKey = localStaticKey
|
self.localStaticKey = localStaticKey
|
||||||
self.keychain = keychain
|
self.keychain = keychain
|
||||||
self.sessionFactory = { peerID, role in
|
self.sessionFactory = { peerID, role in
|
||||||
@@ -38,7 +38,7 @@ public final class NoiseSessionManager {
|
|||||||
#if DEBUG
|
#if DEBUG
|
||||||
init(
|
init(
|
||||||
localStaticKey: Curve25519.KeyAgreement.PrivateKey,
|
localStaticKey: Curve25519.KeyAgreement.PrivateKey,
|
||||||
keychain: SecureMemoryCleaner,
|
keychain: KeychainManagerProtocol,
|
||||||
sessionFactory: @escaping (PeerID, NoiseRole) -> NoiseSession
|
sessionFactory: @escaping (PeerID, NoiseRole) -> NoiseSession
|
||||||
) {
|
) {
|
||||||
self.localStaticKey = localStaticKey
|
self.localStaticKey = localStaticKey
|
||||||
@@ -49,13 +49,13 @@ public final class NoiseSessionManager {
|
|||||||
|
|
||||||
// MARK: - Session Management
|
// MARK: - Session Management
|
||||||
|
|
||||||
public func getSession(for peerID: PeerID) -> NoiseSession? {
|
func getSession(for peerID: PeerID) -> NoiseSession? {
|
||||||
return managerQueue.sync {
|
return managerQueue.sync {
|
||||||
return sessions[peerID]
|
return sessions[peerID]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public func removeSession(for peerID: PeerID) {
|
func removeSession(for peerID: PeerID) {
|
||||||
managerQueue.sync(flags: .barrier) {
|
managerQueue.sync(flags: .barrier) {
|
||||||
if let session = sessions.removeValue(forKey: peerID) {
|
if let session = sessions.removeValue(forKey: peerID) {
|
||||||
session.reset() // Clear sensitive data before removing
|
session.reset() // Clear sensitive data before removing
|
||||||
@@ -63,7 +63,7 @@ public final class NoiseSessionManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public func removeAllSessions() {
|
func removeAllSessions() {
|
||||||
managerQueue.sync(flags: .barrier) {
|
managerQueue.sync(flags: .barrier) {
|
||||||
for (_, session) in sessions {
|
for (_, session) in sessions {
|
||||||
session.reset()
|
session.reset()
|
||||||
@@ -74,7 +74,7 @@ public final class NoiseSessionManager {
|
|||||||
|
|
||||||
// MARK: - Handshake Helpers
|
// MARK: - Handshake Helpers
|
||||||
|
|
||||||
public func initiateHandshake(with peerID: PeerID) throws -> Data {
|
func initiateHandshake(with peerID: PeerID) throws -> Data {
|
||||||
return try managerQueue.sync(flags: .barrier) {
|
return try managerQueue.sync(flags: .barrier) {
|
||||||
// Check if we already have an established session
|
// Check if we already have an established session
|
||||||
if let existingSession = sessions[peerID], existingSession.isEstablished() {
|
if let existingSession = sessions[peerID], existingSession.isEstablished() {
|
||||||
@@ -103,7 +103,7 @@ public final class NoiseSessionManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public func handleIncomingHandshake(from peerID: PeerID, message: Data) throws -> Data? {
|
func handleIncomingHandshake(from peerID: PeerID, message: Data) throws -> Data? {
|
||||||
// Process everything within the synchronized block to prevent race conditions
|
// Process everything within the synchronized block to prevent race conditions
|
||||||
return try managerQueue.sync(flags: .barrier) {
|
return try managerQueue.sync(flags: .barrier) {
|
||||||
var shouldCreateNew = false
|
var shouldCreateNew = false
|
||||||
@@ -173,7 +173,7 @@ public final class NoiseSessionManager {
|
|||||||
|
|
||||||
// MARK: - Encryption/Decryption
|
// MARK: - Encryption/Decryption
|
||||||
|
|
||||||
public func encrypt(_ plaintext: Data, for peerID: PeerID) throws -> Data {
|
func encrypt(_ plaintext: Data, for peerID: PeerID) throws -> Data {
|
||||||
guard let session = getSession(for: peerID) else {
|
guard let session = getSession(for: peerID) else {
|
||||||
throw NoiseSessionError.sessionNotFound
|
throw NoiseSessionError.sessionNotFound
|
||||||
}
|
}
|
||||||
@@ -181,7 +181,7 @@ public final class NoiseSessionManager {
|
|||||||
return try session.encrypt(plaintext)
|
return try session.encrypt(plaintext)
|
||||||
}
|
}
|
||||||
|
|
||||||
public func decrypt(_ ciphertext: Data, from peerID: PeerID) throws -> Data {
|
func decrypt(_ ciphertext: Data, from peerID: PeerID) throws -> Data {
|
||||||
guard let session = getSession(for: peerID) else {
|
guard let session = getSession(for: peerID) else {
|
||||||
throw NoiseSessionError.sessionNotFound
|
throw NoiseSessionError.sessionNotFound
|
||||||
}
|
}
|
||||||
@@ -191,13 +191,13 @@ public final class NoiseSessionManager {
|
|||||||
|
|
||||||
// MARK: - Key Management
|
// MARK: - Key Management
|
||||||
|
|
||||||
public func getRemoteStaticKey(for peerID: PeerID) -> Curve25519.KeyAgreement.PublicKey? {
|
func getRemoteStaticKey(for peerID: PeerID) -> Curve25519.KeyAgreement.PublicKey? {
|
||||||
return getSession(for: peerID)?.getRemoteStaticPublicKey()
|
return getSession(for: peerID)?.getRemoteStaticPublicKey()
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Session Rekeying
|
// MARK: - Session Rekeying
|
||||||
|
|
||||||
public func getSessionsNeedingRekey() -> [(peerID: PeerID, needsRekey: Bool)] {
|
func getSessionsNeedingRekey() -> [(peerID: PeerID, needsRekey: Bool)] {
|
||||||
return managerQueue.sync {
|
return managerQueue.sync {
|
||||||
var needingRekey: [(peerID: PeerID, needsRekey: Bool)] = []
|
var needingRekey: [(peerID: PeerID, needsRekey: Bool)] = []
|
||||||
|
|
||||||
@@ -213,7 +213,7 @@ public final class NoiseSessionManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public func initiateRekey(for peerID: PeerID) throws {
|
func initiateRekey(for peerID: PeerID) throws {
|
||||||
// Remove old session
|
// Remove old session
|
||||||
removeSession(for: peerID)
|
removeSession(for: peerID)
|
||||||
|
|
||||||
@@ -0,0 +1,22 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum Base64URLCoding {
|
||||||
|
static func encode(_ data: Data) -> String {
|
||||||
|
data.base64EncodedString()
|
||||||
|
.replacingOccurrences(of: "+", with: "-")
|
||||||
|
.replacingOccurrences(of: "/", with: "_")
|
||||||
|
.replacingOccurrences(of: "=", with: "")
|
||||||
|
}
|
||||||
|
|
||||||
|
static func decode(_ string: String) -> Data? {
|
||||||
|
var base64 = string
|
||||||
|
let padding = (4 - (base64.count % 4)) % 4
|
||||||
|
if padding > 0 {
|
||||||
|
base64 += String(repeating: "=", count: padding)
|
||||||
|
}
|
||||||
|
base64 = base64
|
||||||
|
.replacingOccurrences(of: "-", with: "+")
|
||||||
|
.replacingOccurrences(of: "_", with: "/")
|
||||||
|
return Data(base64Encoded: base64)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,37 +1,38 @@
|
|||||||
import Foundation
|
import Foundation
|
||||||
|
|
||||||
/// Bech32 encoding for Nostr (minimal implementation)
|
/// Bech32 encoding for Nostr (minimal implementation)
|
||||||
public enum Bech32 {
|
enum Bech32 {
|
||||||
private static let charset = "qpzry9x8gf2tvdw0s3jn54khce6mua7l"
|
private static let charset = "qpzry9x8gf2tvdw0s3jn54khce6mua7l"
|
||||||
private static let generator = [0x3b6a57b2, 0x26508e6d, 0x1ea119fa, 0x3d4233dd, 0x2a1462b3]
|
private static let generator = [0x3b6a57b2, 0x26508e6d, 0x1ea119fa, 0x3d4233dd, 0x2a1462b3]
|
||||||
|
|
||||||
public static func encode(hrp: String, data: Data) throws -> String {
|
static func encode(hrp: String, data: Data) throws -> String {
|
||||||
let values = convertBits(from: 8, to: 5, pad: true, data: Array(data))
|
let values = convertBits(from: 8, to: 5, pad: true, data: Array(data))
|
||||||
let checksum = createChecksum(hrp: hrp, values: values)
|
let checksum = createChecksum(hrp: hrp, values: values)
|
||||||
let combined = values + checksum
|
let combined = values + checksum
|
||||||
|
|
||||||
return hrp + "1" + combined.map {
|
return hrp + "1" + combined.map {
|
||||||
let index = charset.index(charset.startIndex, offsetBy: Int($0))
|
let index = charset.index(charset.startIndex, offsetBy: Int($0))
|
||||||
return String(charset[index])
|
return String(charset[index])
|
||||||
}.joined()
|
}.joined()
|
||||||
}
|
}
|
||||||
|
|
||||||
public static func decode(_ bech32String: String) throws -> (hrp: String, data: Data) {
|
static func decode(_ bech32String: String) throws -> (hrp: String, data: Data) {
|
||||||
|
// Find the last occurrence of '1'
|
||||||
guard let separatorIndex = bech32String.lastIndex(of: "1") else {
|
guard let separatorIndex = bech32String.lastIndex(of: "1") else {
|
||||||
throw Bech32Error.invalidFormat
|
throw Bech32Error.invalidFormat
|
||||||
}
|
}
|
||||||
|
|
||||||
let hrp = String(bech32String[..<separatorIndex])
|
let hrp = String(bech32String[..<separatorIndex])
|
||||||
|
|
||||||
// Validate HRP contains only ASCII characters
|
// Validate HRP contains only ASCII characters
|
||||||
for char in hrp {
|
for char in hrp {
|
||||||
guard char.asciiValue != nil else {
|
guard char.asciiValue != nil else {
|
||||||
throw Bech32Error.invalidCharacter
|
throw Bech32Error.invalidCharacter
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let dataString = String(bech32String[bech32String.index(after: separatorIndex)...])
|
let dataString = String(bech32String[bech32String.index(after: separatorIndex)...])
|
||||||
|
|
||||||
// Convert characters to values
|
// Convert characters to values
|
||||||
var values = [UInt8]()
|
var values = [UInt8]()
|
||||||
for char in dataString {
|
for char in dataString {
|
||||||
@@ -40,84 +41,84 @@ public enum Bech32 {
|
|||||||
}
|
}
|
||||||
values.append(UInt8(charset.distance(from: charset.startIndex, to: index)))
|
values.append(UInt8(charset.distance(from: charset.startIndex, to: index)))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Verify checksum
|
// Verify checksum
|
||||||
guard values.count >= 6 else {
|
guard values.count >= 6 else {
|
||||||
throw Bech32Error.invalidChecksum
|
throw Bech32Error.invalidChecksum
|
||||||
}
|
}
|
||||||
|
|
||||||
let payloadValues = Array(values.dropLast(6))
|
let payloadValues = Array(values.dropLast(6))
|
||||||
let checksum = Array(values.suffix(6))
|
let checksum = Array(values.suffix(6))
|
||||||
let expectedChecksum = createChecksum(hrp: hrp, values: payloadValues)
|
let expectedChecksum = createChecksum(hrp: hrp, values: payloadValues)
|
||||||
|
|
||||||
guard checksum == expectedChecksum else {
|
guard checksum == expectedChecksum else {
|
||||||
throw Bech32Error.invalidChecksum
|
throw Bech32Error.invalidChecksum
|
||||||
}
|
}
|
||||||
|
|
||||||
// Convert back to bytes
|
// Convert back to bytes
|
||||||
let bytes = convertBits(from: 5, to: 8, pad: false, data: payloadValues)
|
let bytes = convertBits(from: 5, to: 8, pad: false, data: payloadValues)
|
||||||
return (hrp: hrp, data: Data(bytes))
|
return (hrp: hrp, data: Data(bytes))
|
||||||
}
|
}
|
||||||
|
|
||||||
enum Bech32Error: Error {
|
enum Bech32Error: Error {
|
||||||
case invalidFormat
|
case invalidFormat
|
||||||
case invalidCharacter
|
case invalidCharacter
|
||||||
case invalidChecksum
|
case invalidChecksum
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func convertBits(from: Int, to: Int, pad: Bool, data: [UInt8]) -> [UInt8] {
|
private static func convertBits(from: Int, to: Int, pad: Bool, data: [UInt8]) -> [UInt8] {
|
||||||
var acc = 0
|
var acc = 0
|
||||||
var bits = 0
|
var bits = 0
|
||||||
var result = [UInt8]()
|
var result = [UInt8]()
|
||||||
let maxv = (1 << to) - 1
|
let maxv = (1 << to) - 1
|
||||||
|
|
||||||
for value in data {
|
for value in data {
|
||||||
acc = (acc << from) | Int(value)
|
acc = (acc << from) | Int(value)
|
||||||
bits += from
|
bits += from
|
||||||
|
|
||||||
while bits >= to {
|
while bits >= to {
|
||||||
bits -= to
|
bits -= to
|
||||||
result.append(UInt8((acc >> bits) & maxv))
|
result.append(UInt8((acc >> bits) & maxv))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if pad && bits > 0 {
|
if pad && bits > 0 {
|
||||||
result.append(UInt8((acc << (to - bits)) & maxv))
|
result.append(UInt8((acc << (to - bits)) & maxv))
|
||||||
}
|
}
|
||||||
|
|
||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func createChecksum(hrp: String, values: [UInt8]) -> [UInt8] {
|
private static func createChecksum(hrp: String, values: [UInt8]) -> [UInt8] {
|
||||||
let checksumValues = hrpExpand(hrp) + values + [0, 0, 0, 0, 0, 0]
|
let checksumValues = hrpExpand(hrp) + values + [0, 0, 0, 0, 0, 0]
|
||||||
let polymod = polymod(checksumValues) ^ 1
|
let polymod = polymod(checksumValues) ^ 1
|
||||||
var checksum = [UInt8]()
|
var checksum = [UInt8]()
|
||||||
|
|
||||||
for i in 0..<6 {
|
for i in 0..<6 {
|
||||||
checksum.append(UInt8((polymod >> (5 * (5 - i))) & 31))
|
checksum.append(UInt8((polymod >> (5 * (5 - i))) & 31))
|
||||||
}
|
}
|
||||||
|
|
||||||
return checksum
|
return checksum
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func hrpExpand(_ hrp: String) -> [UInt8] {
|
private static func hrpExpand(_ hrp: String) -> [UInt8] {
|
||||||
var result = [UInt8]()
|
var result = [UInt8]()
|
||||||
for c in hrp {
|
for c in hrp {
|
||||||
guard let asciiValue = c.asciiValue else {
|
guard let asciiValue = c.asciiValue else {
|
||||||
return []
|
return [] // Return empty array for invalid input
|
||||||
}
|
}
|
||||||
result.append(UInt8(asciiValue >> 5))
|
result.append(UInt8(asciiValue >> 5))
|
||||||
}
|
}
|
||||||
result.append(0)
|
result.append(0)
|
||||||
for c in hrp {
|
for c in hrp {
|
||||||
guard let asciiValue = c.asciiValue else {
|
guard let asciiValue = c.asciiValue else {
|
||||||
return []
|
return [] // Return empty array for invalid input
|
||||||
}
|
}
|
||||||
result.append(UInt8(asciiValue & 31))
|
result.append(UInt8(asciiValue & 31))
|
||||||
}
|
}
|
||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func polymod(_ values: [UInt8]) -> Int {
|
private static func polymod(_ values: [UInt8]) -> Int {
|
||||||
var chk = 1
|
var chk = 1
|
||||||
for value in values {
|
for value in values {
|
||||||
+145
-160
@@ -1,72 +1,106 @@
|
|||||||
import BitLogger
|
import BitLogger
|
||||||
import Foundation
|
import Foundation
|
||||||
|
import Tor
|
||||||
|
#if os(iOS)
|
||||||
|
import UIKit
|
||||||
|
#elseif os(macOS)
|
||||||
|
import AppKit
|
||||||
|
#endif
|
||||||
|
|
||||||
public struct GeoRelayDirectoryDependencies {
|
extension Notification.Name {
|
||||||
public var userDefaults: UserDefaults
|
/// Posted after the geo relay directory successfully refreshes its entries.
|
||||||
public var notificationCenter: NotificationCenter
|
static let geoRelayDirectoryDidRefresh = Notification.Name("bitchat.geoRelayDirectoryDidRefresh")
|
||||||
public var now: () -> Date
|
}
|
||||||
public var remoteURL: URL
|
|
||||||
public var fetchInterval: TimeInterval
|
|
||||||
public var refreshCheckInterval: TimeInterval
|
|
||||||
public var retryInitialSeconds: TimeInterval
|
|
||||||
public var retryMaxSeconds: TimeInterval
|
|
||||||
public var awaitTorReady: @Sendable () async -> Bool
|
|
||||||
public var makeFetchData: @MainActor @Sendable () -> (@Sendable (URLRequest) async throws -> Data)
|
|
||||||
public var readData: (URL) -> Data?
|
|
||||||
public var writeData: (Data, URL) throws -> Void
|
|
||||||
public var cacheURL: () -> URL?
|
|
||||||
public var bundledCSVURLs: () -> [URL]
|
|
||||||
public var currentDirectoryPath: () -> String?
|
|
||||||
public var retrySleep: (TimeInterval) async -> Void
|
|
||||||
public var torReadyNotificationName: Notification.Name?
|
|
||||||
public var activeNotificationName: Notification.Name?
|
|
||||||
public var autoStart: Bool
|
|
||||||
|
|
||||||
public init(
|
/// Directory of online Nostr relays with approximate GPS locations, used for geohash routing.
|
||||||
userDefaults: UserDefaults,
|
struct GeoRelayDirectoryDependencies {
|
||||||
notificationCenter: NotificationCenter,
|
var userDefaults: UserDefaults
|
||||||
now: @escaping () -> Date,
|
var notificationCenter: NotificationCenter
|
||||||
remoteURL: URL,
|
var now: () -> Date
|
||||||
fetchInterval: TimeInterval,
|
var remoteURL: URL
|
||||||
refreshCheckInterval: TimeInterval,
|
var fetchInterval: TimeInterval
|
||||||
retryInitialSeconds: TimeInterval,
|
var refreshCheckInterval: TimeInterval
|
||||||
retryMaxSeconds: TimeInterval,
|
var retryInitialSeconds: TimeInterval
|
||||||
awaitTorReady: @Sendable @escaping () async -> Bool,
|
var retryMaxSeconds: TimeInterval
|
||||||
makeFetchData: @MainActor @Sendable @escaping () -> (@Sendable (URLRequest) async throws -> Data),
|
var awaitTorReady: @Sendable () async -> Bool
|
||||||
readData: @escaping (URL) -> Data?,
|
var makeFetchData: @MainActor @Sendable () -> (@Sendable (URLRequest) async throws -> Data)
|
||||||
writeData: @escaping (Data, URL) throws -> Void,
|
var readData: (URL) -> Data?
|
||||||
cacheURL: @escaping () -> URL?,
|
var writeData: (Data, URL) throws -> Void
|
||||||
bundledCSVURLs: @escaping () -> [URL],
|
var cacheURL: () -> URL?
|
||||||
currentDirectoryPath: @escaping () -> String?,
|
var bundledCSVURLs: () -> [URL]
|
||||||
retrySleep: @escaping (TimeInterval) async -> Void,
|
var currentDirectoryPath: () -> String?
|
||||||
torReadyNotificationName: Notification.Name?,
|
var retrySleep: (TimeInterval) async -> Void
|
||||||
activeNotificationName: Notification.Name?,
|
var activeNotificationName: Notification.Name?
|
||||||
autoStart: Bool
|
var autoStart: Bool
|
||||||
) {
|
}
|
||||||
self.userDefaults = userDefaults
|
|
||||||
self.notificationCenter = notificationCenter
|
private extension GeoRelayDirectoryDependencies {
|
||||||
self.now = now
|
@MainActor
|
||||||
self.remoteURL = remoteURL
|
static func live() -> Self {
|
||||||
self.fetchInterval = fetchInterval
|
#if os(iOS)
|
||||||
self.refreshCheckInterval = refreshCheckInterval
|
let activeNotificationName: Notification.Name? = UIApplication.didBecomeActiveNotification
|
||||||
self.retryInitialSeconds = retryInitialSeconds
|
#elseif os(macOS)
|
||||||
self.retryMaxSeconds = retryMaxSeconds
|
let activeNotificationName: Notification.Name? = NSApplication.didBecomeActiveNotification
|
||||||
self.awaitTorReady = awaitTorReady
|
#else
|
||||||
self.makeFetchData = makeFetchData
|
let activeNotificationName: Notification.Name? = nil
|
||||||
self.readData = readData
|
#endif
|
||||||
self.writeData = writeData
|
|
||||||
self.cacheURL = cacheURL
|
return Self(
|
||||||
self.bundledCSVURLs = bundledCSVURLs
|
userDefaults: .standard,
|
||||||
self.currentDirectoryPath = currentDirectoryPath
|
notificationCenter: .default,
|
||||||
self.retrySleep = retrySleep
|
now: Date.init,
|
||||||
self.torReadyNotificationName = torReadyNotificationName
|
remoteURL: URL(string: "https://raw.githubusercontent.com/permissionlesstech/georelays/refs/heads/main/nostr_relays.csv")!,
|
||||||
self.activeNotificationName = activeNotificationName
|
fetchInterval: TransportConfig.geoRelayFetchIntervalSeconds,
|
||||||
self.autoStart = autoStart
|
refreshCheckInterval: TransportConfig.geoRelayRefreshCheckIntervalSeconds,
|
||||||
|
retryInitialSeconds: TransportConfig.geoRelayRetryInitialSeconds,
|
||||||
|
retryMaxSeconds: TransportConfig.geoRelayRetryMaxSeconds,
|
||||||
|
awaitTorReady: { await TorManager.shared.awaitReady() },
|
||||||
|
makeFetchData: {
|
||||||
|
let session = TorURLSession.shared.session
|
||||||
|
return { request in
|
||||||
|
let (data, _) = try await session.data(for: request)
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
},
|
||||||
|
readData: { try? Data(contentsOf: $0) },
|
||||||
|
writeData: { data, url in
|
||||||
|
try data.write(to: url, options: .atomic)
|
||||||
|
},
|
||||||
|
cacheURL: {
|
||||||
|
do {
|
||||||
|
let base = try FileManager.default.url(
|
||||||
|
for: .applicationSupportDirectory,
|
||||||
|
in: .userDomainMask,
|
||||||
|
appropriateFor: nil,
|
||||||
|
create: true
|
||||||
|
)
|
||||||
|
let dir = base.appendingPathComponent("bitchat", isDirectory: true)
|
||||||
|
try? FileManager.default.createDirectory(at: dir, withIntermediateDirectories: true)
|
||||||
|
return dir.appendingPathComponent("georelays_cache.csv")
|
||||||
|
} catch {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
},
|
||||||
|
bundledCSVURLs: {
|
||||||
|
[
|
||||||
|
Bundle.main.url(forResource: "nostr_relays", withExtension: "csv"),
|
||||||
|
Bundle.main.url(forResource: "online_relays_gps", withExtension: "csv"),
|
||||||
|
Bundle.main.url(forResource: "online_relays_gps", withExtension: "csv", subdirectory: "relays")
|
||||||
|
].compactMap { $0 }
|
||||||
|
},
|
||||||
|
currentDirectoryPath: { FileManager.default.currentDirectoryPath },
|
||||||
|
retrySleep: { delay in
|
||||||
|
let nanoseconds = UInt64(delay * 1_000_000_000)
|
||||||
|
try? await Task.sleep(nanoseconds: nanoseconds)
|
||||||
|
},
|
||||||
|
activeNotificationName: activeNotificationName,
|
||||||
|
autoStart: true
|
||||||
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@MainActor
|
@MainActor
|
||||||
public final class GeoRelayDirectory {
|
final class GeoRelayDirectory {
|
||||||
private final class CleanupState {
|
private final class CleanupState {
|
||||||
let notificationCenter: NotificationCenter
|
let notificationCenter: NotificationCenter
|
||||||
var observers: [NSObjectProtocol] = []
|
var observers: [NSObjectProtocol] = []
|
||||||
@@ -84,16 +118,10 @@ public final class GeoRelayDirectory {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public struct Entry: Hashable, Sendable {
|
struct Entry: Hashable, Sendable {
|
||||||
public let host: String
|
let host: String
|
||||||
public let lat: Double
|
let lat: Double
|
||||||
public let lon: Double
|
let lon: Double
|
||||||
|
|
||||||
public init(host: String, lat: Double, lon: Double) {
|
|
||||||
self.host = host
|
|
||||||
self.lat = lat
|
|
||||||
self.lon = lon
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private enum DetachedFetchOutcome: Sendable {
|
private enum DetachedFetchOutcome: Sendable {
|
||||||
@@ -103,13 +131,9 @@ public final class GeoRelayDirectory {
|
|||||||
case network(String)
|
case network(String)
|
||||||
}
|
}
|
||||||
|
|
||||||
nonisolated(unsafe) public static var shared: GeoRelayDirectory!
|
static let shared = GeoRelayDirectory()
|
||||||
|
|
||||||
public static func setupShared(dependencies: GeoRelayDirectoryDependencies) {
|
private(set) var entries: [Entry] = []
|
||||||
shared = GeoRelayDirectory(dependencies: dependencies)
|
|
||||||
}
|
|
||||||
|
|
||||||
private(set) public var entries: [Entry] = []
|
|
||||||
private let lastFetchKey = "georelay.lastFetchAt"
|
private let lastFetchKey = "georelay.lastFetchAt"
|
||||||
private let dependencies: GeoRelayDirectoryDependencies
|
private let dependencies: GeoRelayDirectoryDependencies
|
||||||
private let cleanupState: CleanupState
|
private let cleanupState: CleanupState
|
||||||
@@ -117,7 +141,18 @@ public final class GeoRelayDirectory {
|
|||||||
private var retryAttempt: Int = 0
|
private var retryAttempt: Int = 0
|
||||||
private var isFetching: Bool = false
|
private var isFetching: Bool = false
|
||||||
|
|
||||||
public init(dependencies: GeoRelayDirectoryDependencies) {
|
private init() {
|
||||||
|
self.dependencies = .live()
|
||||||
|
self.cleanupState = CleanupState(notificationCenter: dependencies.notificationCenter)
|
||||||
|
entries = loadLocalEntries()
|
||||||
|
if dependencies.autoStart {
|
||||||
|
registerObservers()
|
||||||
|
startRefreshTimer()
|
||||||
|
prefetchIfNeeded()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
internal init(dependencies: GeoRelayDirectoryDependencies) {
|
||||||
self.dependencies = dependencies
|
self.dependencies = dependencies
|
||||||
self.cleanupState = CleanupState(notificationCenter: dependencies.notificationCenter)
|
self.cleanupState = CleanupState(notificationCenter: dependencies.notificationCenter)
|
||||||
entries = loadLocalEntries()
|
entries = loadLocalEntries()
|
||||||
@@ -129,43 +164,26 @@ public final class GeoRelayDirectory {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Returns up to `count` relay URLs (wss://) closest to the geohash center.
|
/// Returns up to `count` relay URLs (wss://) closest to the geohash center.
|
||||||
public func closestRelays(toGeohash geohash: String, count: Int = 5) -> [String] {
|
func closestRelays(toGeohash geohash: String, count: Int = 5) -> [String] {
|
||||||
let center = decodeGeohashCenter(geohash)
|
let center = Geohash.decodeCenter(geohash)
|
||||||
return closestRelays(toLat: center.lat, lon: center.lon, count: count)
|
return closestRelays(toLat: center.lat, lon: center.lon, count: count)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns up to `count` relay URLs (wss://) closest to the given coordinate.
|
/// Returns up to `count` relay URLs (wss://) closest to the given coordinate.
|
||||||
public func closestRelays(toLat lat: Double, lon: Double, count: Int = 5) -> [String] {
|
/// Ties break by host so every device with the same directory picks the
|
||||||
|
/// same relay set — publishers and subscribers must agree on relays.
|
||||||
|
func closestRelays(toLat lat: Double, lon: Double, count: Int = 5) -> [String] {
|
||||||
guard !entries.isEmpty, count > 0 else { return [] }
|
guard !entries.isEmpty, count > 0 else { return [] }
|
||||||
|
|
||||||
if entries.count <= count {
|
return entries
|
||||||
return entries
|
.map { (entry: $0, distance: haversineKm(lat, lon, $0.lat, $0.lon)) }
|
||||||
.sorted { a, b in
|
.sorted { ($0.distance, $0.entry.host) < ($1.distance, $1.entry.host) }
|
||||||
haversineKm(lat, lon, a.lat, a.lon) < haversineKm(lat, lon, b.lat, b.lon)
|
.prefix(count)
|
||||||
}
|
.map { "wss://\($0.entry.host)" }
|
||||||
.map { "wss://\($0.host)" }
|
|
||||||
}
|
|
||||||
|
|
||||||
var best: [(entry: Entry, distance: Double)] = []
|
|
||||||
best.reserveCapacity(count)
|
|
||||||
|
|
||||||
for entry in entries {
|
|
||||||
let distance = haversineKm(lat, lon, entry.lat, entry.lon)
|
|
||||||
if best.count < count {
|
|
||||||
let idx = best.firstIndex { $0.distance > distance } ?? best.count
|
|
||||||
best.insert((entry, distance), at: idx)
|
|
||||||
} else if let worstDistance = best.last?.distance, distance < worstDistance {
|
|
||||||
let idx = best.firstIndex { $0.distance > distance } ?? best.count
|
|
||||||
best.insert((entry, distance), at: idx)
|
|
||||||
best.removeLast()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return best.map { "wss://\($0.entry.host)" }
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Remote Fetch
|
// MARK: - Remote Fetch
|
||||||
public func prefetchIfNeeded(force: Bool = false) {
|
func prefetchIfNeeded(force: Bool = false) {
|
||||||
guard !isFetching else { return }
|
guard !isFetching else { return }
|
||||||
|
|
||||||
let now = dependencies.now()
|
let now = dependencies.now()
|
||||||
@@ -175,6 +193,7 @@ public final class GeoRelayDirectory {
|
|||||||
guard now.timeIntervalSince(last) >= dependencies.fetchInterval else { return }
|
guard now.timeIntervalSince(last) >= dependencies.fetchInterval else { return }
|
||||||
} else if last != .distantPast,
|
} else if last != .distantPast,
|
||||||
now.timeIntervalSince(last) < dependencies.retryInitialSeconds {
|
now.timeIntervalSince(last) < dependencies.retryInitialSeconds {
|
||||||
|
// Skip forced fetches if we just refreshed moments ago.
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -258,6 +277,8 @@ public final class GeoRelayDirectory {
|
|||||||
isFetching = false
|
isFetching = false
|
||||||
retryAttempt = 0
|
retryAttempt = 0
|
||||||
cancelRetry()
|
cancelRetry()
|
||||||
|
// Let waiters (e.g. location notes stuck in a "no relays" state) retry.
|
||||||
|
dependencies.notificationCenter.post(name: .geoRelayDirectoryDidRefresh, object: nil)
|
||||||
}
|
}
|
||||||
|
|
||||||
@MainActor
|
@MainActor
|
||||||
@@ -312,6 +333,7 @@ public final class GeoRelayDirectory {
|
|||||||
|
|
||||||
// MARK: - Loading
|
// MARK: - Loading
|
||||||
private func loadLocalEntries() -> [Entry] {
|
private func loadLocalEntries() -> [Entry] {
|
||||||
|
// Prefer cached file if present
|
||||||
if let cache = dependencies.cacheURL(),
|
if let cache = dependencies.cacheURL(),
|
||||||
let data = dependencies.readData(cache),
|
let data = dependencies.readData(cache),
|
||||||
let text = String(data: data, encoding: .utf8) {
|
let text = String(data: data, encoding: .utf8) {
|
||||||
@@ -319,6 +341,7 @@ public final class GeoRelayDirectory {
|
|||||||
if !arr.isEmpty { return arr }
|
if !arr.isEmpty { return arr }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Try bundled resource(s)
|
||||||
let bundleCandidates = dependencies.bundledCSVURLs()
|
let bundleCandidates = dependencies.bundledCSVURLs()
|
||||||
|
|
||||||
for url in bundleCandidates {
|
for url in bundleCandidates {
|
||||||
@@ -329,6 +352,7 @@ public final class GeoRelayDirectory {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Try filesystem path (development/test)
|
||||||
if let cwd = dependencies.currentDirectoryPath(),
|
if let cwd = dependencies.currentDirectoryPath(),
|
||||||
let data = dependencies.readData(URL(fileURLWithPath: cwd).appendingPathComponent("relays/online_relays_gps.csv")),
|
let data = dependencies.readData(URL(fileURLWithPath: cwd).appendingPathComponent("relays/online_relays_gps.csv")),
|
||||||
let text = String(data: data, encoding: .utf8) {
|
let text = String(data: data, encoding: .utf8) {
|
||||||
@@ -339,7 +363,7 @@ public final class GeoRelayDirectory {
|
|||||||
return []
|
return []
|
||||||
}
|
}
|
||||||
|
|
||||||
nonisolated public static func parseCSV(_ text: String) -> [Entry] {
|
nonisolated static func parseCSV(_ text: String) -> [Entry] {
|
||||||
var result: Set<Entry> = []
|
var result: Set<Entry> = []
|
||||||
let lines = text.split(whereSeparator: { $0.isNewline })
|
let lines = text.split(whereSeparator: { $0.isNewline })
|
||||||
for (idx, raw) in lines.enumerated() {
|
for (idx, raw) in lines.enumerated() {
|
||||||
@@ -347,12 +371,7 @@ public final class GeoRelayDirectory {
|
|||||||
if idx == 0 && line.lowercased().contains("relay url") { continue }
|
if idx == 0 && line.lowercased().contains("relay url") { continue }
|
||||||
let parts = line.split(separator: ",").map { $0.trimmed }
|
let parts = line.split(separator: ",").map { $0.trimmed }
|
||||||
guard parts.count >= 3 else { continue }
|
guard parts.count >= 3 else { continue }
|
||||||
var host = parts[0]
|
guard let host = NostrRelayURL.directoryAddress(parts[0]) else { continue }
|
||||||
host = host.replacingOccurrences(of: "https://", with: "")
|
|
||||||
host = host.replacingOccurrences(of: "http://", with: "")
|
|
||||||
host = host.replacingOccurrences(of: "wss://", with: "")
|
|
||||||
host = host.replacingOccurrences(of: "ws://", with: "")
|
|
||||||
host = host.trimmingCharacters(in: CharacterSet(charactersIn: "/"))
|
|
||||||
guard let lat = Double(parts[1]), let lon = Double(parts[2]) else { continue }
|
guard let lat = Double(parts[1]), let lon = Double(parts[2]) else { continue }
|
||||||
result.insert(Entry(host: host, lat: lat, lon: lon))
|
result.insert(Entry(host: host, lat: lat, lon: lon))
|
||||||
}
|
}
|
||||||
@@ -363,19 +382,17 @@ public final class GeoRelayDirectory {
|
|||||||
private func registerObservers() {
|
private func registerObservers() {
|
||||||
let center = dependencies.notificationCenter
|
let center = dependencies.notificationCenter
|
||||||
|
|
||||||
if let torReadyName = dependencies.torReadyNotificationName {
|
let torReady = center.addObserver(
|
||||||
let torReady = center.addObserver(
|
forName: .TorDidBecomeReady,
|
||||||
forName: torReadyName,
|
object: nil,
|
||||||
object: nil,
|
queue: .main
|
||||||
queue: .main
|
) { [weak self] _ in
|
||||||
) { [weak self] _ in
|
guard let self else { return }
|
||||||
guard let self else { return }
|
Task { @MainActor in
|
||||||
Task { @MainActor in
|
self.prefetchIfNeeded(force: true)
|
||||||
self.prefetchIfNeeded(force: true)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
cleanupState.observers.append(torReady)
|
|
||||||
}
|
}
|
||||||
|
cleanupState.observers.append(torReady)
|
||||||
|
|
||||||
if let activeNotificationName = dependencies.activeNotificationName {
|
if let activeNotificationName = dependencies.activeNotificationName {
|
||||||
let didBecomeActive = center.addObserver(
|
let didBecomeActive = center.addObserver(
|
||||||
@@ -407,49 +424,17 @@ public final class GeoRelayDirectory {
|
|||||||
RunLoop.main.add(timer, forMode: .common)
|
RunLoop.main.add(timer, forMode: .common)
|
||||||
}
|
}
|
||||||
|
|
||||||
public var debugRetryAttempt: Int { retryAttempt }
|
var debugRetryAttempt: Int { retryAttempt }
|
||||||
public var debugHasRetryTask: Bool { cleanupState.retryTask != nil }
|
var debugHasRetryTask: Bool { cleanupState.retryTask != nil }
|
||||||
public var debugObserverCount: Int { cleanupState.observers.count }
|
var debugObserverCount: Int { cleanupState.observers.count }
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Distance
|
// MARK: - Distance
|
||||||
private func haversineKm(_ lat1: Double, _ lon1: Double, _ lat2: Double, _ lon2: Double) -> Double {
|
private func haversineKm(_ lat1: Double, _ lon1: Double, _ lat2: Double, _ lon2: Double) -> Double {
|
||||||
let r = 6371.0
|
let r = 6371.0 // Earth radius in km
|
||||||
let dLat = (lat2 - lat1) * .pi / 180
|
let dLat = (lat2 - lat1) * .pi / 180
|
||||||
let dLon = (lon2 - lon1) * .pi / 180
|
let dLon = (lon2 - lon1) * .pi / 180
|
||||||
let a = sin(dLat/2) * sin(dLat/2) + cos(lat1 * .pi/180) * cos(lat2 * .pi/180) * sin(dLon/2) * sin(dLon/2)
|
let a = sin(dLat/2) * sin(dLat/2) + cos(lat1 * .pi/180) * cos(lat2 * .pi/180) * sin(dLon/2) * sin(dLon/2)
|
||||||
let c = 2 * atan2(sqrt(a), sqrt(1 - a))
|
let c = 2 * atan2(sqrt(a), sqrt(1 - a))
|
||||||
return r * c
|
return r * c
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Geohash decode (inline to avoid external dependency)
|
|
||||||
|
|
||||||
private let geohashBase32Map: [Character: Int] = {
|
|
||||||
let chars = Array("0123456789bcdefghjkmnpqrstuvwxyz")
|
|
||||||
var map: [Character: Int] = [:]
|
|
||||||
for (i, c) in chars.enumerated() { map[c] = i }
|
|
||||||
return map
|
|
||||||
}()
|
|
||||||
|
|
||||||
private func decodeGeohashCenter(_ geohash: String) -> (lat: Double, lon: Double) {
|
|
||||||
var latInterval: (Double, Double) = (-90.0, 90.0)
|
|
||||||
var lonInterval: (Double, Double) = (-180.0, 180.0)
|
|
||||||
|
|
||||||
var isEven = true
|
|
||||||
for ch in geohash.lowercased() {
|
|
||||||
guard let cd = geohashBase32Map[ch] else { continue }
|
|
||||||
for mask in [16, 8, 4, 2, 1] {
|
|
||||||
if isEven {
|
|
||||||
let mid = (lonInterval.0 + lonInterval.1) / 2
|
|
||||||
if (cd & mask) != 0 { lonInterval.0 = mid } else { lonInterval.1 = mid }
|
|
||||||
} else {
|
|
||||||
let mid = (latInterval.0 + latInterval.1) / 2
|
|
||||||
if (cd & mask) != 0 { latInterval.0 = mid } else { latInterval.1 = mid }
|
|
||||||
}
|
|
||||||
isEven.toggle()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
let lat = (latInterval.0 + latInterval.1) / 2
|
|
||||||
let lon = (lonInterval.0 + lonInterval.1) / 2
|
|
||||||
return (lat, lon)
|
|
||||||
}
|
|
||||||
+24
-21
@@ -2,56 +2,59 @@ import Foundation
|
|||||||
import P256K
|
import P256K
|
||||||
|
|
||||||
/// Manages Nostr identity (secp256k1 keypair) for NIP-17 private messaging
|
/// Manages Nostr identity (secp256k1 keypair) for NIP-17 private messaging
|
||||||
public struct NostrIdentity: Codable, Sendable {
|
struct NostrIdentity: Codable {
|
||||||
public let privateKey: Data
|
let privateKey: Data
|
||||||
public let publicKey: Data
|
let publicKey: Data
|
||||||
public let npub: String // Bech32-encoded public key
|
let npub: String // Bech32-encoded public key
|
||||||
public let createdAt: Date
|
let createdAt: Date
|
||||||
|
|
||||||
public init(privateKey: Data, publicKey: Data, npub: String, createdAt: Date) {
|
/// Memberwise initializer
|
||||||
|
init(privateKey: Data, publicKey: Data, npub: String, createdAt: Date) {
|
||||||
self.privateKey = privateKey
|
self.privateKey = privateKey
|
||||||
self.publicKey = publicKey
|
self.publicKey = publicKey
|
||||||
self.npub = npub
|
self.npub = npub
|
||||||
self.createdAt = createdAt
|
self.createdAt = createdAt
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Generate a new Nostr identity
|
/// Generate a new Nostr identity
|
||||||
public static func generate() throws -> NostrIdentity {
|
static func generate() throws -> NostrIdentity {
|
||||||
|
// Generate Schnorr key for Nostr
|
||||||
let schnorrKey = try P256K.Schnorr.PrivateKey()
|
let schnorrKey = try P256K.Schnorr.PrivateKey()
|
||||||
let xOnlyPubkey = Data(schnorrKey.xonly.bytes)
|
let xOnlyPubkey = Data(schnorrKey.xonly.bytes)
|
||||||
let npub = try Bech32.encode(hrp: "npub", data: xOnlyPubkey)
|
let npub = try Bech32.encode(hrp: "npub", data: xOnlyPubkey)
|
||||||
|
|
||||||
return NostrIdentity(
|
return NostrIdentity(
|
||||||
privateKey: schnorrKey.dataRepresentation,
|
privateKey: schnorrKey.dataRepresentation,
|
||||||
publicKey: xOnlyPubkey,
|
publicKey: xOnlyPubkey, // Store x-only public key
|
||||||
npub: npub,
|
npub: npub,
|
||||||
createdAt: Date()
|
createdAt: Date()
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Initialize from existing private key data
|
/// Initialize from existing private key data
|
||||||
public init(privateKeyData: Data) throws {
|
init(privateKeyData: Data) throws {
|
||||||
let schnorrKey = try P256K.Schnorr.PrivateKey(dataRepresentation: privateKeyData)
|
let schnorrKey = try P256K.Schnorr.PrivateKey(dataRepresentation: privateKeyData)
|
||||||
let xOnlyPubkey = Data(schnorrKey.xonly.bytes)
|
let xOnlyPubkey = Data(schnorrKey.xonly.bytes)
|
||||||
|
|
||||||
self.privateKey = privateKeyData
|
self.privateKey = privateKeyData
|
||||||
self.publicKey = xOnlyPubkey
|
self.publicKey = xOnlyPubkey
|
||||||
self.npub = try Bech32.encode(hrp: "npub", data: xOnlyPubkey)
|
self.npub = try Bech32.encode(hrp: "npub", data: xOnlyPubkey)
|
||||||
self.createdAt = Date()
|
self.createdAt = Date()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get signing key for event signatures
|
/// Get signing key for event signatures
|
||||||
public func signingKey() throws -> P256K.Signing.PrivateKey {
|
func signingKey() throws -> P256K.Signing.PrivateKey {
|
||||||
try P256K.Signing.PrivateKey(dataRepresentation: privateKey)
|
try P256K.Signing.PrivateKey(dataRepresentation: privateKey)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get Schnorr signing key for Nostr event signatures
|
/// Get Schnorr signing key for Nostr event signatures
|
||||||
public func schnorrSigningKey() throws -> P256K.Schnorr.PrivateKey {
|
func schnorrSigningKey() throws -> P256K.Schnorr.PrivateKey {
|
||||||
try P256K.Schnorr.PrivateKey(dataRepresentation: privateKey)
|
try P256K.Schnorr.PrivateKey(dataRepresentation: privateKey)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get hex-encoded public key (for Nostr events)
|
/// Get hex-encoded public key (for Nostr events)
|
||||||
public var publicKeyHex: String {
|
var publicKeyHex: String {
|
||||||
publicKey.hexEncodedString()
|
// Public key is already stored as x-only (32 bytes)
|
||||||
|
return publicKey.hexEncodedString()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
+42
-44
@@ -1,54 +1,52 @@
|
|||||||
|
import BitFoundation
|
||||||
import Foundation
|
import Foundation
|
||||||
import CryptoKit
|
import CryptoKit
|
||||||
|
|
||||||
/// Minimal keychain access required by NostrIdentityBridge.
|
|
||||||
public protocol NostrKeychainStoring: Sendable {
|
|
||||||
func save(key: String, data: Data, service: String, accessible: CFString?)
|
|
||||||
func load(key: String, service: String) -> Data?
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Bridge between Noise and Nostr identities
|
/// Bridge between Noise and Nostr identities
|
||||||
public final class NostrIdentityBridge {
|
final class NostrIdentityBridge {
|
||||||
private let keychainService = "chat.bitchat.nostr"
|
private let keychainService = "chat.bitchat.nostr"
|
||||||
private let currentIdentityKey = "nostr-current-identity"
|
private let currentIdentityKey = "nostr-current-identity"
|
||||||
private let deviceSeedKey = "nostr-device-seed"
|
private let deviceSeedKey = "nostr-device-seed"
|
||||||
private let deviceSeedCache: NSLock = NSLock()
|
// In-memory cache to avoid transient keychain access issues
|
||||||
private var _deviceSeedCacheValue: Data?
|
private var deviceSeedCache: Data?
|
||||||
// Cache derived identities to avoid repeated crypto during view rendering
|
// Cache derived identities to avoid repeated crypto during view rendering
|
||||||
private var _derivedIdentityCache: [String: NostrIdentity] = [:]
|
private var derivedIdentityCache: [String: NostrIdentity] = [:]
|
||||||
private let cacheLock = NSLock()
|
private let cacheLock = NSLock()
|
||||||
|
|
||||||
private let keychain: any NostrKeychainStoring
|
private let keychain: KeychainManagerProtocol
|
||||||
|
|
||||||
public init(keychain: any NostrKeychainStoring) {
|
init(keychain: KeychainManagerProtocol = KeychainManager()) {
|
||||||
self.keychain = keychain
|
self.keychain = keychain
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get or create the current Nostr identity
|
/// Get or create the current Nostr identity
|
||||||
public func getCurrentNostrIdentity() throws -> NostrIdentity? {
|
func getCurrentNostrIdentity() throws -> NostrIdentity? {
|
||||||
|
// Check if we already have a Nostr identity
|
||||||
if let existingData = keychain.load(key: currentIdentityKey, service: keychainService),
|
if let existingData = keychain.load(key: currentIdentityKey, service: keychainService),
|
||||||
let identity = try? JSONDecoder().decode(NostrIdentity.self, from: existingData) {
|
let identity = try? JSONDecoder().decode(NostrIdentity.self, from: existingData) {
|
||||||
return identity
|
return identity
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Generate new Nostr identity
|
||||||
let nostrIdentity = try NostrIdentity.generate()
|
let nostrIdentity = try NostrIdentity.generate()
|
||||||
|
|
||||||
|
// Store it
|
||||||
let data = try JSONEncoder().encode(nostrIdentity)
|
let data = try JSONEncoder().encode(nostrIdentity)
|
||||||
keychain.save(key: currentIdentityKey, data: data, service: keychainService, accessible: nil)
|
keychain.save(key: currentIdentityKey, data: data, service: keychainService, accessible: nil)
|
||||||
|
|
||||||
return nostrIdentity
|
return nostrIdentity
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Associate a Nostr identity with a Noise public key (for favorites)
|
/// Associate a Nostr identity with a Noise public key (for favorites)
|
||||||
public func associateNostrIdentity(_ nostrPubkey: String, with noisePublicKey: Data) {
|
func associateNostrIdentity(_ nostrPubkey: String, with noisePublicKey: Data) {
|
||||||
let key = "nostr-noise-\(noisePublicKey.base64EncodedString())"
|
let key = "nostr-noise-\(noisePublicKey.base64EncodedString())"
|
||||||
if let data = nostrPubkey.data(using: .utf8) {
|
if let data = nostrPubkey.data(using: .utf8) {
|
||||||
keychain.save(key: key, data: data, service: keychainService, accessible: nil)
|
keychain.save(key: key, data: data, service: keychainService, accessible: nil)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get Nostr public key associated with a Noise public key
|
/// Get Nostr public key associated with a Noise public key
|
||||||
public func getNostrPublicKey(for noisePublicKey: Data) -> String? {
|
func getNostrPublicKey(for noisePublicKey: Data) -> String? {
|
||||||
let key = "nostr-noise-\(noisePublicKey.base64EncodedString())"
|
let key = "nostr-noise-\(noisePublicKey.base64EncodedString())"
|
||||||
guard let data = keychain.load(key: key, service: keychainService),
|
guard let data = keychain.load(key: key, service: keychainService),
|
||||||
let pubkey = String(data: data, encoding: .utf8) else {
|
let pubkey = String(data: data, encoding: .utf8) else {
|
||||||
@@ -56,9 +54,9 @@ public final class NostrIdentityBridge {
|
|||||||
}
|
}
|
||||||
return pubkey
|
return pubkey
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Clear all Nostr identity associations and current identity
|
/// Clear all Nostr identity associations and current identity
|
||||||
public func clearAllAssociations() {
|
func clearAllAssociations() {
|
||||||
let query: [String: Any] = [
|
let query: [String: Any] = [
|
||||||
kSecClass as String: kSecClassGenericPassword,
|
kSecClass as String: kSecClassGenericPassword,
|
||||||
kSecAttrService as String: keychainService,
|
kSecAttrService as String: keychainService,
|
||||||
@@ -79,45 +77,42 @@ public final class NostrIdentityBridge {
|
|||||||
}
|
}
|
||||||
SecItemDelete(deleteQuery as CFDictionary)
|
SecItemDelete(deleteQuery as CFDictionary)
|
||||||
}
|
}
|
||||||
|
} else if status == errSecItemNotFound {
|
||||||
|
// nothing persisted; no action needed
|
||||||
}
|
}
|
||||||
|
|
||||||
deviceSeedCache.lock()
|
deviceSeedCache = nil
|
||||||
_deviceSeedCacheValue = nil
|
|
||||||
deviceSeedCache.unlock()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Per-Geohash Identities (Location Channels)
|
// MARK: - Per-Geohash Identities (Location Channels)
|
||||||
|
|
||||||
|
/// Returns a stable device seed used to derive unlinkable per-geohash identities.
|
||||||
|
/// Stored only on device keychain.
|
||||||
private func getOrCreateDeviceSeed() -> Data {
|
private func getOrCreateDeviceSeed() -> Data {
|
||||||
deviceSeedCache.lock()
|
if let cached = deviceSeedCache { return cached }
|
||||||
if let cached = _deviceSeedCacheValue {
|
|
||||||
deviceSeedCache.unlock()
|
|
||||||
return cached
|
|
||||||
}
|
|
||||||
deviceSeedCache.unlock()
|
|
||||||
|
|
||||||
if let existing = keychain.load(key: deviceSeedKey, service: keychainService) {
|
if let existing = keychain.load(key: deviceSeedKey, service: keychainService) {
|
||||||
|
// Migrate to AfterFirstUnlockThisDeviceOnly for stability during lock
|
||||||
keychain.save(key: deviceSeedKey, data: existing, service: keychainService, accessible: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly)
|
keychain.save(key: deviceSeedKey, data: existing, service: keychainService, accessible: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly)
|
||||||
deviceSeedCache.lock()
|
deviceSeedCache = existing
|
||||||
_deviceSeedCacheValue = existing
|
|
||||||
deviceSeedCache.unlock()
|
|
||||||
return existing
|
return existing
|
||||||
}
|
}
|
||||||
var seed = Data(count: 32)
|
var seed = Data(count: 32)
|
||||||
_ = seed.withUnsafeMutableBytes { ptr in
|
_ = seed.withUnsafeMutableBytes { ptr in
|
||||||
SecRandomCopyBytes(kSecRandomDefault, 32, ptr.baseAddress!)
|
SecRandomCopyBytes(kSecRandomDefault, 32, ptr.baseAddress!)
|
||||||
}
|
}
|
||||||
|
// Ensure availability after first unlock to prevent unintended rotation when locked
|
||||||
keychain.save(key: deviceSeedKey, data: seed, service: keychainService, accessible: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly)
|
keychain.save(key: deviceSeedKey, data: seed, service: keychainService, accessible: kSecAttrAccessibleAfterFirstUnlockThisDeviceOnly)
|
||||||
deviceSeedCache.lock()
|
deviceSeedCache = seed
|
||||||
_deviceSeedCacheValue = seed
|
|
||||||
deviceSeedCache.unlock()
|
|
||||||
return seed
|
return seed
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Derive a deterministic, unlinkable Nostr identity for a given geohash.
|
/// Derive a deterministic, unlinkable Nostr identity for a given geohash.
|
||||||
public func deriveIdentity(forGeohash geohash: String) throws -> NostrIdentity {
|
/// Uses HMAC-SHA256(deviceSeed, geohash) as private key material, with fallback rehashing
|
||||||
|
/// if the candidate is not a valid secp256k1 private key.
|
||||||
|
func deriveIdentity(forGeohash geohash: String) throws -> NostrIdentity {
|
||||||
|
// Check cache first to avoid repeated crypto + keychain I/O during view rendering
|
||||||
cacheLock.lock()
|
cacheLock.lock()
|
||||||
if let cached = _derivedIdentityCache[geohash] {
|
if let cached = derivedIdentityCache[geohash] {
|
||||||
cacheLock.unlock()
|
cacheLock.unlock()
|
||||||
return cached
|
return cached
|
||||||
}
|
}
|
||||||
@@ -138,21 +133,24 @@ public final class NostrIdentityBridge {
|
|||||||
return Data(code)
|
return Data(code)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Try a few iterations to ensure a valid key can be formed
|
||||||
for i in 0..<10 {
|
for i in 0..<10 {
|
||||||
let keyData = candidateKey(iteration: UInt32(i))
|
let keyData = candidateKey(iteration: UInt32(i))
|
||||||
if let identity = try? NostrIdentity(privateKeyData: keyData) {
|
if let identity = try? NostrIdentity(privateKeyData: keyData) {
|
||||||
|
// Cache the result
|
||||||
cacheLock.lock()
|
cacheLock.lock()
|
||||||
_derivedIdentityCache[geohash] = identity
|
derivedIdentityCache[geohash] = identity
|
||||||
cacheLock.unlock()
|
cacheLock.unlock()
|
||||||
return identity
|
return identity
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
// As a final fallback, hash the seed+msg and try again
|
||||||
let fallback = (seed + msg).sha256Hash()
|
let fallback = (seed + msg).sha256Hash()
|
||||||
let identity = try NostrIdentity(privateKeyData: fallback)
|
let identity = try NostrIdentity(privateKeyData: fallback)
|
||||||
|
|
||||||
|
// Cache the result
|
||||||
cacheLock.lock()
|
cacheLock.lock()
|
||||||
_derivedIdentityCache[geohash] = identity
|
derivedIdentityCache[geohash] = identity
|
||||||
cacheLock.unlock()
|
cacheLock.unlock()
|
||||||
|
|
||||||
return identity
|
return identity
|
||||||
@@ -1,118 +0,0 @@
|
|||||||
import Nostr
|
|
||||||
import Combine
|
|
||||||
import Foundation
|
|
||||||
import Tor
|
|
||||||
#if os(iOS)
|
|
||||||
import UIKit
|
|
||||||
#elseif os(macOS)
|
|
||||||
import AppKit
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// MARK: - GeoRelayDirectory Live Dependencies
|
|
||||||
|
|
||||||
extension GeoRelayDirectoryDependencies {
|
|
||||||
@MainActor
|
|
||||||
static func live() -> Self {
|
|
||||||
#if os(iOS)
|
|
||||||
let activeNotificationName: Notification.Name? = UIApplication.didBecomeActiveNotification
|
|
||||||
#elseif os(macOS)
|
|
||||||
let activeNotificationName: Notification.Name? = NSApplication.didBecomeActiveNotification
|
|
||||||
#else
|
|
||||||
let activeNotificationName: Notification.Name? = nil
|
|
||||||
#endif
|
|
||||||
|
|
||||||
return Self(
|
|
||||||
userDefaults: .standard,
|
|
||||||
notificationCenter: .default,
|
|
||||||
now: Date.init,
|
|
||||||
remoteURL: URL(string: "https://raw.githubusercontent.com/permissionlesstech/georelays/refs/heads/main/nostr_relays.csv")!,
|
|
||||||
fetchInterval: TransportConfig.geoRelayFetchIntervalSeconds,
|
|
||||||
refreshCheckInterval: TransportConfig.geoRelayRefreshCheckIntervalSeconds,
|
|
||||||
retryInitialSeconds: TransportConfig.geoRelayRetryInitialSeconds,
|
|
||||||
retryMaxSeconds: TransportConfig.geoRelayRetryMaxSeconds,
|
|
||||||
awaitTorReady: { await TorManager.shared.awaitReady() },
|
|
||||||
makeFetchData: {
|
|
||||||
let session = TorURLSession.shared.session
|
|
||||||
return { request in
|
|
||||||
let (data, _) = try await session.data(for: request)
|
|
||||||
return data
|
|
||||||
}
|
|
||||||
},
|
|
||||||
readData: { try? Data(contentsOf: $0) },
|
|
||||||
writeData: { data, url in
|
|
||||||
try data.write(to: url, options: .atomic)
|
|
||||||
},
|
|
||||||
cacheURL: {
|
|
||||||
do {
|
|
||||||
let base = try FileManager.default.url(
|
|
||||||
for: .applicationSupportDirectory,
|
|
||||||
in: .userDomainMask,
|
|
||||||
appropriateFor: nil,
|
|
||||||
create: true
|
|
||||||
)
|
|
||||||
let dir = base.appendingPathComponent("bitchat", isDirectory: true)
|
|
||||||
try? FileManager.default.createDirectory(at: dir, withIntermediateDirectories: true)
|
|
||||||
return dir.appendingPathComponent("georelays_cache.csv")
|
|
||||||
} catch {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
},
|
|
||||||
bundledCSVURLs: {
|
|
||||||
[
|
|
||||||
Bundle.main.url(forResource: "nostr_relays", withExtension: "csv"),
|
|
||||||
Bundle.main.url(forResource: "online_relays_gps", withExtension: "csv"),
|
|
||||||
Bundle.main.url(forResource: "online_relays_gps", withExtension: "csv", subdirectory: "relays")
|
|
||||||
].compactMap { $0 }
|
|
||||||
},
|
|
||||||
currentDirectoryPath: { FileManager.default.currentDirectoryPath },
|
|
||||||
retrySleep: { delay in
|
|
||||||
let nanoseconds = UInt64(delay * 1_000_000_000)
|
|
||||||
try? await Task.sleep(nanoseconds: nanoseconds)
|
|
||||||
},
|
|
||||||
torReadyNotificationName: .TorDidBecomeReady,
|
|
||||||
activeNotificationName: activeNotificationName,
|
|
||||||
autoStart: true
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// MARK: - NostrRelayManager Live Dependencies
|
|
||||||
|
|
||||||
extension NostrRelayManagerDependencies {
|
|
||||||
@MainActor
|
|
||||||
static func live() -> Self {
|
|
||||||
Self(
|
|
||||||
activationAllowed: { NetworkActivationService.shared.activationAllowed },
|
|
||||||
userTorEnabled: { NetworkActivationService.shared.userTorEnabled },
|
|
||||||
hasMutualFavorites: { !FavoritesPersistenceService.shared.mutualFavorites.isEmpty },
|
|
||||||
hasLocationPermission: { LocationChannelManager.shared.permissionState == .authorized },
|
|
||||||
mutualFavoritesPublisher: FavoritesPersistenceService.shared.$mutualFavorites.eraseToAnyPublisher(),
|
|
||||||
locationPermissionPublisher: LocationChannelManager.shared.$permissionState
|
|
||||||
.map { state -> LocationPermissionState in
|
|
||||||
switch state {
|
|
||||||
case .notDetermined:.notDetermined
|
|
||||||
case .authorized: .authorized
|
|
||||||
case .denied: .denied
|
|
||||||
case .restricted: .denied
|
|
||||||
}
|
|
||||||
}
|
|
||||||
.eraseToAnyPublisher(),
|
|
||||||
torEnforced: { TorManager.shared.torEnforced },
|
|
||||||
torIsReady: { TorManager.shared.isReady },
|
|
||||||
torIsForeground: { TorManager.shared.isForeground() },
|
|
||||||
awaitTorReady: { completion in
|
|
||||||
Task.detached {
|
|
||||||
let ready = await TorManager.shared.awaitReady()
|
|
||||||
await MainActor.run {
|
|
||||||
completion(ready)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
makeSession: { NostrRelayManager.makeURLSession(TorURLSession.shared.session) },
|
|
||||||
scheduleAfter: { delay, action in
|
|
||||||
DispatchQueue.main.asyncAfter(deadline: .now() + delay, execute: action)
|
|
||||||
},
|
|
||||||
now: Date.init
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
+190
-96
@@ -1,15 +1,17 @@
|
|||||||
import BitFoundation
|
|
||||||
import BitLogger
|
import BitLogger
|
||||||
import Foundation
|
import Foundation
|
||||||
import CryptoKit
|
import CryptoKit
|
||||||
import P256K
|
import P256K
|
||||||
import Security
|
import Security
|
||||||
|
|
||||||
/// NIP-17 Protocol Implementation for Private Direct Messages
|
// Note: This file depends on Data extension from BinaryEncodingUtils.swift
|
||||||
public struct NostrProtocol {
|
// Make sure BinaryEncodingUtils.swift is included in the target
|
||||||
|
|
||||||
|
/// NIP-17 Protocol Implementation for Private Direct Messages
|
||||||
|
struct NostrProtocol {
|
||||||
|
|
||||||
/// Nostr event kinds
|
/// Nostr event kinds
|
||||||
public enum EventKind: Int, Sendable {
|
enum EventKind: Int {
|
||||||
case metadata = 0
|
case metadata = 0
|
||||||
case textNote = 1
|
case textNote = 1
|
||||||
case dm = 14 // NIP-17 DM rumor kind
|
case dm = 14 // NIP-17 DM rumor kind
|
||||||
@@ -18,71 +20,88 @@ public struct NostrProtocol {
|
|||||||
case ephemeralEvent = 20000
|
case ephemeralEvent = 20000
|
||||||
case geohashPresence = 20001
|
case geohashPresence = 20001
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Create a NIP-17 private message
|
/// Create a NIP-17 private message
|
||||||
public static func createPrivateMessage(
|
static func createPrivateMessage(
|
||||||
content: String,
|
content: String,
|
||||||
recipientPubkey: String,
|
recipientPubkey: String,
|
||||||
senderIdentity: NostrIdentity
|
senderIdentity: NostrIdentity
|
||||||
) throws -> NostrEvent {
|
) throws -> NostrEvent {
|
||||||
|
|
||||||
|
// Creating private message
|
||||||
|
|
||||||
|
// 1. Create the rumor (unsigned event)
|
||||||
let rumor = NostrEvent(
|
let rumor = NostrEvent(
|
||||||
pubkey: senderIdentity.publicKeyHex,
|
pubkey: senderIdentity.publicKeyHex,
|
||||||
createdAt: Date(),
|
createdAt: Date(),
|
||||||
kind: .dm,
|
kind: .dm, // NIP-17: DM rumor kind 14
|
||||||
tags: [],
|
tags: [],
|
||||||
content: content
|
content: content
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// 2. Create ephemeral key for this message
|
||||||
let ephemeralKey = try P256K.Schnorr.PrivateKey()
|
let ephemeralKey = try P256K.Schnorr.PrivateKey()
|
||||||
|
// Created ephemeral key for seal
|
||||||
|
|
||||||
|
// 3. Seal the rumor (encrypt to recipient)
|
||||||
let sealedEvent = try createSeal(
|
let sealedEvent = try createSeal(
|
||||||
rumor: rumor,
|
rumor: rumor,
|
||||||
recipientPubkey: recipientPubkey,
|
recipientPubkey: recipientPubkey,
|
||||||
senderKey: ephemeralKey
|
senderKey: ephemeralKey
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// 4. Gift wrap the sealed event (encrypt to recipient again)
|
||||||
let giftWrap = try createGiftWrap(
|
let giftWrap = try createGiftWrap(
|
||||||
seal: sealedEvent,
|
seal: sealedEvent,
|
||||||
recipientPubkey: recipientPubkey,
|
recipientPubkey: recipientPubkey,
|
||||||
senderKey: ephemeralKey
|
senderKey: ephemeralKey
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Created gift wrap
|
||||||
|
|
||||||
return giftWrap
|
return giftWrap
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Decrypt a received NIP-17 message
|
/// Decrypt a received NIP-17 message
|
||||||
/// Returns the content, sender pubkey, and the actual message timestamp (not the randomized gift wrap timestamp)
|
/// Returns the content, sender pubkey, and the actual message timestamp (not the randomized gift wrap timestamp)
|
||||||
public static func decryptPrivateMessage(
|
static func decryptPrivateMessage(
|
||||||
giftWrap: NostrEvent,
|
giftWrap: NostrEvent,
|
||||||
recipientIdentity: NostrIdentity
|
recipientIdentity: NostrIdentity
|
||||||
) throws -> (content: String, senderPubkey: String, timestamp: Int) {
|
) throws -> (content: String, senderPubkey: String, timestamp: Int) {
|
||||||
|
|
||||||
|
// Starting decryption
|
||||||
|
|
||||||
|
// 1. Unwrap the gift wrap
|
||||||
let seal: NostrEvent
|
let seal: NostrEvent
|
||||||
do {
|
do {
|
||||||
seal = try unwrapGiftWrap(
|
seal = try unwrapGiftWrap(
|
||||||
giftWrap: giftWrap,
|
giftWrap: giftWrap,
|
||||||
recipientKey: recipientIdentity.schnorrSigningKey()
|
recipientKey: recipientIdentity.schnorrSigningKey()
|
||||||
)
|
)
|
||||||
|
// Successfully unwrapped gift wrap
|
||||||
} catch {
|
} catch {
|
||||||
SecureLogger.error("❌ Failed to unwrap gift wrap: \(error)", category: .session)
|
SecureLogger.error("❌ Failed to unwrap gift wrap: \(error)", category: .session)
|
||||||
throw error
|
throw error
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 2. Open the seal
|
||||||
let rumor: NostrEvent
|
let rumor: NostrEvent
|
||||||
do {
|
do {
|
||||||
rumor = try openSeal(
|
rumor = try openSeal(
|
||||||
seal: seal,
|
seal: seal,
|
||||||
recipientKey: recipientIdentity.schnorrSigningKey()
|
recipientKey: recipientIdentity.schnorrSigningKey()
|
||||||
)
|
)
|
||||||
|
// Successfully opened seal
|
||||||
} catch {
|
} catch {
|
||||||
SecureLogger.error("❌ Failed to open seal: \(error)", category: .session)
|
SecureLogger.error("❌ Failed to open seal: \(error)", category: .session)
|
||||||
throw error
|
throw error
|
||||||
}
|
}
|
||||||
|
|
||||||
return (content: rumor.content, senderPubkey: rumor.pubkey, timestamp: rumor.created_at)
|
return (content: rumor.content, senderPubkey: rumor.pubkey, timestamp: rumor.created_at)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Create a geohash-scoped ephemeral public message (kind 20000)
|
/// Create a geohash-scoped ephemeral public message (kind 20000)
|
||||||
public static func createEphemeralGeohashEvent(
|
static func createEphemeralGeohashEvent(
|
||||||
content: String,
|
content: String,
|
||||||
geohash: String,
|
geohash: String,
|
||||||
senderIdentity: NostrIdentity,
|
senderIdentity: NostrIdentity,
|
||||||
@@ -109,7 +128,7 @@ public struct NostrProtocol {
|
|||||||
|
|
||||||
/// Create a geohash presence heartbeat (kind 20001)
|
/// Create a geohash presence heartbeat (kind 20001)
|
||||||
/// Must contain empty content and NO nickname tag
|
/// Must contain empty content and NO nickname tag
|
||||||
public static func createGeohashPresenceEvent(
|
static func createGeohashPresenceEvent(
|
||||||
geohash: String,
|
geohash: String,
|
||||||
senderIdentity: NostrIdentity
|
senderIdentity: NostrIdentity
|
||||||
) throws -> NostrEvent {
|
) throws -> NostrEvent {
|
||||||
@@ -126,7 +145,7 @@ public struct NostrProtocol {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Create a persistent location note (kind 1: text note) tagged to a street-level geohash.
|
/// Create a persistent location note (kind 1: text note) tagged to a street-level geohash.
|
||||||
public static func createGeohashTextNote(
|
static func createGeohashTextNote(
|
||||||
content: String,
|
content: String,
|
||||||
geohash: String,
|
geohash: String,
|
||||||
senderIdentity: NostrIdentity,
|
senderIdentity: NostrIdentity,
|
||||||
@@ -146,21 +165,22 @@ public struct NostrProtocol {
|
|||||||
let schnorrKey = try senderIdentity.schnorrSigningKey()
|
let schnorrKey = try senderIdentity.schnorrSigningKey()
|
||||||
return try event.sign(with: schnorrKey)
|
return try event.sign(with: schnorrKey)
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Private Methods
|
// MARK: - Private Methods
|
||||||
|
|
||||||
private static func createSeal(
|
private static func createSeal(
|
||||||
rumor: NostrEvent,
|
rumor: NostrEvent,
|
||||||
recipientPubkey: String,
|
recipientPubkey: String,
|
||||||
senderKey: P256K.Schnorr.PrivateKey
|
senderKey: P256K.Schnorr.PrivateKey
|
||||||
) throws -> NostrEvent {
|
) throws -> NostrEvent {
|
||||||
|
|
||||||
let rumorJSON = try rumor.jsonString()
|
let rumorJSON = try rumor.jsonString()
|
||||||
let encrypted = try encrypt(
|
let encrypted = try encrypt(
|
||||||
plaintext: rumorJSON,
|
plaintext: rumorJSON,
|
||||||
recipientPubkey: recipientPubkey,
|
recipientPubkey: recipientPubkey,
|
||||||
senderKey: senderKey
|
senderKey: senderKey
|
||||||
)
|
)
|
||||||
|
|
||||||
let seal = NostrEvent(
|
let seal = NostrEvent(
|
||||||
pubkey: Data(senderKey.xonly.bytes).hexEncodedString(),
|
pubkey: Data(senderKey.xonly.bytes).hexEncodedString(),
|
||||||
createdAt: randomizedTimestamp(),
|
createdAt: randomizedTimestamp(),
|
||||||
@@ -168,111 +188,133 @@ public struct NostrProtocol {
|
|||||||
tags: [],
|
tags: [],
|
||||||
content: encrypted
|
content: encrypted
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Sign the seal with the sender's Schnorr private key
|
||||||
return try seal.sign(with: senderKey)
|
return try seal.sign(with: senderKey)
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func createGiftWrap(
|
private static func createGiftWrap(
|
||||||
seal: NostrEvent,
|
seal: NostrEvent,
|
||||||
recipientPubkey: String,
|
recipientPubkey: String,
|
||||||
senderKey: P256K.Schnorr.PrivateKey
|
senderKey: P256K.Schnorr.PrivateKey // This is the ephemeral key used for the seal
|
||||||
) throws -> NostrEvent {
|
) throws -> NostrEvent {
|
||||||
|
|
||||||
let sealJSON = try seal.jsonString()
|
let sealJSON = try seal.jsonString()
|
||||||
|
|
||||||
|
// Create new ephemeral key for gift wrap
|
||||||
let wrapKey = try P256K.Schnorr.PrivateKey()
|
let wrapKey = try P256K.Schnorr.PrivateKey()
|
||||||
|
// Creating gift wrap with ephemeral key
|
||||||
|
|
||||||
|
// Encrypt the seal with the new ephemeral key (not the seal's key)
|
||||||
let encrypted = try encrypt(
|
let encrypted = try encrypt(
|
||||||
plaintext: sealJSON,
|
plaintext: sealJSON,
|
||||||
recipientPubkey: recipientPubkey,
|
recipientPubkey: recipientPubkey,
|
||||||
senderKey: wrapKey
|
senderKey: wrapKey // Use the gift wrap ephemeral key
|
||||||
)
|
)
|
||||||
|
|
||||||
let giftWrap = NostrEvent(
|
let giftWrap = NostrEvent(
|
||||||
pubkey: Data(wrapKey.xonly.bytes).hexEncodedString(),
|
pubkey: Data(wrapKey.xonly.bytes).hexEncodedString(),
|
||||||
createdAt: randomizedTimestamp(),
|
createdAt: randomizedTimestamp(),
|
||||||
kind: .giftWrap,
|
kind: .giftWrap,
|
||||||
tags: [["p", recipientPubkey]],
|
tags: [["p", recipientPubkey]], // Tag recipient
|
||||||
content: encrypted
|
content: encrypted
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Sign the gift wrap with the wrap Schnorr private key
|
||||||
return try giftWrap.sign(with: wrapKey)
|
return try giftWrap.sign(with: wrapKey)
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func unwrapGiftWrap(
|
private static func unwrapGiftWrap(
|
||||||
giftWrap: NostrEvent,
|
giftWrap: NostrEvent,
|
||||||
recipientKey: P256K.Schnorr.PrivateKey
|
recipientKey: P256K.Schnorr.PrivateKey
|
||||||
) throws -> NostrEvent {
|
) throws -> NostrEvent {
|
||||||
|
|
||||||
|
// Unwrapping gift wrap
|
||||||
|
|
||||||
let decrypted = try decrypt(
|
let decrypted = try decrypt(
|
||||||
ciphertext: giftWrap.content,
|
ciphertext: giftWrap.content,
|
||||||
senderPubkey: giftWrap.pubkey,
|
senderPubkey: giftWrap.pubkey,
|
||||||
recipientKey: recipientKey
|
recipientKey: recipientKey
|
||||||
)
|
)
|
||||||
|
|
||||||
guard let data = decrypted.data(using: .utf8),
|
guard let data = decrypted.data(using: .utf8),
|
||||||
let sealDict = try JSONSerialization.jsonObject(with: data) as? [String: Any] else {
|
let sealDict = try JSONSerialization.jsonObject(with: data) as? [String: Any] else {
|
||||||
throw NostrError.invalidEvent
|
throw NostrError.invalidEvent
|
||||||
}
|
}
|
||||||
|
|
||||||
return try NostrEvent(from: sealDict)
|
let seal = try NostrEvent(from: sealDict)
|
||||||
|
// Unwrapped seal
|
||||||
|
|
||||||
|
return seal
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func openSeal(
|
private static func openSeal(
|
||||||
seal: NostrEvent,
|
seal: NostrEvent,
|
||||||
recipientKey: P256K.Schnorr.PrivateKey
|
recipientKey: P256K.Schnorr.PrivateKey
|
||||||
) throws -> NostrEvent {
|
) throws -> NostrEvent {
|
||||||
|
|
||||||
let decrypted = try decrypt(
|
let decrypted = try decrypt(
|
||||||
ciphertext: seal.content,
|
ciphertext: seal.content,
|
||||||
senderPubkey: seal.pubkey,
|
senderPubkey: seal.pubkey,
|
||||||
recipientKey: recipientKey
|
recipientKey: recipientKey
|
||||||
)
|
)
|
||||||
|
|
||||||
guard let data = decrypted.data(using: .utf8),
|
guard let data = decrypted.data(using: .utf8),
|
||||||
let rumorDict = try JSONSerialization.jsonObject(with: data) as? [String: Any] else {
|
let rumorDict = try JSONSerialization.jsonObject(with: data) as? [String: Any] else {
|
||||||
throw NostrError.invalidEvent
|
throw NostrError.invalidEvent
|
||||||
}
|
}
|
||||||
|
|
||||||
return try NostrEvent(from: rumorDict)
|
return try NostrEvent(from: rumorDict)
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Encryption (NIP-44 v2)
|
// MARK: - Encryption (NIP-44 v2)
|
||||||
|
|
||||||
private static func encrypt(
|
private static func encrypt(
|
||||||
plaintext: String,
|
plaintext: String,
|
||||||
recipientPubkey: String,
|
recipientPubkey: String,
|
||||||
senderKey: P256K.Schnorr.PrivateKey
|
senderKey: P256K.Schnorr.PrivateKey
|
||||||
) throws -> String {
|
) throws -> String {
|
||||||
|
|
||||||
guard let recipientPubkeyData = Data(hexString: recipientPubkey) else {
|
guard let recipientPubkeyData = Data(hexString: recipientPubkey) else {
|
||||||
throw NostrError.invalidPublicKey
|
throw NostrError.invalidPublicKey
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Encrypting message (NIP-44 v2: XChaCha20-Poly1305, versioned)
|
||||||
|
|
||||||
|
// Derive shared secret
|
||||||
let sharedSecret = try deriveSharedSecret(
|
let sharedSecret = try deriveSharedSecret(
|
||||||
privateKey: senderKey,
|
privateKey: senderKey,
|
||||||
publicKey: recipientPubkeyData
|
publicKey: recipientPubkeyData
|
||||||
)
|
)
|
||||||
|
// Derive NIP-44 v2 symmetric key (HKDF-SHA256 with label in info)
|
||||||
let key = try deriveNIP44V2Key(from: sharedSecret)
|
let key = try deriveNIP44V2Key(from: sharedSecret)
|
||||||
|
|
||||||
|
// 24-byte random nonce for XChaCha20-Poly1305
|
||||||
var nonce24 = Data(count: 24)
|
var nonce24 = Data(count: 24)
|
||||||
_ = nonce24.withUnsafeMutableBytes { ptr in
|
_ = nonce24.withUnsafeMutableBytes { ptr in
|
||||||
SecRandomCopyBytes(kSecRandomDefault, 24, ptr.baseAddress!)
|
SecRandomCopyBytes(kSecRandomDefault, 24, ptr.baseAddress!)
|
||||||
}
|
}
|
||||||
|
|
||||||
let pt = Data(plaintext.utf8)
|
let pt = Data(plaintext.utf8)
|
||||||
let sealed = try XChaCha20Poly1305Compat.seal(plaintext: pt, key: key, nonce24: nonce24)
|
let sealed = try XChaCha20Poly1305Compat.seal(plaintext: pt, key: key, nonce24: nonce24)
|
||||||
|
|
||||||
|
// v2: base64url(nonce24 || ciphertext || tag)
|
||||||
var combined = Data()
|
var combined = Data()
|
||||||
combined.append(nonce24)
|
combined.append(nonce24)
|
||||||
combined.append(sealed.ciphertext)
|
combined.append(sealed.ciphertext)
|
||||||
combined.append(sealed.tag)
|
combined.append(sealed.tag)
|
||||||
return "v2:" + base64URLEncode(combined)
|
return "v2:" + Base64URLCoding.encode(combined)
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func decrypt(
|
private static func decrypt(
|
||||||
ciphertext: String,
|
ciphertext: String,
|
||||||
senderPubkey: String,
|
senderPubkey: String,
|
||||||
recipientKey: P256K.Schnorr.PrivateKey
|
recipientKey: P256K.Schnorr.PrivateKey
|
||||||
) throws -> String {
|
) throws -> String {
|
||||||
|
// Expect NIP-44 v2 format
|
||||||
guard ciphertext.hasPrefix("v2:") else { throw NostrError.invalidCiphertext }
|
guard ciphertext.hasPrefix("v2:") else { throw NostrError.invalidCiphertext }
|
||||||
let encoded = String(ciphertext.dropFirst(3))
|
let encoded = String(ciphertext.dropFirst(3))
|
||||||
guard let data = base64URLDecode(encoded),
|
guard let data = Base64URLCoding.decode(encoded),
|
||||||
data.count > (24 + 16),
|
data.count > (24 + 16),
|
||||||
let senderPubkeyData = Data(hexString: senderPubkey) else {
|
let senderPubkeyData = Data(hexString: senderPubkey) else {
|
||||||
throw NostrError.invalidCiphertext
|
throw NostrError.invalidCiphertext
|
||||||
@@ -283,6 +325,7 @@ public struct NostrProtocol {
|
|||||||
let tag = rest.suffix(16)
|
let tag = rest.suffix(16)
|
||||||
let ct = rest.dropLast(16)
|
let ct = rest.dropLast(16)
|
||||||
|
|
||||||
|
// Try decryption with even-Y then odd-Y when sender pubkey is x-only
|
||||||
func attemptDecrypt(using pubKeyData: Data) throws -> Data {
|
func attemptDecrypt(using pubKeyData: Data) throws -> Data {
|
||||||
let ss = try deriveSharedSecret(privateKey: recipientKey, publicKey: pubKeyData)
|
let ss = try deriveSharedSecret(privateKey: recipientKey, publicKey: pubKeyData)
|
||||||
let key = try deriveNIP44V2Key(from: ss)
|
let key = try deriveNIP44V2Key(from: ss)
|
||||||
@@ -294,6 +337,7 @@ public struct NostrProtocol {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// If 32 bytes (x-only) try both parities, otherwise single try
|
||||||
if senderPubkeyData.count == 32 {
|
if senderPubkeyData.count == 32 {
|
||||||
let even = Data([0x02]) + senderPubkeyData
|
let even = Data([0x02]) + senderPubkeyData
|
||||||
if let pt = try? attemptDecrypt(using: even) {
|
if let pt = try? attemptDecrypt(using: even) {
|
||||||
@@ -307,23 +351,32 @@ public struct NostrProtocol {
|
|||||||
return String(data: pt, encoding: .utf8) ?? ""
|
return String(data: pt, encoding: .utf8) ?? ""
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func deriveSharedSecret(
|
private static func deriveSharedSecret(
|
||||||
privateKey: P256K.Schnorr.PrivateKey,
|
privateKey: P256K.Schnorr.PrivateKey,
|
||||||
publicKey: Data
|
publicKey: Data
|
||||||
) throws -> Data {
|
) throws -> Data {
|
||||||
|
// Deriving shared secret
|
||||||
|
|
||||||
|
// Convert Schnorr private key to KeyAgreement private key
|
||||||
let keyAgreementPrivateKey = try P256K.KeyAgreement.PrivateKey(
|
let keyAgreementPrivateKey = try P256K.KeyAgreement.PrivateKey(
|
||||||
dataRepresentation: privateKey.dataRepresentation
|
dataRepresentation: privateKey.dataRepresentation
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Create KeyAgreement public key from the public key data
|
||||||
|
// For ECDH, we need the full 33-byte compressed public key (with 0x02 or 0x03 prefix)
|
||||||
var fullPublicKey = Data()
|
var fullPublicKey = Data()
|
||||||
if publicKey.count == 32 {
|
if publicKey.count == 32 { // X-only key, need to add prefix
|
||||||
|
// For x-only keys in Nostr/Bitcoin, we need to try both possible Y coordinates
|
||||||
|
// First try with even Y (0x02 prefix)
|
||||||
fullPublicKey.append(0x02)
|
fullPublicKey.append(0x02)
|
||||||
fullPublicKey.append(publicKey)
|
fullPublicKey.append(publicKey)
|
||||||
|
// Trying with even Y coordinate
|
||||||
} else {
|
} else {
|
||||||
fullPublicKey = publicKey
|
fullPublicKey = publicKey
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Try to create public key, if it fails with even Y, try odd Y
|
||||||
let keyAgreementPublicKey: P256K.KeyAgreement.PublicKey
|
let keyAgreementPublicKey: P256K.KeyAgreement.PublicKey
|
||||||
do {
|
do {
|
||||||
keyAgreementPublicKey = try P256K.KeyAgreement.PublicKey(
|
keyAgreementPublicKey = try P256K.KeyAgreement.PublicKey(
|
||||||
@@ -332,6 +385,8 @@ public struct NostrProtocol {
|
|||||||
)
|
)
|
||||||
} catch {
|
} catch {
|
||||||
if publicKey.count == 32 {
|
if publicKey.count == 32 {
|
||||||
|
// Try with odd Y (0x03 prefix)
|
||||||
|
// Even Y failed, trying odd Y
|
||||||
fullPublicKey = Data()
|
fullPublicKey = Data()
|
||||||
fullPublicKey.append(0x03)
|
fullPublicKey.append(0x03)
|
||||||
fullPublicKey.append(publicKey)
|
fullPublicKey.append(publicKey)
|
||||||
@@ -343,32 +398,85 @@ public struct NostrProtocol {
|
|||||||
throw error
|
throw error
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Perform ECDH
|
||||||
let sharedSecret = try keyAgreementPrivateKey.sharedSecretFromKeyAgreement(
|
let sharedSecret = try keyAgreementPrivateKey.sharedSecretFromKeyAgreement(
|
||||||
with: keyAgreementPublicKey,
|
with: keyAgreementPublicKey,
|
||||||
format: .compressed
|
format: .compressed
|
||||||
)
|
)
|
||||||
|
|
||||||
return sharedSecret.withUnsafeBytes { Data($0) }
|
// Convert SharedSecret to Data
|
||||||
|
let sharedSecretData = sharedSecret.withUnsafeBytes { Data($0) }
|
||||||
|
// ECDH shared secret derived
|
||||||
|
|
||||||
|
// Return raw ECDH shared secret; HKDF is applied by deriveNIP44V2Key
|
||||||
|
return sharedSecretData
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Direct version that doesn't try to add prefixes
|
||||||
|
private static func deriveSharedSecretDirect(
|
||||||
|
privateKey: P256K.Schnorr.PrivateKey,
|
||||||
|
publicKey: Data
|
||||||
|
) throws -> Data {
|
||||||
|
// Direct shared secret calculation
|
||||||
|
|
||||||
|
// Convert Schnorr private key to KeyAgreement private key
|
||||||
|
let keyAgreementPrivateKey = try P256K.KeyAgreement.PrivateKey(
|
||||||
|
dataRepresentation: privateKey.dataRepresentation
|
||||||
|
)
|
||||||
|
|
||||||
|
// Use the public key as-is (should already have prefix)
|
||||||
|
let keyAgreementPublicKey = try P256K.KeyAgreement.PublicKey(
|
||||||
|
dataRepresentation: publicKey,
|
||||||
|
format: .compressed
|
||||||
|
)
|
||||||
|
|
||||||
|
// Perform ECDH
|
||||||
|
let sharedSecret = try keyAgreementPrivateKey.sharedSecretFromKeyAgreement(
|
||||||
|
with: keyAgreementPublicKey,
|
||||||
|
format: .compressed
|
||||||
|
)
|
||||||
|
|
||||||
|
// Convert SharedSecret to Data
|
||||||
|
let sharedSecretData = sharedSecret.withUnsafeBytes { Data($0) }
|
||||||
|
|
||||||
|
// Return raw ECDH shared secret; HKDF is applied by deriveNIP44V2Key
|
||||||
|
return sharedSecretData
|
||||||
|
}
|
||||||
|
|
||||||
private static func randomizedTimestamp() -> Date {
|
private static func randomizedTimestamp() -> Date {
|
||||||
let offset = TimeInterval.random(in: -900...900)
|
// Add random offset to current time for privacy
|
||||||
return Date().addingTimeInterval(offset)
|
// This prevents timing correlation attacks while the actual message timestamp
|
||||||
|
// is preserved in the encrypted rumor
|
||||||
|
let offset = TimeInterval.random(in: -900...900) // +/- 15 minutes
|
||||||
|
let now = Date()
|
||||||
|
let randomized = now.addingTimeInterval(offset)
|
||||||
|
|
||||||
|
// Log with explicit UTC and local time for debugging
|
||||||
|
let formatter = DateFormatter()
|
||||||
|
//
|
||||||
|
formatter.dateFormat = "yyyy-MM-dd HH:mm:ss"
|
||||||
|
formatter.timeZone = TimeZone(abbreviation: "UTC")
|
||||||
|
|
||||||
|
formatter.timeZone = TimeZone.current
|
||||||
|
|
||||||
|
// Timestamp randomized for privacy
|
||||||
|
|
||||||
|
return randomized
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Nostr Event structure
|
/// Nostr Event structure
|
||||||
public struct NostrEvent: Codable, Sendable {
|
struct NostrEvent: Codable {
|
||||||
public var id: String
|
var id: String
|
||||||
public let pubkey: String
|
let pubkey: String
|
||||||
public let created_at: Int
|
let created_at: Int
|
||||||
public let kind: Int
|
let kind: Int
|
||||||
public let tags: [[String]]
|
let tags: [[String]]
|
||||||
public let content: String
|
let content: String
|
||||||
public var sig: String?
|
var sig: String?
|
||||||
|
|
||||||
public init(
|
init(
|
||||||
pubkey: String,
|
pubkey: String,
|
||||||
createdAt: Date,
|
createdAt: Date,
|
||||||
kind: NostrProtocol.EventKind,
|
kind: NostrProtocol.EventKind,
|
||||||
@@ -381,10 +489,10 @@ public struct NostrEvent: Codable, Sendable {
|
|||||||
self.tags = tags
|
self.tags = tags
|
||||||
self.content = content
|
self.content = content
|
||||||
self.sig = nil
|
self.sig = nil
|
||||||
self.id = ""
|
self.id = "" // Will be set during signing
|
||||||
}
|
}
|
||||||
|
|
||||||
public init(from dict: [String: Any]) throws {
|
init(from dict: [String: Any]) throws {
|
||||||
guard let pubkey = dict["pubkey"] as? String,
|
guard let pubkey = dict["pubkey"] as? String,
|
||||||
let createdAt = dict["created_at"] as? Int,
|
let createdAt = dict["created_at"] as? Int,
|
||||||
let kind = dict["kind"] as? Int,
|
let kind = dict["kind"] as? Int,
|
||||||
@@ -392,7 +500,7 @@ public struct NostrEvent: Codable, Sendable {
|
|||||||
let content = dict["content"] as? String else {
|
let content = dict["content"] as? String else {
|
||||||
throw NostrError.invalidEvent
|
throw NostrError.invalidEvent
|
||||||
}
|
}
|
||||||
|
|
||||||
self.id = dict["id"] as? String ?? ""
|
self.id = dict["id"] as? String ?? ""
|
||||||
self.pubkey = pubkey
|
self.pubkey = pubkey
|
||||||
self.created_at = createdAt
|
self.created_at = createdAt
|
||||||
@@ -401,19 +509,20 @@ public struct NostrEvent: Codable, Sendable {
|
|||||||
self.content = content
|
self.content = content
|
||||||
self.sig = dict["sig"] as? String
|
self.sig = dict["sig"] as? String
|
||||||
}
|
}
|
||||||
|
|
||||||
public func sign(with key: P256K.Schnorr.PrivateKey) throws -> NostrEvent {
|
func sign(with key: P256K.Schnorr.PrivateKey) throws -> NostrEvent {
|
||||||
let (eventId, eventIdHash) = try calculateEventId()
|
let (eventId, eventIdHash) = try calculateEventId()
|
||||||
|
|
||||||
|
// Sign with Schnorr (BIP-340)
|
||||||
var messageBytes = [UInt8](eventIdHash)
|
var messageBytes = [UInt8](eventIdHash)
|
||||||
var auxRand = [UInt8](repeating: 0, count: 32)
|
var auxRand = [UInt8](repeating: 0, count: 32)
|
||||||
_ = auxRand.withUnsafeMutableBytes { ptr in
|
_ = auxRand.withUnsafeMutableBytes { ptr in
|
||||||
SecRandomCopyBytes(kSecRandomDefault, 32, ptr.baseAddress!)
|
SecRandomCopyBytes(kSecRandomDefault, 32, ptr.baseAddress!)
|
||||||
}
|
}
|
||||||
let schnorrSignature = try key.signature(message: &messageBytes, auxiliaryRand: &auxRand)
|
let schnorrSignature = try key.signature(message: &messageBytes, auxiliaryRand: &auxRand)
|
||||||
|
|
||||||
let signatureHex = schnorrSignature.dataRepresentation.hexEncodedString()
|
let signatureHex = schnorrSignature.dataRepresentation.hexEncodedString()
|
||||||
|
|
||||||
var signed = self
|
var signed = self
|
||||||
signed.id = eventId
|
signed.id = eventId
|
||||||
signed.sig = signatureHex
|
signed.sig = signatureHex
|
||||||
@@ -422,7 +531,7 @@ public struct NostrEvent: Codable, Sendable {
|
|||||||
|
|
||||||
/// Validate that the event ID and Schnorr signature match the content and pubkey.
|
/// Validate that the event ID and Schnorr signature match the content and pubkey.
|
||||||
/// Returns false when the signature is missing, malformed, or does not verify.
|
/// Returns false when the signature is missing, malformed, or does not verify.
|
||||||
public func isValidSignature() -> Bool {
|
func isValidSignature() -> Bool {
|
||||||
guard let sig = sig,
|
guard let sig = sig,
|
||||||
let sigData = Data(hexString: sig),
|
let sigData = Data(hexString: sig),
|
||||||
let pubData = Data(hexString: pubkey),
|
let pubData = Data(hexString: pubkey),
|
||||||
@@ -439,7 +548,7 @@ public struct NostrEvent: Codable, Sendable {
|
|||||||
let xonly = P256K.Schnorr.XonlyKey(dataRepresentation: pubData)
|
let xonly = P256K.Schnorr.XonlyKey(dataRepresentation: pubData)
|
||||||
return xonly.isValid(signature, for: &messageBytes)
|
return xonly.isValid(signature, for: &messageBytes)
|
||||||
}
|
}
|
||||||
|
|
||||||
private func calculateEventId() throws -> (String, Data) {
|
private func calculateEventId() throws -> (String, Data) {
|
||||||
let serialized = [
|
let serialized = [
|
||||||
0,
|
0,
|
||||||
@@ -449,12 +558,12 @@ public struct NostrEvent: Codable, Sendable {
|
|||||||
tags,
|
tags,
|
||||||
content
|
content
|
||||||
] as [Any]
|
] as [Any]
|
||||||
|
|
||||||
let data = try JSONSerialization.data(withJSONObject: serialized, options: [.withoutEscapingSlashes])
|
let data = try JSONSerialization.data(withJSONObject: serialized, options: [.withoutEscapingSlashes])
|
||||||
return (data.sha256Fingerprint(), data.sha256Hash())
|
return (data.sha256Fingerprint(), data.sha256Hash())
|
||||||
}
|
}
|
||||||
|
|
||||||
public func jsonString() throws -> String {
|
func jsonString() throws -> String {
|
||||||
let encoder = JSONEncoder()
|
let encoder = JSONEncoder()
|
||||||
encoder.outputFormatting = [.withoutEscapingSlashes]
|
encoder.outputFormatting = [.withoutEscapingSlashes]
|
||||||
let data = try encoder.encode(self)
|
let data = try encoder.encode(self)
|
||||||
@@ -462,7 +571,7 @@ public struct NostrEvent: Codable, Sendable {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public enum NostrError: Error, Sendable {
|
enum NostrError: Error {
|
||||||
case invalidPublicKey
|
case invalidPublicKey
|
||||||
case invalidPrivateKey
|
case invalidPrivateKey
|
||||||
case invalidEvent
|
case invalidEvent
|
||||||
@@ -471,24 +580,9 @@ public enum NostrError: Error, Sendable {
|
|||||||
case encryptionFailed
|
case encryptionFailed
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - NIP-44 v2 helpers (XChaCha20-Poly1305 + base64url)
|
// MARK: - NIP-44 v2 helpers (XChaCha20-Poly1305)
|
||||||
|
|
||||||
private extension NostrProtocol {
|
private extension NostrProtocol {
|
||||||
static func base64URLEncode(_ data: Data) -> String {
|
|
||||||
return data.base64EncodedString()
|
|
||||||
.replacingOccurrences(of: "+", with: "-")
|
|
||||||
.replacingOccurrences(of: "/", with: "_")
|
|
||||||
.replacingOccurrences(of: "=", with: "")
|
|
||||||
}
|
|
||||||
|
|
||||||
static func base64URLDecode(_ s: String) -> Data? {
|
|
||||||
var str = s
|
|
||||||
let pad = (4 - (str.count % 4)) % 4
|
|
||||||
if pad > 0 { str += String(repeating: "=", count: pad) }
|
|
||||||
str = str.replacingOccurrences(of: "-", with: "+").replacingOccurrences(of: "_", with: "/")
|
|
||||||
return Data(base64Encoded: str)
|
|
||||||
}
|
|
||||||
|
|
||||||
static func deriveNIP44V2Key(from sharedSecretData: Data) throws -> Data {
|
static func deriveNIP44V2Key(from sharedSecretData: Data) throws -> Data {
|
||||||
let derivedKey = HKDF<CryptoKit.SHA256>.deriveKey(
|
let derivedKey = HKDF<CryptoKit.SHA256>.deriveKey(
|
||||||
inputKeyMaterial: SymmetricKey(data: sharedSecretData),
|
inputKeyMaterial: SymmetricKey(data: sharedSecretData),
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,51 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum NostrRelayURL {
|
||||||
|
static func normalized(_ rawValue: String, defaultScheme: String? = nil) -> String? {
|
||||||
|
var value = rawValue.trimmingCharacters(in: .whitespacesAndNewlines)
|
||||||
|
guard !value.isEmpty else { return nil }
|
||||||
|
|
||||||
|
if !value.contains("://"), let defaultScheme {
|
||||||
|
value = "\(defaultScheme)://\(value)"
|
||||||
|
}
|
||||||
|
|
||||||
|
guard var components = URLComponents(string: value),
|
||||||
|
let rawScheme = components.scheme?.lowercased(),
|
||||||
|
let rawHost = components.host?.lowercased(),
|
||||||
|
!rawHost.isEmpty else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
switch rawScheme {
|
||||||
|
case "wss", "https":
|
||||||
|
components.scheme = "wss"
|
||||||
|
if components.port == 443 {
|
||||||
|
components.port = nil
|
||||||
|
}
|
||||||
|
case "ws", "http":
|
||||||
|
components.scheme = "ws"
|
||||||
|
if components.port == 80 {
|
||||||
|
components.port = nil
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
components.host = rawHost
|
||||||
|
if components.path == "/" {
|
||||||
|
components.path = ""
|
||||||
|
}
|
||||||
|
components.fragment = nil
|
||||||
|
|
||||||
|
return components.string
|
||||||
|
}
|
||||||
|
|
||||||
|
static func directoryAddress(_ rawValue: String) -> String? {
|
||||||
|
guard var normalized = normalized(rawValue, defaultScheme: "wss") else { return nil }
|
||||||
|
for prefix in ["wss://", "ws://"] where normalized.hasPrefix(prefix) {
|
||||||
|
normalized.removeFirst(prefix.count)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
return normalized
|
||||||
|
}
|
||||||
|
}
|
||||||
+9
@@ -6,6 +6,7 @@ import CryptoKit
|
|||||||
/// as per XChaCha20 construction.
|
/// as per XChaCha20 construction.
|
||||||
enum XChaCha20Poly1305Compat {
|
enum XChaCha20Poly1305Compat {
|
||||||
|
|
||||||
|
/// Errors that can occur during XChaCha20-Poly1305 operations
|
||||||
enum Error: Swift.Error {
|
enum Error: Swift.Error {
|
||||||
case invalidKeyLength(expected: Int, got: Int)
|
case invalidKeyLength(expected: Int, got: Int)
|
||||||
case invalidNonceLength(expected: Int, got: Int)
|
case invalidNonceLength(expected: Int, got: Int)
|
||||||
@@ -58,6 +59,7 @@ enum XChaCha20Poly1305Compat {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private static func hchacha20(key: Data, nonce16: Data) throws -> Data {
|
private static func hchacha20(key: Data, nonce16: Data) throws -> Data {
|
||||||
|
// HChaCha20 based on the original ChaCha20 core with a 16-byte nonce.
|
||||||
guard key.count == 32 else {
|
guard key.count == 32 else {
|
||||||
throw Error.invalidKeyLength(expected: 32, got: key.count)
|
throw Error.invalidKeyLength(expected: 32, got: key.count)
|
||||||
}
|
}
|
||||||
@@ -68,22 +70,28 @@ enum XChaCha20Poly1305Compat {
|
|||||||
// Constants "expand 32-byte k"
|
// Constants "expand 32-byte k"
|
||||||
var state: [UInt32] = [
|
var state: [UInt32] = [
|
||||||
0x61707865, 0x3320646e, 0x79622d32, 0x6b206574,
|
0x61707865, 0x3320646e, 0x79622d32, 0x6b206574,
|
||||||
|
// key (8 words)
|
||||||
key.loadLEWord(0), key.loadLEWord(4), key.loadLEWord(8), key.loadLEWord(12),
|
key.loadLEWord(0), key.loadLEWord(4), key.loadLEWord(8), key.loadLEWord(12),
|
||||||
key.loadLEWord(16), key.loadLEWord(20), key.loadLEWord(24), key.loadLEWord(28),
|
key.loadLEWord(16), key.loadLEWord(20), key.loadLEWord(24), key.loadLEWord(28),
|
||||||
|
// nonce (4 words)
|
||||||
nonce16.loadLEWord(0), nonce16.loadLEWord(4), nonce16.loadLEWord(8), nonce16.loadLEWord(12)
|
nonce16.loadLEWord(0), nonce16.loadLEWord(4), nonce16.loadLEWord(8), nonce16.loadLEWord(12)
|
||||||
]
|
]
|
||||||
|
|
||||||
|
// 20 rounds (10 double rounds)
|
||||||
for _ in 0..<10 {
|
for _ in 0..<10 {
|
||||||
|
// Column rounds
|
||||||
quarterRound(&state, 0, 4, 8, 12)
|
quarterRound(&state, 0, 4, 8, 12)
|
||||||
quarterRound(&state, 1, 5, 9, 13)
|
quarterRound(&state, 1, 5, 9, 13)
|
||||||
quarterRound(&state, 2, 6, 10, 14)
|
quarterRound(&state, 2, 6, 10, 14)
|
||||||
quarterRound(&state, 3, 7, 11, 15)
|
quarterRound(&state, 3, 7, 11, 15)
|
||||||
|
// Diagonal rounds
|
||||||
quarterRound(&state, 0, 5, 10, 15)
|
quarterRound(&state, 0, 5, 10, 15)
|
||||||
quarterRound(&state, 1, 6, 11, 12)
|
quarterRound(&state, 1, 6, 11, 12)
|
||||||
quarterRound(&state, 2, 7, 8, 13)
|
quarterRound(&state, 2, 7, 8, 13)
|
||||||
quarterRound(&state, 3, 4, 9, 14)
|
quarterRound(&state, 3, 4, 9, 14)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Output subkey: state[0..3] and state[12..15]
|
||||||
var out = Data(count: 32)
|
var out = Data(count: 32)
|
||||||
out.storeLEWord(state[0], at: 0)
|
out.storeLEWord(state[0], at: 0)
|
||||||
out.storeLEWord(state[1], at: 4)
|
out.storeLEWord(state[1], at: 4)
|
||||||
@@ -124,3 +132,4 @@ private extension Data {
|
|||||||
replaceSubrange(offset..<(offset+4), with: bytes)
|
replaceSubrange(offset..<(offset+4), with: bytes)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -7,6 +7,7 @@
|
|||||||
//
|
//
|
||||||
|
|
||||||
import Foundation
|
import Foundation
|
||||||
|
import BitFoundation
|
||||||
import BitLogger
|
import BitLogger
|
||||||
|
|
||||||
/// TLV payload for Bluetooth mesh file transfers (voice notes, images, generic files).
|
/// TLV payload for Bluetooth mesh file transfers (voice notes, images, generic files).
|
||||||
|
|||||||
@@ -62,40 +62,6 @@ import Foundation
|
|||||||
import CoreBluetooth
|
import CoreBluetooth
|
||||||
import BitFoundation
|
import BitFoundation
|
||||||
|
|
||||||
// MARK: - Message Types
|
|
||||||
|
|
||||||
/// Simplified BitChat protocol message types.
|
|
||||||
/// Reduced from 24 types to just 6 essential ones.
|
|
||||||
/// All private communication metadata (receipts, status) is embedded in noiseEncrypted payloads.
|
|
||||||
enum MessageType: UInt8 {
|
|
||||||
// Public messages (unencrypted)
|
|
||||||
case announce = 0x01 // "I'm here" with nickname
|
|
||||||
case message = 0x02 // Public chat message
|
|
||||||
case leave = 0x03 // "I'm leaving"
|
|
||||||
case requestSync = 0x21 // GCS filter-based sync request (local-only)
|
|
||||||
|
|
||||||
// Noise encryption
|
|
||||||
case noiseHandshake = 0x10 // Handshake (init or response determined by payload)
|
|
||||||
case noiseEncrypted = 0x11 // All encrypted payloads (messages, receipts, etc.)
|
|
||||||
|
|
||||||
// Fragmentation (simplified)
|
|
||||||
case fragment = 0x20 // Single fragment type for large messages
|
|
||||||
case fileTransfer = 0x22 // Binary file/audio/image payloads
|
|
||||||
|
|
||||||
var description: String {
|
|
||||||
switch self {
|
|
||||||
case .announce: return "announce"
|
|
||||||
case .message: return "message"
|
|
||||||
case .leave: return "leave"
|
|
||||||
case .requestSync: return "requestSync"
|
|
||||||
case .noiseHandshake: return "noiseHandshake"
|
|
||||||
case .noiseEncrypted: return "noiseEncrypted"
|
|
||||||
case .fragment: return "fragment"
|
|
||||||
case .fileTransfer: return "fileTransfer"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// MARK: - Noise Payload Types
|
// MARK: - Noise Payload Types
|
||||||
|
|
||||||
/// Types of payloads embedded within noiseEncrypted messages.
|
/// Types of payloads embedded within noiseEncrypted messages.
|
||||||
@@ -132,35 +98,6 @@ enum LazyHandshakeState {
|
|||||||
case failed(Error) // Handshake failed
|
case failed(Error) // Handshake failed
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Delivery Status
|
|
||||||
|
|
||||||
// Delivery status for messages
|
|
||||||
enum DeliveryStatus: Codable, Equatable, Hashable {
|
|
||||||
case sending
|
|
||||||
case sent // Left our device
|
|
||||||
case delivered(to: String, at: Date) // Confirmed by recipient
|
|
||||||
case read(by: String, at: Date) // Seen by recipient
|
|
||||||
case failed(reason: String)
|
|
||||||
case partiallyDelivered(reached: Int, total: Int) // For rooms
|
|
||||||
|
|
||||||
var displayText: String {
|
|
||||||
switch self {
|
|
||||||
case .sending:
|
|
||||||
return "Sending..."
|
|
||||||
case .sent:
|
|
||||||
return "Sent"
|
|
||||||
case .delivered(let nickname, _):
|
|
||||||
return "Delivered to \(nickname)"
|
|
||||||
case .read(let nickname, _):
|
|
||||||
return "Read by \(nickname)"
|
|
||||||
case .failed(let reason):
|
|
||||||
return "Failed: \(reason)"
|
|
||||||
case .partiallyDelivered(let reached, let total):
|
|
||||||
return "Delivered to \(reached)/\(total)"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// MARK: - Delegate Protocol
|
// MARK: - Delegate Protocol
|
||||||
|
|
||||||
protocol BitchatDelegate: AnyObject {
|
protocol BitchatDelegate: AnyObject {
|
||||||
|
|||||||
@@ -0,0 +1,209 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import BitLogger
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
/// Narrow environment for `BLEAnnounceHandler`.
|
||||||
|
///
|
||||||
|
/// All queue hops (collections barrier, BLE-queue link-state reads, main-actor
|
||||||
|
/// UI notification, delayed re-announce) live inside the closures supplied by
|
||||||
|
/// `BLEService`, keeping the handler queue-agnostic and synchronously testable.
|
||||||
|
struct BLEAnnounceHandlerEnvironment {
|
||||||
|
/// Local peer identity at the time the announce is handled.
|
||||||
|
let localPeerID: () -> PeerID
|
||||||
|
/// TTL value used for direct (non-relayed) packets.
|
||||||
|
let messageTTL: UInt8
|
||||||
|
/// Current time source.
|
||||||
|
let now: () -> Date
|
||||||
|
/// Noise public key already recorded for the peer, if any (registry read).
|
||||||
|
let existingNoisePublicKey: (PeerID) -> Data?
|
||||||
|
/// Verifies the packet signature against the announced signing key.
|
||||||
|
let verifySignature: (_ packet: BitchatPacket, _ signingPublicKey: Data) -> Bool
|
||||||
|
/// Direct link state for the peer (BLE-queue read).
|
||||||
|
let linkState: (PeerID) -> (hasPeripheral: Bool, hasCentral: Bool)
|
||||||
|
/// Runs the registry mutation phase under the collections barrier.
|
||||||
|
let withRegistryBarrier: (() -> Void) -> Void
|
||||||
|
/// Upserts the verified announce into the peer registry.
|
||||||
|
/// Must only be called from inside `withRegistryBarrier`.
|
||||||
|
let upsertVerifiedAnnounce: (
|
||||||
|
_ peerID: PeerID,
|
||||||
|
_ announcement: AnnouncementPacket,
|
||||||
|
_ isConnected: Bool,
|
||||||
|
_ now: Date
|
||||||
|
) -> BLEPeerAnnounceUpdate
|
||||||
|
/// Debounced reconnect-log decision.
|
||||||
|
/// Must only be called from inside `withRegistryBarrier`.
|
||||||
|
let shouldEmitReconnectLog: (_ peerID: PeerID, _ now: Date) -> Bool
|
||||||
|
/// Records verified direct-neighbor claims in the mesh topology.
|
||||||
|
let updateTopology: (_ peerID: PeerID, _ neighbors: [Data]) -> Void
|
||||||
|
/// Persists the announced cryptographic identity for offline verification.
|
||||||
|
let persistIdentity: (AnnouncementPacket) -> Void
|
||||||
|
/// Announce-back dedup check.
|
||||||
|
let dedupContains: (String) -> Bool
|
||||||
|
/// Announce-back dedup marking.
|
||||||
|
let dedupMarkProcessed: (String) -> Void
|
||||||
|
/// Delivers the announce UI events as one ordered main-actor hop:
|
||||||
|
/// `.peerConnected` (if flagged) → initial gossip sync scheduling (if
|
||||||
|
/// flagged) → peer-ID snapshot + data publish + `.peerListUpdated`.
|
||||||
|
/// A single closure keeps the original in-order delivery guarantee that
|
||||||
|
/// separate unstructured tasks would not provide.
|
||||||
|
let deliverAnnounceUIEvents: (
|
||||||
|
_ peerID: PeerID,
|
||||||
|
_ notifyPeerConnected: Bool,
|
||||||
|
_ scheduleInitialSync: Bool
|
||||||
|
) -> Void
|
||||||
|
/// Tracks the announce packet for gossip sync.
|
||||||
|
let trackPacketSeen: (BitchatPacket) -> Void
|
||||||
|
/// Reciprocates the announce for bidirectional discovery.
|
||||||
|
let sendAnnounceBack: () -> Void
|
||||||
|
/// Schedules a delayed re-announce (afterglow) after the given delay.
|
||||||
|
let scheduleAfterglow: (TimeInterval) -> Void
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Orchestrates inbound announce packets: preflight validation, signature
|
||||||
|
/// trust, registry/topology updates, identity persistence, UI notification,
|
||||||
|
/// gossip tracking, and the reciprocal announce response.
|
||||||
|
final class BLEAnnounceHandler {
|
||||||
|
private let environment: BLEAnnounceHandlerEnvironment
|
||||||
|
|
||||||
|
init(environment: BLEAnnounceHandlerEnvironment) {
|
||||||
|
self.environment = environment
|
||||||
|
}
|
||||||
|
|
||||||
|
func handle(_ packet: BitchatPacket, from peerID: PeerID) {
|
||||||
|
let env = environment
|
||||||
|
let now = env.now()
|
||||||
|
let preflight = BLEAnnouncePreflightPolicy.evaluate(
|
||||||
|
packet: packet,
|
||||||
|
from: peerID,
|
||||||
|
localPeerID: env.localPeerID(),
|
||||||
|
now: now
|
||||||
|
)
|
||||||
|
|
||||||
|
let announcement: AnnouncementPacket
|
||||||
|
switch preflight {
|
||||||
|
case .accept(let acceptance):
|
||||||
|
announcement = acceptance.announcement
|
||||||
|
case .reject(.malformed):
|
||||||
|
SecureLogger.error("❌ Failed to decode announce packet from \(peerID.id.prefix(8))…", category: .session)
|
||||||
|
return
|
||||||
|
case .reject(.senderMismatch(let derivedFromKey)):
|
||||||
|
SecureLogger.warning("⚠️ Announce sender mismatch: derived \(derivedFromKey.id.prefix(8))… vs packet \(peerID.id.prefix(8))…", category: .security)
|
||||||
|
return
|
||||||
|
case .reject(.selfAnnounce):
|
||||||
|
return
|
||||||
|
case .reject(.stale(let ageSeconds)):
|
||||||
|
SecureLogger.debug("⏰ Ignoring stale announce from \(peerID.id.prefix(8))… (age: \(ageSeconds)s)", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Suppress announce logs to reduce noise
|
||||||
|
|
||||||
|
// Precompute signature verification outside barrier to reduce contention
|
||||||
|
let existingNoisePublicKey = env.existingNoisePublicKey(peerID)
|
||||||
|
let hasSignature = packet.signature != nil
|
||||||
|
let signatureValid: Bool
|
||||||
|
if hasSignature {
|
||||||
|
signatureValid = env.verifySignature(packet, announcement.signingPublicKey)
|
||||||
|
if !signatureValid {
|
||||||
|
SecureLogger.warning("⚠️ Signature verification for announce failed \(peerID.id.prefix(8))", category: .security)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
signatureValid = false
|
||||||
|
}
|
||||||
|
let trustDecision = BLEAnnounceTrustPolicy.evaluate(
|
||||||
|
hasSignature: hasSignature,
|
||||||
|
signatureValid: signatureValid,
|
||||||
|
existingNoisePublicKey: existingNoisePublicKey,
|
||||||
|
announcedNoisePublicKey: announcement.noisePublicKey
|
||||||
|
)
|
||||||
|
if case .reject(.keyMismatch) = trustDecision {
|
||||||
|
SecureLogger.warning("⚠️ Announce key mismatch for \(peerID.id.prefix(8))… — keeping unverified", category: .security)
|
||||||
|
}
|
||||||
|
let verifiedAnnounce = trustDecision.isVerified
|
||||||
|
|
||||||
|
var isNewPeer = false
|
||||||
|
var isReconnectedPeer = false
|
||||||
|
let directLinkState = env.linkState(peerID)
|
||||||
|
let isDirectAnnounce = packet.ttl == env.messageTTL
|
||||||
|
|
||||||
|
env.withRegistryBarrier {
|
||||||
|
let hasPeripheralConnection = directLinkState.hasPeripheral
|
||||||
|
let hasCentralSubscription = directLinkState.hasCentral
|
||||||
|
|
||||||
|
// Require verified announce; ignore otherwise (no backward compatibility)
|
||||||
|
if !verifiedAnnounce {
|
||||||
|
SecureLogger.warning("❌ Ignoring unverified announce from \(peerID.id.prefix(8))…", category: .security)
|
||||||
|
// Reset flags to prevent post-barrier code from acting on unverified announces
|
||||||
|
isNewPeer = false
|
||||||
|
isReconnectedPeer = false
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let update = env.upsertVerifiedAnnounce(
|
||||||
|
peerID,
|
||||||
|
announcement,
|
||||||
|
isDirectAnnounce || hasPeripheralConnection || hasCentralSubscription,
|
||||||
|
now
|
||||||
|
)
|
||||||
|
isNewPeer = update.isNewPeer
|
||||||
|
isReconnectedPeer = update.wasDisconnected
|
||||||
|
|
||||||
|
// Log connection status only for direct connectivity changes; debounce to reduce spam
|
||||||
|
if isDirectAnnounce || hasPeripheralConnection || hasCentralSubscription {
|
||||||
|
let now = env.now()
|
||||||
|
if update.isNewPeer {
|
||||||
|
SecureLogger.debug("🆕 New peer: \(announcement.nickname)", category: .session)
|
||||||
|
} else if update.wasDisconnected {
|
||||||
|
if env.shouldEmitReconnectLog(peerID, now) {
|
||||||
|
SecureLogger.debug("🔄 Peer \(announcement.nickname) reconnected", category: .session)
|
||||||
|
}
|
||||||
|
} else if let previousNickname = update.previousNickname, previousNickname != announcement.nickname {
|
||||||
|
SecureLogger.debug("🔄 Peer \(peerID.id.prefix(8))… changed nickname: \(previousNickname) -> \(announcement.nickname)", category: .session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update topology with verified neighbor claims (only for authenticated announces)
|
||||||
|
if verifiedAnnounce, let neighbors = announcement.directNeighbors {
|
||||||
|
env.updateTopology(peerID, neighbors)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Persist cryptographic identity and signing key for robust offline verification
|
||||||
|
env.persistIdentity(announcement)
|
||||||
|
|
||||||
|
let announceBackID = "announce-back-\(peerID)"
|
||||||
|
let shouldSendBack = !env.dedupContains(announceBackID)
|
||||||
|
if shouldSendBack {
|
||||||
|
env.dedupMarkProcessed(announceBackID)
|
||||||
|
}
|
||||||
|
let responsePlan = BLEAnnounceResponsePolicy.plan(
|
||||||
|
isDirectAnnounce: isDirectAnnounce,
|
||||||
|
isNewPeer: isNewPeer,
|
||||||
|
isReconnectedPeer: isReconnectedPeer,
|
||||||
|
shouldSendAnnounceBack: shouldSendBack
|
||||||
|
)
|
||||||
|
|
||||||
|
// Only notify of connection for new or reconnected peers when it is a
|
||||||
|
// direct announce; the list update always follows in the same hop.
|
||||||
|
env.deliverAnnounceUIEvents(
|
||||||
|
peerID,
|
||||||
|
responsePlan.shouldNotifyPeerConnected,
|
||||||
|
responsePlan.shouldNotifyPeerConnected && responsePlan.shouldScheduleInitialSync
|
||||||
|
)
|
||||||
|
|
||||||
|
// Track for sync (include our own and others' announces)
|
||||||
|
env.trackPacketSeen(packet)
|
||||||
|
|
||||||
|
if responsePlan.shouldSendAnnounceBack {
|
||||||
|
// Reciprocate announce for bidirectional discovery
|
||||||
|
// Force send to ensure the peer receives our announce
|
||||||
|
env.sendAnnounceBack()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Afterglow: on first-seen peers, schedule a short re-announce to push presence one more hop
|
||||||
|
if responsePlan.shouldScheduleAfterglow {
|
||||||
|
let delay = Double.random(in: 0.3...0.6)
|
||||||
|
env.scheduleAfterglow(delay)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,116 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEAnnouncePreflightAcceptance {
|
||||||
|
let announcement: AnnouncementPacket
|
||||||
|
let derivedPeerID: PeerID
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEAnnouncePreflightRejection: Equatable {
|
||||||
|
case malformed
|
||||||
|
case senderMismatch(derivedPeerID: PeerID)
|
||||||
|
case selfAnnounce
|
||||||
|
case stale(ageSeconds: Double)
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEAnnouncePreflightDecision {
|
||||||
|
case accept(BLEAnnouncePreflightAcceptance)
|
||||||
|
case reject(BLEAnnouncePreflightRejection)
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEAnnouncePreflightPolicy {
|
||||||
|
static func evaluate(
|
||||||
|
packet: BitchatPacket,
|
||||||
|
from peerID: PeerID,
|
||||||
|
localPeerID: PeerID,
|
||||||
|
now: Date
|
||||||
|
) -> BLEAnnouncePreflightDecision {
|
||||||
|
guard let announcement = AnnouncementPacket.decode(from: packet.payload) else {
|
||||||
|
return .reject(.malformed)
|
||||||
|
}
|
||||||
|
|
||||||
|
let derivedPeerID = PeerID(publicKey: announcement.noisePublicKey)
|
||||||
|
guard derivedPeerID == peerID else {
|
||||||
|
return .reject(.senderMismatch(derivedPeerID: derivedPeerID))
|
||||||
|
}
|
||||||
|
|
||||||
|
guard peerID != localPeerID else {
|
||||||
|
return .reject(.selfAnnounce)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard !BLEPacketFreshnessPolicy.isStale(timestampMilliseconds: packet.timestamp, now: now) else {
|
||||||
|
return .reject(.stale(ageSeconds: BLEPacketFreshnessPolicy.ageSeconds(
|
||||||
|
timestampMilliseconds: packet.timestamp,
|
||||||
|
now: now
|
||||||
|
)))
|
||||||
|
}
|
||||||
|
|
||||||
|
return .accept(BLEAnnouncePreflightAcceptance(
|
||||||
|
announcement: announcement,
|
||||||
|
derivedPeerID: derivedPeerID
|
||||||
|
))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEAnnounceTrustRejection: Equatable {
|
||||||
|
case missingSignature
|
||||||
|
case invalidSignature
|
||||||
|
case keyMismatch
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEAnnounceTrustDecision: Equatable {
|
||||||
|
case verified
|
||||||
|
case reject(BLEAnnounceTrustRejection)
|
||||||
|
|
||||||
|
var isVerified: Bool {
|
||||||
|
self == .verified
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEAnnounceTrustPolicy {
|
||||||
|
static func evaluate(
|
||||||
|
hasSignature: Bool,
|
||||||
|
signatureValid: Bool,
|
||||||
|
existingNoisePublicKey: Data?,
|
||||||
|
announcedNoisePublicKey: Data
|
||||||
|
) -> BLEAnnounceTrustDecision {
|
||||||
|
if let existingNoisePublicKey, existingNoisePublicKey != announcedNoisePublicKey {
|
||||||
|
return .reject(.keyMismatch)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard hasSignature else {
|
||||||
|
return .reject(.missingSignature)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard signatureValid else {
|
||||||
|
return .reject(.invalidSignature)
|
||||||
|
}
|
||||||
|
|
||||||
|
return .verified
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEAnnounceResponsePlan: Equatable {
|
||||||
|
let shouldNotifyPeerConnected: Bool
|
||||||
|
let shouldScheduleInitialSync: Bool
|
||||||
|
let shouldSendAnnounceBack: Bool
|
||||||
|
let shouldScheduleAfterglow: Bool
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEAnnounceResponsePolicy {
|
||||||
|
static func plan(
|
||||||
|
isDirectAnnounce: Bool,
|
||||||
|
isNewPeer: Bool,
|
||||||
|
isReconnectedPeer: Bool,
|
||||||
|
shouldSendAnnounceBack: Bool
|
||||||
|
) -> BLEAnnounceResponsePlan {
|
||||||
|
let shouldNotifyPeerConnected = isDirectAnnounce && (isNewPeer || isReconnectedPeer)
|
||||||
|
|
||||||
|
return BLEAnnounceResponsePlan(
|
||||||
|
shouldNotifyPeerConnected: shouldNotifyPeerConnected,
|
||||||
|
shouldScheduleInitialSync: shouldNotifyPeerConnected,
|
||||||
|
shouldSendAnnounceBack: shouldSendAnnounceBack,
|
||||||
|
shouldScheduleAfterglow: isNewPeer
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,31 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEAnnounceThrottle {
|
||||||
|
private var lastSent: Date
|
||||||
|
private let normalMinimumInterval: TimeInterval
|
||||||
|
private let forcedMinimumInterval: TimeInterval
|
||||||
|
|
||||||
|
init(
|
||||||
|
lastSent: Date = .distantPast,
|
||||||
|
normalMinimumInterval: TimeInterval = TransportConfig.bleAnnounceMinInterval,
|
||||||
|
forcedMinimumInterval: TimeInterval = TransportConfig.bleForceAnnounceMinIntervalSeconds
|
||||||
|
) {
|
||||||
|
self.lastSent = lastSent
|
||||||
|
self.normalMinimumInterval = normalMinimumInterval
|
||||||
|
self.forcedMinimumInterval = forcedMinimumInterval
|
||||||
|
}
|
||||||
|
|
||||||
|
func elapsed(since now: Date) -> TimeInterval {
|
||||||
|
now.timeIntervalSince(lastSent)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func shouldSend(force: Bool, now: Date) -> Bool {
|
||||||
|
let minimumInterval = force ? forcedMinimumInterval : normalMinimumInterval
|
||||||
|
guard elapsed(since: now) >= minimumInterval else {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
lastSent = now
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,293 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEConnectionCandidate<Peripheral> {
|
||||||
|
let peripheral: Peripheral
|
||||||
|
let peripheralID: String
|
||||||
|
let rssi: Int
|
||||||
|
let name: String
|
||||||
|
let isConnectable: Bool
|
||||||
|
let discoveredAt: Date
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEExistingConnectionState {
|
||||||
|
let isConnecting: Bool
|
||||||
|
let isConnected: Bool
|
||||||
|
let lastConnectionAttempt: Date?
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEPeripheralConnectionState {
|
||||||
|
case disconnected
|
||||||
|
case connecting
|
||||||
|
case connected
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEDiscoveryDecision: Equatable {
|
||||||
|
case ignore
|
||||||
|
case queued
|
||||||
|
case scheduleRetry(after: TimeInterval)
|
||||||
|
case cancelStaleConnection
|
||||||
|
case connectNow
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEConnectionQueueDecision<Peripheral> {
|
||||||
|
case none
|
||||||
|
case retryAfter(TimeInterval)
|
||||||
|
case connect(BLEConnectionCandidate<Peripheral>)
|
||||||
|
}
|
||||||
|
|
||||||
|
final class BLEConnectionScheduler<Peripheral> {
|
||||||
|
private let maxCentralLinks: Int
|
||||||
|
private let connectRateLimitInterval: TimeInterval
|
||||||
|
private let candidateCap: Int
|
||||||
|
private let weakLinkCooldownSeconds: TimeInterval
|
||||||
|
private let weakLinkRSSICutoff: Int
|
||||||
|
|
||||||
|
private var lastGlobalConnectAttempt: Date = .distantPast
|
||||||
|
private var candidates: [BLEConnectionCandidate<Peripheral>] = []
|
||||||
|
private var failureCounts: [String: Int] = [:]
|
||||||
|
private var recentConnectTimeouts: [String: Date] = [:]
|
||||||
|
// Tracked separately from connect timeouts: a peer we held a connection
|
||||||
|
// with and lost (walked out of range) usually comes back, so it only gets
|
||||||
|
// a brief rediscovery ignore — not the timeout backoff/cooldown treatment
|
||||||
|
// reserved for peers that never answered a connect attempt.
|
||||||
|
private var recentDisconnects: [String: Date] = [:]
|
||||||
|
private var lastIsolatedAt: Date?
|
||||||
|
|
||||||
|
private let initialDynamicRSSIThreshold: Int
|
||||||
|
private(set) var dynamicRSSIThreshold: Int
|
||||||
|
|
||||||
|
var candidateCount: Int {
|
||||||
|
candidates.count
|
||||||
|
}
|
||||||
|
|
||||||
|
init(
|
||||||
|
maxCentralLinks: Int = TransportConfig.bleMaxCentralLinks,
|
||||||
|
connectRateLimitInterval: TimeInterval = TransportConfig.bleConnectRateLimitInterval,
|
||||||
|
candidateCap: Int = TransportConfig.bleConnectionCandidatesMax,
|
||||||
|
weakLinkCooldownSeconds: TimeInterval = TransportConfig.bleWeakLinkCooldownSeconds,
|
||||||
|
weakLinkRSSICutoff: Int = TransportConfig.bleWeakLinkRSSICutoff,
|
||||||
|
dynamicRSSIThreshold: Int = TransportConfig.bleDynamicRSSIThresholdDefault
|
||||||
|
) {
|
||||||
|
self.maxCentralLinks = maxCentralLinks
|
||||||
|
self.connectRateLimitInterval = connectRateLimitInterval
|
||||||
|
self.candidateCap = candidateCap
|
||||||
|
self.weakLinkCooldownSeconds = weakLinkCooldownSeconds
|
||||||
|
self.weakLinkRSSICutoff = weakLinkRSSICutoff
|
||||||
|
self.initialDynamicRSSIThreshold = dynamicRSSIThreshold
|
||||||
|
self.dynamicRSSIThreshold = dynamicRSSIThreshold
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleDiscovery(
|
||||||
|
_ candidate: BLEConnectionCandidate<Peripheral>,
|
||||||
|
connectedOrConnectingCount: Int,
|
||||||
|
existingState: BLEExistingConnectionState?,
|
||||||
|
peripheralState: BLEPeripheralConnectionState,
|
||||||
|
now: Date
|
||||||
|
) -> BLEDiscoveryDecision {
|
||||||
|
guard candidate.isConnectable else { return .ignore }
|
||||||
|
|
||||||
|
if candidate.rssi <= dynamicRSSIThreshold {
|
||||||
|
enqueue(candidate)
|
||||||
|
return .queued
|
||||||
|
}
|
||||||
|
|
||||||
|
if connectedOrConnectingCount >= maxCentralLinks {
|
||||||
|
enqueue(candidate)
|
||||||
|
return .queued
|
||||||
|
}
|
||||||
|
|
||||||
|
if let retryDelay = rateLimitRetryDelay(now: now) {
|
||||||
|
enqueue(candidate)
|
||||||
|
return .scheduleRetry(after: retryDelay)
|
||||||
|
}
|
||||||
|
|
||||||
|
if let existingState {
|
||||||
|
if existingState.isConnected || existingState.isConnecting {
|
||||||
|
return .ignore
|
||||||
|
}
|
||||||
|
|
||||||
|
if let lastAttempt = existingState.lastConnectionAttempt,
|
||||||
|
now.timeIntervalSince(lastAttempt) < 2.0 {
|
||||||
|
return .ignore
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let lastTimeout = recentConnectTimeouts[candidate.peripheralID],
|
||||||
|
now.timeIntervalSince(lastTimeout) < TransportConfig.bleTimeoutDiscoveryIgnoreSeconds {
|
||||||
|
return .ignore
|
||||||
|
}
|
||||||
|
|
||||||
|
if let lastDisconnect = recentDisconnects[candidate.peripheralID],
|
||||||
|
now.timeIntervalSince(lastDisconnect) < TransportConfig.bleDisconnectDiscoveryIgnoreSeconds {
|
||||||
|
return .ignore
|
||||||
|
}
|
||||||
|
|
||||||
|
switch peripheralState {
|
||||||
|
case .disconnected:
|
||||||
|
return .connectNow
|
||||||
|
case .connecting, .connected:
|
||||||
|
return .cancelStaleConnection
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func enqueue(_ candidate: BLEConnectionCandidate<Peripheral>) {
|
||||||
|
if let existingIndex = candidates.firstIndex(where: { $0.peripheralID == candidate.peripheralID }) {
|
||||||
|
candidates[existingIndex] = candidate
|
||||||
|
} else {
|
||||||
|
candidates.append(candidate)
|
||||||
|
}
|
||||||
|
|
||||||
|
candidates.sort {
|
||||||
|
if $0.rssi != $1.rssi { return $0.rssi > $1.rssi }
|
||||||
|
return $0.discoveredAt < $1.discoveredAt
|
||||||
|
}
|
||||||
|
if candidates.count > candidateCap {
|
||||||
|
candidates.removeLast(candidates.count - candidateCap)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func nextCandidate(
|
||||||
|
connectedOrConnectingCount: Int,
|
||||||
|
isAlreadyConnectingOrConnected: (String) -> Bool,
|
||||||
|
now: Date
|
||||||
|
) -> BLEConnectionQueueDecision<Peripheral> {
|
||||||
|
guard connectedOrConnectingCount < maxCentralLinks else { return .none }
|
||||||
|
|
||||||
|
if let retryDelay = rateLimitRetryDelay(now: now) {
|
||||||
|
return .retryAfter(retryDelay)
|
||||||
|
}
|
||||||
|
|
||||||
|
while !candidates.isEmpty {
|
||||||
|
candidates.sort { score($0, now: now) > score($1, now: now) }
|
||||||
|
let candidate = candidates.removeFirst()
|
||||||
|
guard candidate.isConnectable else { continue }
|
||||||
|
|
||||||
|
if let delay = weakLinkRetryDelay(for: candidate, now: now) {
|
||||||
|
enqueue(candidate)
|
||||||
|
return .retryAfter(delay)
|
||||||
|
}
|
||||||
|
|
||||||
|
if let delay = disconnectSettleDelay(for: candidate, now: now) {
|
||||||
|
enqueue(candidate)
|
||||||
|
return .retryAfter(delay)
|
||||||
|
}
|
||||||
|
|
||||||
|
if isAlreadyConnectingOrConnected(candidate.peripheralID) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
return .connect(candidate)
|
||||||
|
}
|
||||||
|
|
||||||
|
return .none
|
||||||
|
}
|
||||||
|
|
||||||
|
func recordConnectionAttempt(at now: Date) {
|
||||||
|
lastGlobalConnectAttempt = now
|
||||||
|
}
|
||||||
|
|
||||||
|
func recordConnectionSuccess(peripheralID: String) {
|
||||||
|
failureCounts[peripheralID] = 0
|
||||||
|
recentConnectTimeouts.removeValue(forKey: peripheralID)
|
||||||
|
recentDisconnects.removeValue(forKey: peripheralID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func recordConnectionFailure(peripheralID: String) {
|
||||||
|
failureCounts[peripheralID, default: 0] += 1
|
||||||
|
}
|
||||||
|
|
||||||
|
func recordDisconnectError(peripheralID: String, at now: Date) {
|
||||||
|
recentDisconnects[peripheralID] = now
|
||||||
|
}
|
||||||
|
|
||||||
|
func recordConnectionTimeout(peripheralID: String, at now: Date) {
|
||||||
|
recentConnectTimeouts[peripheralID] = now
|
||||||
|
recordConnectionFailure(peripheralID: peripheralID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func pruneConnectionTimeouts(before cutoff: Date) {
|
||||||
|
recentConnectTimeouts = recentConnectTimeouts.filter { $0.value >= cutoff }
|
||||||
|
recentDisconnects = recentDisconnects.filter { $0.value >= cutoff }
|
||||||
|
}
|
||||||
|
|
||||||
|
func reset() {
|
||||||
|
lastGlobalConnectAttempt = .distantPast
|
||||||
|
candidates.removeAll()
|
||||||
|
failureCounts.removeAll()
|
||||||
|
recentConnectTimeouts.removeAll()
|
||||||
|
recentDisconnects.removeAll()
|
||||||
|
lastIsolatedAt = nil
|
||||||
|
dynamicRSSIThreshold = initialDynamicRSSIThreshold
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
func updateRSSIThreshold(
|
||||||
|
connectedCount: Int,
|
||||||
|
connectedOrConnectingLinkCount: Int,
|
||||||
|
now: Date
|
||||||
|
) -> Int {
|
||||||
|
if connectedCount == 0 {
|
||||||
|
if lastIsolatedAt == nil { lastIsolatedAt = now }
|
||||||
|
let isolatedAt = lastIsolatedAt ?? now
|
||||||
|
let elapsed = now.timeIntervalSince(isolatedAt)
|
||||||
|
dynamicRSSIThreshold = elapsed > TransportConfig.bleIsolationRelaxThresholdSeconds
|
||||||
|
? TransportConfig.bleRSSIIsolatedRelaxed
|
||||||
|
: TransportConfig.bleRSSIIsolatedBase
|
||||||
|
return dynamicRSSIThreshold
|
||||||
|
}
|
||||||
|
|
||||||
|
lastIsolatedAt = nil
|
||||||
|
// Flaky links are handled per-peripheral (weak-link cooldown, discovery
|
||||||
|
// ignore window, score bias) — never globally, so one flaky distant peer
|
||||||
|
// can't blind us to every other edge-of-range peer.
|
||||||
|
var threshold = TransportConfig.bleDynamicRSSIThresholdDefault
|
||||||
|
if connectedOrConnectingLinkCount >= maxCentralLinks || candidates.count >= candidateCap {
|
||||||
|
threshold = TransportConfig.bleRSSIConnectedThreshold
|
||||||
|
}
|
||||||
|
|
||||||
|
dynamicRSSIThreshold = threshold
|
||||||
|
return threshold
|
||||||
|
}
|
||||||
|
|
||||||
|
private func rateLimitRetryDelay(now: Date) -> TimeInterval? {
|
||||||
|
let elapsed = now.timeIntervalSince(lastGlobalConnectAttempt)
|
||||||
|
guard elapsed < connectRateLimitInterval else { return nil }
|
||||||
|
return connectRateLimitInterval - elapsed + 0.05
|
||||||
|
}
|
||||||
|
|
||||||
|
private func weakLinkRetryDelay(
|
||||||
|
for candidate: BLEConnectionCandidate<Peripheral>,
|
||||||
|
now: Date
|
||||||
|
) -> TimeInterval? {
|
||||||
|
guard let lastTimeout = recentConnectTimeouts[candidate.peripheralID] else { return nil }
|
||||||
|
let elapsed = now.timeIntervalSince(lastTimeout)
|
||||||
|
guard elapsed < weakLinkCooldownSeconds && candidate.rssi <= weakLinkRSSICutoff else { return nil }
|
||||||
|
let remaining = weakLinkCooldownSeconds - elapsed
|
||||||
|
return min(max(2.0, remaining), 15.0)
|
||||||
|
}
|
||||||
|
|
||||||
|
// The disconnect settle window must hold on the queue path too: a stale
|
||||||
|
// candidate enqueued while the peripheral was still connected would
|
||||||
|
// otherwise reconnect immediately via the post-disconnect queue drain,
|
||||||
|
// bypassing the window and recreating reconnect/cancel thrash.
|
||||||
|
private func disconnectSettleDelay(
|
||||||
|
for candidate: BLEConnectionCandidate<Peripheral>,
|
||||||
|
now: Date
|
||||||
|
) -> TimeInterval? {
|
||||||
|
guard let lastDisconnect = recentDisconnects[candidate.peripheralID] else { return nil }
|
||||||
|
let remaining = TransportConfig.bleDisconnectDiscoveryIgnoreSeconds - now.timeIntervalSince(lastDisconnect)
|
||||||
|
guard remaining > 0 else { return nil }
|
||||||
|
return remaining + 0.05
|
||||||
|
}
|
||||||
|
|
||||||
|
private func score(_ candidate: BLEConnectionCandidate<Peripheral>, now: Date) -> Int {
|
||||||
|
let failures = failureCounts[candidate.peripheralID] ?? 0
|
||||||
|
let penalty = min(20, 1 << min(4, failures))
|
||||||
|
let timeoutBias = recentConnectTimeouts[candidate.peripheralID].map {
|
||||||
|
now.timeIntervalSince($0) < 60 ? 10 : 0
|
||||||
|
} ?? 0
|
||||||
|
let base = (candidate.isConnectable ? 1000 : 0) + (candidate.rssi + 100) * 2
|
||||||
|
let recency = -Int(now.timeIntervalSince(candidate.discoveredAt) * 10)
|
||||||
|
return base + recency - penalty - timeoutBias
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,71 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEDirectedRelaySpoolEntry {
|
||||||
|
let recipient: PeerID
|
||||||
|
let packet: BitchatPacket
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEDirectedRelaySpool {
|
||||||
|
private struct StoredPacket {
|
||||||
|
let packet: BitchatPacket
|
||||||
|
let enqueuedAt: Date
|
||||||
|
}
|
||||||
|
|
||||||
|
private var packetsByRecipient: [PeerID: [String: StoredPacket]] = [:]
|
||||||
|
|
||||||
|
var isEmpty: Bool {
|
||||||
|
packetsByRecipient.isEmpty
|
||||||
|
}
|
||||||
|
|
||||||
|
var count: Int {
|
||||||
|
packetsByRecipient.values.reduce(0) { $0 + $1.count }
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
mutating func enqueue(
|
||||||
|
packet: BitchatPacket,
|
||||||
|
recipient: PeerID,
|
||||||
|
messageID: String,
|
||||||
|
enqueuedAt: Date
|
||||||
|
) -> Bool {
|
||||||
|
var packets = packetsByRecipient[recipient] ?? [:]
|
||||||
|
guard packets[messageID] == nil else {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
packets[messageID] = StoredPacket(packet: packet, enqueuedAt: enqueuedAt)
|
||||||
|
packetsByRecipient[recipient] = packets
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func drainUnexpired(now: Date, window: TimeInterval) -> [BLEDirectedRelaySpoolEntry] {
|
||||||
|
var entries: [BLEDirectedRelaySpoolEntry] = []
|
||||||
|
|
||||||
|
for (recipient, packets) in packetsByRecipient {
|
||||||
|
for stored in packets.values where now.timeIntervalSince(stored.enqueuedAt) <= window {
|
||||||
|
entries.append(BLEDirectedRelaySpoolEntry(recipient: recipient, packet: stored.packet))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
packetsByRecipient.removeAll()
|
||||||
|
return entries
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func pruneExpired(now: Date, window: TimeInterval) {
|
||||||
|
guard !packetsByRecipient.isEmpty else { return }
|
||||||
|
|
||||||
|
var pruned: [PeerID: [String: StoredPacket]] = [:]
|
||||||
|
for (recipient, packets) in packetsByRecipient {
|
||||||
|
let freshPackets = packets.filter { now.timeIntervalSince($0.value.enqueuedAt) <= window }
|
||||||
|
if !freshPackets.isEmpty {
|
||||||
|
pruned[recipient] = freshPackets
|
||||||
|
}
|
||||||
|
}
|
||||||
|
packetsByRecipient = pruned
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
packetsByRecipient.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,152 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import CryptoKit
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEFanoutSelection: Equatable {
|
||||||
|
let peripheralIDs: Set<String>
|
||||||
|
let centralIDs: Set<String>
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEFanoutSelector {
|
||||||
|
static func selectLinks(
|
||||||
|
peripheralIDs: [String],
|
||||||
|
centralIDs: [String],
|
||||||
|
ingressLink: BLEIngressLinkID?,
|
||||||
|
excludedLinks: Set<BLEIngressLinkID> = [],
|
||||||
|
peripheralPeerBindings: [String: PeerID] = [:],
|
||||||
|
centralPeerBindings: [String: PeerID] = [:],
|
||||||
|
directedPeerHint: PeerID?,
|
||||||
|
packetType: UInt8,
|
||||||
|
messageID: String
|
||||||
|
) -> BLEFanoutSelection {
|
||||||
|
let allowed = collapseDuplicateLinksPerPeer(
|
||||||
|
allowedLinks(
|
||||||
|
peripheralIDs: peripheralIDs,
|
||||||
|
centralIDs: centralIDs,
|
||||||
|
ingressLink: ingressLink,
|
||||||
|
excludedLinks: excludedLinks
|
||||||
|
),
|
||||||
|
peripheralPeerBindings: peripheralPeerBindings,
|
||||||
|
centralPeerBindings: centralPeerBindings
|
||||||
|
)
|
||||||
|
|
||||||
|
guard shouldSubset(packetType: packetType, directedPeerHint: directedPeerHint) else {
|
||||||
|
return BLEFanoutSelection(
|
||||||
|
peripheralIDs: Set(allowed.peripheralIDs),
|
||||||
|
centralIDs: Set(allowed.centralIDs)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
return BLEFanoutSelection(
|
||||||
|
peripheralIDs: deterministicSubset(
|
||||||
|
ids: allowed.peripheralIDs,
|
||||||
|
k: subsetSize(for: allowed.peripheralIDs.count),
|
||||||
|
seed: messageID
|
||||||
|
),
|
||||||
|
centralIDs: deterministicSubset(
|
||||||
|
ids: allowed.centralIDs,
|
||||||
|
k: subsetSize(for: allowed.centralIDs.count),
|
||||||
|
seed: messageID
|
||||||
|
)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func allowedLinks(
|
||||||
|
peripheralIDs: [String],
|
||||||
|
centralIDs: [String],
|
||||||
|
ingressLink: BLEIngressLinkID?,
|
||||||
|
excludedLinks: Set<BLEIngressLinkID>
|
||||||
|
) -> (peripheralIDs: [String], centralIDs: [String]) {
|
||||||
|
var allowedPeripheralIDs = peripheralIDs
|
||||||
|
var allowedCentralIDs = centralIDs
|
||||||
|
var blockedLinks = excludedLinks
|
||||||
|
|
||||||
|
if let ingressLink {
|
||||||
|
blockedLinks.insert(ingressLink)
|
||||||
|
}
|
||||||
|
|
||||||
|
allowedPeripheralIDs.removeAll { blockedLinks.contains(.peripheral($0)) }
|
||||||
|
allowedCentralIDs.removeAll { blockedLinks.contains(.central($0)) }
|
||||||
|
|
||||||
|
return (allowedPeripheralIDs, allowedCentralIDs)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Dual-role pairs hold two live links (we-as-central writing to their
|
||||||
|
// peripheral, and they-as-central subscribed to ours). Sending the same
|
||||||
|
// packet down both doubles airtime for nothing — the receiver's assembler
|
||||||
|
// and deduplicator just discard the copy. Keep one link per bound peer,
|
||||||
|
// preferring the peripheral (write) side: it has per-link flow control
|
||||||
|
// via canSendWriteWithoutResponse, while notifications share the
|
||||||
|
// peripheral manager's update queue across all centrals. Links with no
|
||||||
|
// bound peer yet (pre-announce) pass through untouched.
|
||||||
|
private static func collapseDuplicateLinksPerPeer(
|
||||||
|
_ links: (peripheralIDs: [String], centralIDs: [String]),
|
||||||
|
peripheralPeerBindings: [String: PeerID],
|
||||||
|
centralPeerBindings: [String: PeerID]
|
||||||
|
) -> (peripheralIDs: [String], centralIDs: [String]) {
|
||||||
|
guard !peripheralPeerBindings.isEmpty || !centralPeerBindings.isEmpty else {
|
||||||
|
return links
|
||||||
|
}
|
||||||
|
|
||||||
|
var seenPeers = Set<PeerID>()
|
||||||
|
var keptPeripheralIDs: [String] = []
|
||||||
|
for id in links.peripheralIDs {
|
||||||
|
if let peer = peripheralPeerBindings[id], !seenPeers.insert(peer).inserted {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
keptPeripheralIDs.append(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
var keptCentralIDs: [String] = []
|
||||||
|
for id in links.centralIDs {
|
||||||
|
if let peer = centralPeerBindings[id], !seenPeers.insert(peer).inserted {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
keptCentralIDs.append(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
return (keptPeripheralIDs, keptCentralIDs)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func shouldSubset(packetType: UInt8, directedPeerHint: PeerID?) -> Bool {
|
||||||
|
directedPeerHint == nil
|
||||||
|
&& packetType != MessageType.fragment.rawValue
|
||||||
|
&& packetType != MessageType.announce.rawValue
|
||||||
|
&& packetType != MessageType.requestSync.rawValue
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func subsetSize(for count: Int) -> Int {
|
||||||
|
guard count > 0 else { return 0 }
|
||||||
|
if count <= 2 { return count }
|
||||||
|
|
||||||
|
var value = count - 1
|
||||||
|
var bits = 0
|
||||||
|
while value > 0 {
|
||||||
|
value >>= 1
|
||||||
|
bits += 1
|
||||||
|
}
|
||||||
|
return min(count, max(1, bits + 1))
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func deterministicSubset(ids: [String], k: Int, seed: String) -> Set<String> {
|
||||||
|
guard k > 0 && ids.count > k else { return Set(ids) }
|
||||||
|
|
||||||
|
var scored: [(score: [UInt8], id: String)] = []
|
||||||
|
for id in ids {
|
||||||
|
let data = (seed + "::" + id).data(using: .utf8) ?? Data()
|
||||||
|
let digest = Array(SHA256.hash(data: data))
|
||||||
|
scored.append((digest, id))
|
||||||
|
}
|
||||||
|
|
||||||
|
scored.sort { lhs, rhs in
|
||||||
|
for index in 0..<min(lhs.score.count, rhs.score.count) {
|
||||||
|
if lhs.score[index] != rhs.score[index] {
|
||||||
|
return lhs.score[index] < rhs.score[index]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return lhs.id < rhs.id
|
||||||
|
}
|
||||||
|
|
||||||
|
return Set(scored.prefix(k).map(\.id))
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,123 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import BitLogger
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
/// Narrow environment for `BLEFileTransferHandler`.
|
||||||
|
///
|
||||||
|
/// All queue hops (collections registry reads/writes, main-actor UI
|
||||||
|
/// notification) live inside the closures supplied by `BLEService`, keeping
|
||||||
|
/// the handler queue-agnostic and synchronously testable.
|
||||||
|
struct BLEFileTransferHandlerEnvironment {
|
||||||
|
/// Local peer identity at the time the transfer is handled.
|
||||||
|
let localPeerID: () -> PeerID
|
||||||
|
/// Local nickname used for sender resolution and collision checks.
|
||||||
|
let localNickname: () -> String
|
||||||
|
/// Snapshot of known peers keyed by ID (registry read).
|
||||||
|
let peersSnapshot: () -> [PeerID: BLEPeerInfo]
|
||||||
|
/// Resolves a display name from a verified packet signature for peers missing from the registry.
|
||||||
|
let signedSenderDisplayName: (_ packet: BitchatPacket, _ peerID: PeerID) -> String?
|
||||||
|
/// Tracks the broadcast file packet for gossip sync.
|
||||||
|
let trackPacketSeen: (BitchatPacket) -> Void
|
||||||
|
/// Enforces the incoming-media storage quota before saving (BCH-01-002).
|
||||||
|
let enforceStorageQuota: (_ reservingBytes: Int) -> Void
|
||||||
|
/// Persists the validated file to the incoming-media store; returns the destination URL.
|
||||||
|
let saveIncomingFile: (
|
||||||
|
_ data: Data,
|
||||||
|
_ preferredName: String?,
|
||||||
|
_ subdirectory: String,
|
||||||
|
_ fallbackExtension: String?,
|
||||||
|
_ defaultPrefix: String
|
||||||
|
) -> URL?
|
||||||
|
/// Updates the registry last-seen timestamp for the peer (async barrier write).
|
||||||
|
let updatePeerLastSeen: (PeerID) -> Void
|
||||||
|
/// Delivers `.messageReceived` to the UI as one main-actor hop.
|
||||||
|
let deliverMessage: (BitchatMessage) -> Void
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Orchestrates inbound file transfers: self-echo policy, sender display-name
|
||||||
|
/// resolution, delivery planning, payload validation, quota-checked storage,
|
||||||
|
/// and UI delivery.
|
||||||
|
final class BLEFileTransferHandler {
|
||||||
|
private let environment: BLEFileTransferHandlerEnvironment
|
||||||
|
|
||||||
|
init(environment: BLEFileTransferHandlerEnvironment) {
|
||||||
|
self.environment = environment
|
||||||
|
}
|
||||||
|
|
||||||
|
func handle(_ packet: BitchatPacket, from peerID: PeerID) {
|
||||||
|
let env = environment
|
||||||
|
if BLEFileTransferPolicy.isSelfEcho(packet: packet, from: peerID, localPeerID: env.localPeerID()) { return }
|
||||||
|
|
||||||
|
let peersSnapshot = env.peersSnapshot()
|
||||||
|
guard let senderNickname = BLEPeerSenderDisplayName.resolveKnownPeer(
|
||||||
|
peerID: peerID,
|
||||||
|
localPeerID: env.localPeerID(),
|
||||||
|
localNickname: env.localNickname(),
|
||||||
|
peers: peersSnapshot,
|
||||||
|
allowConnectedUnverified: true
|
||||||
|
) ?? env.signedSenderDisplayName(packet, peerID) else {
|
||||||
|
SecureLogger.warning("🚫 Dropping file transfer from unverified or unknown peer \(peerID.id.prefix(8))…", category: .security)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
guard let deliveryPlan = BLEFileTransferPolicy.deliveryPlan(packet: packet, localPeerID: env.localPeerID()) else {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if deliveryPlan.shouldTrackForSync {
|
||||||
|
env.trackPacketSeen(packet)
|
||||||
|
}
|
||||||
|
|
||||||
|
let filePacket: BitchatFilePacket
|
||||||
|
let mime: MimeType
|
||||||
|
switch BLEIncomingFileValidator.validate(payload: packet.payload) {
|
||||||
|
case .success(let acceptance):
|
||||||
|
filePacket = acceptance.filePacket
|
||||||
|
mime = acceptance.mime
|
||||||
|
case .failure(.malformedPayload):
|
||||||
|
SecureLogger.error("❌ Failed to decode file transfer payload", category: .session)
|
||||||
|
return
|
||||||
|
case .failure(.payloadTooLarge(let bytes)):
|
||||||
|
SecureLogger.warning("🚫 Dropping file transfer exceeding size cap (\(bytes) bytes)", category: .security)
|
||||||
|
return
|
||||||
|
case .failure(.unsupportedMime(let mimeType, let bytes)):
|
||||||
|
SecureLogger.warning("🚫 MIME REJECT: '\(mimeType ?? "<empty>")' not supported. Size=\(bytes)b from \(peerID.id.prefix(8))...", category: .security)
|
||||||
|
return
|
||||||
|
case .failure(.magicMismatch(let mime, let bytes, let prefixHex)):
|
||||||
|
SecureLogger.warning("🚫 MAGIC REJECT: MIME='\(mime)' size=\(bytes)b prefix=[\(prefixHex)] from \(peerID.id.prefix(8))...", category: .security)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// BCH-01-002: Enforce storage quota before saving
|
||||||
|
env.enforceStorageQuota(filePacket.content.count)
|
||||||
|
|
||||||
|
guard let destination = env.saveIncomingFile(
|
||||||
|
filePacket.content,
|
||||||
|
filePacket.fileName,
|
||||||
|
"\(mime.category.mediaDir)/incoming",
|
||||||
|
mime.defaultExtension,
|
||||||
|
mime.category.rawValue
|
||||||
|
) else {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if deliveryPlan.isPrivateMessage {
|
||||||
|
env.updatePeerLastSeen(peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
let ts = Date(timeIntervalSince1970: Double(packet.timestamp) / 1000)
|
||||||
|
let message = BitchatMessage(
|
||||||
|
sender: senderNickname,
|
||||||
|
content: "\(mime.category.messagePrefix)\(destination.lastPathComponent)",
|
||||||
|
timestamp: ts,
|
||||||
|
isRelay: false,
|
||||||
|
originalSender: nil,
|
||||||
|
isPrivate: deliveryPlan.isPrivateMessage,
|
||||||
|
recipientNickname: nil,
|
||||||
|
senderPeerID: peerID
|
||||||
|
)
|
||||||
|
|
||||||
|
SecureLogger.debug("📁 Stored incoming media from \(peerID.id.prefix(8))… -> \(destination.lastPathComponent)", category: .session)
|
||||||
|
|
||||||
|
env.deliverMessage(message)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,71 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEFileTransferDeliveryPlan: Equatable {
|
||||||
|
let isPrivateMessage: Bool
|
||||||
|
let shouldTrackForSync: Bool
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEFileTransferPolicy {
|
||||||
|
static func isSelfEcho(packet: BitchatPacket, from peerID: PeerID, localPeerID: PeerID) -> Bool {
|
||||||
|
peerID == localPeerID && packet.ttl != 0
|
||||||
|
}
|
||||||
|
|
||||||
|
static func deliveryPlan(packet: BitchatPacket, localPeerID: PeerID) -> BLEFileTransferDeliveryPlan? {
|
||||||
|
guard let recipientID = packet.recipientID else {
|
||||||
|
return BLEFileTransferDeliveryPlan(isPrivateMessage: false, shouldTrackForSync: true)
|
||||||
|
}
|
||||||
|
|
||||||
|
let isBroadcast = recipientID.allSatisfy { $0 == 0xFF }
|
||||||
|
if isBroadcast {
|
||||||
|
return BLEFileTransferDeliveryPlan(isPrivateMessage: false, shouldTrackForSync: true)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard PeerID(hexData: recipientID) == localPeerID else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return BLEFileTransferDeliveryPlan(isPrivateMessage: true, shouldTrackForSync: false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEIncomingFileAcceptance {
|
||||||
|
let filePacket: BitchatFilePacket
|
||||||
|
let mime: MimeType
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEIncomingFileRejection: Error, Equatable {
|
||||||
|
case malformedPayload
|
||||||
|
case payloadTooLarge(bytes: Int)
|
||||||
|
case unsupportedMime(mimeType: String?, bytes: Int)
|
||||||
|
case magicMismatch(mime: MimeType, bytes: Int, prefixHex: String)
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEIncomingFileValidator {
|
||||||
|
static func validate(payload: Data) -> Result<BLEIncomingFileAcceptance, BLEIncomingFileRejection> {
|
||||||
|
guard let filePacket = BitchatFilePacket.decode(payload) else {
|
||||||
|
return .failure(.malformedPayload)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard FileTransferLimits.isValidPayload(filePacket.content.count) else {
|
||||||
|
return .failure(.payloadTooLarge(bytes: filePacket.content.count))
|
||||||
|
}
|
||||||
|
|
||||||
|
guard let mime = MimeType(filePacket.mimeType), mime.isAllowed else {
|
||||||
|
return .failure(.unsupportedMime(
|
||||||
|
mimeType: filePacket.mimeType,
|
||||||
|
bytes: filePacket.content.count
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
guard mime.matches(data: filePacket.content) else {
|
||||||
|
return .failure(.magicMismatch(
|
||||||
|
mime: mime,
|
||||||
|
bytes: filePacket.content.count,
|
||||||
|
prefixHex: filePacket.content.prefix(20).map { String(format: "%02x", $0) }.joined(separator: " ")
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
return .success(BLEIncomingFileAcceptance(filePacket: filePacket, mime: mime))
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,153 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEFragmentKey: Hashable, Equatable {
|
||||||
|
let sender: UInt64
|
||||||
|
let id: UInt64
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEFragmentHeader: Equatable {
|
||||||
|
let key: BLEFragmentKey
|
||||||
|
let index: Int
|
||||||
|
let total: Int
|
||||||
|
let originalType: UInt8
|
||||||
|
let fragmentData: Data
|
||||||
|
let isBroadcastFragment: Bool
|
||||||
|
|
||||||
|
var idLogString: String {
|
||||||
|
String(format: "%016llx", key.id)
|
||||||
|
}
|
||||||
|
|
||||||
|
init?(packet: BitchatPacket) {
|
||||||
|
// Minimum header: 8 bytes ID + 2 index + 2 total + 1 type.
|
||||||
|
guard packet.payload.count >= 13 else { return nil }
|
||||||
|
|
||||||
|
var senderU64: UInt64 = 0
|
||||||
|
for byte in packet.senderID.prefix(8) {
|
||||||
|
senderU64 = (senderU64 << 8) | UInt64(byte)
|
||||||
|
}
|
||||||
|
|
||||||
|
var fragmentU64: UInt64 = 0
|
||||||
|
for byte in packet.payload.prefix(8) {
|
||||||
|
fragmentU64 = (fragmentU64 << 8) | UInt64(byte)
|
||||||
|
}
|
||||||
|
|
||||||
|
let index = Int((UInt16(packet.payload[8]) << 8) | UInt16(packet.payload[9]))
|
||||||
|
let total = Int((UInt16(packet.payload[10]) << 8) | UInt16(packet.payload[11]))
|
||||||
|
|
||||||
|
guard total > 0 && total <= 10_000 && index >= 0 && index < total else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
let isBroadcastFragment: Bool = {
|
||||||
|
guard let recipient = packet.recipientID else { return true }
|
||||||
|
return recipient.count == 8 && recipient.allSatisfy { $0 == 0xFF }
|
||||||
|
}()
|
||||||
|
|
||||||
|
self.key = BLEFragmentKey(sender: senderU64, id: fragmentU64)
|
||||||
|
self.index = index
|
||||||
|
self.total = total
|
||||||
|
self.originalType = packet.payload[12]
|
||||||
|
self.fragmentData = Data(packet.payload.suffix(from: 13))
|
||||||
|
self.isBroadcastFragment = isBroadcastFragment
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEFragmentAssemblyBuffer {
|
||||||
|
enum AppendResult: Equatable {
|
||||||
|
case stored(header: BLEFragmentHeader, started: Bool)
|
||||||
|
case complete(header: BLEFragmentHeader, reassembledData: Data, started: Bool)
|
||||||
|
case oversized(header: BLEFragmentHeader, projectedSize: Int, limit: Int, started: Bool)
|
||||||
|
}
|
||||||
|
|
||||||
|
private struct Metadata {
|
||||||
|
let type: UInt8
|
||||||
|
let total: Int
|
||||||
|
let timestamp: Date
|
||||||
|
}
|
||||||
|
|
||||||
|
private var fragmentsByKey: [BLEFragmentKey: [Int: Data]] = [:]
|
||||||
|
private var metadataByKey: [BLEFragmentKey: Metadata] = [:]
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
fragmentsByKey.removeAll()
|
||||||
|
metadataByKey.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
mutating func removeExpired(before cutoff: Date) -> Int {
|
||||||
|
let expiredKeys = metadataByKey
|
||||||
|
.filter { $0.value.timestamp < cutoff }
|
||||||
|
.map(\.key)
|
||||||
|
|
||||||
|
for key in expiredKeys {
|
||||||
|
fragmentsByKey.removeValue(forKey: key)
|
||||||
|
metadataByKey.removeValue(forKey: key)
|
||||||
|
}
|
||||||
|
|
||||||
|
return expiredKeys.count
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func append(
|
||||||
|
_ header: BLEFragmentHeader,
|
||||||
|
maxInFlightAssemblies: Int,
|
||||||
|
now: Date = Date()
|
||||||
|
) -> AppendResult {
|
||||||
|
let started = startAssemblyIfNeeded(for: header, maxInFlightAssemblies: maxInFlightAssemblies, now: now)
|
||||||
|
|
||||||
|
let currentSize = fragmentsByKey[header.key]?.values.reduce(0) { $0 + $1.count } ?? 0
|
||||||
|
let limit = Self.assemblyLimit(for: header.originalType)
|
||||||
|
let projectedSize = currentSize + header.fragmentData.count
|
||||||
|
|
||||||
|
guard projectedSize <= limit else {
|
||||||
|
fragmentsByKey.removeValue(forKey: header.key)
|
||||||
|
metadataByKey.removeValue(forKey: header.key)
|
||||||
|
return .oversized(header: header, projectedSize: projectedSize, limit: limit, started: started)
|
||||||
|
}
|
||||||
|
|
||||||
|
fragmentsByKey[header.key]?[header.index] = header.fragmentData
|
||||||
|
|
||||||
|
guard let fragments = fragmentsByKey[header.key],
|
||||||
|
fragments.count == header.total else {
|
||||||
|
return .stored(header: header, started: started)
|
||||||
|
}
|
||||||
|
|
||||||
|
let reassembled = (0..<header.total).reduce(into: Data()) { data, index in
|
||||||
|
if let fragment = fragments[index] {
|
||||||
|
data.append(fragment)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fragmentsByKey.removeValue(forKey: header.key)
|
||||||
|
metadataByKey.removeValue(forKey: header.key)
|
||||||
|
|
||||||
|
return .complete(header: header, reassembledData: reassembled, started: started)
|
||||||
|
}
|
||||||
|
|
||||||
|
private mutating func startAssemblyIfNeeded(
|
||||||
|
for header: BLEFragmentHeader,
|
||||||
|
maxInFlightAssemblies: Int,
|
||||||
|
now: Date
|
||||||
|
) -> Bool {
|
||||||
|
guard fragmentsByKey[header.key] == nil else { return false }
|
||||||
|
|
||||||
|
if fragmentsByKey.count >= maxInFlightAssemblies,
|
||||||
|
let oldest = metadataByKey.min(by: { $0.value.timestamp < $1.value.timestamp })?.key {
|
||||||
|
fragmentsByKey.removeValue(forKey: oldest)
|
||||||
|
metadataByKey.removeValue(forKey: oldest)
|
||||||
|
}
|
||||||
|
|
||||||
|
fragmentsByKey[header.key] = [:]
|
||||||
|
metadataByKey[header.key] = Metadata(type: header.originalType, total: header.total, timestamp: now)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func assemblyLimit(for originalType: UInt8) -> Int {
|
||||||
|
if originalType == MessageType.fileTransfer.rawValue {
|
||||||
|
// Allow headroom for TLV metadata and binary framing overhead.
|
||||||
|
return FileTransferLimits.maxFramedFileBytes
|
||||||
|
}
|
||||||
|
|
||||||
|
return FileTransferLimits.maxPayloadBytes
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,94 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import BitLogger
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
/// Narrow environment for `BLEFragmentHandler`.
|
||||||
|
///
|
||||||
|
/// All queue hops (the message-queue entry hop and the collections barrier
|
||||||
|
/// around the assembly buffer) live on the `BLEService` side — the entry hop
|
||||||
|
/// in `BLEService.handleFragment`, the barrier inside the supplied closures —
|
||||||
|
/// keeping the handler queue-agnostic and synchronously testable.
|
||||||
|
struct BLEFragmentHandlerEnvironment {
|
||||||
|
/// Local peer identity at the time the fragment is handled.
|
||||||
|
let localPeerID: () -> PeerID
|
||||||
|
/// Tracks broadcast fragments for gossip sync.
|
||||||
|
let trackPacketSeen: (BitchatPacket) -> Void
|
||||||
|
/// Appends the fragment to the assembly buffer (collections barrier write).
|
||||||
|
let appendFragment: (BLEFragmentHeader) -> BLEFragmentAssemblyBuffer.AppendResult
|
||||||
|
/// Ingress acceptance check for the reassembled inner packet.
|
||||||
|
let isAcceptedIngressPayload: (_ packet: BitchatPacket, _ innerSender: PeerID) -> Bool
|
||||||
|
/// Re-enters the receive pipeline with the reassembled packet (TTL already zeroed).
|
||||||
|
let processReassembledPacket: (_ packet: BitchatPacket, _ from: PeerID) -> Void
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Orchestrates inbound fragments: self-fragment suppression, gossip tracking,
|
||||||
|
/// assembly-buffer appends, and reassembled-packet validation and re-injection
|
||||||
|
/// into the receive pipeline.
|
||||||
|
final class BLEFragmentHandler {
|
||||||
|
private let environment: BLEFragmentHandlerEnvironment
|
||||||
|
|
||||||
|
init(environment: BLEFragmentHandlerEnvironment) {
|
||||||
|
self.environment = environment
|
||||||
|
}
|
||||||
|
|
||||||
|
func handle(_ packet: BitchatPacket, from peerID: PeerID) {
|
||||||
|
let env = environment
|
||||||
|
// Don't process our own fragments
|
||||||
|
if peerID == env.localPeerID() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
guard let header = BLEFragmentHeader(packet: packet) else { return }
|
||||||
|
|
||||||
|
if header.isBroadcastFragment {
|
||||||
|
env.trackPacketSeen(packet)
|
||||||
|
}
|
||||||
|
|
||||||
|
let assemblyResult = env.appendFragment(header)
|
||||||
|
|
||||||
|
logFragmentAssemblyResult(assemblyResult)
|
||||||
|
|
||||||
|
guard case let .complete(completedHeader, reassembled, _) = assemblyResult else { return }
|
||||||
|
|
||||||
|
// Decode the original packet bytes we reassembled, so flags/compression are preserved
|
||||||
|
if var originalPacket = BinaryProtocol.decode(reassembled) {
|
||||||
|
|
||||||
|
// Reassembled packet validation
|
||||||
|
let innerSender = PeerID(hexData: originalPacket.senderID)
|
||||||
|
if !env.isAcceptedIngressPayload(originalPacket, innerSender) {
|
||||||
|
// Cleanup below
|
||||||
|
} else {
|
||||||
|
SecureLogger.debug("✅ Reassembled packet id=\(completedHeader.idLogString) type=\(originalPacket.type) bytes=\(reassembled.count)", category: .session)
|
||||||
|
originalPacket.ttl = 0
|
||||||
|
env.processReassembledPacket(originalPacket, peerID)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
SecureLogger.error("❌ Failed to decode reassembled packet (type=\(completedHeader.originalType), total=\(completedHeader.total))", category: .session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func logFragmentAssemblyResult(_ result: BLEFragmentAssemblyBuffer.AppendResult) {
|
||||||
|
func logStartedIfNeeded(header: BLEFragmentHeader, started: Bool) {
|
||||||
|
if started {
|
||||||
|
SecureLogger.debug("📦 Started fragment assembly id=\(header.idLogString) total=\(header.total)", category: .session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
switch result {
|
||||||
|
case let .stored(header, started):
|
||||||
|
logStartedIfNeeded(header: header, started: started)
|
||||||
|
SecureLogger.debug("📦 Fragment \(header.index + 1)/\(header.total) (len=\(header.fragmentData.count)) for id=\(header.idLogString)", category: .session)
|
||||||
|
|
||||||
|
case let .complete(header, _, started):
|
||||||
|
logStartedIfNeeded(header: header, started: started)
|
||||||
|
SecureLogger.debug("📦 Fragment \(header.index + 1)/\(header.total) (len=\(header.fragmentData.count)) for id=\(header.idLogString)", category: .session)
|
||||||
|
|
||||||
|
case let .oversized(header, projectedSize, limit, started):
|
||||||
|
logStartedIfNeeded(header: header, started: started)
|
||||||
|
SecureLogger.warning(
|
||||||
|
"🚫 Fragment assembly exceeds size limit (\(projectedSize) bytes > \(limit)), evicting. Type=\(header.originalType) Index=\(header.index)/\(header.total)",
|
||||||
|
category: .security
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,70 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEInboundWriteChunk: Equatable {
|
||||||
|
let offset: Int
|
||||||
|
let data: Data
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEInboundWriteAppendMetadata: Equatable {
|
||||||
|
let accumulatedBytes: Int
|
||||||
|
let appendedBytes: Int
|
||||||
|
let offsets: [Int]
|
||||||
|
let packetType: UInt8?
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEInboundWriteBuffer {
|
||||||
|
enum AppendResult {
|
||||||
|
case decoded(packet: BitchatPacket, metadata: BLEInboundWriteAppendMetadata)
|
||||||
|
case waiting(metadata: BLEInboundWriteAppendMetadata)
|
||||||
|
case oversized(metadata: BLEInboundWriteAppendMetadata)
|
||||||
|
}
|
||||||
|
|
||||||
|
private var buffersByCentralID: [String: Data] = [:]
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
buffersByCentralID.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func append(
|
||||||
|
chunks: [BLEInboundWriteChunk],
|
||||||
|
for centralID: String,
|
||||||
|
capBytes: Int
|
||||||
|
) -> AppendResult {
|
||||||
|
var combined = buffersByCentralID[centralID] ?? Data()
|
||||||
|
var appendedBytes = 0
|
||||||
|
var offsets: [Int] = []
|
||||||
|
|
||||||
|
for chunk in chunks where !chunk.data.isEmpty {
|
||||||
|
offsets.append(chunk.offset)
|
||||||
|
let end = chunk.offset + chunk.data.count
|
||||||
|
|
||||||
|
if combined.count < end {
|
||||||
|
combined.append(Data(repeating: 0, count: end - combined.count))
|
||||||
|
}
|
||||||
|
|
||||||
|
combined.replaceSubrange(chunk.offset..<end, with: chunk.data)
|
||||||
|
appendedBytes += chunk.data.count
|
||||||
|
}
|
||||||
|
|
||||||
|
let metadata = BLEInboundWriteAppendMetadata(
|
||||||
|
accumulatedBytes: combined.count,
|
||||||
|
appendedBytes: appendedBytes,
|
||||||
|
offsets: offsets,
|
||||||
|
packetType: combined.count >= 2 ? combined[1] : nil
|
||||||
|
)
|
||||||
|
|
||||||
|
if let packet = BinaryProtocol.decode(combined) {
|
||||||
|
buffersByCentralID.removeValue(forKey: centralID)
|
||||||
|
return .decoded(packet: packet, metadata: metadata)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard combined.count <= capBytes else {
|
||||||
|
buffersByCentralID.removeValue(forKey: centralID)
|
||||||
|
return .oversized(metadata: metadata)
|
||||||
|
}
|
||||||
|
|
||||||
|
buffersByCentralID[centralID] = combined
|
||||||
|
return .waiting(metadata: metadata)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,168 @@
|
|||||||
|
import BitLogger
|
||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEIncomingFileStore {
|
||||||
|
private static let quotaBytes: Int64 = 100 * 1024 * 1024
|
||||||
|
|
||||||
|
private let fileManager: FileManager
|
||||||
|
private let baseDirectory: URL?
|
||||||
|
private let dateProvider: () -> Date
|
||||||
|
|
||||||
|
init(fileManager: FileManager = .default, baseDirectory: URL? = nil, dateProvider: @escaping () -> Date = Date.init) {
|
||||||
|
self.fileManager = fileManager
|
||||||
|
self.baseDirectory = baseDirectory
|
||||||
|
self.dateProvider = dateProvider
|
||||||
|
}
|
||||||
|
|
||||||
|
func save(
|
||||||
|
data: Data,
|
||||||
|
preferredName: String?,
|
||||||
|
subdirectory: String,
|
||||||
|
fallbackExtension: String?,
|
||||||
|
defaultPrefix: String
|
||||||
|
) -> URL? {
|
||||||
|
do {
|
||||||
|
let base = try filesDirectory().appendingPathComponent(subdirectory, isDirectory: true)
|
||||||
|
try fileManager.createDirectory(at: base, withIntermediateDirectories: true, attributes: nil)
|
||||||
|
let sanitized = sanitizedFileName(
|
||||||
|
preferredName,
|
||||||
|
defaultName: "\(defaultPrefix)_\(Self.timestampString(from: dateProvider()))",
|
||||||
|
fallbackExtension: fallbackExtension
|
||||||
|
)
|
||||||
|
let destination = uniqueFileURL(in: base, fileName: sanitized)
|
||||||
|
try data.write(to: destination, options: .atomic)
|
||||||
|
return destination
|
||||||
|
} catch {
|
||||||
|
SecureLogger.error("❌ Failed to persist incoming media: \(error)", category: .session)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func enforceQuota(reservingBytes: Int) {
|
||||||
|
do {
|
||||||
|
let base = try filesDirectory()
|
||||||
|
let incomingDirs = [
|
||||||
|
base.appendingPathComponent("voicenotes/incoming", isDirectory: true),
|
||||||
|
base.appendingPathComponent("images/incoming", isDirectory: true),
|
||||||
|
base.appendingPathComponent("files/incoming", isDirectory: true)
|
||||||
|
]
|
||||||
|
var allFiles: [(url: URL, size: Int64, modified: Date)] = []
|
||||||
|
|
||||||
|
for dir in incomingDirs where fileManager.fileExists(atPath: dir.path) {
|
||||||
|
guard let contents = try? fileManager.contentsOfDirectory(
|
||||||
|
at: dir,
|
||||||
|
includingPropertiesForKeys: [.fileSizeKey, .contentModificationDateKey],
|
||||||
|
options: [.skipsHiddenFiles]
|
||||||
|
) else { continue }
|
||||||
|
|
||||||
|
for fileURL in contents {
|
||||||
|
guard let attrs = try? fileURL.resourceValues(forKeys: [.fileSizeKey, .contentModificationDateKey]),
|
||||||
|
let size = attrs.fileSize,
|
||||||
|
let modified = attrs.contentModificationDate else { continue }
|
||||||
|
allFiles.append((url: fileURL, size: Int64(size), modified: modified))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let currentUsage = allFiles.reduce(0) { $0 + $1.size }
|
||||||
|
let targetUsage = Self.quotaBytes - Int64(reservingBytes)
|
||||||
|
guard currentUsage > targetUsage else { return }
|
||||||
|
|
||||||
|
let needToFree = currentUsage - targetUsage
|
||||||
|
var freedSpace: Int64 = 0
|
||||||
|
for file in allFiles.sorted(by: { $0.modified < $1.modified }) {
|
||||||
|
guard freedSpace < needToFree else { break }
|
||||||
|
do {
|
||||||
|
try fileManager.removeItem(at: file.url)
|
||||||
|
freedSpace += file.size
|
||||||
|
SecureLogger.debug("🗑️ BCH-01-002: Deleted old incoming file to free space: \(file.url.lastPathComponent)", category: .security)
|
||||||
|
} catch {
|
||||||
|
SecureLogger.warning("⚠️ Failed to delete old file for quota: \(error)", category: .security)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if freedSpace > 0 {
|
||||||
|
SecureLogger.info("📊 BCH-01-002: Freed \(ByteCountFormatter.string(fromByteCount: freedSpace, countStyle: .file)) to stay within incoming files quota", category: .security)
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
SecureLogger.warning("⚠️ Could not enforce storage quota: \(error)", category: .security)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func filesDirectory() throws -> URL {
|
||||||
|
let root = try baseDirectory ?? fileManager.url(
|
||||||
|
for: .applicationSupportDirectory,
|
||||||
|
in: .userDomainMask,
|
||||||
|
appropriateFor: nil,
|
||||||
|
create: true
|
||||||
|
)
|
||||||
|
let filesDir = root.appendingPathComponent("files", isDirectory: true)
|
||||||
|
try fileManager.createDirectory(at: filesDir, withIntermediateDirectories: true, attributes: nil)
|
||||||
|
return filesDir
|
||||||
|
}
|
||||||
|
|
||||||
|
private func sanitizedFileName(_ name: String?, defaultName: String, fallbackExtension: String?) -> String {
|
||||||
|
var candidate = (name ?? "")
|
||||||
|
.replacingOccurrences(of: "\0", with: "")
|
||||||
|
.precomposedStringWithCanonicalMapping
|
||||||
|
.replacingOccurrences(of: "/", with: "_")
|
||||||
|
.replacingOccurrences(of: "\\", with: "_")
|
||||||
|
|
||||||
|
let invalid = CharacterSet(charactersIn: "<>:\"|?*\0").union(.controlCharacters)
|
||||||
|
candidate = candidate.components(separatedBy: invalid).joined(separator: "_").trimmed
|
||||||
|
if candidate.isEmpty { candidate = defaultName }
|
||||||
|
if candidate.hasPrefix(".") { candidate = "_" + candidate }
|
||||||
|
|
||||||
|
if candidate.count > 120 {
|
||||||
|
let ext = (candidate as NSString).pathExtension
|
||||||
|
let base = (candidate as NSString).deletingPathExtension
|
||||||
|
candidate = ext.isEmpty
|
||||||
|
? String(candidate.prefix(120))
|
||||||
|
: String(base.prefix(max(10, 120 - ext.count - 1))) + "." + ext
|
||||||
|
}
|
||||||
|
|
||||||
|
if let fallbackExtension, (candidate as NSString).pathExtension.isEmpty {
|
||||||
|
candidate += ".\(fallbackExtension)"
|
||||||
|
}
|
||||||
|
|
||||||
|
return candidate.isEmpty ? defaultName : candidate
|
||||||
|
}
|
||||||
|
|
||||||
|
private func uniqueFileURL(in directory: URL, fileName: String) -> URL {
|
||||||
|
let directoryPath = directory.standardizedFileURL.path
|
||||||
|
func isInsideDirectory(_ url: URL) -> Bool {
|
||||||
|
url.standardizedFileURL.path.hasPrefix(directoryPath + "/")
|
||||||
|
}
|
||||||
|
|
||||||
|
var candidate = directory.appendingPathComponent(fileName)
|
||||||
|
guard isInsideDirectory(candidate) else {
|
||||||
|
SecureLogger.warning("⚠️ Path traversal blocked: \(fileName)", category: .security)
|
||||||
|
return directory.appendingPathComponent("blocked_\(UUID().uuidString)")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !fileManager.fileExists(atPath: candidate.path) {
|
||||||
|
return candidate
|
||||||
|
}
|
||||||
|
|
||||||
|
let baseName = (fileName as NSString).deletingPathExtension
|
||||||
|
let ext = (fileName as NSString).pathExtension
|
||||||
|
for counter in 1..<100 {
|
||||||
|
let newName = ext.isEmpty ? "\(baseName) (\(counter))" : "\(baseName) (\(counter)).\(ext)"
|
||||||
|
candidate = directory.appendingPathComponent(newName)
|
||||||
|
guard isInsideDirectory(candidate) else {
|
||||||
|
return directory.appendingPathComponent("blocked_\(UUID().uuidString)")
|
||||||
|
}
|
||||||
|
if !fileManager.fileExists(atPath: candidate.path) {
|
||||||
|
return candidate
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return directory.appendingPathComponent("\(baseName)_\(UUID().uuidString).\(ext.isEmpty ? "dat" : ext)")
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func timestampString(from date: Date) -> String {
|
||||||
|
let formatter = DateFormatter()
|
||||||
|
formatter.dateFormat = "yyyyMMdd_HHmmss"
|
||||||
|
return formatter.string(from: date)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,108 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum BLEIngressLinkID: Hashable, Equatable {
|
||||||
|
case peripheral(String)
|
||||||
|
case central(String)
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEIngressPacketContext: Equatable {
|
||||||
|
let receivedFromPeerID: PeerID
|
||||||
|
let validationPeerID: PeerID
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEIngressLinkRecord: Equatable {
|
||||||
|
let link: BLEIngressLinkID
|
||||||
|
let peerID: PeerID
|
||||||
|
let timestamp: Date
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEIngressRejection: Error, Equatable {
|
||||||
|
case selfLoopback(packetType: UInt8)
|
||||||
|
case directSenderMismatch(boundPeerID: PeerID, claimedSenderID: PeerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEIngressLinkRegistry {
|
||||||
|
private var ingressByMessageID: [String: BLEIngressLinkRecord] = [:]
|
||||||
|
|
||||||
|
var isEmpty: Bool {
|
||||||
|
ingressByMessageID.isEmpty
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
ingressByMessageID.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
func record(for packet: BitchatPacket) -> BLEIngressLinkRecord? {
|
||||||
|
ingressByMessageID[Self.messageID(for: packet)]
|
||||||
|
}
|
||||||
|
|
||||||
|
func link(for packet: BitchatPacket) -> BLEIngressLinkID? {
|
||||||
|
record(for: packet)?.link
|
||||||
|
}
|
||||||
|
|
||||||
|
func peerID(for packet: BitchatPacket) -> PeerID? {
|
||||||
|
record(for: packet)?.peerID
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func recordIfNew(
|
||||||
|
_ packet: BitchatPacket,
|
||||||
|
link: BLEIngressLinkID,
|
||||||
|
peerID: PeerID,
|
||||||
|
now: Date = Date(),
|
||||||
|
lifetime: TimeInterval
|
||||||
|
) -> Bool {
|
||||||
|
let messageID = Self.messageID(for: packet)
|
||||||
|
if let existing = ingressByMessageID[messageID],
|
||||||
|
now.timeIntervalSince(existing.timestamp) <= lifetime {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
ingressByMessageID[messageID] = BLEIngressLinkRecord(link: link, peerID: peerID, timestamp: now)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func prune(before cutoff: Date) {
|
||||||
|
ingressByMessageID = ingressByMessageID.filter { $0.value.timestamp >= cutoff }
|
||||||
|
}
|
||||||
|
|
||||||
|
static func packetContext(
|
||||||
|
for packet: BitchatPacket,
|
||||||
|
claimedSenderID: PeerID,
|
||||||
|
boundPeerID: PeerID?,
|
||||||
|
localPeerID: PeerID,
|
||||||
|
directAnnounceTTL: UInt8
|
||||||
|
) -> Result<BLEIngressPacketContext, BLEIngressRejection> {
|
||||||
|
if claimedSenderID == localPeerID,
|
||||||
|
!isSelfAuthoredSyncResponse(packet) {
|
||||||
|
return .failure(.selfLoopback(packetType: packet.type))
|
||||||
|
}
|
||||||
|
|
||||||
|
if let boundPeerID,
|
||||||
|
boundPeerID != claimedSenderID,
|
||||||
|
requiresDirectSenderBinding(packet, directAnnounceTTL: directAnnounceTTL) {
|
||||||
|
return .failure(.directSenderMismatch(boundPeerID: boundPeerID, claimedSenderID: claimedSenderID))
|
||||||
|
}
|
||||||
|
|
||||||
|
let receivedFromPeerID = boundPeerID ?? claimedSenderID
|
||||||
|
let validationPeerID = packet.isRSR ? receivedFromPeerID : claimedSenderID
|
||||||
|
return .success(BLEIngressPacketContext(
|
||||||
|
receivedFromPeerID: receivedFromPeerID,
|
||||||
|
validationPeerID: validationPeerID
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
static func messageID(for packet: BitchatPacket) -> String {
|
||||||
|
let senderID = packet.senderID.hexEncodedString()
|
||||||
|
let digestPrefix = packet.payload.sha256Hash().prefix(4).hexEncodedString()
|
||||||
|
return "\(senderID)-\(packet.timestamp)-\(packet.type)-\(digestPrefix)"
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func requiresDirectSenderBinding(_ packet: BitchatPacket, directAnnounceTTL: UInt8) -> Bool {
|
||||||
|
packet.type == MessageType.announce.rawValue && packet.ttl == directAnnounceTTL
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func isSelfAuthoredSyncResponse(_ packet: BitchatPacket) -> Bool {
|
||||||
|
packet.isRSR && packet.ttl == 0
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,76 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum BLEIngressPacketGuard {
|
||||||
|
enum Rejection: Error, Equatable {
|
||||||
|
case selfLoopback(packetType: UInt8)
|
||||||
|
case directSenderMismatch(boundPeerID: PeerID, claimedSenderID: PeerID)
|
||||||
|
case invalidRSR(peerID: PeerID)
|
||||||
|
case timestampSkew(peerID: PeerID, skewMs: UInt64, maxSkewMs: UInt64)
|
||||||
|
}
|
||||||
|
|
||||||
|
static func evaluate(
|
||||||
|
packet: BitchatPacket,
|
||||||
|
claimedSenderID: PeerID,
|
||||||
|
boundPeerID: PeerID?,
|
||||||
|
localPeerID: PeerID,
|
||||||
|
directAnnounceTTL: UInt8,
|
||||||
|
nowMs: UInt64 = UInt64(Date().timeIntervalSince1970 * 1000),
|
||||||
|
maxTimestampSkewMs: UInt64 = 120_000,
|
||||||
|
isValidSyncResponse: (PeerID) -> Bool
|
||||||
|
) -> Result<BLEIngressPacketContext, Rejection> {
|
||||||
|
let contextResult = BLEIngressLinkRegistry.packetContext(
|
||||||
|
for: packet,
|
||||||
|
claimedSenderID: claimedSenderID,
|
||||||
|
boundPeerID: boundPeerID,
|
||||||
|
localPeerID: localPeerID,
|
||||||
|
directAnnounceTTL: directAnnounceTTL
|
||||||
|
)
|
||||||
|
|
||||||
|
let context: BLEIngressPacketContext
|
||||||
|
switch contextResult {
|
||||||
|
case .success(let acceptedContext):
|
||||||
|
context = acceptedContext
|
||||||
|
case .failure(.selfLoopback(let packetType)):
|
||||||
|
return .failure(.selfLoopback(packetType: packetType))
|
||||||
|
case .failure(.directSenderMismatch(let boundPeerID, let claimedSenderID)):
|
||||||
|
return .failure(.directSenderMismatch(boundPeerID: boundPeerID, claimedSenderID: claimedSenderID))
|
||||||
|
}
|
||||||
|
|
||||||
|
switch validatePayload(
|
||||||
|
packet,
|
||||||
|
from: context.validationPeerID,
|
||||||
|
nowMs: nowMs,
|
||||||
|
maxTimestampSkewMs: maxTimestampSkewMs,
|
||||||
|
isValidSyncResponse: isValidSyncResponse
|
||||||
|
) {
|
||||||
|
case .success:
|
||||||
|
return .success(context)
|
||||||
|
case .failure(let rejection):
|
||||||
|
return .failure(rejection)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static func validatePayload(
|
||||||
|
_ packet: BitchatPacket,
|
||||||
|
from peerID: PeerID,
|
||||||
|
nowMs: UInt64 = UInt64(Date().timeIntervalSince1970 * 1000),
|
||||||
|
maxTimestampSkewMs: UInt64 = 120_000,
|
||||||
|
isValidSyncResponse: (PeerID) -> Bool
|
||||||
|
) -> Result<Void, Rejection> {
|
||||||
|
if packet.isRSR {
|
||||||
|
guard isValidSyncResponse(peerID) else {
|
||||||
|
return .failure(.invalidRSR(peerID: peerID))
|
||||||
|
}
|
||||||
|
return .success(())
|
||||||
|
}
|
||||||
|
|
||||||
|
let packetTime = packet.timestamp
|
||||||
|
let skew = packetTime > nowMs ? packetTime - nowMs : nowMs - packetTime
|
||||||
|
guard skew <= maxTimestampSkewMs else {
|
||||||
|
return .failure(.timestampSkew(peerID: peerID, skewMs: skew, maxSkewMs: maxTimestampSkewMs))
|
||||||
|
}
|
||||||
|
|
||||||
|
return .success(())
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,243 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import CoreBluetooth
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEPeripheralLinkState {
|
||||||
|
let peripheral: CBPeripheral
|
||||||
|
var characteristic: CBCharacteristic?
|
||||||
|
var peerID: PeerID?
|
||||||
|
var isConnecting: Bool
|
||||||
|
var isConnected: Bool
|
||||||
|
var lastConnectionAttempt: Date?
|
||||||
|
var assembler: NotificationStreamAssembler
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEDirectLinkState: Equatable {
|
||||||
|
let hasPeripheral: Bool
|
||||||
|
let hasCentral: Bool
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLESubscribedCentralSnapshot {
|
||||||
|
let centrals: [CBCentral]
|
||||||
|
let peerIDsByCentralUUID: [String: PeerID]
|
||||||
|
|
||||||
|
func central(for peerID: PeerID) -> CBCentral? {
|
||||||
|
centrals.first { peerIDsByCentralUUID[$0.identifier.uuidString] == peerID }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Owns all BLE link state (peripheral connections we hold as central, and
|
||||||
|
/// central subscriptions we serve as peripheral). The store has no internal
|
||||||
|
/// locking: every access must happen on the single owning queue (the BLE
|
||||||
|
/// queue). Other queues must go through BLEService's `readLinkState`, which
|
||||||
|
/// hops to that queue. Call `assumeOwnership(of:)` to have debug builds trap
|
||||||
|
/// any access from the wrong queue.
|
||||||
|
final class BLELinkStateStore {
|
||||||
|
private(set) var peripherals: [String: BLEPeripheralLinkState] = [:]
|
||||||
|
private(set) var peerToPeripheralUUID: [PeerID: String] = [:]
|
||||||
|
private(set) var subscribedCentrals: [CBCentral] = []
|
||||||
|
private(set) var centralToPeerID: [String: PeerID] = [:]
|
||||||
|
|
||||||
|
#if DEBUG
|
||||||
|
private var ownerQueue: DispatchQueue?
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/// Pin the store to its owning queue. Debug-only enforcement; release
|
||||||
|
/// builds are unchanged.
|
||||||
|
func assumeOwnership(of queue: DispatchQueue) {
|
||||||
|
#if DEBUG
|
||||||
|
ownerQueue = queue
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
@inline(__always)
|
||||||
|
private func assertOwned() {
|
||||||
|
#if DEBUG
|
||||||
|
if let queue = ownerQueue {
|
||||||
|
dispatchPrecondition(condition: .onQueue(queue))
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
var peripheralStates: [BLEPeripheralLinkState] {
|
||||||
|
assertOwned()
|
||||||
|
return Array(peripherals.values)
|
||||||
|
}
|
||||||
|
|
||||||
|
var subscribedCentralSnapshot: BLESubscribedCentralSnapshot {
|
||||||
|
assertOwned()
|
||||||
|
return BLESubscribedCentralSnapshot(
|
||||||
|
centrals: subscribedCentrals,
|
||||||
|
peerIDsByCentralUUID: centralToPeerID
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
var subscribedCentralCount: Int {
|
||||||
|
assertOwned()
|
||||||
|
return subscribedCentrals.count
|
||||||
|
}
|
||||||
|
|
||||||
|
var connectedOrConnectingPeripheralCount: Int {
|
||||||
|
assertOwned()
|
||||||
|
return peripherals.values.filter { $0.isConnected || $0.isConnecting }.count
|
||||||
|
}
|
||||||
|
|
||||||
|
func state(forPeripheralID peripheralID: String) -> BLEPeripheralLinkState? {
|
||||||
|
assertOwned()
|
||||||
|
return peripherals[peripheralID]
|
||||||
|
}
|
||||||
|
|
||||||
|
func setPeripheralState(_ state: BLEPeripheralLinkState, for peripheralID: String) {
|
||||||
|
assertOwned()
|
||||||
|
peripherals[peripheralID] = state
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
func updatePeripheral(
|
||||||
|
_ peripheralID: String,
|
||||||
|
_ update: (inout BLEPeripheralLinkState) -> Void
|
||||||
|
) -> BLEPeripheralLinkState? {
|
||||||
|
assertOwned()
|
||||||
|
guard var state = peripherals[peripheralID] else { return nil }
|
||||||
|
update(&state)
|
||||||
|
peripherals[peripheralID] = state
|
||||||
|
return state
|
||||||
|
}
|
||||||
|
|
||||||
|
func beginConnecting(to peripheral: CBPeripheral, at date: Date) {
|
||||||
|
setPeripheralState(
|
||||||
|
BLEPeripheralLinkState(
|
||||||
|
peripheral: peripheral,
|
||||||
|
characteristic: nil,
|
||||||
|
peerID: nil,
|
||||||
|
isConnecting: true,
|
||||||
|
isConnected: false,
|
||||||
|
lastConnectionAttempt: date,
|
||||||
|
assembler: NotificationStreamAssembler()
|
||||||
|
),
|
||||||
|
for: peripheral.identifier.uuidString
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
func markConnected(_ peripheral: CBPeripheral) {
|
||||||
|
let peripheralID = peripheral.identifier.uuidString
|
||||||
|
if updatePeripheral(peripheralID, {
|
||||||
|
$0.isConnecting = false
|
||||||
|
$0.isConnected = true
|
||||||
|
}) == nil {
|
||||||
|
setPeripheralState(
|
||||||
|
BLEPeripheralLinkState(
|
||||||
|
peripheral: peripheral,
|
||||||
|
characteristic: nil,
|
||||||
|
peerID: nil,
|
||||||
|
isConnecting: false,
|
||||||
|
isConnected: true,
|
||||||
|
lastConnectionAttempt: nil,
|
||||||
|
assembler: NotificationStreamAssembler()
|
||||||
|
),
|
||||||
|
for: peripheralID
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func updateCharacteristic(_ characteristic: CBCharacteristic, forPeripheralID peripheralID: String) {
|
||||||
|
updatePeripheral(peripheralID) {
|
||||||
|
$0.characteristic = characteristic
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func directPeripheralState(for peerID: PeerID) -> BLEPeripheralLinkState? {
|
||||||
|
assertOwned()
|
||||||
|
return peerToPeripheralUUID[peerID].flatMap { peripherals[$0] }
|
||||||
|
}
|
||||||
|
|
||||||
|
func directLinkState(for peerID: PeerID) -> BLEDirectLinkState {
|
||||||
|
assertOwned()
|
||||||
|
let peripheralUUID = peerToPeripheralUUID[peerID]
|
||||||
|
let hasPeripheral = peripheralUUID.flatMap { peripherals[$0]?.isConnected } ?? false
|
||||||
|
let hasCentral = centralToPeerID.values.contains(peerID)
|
||||||
|
return BLEDirectLinkState(hasPeripheral: hasPeripheral, hasCentral: hasCentral)
|
||||||
|
}
|
||||||
|
|
||||||
|
func links(to peerID: PeerID?) -> Set<BLEIngressLinkID> {
|
||||||
|
assertOwned()
|
||||||
|
guard let peerID else { return [] }
|
||||||
|
|
||||||
|
var links: Set<BLEIngressLinkID> = []
|
||||||
|
if let peripheralUUID = peerToPeripheralUUID[peerID] {
|
||||||
|
links.insert(.peripheral(peripheralUUID))
|
||||||
|
}
|
||||||
|
for (centralUUID, mappedPeerID) in centralToPeerID where mappedPeerID == peerID {
|
||||||
|
links.insert(.central(centralUUID))
|
||||||
|
}
|
||||||
|
return links
|
||||||
|
}
|
||||||
|
|
||||||
|
func peerID(forPeripheralID peripheralID: String) -> PeerID? {
|
||||||
|
assertOwned()
|
||||||
|
return peripherals[peripheralID]?.peerID
|
||||||
|
}
|
||||||
|
|
||||||
|
func peerID(forCentralUUID centralUUID: String) -> PeerID? {
|
||||||
|
assertOwned()
|
||||||
|
return centralToPeerID[centralUUID]
|
||||||
|
}
|
||||||
|
|
||||||
|
func addSubscribedCentral(_ central: CBCentral) {
|
||||||
|
assertOwned()
|
||||||
|
guard !subscribedCentrals.contains(central) else { return }
|
||||||
|
subscribedCentrals.append(central)
|
||||||
|
}
|
||||||
|
|
||||||
|
func removeSubscribedCentral(_ central: CBCentral) -> PeerID? {
|
||||||
|
assertOwned()
|
||||||
|
let centralUUID = central.identifier.uuidString
|
||||||
|
subscribedCentrals.removeAll { $0.identifier == central.identifier }
|
||||||
|
return centralToPeerID.removeValue(forKey: centralUUID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func bindCentral(_ centralUUID: String, to peerID: PeerID) {
|
||||||
|
assertOwned()
|
||||||
|
centralToPeerID[centralUUID] = peerID
|
||||||
|
}
|
||||||
|
|
||||||
|
func bindPeripheral(_ peripheralUUID: String, to peerID: PeerID) {
|
||||||
|
assertOwned()
|
||||||
|
if updatePeripheral(peripheralUUID, { $0.peerID = peerID }) != nil {
|
||||||
|
peerToPeripheralUUID[peerID] = peripheralUUID
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func removePeripheral(_ peripheralID: String) -> PeerID? {
|
||||||
|
assertOwned()
|
||||||
|
let peerID = peripherals.removeValue(forKey: peripheralID)?.peerID
|
||||||
|
if let peerID {
|
||||||
|
peerToPeripheralUUID.removeValue(forKey: peerID)
|
||||||
|
}
|
||||||
|
return peerID
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearPeripherals() -> [PeerID] {
|
||||||
|
assertOwned()
|
||||||
|
let peerIDs = peripherals.compactMap { $0.value.peerID }
|
||||||
|
peripherals.removeAll()
|
||||||
|
peerToPeripheralUUID.removeAll()
|
||||||
|
return peerIDs
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearCentrals() -> [PeerID] {
|
||||||
|
assertOwned()
|
||||||
|
let peerIDs = Array(centralToPeerID.values)
|
||||||
|
subscribedCentrals.removeAll()
|
||||||
|
centralToPeerID.removeAll()
|
||||||
|
return peerIDs
|
||||||
|
}
|
||||||
|
|
||||||
|
func clearAll() {
|
||||||
|
assertOwned()
|
||||||
|
peripherals.removeAll()
|
||||||
|
peerToPeripheralUUID.removeAll()
|
||||||
|
subscribedCentrals.removeAll()
|
||||||
|
centralToPeerID.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,33 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
final class BLELogRateLimiter {
|
||||||
|
private let defaultMinimumInterval: TimeInterval
|
||||||
|
private let queue = DispatchQueue(label: "chat.bitchat.ble.log-rate-limiter")
|
||||||
|
private var lastLogTimeByKey: [String: Date] = [:]
|
||||||
|
|
||||||
|
init(defaultMinimumInterval: TimeInterval) {
|
||||||
|
self.defaultMinimumInterval = defaultMinimumInterval
|
||||||
|
}
|
||||||
|
|
||||||
|
func shouldLog(
|
||||||
|
key: String,
|
||||||
|
now: Date = Date(),
|
||||||
|
minimumInterval: TimeInterval? = nil
|
||||||
|
) -> Bool {
|
||||||
|
queue.sync {
|
||||||
|
let interval = minimumInterval ?? defaultMinimumInterval
|
||||||
|
if let lastLogTime = lastLogTimeByKey[key],
|
||||||
|
now.timeIntervalSince(lastLogTime) < interval {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
lastLogTimeByKey[key] = now
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func removeAll() {
|
||||||
|
queue.sync {
|
||||||
|
lastLogTimeByKey.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,62 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEMaintenancePlan: Equatable {
|
||||||
|
let shouldSendAnnounce: Bool
|
||||||
|
let shouldEnsureAdvertising: Bool
|
||||||
|
let shouldRunCleanup: Bool
|
||||||
|
let shouldFlushDirectedSpool: Bool
|
||||||
|
let shouldResetCounter: Bool
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEMaintenancePolicy {
|
||||||
|
static func plan(
|
||||||
|
cycle: Int,
|
||||||
|
connectedCount: Int,
|
||||||
|
peerRegistryIsEmpty: Bool,
|
||||||
|
elapsedSinceLastAnnounce: TimeInterval,
|
||||||
|
hasRecentTraffic: Bool,
|
||||||
|
connectedAnnounceJitterOffset: TimeInterval? = nil,
|
||||||
|
highDegreeThreshold: Int = TransportConfig.bleHighDegreeThreshold
|
||||||
|
) -> BLEMaintenancePlan {
|
||||||
|
BLEMaintenancePlan(
|
||||||
|
shouldSendAnnounce: shouldSendAnnounce(
|
||||||
|
connectedCount: connectedCount,
|
||||||
|
elapsedSinceLastAnnounce: elapsedSinceLastAnnounce,
|
||||||
|
hasRecentTraffic: hasRecentTraffic,
|
||||||
|
connectedAnnounceJitterOffset: connectedAnnounceJitterOffset,
|
||||||
|
highDegreeThreshold: highDegreeThreshold
|
||||||
|
),
|
||||||
|
shouldEnsureAdvertising: peerRegistryIsEmpty,
|
||||||
|
shouldRunCleanup: cycle.isMultiple(of: 3),
|
||||||
|
shouldFlushDirectedSpool: !cycle.isMultiple(of: 2),
|
||||||
|
shouldResetCounter: cycle >= 6
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
static func shouldSendAnnounce(
|
||||||
|
connectedCount: Int,
|
||||||
|
elapsedSinceLastAnnounce: TimeInterval,
|
||||||
|
hasRecentTraffic: Bool,
|
||||||
|
connectedAnnounceJitterOffset: TimeInterval? = nil,
|
||||||
|
highDegreeThreshold: Int = TransportConfig.bleHighDegreeThreshold
|
||||||
|
) -> Bool {
|
||||||
|
if hasRecentTraffic && elapsedSinceLastAnnounce >= 10.0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
guard connectedCount > 0 else {
|
||||||
|
return elapsedSinceLastAnnounce >= TransportConfig.bleAnnounceIntervalSeconds
|
||||||
|
}
|
||||||
|
|
||||||
|
let highDegree = connectedCount >= highDegreeThreshold
|
||||||
|
let base = highDegree ?
|
||||||
|
TransportConfig.bleConnectedAnnounceBaseSecondsDense :
|
||||||
|
TransportConfig.bleConnectedAnnounceBaseSecondsSparse
|
||||||
|
let jitter = highDegree ?
|
||||||
|
TransportConfig.bleConnectedAnnounceJitterDense :
|
||||||
|
TransportConfig.bleConnectedAnnounceJitterSparse
|
||||||
|
let jitterOffset = connectedAnnounceJitterOffset ?? Double.random(in: -jitter...jitter)
|
||||||
|
|
||||||
|
return elapsedSinceLastAnnounce >= base + jitterOffset
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,132 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import BitLogger
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
/// Narrow environment for `BLENoisePacketHandler`.
|
||||||
|
///
|
||||||
|
/// All queue hops (collections barrier writes, main-actor UI notification)
|
||||||
|
/// and every `noiseService.*` crypto call live inside the closures supplied by
|
||||||
|
/// `BLEService`, keeping the handler queue-agnostic and synchronously testable.
|
||||||
|
struct BLENoisePacketHandlerEnvironment {
|
||||||
|
/// Local peer identity at the time the packet is handled.
|
||||||
|
let localPeerID: () -> PeerID
|
||||||
|
/// Local peer ID bytes used as the sender of handshake responses.
|
||||||
|
let localPeerIDData: () -> Data
|
||||||
|
/// TTL value used for direct (non-relayed) packets.
|
||||||
|
let messageTTL: UInt8
|
||||||
|
/// Current time source.
|
||||||
|
let now: () -> Date
|
||||||
|
/// Processes an inbound handshake message, returning an optional response payload (crypto).
|
||||||
|
let processHandshakeMessage: (_ peerID: PeerID, _ message: Data) throws -> Data?
|
||||||
|
/// Whether any Noise session (established or pending) exists for the peer (crypto).
|
||||||
|
let hasNoiseSession: (PeerID) -> Bool
|
||||||
|
/// Initiates a fresh Noise handshake with the peer (crypto + send).
|
||||||
|
let initiateHandshake: (PeerID) -> Void
|
||||||
|
/// Broadcasts a packet on the mesh (caller is already on the message queue).
|
||||||
|
let broadcastPacket: (BitchatPacket) -> Void
|
||||||
|
/// Updates the registry last-seen timestamp for the peer (async barrier write).
|
||||||
|
let updatePeerLastSeen: (PeerID) -> Void
|
||||||
|
/// Decrypts an encrypted payload from the peer (crypto).
|
||||||
|
let decrypt: (_ payload: Data, _ peerID: PeerID) throws -> Data
|
||||||
|
/// Clears the peer's Noise session after an unrecoverable decrypt failure (crypto).
|
||||||
|
let clearSession: (PeerID) -> Void
|
||||||
|
/// Delivers `.noisePayloadReceived` to the UI as one main-actor hop.
|
||||||
|
let deliverNoisePayload: (
|
||||||
|
_ peerID: PeerID,
|
||||||
|
_ type: NoisePayloadType,
|
||||||
|
_ payload: Data,
|
||||||
|
_ timestamp: Date
|
||||||
|
) -> Void
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Orchestrates the Noise session domain for inbound packets: handshake
|
||||||
|
/// processing (with response), encrypted payload decryption and dispatch,
|
||||||
|
/// and session recovery on decrypt failure.
|
||||||
|
final class BLENoisePacketHandler {
|
||||||
|
private let environment: BLENoisePacketHandlerEnvironment
|
||||||
|
|
||||||
|
init(environment: BLENoisePacketHandlerEnvironment) {
|
||||||
|
self.environment = environment
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleHandshake(_ packet: BitchatPacket, from peerID: PeerID) {
|
||||||
|
let env = environment
|
||||||
|
// Use NoiseEncryptionService for handshake processing
|
||||||
|
if PeerID(hexData: packet.recipientID) == env.localPeerID() {
|
||||||
|
// Handshake is for us
|
||||||
|
do {
|
||||||
|
if let response = try env.processHandshakeMessage(peerID, packet.payload) {
|
||||||
|
// Send response
|
||||||
|
let responsePacket = BitchatPacket(
|
||||||
|
type: MessageType.noiseHandshake.rawValue,
|
||||||
|
senderID: env.localPeerIDData(),
|
||||||
|
recipientID: Data(hexString: peerID.id),
|
||||||
|
timestamp: UInt64(env.now().timeIntervalSince1970 * 1000),
|
||||||
|
payload: response,
|
||||||
|
signature: nil,
|
||||||
|
ttl: env.messageTTL
|
||||||
|
)
|
||||||
|
// We're on messageQueue from delegate callback
|
||||||
|
env.broadcastPacket(responsePacket)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Session establishment will trigger onPeerAuthenticated callback
|
||||||
|
// which will send any pending messages at the right time
|
||||||
|
} catch {
|
||||||
|
SecureLogger.error("Failed to process handshake: \(error)")
|
||||||
|
// Try initiating a new handshake
|
||||||
|
if !env.hasNoiseSession(peerID) {
|
||||||
|
env.initiateHandshake(peerID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleEncrypted(_ packet: BitchatPacket, from peerID: PeerID) {
|
||||||
|
let env = environment
|
||||||
|
guard let recipientID = PeerID(hexData: packet.recipientID) else {
|
||||||
|
SecureLogger.warning("⚠️ Encrypted message has no recipient ID", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if recipientID != env.localPeerID() {
|
||||||
|
SecureLogger.debug("🔐 Encrypted message not for me (for \(recipientID.id.prefix(8))…, I am \(env.localPeerID().id.prefix(8))…)", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update lastSeen for the peer we received from (important for private messages)
|
||||||
|
env.updatePeerLastSeen(peerID)
|
||||||
|
|
||||||
|
do {
|
||||||
|
let decrypted = try env.decrypt(packet.payload, peerID)
|
||||||
|
guard decrypted.count > 0 else { return }
|
||||||
|
|
||||||
|
// First byte indicates the payload type
|
||||||
|
let payloadType = decrypted[0]
|
||||||
|
let payloadData = decrypted.dropFirst()
|
||||||
|
|
||||||
|
guard let noisePayloadType = NoisePayloadType(rawValue: payloadType) else {
|
||||||
|
SecureLogger.warning("⚠️ Unknown noise payload type: \(payloadType)")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
SecureLogger.debug("🔐 Decrypted noise payload type \(noisePayloadType.description) from \(peerID.id.prefix(8))…", category: .session)
|
||||||
|
|
||||||
|
let ts = Date(timeIntervalSince1970: Double(packet.timestamp) / 1000)
|
||||||
|
env.deliverNoisePayload(peerID, noisePayloadType, Data(payloadData), ts)
|
||||||
|
} catch NoiseEncryptionError.sessionNotEstablished {
|
||||||
|
// We received an encrypted message before establishing a session with this peer.
|
||||||
|
// Trigger a handshake so future messages can be decrypted.
|
||||||
|
SecureLogger.debug("🔑 Encrypted message from \(peerID.id.prefix(8))… without session; initiating handshake")
|
||||||
|
if !env.hasNoiseSession(peerID) {
|
||||||
|
env.initiateHandshake(peerID)
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
// Decryption failed - clear the corrupted session and re-initiate handshake
|
||||||
|
// This handles cases where session state got out of sync (nonce mismatch, etc.)
|
||||||
|
SecureLogger.error("❌ Failed to decrypt message from \(peerID.id.prefix(8))…: \(error) - clearing session and re-initiating handshake")
|
||||||
|
env.clearSession(peerID)
|
||||||
|
env.initiateHandshake(peerID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,25 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum BLENoisePayloadFactory {
|
||||||
|
static func privateMessage(content: String, messageID: String) -> Data? {
|
||||||
|
guard let payload = PrivateMessagePacket(messageID: messageID, content: content).encode() else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return typedPayload(.privateMessage, payload: payload)
|
||||||
|
}
|
||||||
|
|
||||||
|
static func readReceipt(originalMessageID: String) -> Data {
|
||||||
|
typedPayload(.readReceipt, payload: Data(originalMessageID.utf8))
|
||||||
|
}
|
||||||
|
|
||||||
|
static func delivered(messageID: String) -> Data {
|
||||||
|
typedPayload(.delivered, payload: Data(messageID.utf8))
|
||||||
|
}
|
||||||
|
|
||||||
|
static func typedPayload(_ type: NoisePayloadType, payload: Data) -> Data {
|
||||||
|
var typed = Data([type.rawValue])
|
||||||
|
typed.append(payload)
|
||||||
|
return typed
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,46 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEPendingPrivateMessage: Equatable {
|
||||||
|
let content: String
|
||||||
|
let messageID: String
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLENoiseSessionQueues {
|
||||||
|
private var privateMessagesByPeerID: [PeerID: [BLEPendingPrivateMessage]] = [:]
|
||||||
|
private var typedPayloadsByPeerID: [PeerID: [Data]] = [:]
|
||||||
|
|
||||||
|
var isEmpty: Bool {
|
||||||
|
privateMessagesByPeerID.isEmpty && typedPayloadsByPeerID.isEmpty
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
privateMessagesByPeerID.removeAll()
|
||||||
|
typedPayloadsByPeerID.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func appendPrivateMessage(content: String, messageID: String, for peerID: PeerID) {
|
||||||
|
privateMessagesByPeerID[peerID, default: []].append(BLEPendingPrivateMessage(content: content, messageID: messageID))
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func takePrivateMessages(for peerID: PeerID) -> [BLEPendingPrivateMessage] {
|
||||||
|
let messages = privateMessagesByPeerID[peerID] ?? []
|
||||||
|
privateMessagesByPeerID.removeValue(forKey: peerID)
|
||||||
|
return messages
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func prependPrivateMessages(_ messages: [BLEPendingPrivateMessage], for peerID: PeerID) {
|
||||||
|
guard !messages.isEmpty else { return }
|
||||||
|
privateMessagesByPeerID[peerID, default: []].insert(contentsOf: messages, at: 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func appendTypedPayload(_ payload: Data, for peerID: PeerID) {
|
||||||
|
typedPayloadsByPeerID[peerID, default: []].append(payload)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func takeTypedPayloads(for peerID: PeerID) -> [Data] {
|
||||||
|
let payloads = typedPayloadsByPeerID[peerID] ?? []
|
||||||
|
typedPayloadsByPeerID.removeValue(forKey: peerID)
|
||||||
|
return payloads
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,137 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEOutboundFragmentPlan {
|
||||||
|
let fragmentPackets: [BitchatPacket]
|
||||||
|
let fragmentVersion: UInt8
|
||||||
|
let chunkSize: Int
|
||||||
|
let spacingMs: Int
|
||||||
|
|
||||||
|
var totalFragments: Int {
|
||||||
|
fragmentPackets.count
|
||||||
|
}
|
||||||
|
|
||||||
|
var shouldPauseScanning: Bool {
|
||||||
|
totalFragments > 4
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEOutboundFragmentPlanner {
|
||||||
|
private static let minimumChunkSize = 64
|
||||||
|
private static let fragmentIDLength = 8
|
||||||
|
|
||||||
|
static func makePlan(
|
||||||
|
for request: BLEOutboundFragmentTransferRequest,
|
||||||
|
defaultChunkSize: Int,
|
||||||
|
bleMaxMTU: Int,
|
||||||
|
fragmentID: Data = randomFragmentID()
|
||||||
|
) -> BLEOutboundFragmentPlan? {
|
||||||
|
guard fragmentID.count == fragmentIDLength,
|
||||||
|
let fullData = request.packet.toBinaryData(padding: request.pad) else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
let sizing = sizingPolicy(
|
||||||
|
for: request.packet,
|
||||||
|
requestedMaxChunk: request.maxChunk,
|
||||||
|
defaultChunkSize: defaultChunkSize,
|
||||||
|
bleMaxMTU: bleMaxMTU
|
||||||
|
)
|
||||||
|
|
||||||
|
let chunks = stride(from: 0, to: fullData.count, by: sizing.chunkSize).map { offset in
|
||||||
|
Data(fullData[offset..<min(offset + sizing.chunkSize, fullData.count)])
|
||||||
|
}
|
||||||
|
|
||||||
|
guard !chunks.isEmpty else { return nil }
|
||||||
|
|
||||||
|
let fragmentRecipient: Data? = {
|
||||||
|
if let directedPeer = request.directedPeer {
|
||||||
|
return Data(hexString: directedPeer.id)
|
||||||
|
}
|
||||||
|
return request.packet.recipientID
|
||||||
|
}()
|
||||||
|
|
||||||
|
let fragmentPackets = chunks.enumerated().map { index, chunk in
|
||||||
|
makeFragmentPacket(
|
||||||
|
original: request.packet,
|
||||||
|
fragmentID: fragmentID,
|
||||||
|
index: index,
|
||||||
|
total: chunks.count,
|
||||||
|
fragmentData: chunk,
|
||||||
|
fragmentRecipient: fragmentRecipient,
|
||||||
|
fragmentVersion: sizing.fragmentVersion
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
return BLEOutboundFragmentPlan(
|
||||||
|
fragmentPackets: fragmentPackets,
|
||||||
|
fragmentVersion: sizing.fragmentVersion,
|
||||||
|
chunkSize: sizing.chunkSize,
|
||||||
|
spacingMs: spacingMs(for: request)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func sizingPolicy(
|
||||||
|
for packet: BitchatPacket,
|
||||||
|
requestedMaxChunk: Int?,
|
||||||
|
defaultChunkSize: Int,
|
||||||
|
bleMaxMTU: Int
|
||||||
|
) -> (fragmentVersion: UInt8, chunkSize: Int) {
|
||||||
|
var fragmentVersion: UInt8 = 1
|
||||||
|
var calculatedChunk = defaultChunkSize
|
||||||
|
|
||||||
|
if let route = packet.route, !route.isEmpty {
|
||||||
|
fragmentVersion = 2
|
||||||
|
let routeSize = 1 + (route.count * 8)
|
||||||
|
let overhead = 16 + 8 + 8 + routeSize + 13 + 16
|
||||||
|
calculatedChunk = max(minimumChunkSize, bleMaxMTU - overhead)
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
fragmentVersion: fragmentVersion,
|
||||||
|
chunkSize: max(minimumChunkSize, requestedMaxChunk ?? calculatedChunk)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func makeFragmentPacket(
|
||||||
|
original packet: BitchatPacket,
|
||||||
|
fragmentID: Data,
|
||||||
|
index: Int,
|
||||||
|
total: Int,
|
||||||
|
fragmentData: Data,
|
||||||
|
fragmentRecipient: Data?,
|
||||||
|
fragmentVersion: UInt8
|
||||||
|
) -> BitchatPacket {
|
||||||
|
var payload = Data()
|
||||||
|
payload.append(fragmentID)
|
||||||
|
payload.append(contentsOf: withUnsafeBytes(of: UInt16(index).bigEndian) { Data($0) })
|
||||||
|
payload.append(contentsOf: withUnsafeBytes(of: UInt16(total).bigEndian) { Data($0) })
|
||||||
|
payload.append(packet.type)
|
||||||
|
payload.append(fragmentData)
|
||||||
|
|
||||||
|
return BitchatPacket(
|
||||||
|
type: MessageType.fragment.rawValue,
|
||||||
|
senderID: packet.senderID,
|
||||||
|
recipientID: fragmentRecipient,
|
||||||
|
timestamp: packet.timestamp,
|
||||||
|
payload: payload,
|
||||||
|
signature: nil,
|
||||||
|
ttl: packet.ttl,
|
||||||
|
version: fragmentVersion,
|
||||||
|
route: packet.route,
|
||||||
|
isRSR: packet.isRSR
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func spacingMs(for request: BLEOutboundFragmentTransferRequest) -> Int {
|
||||||
|
if request.directedPeer != nil || request.packet.recipientID != nil {
|
||||||
|
return TransportConfig.bleFragmentSpacingDirectedMs
|
||||||
|
}
|
||||||
|
|
||||||
|
return TransportConfig.bleFragmentSpacingMs
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func randomFragmentID() -> Data {
|
||||||
|
Data((0..<fragmentIDLength).map { _ in UInt8.random(in: 0...255) })
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,181 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEOutboundFragmentTransferRequest {
|
||||||
|
let packet: BitchatPacket
|
||||||
|
let pad: Bool
|
||||||
|
let maxChunk: Int?
|
||||||
|
let directedPeer: PeerID?
|
||||||
|
let transferId: String?
|
||||||
|
|
||||||
|
var resolvedTransferId: String? {
|
||||||
|
guard packet.type == MessageType.fileTransfer.rawValue else { return nil }
|
||||||
|
return transferId ?? packet.payload.sha256Hex()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEOutboundFragmentTransferScheduler {
|
||||||
|
enum QueuePosition {
|
||||||
|
case front
|
||||||
|
case back
|
||||||
|
}
|
||||||
|
|
||||||
|
enum SubmitResult {
|
||||||
|
case start(request: BLEOutboundFragmentTransferRequest, reservedTransferId: String?)
|
||||||
|
case queued(request: BLEOutboundFragmentTransferRequest, transferId: String?, position: QueuePosition)
|
||||||
|
}
|
||||||
|
|
||||||
|
enum CancelResult {
|
||||||
|
case active(transferId: String, workItems: [DispatchWorkItem])
|
||||||
|
case pending(transferId: String)
|
||||||
|
case missing
|
||||||
|
}
|
||||||
|
|
||||||
|
enum SentResult: Equatable {
|
||||||
|
case progress(sentFragments: Int, totalFragments: Int)
|
||||||
|
case complete(sentFragments: Int, totalFragments: Int)
|
||||||
|
case missing
|
||||||
|
}
|
||||||
|
|
||||||
|
private struct ActiveTransferState {
|
||||||
|
let totalFragments: Int
|
||||||
|
var sentFragments: Int
|
||||||
|
var workItems: [DispatchWorkItem]
|
||||||
|
}
|
||||||
|
|
||||||
|
private var activeTransfers: [String: ActiveTransferState] = [:]
|
||||||
|
private var pendingTransfers: [BLEOutboundFragmentTransferRequest] = []
|
||||||
|
|
||||||
|
var activeCount: Int {
|
||||||
|
activeTransfers.count
|
||||||
|
}
|
||||||
|
|
||||||
|
var pendingCount: Int {
|
||||||
|
pendingTransfers.count
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() -> [(id: String, workItems: [DispatchWorkItem])] {
|
||||||
|
let active = activeTransfers.map { ($0.key, $0.value.workItems) }
|
||||||
|
activeTransfers.removeAll()
|
||||||
|
pendingTransfers.removeAll()
|
||||||
|
return active
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func submit(
|
||||||
|
_ request: BLEOutboundFragmentTransferRequest,
|
||||||
|
maxConcurrentTransfers: Int
|
||||||
|
) -> SubmitResult {
|
||||||
|
guard let transferId = request.resolvedTransferId else {
|
||||||
|
return .start(request: request, reservedTransferId: nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard activeTransfers.count < maxConcurrentTransfers else {
|
||||||
|
pendingTransfers.append(request)
|
||||||
|
return .queued(request: request, transferId: transferId, position: .back)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard activeTransfers[transferId] == nil else {
|
||||||
|
pendingTransfers.insert(request, at: 0)
|
||||||
|
return .queued(request: request, transferId: transferId, position: .front)
|
||||||
|
}
|
||||||
|
|
||||||
|
activeTransfers[transferId] = ActiveTransferState(totalFragments: 0, sentFragments: 0, workItems: [])
|
||||||
|
return .start(request: request, reservedTransferId: transferId)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func activateReservedTransfer(
|
||||||
|
id transferId: String,
|
||||||
|
totalFragments: Int,
|
||||||
|
workItems: [DispatchWorkItem]
|
||||||
|
) -> Bool {
|
||||||
|
guard activeTransfers[transferId] != nil else { return false }
|
||||||
|
activeTransfers[transferId] = ActiveTransferState(
|
||||||
|
totalFragments: totalFragments,
|
||||||
|
sentFragments: 0,
|
||||||
|
workItems: workItems
|
||||||
|
)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func updateWorkItems(_ workItems: [DispatchWorkItem], for transferId: String) -> Bool {
|
||||||
|
guard var state = activeTransfers[transferId] else { return false }
|
||||||
|
state.workItems = workItems
|
||||||
|
activeTransfers[transferId] = state
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func releaseReservation(_ transferId: String) -> [DispatchWorkItem]? {
|
||||||
|
activeTransfers.removeValue(forKey: transferId)?.workItems
|
||||||
|
}
|
||||||
|
|
||||||
|
func isActive(_ transferId: String) -> Bool {
|
||||||
|
activeTransfers[transferId] != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func cancelTransfer(_ transferId: String) -> CancelResult {
|
||||||
|
if let active = activeTransfers.removeValue(forKey: transferId) {
|
||||||
|
return .active(transferId: transferId, workItems: active.workItems)
|
||||||
|
}
|
||||||
|
|
||||||
|
if let pendingIndex = pendingTransfers.firstIndex(where: { $0.resolvedTransferId == transferId || $0.transferId == transferId }) {
|
||||||
|
pendingTransfers.remove(at: pendingIndex)
|
||||||
|
return .pending(transferId: transferId)
|
||||||
|
}
|
||||||
|
|
||||||
|
return .missing
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func markFragmentSent(transferId: String) -> SentResult {
|
||||||
|
guard var state = activeTransfers[transferId] else { return .missing }
|
||||||
|
|
||||||
|
state.sentFragments = min(state.sentFragments + 1, state.totalFragments)
|
||||||
|
let isComplete = state.sentFragments >= state.totalFragments
|
||||||
|
|
||||||
|
if isComplete {
|
||||||
|
activeTransfers.removeValue(forKey: transferId)
|
||||||
|
return .complete(sentFragments: state.sentFragments, totalFragments: state.totalFragments)
|
||||||
|
}
|
||||||
|
|
||||||
|
activeTransfers[transferId] = state
|
||||||
|
return .progress(sentFragments: state.sentFragments, totalFragments: state.totalFragments)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func reservePendingStarts(maxConcurrentTransfers: Int) -> [SubmitResult] {
|
||||||
|
var availableSlots = max(0, maxConcurrentTransfers - activeTransfers.count)
|
||||||
|
guard availableSlots > 0, !pendingTransfers.isEmpty else { return [] }
|
||||||
|
|
||||||
|
var results: [SubmitResult] = []
|
||||||
|
var blockedFront: [BLEOutboundFragmentTransferRequest] = []
|
||||||
|
|
||||||
|
while availableSlots > 0, !pendingTransfers.isEmpty {
|
||||||
|
let request = pendingTransfers.removeFirst()
|
||||||
|
availableSlots -= 1
|
||||||
|
|
||||||
|
guard let transferId = request.resolvedTransferId else {
|
||||||
|
results.append(.start(request: request, reservedTransferId: nil))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
guard activeTransfers.count < maxConcurrentTransfers else {
|
||||||
|
pendingTransfers.insert(request, at: 0)
|
||||||
|
results.append(.queued(request: request, transferId: transferId, position: .front))
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
guard activeTransfers[transferId] == nil else {
|
||||||
|
blockedFront.append(request)
|
||||||
|
results.append(.queued(request: request, transferId: transferId, position: .front))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
activeTransfers[transferId] = ActiveTransferState(totalFragments: 0, sentFragments: 0, workItems: [])
|
||||||
|
results.append(.start(request: request, reservedTransferId: transferId))
|
||||||
|
}
|
||||||
|
|
||||||
|
if !blockedFront.isEmpty {
|
||||||
|
pendingTransfers.insert(contentsOf: blockedFront, at: 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
return results
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,91 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEOutboundLinkPlan: Equatable {
|
||||||
|
let directedPeerHint: PeerID?
|
||||||
|
let fragmentChunkSize: Int?
|
||||||
|
let selectedLinks: BLEFanoutSelection
|
||||||
|
let shouldSpoolDirectedPacket: Bool
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEOutboundLinkPlanner {
|
||||||
|
static func plan(
|
||||||
|
packet: BitchatPacket,
|
||||||
|
dataCount: Int,
|
||||||
|
peripheralIDs: [String],
|
||||||
|
peripheralWriteLimits: [Int],
|
||||||
|
centralIDs: [String],
|
||||||
|
centralNotifyLimits: [Int],
|
||||||
|
ingressRecord: BLEIngressLinkRecord?,
|
||||||
|
excludedLinks: Set<BLEIngressLinkID>,
|
||||||
|
peripheralPeerBindings: [String: PeerID] = [:],
|
||||||
|
centralPeerBindings: [String: PeerID] = [:],
|
||||||
|
directedOnlyPeer: PeerID?
|
||||||
|
) -> BLEOutboundLinkPlan {
|
||||||
|
if let minLimit = minimumLinkLimit(
|
||||||
|
peripheralWriteLimits: peripheralWriteLimits,
|
||||||
|
centralNotifyLimits: centralNotifyLimits
|
||||||
|
), packet.type != MessageType.fragment.rawValue,
|
||||||
|
dataCount > minLimit {
|
||||||
|
return BLEOutboundLinkPlan(
|
||||||
|
directedPeerHint: directedPeerHint(for: packet, explicitPeer: directedOnlyPeer),
|
||||||
|
fragmentChunkSize: BLEOutboundPacketPolicy.fragmentChunkSize(forLinkLimit: minLimit),
|
||||||
|
selectedLinks: BLEFanoutSelection(peripheralIDs: [], centralIDs: []),
|
||||||
|
shouldSpoolDirectedPacket: false
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
let directedPeerHint = directedPeerHint(for: packet, explicitPeer: directedOnlyPeer)
|
||||||
|
let selectedLinks = BLEFanoutSelector.selectLinks(
|
||||||
|
peripheralIDs: peripheralIDs,
|
||||||
|
centralIDs: centralIDs,
|
||||||
|
ingressLink: ingressRecord?.link,
|
||||||
|
excludedLinks: excludedLinks,
|
||||||
|
peripheralPeerBindings: peripheralPeerBindings,
|
||||||
|
centralPeerBindings: centralPeerBindings,
|
||||||
|
directedPeerHint: directedPeerHint,
|
||||||
|
packetType: packet.type,
|
||||||
|
messageID: BLEOutboundPacketPolicy.messageID(for: packet)
|
||||||
|
)
|
||||||
|
|
||||||
|
return BLEOutboundLinkPlan(
|
||||||
|
directedPeerHint: directedPeerHint,
|
||||||
|
fragmentChunkSize: nil,
|
||||||
|
selectedLinks: selectedLinks,
|
||||||
|
shouldSpoolDirectedPacket: shouldSpoolDirectedPacket(
|
||||||
|
directedPeerHint: directedPeerHint,
|
||||||
|
selectedLinks: selectedLinks,
|
||||||
|
packetType: packet.type
|
||||||
|
)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
static func directedPeerHint(for packet: BitchatPacket, explicitPeer: PeerID?) -> PeerID? {
|
||||||
|
if let explicitPeer { return explicitPeer }
|
||||||
|
if let recipient = PeerID(str: packet.recipientID?.hexEncodedString()), !recipient.isEmpty {
|
||||||
|
return recipient
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
static func minimumLinkLimit(peripheralWriteLimits: [Int], centralNotifyLimits: [Int]) -> Int? {
|
||||||
|
[peripheralWriteLimits.min(), centralNotifyLimits.min()]
|
||||||
|
.compactMap { $0 }
|
||||||
|
.min()
|
||||||
|
}
|
||||||
|
|
||||||
|
static func shouldSpoolDirectedPacket(
|
||||||
|
directedPeerHint: PeerID?,
|
||||||
|
selectedLinks: BLEFanoutSelection,
|
||||||
|
packetType: UInt8
|
||||||
|
) -> Bool {
|
||||||
|
guard directedPeerHint != nil,
|
||||||
|
selectedLinks.peripheralIDs.isEmpty,
|
||||||
|
selectedLinks.centralIDs.isEmpty else {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return packetType == MessageType.noiseEncrypted.rawValue ||
|
||||||
|
packetType == MessageType.noiseHandshake.rawValue
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,47 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEPendingNotification<Target> {
|
||||||
|
let data: Data
|
||||||
|
let targets: [Target]?
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEOutboundNotificationBuffer<Target> {
|
||||||
|
enum EnqueueResult {
|
||||||
|
case enqueued(count: Int)
|
||||||
|
case full(count: Int)
|
||||||
|
}
|
||||||
|
|
||||||
|
private var notifications: [BLEPendingNotification<Target>] = []
|
||||||
|
|
||||||
|
var count: Int {
|
||||||
|
notifications.count
|
||||||
|
}
|
||||||
|
|
||||||
|
var isEmpty: Bool {
|
||||||
|
notifications.isEmpty
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
notifications.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func enqueue(data: Data, targets: [Target]?, capCount: Int) -> EnqueueResult {
|
||||||
|
guard notifications.count < capCount else {
|
||||||
|
return .full(count: notifications.count)
|
||||||
|
}
|
||||||
|
|
||||||
|
notifications.append(BLEPendingNotification(data: data, targets: targets))
|
||||||
|
return .enqueued(count: notifications.count)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func takeAll() -> [BLEPendingNotification<Target>] {
|
||||||
|
let pending = notifications
|
||||||
|
notifications.removeAll()
|
||||||
|
return pending
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func prepend(_ pending: [BLEPendingNotification<Target>]) {
|
||||||
|
guard !pending.isEmpty else { return }
|
||||||
|
notifications.insert(contentsOf: pending, at: 0)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,43 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum BLEOutboundPacketPolicy {
|
||||||
|
private static let fragmentFrameOverhead = 13 + 8 + 8 + 13
|
||||||
|
|
||||||
|
static func messageID(for packet: BitchatPacket) -> String {
|
||||||
|
BLEIngressLinkRegistry.messageID(for: packet)
|
||||||
|
}
|
||||||
|
|
||||||
|
static func padsBLEFrame(for packetType: UInt8) -> Bool {
|
||||||
|
switch MessageType(rawValue: packetType) {
|
||||||
|
case .noiseEncrypted, .noiseHandshake:
|
||||||
|
return true
|
||||||
|
case .none, .announce, .message, .leave, .requestSync, .fragment, .fileTransfer:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static func priority(for packet: BitchatPacket, data: Data) -> BLEOutboundWritePriority {
|
||||||
|
guard let messageType = MessageType(rawValue: packet.type) else { return .low }
|
||||||
|
switch messageType {
|
||||||
|
case .fragment:
|
||||||
|
return .fragment(totalFragments: fragmentTotalCount(from: packet.payload))
|
||||||
|
case .fileTransfer:
|
||||||
|
return .fileTransfer
|
||||||
|
default:
|
||||||
|
return .high
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static func fragmentChunkSize(forLinkLimit limit: Int) -> Int {
|
||||||
|
max(64, limit - fragmentFrameOverhead)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func fragmentTotalCount(from payload: Data) -> Int {
|
||||||
|
guard payload.count >= 12 else { return Int(UInt16.max) }
|
||||||
|
let totalHigh = Int(payload[10])
|
||||||
|
let totalLow = Int(payload[11])
|
||||||
|
let total = (totalHigh << 8) | totalLow
|
||||||
|
return max(total, 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,83 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEOutboundWritePriority: Comparable {
|
||||||
|
let level: Int
|
||||||
|
let suborder: Int
|
||||||
|
|
||||||
|
static let high = BLEOutboundWritePriority(level: 0, suborder: 0)
|
||||||
|
|
||||||
|
static func fragment(totalFragments: Int) -> BLEOutboundWritePriority {
|
||||||
|
BLEOutboundWritePriority(level: 1, suborder: max(1, min(totalFragments, Int(UInt16.max))))
|
||||||
|
}
|
||||||
|
|
||||||
|
static let fileTransfer = BLEOutboundWritePriority(level: 2, suborder: Int.max - 1)
|
||||||
|
static let low = BLEOutboundWritePriority(level: 2, suborder: Int.max)
|
||||||
|
|
||||||
|
static func < (lhs: BLEOutboundWritePriority, rhs: BLEOutboundWritePriority) -> Bool {
|
||||||
|
if lhs.level != rhs.level { return lhs.level < rhs.level }
|
||||||
|
return lhs.suborder < rhs.suborder
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEPendingWrite {
|
||||||
|
let priority: BLEOutboundWritePriority
|
||||||
|
let data: Data
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEOutboundWriteBuffer {
|
||||||
|
enum EnqueueResult {
|
||||||
|
case enqueued(trimmedBytes: Int, remainingBytes: Int)
|
||||||
|
case oversized(bytes: Int)
|
||||||
|
}
|
||||||
|
|
||||||
|
private var writesByPeripheralID: [String: [BLEPendingWrite]] = [:]
|
||||||
|
|
||||||
|
var peripheralIDs: [String] {
|
||||||
|
Array(writesByPeripheralID.keys)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
writesByPeripheralID.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func enqueue(
|
||||||
|
data: Data,
|
||||||
|
for peripheralID: String,
|
||||||
|
priority: BLEOutboundWritePriority,
|
||||||
|
capBytes: Int
|
||||||
|
) -> EnqueueResult {
|
||||||
|
guard data.count <= capBytes else {
|
||||||
|
return .oversized(bytes: data.count)
|
||||||
|
}
|
||||||
|
|
||||||
|
var queue = writesByPeripheralID[peripheralID] ?? []
|
||||||
|
let item = BLEPendingWrite(priority: priority, data: data)
|
||||||
|
let insertIndex = queue.firstIndex { item.priority < $0.priority } ?? queue.count
|
||||||
|
queue.insert(item, at: insertIndex)
|
||||||
|
|
||||||
|
var total = queue.reduce(0) { $0 + $1.data.count }
|
||||||
|
var trimmedBytes = 0
|
||||||
|
|
||||||
|
while total > capBytes && !queue.isEmpty {
|
||||||
|
let removed = queue.removeLast()
|
||||||
|
trimmedBytes += removed.data.count
|
||||||
|
total -= removed.data.count
|
||||||
|
}
|
||||||
|
|
||||||
|
writesByPeripheralID[peripheralID] = queue.isEmpty ? nil : queue
|
||||||
|
return .enqueued(trimmedBytes: trimmedBytes, remainingBytes: total)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func takeAll(for peripheralID: String) -> [BLEPendingWrite] {
|
||||||
|
let items = writesByPeripheralID[peripheralID] ?? []
|
||||||
|
writesByPeripheralID[peripheralID] = nil
|
||||||
|
return items
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func prepend(_ items: [BLEPendingWrite], for peripheralID: String) {
|
||||||
|
guard !items.isEmpty else { return }
|
||||||
|
var existing = writesByPeripheralID[peripheralID] ?? []
|
||||||
|
existing.insert(contentsOf: items, at: 0)
|
||||||
|
writesByPeripheralID[peripheralID] = existing
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,27 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum BLEPacketFreshnessPolicy {
|
||||||
|
static let defaultMaxAgeSeconds: TimeInterval = 900
|
||||||
|
|
||||||
|
static func isBroadcastRecipient(_ recipientID: Data?) -> Bool {
|
||||||
|
guard let recipientID else { return true }
|
||||||
|
return recipientID.count == 8 && recipientID.allSatisfy { $0 == 0xFF }
|
||||||
|
}
|
||||||
|
|
||||||
|
static func isStale(
|
||||||
|
timestampMilliseconds: UInt64,
|
||||||
|
now: Date,
|
||||||
|
maxAgeSeconds: TimeInterval = defaultMaxAgeSeconds
|
||||||
|
) -> Bool {
|
||||||
|
let nowMilliseconds = UInt64(now.timeIntervalSince1970 * 1000)
|
||||||
|
let maxAgeMilliseconds = UInt64(maxAgeSeconds * 1000)
|
||||||
|
guard nowMilliseconds >= maxAgeMilliseconds else { return false }
|
||||||
|
return timestampMilliseconds < nowMilliseconds - maxAgeMilliseconds
|
||||||
|
}
|
||||||
|
|
||||||
|
static func ageSeconds(timestampMilliseconds: UInt64, now: Date) -> Double {
|
||||||
|
let nowMilliseconds = UInt64(now.timeIntervalSince1970 * 1000)
|
||||||
|
guard nowMilliseconds >= timestampMilliseconds else { return 0 }
|
||||||
|
return Double(nowMilliseconds - timestampMilliseconds) / 1000.0
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,25 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEPeerEventDebouncer {
|
||||||
|
private var lastEmitByPeer: [PeerID: Date] = [:]
|
||||||
|
|
||||||
|
var count: Int {
|
||||||
|
lastEmitByPeer.count
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
mutating func shouldEmit(peerID: PeerID, now: Date, minimumInterval: TimeInterval) -> Bool {
|
||||||
|
if let lastEmit = lastEmitByPeer[peerID],
|
||||||
|
now.timeIntervalSince(lastEmit) < minimumInterval {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
lastEmitByPeer[peerID] = now
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
lastEmitByPeer.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,43 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum BLEPeerPublishDecision: Equatable {
|
||||||
|
case publishNow
|
||||||
|
case schedule(delay: TimeInterval)
|
||||||
|
case skip
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEPeerPublishCoalescer {
|
||||||
|
private var lastPublishAt: Date
|
||||||
|
private var publishPending: Bool
|
||||||
|
private let minimumInterval: TimeInterval
|
||||||
|
|
||||||
|
init(
|
||||||
|
lastPublishAt: Date = .distantPast,
|
||||||
|
publishPending: Bool = false,
|
||||||
|
minimumInterval: TimeInterval = 0.1
|
||||||
|
) {
|
||||||
|
self.lastPublishAt = lastPublishAt
|
||||||
|
self.publishPending = publishPending
|
||||||
|
self.minimumInterval = minimumInterval
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func requestPublish(now: Date) -> BLEPeerPublishDecision {
|
||||||
|
let elapsed = now.timeIntervalSince(lastPublishAt)
|
||||||
|
if elapsed >= minimumInterval {
|
||||||
|
lastPublishAt = now
|
||||||
|
return .publishNow
|
||||||
|
}
|
||||||
|
|
||||||
|
guard !publishPending else {
|
||||||
|
return .skip
|
||||||
|
}
|
||||||
|
|
||||||
|
publishPending = true
|
||||||
|
return .schedule(delay: minimumInterval - elapsed)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func scheduledPublishFired(now: Date) {
|
||||||
|
lastPublishAt = now
|
||||||
|
publishPending = false
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,211 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEPeerInfo: Equatable {
|
||||||
|
let peerID: PeerID
|
||||||
|
var nickname: String
|
||||||
|
var isConnected: Bool
|
||||||
|
var noisePublicKey: Data?
|
||||||
|
var signingPublicKey: Data?
|
||||||
|
var isVerifiedNickname: Bool
|
||||||
|
var lastSeen: Date
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEPeerAnnounceUpdate: Equatable {
|
||||||
|
let isNewPeer: Bool
|
||||||
|
let wasDisconnected: Bool
|
||||||
|
let previousNickname: String?
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEPeerLinkPresence: Equatable {
|
||||||
|
var hasPeripheral: Bool
|
||||||
|
var hasCentral: Bool
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLERemovedPeer: Equatable {
|
||||||
|
let peerID: PeerID
|
||||||
|
let nickname: String
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEPeerConnectivityChanges: Equatable {
|
||||||
|
var disconnectedPeerIDs: [PeerID] = []
|
||||||
|
var removedPeers: [BLERemovedPeer] = []
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEPeerRegistry {
|
||||||
|
private var peers: [PeerID: BLEPeerInfo] = [:]
|
||||||
|
|
||||||
|
var isEmpty: Bool {
|
||||||
|
peers.isEmpty
|
||||||
|
}
|
||||||
|
|
||||||
|
var count: Int {
|
||||||
|
peers.count
|
||||||
|
}
|
||||||
|
|
||||||
|
var peerIDs: [PeerID] {
|
||||||
|
Array(peers.keys)
|
||||||
|
}
|
||||||
|
|
||||||
|
var connectedCount: Int {
|
||||||
|
peers.values.filter(\.isConnected).count
|
||||||
|
}
|
||||||
|
|
||||||
|
var connectedPeerIDs: [PeerID] {
|
||||||
|
peers.values.compactMap { $0.isConnected ? $0.peerID : nil }
|
||||||
|
}
|
||||||
|
|
||||||
|
var connectedRoutingData: [Data] {
|
||||||
|
peers.values.filter(\.isConnected).compactMap { $0.peerID.routingData }
|
||||||
|
}
|
||||||
|
|
||||||
|
var snapshotByID: [PeerID: BLEPeerInfo] {
|
||||||
|
peers
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
peers.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
func info(for peerID: PeerID) -> BLEPeerInfo? {
|
||||||
|
peers[peerID]
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func upsert(_ info: BLEPeerInfo) {
|
||||||
|
peers[info.peerID] = info
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
mutating func remove(_ peerID: PeerID) -> BLEPeerInfo? {
|
||||||
|
peers.removeValue(forKey: peerID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func isConnected(_ peerID: PeerID) -> Bool {
|
||||||
|
peers[peerID.toShort()]?.isConnected ?? false
|
||||||
|
}
|
||||||
|
|
||||||
|
func isReachable(_ peerID: PeerID, now: Date) -> Bool {
|
||||||
|
let shortID = peerID.toShort()
|
||||||
|
let meshAttached = connectedCount > 0
|
||||||
|
guard let info = peers[shortID] else { return false }
|
||||||
|
if info.isConnected { return true }
|
||||||
|
guard meshAttached else { return false }
|
||||||
|
|
||||||
|
let retention: TimeInterval = info.isVerifiedNickname
|
||||||
|
? TransportConfig.bleReachabilityRetentionVerifiedSeconds
|
||||||
|
: TransportConfig.bleReachabilityRetentionUnverifiedSeconds
|
||||||
|
return now.timeIntervalSince(info.lastSeen) <= retention
|
||||||
|
}
|
||||||
|
|
||||||
|
func nickname(for peerID: PeerID, connectedOnly: Bool) -> String? {
|
||||||
|
guard let peer = peers[peerID] else { return nil }
|
||||||
|
if connectedOnly && !peer.isConnected { return nil }
|
||||||
|
return peer.nickname
|
||||||
|
}
|
||||||
|
|
||||||
|
func fingerprint(for peerID: PeerID) -> String? {
|
||||||
|
peers[peerID]?.noisePublicKey?.sha256Fingerprint()
|
||||||
|
}
|
||||||
|
|
||||||
|
func displayNicknames(selfNickname: String) -> [PeerID: String] {
|
||||||
|
let connected = peers.filter { $0.value.isConnected }
|
||||||
|
let tuples = connected.map { ($0.key, $0.value.nickname, true) }
|
||||||
|
return PeerDisplayNameResolver.resolve(tuples, selfNickname: selfNickname)
|
||||||
|
}
|
||||||
|
|
||||||
|
func transportSnapshots(selfNickname: String) -> [TransportPeerSnapshot] {
|
||||||
|
let snapshot = Array(peers.values)
|
||||||
|
let resolvedNames = PeerDisplayNameResolver.resolve(
|
||||||
|
snapshot.map { ($0.peerID, $0.nickname, $0.isConnected) },
|
||||||
|
selfNickname: selfNickname
|
||||||
|
)
|
||||||
|
return snapshot.map { info in
|
||||||
|
TransportPeerSnapshot(
|
||||||
|
peerID: info.peerID,
|
||||||
|
nickname: resolvedNames[info.peerID] ?? info.nickname,
|
||||||
|
isConnected: info.isConnected,
|
||||||
|
noisePublicKey: info.noisePublicKey,
|
||||||
|
lastSeen: info.lastSeen
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func collisionResolvedNickname(for peerID: PeerID, selfNickname: String) -> String? {
|
||||||
|
guard let info = peers[peerID], info.isVerifiedNickname else { return nil }
|
||||||
|
let hasCollision = peers.values.contains {
|
||||||
|
$0.isConnected && $0.nickname == info.nickname && $0.peerID != peerID
|
||||||
|
} || selfNickname == info.nickname
|
||||||
|
return hasCollision ? info.nickname + "#" + String(peerID.id.prefix(4)) : info.nickname
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func markDisconnected(_ peerID: PeerID) {
|
||||||
|
guard var info = peers[peerID] else { return }
|
||||||
|
info.isConnected = false
|
||||||
|
peers[peerID] = info
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func updateLastSeen(_ peerID: PeerID, at date: Date) {
|
||||||
|
guard var peer = peers[peerID] else { return }
|
||||||
|
peer.lastSeen = date
|
||||||
|
peers[peerID] = peer
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func upsertVerifiedAnnounce(
|
||||||
|
peerID: PeerID,
|
||||||
|
nickname: String,
|
||||||
|
noisePublicKey: Data,
|
||||||
|
signingPublicKey: Data?,
|
||||||
|
isConnected: Bool,
|
||||||
|
now: Date
|
||||||
|
) -> BLEPeerAnnounceUpdate {
|
||||||
|
let existing = peers[peerID]
|
||||||
|
let update = BLEPeerAnnounceUpdate(
|
||||||
|
isNewPeer: existing == nil,
|
||||||
|
wasDisconnected: existing?.isConnected == false,
|
||||||
|
previousNickname: existing?.nickname
|
||||||
|
)
|
||||||
|
|
||||||
|
peers[peerID] = BLEPeerInfo(
|
||||||
|
peerID: existing?.peerID ?? peerID,
|
||||||
|
nickname: nickname,
|
||||||
|
isConnected: isConnected,
|
||||||
|
noisePublicKey: noisePublicKey,
|
||||||
|
signingPublicKey: signingPublicKey,
|
||||||
|
isVerifiedNickname: true,
|
||||||
|
lastSeen: now
|
||||||
|
)
|
||||||
|
|
||||||
|
return update
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func reconcileConnectivity(
|
||||||
|
now: Date,
|
||||||
|
linkStates: [PeerID: BLEPeerLinkPresence]
|
||||||
|
) -> BLEPeerConnectivityChanges {
|
||||||
|
var changes = BLEPeerConnectivityChanges()
|
||||||
|
|
||||||
|
for (peerID, peer) in Array(peers) {
|
||||||
|
let age = now.timeIntervalSince(peer.lastSeen)
|
||||||
|
let retention: TimeInterval = peer.isVerifiedNickname
|
||||||
|
? TransportConfig.bleReachabilityRetentionVerifiedSeconds
|
||||||
|
: TransportConfig.bleReachabilityRetentionUnverifiedSeconds
|
||||||
|
|
||||||
|
if peer.isConnected && age > TransportConfig.blePeerInactivityTimeoutSeconds {
|
||||||
|
let state = linkStates[peerID] ?? BLEPeerLinkPresence(hasPeripheral: false, hasCentral: false)
|
||||||
|
if !state.hasPeripheral && !state.hasCentral {
|
||||||
|
var updated = peer
|
||||||
|
updated.isConnected = false
|
||||||
|
peers[peerID] = updated
|
||||||
|
changes.disconnectedPeerIDs.append(peerID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !peer.isConnected && age > retention {
|
||||||
|
peers.removeValue(forKey: peerID)
|
||||||
|
changes.removedPeers.append(BLERemovedPeer(peerID: peerID, nickname: peer.nickname))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return changes
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,60 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum BLEPeerSenderDisplayName {
|
||||||
|
static func resolveKnownPeer(
|
||||||
|
peerID: PeerID,
|
||||||
|
localPeerID: PeerID,
|
||||||
|
localNickname: String,
|
||||||
|
peers: [PeerID: BLEPeerInfo],
|
||||||
|
allowConnectedUnverified: Bool
|
||||||
|
) -> String? {
|
||||||
|
if peerID == localPeerID {
|
||||||
|
return localNickname
|
||||||
|
}
|
||||||
|
|
||||||
|
guard let info = peers[peerID] else { return nil }
|
||||||
|
|
||||||
|
if info.isVerifiedNickname {
|
||||||
|
return collisionResolvedName(
|
||||||
|
displayName: info.nickname,
|
||||||
|
collisionNickname: info.nickname,
|
||||||
|
peerID: peerID,
|
||||||
|
localNickname: localNickname,
|
||||||
|
peers: peers
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
if allowConnectedUnverified, info.isConnected {
|
||||||
|
let displayName = info.nickname.isEmpty ? anonymousNickname(for: peerID) : info.nickname
|
||||||
|
return collisionResolvedName(
|
||||||
|
displayName: displayName,
|
||||||
|
collisionNickname: info.nickname,
|
||||||
|
peerID: peerID,
|
||||||
|
localNickname: localNickname,
|
||||||
|
peers: peers
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
static func anonymousNickname(for peerID: PeerID) -> String {
|
||||||
|
"anon" + String(peerID.id.prefix(4))
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func collisionResolvedName(
|
||||||
|
displayName: String,
|
||||||
|
collisionNickname: String,
|
||||||
|
peerID: PeerID,
|
||||||
|
localNickname: String,
|
||||||
|
peers: [PeerID: BLEPeerInfo]
|
||||||
|
) -> String {
|
||||||
|
let hasCollision = peers.values.contains {
|
||||||
|
$0.isConnected && $0.nickname == collisionNickname && $0.peerID != peerID
|
||||||
|
} || localNickname == collisionNickname
|
||||||
|
|
||||||
|
guard hasCollision else { return displayName }
|
||||||
|
return displayName + "#" + String(peerID.id.prefix(4))
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,104 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import BitLogger
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
/// Narrow environment for `BLEPublicMessageHandler`.
|
||||||
|
///
|
||||||
|
/// All queue hops (collections registry reads, BLE-queue link-state reads,
|
||||||
|
/// main-actor UI notification) live inside the closures supplied by
|
||||||
|
/// `BLEService`, keeping the handler queue-agnostic and synchronously testable.
|
||||||
|
struct BLEPublicMessageHandlerEnvironment {
|
||||||
|
/// Local peer identity at the time the message is handled.
|
||||||
|
let localPeerID: () -> PeerID
|
||||||
|
/// Local nickname used for sender resolution and collision checks.
|
||||||
|
let localNickname: () -> String
|
||||||
|
/// Current time source.
|
||||||
|
let now: () -> Date
|
||||||
|
/// Snapshot of known peers keyed by ID (registry read).
|
||||||
|
let peersSnapshot: () -> [PeerID: BLEPeerInfo]
|
||||||
|
/// Resolves a display name from a verified packet signature for peers missing from the registry.
|
||||||
|
let signedSenderDisplayName: (_ packet: BitchatPacket, _ peerID: PeerID) -> String?
|
||||||
|
/// Tracks the broadcast message packet for gossip sync.
|
||||||
|
let trackPacketSeen: (BitchatPacket) -> Void
|
||||||
|
/// Direct link state for the peer (BLE-queue read).
|
||||||
|
let linkState: (PeerID) -> (hasPeripheral: Bool, hasCentral: Bool)
|
||||||
|
/// Resolves and consumes the original message ID for our own re-broadcast.
|
||||||
|
let takeSelfBroadcastMessageID: (BitchatPacket) -> String?
|
||||||
|
/// Delivers `.publicMessageReceived` to the UI as one main-actor hop.
|
||||||
|
let deliverPublicMessage: (
|
||||||
|
_ peerID: PeerID,
|
||||||
|
_ nickname: String,
|
||||||
|
_ content: String,
|
||||||
|
_ timestamp: Date,
|
||||||
|
_ messageID: String?
|
||||||
|
) -> Void
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Orchestrates inbound public (broadcast) messages: freshness/self-echo
|
||||||
|
/// policy, sender display-name resolution, gossip tracking, payload decoding,
|
||||||
|
/// and UI delivery.
|
||||||
|
final class BLEPublicMessageHandler {
|
||||||
|
private let environment: BLEPublicMessageHandlerEnvironment
|
||||||
|
|
||||||
|
init(environment: BLEPublicMessageHandlerEnvironment) {
|
||||||
|
self.environment = environment
|
||||||
|
}
|
||||||
|
|
||||||
|
func handle(_ packet: BitchatPacket, from peerID: PeerID) {
|
||||||
|
let env = environment
|
||||||
|
let now = env.now()
|
||||||
|
let messageDecision = BLEPublicMessagePolicy.evaluate(
|
||||||
|
packet: packet,
|
||||||
|
from: peerID,
|
||||||
|
localPeerID: env.localPeerID(),
|
||||||
|
now: now
|
||||||
|
)
|
||||||
|
|
||||||
|
let messagePolicy: BLEPublicMessageAcceptance
|
||||||
|
switch messageDecision {
|
||||||
|
case .accept(let acceptance):
|
||||||
|
messagePolicy = acceptance
|
||||||
|
case .reject(.selfEcho):
|
||||||
|
return
|
||||||
|
case .reject(.staleBroadcast(let ageSeconds)):
|
||||||
|
SecureLogger.debug("⏰ Ignoring stale broadcast message from \(peerID.id.prefix(8))… (age: \(ageSeconds)s)", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Snapshot peers to avoid concurrent mutation while iterating during nickname collision checks.
|
||||||
|
let peersSnapshot = env.peersSnapshot()
|
||||||
|
|
||||||
|
guard let senderNickname = BLEPeerSenderDisplayName.resolveKnownPeer(
|
||||||
|
peerID: peerID,
|
||||||
|
localPeerID: env.localPeerID(),
|
||||||
|
localNickname: env.localNickname(),
|
||||||
|
peers: peersSnapshot,
|
||||||
|
allowConnectedUnverified: false
|
||||||
|
) ?? env.signedSenderDisplayName(packet, peerID) else {
|
||||||
|
SecureLogger.warning("🚫 Dropping public message from unverified or unknown peer \(peerID.id.prefix(8))…", category: .security)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if messagePolicy.shouldTrackForSync {
|
||||||
|
env.trackPacketSeen(packet)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard let content = String(data: packet.payload, encoding: .utf8) else {
|
||||||
|
SecureLogger.error("❌ Failed to decode message payload as UTF-8", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// Determine if we have a direct link to the sender
|
||||||
|
let directLink = env.linkState(peerID)
|
||||||
|
let hasDirectLink = directLink.hasPeripheral || directLink.hasCentral
|
||||||
|
|
||||||
|
let pathTag = hasDirectLink ? "direct" : "mesh"
|
||||||
|
SecureLogger.debug("💬 [\(senderNickname)] TTL:\(packet.ttl) (\(pathTag)) chars=\(content.count) bytes=\(packet.payload.count)", category: .session)
|
||||||
|
|
||||||
|
let ts = Date(timeIntervalSince1970: Double(packet.timestamp) / 1000)
|
||||||
|
var resolvedSelfMessageID: String? = nil
|
||||||
|
if peerID == env.localPeerID() {
|
||||||
|
resolvedSelfMessageID = env.takeSelfBroadcastMessageID(packet)
|
||||||
|
}
|
||||||
|
env.deliverPublicMessage(peerID, senderNickname, content, ts, resolvedSelfMessageID)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,42 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEPublicMessageAcceptance: Equatable {
|
||||||
|
let shouldTrackForSync: Bool
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEPublicMessageRejection: Equatable {
|
||||||
|
case selfEcho
|
||||||
|
case staleBroadcast(ageSeconds: Double)
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEPublicMessageDecision: Equatable {
|
||||||
|
case accept(BLEPublicMessageAcceptance)
|
||||||
|
case reject(BLEPublicMessageRejection)
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEPublicMessagePolicy {
|
||||||
|
static func evaluate(
|
||||||
|
packet: BitchatPacket,
|
||||||
|
from peerID: PeerID,
|
||||||
|
localPeerID: PeerID,
|
||||||
|
now: Date
|
||||||
|
) -> BLEPublicMessageDecision {
|
||||||
|
if peerID == localPeerID && packet.ttl != 0 {
|
||||||
|
return .reject(.selfEcho)
|
||||||
|
}
|
||||||
|
|
||||||
|
let isBroadcast = BLEPacketFreshnessPolicy.isBroadcastRecipient(packet.recipientID)
|
||||||
|
if isBroadcast,
|
||||||
|
BLEPacketFreshnessPolicy.isStale(timestampMilliseconds: packet.timestamp, now: now) {
|
||||||
|
return .reject(.staleBroadcast(ageSeconds: BLEPacketFreshnessPolicy.ageSeconds(
|
||||||
|
timestampMilliseconds: packet.timestamp,
|
||||||
|
now: now
|
||||||
|
)))
|
||||||
|
}
|
||||||
|
|
||||||
|
return .accept(BLEPublicMessageAcceptance(
|
||||||
|
shouldTrackForSync: isBroadcast && packet.type == MessageType.message.rawValue
|
||||||
|
))
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,84 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEReceivedPacketContext: Equatable {
|
||||||
|
let senderID: PeerID
|
||||||
|
let messageID: String
|
||||||
|
let messageType: MessageType?
|
||||||
|
let shouldDeduplicate: Bool
|
||||||
|
let logsHandlingDetails: Bool
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLEReceivePipeline {
|
||||||
|
static func context(for packet: BitchatPacket, localPeerID: PeerID) -> BLEReceivedPacketContext {
|
||||||
|
let senderID = PeerID(hexData: packet.senderID)
|
||||||
|
let messageID = "\(senderID)-\(packet.timestamp)-\(packet.type)"
|
||||||
|
let messageType = MessageType(rawValue: packet.type)
|
||||||
|
let allowSelfSyncReplay = packet.ttl == 0 && senderID == localPeerID
|
||||||
|
let shouldDeduplicate = messageType != .fragment && !allowSelfSyncReplay
|
||||||
|
|
||||||
|
return BLEReceivedPacketContext(
|
||||||
|
senderID: senderID,
|
||||||
|
messageID: messageID,
|
||||||
|
messageType: messageType,
|
||||||
|
shouldDeduplicate: shouldDeduplicate,
|
||||||
|
logsHandlingDetails: messageType != .announce
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
static func shouldCancelScheduledRelayForDuplicate(connectedPeerCount: Int) -> Bool {
|
||||||
|
connectedPeerCount > 2
|
||||||
|
}
|
||||||
|
|
||||||
|
static func relayDecision(
|
||||||
|
for packet: BitchatPacket,
|
||||||
|
senderID: PeerID,
|
||||||
|
localPeerID: PeerID,
|
||||||
|
degree: Int,
|
||||||
|
highDegreeThreshold: Int
|
||||||
|
) -> RelayDecision {
|
||||||
|
RelayController.decide(
|
||||||
|
ttl: packet.ttl,
|
||||||
|
senderIsSelf: senderID == localPeerID,
|
||||||
|
recipientIsSelf: PeerID(hexData: packet.recipientID) == localPeerID,
|
||||||
|
isEncrypted: packet.type == MessageType.noiseEncrypted.rawValue,
|
||||||
|
isDirectedEncrypted: packet.type == MessageType.noiseEncrypted.rawValue && packet.recipientID != nil,
|
||||||
|
isFragment: packet.type == MessageType.fragment.rawValue,
|
||||||
|
isDirectedFragment: packet.type == MessageType.fragment.rawValue && packet.recipientID != nil,
|
||||||
|
isHandshake: packet.type == MessageType.noiseHandshake.rawValue,
|
||||||
|
isAnnounce: packet.type == MessageType.announce.rawValue,
|
||||||
|
degree: degree,
|
||||||
|
highDegreeThreshold: highDegreeThreshold
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLERecentTrafficTracker: Equatable {
|
||||||
|
private var packetTimestamps: [Date] = []
|
||||||
|
|
||||||
|
var count: Int {
|
||||||
|
packetTimestamps.count
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
packetTimestamps.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func recordPacket(at now: Date) {
|
||||||
|
packetTimestamps.append(now)
|
||||||
|
prune(at: now)
|
||||||
|
}
|
||||||
|
|
||||||
|
func hasTraffic(within seconds: TimeInterval, now: Date) -> Bool {
|
||||||
|
let cutoff = now.addingTimeInterval(-seconds)
|
||||||
|
return packetTimestamps.contains { $0 >= cutoff }
|
||||||
|
}
|
||||||
|
|
||||||
|
private mutating func prune(at now: Date) {
|
||||||
|
let cutoff = now.addingTimeInterval(-TransportConfig.bleRecentPacketWindowSeconds)
|
||||||
|
if packetTimestamps.count > TransportConfig.bleRecentPacketWindowMaxCount {
|
||||||
|
packetTimestamps.removeFirst(packetTimestamps.count - TransportConfig.bleRecentPacketWindowMaxCount)
|
||||||
|
}
|
||||||
|
packetTimestamps.removeAll { $0 < cutoff }
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,93 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLERouteForwardingPlan {
|
||||||
|
let shouldSuppressFloodRelay: Bool
|
||||||
|
let forwardPacket: BitchatPacket?
|
||||||
|
let nextHop: PeerID?
|
||||||
|
|
||||||
|
static let allowFloodRelay = BLERouteForwardingPlan(
|
||||||
|
shouldSuppressFloodRelay: false,
|
||||||
|
forwardPacket: nil,
|
||||||
|
nextHop: nil
|
||||||
|
)
|
||||||
|
|
||||||
|
static let suppressFloodRelay = BLERouteForwardingPlan(
|
||||||
|
shouldSuppressFloodRelay: true,
|
||||||
|
forwardPacket: nil,
|
||||||
|
nextHop: nil
|
||||||
|
)
|
||||||
|
|
||||||
|
static func forward(_ packet: BitchatPacket, to nextHop: PeerID) -> BLERouteForwardingPlan {
|
||||||
|
BLERouteForwardingPlan(
|
||||||
|
shouldSuppressFloodRelay: true,
|
||||||
|
forwardPacket: packet,
|
||||||
|
nextHop: nextHop
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLERouteForwardingPolicy {
|
||||||
|
static func plan(
|
||||||
|
for packet: BitchatPacket,
|
||||||
|
localPeerID: PeerID,
|
||||||
|
localRoutingData: Data?,
|
||||||
|
routingPeer: (Data) -> PeerID?,
|
||||||
|
isPeerConnected: (PeerID) -> Bool
|
||||||
|
) -> BLERouteForwardingPlan {
|
||||||
|
if PeerID(hexData: packet.recipientID) == localPeerID {
|
||||||
|
return .suppressFloodRelay
|
||||||
|
}
|
||||||
|
|
||||||
|
guard let route = packet.route, !route.isEmpty else {
|
||||||
|
return .allowFloodRelay
|
||||||
|
}
|
||||||
|
|
||||||
|
guard packet.ttl > 1 else {
|
||||||
|
return .suppressFloodRelay
|
||||||
|
}
|
||||||
|
|
||||||
|
guard let localRoutingData else {
|
||||||
|
return .allowFloodRelay
|
||||||
|
}
|
||||||
|
|
||||||
|
guard let localIndex = route.firstIndex(of: localRoutingData) else {
|
||||||
|
return forward(packet, toRouteData: route[0], routingPeer: routingPeer, isPeerConnected: isPeerConnected)
|
||||||
|
}
|
||||||
|
|
||||||
|
if localIndex == route.count - 1 {
|
||||||
|
guard let destinationPeer = PeerID(hexData: packet.recipientID),
|
||||||
|
isPeerConnected(destinationPeer) else {
|
||||||
|
return .allowFloodRelay
|
||||||
|
}
|
||||||
|
return .forward(relayed(packet), to: destinationPeer)
|
||||||
|
}
|
||||||
|
|
||||||
|
return forward(
|
||||||
|
packet,
|
||||||
|
toRouteData: route[localIndex + 1],
|
||||||
|
routingPeer: routingPeer,
|
||||||
|
isPeerConnected: isPeerConnected
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func forward(
|
||||||
|
_ packet: BitchatPacket,
|
||||||
|
toRouteData routeData: Data,
|
||||||
|
routingPeer: (Data) -> PeerID?,
|
||||||
|
isPeerConnected: (PeerID) -> Bool
|
||||||
|
) -> BLERouteForwardingPlan {
|
||||||
|
guard let nextPeer = routingPeer(routeData),
|
||||||
|
isPeerConnected(nextPeer) else {
|
||||||
|
return .allowFloodRelay
|
||||||
|
}
|
||||||
|
|
||||||
|
return .forward(relayed(packet), to: nextPeer)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func relayed(_ packet: BitchatPacket) -> BitchatPacket {
|
||||||
|
var relayPacket = packet
|
||||||
|
relayPacket.ttl = packet.ttl - 1
|
||||||
|
return relayPacket
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,35 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum BLEScanDutyPlan: Equatable {
|
||||||
|
case continuous
|
||||||
|
case dutyCycle(onDuration: TimeInterval, offDuration: TimeInterval)
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BLEScanDutyPolicy {
|
||||||
|
static func plan(
|
||||||
|
dutyEnabled: Bool,
|
||||||
|
appIsActive: Bool,
|
||||||
|
connectedCount: Int,
|
||||||
|
hasRecentTraffic: Bool,
|
||||||
|
highDegreeThreshold: Int = TransportConfig.bleHighDegreeThreshold
|
||||||
|
) -> BLEScanDutyPlan {
|
||||||
|
let forceContinuousScan = connectedCount <= 2 || hasRecentTraffic
|
||||||
|
let shouldDutyCycle = dutyEnabled && appIsActive && connectedCount > 0 && !forceContinuousScan
|
||||||
|
|
||||||
|
guard shouldDutyCycle else {
|
||||||
|
return .continuous
|
||||||
|
}
|
||||||
|
|
||||||
|
if connectedCount >= highDegreeThreshold {
|
||||||
|
return .dutyCycle(
|
||||||
|
onDuration: TransportConfig.bleDutyOnDurationDense,
|
||||||
|
offDuration: TransportConfig.bleDutyOffDurationDense
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
return .dutyCycle(
|
||||||
|
onDuration: TransportConfig.bleDutyOnDuration,
|
||||||
|
offDuration: TransportConfig.bleDutyOffDuration
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,43 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLEScheduledRelayStore {
|
||||||
|
private var relays: [String: DispatchWorkItem] = [:]
|
||||||
|
|
||||||
|
var count: Int {
|
||||||
|
relays.count
|
||||||
|
}
|
||||||
|
|
||||||
|
var isEmpty: Bool {
|
||||||
|
relays.isEmpty
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func schedule(_ workItem: DispatchWorkItem, messageID: String) {
|
||||||
|
relays[messageID] = workItem
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
mutating func remove(messageID: String) -> DispatchWorkItem? {
|
||||||
|
relays.removeValue(forKey: messageID)
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
mutating func cancel(messageID: String) -> Bool {
|
||||||
|
guard let workItem = relays.removeValue(forKey: messageID) else {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
workItem.cancel()
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func cancelAll() {
|
||||||
|
relays.values.forEach { $0.cancel() }
|
||||||
|
relays.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAllIfOverCapacity(_ maxCount: Int) {
|
||||||
|
if relays.count > maxCount {
|
||||||
|
relays.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,40 @@
|
|||||||
|
import BitFoundation
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct BLESelfBroadcastTracker {
|
||||||
|
private struct Entry {
|
||||||
|
let messageID: String
|
||||||
|
let sentAt: Date
|
||||||
|
}
|
||||||
|
|
||||||
|
private var entriesByDedupID: [String: Entry] = [:]
|
||||||
|
|
||||||
|
var isEmpty: Bool {
|
||||||
|
entriesByDedupID.isEmpty
|
||||||
|
}
|
||||||
|
|
||||||
|
var count: Int {
|
||||||
|
entriesByDedupID.count
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func record(messageID: String, packet: BitchatPacket, sentAt: Date) {
|
||||||
|
entriesByDedupID[Self.dedupID(for: packet)] = Entry(messageID: messageID, sentAt: sentAt)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func takeMessageID(for packet: BitchatPacket) -> String? {
|
||||||
|
entriesByDedupID.removeValue(forKey: Self.dedupID(for: packet))?.messageID
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func prune(before cutoff: Date) {
|
||||||
|
guard !entriesByDedupID.isEmpty else { return }
|
||||||
|
entriesByDedupID = entriesByDedupID.filter { cutoff <= $0.value.sentAt }
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
entriesByDedupID.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
static func dedupID(for packet: BitchatPacket) -> String {
|
||||||
|
"\(packet.senderID.hexEncodedString())-\(packet.timestamp)-\(packet.type)"
|
||||||
|
}
|
||||||
|
}
|
||||||
+1262
-2371
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,71 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
enum BLESubscriptionAnnounceDecision: Equatable {
|
||||||
|
case allowed
|
||||||
|
case rateLimited(backoffSeconds: TimeInterval, attemptCount: Int, suppressAnnounce: Bool)
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BLESubscriptionAnnounceLimiter {
|
||||||
|
private struct State {
|
||||||
|
var lastAnnounceTime: Date
|
||||||
|
var attemptCount: Int
|
||||||
|
var currentBackoffSeconds: TimeInterval
|
||||||
|
}
|
||||||
|
|
||||||
|
private var states: [String: State] = [:]
|
||||||
|
|
||||||
|
var trackedCentralCount: Int {
|
||||||
|
states.count
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeAll() {
|
||||||
|
states.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func decision(for centralID: String, now: Date) -> BLESubscriptionAnnounceDecision {
|
||||||
|
pruneStaleEntries(now: now)
|
||||||
|
|
||||||
|
guard let existing = states[centralID] else {
|
||||||
|
recordAllowedAttempt(for: centralID, now: now)
|
||||||
|
return .allowed
|
||||||
|
}
|
||||||
|
|
||||||
|
let timeSinceLastAnnounce = now.timeIntervalSince(existing.lastAnnounceTime)
|
||||||
|
guard timeSinceLastAnnounce < existing.currentBackoffSeconds else {
|
||||||
|
recordAllowedAttempt(for: centralID, now: now)
|
||||||
|
return .allowed
|
||||||
|
}
|
||||||
|
|
||||||
|
let newAttemptCount = existing.attemptCount + 1
|
||||||
|
let newBackoff = min(
|
||||||
|
existing.currentBackoffSeconds * TransportConfig.bleSubscriptionRateLimitBackoffFactor,
|
||||||
|
TransportConfig.bleSubscriptionRateLimitMaxBackoffSeconds
|
||||||
|
)
|
||||||
|
states[centralID] = State(
|
||||||
|
lastAnnounceTime: now,
|
||||||
|
attemptCount: newAttemptCount,
|
||||||
|
currentBackoffSeconds: newBackoff
|
||||||
|
)
|
||||||
|
|
||||||
|
return .rateLimited(
|
||||||
|
backoffSeconds: existing.currentBackoffSeconds,
|
||||||
|
attemptCount: existing.attemptCount,
|
||||||
|
suppressAnnounce: newAttemptCount >= TransportConfig.bleSubscriptionRateLimitMaxAttempts
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private mutating func recordAllowedAttempt(for centralID: String, now: Date) {
|
||||||
|
states[centralID] = State(
|
||||||
|
lastAnnounceTime: now,
|
||||||
|
attemptCount: 1,
|
||||||
|
currentBackoffSeconds: TransportConfig.bleSubscriptionRateLimitMinSeconds
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private mutating func pruneStaleEntries(now: Date) {
|
||||||
|
let windowSeconds = TransportConfig.bleSubscriptionRateLimitWindowSeconds
|
||||||
|
states = states.filter { _, state in
|
||||||
|
now.timeIntervalSince(state.lastAnnounceTime) < windowSeconds
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -6,7 +6,6 @@
|
|||||||
// This is free and unencumbered software released into the public domain.
|
// This is free and unencumbered software released into the public domain.
|
||||||
//
|
//
|
||||||
|
|
||||||
import Nostr
|
|
||||||
import Foundation
|
import Foundation
|
||||||
import BitFoundation
|
import BitFoundation
|
||||||
|
|
||||||
@@ -29,9 +28,9 @@ struct CommandGeoParticipant {
|
|||||||
protocol CommandContextProvider: AnyObject {
|
protocol CommandContextProvider: AnyObject {
|
||||||
// MARK: - State Properties
|
// MARK: - State Properties
|
||||||
var nickname: String { get }
|
var nickname: String { get }
|
||||||
|
var activeChannel: ChannelID { get }
|
||||||
var selectedPrivateChatPeer: PeerID? { get }
|
var selectedPrivateChatPeer: PeerID? { get }
|
||||||
var blockedUsers: Set<String> { get }
|
var blockedUsers: Set<String> { get }
|
||||||
var privateChats: [PeerID: [BitchatMessage]] { get set }
|
|
||||||
var idBridge: NostrIdentityBridge { get }
|
var idBridge: NostrIdentityBridge { get }
|
||||||
|
|
||||||
// MARK: - Peer Lookup
|
// MARK: - Peer Lookup
|
||||||
@@ -43,6 +42,8 @@ protocol CommandContextProvider: AnyObject {
|
|||||||
func startPrivateChat(with peerID: PeerID)
|
func startPrivateChat(with peerID: PeerID)
|
||||||
func sendPrivateMessage(_ content: String, to peerID: PeerID)
|
func sendPrivateMessage(_ content: String, to peerID: PeerID)
|
||||||
func clearCurrentPublicTimeline()
|
func clearCurrentPublicTimeline()
|
||||||
|
/// Empties the peer's chat (single-writer store intent for `/clear`).
|
||||||
|
func clearPrivateChat(_ peerID: PeerID)
|
||||||
func sendPublicRaw(_ content: String)
|
func sendPublicRaw(_ content: String)
|
||||||
|
|
||||||
// MARK: - System Messages
|
// MARK: - System Messages
|
||||||
@@ -76,7 +77,7 @@ final class CommandProcessor {
|
|||||||
|
|
||||||
// Geohash context: disable favoriting in public geohash or GeoDM
|
// Geohash context: disable favoriting in public geohash or GeoDM
|
||||||
let inGeoPublic: Bool = {
|
let inGeoPublic: Bool = {
|
||||||
switch LocationChannelManager.shared.selectedChannel {
|
switch contextProvider?.activeChannel ?? .mesh {
|
||||||
case .mesh: return false
|
case .mesh: return false
|
||||||
case .location: return true
|
case .location: return true
|
||||||
}
|
}
|
||||||
@@ -136,7 +137,7 @@ final class CommandProcessor {
|
|||||||
|
|
||||||
private func handleWho() -> CommandResult {
|
private func handleWho() -> CommandResult {
|
||||||
// Show geohash participants when in a geohash channel; otherwise mesh peers
|
// Show geohash participants when in a geohash channel; otherwise mesh peers
|
||||||
switch LocationChannelManager.shared.selectedChannel {
|
switch contextProvider?.activeChannel ?? .mesh {
|
||||||
case .location(let ch):
|
case .location(let ch):
|
||||||
// Geohash context: show visible geohash participants (exclude self)
|
// Geohash context: show visible geohash participants (exclude self)
|
||||||
guard let vm = contextProvider else { return .success(message: "nobody around") }
|
guard let vm = contextProvider else { return .success(message: "nobody around") }
|
||||||
@@ -160,7 +161,7 @@ final class CommandProcessor {
|
|||||||
|
|
||||||
private func handleClear() -> CommandResult {
|
private func handleClear() -> CommandResult {
|
||||||
if let peerID = contextProvider?.selectedPrivateChatPeer {
|
if let peerID = contextProvider?.selectedPrivateChatPeer {
|
||||||
contextProvider?.privateChats[peerID]?.removeAll()
|
contextProvider?.clearPrivateChat(peerID)
|
||||||
} else {
|
} else {
|
||||||
contextProvider?.clearCurrentPublicTimeline()
|
contextProvider?.clearCurrentPublicTimeline()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -34,7 +34,23 @@ final class FavoritesPersistenceService: ObservableObject {
|
|||||||
|
|
||||||
static let shared = FavoritesPersistenceService()
|
static let shared = FavoritesPersistenceService()
|
||||||
|
|
||||||
init(keychain: KeychainManagerProtocol = KeychainManager()) {
|
/// Default keychain for the `shared` singleton. Under test this is an
|
||||||
|
/// in-memory keychain so touching `shared` never blocks on securityd
|
||||||
|
/// (`SecItemCopyMatching` can hang in test environments) and never reads
|
||||||
|
/// or writes the developer's real keychain. Production behavior is
|
||||||
|
/// unchanged. Tests that need their own instance keep injecting a mock
|
||||||
|
/// via `init(keychain:)`.
|
||||||
|
private nonisolated static func makeDefaultKeychain() -> KeychainManagerProtocol {
|
||||||
|
// PreviewKeychainManager lives in _PreviewHelpers, a development
|
||||||
|
// asset excluded from archive builds — release code must not
|
||||||
|
// reference it. Tests always run Debug, so the guard is lossless.
|
||||||
|
#if DEBUG
|
||||||
|
if TestEnvironment.isRunningTests { return PreviewKeychainManager() }
|
||||||
|
#endif
|
||||||
|
return KeychainManager()
|
||||||
|
}
|
||||||
|
|
||||||
|
init(keychain: KeychainManagerProtocol = FavoritesPersistenceService.makeDefaultKeychain()) {
|
||||||
self.keychain = keychain
|
self.keychain = keychain
|
||||||
loadFavorites()
|
loadFavorites()
|
||||||
|
|
||||||
|
|||||||
@@ -8,7 +8,6 @@
|
|||||||
// This is free and unencumbered software released into the public domain.
|
// This is free and unencumbered software released into the public domain.
|
||||||
//
|
//
|
||||||
|
|
||||||
import Nostr
|
|
||||||
import Foundation
|
import Foundation
|
||||||
import Combine
|
import Combine
|
||||||
import BitLogger
|
import BitLogger
|
||||||
@@ -75,7 +74,7 @@ final class GeohashPresenceService: ObservableObject {
|
|||||||
]
|
]
|
||||||
|
|
||||||
private init() {
|
private init() {
|
||||||
let idBridge = NostrIdentityBridge(keychain: KeychainManager())
|
let idBridge = NostrIdentityBridge()
|
||||||
self.availableChannelsProvider = { LocationStateManager.shared.availableChannels }
|
self.availableChannelsProvider = { LocationStateManager.shared.availableChannels }
|
||||||
self.locationChanges = LocationStateManager.shared.$availableChannels.eraseToAnyPublisher()
|
self.locationChanges = LocationStateManager.shared.$availableChannels.eraseToAnyPublisher()
|
||||||
self.torReadyPublisher = NotificationCenter.default.publisher(for: .TorDidBecomeReady)
|
self.torReadyPublisher = NotificationCenter.default.publisher(for: .TorDidBecomeReady)
|
||||||
|
|||||||
@@ -7,78 +7,10 @@
|
|||||||
//
|
//
|
||||||
|
|
||||||
import BitLogger
|
import BitLogger
|
||||||
import protocol Nostr.NostrKeychainStoring
|
import BitFoundation
|
||||||
import protocol Noise.SecureMemoryCleaner
|
|
||||||
import Foundation
|
import Foundation
|
||||||
import Security
|
import Security
|
||||||
|
|
||||||
// MARK: - Keychain Error Types
|
|
||||||
// BCH-01-009: Proper error classification to distinguish expected states from critical failures
|
|
||||||
|
|
||||||
/// Result of a keychain read operation with proper error classification
|
|
||||||
enum KeychainReadResult {
|
|
||||||
case success(Data)
|
|
||||||
case itemNotFound // Expected: key doesn't exist yet
|
|
||||||
case accessDenied // Critical: app lacks keychain access
|
|
||||||
case deviceLocked // Recoverable: device is locked
|
|
||||||
case authenticationFailed // Recoverable: biometric/passcode failed
|
|
||||||
case otherError(OSStatus) // Unexpected error
|
|
||||||
|
|
||||||
var isRecoverableError: Bool {
|
|
||||||
switch self {
|
|
||||||
case .deviceLocked, .authenticationFailed:
|
|
||||||
return true
|
|
||||||
default:
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Result of a keychain save operation with proper error classification
|
|
||||||
enum KeychainSaveResult {
|
|
||||||
case success
|
|
||||||
case duplicateItem // Can retry with update
|
|
||||||
case accessDenied // Critical: app lacks keychain access
|
|
||||||
case deviceLocked // Recoverable: device is locked
|
|
||||||
case storageFull // Critical: no space available
|
|
||||||
case otherError(OSStatus)
|
|
||||||
|
|
||||||
var isRecoverableError: Bool {
|
|
||||||
switch self {
|
|
||||||
case .duplicateItem, .deviceLocked:
|
|
||||||
return true
|
|
||||||
default:
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
protocol KeychainManagerProtocol: SecureMemoryCleaner, NostrKeychainStoring {
|
|
||||||
func saveIdentityKey(_ keyData: Data, forKey key: String) -> Bool
|
|
||||||
func getIdentityKey(forKey key: String) -> Data?
|
|
||||||
func deleteIdentityKey(forKey key: String) -> Bool
|
|
||||||
func deleteAllKeychainData() -> Bool
|
|
||||||
|
|
||||||
func secureClear(_ data: inout Data)
|
|
||||||
func secureClear(_ string: inout String)
|
|
||||||
|
|
||||||
func verifyIdentityKeyExists() -> Bool
|
|
||||||
|
|
||||||
// BCH-01-009: Methods with proper error classification
|
|
||||||
/// Get identity key with detailed result for error handling
|
|
||||||
func getIdentityKeyWithResult(forKey key: String) -> KeychainReadResult
|
|
||||||
/// Save identity key with detailed result for error handling
|
|
||||||
func saveIdentityKeyWithResult(_ keyData: Data, forKey key: String) -> KeychainSaveResult
|
|
||||||
|
|
||||||
// MARK: - Generic Data Storage (consolidated from KeychainHelper)
|
|
||||||
/// Save data with a custom service name
|
|
||||||
func save(key: String, data: Data, service: String, accessible: CFString?)
|
|
||||||
/// Load data from a custom service
|
|
||||||
func load(key: String, service: String) -> Data?
|
|
||||||
/// Delete data from a custom service
|
|
||||||
func delete(key: String, service: String)
|
|
||||||
}
|
|
||||||
|
|
||||||
final class KeychainManager: KeychainManagerProtocol {
|
final class KeychainManager: KeychainManagerProtocol {
|
||||||
// Use consistent service name for all keychain items
|
// Use consistent service name for all keychain items
|
||||||
private let service = BitchatApp.bundleID
|
private let service = BitchatApp.bundleID
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import BitLogger
|
import BitLogger
|
||||||
import Nostr
|
import Combine
|
||||||
import Foundation
|
import Foundation
|
||||||
|
|
||||||
/// Dependencies for location notes, allowing tests to stub relay/identity behavior.
|
/// Dependencies for location notes, allowing tests to stub relay/identity behavior.
|
||||||
@@ -15,8 +15,10 @@ struct LocationNotesDependencies {
|
|||||||
var sendEvent: SendEvent
|
var sendEvent: SendEvent
|
||||||
var deriveIdentity: (_ geohash: String) throws -> NostrIdentity
|
var deriveIdentity: (_ geohash: String) throws -> NostrIdentity
|
||||||
var now: () -> Date
|
var now: () -> Date
|
||||||
|
// Fires when the geo relay directory refreshes; used to retry after "no relays".
|
||||||
private static let idBridge = NostrIdentityBridge(keychain: KeychainManager())
|
var relayDirectoryUpdates: AnyPublisher<Void, Never> = Empty(completeImmediately: false).eraseToAnyPublisher()
|
||||||
|
|
||||||
|
private static let idBridge = NostrIdentityBridge()
|
||||||
|
|
||||||
static let live = LocationNotesDependencies(
|
static let live = LocationNotesDependencies(
|
||||||
relayLookup: { geohash, count in
|
relayLookup: { geohash, count in
|
||||||
@@ -40,7 +42,11 @@ struct LocationNotesDependencies {
|
|||||||
deriveIdentity: { geohash in
|
deriveIdentity: { geohash in
|
||||||
try idBridge.deriveIdentity(forGeohash: geohash)
|
try idBridge.deriveIdentity(forGeohash: geohash)
|
||||||
},
|
},
|
||||||
now: { Date() }
|
now: { Date() },
|
||||||
|
relayDirectoryUpdates: NotificationCenter.default
|
||||||
|
.publisher(for: .geoRelayDirectoryDidRefresh)
|
||||||
|
.map { _ in () }
|
||||||
|
.eraseToAnyPublisher()
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -78,6 +84,7 @@ final class LocationNotesManager: ObservableObject {
|
|||||||
@Published private(set) var errorMessage: String?
|
@Published private(set) var errorMessage: String?
|
||||||
private var subscriptionID: String?
|
private var subscriptionID: String?
|
||||||
private var noteIDs = Set<String>() // O(1) duplicate detection
|
private var noteIDs = Set<String>() // O(1) duplicate detection
|
||||||
|
private var directoryUpdateCancellable: AnyCancellable?
|
||||||
private let dependencies: LocationNotesDependencies
|
private let dependencies: LocationNotesDependencies
|
||||||
private let maxNotesInMemory = 500 // Defensive cap (relay limit is 200)
|
private let maxNotesInMemory = 500 // Defensive cap (relay limit is 200)
|
||||||
|
|
||||||
@@ -102,6 +109,15 @@ final class LocationNotesManager: ObservableObject {
|
|||||||
SecureLogger.warning("LocationNotesManager: invalid geohash '\(norm)' (expected 8 valid base32 chars)", category: .session)
|
SecureLogger.warning("LocationNotesManager: invalid geohash '\(norm)' (expected 8 valid base32 chars)", category: .session)
|
||||||
}
|
}
|
||||||
subscribe()
|
subscribe()
|
||||||
|
// The relay directory may load after init (remote fetch over Tor);
|
||||||
|
// retry automatically instead of staying stuck on "no relays".
|
||||||
|
directoryUpdateCancellable = dependencies.relayDirectoryUpdates
|
||||||
|
.sink { [weak self] in
|
||||||
|
Task { @MainActor [weak self] in
|
||||||
|
guard let self, self.state == .noRelays else { return }
|
||||||
|
self.subscribe()
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func setGeohash(_ newGeohash: String) {
|
func setGeohash(_ newGeohash: String) {
|
||||||
|
|||||||
@@ -6,6 +6,13 @@ import Foundation
|
|||||||
@MainActor
|
@MainActor
|
||||||
final class MessageRouter {
|
final class MessageRouter {
|
||||||
private let transports: [Transport]
|
private let transports: [Transport]
|
||||||
|
private let now: () -> Date
|
||||||
|
|
||||||
|
/// Invoked whenever a retained private message is dropped without a
|
||||||
|
/// delivery ack (attempt cap, TTL expiry, or per-peer overflow eviction)
|
||||||
|
/// so the UI can surface the failure instead of leaving the message in a
|
||||||
|
/// stale "sending/sent" state forever.
|
||||||
|
var onMessageDropped: ((_ messageID: String, _ peerID: PeerID) -> Void)?
|
||||||
|
|
||||||
// Outbox entry with timestamp for TTL-based eviction
|
// Outbox entry with timestamp for TTL-based eviction
|
||||||
private struct QueuedMessage {
|
private struct QueuedMessage {
|
||||||
@@ -13,6 +20,7 @@ final class MessageRouter {
|
|||||||
let nickname: String
|
let nickname: String
|
||||||
let messageID: String
|
let messageID: String
|
||||||
let timestamp: Date
|
let timestamp: Date
|
||||||
|
var sendAttempts: Int = 0
|
||||||
}
|
}
|
||||||
|
|
||||||
private var outbox: [PeerID: [QueuedMessage]] = [:]
|
private var outbox: [PeerID: [QueuedMessage]] = [:]
|
||||||
@@ -20,9 +28,13 @@ final class MessageRouter {
|
|||||||
// Outbox limits to prevent unbounded memory growth
|
// Outbox limits to prevent unbounded memory growth
|
||||||
private static let maxMessagesPerPeer = 100
|
private static let maxMessagesPerPeer = 100
|
||||||
private static let messageTTLSeconds: TimeInterval = 24 * 60 * 60 // 24 hours
|
private static let messageTTLSeconds: TimeInterval = 24 * 60 * 60 // 24 hours
|
||||||
|
// Bound resends of messages sent on a weak reachability signal that never
|
||||||
|
// get a delivery ack (e.g. peer on an old client that doesn't ack).
|
||||||
|
private static let maxSendAttempts = 8
|
||||||
|
|
||||||
init(transports: [Transport]) {
|
init(transports: [Transport], now: @escaping () -> Date = Date.init) {
|
||||||
self.transports = transports
|
self.transports = transports
|
||||||
|
self.now = now
|
||||||
|
|
||||||
// Observe favorites changes to learn Nostr mapping and flush queued messages
|
// Observe favorites changes to learn Nostr mapping and flush queued messages
|
||||||
NotificationCenter.default.addObserver(
|
NotificationCenter.default.addObserver(
|
||||||
@@ -61,26 +73,54 @@ final class MessageRouter {
|
|||||||
// MARK: - Message Sending
|
// MARK: - Message Sending
|
||||||
|
|
||||||
func sendPrivate(_ content: String, to peerID: PeerID, recipientNickname: String, messageID: String) {
|
func sendPrivate(_ content: String, to peerID: PeerID, recipientNickname: String, messageID: String) {
|
||||||
if let transport = reachableTransport(for: peerID) {
|
if let transport = connectedTransport(for: peerID) {
|
||||||
SecureLogger.debug("Routing PM via \(type(of: transport)) to \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
// A live link is a strong delivery signal; trust it outright.
|
||||||
|
SecureLogger.debug("Routing PM via \(type(of: transport)) (connected) to \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
||||||
transport.sendPrivateMessage(content, to: peerID, recipientNickname: recipientNickname, messageID: messageID)
|
transport.sendPrivateMessage(content, to: peerID, recipientNickname: recipientNickname, messageID: messageID)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let message = QueuedMessage(content: content, nickname: recipientNickname, messageID: messageID, timestamp: now(), sendAttempts: 1)
|
||||||
|
if let transport = reachableTransport(for: peerID) {
|
||||||
|
// Reachability without a connection is a freshness heuristic (e.g.
|
||||||
|
// the mesh retention window), so the send can silently go nowhere.
|
||||||
|
// Send now, but retain a copy until a delivery/read ack clears it;
|
||||||
|
// receivers dedup resends by message ID.
|
||||||
|
SecureLogger.debug("Routing PM via \(type(of: transport)) (reachable) to \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
||||||
|
transport.sendPrivateMessage(content, to: peerID, recipientNickname: recipientNickname, messageID: messageID)
|
||||||
|
enqueue(message, for: peerID)
|
||||||
} else {
|
} else {
|
||||||
// Queue for later with timestamp for TTL tracking
|
var unsent = message
|
||||||
if outbox[peerID] == nil { outbox[peerID] = [] }
|
unsent.sendAttempts = 0
|
||||||
|
enqueue(unsent, for: peerID)
|
||||||
let message = QueuedMessage(content: content, nickname: recipientNickname, messageID: messageID, timestamp: Date())
|
|
||||||
outbox[peerID]?.append(message)
|
|
||||||
|
|
||||||
// Enforce per-peer size limit with FIFO eviction
|
|
||||||
if let count = outbox[peerID]?.count, count > Self.maxMessagesPerPeer {
|
|
||||||
let evicted = outbox[peerID]?.removeFirst()
|
|
||||||
SecureLogger.warning("📤 Outbox overflow for \(peerID.id.prefix(8))… - evicted oldest message: \(evicted?.messageID.prefix(8) ?? "?")…", category: .session)
|
|
||||||
}
|
|
||||||
|
|
||||||
SecureLogger.debug("Queued PM for \(peerID.id.prefix(8))… (no reachable transport) id=\(messageID.prefix(8))… queue=\(outbox[peerID]?.count ?? 0)", category: .session)
|
SecureLogger.debug("Queued PM for \(peerID.id.prefix(8))… (no reachable transport) id=\(messageID.prefix(8))… queue=\(outbox[peerID]?.count ?? 0)", category: .session)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// A delivery or read ack confirms receipt; stop retaining the message.
|
||||||
|
func markDelivered(_ messageID: String) {
|
||||||
|
for (peerID, queue) in outbox {
|
||||||
|
let filtered = queue.filter { $0.messageID != messageID }
|
||||||
|
guard filtered.count != queue.count else { continue }
|
||||||
|
outbox[peerID] = filtered.isEmpty ? nil : filtered
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func enqueue(_ message: QueuedMessage, for peerID: PeerID) {
|
||||||
|
var queue = outbox[peerID] ?? []
|
||||||
|
// Re-sending an already-queued ID replaces the entry (keeps attempt count fresh)
|
||||||
|
queue.removeAll { $0.messageID == message.messageID }
|
||||||
|
queue.append(message)
|
||||||
|
|
||||||
|
// Enforce per-peer size limit with FIFO eviction
|
||||||
|
if queue.count > Self.maxMessagesPerPeer {
|
||||||
|
let evicted = queue.removeFirst()
|
||||||
|
SecureLogger.warning("📤 Outbox overflow for \(peerID.id.prefix(8))… - evicted oldest message: \(evicted.messageID.prefix(8))…", category: .session)
|
||||||
|
onMessageDropped?(evicted.messageID, peerID)
|
||||||
|
}
|
||||||
|
outbox[peerID] = queue
|
||||||
|
}
|
||||||
|
|
||||||
func sendReadReceipt(_ receipt: ReadReceipt, to peerID: PeerID) {
|
func sendReadReceipt(_ receipt: ReadReceipt, to peerID: PeerID) {
|
||||||
if let transport = reachableTransport(for: peerID) {
|
if let transport = reachableTransport(for: peerID) {
|
||||||
SecureLogger.debug("Routing READ ack via \(type(of: transport)) to \(peerID.id.prefix(8))… id=\(receipt.originalMessageID.prefix(8))…", category: .session)
|
SecureLogger.debug("Routing READ ack via \(type(of: transport)) to \(peerID.id.prefix(8))… id=\(receipt.originalMessageID.prefix(8))…", category: .session)
|
||||||
@@ -111,19 +151,34 @@ final class MessageRouter {
|
|||||||
guard let queued = outbox[peerID], !queued.isEmpty else { return }
|
guard let queued = outbox[peerID], !queued.isEmpty else { return }
|
||||||
SecureLogger.debug("Flushing outbox for \(peerID.id.prefix(8))… count=\(queued.count)", category: .session)
|
SecureLogger.debug("Flushing outbox for \(peerID.id.prefix(8))… count=\(queued.count)", category: .session)
|
||||||
|
|
||||||
let now = Date()
|
let now = now()
|
||||||
var remaining: [QueuedMessage] = []
|
var remaining: [QueuedMessage] = []
|
||||||
|
|
||||||
for message in queued {
|
for message in queued {
|
||||||
// Skip expired messages (TTL exceeded)
|
// Skip expired messages (TTL exceeded)
|
||||||
if now.timeIntervalSince(message.timestamp) > Self.messageTTLSeconds {
|
if now.timeIntervalSince(message.timestamp) > Self.messageTTLSeconds {
|
||||||
SecureLogger.debug("⏰ Expired queued message for \(peerID.id.prefix(8))… id=\(message.messageID.prefix(8))… (age: \(Int(now.timeIntervalSince(message.timestamp)))s)", category: .session)
|
SecureLogger.debug("⏰ Expired queued message for \(peerID.id.prefix(8))… id=\(message.messageID.prefix(8))… (age: \(Int(now.timeIntervalSince(message.timestamp)))s)", category: .session)
|
||||||
|
onMessageDropped?(message.messageID, peerID)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
if let transport = reachableTransport(for: peerID) {
|
if let transport = connectedTransport(for: peerID) {
|
||||||
SecureLogger.debug("Outbox -> \(type(of: transport)) for \(peerID.id.prefix(8))… id=\(message.messageID.prefix(8))…", category: .session)
|
// Live link: send and stop retaining.
|
||||||
|
SecureLogger.debug("Outbox -> \(type(of: transport)) (connected) for \(peerID.id.prefix(8))… id=\(message.messageID.prefix(8))…", category: .session)
|
||||||
transport.sendPrivateMessage(message.content, to: peerID, recipientNickname: message.nickname, messageID: message.messageID)
|
transport.sendPrivateMessage(message.content, to: peerID, recipientNickname: message.nickname, messageID: message.messageID)
|
||||||
|
} else if let transport = reachableTransport(for: peerID) {
|
||||||
|
// Weak signal: send but keep retaining until an ack clears it,
|
||||||
|
// bounded by attempt count for peers that never ack.
|
||||||
|
guard message.sendAttempts < Self.maxSendAttempts else {
|
||||||
|
SecureLogger.warning("📤 Dropping unacked PM for \(peerID.id.prefix(8))… id=\(message.messageID.prefix(8))… after \(message.sendAttempts) attempts", category: .session)
|
||||||
|
onMessageDropped?(message.messageID, peerID)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
SecureLogger.debug("Outbox -> \(type(of: transport)) (reachable) for \(peerID.id.prefix(8))… id=\(message.messageID.prefix(8))…", category: .session)
|
||||||
|
transport.sendPrivateMessage(message.content, to: peerID, recipientNickname: message.nickname, messageID: message.messageID)
|
||||||
|
var retained = message
|
||||||
|
retained.sendAttempts += 1
|
||||||
|
remaining.append(retained)
|
||||||
} else {
|
} else {
|
||||||
remaining.append(message)
|
remaining.append(message)
|
||||||
}
|
}
|
||||||
@@ -142,12 +197,21 @@ final class MessageRouter {
|
|||||||
|
|
||||||
/// Periodically clean up expired messages from all outboxes
|
/// Periodically clean up expired messages from all outboxes
|
||||||
func cleanupExpiredMessages() {
|
func cleanupExpiredMessages() {
|
||||||
let now = Date()
|
let now = now()
|
||||||
for peerID in Array(outbox.keys) {
|
for peerID in Array(outbox.keys) {
|
||||||
outbox[peerID]?.removeAll { now.timeIntervalSince($0.timestamp) > Self.messageTTLSeconds }
|
var expiredMessageIDs: [String] = []
|
||||||
|
outbox[peerID]?.removeAll { message in
|
||||||
|
guard now.timeIntervalSince(message.timestamp) > Self.messageTTLSeconds else { return false }
|
||||||
|
expiredMessageIDs.append(message.messageID)
|
||||||
|
return true
|
||||||
|
}
|
||||||
if outbox[peerID]?.isEmpty == true {
|
if outbox[peerID]?.isEmpty == true {
|
||||||
outbox.removeValue(forKey: peerID)
|
outbox.removeValue(forKey: peerID)
|
||||||
}
|
}
|
||||||
|
for messageID in expiredMessageIDs {
|
||||||
|
SecureLogger.debug("⏰ Expired queued message for \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
||||||
|
onMessageDropped?(messageID, peerID)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,4 +1,3 @@
|
|||||||
import Nostr
|
|
||||||
import Foundation
|
import Foundation
|
||||||
import BitLogger
|
import BitLogger
|
||||||
import Combine
|
import Combine
|
||||||
|
|||||||
@@ -84,7 +84,6 @@
|
|||||||
|
|
||||||
import BitLogger
|
import BitLogger
|
||||||
import BitFoundation
|
import BitFoundation
|
||||||
import Noise
|
|
||||||
import Foundation
|
import Foundation
|
||||||
import CryptoKit
|
import CryptoKit
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
import BitLogger
|
import BitLogger
|
||||||
import BitFoundation
|
import BitFoundation
|
||||||
import Nostr
|
|
||||||
import Foundation
|
import Foundation
|
||||||
import Combine
|
import Combine
|
||||||
|
|
||||||
@@ -107,6 +106,7 @@ final class NostrTransport: Transport, @unchecked Sendable {
|
|||||||
// MARK: - Transport Protocol Conformance
|
// MARK: - Transport Protocol Conformance
|
||||||
|
|
||||||
weak var delegate: BitchatDelegate?
|
weak var delegate: BitchatDelegate?
|
||||||
|
weak var eventDelegate: TransportEventDelegate?
|
||||||
weak var peerEventsDelegate: TransportPeerEventsDelegate?
|
weak var peerEventsDelegate: TransportPeerEventsDelegate?
|
||||||
|
|
||||||
var peerSnapshotPublisher: AnyPublisher<[TransportPeerSnapshot], Never> {
|
var peerSnapshotPublisher: AnyPublisher<[TransportPeerSnapshot], Never> {
|
||||||
@@ -142,17 +142,9 @@ final class NostrTransport: Transport, @unchecked Sendable {
|
|||||||
func getFingerprint(for peerID: PeerID) -> String? { nil }
|
func getFingerprint(for peerID: PeerID) -> String? { nil }
|
||||||
func getNoiseSessionState(for peerID: PeerID) -> LazyHandshakeState { .none }
|
func getNoiseSessionState(for peerID: PeerID) -> LazyHandshakeState { .none }
|
||||||
func triggerHandshake(with peerID: PeerID) { /* no-op */ }
|
func triggerHandshake(with peerID: PeerID) { /* no-op */ }
|
||||||
|
|
||||||
// Nostr does not use Noise sessions here; return a cached placeholder to avoid reallocation
|
// Nostr does not use Noise sessions here; the inert Transport defaults
|
||||||
private static var cachedNoiseService: NoiseEncryptionService?
|
// for the noise* identity hooks apply.
|
||||||
func getNoiseService() -> NoiseEncryptionService {
|
|
||||||
if let noiseService = Self.cachedNoiseService {
|
|
||||||
return noiseService
|
|
||||||
}
|
|
||||||
let noiseService = NoiseEncryptionService(keychain: keychain)
|
|
||||||
Self.cachedNoiseService = noiseService
|
|
||||||
return noiseService
|
|
||||||
}
|
|
||||||
|
|
||||||
// Public broadcast not supported over Nostr here
|
// Public broadcast not supported over Nostr here
|
||||||
func sendMessage(_ content: String, mentions: [String]) { /* no-op */ }
|
func sendMessage(_ content: String, mentions: [String]) { /* no-op */ }
|
||||||
@@ -174,9 +166,9 @@ final class NostrTransport: Transport, @unchecked Sendable {
|
|||||||
func sendReadReceipt(_ receipt: ReadReceipt, to peerID: PeerID) {
|
func sendReadReceipt(_ receipt: ReadReceipt, to peerID: PeerID) {
|
||||||
// Enqueue and process with throttling to avoid relay rate limits
|
// Enqueue and process with throttling to avoid relay rate limits
|
||||||
// Use barrier to synchronize access to readQueue
|
// Use barrier to synchronize access to readQueue
|
||||||
queue.async(flags: .barrier) { [weak self] in
|
queue.async(flags: .barrier) {
|
||||||
self?.readQueue.append(QueuedRead(receipt: receipt, peerID: peerID))
|
self.readQueue.append(QueuedRead(receipt: receipt, peerID: peerID))
|
||||||
self?.processReadQueueIfNeeded()
|
self.processReadQueueIfNeeded()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -7,6 +7,7 @@
|
|||||||
//
|
//
|
||||||
|
|
||||||
import BitLogger
|
import BitLogger
|
||||||
|
import BitFoundation
|
||||||
import Foundation
|
import Foundation
|
||||||
|
|
||||||
struct NotificationStreamAssembler {
|
struct NotificationStreamAssembler {
|
||||||
@@ -20,6 +21,19 @@ struct NotificationStreamAssembler {
|
|||||||
pendingFrameExpectedLength = 0
|
pendingFrameExpectedLength = 0
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private mutating func discardLeadingPaddingIfPresent() -> Bool {
|
||||||
|
guard let first = buffer.first else { return false }
|
||||||
|
guard first != 1 && first != 2 else { return false }
|
||||||
|
let paddingLength = Int(first)
|
||||||
|
guard paddingLength > 0, paddingLength <= buffer.count else { return false }
|
||||||
|
guard buffer.prefix(paddingLength).allSatisfy({ $0 == first }) else { return false }
|
||||||
|
|
||||||
|
buffer.removeFirst(paddingLength)
|
||||||
|
pendingFrameStartedAt = nil
|
||||||
|
pendingFrameExpectedLength = 0
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
mutating func append(_ chunk: Data) -> (frames: [Data], droppedPrefixes: [UInt8], reset: Bool) {
|
mutating func append(_ chunk: Data) -> (frames: [Data], droppedPrefixes: [UInt8], reset: Bool) {
|
||||||
guard !chunk.isEmpty else { return ([], [], false) }
|
guard !chunk.isEmpty else { return ([], [], false) }
|
||||||
|
|
||||||
@@ -41,6 +55,9 @@ struct NotificationStreamAssembler {
|
|||||||
while buffer.count >= minimumFramePrefix {
|
while buffer.count >= minimumFramePrefix {
|
||||||
guard let version = buffer.first else { break }
|
guard let version = buffer.first else { break }
|
||||||
guard version == 1 || version == 2 else {
|
guard version == 1 || version == 2 else {
|
||||||
|
if discardLeadingPaddingIfPresent() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
dropped.append(buffer.removeFirst())
|
dropped.append(buffer.removeFirst())
|
||||||
pendingFrameStartedAt = nil
|
pendingFrameStartedAt = nil
|
||||||
pendingFrameExpectedLength = 0
|
pendingFrameExpectedLength = 0
|
||||||
@@ -132,6 +149,11 @@ struct NotificationStreamAssembler {
|
|||||||
let frame = Data(buffer.prefix(frameLength))
|
let frame = Data(buffer.prefix(frameLength))
|
||||||
frames.append(frame)
|
frames.append(frame)
|
||||||
buffer.removeFirst(frameLength)
|
buffer.removeFirst(frameLength)
|
||||||
|
_ = discardLeadingPaddingIfPresent()
|
||||||
|
}
|
||||||
|
|
||||||
|
if discardLeadingPaddingIfPresent() {
|
||||||
|
return (frames, dropped, didReset)
|
||||||
}
|
}
|
||||||
|
|
||||||
if !buffer.isEmpty, buffer.allSatisfy({ $0 == 0 }) {
|
if !buffer.isEmpty, buffer.allSatisfy({ $0 == 0 }) {
|
||||||
|
|||||||
@@ -8,14 +8,25 @@
|
|||||||
|
|
||||||
import BitLogger
|
import BitLogger
|
||||||
import BitFoundation
|
import BitFoundation
|
||||||
|
import Combine
|
||||||
import Foundation
|
import Foundation
|
||||||
import SwiftUI
|
import SwiftUI
|
||||||
|
|
||||||
/// Manages all private chat functionality
|
/// Manages private chat session policy (selection, read receipts,
|
||||||
|
/// consolidation). Message storage lives in the single-writer
|
||||||
|
/// `ConversationStore` (docs/CONVERSATION-STORE-DESIGN.md); the
|
||||||
|
/// `privateChats` / `unreadMessages` properties below are read-only views
|
||||||
|
/// derived from it.
|
||||||
|
@MainActor
|
||||||
final class PrivateChatManager: ObservableObject {
|
final class PrivateChatManager: ObservableObject {
|
||||||
@Published var privateChats: [PeerID: [BitchatMessage]] = [:]
|
/// Read-only mirror of `ConversationStore.selectedPrivatePeerID` — the
|
||||||
@Published var selectedPeer: PeerID? = nil
|
/// store is the sole owner of conversation selection. Kept `@Published`
|
||||||
@Published var unreadMessages: Set<PeerID> = []
|
/// so existing observers (`objectWillChange` forwarding into
|
||||||
|
/// `ChatViewModel`) keep firing on selection changes. Mutate via
|
||||||
|
/// `startChat(with:)` / `endChat()`, which route through the store's
|
||||||
|
/// `setSelectedPrivatePeer` intent.
|
||||||
|
@Published private(set) var selectedPeer: PeerID? = nil
|
||||||
|
private var selectedPeerMirrorCancellable: AnyCancellable? = nil
|
||||||
|
|
||||||
private var selectedPeerFingerprint: String? = nil
|
private var selectedPeerFingerprint: String? = nil
|
||||||
var sentReadReceipts: Set<String> = [] // Made accessible for ChatViewModel
|
var sentReadReceipts: Set<String> = [] // Made accessible for ChatViewModel
|
||||||
@@ -25,13 +36,51 @@ final class PrivateChatManager: ObservableObject {
|
|||||||
weak var messageRouter: MessageRouter?
|
weak var messageRouter: MessageRouter?
|
||||||
// Peer service for looking up peer info during consolidation
|
// Peer service for looking up peer info during consolidation
|
||||||
weak var unifiedPeerService: UnifiedPeerService?
|
weak var unifiedPeerService: UnifiedPeerService?
|
||||||
|
/// Single source of truth for message and selection state; injected by
|
||||||
init(meshService: Transport? = nil) {
|
/// the bootstrapper (`wireServiceGraph`).
|
||||||
self.meshService = meshService
|
var conversationStore: ConversationStore? {
|
||||||
|
didSet { bindSelectionMirror() }
|
||||||
}
|
}
|
||||||
|
|
||||||
// Cap for messages stored per private chat
|
init(meshService: Transport? = nil, conversationStore: ConversationStore? = nil) {
|
||||||
private let privateChatCap = TransportConfig.privateChatCap
|
self.meshService = meshService
|
||||||
|
self.conversationStore = conversationStore
|
||||||
|
bindSelectionMirror() // didSet does not fire during init
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Keeps `selectedPeer` in lock-step with the store's selection axis
|
||||||
|
/// (including store-internal handoffs such as conversation migration).
|
||||||
|
private func bindSelectionMirror() {
|
||||||
|
guard let store = conversationStore else {
|
||||||
|
selectedPeerMirrorCancellable = nil
|
||||||
|
return
|
||||||
|
}
|
||||||
|
selectedPeerMirrorCancellable = store.$selectedPrivatePeerID
|
||||||
|
.sink { [weak self] peerID in
|
||||||
|
guard let self, self.selectedPeer != peerID else { return }
|
||||||
|
self.selectedPeer = peerID
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Derived message state (read-only compat views)
|
||||||
|
|
||||||
|
/// All private chats keyed by routing peer ID, derived from the store.
|
||||||
|
/// Mutations go through the store's intent API only.
|
||||||
|
@MainActor
|
||||||
|
var privateChats: [PeerID: [BitchatMessage]] {
|
||||||
|
conversationStore?.directMessagesByRoutingPeerID() ?? [:]
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Unread chats, derived from the store's unread state.
|
||||||
|
@MainActor
|
||||||
|
var unreadMessages: Set<PeerID> {
|
||||||
|
conversationStore?.unreadDirectRoutingPeerIDs() ?? []
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
private func messages(for peerID: PeerID) -> [BitchatMessage] {
|
||||||
|
conversationStore?.conversationsByID[.directPeer(peerID)]?.messages ?? []
|
||||||
|
}
|
||||||
|
|
||||||
// MARK: - Message Consolidation
|
// MARK: - Message Consolidation
|
||||||
|
|
||||||
@@ -44,57 +93,51 @@ final class PrivateChatManager: ObservableObject {
|
|||||||
/// - Returns: True if any unread messages were found during consolidation
|
/// - Returns: True if any unread messages were found during consolidation
|
||||||
@MainActor
|
@MainActor
|
||||||
func consolidateMessages(for peerID: PeerID, peerNickname: String, persistedReadReceipts: Set<String>) -> Bool {
|
func consolidateMessages(for peerID: PeerID, peerNickname: String, persistedReadReceipts: Set<String>) -> Bool {
|
||||||
guard let meshService = meshService else { return false }
|
guard let meshService = meshService, let store = conversationStore else { return false }
|
||||||
var hasUnreadMessages = false
|
var hasUnreadMessages = false
|
||||||
|
|
||||||
// 1. Consolidate from stable Noise key (64-char hex)
|
// 1. Consolidate from stable Noise key (64-char hex)
|
||||||
if let peer = unifiedPeerService?.getPeer(by: peerID) {
|
if let peer = unifiedPeerService?.getPeer(by: peerID) {
|
||||||
let noiseKeyHex = PeerID(hexData: peer.noisePublicKey)
|
let noiseKeyHex = PeerID(hexData: peer.noisePublicKey)
|
||||||
|
let nostrMessages = messages(for: noiseKeyHex)
|
||||||
|
|
||||||
if noiseKeyHex != peerID, let nostrMessages = privateChats[noiseKeyHex], !nostrMessages.isEmpty {
|
if noiseKeyHex != peerID, !nostrMessages.isEmpty {
|
||||||
if privateChats[peerID] == nil {
|
|
||||||
privateChats[peerID] = []
|
|
||||||
}
|
|
||||||
|
|
||||||
let existingMessageIds = Set(privateChats[peerID]?.map { $0.id } ?? [])
|
|
||||||
for message in nostrMessages {
|
for message in nostrMessages {
|
||||||
if !existingMessageIds.contains(message.id) {
|
// Update senderPeerID for correct read receipts
|
||||||
// Update senderPeerID for correct read receipts
|
let updatedMessage = BitchatMessage(
|
||||||
let updatedMessage = BitchatMessage(
|
id: message.id,
|
||||||
id: message.id,
|
sender: message.sender,
|
||||||
sender: message.sender,
|
content: message.content,
|
||||||
content: message.content,
|
timestamp: message.timestamp,
|
||||||
timestamp: message.timestamp,
|
isRelay: message.isRelay,
|
||||||
isRelay: message.isRelay,
|
originalSender: message.originalSender,
|
||||||
originalSender: message.originalSender,
|
isPrivate: message.isPrivate,
|
||||||
isPrivate: message.isPrivate,
|
recipientNickname: message.recipientNickname,
|
||||||
recipientNickname: message.recipientNickname,
|
senderPeerID: message.senderPeerID == meshService.myPeerID ? meshService.myPeerID : peerID,
|
||||||
senderPeerID: message.senderPeerID == meshService.myPeerID ? meshService.myPeerID : peerID,
|
mentions: message.mentions,
|
||||||
mentions: message.mentions,
|
deliveryStatus: message.deliveryStatus
|
||||||
deliveryStatus: message.deliveryStatus
|
)
|
||||||
)
|
// Store append dedups by message ID (skips ones the
|
||||||
privateChats[peerID]?.append(updatedMessage)
|
// target chat already has).
|
||||||
|
guard store.append(updatedMessage, to: .directPeer(peerID)) else { continue }
|
||||||
|
|
||||||
// Check for recent unread messages (< 60s, not sent by us, not already read)
|
// Check for recent unread messages (< 60s, not sent by us, not already read)
|
||||||
// Use persistedReadReceipts to correctly identify already-read messages after app restart
|
// Use persistedReadReceipts to correctly identify already-read messages after app restart
|
||||||
if message.senderPeerID != meshService.myPeerID {
|
if message.senderPeerID != meshService.myPeerID {
|
||||||
let messageAge = Date().timeIntervalSince(message.timestamp)
|
let messageAge = Date().timeIntervalSince(message.timestamp)
|
||||||
if messageAge < 60 && !persistedReadReceipts.contains(message.id) {
|
if messageAge < 60 && !persistedReadReceipts.contains(message.id) {
|
||||||
hasUnreadMessages = true
|
hasUnreadMessages = true
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
privateChats[peerID]?.sort { $0.timestamp < $1.timestamp }
|
|
||||||
|
|
||||||
if hasUnreadMessages {
|
if hasUnreadMessages {
|
||||||
unreadMessages.insert(peerID)
|
store.markUnread(.directPeer(peerID))
|
||||||
} else if unreadMessages.contains(noiseKeyHex) {
|
} else {
|
||||||
unreadMessages.remove(noiseKeyHex)
|
store.markRead(.directPeer(noiseKeyHex))
|
||||||
}
|
}
|
||||||
|
|
||||||
privateChats.removeValue(forKey: noiseKeyHex)
|
store.removeConversation(.directPeer(noiseKeyHex))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -112,52 +155,43 @@ final class PrivateChatManager: ObservableObject {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if !tempPeerIDsToConsolidate.isEmpty {
|
if !tempPeerIDsToConsolidate.isEmpty {
|
||||||
if privateChats[peerID] == nil {
|
|
||||||
privateChats[peerID] = []
|
|
||||||
}
|
|
||||||
|
|
||||||
let existingMessageIds = Set(privateChats[peerID]?.map { $0.id } ?? [])
|
|
||||||
var consolidatedCount = 0
|
var consolidatedCount = 0
|
||||||
var hadUnreadTemp = false
|
var hadUnreadTemp = false
|
||||||
|
let unreadPeerIDs = unreadMessages
|
||||||
|
|
||||||
for tempPeerID in tempPeerIDsToConsolidate {
|
for tempPeerID in tempPeerIDsToConsolidate {
|
||||||
if unreadMessages.contains(tempPeerID) {
|
if unreadPeerIDs.contains(tempPeerID) {
|
||||||
hadUnreadTemp = true
|
hadUnreadTemp = true
|
||||||
}
|
}
|
||||||
|
|
||||||
if let tempMessages = privateChats[tempPeerID] {
|
for message in messages(for: tempPeerID) {
|
||||||
for message in tempMessages {
|
let updatedMessage = BitchatMessage(
|
||||||
if !existingMessageIds.contains(message.id) {
|
id: message.id,
|
||||||
let updatedMessage = BitchatMessage(
|
sender: message.sender,
|
||||||
id: message.id,
|
content: message.content,
|
||||||
sender: message.sender,
|
timestamp: message.timestamp,
|
||||||
content: message.content,
|
isRelay: message.isRelay,
|
||||||
timestamp: message.timestamp,
|
originalSender: message.originalSender,
|
||||||
isRelay: message.isRelay,
|
isPrivate: message.isPrivate,
|
||||||
originalSender: message.originalSender,
|
recipientNickname: message.recipientNickname,
|
||||||
isPrivate: message.isPrivate,
|
senderPeerID: peerID,
|
||||||
recipientNickname: message.recipientNickname,
|
mentions: message.mentions,
|
||||||
senderPeerID: peerID,
|
deliveryStatus: message.deliveryStatus
|
||||||
mentions: message.mentions,
|
)
|
||||||
deliveryStatus: message.deliveryStatus
|
if store.append(updatedMessage, to: .directPeer(peerID)) {
|
||||||
)
|
consolidatedCount += 1
|
||||||
privateChats[peerID]?.append(updatedMessage)
|
|
||||||
consolidatedCount += 1
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
privateChats.removeValue(forKey: tempPeerID)
|
|
||||||
unreadMessages.remove(tempPeerID)
|
|
||||||
}
|
}
|
||||||
|
store.removeConversation(.directPeer(tempPeerID))
|
||||||
}
|
}
|
||||||
|
|
||||||
if hadUnreadTemp {
|
if hadUnreadTemp {
|
||||||
unreadMessages.insert(peerID)
|
store.markUnread(.directPeer(peerID))
|
||||||
hasUnreadMessages = true
|
hasUnreadMessages = true
|
||||||
SecureLogger.debug("📬 Transferred unread status from temp peer IDs to \(peerID)", category: .session)
|
SecureLogger.debug("📬 Transferred unread status from temp peer IDs to \(peerID)", category: .session)
|
||||||
}
|
}
|
||||||
|
|
||||||
if consolidatedCount > 0 {
|
if consolidatedCount > 0 {
|
||||||
privateChats[peerID]?.sort { $0.timestamp < $1.timestamp }
|
|
||||||
SecureLogger.info("📥 Consolidated \(consolidatedCount) Nostr messages from temporary peer IDs to \(peerNickname)", category: .session)
|
SecureLogger.info("📥 Consolidated \(consolidatedCount) Nostr messages from temporary peer IDs to \(peerNickname)", category: .session)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -168,9 +202,7 @@ final class PrivateChatManager: ObservableObject {
|
|||||||
/// Syncs the read receipt tracking between manager and view model for sent messages
|
/// Syncs the read receipt tracking between manager and view model for sent messages
|
||||||
@MainActor
|
@MainActor
|
||||||
func syncReadReceiptsForSentMessages(peerID: PeerID, nickname: String, externalReceipts: inout Set<String>) {
|
func syncReadReceiptsForSentMessages(peerID: PeerID, nickname: String, externalReceipts: inout Set<String>) {
|
||||||
guard let messages = privateChats[peerID] else { return }
|
for message in messages(for: peerID) {
|
||||||
|
|
||||||
for message in messages {
|
|
||||||
if message.sender == nickname {
|
if message.sender == nickname {
|
||||||
if let status = message.deliveryStatus {
|
if let status = message.deliveryStatus {
|
||||||
switch status {
|
switch status {
|
||||||
@@ -184,86 +216,68 @@ final class PrivateChatManager: ObservableObject {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Start a private chat with a peer
|
/// Start a private chat with a peer. Selection is mutated through the
|
||||||
|
/// store's intent (the store owns it); the manager keeps its side
|
||||||
|
/// effects (fingerprint tracking, read receipts, unread clearing).
|
||||||
|
@MainActor
|
||||||
func startChat(with peerID: PeerID) {
|
func startChat(with peerID: PeerID) {
|
||||||
selectedPeer = peerID
|
// Also creates the conversation if needed and updates the derived
|
||||||
|
// `selectedConversationID`; `selectedPeer` mirrors the change.
|
||||||
|
conversationStore?.setSelectedPrivatePeer(peerID)
|
||||||
|
|
||||||
// Store fingerprint for persistence across reconnections
|
// Store fingerprint for persistence across reconnections
|
||||||
if let fingerprint = meshService?.getFingerprint(for: peerID) {
|
if let fingerprint = meshService?.getFingerprint(for: peerID) {
|
||||||
selectedPeerFingerprint = fingerprint
|
selectedPeerFingerprint = fingerprint
|
||||||
}
|
}
|
||||||
|
|
||||||
// Mark messages as read
|
// Mark messages as read
|
||||||
markAsRead(from: peerID)
|
markAsRead(from: peerID)
|
||||||
|
|
||||||
// Initialize chat if needed
|
|
||||||
if privateChats[peerID] == nil {
|
|
||||||
privateChats[peerID] = []
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// End the current private chat
|
/// End the current private chat (selection returns to the active public
|
||||||
|
/// channel's conversation).
|
||||||
func endChat() {
|
func endChat() {
|
||||||
selectedPeer = nil
|
conversationStore?.setSelectedPrivatePeer(nil)
|
||||||
selectedPeerFingerprint = nil
|
selectedPeerFingerprint = nil
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Remove duplicate messages by ID and keep chronological order
|
/// No-op since the `ConversationStore` cutover: the store maintains
|
||||||
func sanitizeChat(for peerID: PeerID) {
|
/// chronological order and dedups by message ID on every insert, so the
|
||||||
guard let arr = privateChats[peerID] else { return }
|
/// per-append re-sort/dedup sweep this performed is no longer needed.
|
||||||
if arr.count <= 1 {
|
/// Kept only for API compatibility until step 5 removes the callers.
|
||||||
return
|
func sanitizeChat(for peerID: PeerID) {}
|
||||||
}
|
|
||||||
|
|
||||||
var indexByID: [String: Int] = [:]
|
|
||||||
indexByID.reserveCapacity(arr.count)
|
|
||||||
var deduped: [BitchatMessage] = []
|
|
||||||
deduped.reserveCapacity(arr.count)
|
|
||||||
|
|
||||||
for msg in arr.sorted(by: { $0.timestamp < $1.timestamp }) {
|
|
||||||
if let existing = indexByID[msg.id] {
|
|
||||||
deduped[existing] = msg
|
|
||||||
} else {
|
|
||||||
indexByID[msg.id] = deduped.count
|
|
||||||
deduped.append(msg)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
privateChats[peerID] = deduped
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Mark messages from a peer as read
|
/// Mark messages from a peer as read
|
||||||
|
@MainActor
|
||||||
func markAsRead(from peerID: PeerID) {
|
func markAsRead(from peerID: PeerID) {
|
||||||
unreadMessages.remove(peerID)
|
conversationStore?.markRead(.directPeer(peerID))
|
||||||
|
|
||||||
// Send read receipts for unread messages that haven't been sent yet
|
// Send read receipts for unread messages that haven't been sent yet
|
||||||
if let messages = privateChats[peerID] {
|
for message in messages(for: peerID) {
|
||||||
for message in messages {
|
if message.senderPeerID == peerID && !message.isRelay && !sentReadReceipts.contains(message.id) {
|
||||||
if message.senderPeerID == peerID && !message.isRelay && !sentReadReceipts.contains(message.id) {
|
sendReadReceipt(for: message)
|
||||||
sendReadReceipt(for: message)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Private Methods
|
// MARK: - Private Methods
|
||||||
|
|
||||||
private func sendReadReceipt(for message: BitchatMessage) {
|
private func sendReadReceipt(for message: BitchatMessage) {
|
||||||
guard !sentReadReceipts.contains(message.id),
|
guard !sentReadReceipts.contains(message.id),
|
||||||
let senderPeerID = message.senderPeerID else {
|
let senderPeerID = message.senderPeerID else {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
sentReadReceipts.insert(message.id)
|
sentReadReceipts.insert(message.id)
|
||||||
|
|
||||||
// Create read receipt using the simplified method
|
// Create read receipt using the simplified method
|
||||||
let receipt = ReadReceipt(
|
let receipt = ReadReceipt(
|
||||||
originalMessageID: message.id,
|
originalMessageID: message.id,
|
||||||
readerID: meshService?.myPeerID ?? PeerID(str: ""),
|
readerID: meshService?.myPeerID ?? PeerID(str: ""),
|
||||||
readerNickname: meshService?.myNickname ?? ""
|
readerNickname: meshService?.myNickname ?? ""
|
||||||
)
|
)
|
||||||
|
|
||||||
// Route via MessageRouter to avoid handshakeRequired spam when session isn't established
|
// Route via MessageRouter to avoid handshakeRequired spam when session isn't established
|
||||||
if let router = messageRouter {
|
if let router = messageRouter {
|
||||||
SecureLogger.debug("PrivateChatManager: sending READ ack for \(message.id.prefix(8))… to \(senderPeerID.id.prefix(8))… via router", category: .session)
|
SecureLogger.debug("PrivateChatManager: sending READ ack for \(message.id.prefix(8))… to \(senderPeerID.id.prefix(8))… via router", category: .session)
|
||||||
|
|||||||
@@ -11,6 +11,7 @@ struct RelayDecision {
|
|||||||
struct RelayController {
|
struct RelayController {
|
||||||
static func decide(ttl: UInt8,
|
static func decide(ttl: UInt8,
|
||||||
senderIsSelf: Bool,
|
senderIsSelf: Bool,
|
||||||
|
recipientIsSelf: Bool = false,
|
||||||
isEncrypted: Bool,
|
isEncrypted: Bool,
|
||||||
isDirectedEncrypted: Bool,
|
isDirectedEncrypted: Bool,
|
||||||
isFragment: Bool,
|
isFragment: Bool,
|
||||||
@@ -22,7 +23,7 @@ struct RelayController {
|
|||||||
let ttlCap = min(ttl, TransportConfig.messageTTLDefault)
|
let ttlCap = min(ttl, TransportConfig.messageTTLDefault)
|
||||||
|
|
||||||
// Suppress obvious non-relays
|
// Suppress obvious non-relays
|
||||||
if ttlCap <= 1 || senderIsSelf {
|
if ttlCap <= 1 || senderIsSelf || recipientIsSelf {
|
||||||
return RelayDecision(shouldRelay: false, newTTL: ttlCap, delayMs: 0)
|
return RelayDecision(shouldRelay: false, newTTL: ttlCap, delayMs: 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -38,7 +39,12 @@ struct RelayController {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if isFragment {
|
if isFragment {
|
||||||
let ttlLimit = min(ttlCap, TransportConfig.bleFragmentRelayTtlCap)
|
// Dense graphs clamp harder to contain full-fanout fragment floods;
|
||||||
|
// sparse graphs get full depth so media reaches as far as text.
|
||||||
|
let fragmentCap = degree >= highDegreeThreshold
|
||||||
|
? TransportConfig.bleFragmentRelayTtlCapDense
|
||||||
|
: TransportConfig.bleFragmentRelayTtlCap
|
||||||
|
let ttlLimit = min(ttlCap, fragmentCap)
|
||||||
guard ttlLimit > 1 else {
|
guard ttlLimit > 1 else {
|
||||||
return RelayDecision(shouldRelay: false, newTTL: ttlLimit, delayMs: 0)
|
return RelayDecision(shouldRelay: false, newTTL: ttlLimit, delayMs: 0)
|
||||||
}
|
}
|
||||||
@@ -49,11 +55,16 @@ struct RelayController {
|
|||||||
|
|
||||||
// TTL clamping for broadcast
|
// TTL clamping for broadcast
|
||||||
// - Dense graphs: keep lower but still allow multi-hop bridging
|
// - Dense graphs: keep lower but still allow multi-hop bridging
|
||||||
|
// - Thin chains (degree <= 2): every hop counts and flood cost is
|
||||||
|
// minimal, so relay at full incoming depth
|
||||||
// - Announces get a bit more headroom
|
// - Announces get a bit more headroom
|
||||||
let ttlLimit: UInt8 = {
|
let ttlLimit: UInt8 = {
|
||||||
if degree >= highDegreeThreshold {
|
if degree >= highDegreeThreshold {
|
||||||
return max(UInt8(2), min(ttlCap, UInt8(5)))
|
return max(UInt8(2), min(ttlCap, UInt8(5)))
|
||||||
}
|
}
|
||||||
|
if degree <= 2 {
|
||||||
|
return ttlCap
|
||||||
|
}
|
||||||
let preferred = UInt8(isAnnounce ? 7 : 6)
|
let preferred = UInt8(isAnnounce ? 7 : 6)
|
||||||
return max(UInt8(2), min(ttlCap, preferred))
|
return max(UInt8(2), min(ttlCap, preferred))
|
||||||
}()
|
}()
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user