mirror of
https://github.com/permissionlesstech/bitchat.git
synced 2026-07-25 23:25:20 +00:00
Compare commits
187
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
4fbbd24021 | ||
|
|
ec54877140 | ||
|
|
293d627c28 | ||
|
|
0c3f84224c | ||
|
|
7323c0b96c | ||
|
|
7bb835ffc9 | ||
|
|
23d63ab4df | ||
|
|
fa3c74f941 | ||
|
|
9bac52051a | ||
|
|
7b9ffe464a | ||
|
|
7b940485d9 | ||
|
|
5a87ee3e62 | ||
|
|
342eabbc00 | ||
|
|
241ce2d52c | ||
|
|
18b56e7393 | ||
|
|
beb04fc887 | ||
|
|
5fcffefa28 | ||
|
|
46ae039587 | ||
|
|
917f7ebe5f | ||
|
|
b47fb736f4 | ||
|
|
ebb5bb6558 | ||
|
|
7cfdcfe174 | ||
|
|
21e0cbc607 | ||
|
|
1b439a543e | ||
|
|
d30a3b14cf | ||
|
|
d03128612d | ||
|
|
ae294aab6d | ||
|
|
07b0e146f2 | ||
|
|
cc5939cb13 | ||
|
|
cada784844 | ||
|
|
b1aeb931bc | ||
|
|
b675738664 | ||
|
|
4091a30f10 | ||
|
|
238311aefb | ||
|
|
6defae71c6 | ||
|
|
b533d9560d | ||
|
|
f41a390a94 | ||
|
|
f83316bd1f | ||
|
|
09818a02ed | ||
|
|
9cd955ae2f | ||
|
|
49b1413d85 | ||
|
|
31be6b83a7 | ||
|
|
1563209797 | ||
|
|
99896bcded | ||
|
|
4b3077169a | ||
|
|
b84c36c6fa | ||
|
|
3eac5858e4 | ||
|
|
10b7c1fd80 | ||
|
|
bf3249aef7 | ||
|
|
95a6ec7315 | ||
|
|
74864472c7 | ||
|
|
9404c03477 | ||
|
|
6faa46a22f | ||
|
|
e02f4327c0 | ||
|
|
ce31d85323 | ||
|
|
b6d8a5b758 | ||
|
|
6630f5a792 | ||
|
|
0f5299a0f5 | ||
|
|
eb3bbfd861 | ||
|
|
a37243e780 | ||
|
|
90b134186b | ||
|
|
3c7e14f49d | ||
|
|
31275856dd | ||
|
|
b6cf44a824 | ||
|
|
b6ae08be60 | ||
|
|
d469704c34 | ||
|
|
c975abf2ff | ||
|
|
aff700a15e | ||
|
|
869d766f8d | ||
|
|
1b4f120014 | ||
|
|
bc312e4aef | ||
|
|
6e7509f2be | ||
|
|
ca06f4d51d | ||
|
|
04f57e8713 | ||
|
|
59c3c4e236 | ||
|
|
e887e04f40 | ||
|
|
151b68a497 | ||
|
|
b4a3ee5777 | ||
|
|
8d19d6d62c | ||
|
|
84fd92ef4b | ||
|
|
f71bd506fd | ||
|
|
ddd7ef5668 | ||
|
|
548a20e77d | ||
|
|
6e231d10c5 | ||
|
|
4c9f6e689e | ||
|
|
7e73b65240 | ||
|
|
f5e5f7b98e | ||
|
|
b15d92ebb5 | ||
|
|
6efe9d02fb | ||
|
|
5a66f03400 | ||
|
|
a221b22691 | ||
|
|
4b38e9a23c | ||
|
|
2686d9c82c | ||
|
|
832e1f006c | ||
|
|
090c4cd919 | ||
|
|
37c2b3a8c2 | ||
|
|
6f1c879237 | ||
|
|
81f5101390 | ||
|
|
d56674706d | ||
|
|
33bcfa3147 | ||
|
|
3a54160793 | ||
|
|
5d2745eae6 | ||
|
|
5138ce4a33 | ||
|
|
07a4997192 | ||
|
|
2d703f4dd9 | ||
|
|
56b9457fcf | ||
|
|
9e57bce5c3 | ||
|
|
6eef030386 | ||
|
|
9d8754099d | ||
|
|
7fae4e71fb | ||
|
|
76d8b3fa7f | ||
|
|
b8a8b940b7 | ||
|
|
7a5ab52f4c | ||
|
|
d2edb651c1 | ||
|
|
858e878959 | ||
|
|
fa0a15cbcf | ||
|
|
3e680f40bf | ||
|
|
ec3c16176e | ||
|
|
2a8de7e048 | ||
|
|
e9dd30ccbf | ||
|
|
af28faa91d | ||
|
|
b33fe8086d | ||
|
|
5495874b5a | ||
|
|
0fca551966 | ||
|
|
0492480598 | ||
|
|
86f46c8b90 | ||
|
|
7057fe75b0 | ||
|
|
873788b24f | ||
|
|
b6dd31b285 | ||
|
|
ff7e5a2c2c | ||
|
|
708a5e33cd | ||
|
|
f893f0605a | ||
|
|
ce33132a0f | ||
|
|
c538837300 | ||
|
|
6888bfa351 | ||
|
|
2fed4b7c31 | ||
|
|
b956e407ff | ||
|
|
477cc7fa05 | ||
|
|
dffce9d63f | ||
|
|
792eaa3166 | ||
|
|
6ff9b59ff2 | ||
|
|
bc04cd0bde | ||
|
|
3fe417395e | ||
|
|
e35bd57c6e | ||
|
|
5aefb05a8b | ||
|
|
e4cbf382ce | ||
|
|
7609c6eeba | ||
|
|
f37acf7e4b | ||
|
|
cf528b0daf | ||
|
|
209ccb4ade | ||
|
|
1876e85f8b | ||
|
|
109b7d0e03 | ||
|
|
09a319fb0f | ||
|
|
97ca55cc54 | ||
|
|
394836f247 | ||
|
|
97fc21c23f | ||
|
|
79bd4af912 | ||
|
|
96c78ef5a2 | ||
|
|
01256f3041 | ||
|
|
2edbe2bbbd | ||
|
|
a91978c10e | ||
|
|
14c4e586fc | ||
|
|
83779240ae | ||
|
|
5034732515 | ||
|
|
b6ce4fae43 | ||
|
|
98fa02cc16 | ||
|
|
0812cafd55 | ||
|
|
450955525c | ||
|
|
475bc70c71 | ||
|
|
af6136c01c | ||
|
|
b839ce5f6c | ||
|
|
0776c9813c | ||
|
|
0dd999af6b | ||
|
|
c3a1af7023 | ||
|
|
880813f256 | ||
|
|
64fb634166 | ||
|
|
13b19fb8eb | ||
|
|
d4967ae9c3 | ||
|
|
435744a977 | ||
|
|
70caa9e24a | ||
|
|
5084f87fe5 | ||
|
|
8f56e4f0fb | ||
|
|
aca44f9f55 | ||
|
|
3f00cf9467 | ||
|
|
40e54a5120 | ||
|
|
7040d9ecb0 | ||
|
|
88bafb41cc |
@@ -7,6 +7,7 @@ on:
|
|||||||
|
|
||||||
permissions:
|
permissions:
|
||||||
contents: write
|
contents: write
|
||||||
|
pull-requests: write
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
update-relay-data:
|
update-relay-data:
|
||||||
@@ -17,10 +18,11 @@ jobs:
|
|||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
token: ${{ secrets.GITHUB_TOKEN }}
|
token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
- name: Fetch GeoRelays
|
- name: Fetch GeoRelays
|
||||||
run: |
|
run: |
|
||||||
wget https://raw.githubusercontent.com/permissionlesstech/georelays/refs/heads/main/nostr_relays.csv
|
wget -q https://raw.githubusercontent.com/permissionlesstech/georelays/refs/heads/main/nostr_relays.csv
|
||||||
mv nostr_relays.csv ./relays/online_relays_gps.csv
|
mv nostr_relays.csv ./relays/online_relays_gps.csv
|
||||||
|
|
||||||
- name: Check for changes
|
- name: Check for changes
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
MARKETING_VERSION = 1.4.4
|
MARKETING_VERSION = 1.5.0
|
||||||
CURRENT_PROJECT_VERSION = 1
|
CURRENT_PROJECT_VERSION = 1
|
||||||
|
|
||||||
IPHONEOS_DEPLOYMENT_TARGET = 16.0
|
IPHONEOS_DEPLOYMENT_TARGET = 16.0
|
||||||
|
|||||||
@@ -47,7 +47,7 @@ patch-for-macos: backup
|
|||||||
# Build the macOS app
|
# Build the macOS app
|
||||||
build: #check generate
|
build: #check generate
|
||||||
@echo "Building BitChat for macOS..."
|
@echo "Building BitChat for macOS..."
|
||||||
@xcodebuild -project bitchat.xcodeproj -scheme "bitchat_macOS" -configuration Debug CODE_SIGN_IDENTITY="" CODE_SIGNING_REQUIRED=NO CODE_SIGN_ENTITLEMENTS="" build
|
@xcodebuild -project bitchat.xcodeproj -scheme "bitchat (macOS)" -configuration Debug CODE_SIGN_IDENTITY="" CODE_SIGNING_REQUIRED=NO CODE_SIGN_ENTITLEMENTS="" build
|
||||||
|
|
||||||
# Run the macOS app
|
# Run the macOS app
|
||||||
run: build
|
run: build
|
||||||
|
|||||||
+6
-4
@@ -16,7 +16,7 @@ let package = Package(
|
|||||||
),
|
),
|
||||||
],
|
],
|
||||||
dependencies:[
|
dependencies:[
|
||||||
.package(path: "localPackages/Tor"),
|
.package(path: "localPackages/Arti"),
|
||||||
.package(path: "localPackages/BitLogger"),
|
.package(path: "localPackages/BitLogger"),
|
||||||
.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")
|
||||||
],
|
],
|
||||||
@@ -26,7 +26,7 @@ let package = Package(
|
|||||||
dependencies: [
|
dependencies: [
|
||||||
.product(name: "P256K", package: "swift-secp256k1"),
|
.product(name: "P256K", package: "swift-secp256k1"),
|
||||||
.product(name: "BitLogger", package: "BitLogger"),
|
.product(name: "BitLogger", package: "BitLogger"),
|
||||||
.product(name: "Tor", package: "Tor")
|
.product(name: "Tor", package: "Arti")
|
||||||
],
|
],
|
||||||
path: "bitchat",
|
path: "bitchat",
|
||||||
exclude: [
|
exclude: [
|
||||||
@@ -34,7 +34,8 @@ let package = Package(
|
|||||||
"Assets.xcassets",
|
"Assets.xcassets",
|
||||||
"bitchat.entitlements",
|
"bitchat.entitlements",
|
||||||
"bitchat-macOS.entitlements",
|
"bitchat-macOS.entitlements",
|
||||||
"LaunchScreen.storyboard"
|
"LaunchScreen.storyboard",
|
||||||
|
"ViewModels/Extensions/README.md"
|
||||||
],
|
],
|
||||||
resources: [
|
resources: [
|
||||||
.process("Localizable.xcstrings")
|
.process("Localizable.xcstrings")
|
||||||
@@ -49,7 +50,8 @@ let package = Package(
|
|||||||
"README.md"
|
"README.md"
|
||||||
],
|
],
|
||||||
resources: [
|
resources: [
|
||||||
.process("Localization")
|
.process("Localization"),
|
||||||
|
.process("Noise")
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
]
|
]
|
||||||
|
|||||||
Generated
+39
-7
@@ -14,7 +14,6 @@
|
|||||||
A6E3E5722E7703760032EA8A /* BitLogger in Frameworks */ = {isa = PBXBuildFile; productRef = A6E3E5712E7703760032EA8A /* BitLogger */; };
|
A6E3E5722E7703760032EA8A /* BitLogger in Frameworks */ = {isa = PBXBuildFile; productRef = A6E3E5712E7703760032EA8A /* BitLogger */; };
|
||||||
A6E3EA7F2E7706720032EA8A /* Tor in Frameworks */ = {isa = PBXBuildFile; productRef = A6E3EA7E2E7706720032EA8A /* Tor */; };
|
A6E3EA7F2E7706720032EA8A /* Tor in Frameworks */ = {isa = PBXBuildFile; productRef = A6E3EA7E2E7706720032EA8A /* Tor */; };
|
||||||
A6E3EA812E7706A80032EA8A /* Tor in Frameworks */ = {isa = PBXBuildFile; productRef = A6E3EA802E7706A80032EA8A /* Tor */; };
|
A6E3EA812E7706A80032EA8A /* Tor in Frameworks */ = {isa = PBXBuildFile; productRef = A6E3EA802E7706A80032EA8A /* Tor */; };
|
||||||
A6F183FD2E948783006A9046 /* tor-nolzma.xcframework in Frameworks */ = {isa = PBXBuildFile; fileRef = A6F183FC2E948783006A9046 /* tor-nolzma.xcframework */; };
|
|
||||||
E0A1B2C3D4E5F6012345678D /* relays/online_relays_gps.csv in Resources */ = {isa = PBXBuildFile; fileRef = E0A1B2C3D4E5F6012345678A /* relays/online_relays_gps.csv */; };
|
E0A1B2C3D4E5F6012345678D /* relays/online_relays_gps.csv in Resources */ = {isa = PBXBuildFile; fileRef = E0A1B2C3D4E5F6012345678A /* relays/online_relays_gps.csv */; };
|
||||||
E0A1B2C3D4E5F6012345678E /* relays/online_relays_gps.csv in Resources */ = {isa = PBXBuildFile; fileRef = E0A1B2C3D4E5F6012345678A /* relays/online_relays_gps.csv */; };
|
E0A1B2C3D4E5F6012345678E /* relays/online_relays_gps.csv in Resources */ = {isa = PBXBuildFile; fileRef = E0A1B2C3D4E5F6012345678A /* relays/online_relays_gps.csv */; };
|
||||||
/* End PBXBuildFile section */
|
/* End PBXBuildFile section */
|
||||||
@@ -95,6 +94,24 @@
|
|||||||
);
|
);
|
||||||
target = 57CA17A36A2532A6CFF367BB /* bitchatShareExtension */;
|
target = 57CA17A36A2532A6CFF367BB /* bitchatShareExtension */;
|
||||||
};
|
};
|
||||||
|
C5E027A52ECCDFD700BD6012 /* Exceptions for "bitchatTests" folder in "bitchatTests_macOS" target */ = {
|
||||||
|
isa = PBXFileSystemSynchronizedBuildFileExceptionSet;
|
||||||
|
membershipExceptions = (
|
||||||
|
Info.plist,
|
||||||
|
Localization/PrimaryLocalizationKeys.json,
|
||||||
|
README.md,
|
||||||
|
);
|
||||||
|
target = 47FF23248747DD7CB666CB91 /* bitchatTests_macOS */;
|
||||||
|
};
|
||||||
|
C5E027A82ECCDFE200BD6012 /* Exceptions for "bitchatTests" folder in "bitchatTests_iOS" target */ = {
|
||||||
|
isa = PBXFileSystemSynchronizedBuildFileExceptionSet;
|
||||||
|
membershipExceptions = (
|
||||||
|
Info.plist,
|
||||||
|
Localization/PrimaryLocalizationKeys.json,
|
||||||
|
README.md,
|
||||||
|
);
|
||||||
|
target = 6CB97DF2EA57234CB3E563B8 /* bitchatTests_iOS */;
|
||||||
|
};
|
||||||
/* End PBXFileSystemSynchronizedBuildFileExceptionSet section */
|
/* End PBXFileSystemSynchronizedBuildFileExceptionSet section */
|
||||||
|
|
||||||
/* Begin PBXFileSystemSynchronizedRootGroup section */
|
/* Begin PBXFileSystemSynchronizedRootGroup section */
|
||||||
@@ -118,6 +135,10 @@
|
|||||||
};
|
};
|
||||||
A6E32D412E762EAE0032EA8A /* bitchatTests */ = {
|
A6E32D412E762EAE0032EA8A /* bitchatTests */ = {
|
||||||
isa = PBXFileSystemSynchronizedRootGroup;
|
isa = PBXFileSystemSynchronizedRootGroup;
|
||||||
|
exceptions = (
|
||||||
|
C5E027A82ECCDFE200BD6012 /* Exceptions for "bitchatTests" folder in "bitchatTests_iOS" target */,
|
||||||
|
C5E027A52ECCDFD700BD6012 /* Exceptions for "bitchatTests" folder in "bitchatTests_macOS" target */,
|
||||||
|
);
|
||||||
path = bitchatTests;
|
path = bitchatTests;
|
||||||
sourceTree = "<group>";
|
sourceTree = "<group>";
|
||||||
};
|
};
|
||||||
@@ -140,7 +161,6 @@
|
|||||||
B5A5CC493FFB3D8966548140 /* Frameworks */ = {
|
B5A5CC493FFB3D8966548140 /* Frameworks */ = {
|
||||||
isa = PBXFrameworksBuildPhase;
|
isa = PBXFrameworksBuildPhase;
|
||||||
files = (
|
files = (
|
||||||
A6F183FD2E948783006A9046 /* tor-nolzma.xcframework in Frameworks */,
|
|
||||||
A6E3E5702E77036A0032EA8A /* BitLogger in Frameworks */,
|
A6E3E5702E77036A0032EA8A /* BitLogger in Frameworks */,
|
||||||
885BBED78092484A5B069461 /* P256K in Frameworks */,
|
885BBED78092484A5B069461 /* P256K in Frameworks */,
|
||||||
A6E3EA7F2E7706720032EA8A /* Tor in Frameworks */,
|
A6E3EA7F2E7706720032EA8A /* Tor in Frameworks */,
|
||||||
@@ -213,6 +233,7 @@
|
|||||||
buildConfigurationList = 1C27B5BA3DB46DDF0DBFEF62 /* Build configuration list for PBXNativeTarget "bitchatTests_macOS" */;
|
buildConfigurationList = 1C27B5BA3DB46DDF0DBFEF62 /* Build configuration list for PBXNativeTarget "bitchatTests_macOS" */;
|
||||||
buildPhases = (
|
buildPhases = (
|
||||||
5C22AA7B9ACC5A861445C769 /* Sources */,
|
5C22AA7B9ACC5A861445C769 /* Sources */,
|
||||||
|
C5E027A42ECCDFD700BD6012 /* Resources */,
|
||||||
);
|
);
|
||||||
buildRules = (
|
buildRules = (
|
||||||
);
|
);
|
||||||
@@ -245,6 +266,7 @@
|
|||||||
buildConfigurationList = 38C4AF6313E5037F25CEF30B /* Build configuration list for PBXNativeTarget "bitchatTests_iOS" */;
|
buildConfigurationList = 38C4AF6313E5037F25CEF30B /* Build configuration list for PBXNativeTarget "bitchatTests_iOS" */;
|
||||||
buildPhases = (
|
buildPhases = (
|
||||||
865C8403EF02C089369A9FCB /* Sources */,
|
865C8403EF02C089369A9FCB /* Sources */,
|
||||||
|
C5E027A72ECCDFE200BD6012 /* Resources */,
|
||||||
);
|
);
|
||||||
buildRules = (
|
buildRules = (
|
||||||
);
|
);
|
||||||
@@ -321,7 +343,7 @@
|
|||||||
packageReferences = (
|
packageReferences = (
|
||||||
B8C407587481BBB190741C93 /* XCRemoteSwiftPackageReference "swift-secp256k1" */,
|
B8C407587481BBB190741C93 /* XCRemoteSwiftPackageReference "swift-secp256k1" */,
|
||||||
A6E3E56E2E77036A0032EA8A /* XCLocalSwiftPackageReference "localPackages/BitLogger" */,
|
A6E3E56E2E77036A0032EA8A /* XCLocalSwiftPackageReference "localPackages/BitLogger" */,
|
||||||
A6E3EA7D2E7706720032EA8A /* XCLocalSwiftPackageReference "localPackages/Tor" */,
|
A6E3EA7D2E7706720032EA8A /* XCLocalSwiftPackageReference "localPackages/Arti" */,
|
||||||
);
|
);
|
||||||
preferredProjectObjectVersion = 90;
|
preferredProjectObjectVersion = 90;
|
||||||
projectDirPath = "";
|
projectDirPath = "";
|
||||||
@@ -343,6 +365,16 @@
|
|||||||
E0A1B2C3D4E5F6012345678D /* relays/online_relays_gps.csv in Resources */,
|
E0A1B2C3D4E5F6012345678D /* relays/online_relays_gps.csv in Resources */,
|
||||||
);
|
);
|
||||||
};
|
};
|
||||||
|
C5E027A42ECCDFD700BD6012 /* Resources */ = {
|
||||||
|
isa = PBXResourcesBuildPhase;
|
||||||
|
files = (
|
||||||
|
);
|
||||||
|
};
|
||||||
|
C5E027A72ECCDFE200BD6012 /* Resources */ = {
|
||||||
|
isa = PBXResourcesBuildPhase;
|
||||||
|
files = (
|
||||||
|
);
|
||||||
|
};
|
||||||
CD6E8F32BC38357473954F97 /* Resources */ = {
|
CD6E8F32BC38357473954F97 /* Resources */ = {
|
||||||
isa = PBXResourcesBuildPhase;
|
isa = PBXResourcesBuildPhase;
|
||||||
files = (
|
files = (
|
||||||
@@ -879,9 +911,9 @@
|
|||||||
isa = XCLocalSwiftPackageReference;
|
isa = XCLocalSwiftPackageReference;
|
||||||
relativePath = localPackages/BitLogger;
|
relativePath = localPackages/BitLogger;
|
||||||
};
|
};
|
||||||
A6E3EA7D2E7706720032EA8A /* XCLocalSwiftPackageReference "localPackages/Tor" */ = {
|
A6E3EA7D2E7706720032EA8A /* XCLocalSwiftPackageReference "localPackages/Arti" */ = {
|
||||||
isa = XCLocalSwiftPackageReference;
|
isa = XCLocalSwiftPackageReference;
|
||||||
relativePath = localPackages/Tor;
|
relativePath = localPackages/Arti;
|
||||||
};
|
};
|
||||||
/* End XCLocalSwiftPackageReference section */
|
/* End XCLocalSwiftPackageReference section */
|
||||||
|
|
||||||
@@ -890,8 +922,8 @@
|
|||||||
isa = XCRemoteSwiftPackageReference;
|
isa = XCRemoteSwiftPackageReference;
|
||||||
repositoryURL = "https://github.com/21-DOT-DEV/swift-secp256k1";
|
repositoryURL = "https://github.com/21-DOT-DEV/swift-secp256k1";
|
||||||
requirement = {
|
requirement = {
|
||||||
kind = upToNextMajorVersion;
|
kind = exactVersion;
|
||||||
minimumVersion = 0.21.1;
|
version = 0.21.1;
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
/* End XCRemoteSwiftPackageReference section */
|
/* End XCRemoteSwiftPackageReference section */
|
||||||
|
|||||||
@@ -98,8 +98,8 @@
|
|||||||
</BuildableProductRunnable>
|
</BuildableProductRunnable>
|
||||||
<EnvironmentVariables>
|
<EnvironmentVariables>
|
||||||
<EnvironmentVariable
|
<EnvironmentVariable
|
||||||
key = "-DBITCHAT_DEV_ALLOW_CLEARNET"
|
key = "BITCHAT_LOG_LEVEL"
|
||||||
value = ""
|
value = "debug"
|
||||||
isEnabled = "YES">
|
isEnabled = "YES">
|
||||||
</EnvironmentVariable>
|
</EnvironmentVariable>
|
||||||
</EnvironmentVariables>
|
</EnvironmentVariables>
|
||||||
|
|||||||
@@ -53,17 +53,20 @@ struct BitchatApp: App {
|
|||||||
// Inject live Noise service into VerificationService to avoid creating new BLE instances
|
// Inject live Noise service into VerificationService to avoid creating new BLE instances
|
||||||
VerificationService.shared.configure(with: chatViewModel.meshService.getNoiseService())
|
VerificationService.shared.configure(with: chatViewModel.meshService.getNoiseService())
|
||||||
// Prewarm Nostr identity and QR to make first VERIFY sheet fast
|
// Prewarm Nostr identity and QR to make first VERIFY sheet fast
|
||||||
|
let nickname = chatViewModel.nickname
|
||||||
DispatchQueue.global(qos: .utility).async {
|
DispatchQueue.global(qos: .utility).async {
|
||||||
let npub = try? idBridge.getCurrentNostrIdentity()?.npub
|
let npub = try? idBridge.getCurrentNostrIdentity()?.npub
|
||||||
_ = VerificationService.shared.buildMyQRString(nickname: chatViewModel.nickname, npub: npub)
|
_ = VerificationService.shared.buildMyQRString(nickname: nickname, npub: npub)
|
||||||
}
|
}
|
||||||
#if os(iOS)
|
|
||||||
appDelegate.chatViewModel = chatViewModel
|
appDelegate.chatViewModel = chatViewModel
|
||||||
#elseif os(macOS)
|
|
||||||
appDelegate.chatViewModel = chatViewModel
|
|
||||||
#endif
|
|
||||||
// Initialize network activation policy; will start Tor/Nostr only when allowed
|
// Initialize network activation policy; will start Tor/Nostr only when allowed
|
||||||
NetworkActivationService.shared.start()
|
NetworkActivationService.shared.start()
|
||||||
|
|
||||||
|
// Start presence service (will wait for Tor readiness)
|
||||||
|
GeohashPresenceService.shared.start()
|
||||||
|
|
||||||
// Check for shared content
|
// Check for shared content
|
||||||
checkForSharedContent()
|
checkForSharedContent()
|
||||||
}
|
}
|
||||||
@@ -189,6 +192,10 @@ final class AppDelegate: NSObject, UIApplicationDelegate {
|
|||||||
func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey : Any]? = nil) -> Bool {
|
func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey : Any]? = nil) -> Bool {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func applicationWillTerminate(_ application: UIApplication) {
|
||||||
|
chatViewModel?.applicationWillTerminate()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
@@ -246,11 +253,16 @@ final class NotificationDelegate: NSObject, UNUserNotificationCenterDelegate {
|
|||||||
// Get peer ID from userInfo
|
// Get peer ID from userInfo
|
||||||
if let peerID = userInfo["peerID"] as? String {
|
if let peerID = userInfo["peerID"] as? String {
|
||||||
// Don't show notification if the private chat is already open
|
// Don't show notification if the private chat is already open
|
||||||
if chatViewModel?.selectedPrivateChatPeer == peerID {
|
// Access main-actor-isolated property via Task
|
||||||
|
Task { @MainActor in
|
||||||
|
if self.chatViewModel?.selectedPrivateChatPeer == PeerID(str: peerID) {
|
||||||
completionHandler([])
|
completionHandler([])
|
||||||
return
|
} else {
|
||||||
|
completionHandler([.banner, .sound])
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
}
|
}
|
||||||
// Suppress geohash activity notification if we're already in that geohash channel
|
// Suppress geohash activity notification if we're already in that geohash channel
|
||||||
if identifier.hasPrefix("geo-activity-"),
|
if identifier.hasPrefix("geo-activity-"),
|
||||||
|
|||||||
@@ -0,0 +1,203 @@
|
|||||||
|
import Foundation
|
||||||
|
import ImageIO
|
||||||
|
import UniformTypeIdentifiers
|
||||||
|
#if os(iOS)
|
||||||
|
import UIKit
|
||||||
|
#else
|
||||||
|
import AppKit
|
||||||
|
#endif
|
||||||
|
|
||||||
|
enum ImageUtilsError: Error {
|
||||||
|
case invalidImage
|
||||||
|
case encodingFailed
|
||||||
|
}
|
||||||
|
|
||||||
|
enum ImageUtils {
|
||||||
|
private static let compressionQuality: CGFloat = 0.82
|
||||||
|
private static let targetImageBytes: Int = 45_000
|
||||||
|
|
||||||
|
static func processImage(at url: URL, maxDimension: CGFloat = 448) throws -> URL {
|
||||||
|
// 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 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)
|
||||||
|
#if os(iOS)
|
||||||
|
guard let image = UIImage(data: data) else { throw ImageUtilsError.invalidImage }
|
||||||
|
return try processImage(image, maxDimension: maxDimension)
|
||||||
|
#else
|
||||||
|
guard let image = NSImage(data: data) else { throw ImageUtilsError.invalidImage }
|
||||||
|
return try processImage(image, maxDimension: maxDimension)
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
#if os(iOS)
|
||||||
|
static func processImage(_ image: UIImage, maxDimension: CGFloat = 448) throws -> URL {
|
||||||
|
return try autoreleasepool {
|
||||||
|
// Scale the image first
|
||||||
|
let scaled = scaledImage(image, maxDimension: maxDimension)
|
||||||
|
|
||||||
|
// Get CGImage from UIImage - this is the key to stripping metadata
|
||||||
|
guard let cgImage = scaled.cgImage else {
|
||||||
|
throw ImageUtilsError.encodingFailed
|
||||||
|
}
|
||||||
|
|
||||||
|
// Use CGImageDestination to encode without metadata (same as macOS)
|
||||||
|
var quality = compressionQuality
|
||||||
|
guard var jpegData = encodeJPEG(from: cgImage, quality: quality) else {
|
||||||
|
throw ImageUtilsError.encodingFailed
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compress to target size
|
||||||
|
while jpegData.count > targetImageBytes && quality > 0.3 {
|
||||||
|
quality -= 0.1
|
||||||
|
autoreleasepool {
|
||||||
|
if let next = encodeJPEG(from: cgImage, quality: quality) {
|
||||||
|
jpegData = next
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let outputURL = try makeOutputURL()
|
||||||
|
try jpegData.write(to: outputURL, options: .atomic)
|
||||||
|
return outputURL
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func scaledImage(_ image: UIImage, maxDimension: CGFloat) -> UIImage {
|
||||||
|
let size = image.size
|
||||||
|
let maxSide = max(size.width, size.height)
|
||||||
|
guard maxSide > maxDimension else { return image }
|
||||||
|
let scale = maxDimension / maxSide
|
||||||
|
let newSize = CGSize(width: size.width * scale, height: size.height * scale)
|
||||||
|
|
||||||
|
// Draw into a new context to get a clean CGImage without metadata
|
||||||
|
UIGraphicsBeginImageContextWithOptions(newSize, true, 1.0)
|
||||||
|
image.draw(in: CGRect(origin: .zero, size: newSize))
|
||||||
|
let rendered = UIGraphicsGetImageFromCurrentImageContext()
|
||||||
|
UIGraphicsEndImageContext()
|
||||||
|
return rendered ?? image
|
||||||
|
}
|
||||||
|
|
||||||
|
// Shared EXIF-stripping JPEG encoder for both iOS and macOS
|
||||||
|
private static func encodeJPEG(from cgImage: CGImage, quality: CGFloat) -> Data? {
|
||||||
|
guard let data = CFDataCreateMutable(nil, 0) else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
guard let destination = CGImageDestinationCreateWithData(data, UTType.jpeg.identifier as CFString, 1, nil) else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
// Security: Strip ALL metadata (EXIF, GPS, TIFF, IPTC, XMP)
|
||||||
|
// By only specifying compression quality and no metadata keys,
|
||||||
|
// we ensure a clean JPEG with no privacy-leaking information
|
||||||
|
let options: [CFString: Any] = [
|
||||||
|
kCGImageDestinationLossyCompressionQuality: quality
|
||||||
|
]
|
||||||
|
CGImageDestinationAddImage(destination, cgImage, options as CFDictionary)
|
||||||
|
guard CGImageDestinationFinalize(destination) else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return data as Data
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
static func processImage(_ image: NSImage, maxDimension: CGFloat = 448) throws -> URL {
|
||||||
|
return try autoreleasepool {
|
||||||
|
let scaled = scaledImage(image, maxDimension: maxDimension)
|
||||||
|
guard let inputCG = scaled.cgImage(forProposedRect: nil, context: nil, hints: nil) else {
|
||||||
|
throw ImageUtilsError.encodingFailed
|
||||||
|
}
|
||||||
|
let width = inputCG.width
|
||||||
|
let height = inputCG.height
|
||||||
|
let colorSpace = CGColorSpace(name: CGColorSpace.sRGB) ?? CGColorSpaceCreateDeviceRGB()
|
||||||
|
guard let context = CGContext(
|
||||||
|
data: nil,
|
||||||
|
width: width,
|
||||||
|
height: height,
|
||||||
|
bitsPerComponent: 8,
|
||||||
|
bytesPerRow: 0,
|
||||||
|
space: colorSpace,
|
||||||
|
bitmapInfo: CGImageAlphaInfo.premultipliedLast.rawValue
|
||||||
|
) else {
|
||||||
|
throw ImageUtilsError.encodingFailed
|
||||||
|
}
|
||||||
|
context.draw(inputCG, in: CGRect(x: 0, y: 0, width: width, height: height))
|
||||||
|
guard let cgImage = context.makeImage() else {
|
||||||
|
throw ImageUtilsError.encodingFailed
|
||||||
|
}
|
||||||
|
var quality = compressionQuality
|
||||||
|
guard var jpegData = encodeJPEG(from: cgImage, quality: quality) else {
|
||||||
|
throw ImageUtilsError.encodingFailed
|
||||||
|
}
|
||||||
|
while jpegData.count > targetImageBytes && quality > 0.3 {
|
||||||
|
quality -= 0.1
|
||||||
|
autoreleasepool {
|
||||||
|
if let next = encodeJPEG(from: cgImage, quality: quality) {
|
||||||
|
jpegData = next
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let outputURL = try makeOutputURL()
|
||||||
|
try jpegData.write(to: outputURL, options: .atomic)
|
||||||
|
return outputURL
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func scaledImage(_ image: NSImage, maxDimension: CGFloat) -> NSImage {
|
||||||
|
let size = image.size
|
||||||
|
let maxSide = max(size.width, size.height)
|
||||||
|
guard maxSide > maxDimension else { return image }
|
||||||
|
let scale = maxDimension / maxSide
|
||||||
|
let newSize = NSSize(width: size.width * scale, height: size.height * scale)
|
||||||
|
let scaledImage = NSImage(size: newSize)
|
||||||
|
scaledImage.lockFocus()
|
||||||
|
image.draw(in: NSRect(origin: .zero, size: newSize),
|
||||||
|
from: NSRect(origin: .zero, size: size),
|
||||||
|
operation: .copy,
|
||||||
|
fraction: 1.0)
|
||||||
|
scaledImage.unlockFocus()
|
||||||
|
return scaledImage
|
||||||
|
}
|
||||||
|
|
||||||
|
// Shared EXIF-stripping JPEG encoder for both iOS and macOS
|
||||||
|
private static func encodeJPEG(from cgImage: CGImage, quality: CGFloat) -> Data? {
|
||||||
|
guard let data = CFDataCreateMutable(nil, 0) else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
guard let destination = CGImageDestinationCreateWithData(data, UTType.jpeg.identifier as CFString, 1, nil) else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
// Security: Strip ALL metadata (EXIF, GPS, TIFF, IPTC, XMP)
|
||||||
|
// By only specifying compression quality and no metadata keys,
|
||||||
|
// we ensure a clean JPEG with no privacy-leaking information
|
||||||
|
let options: [CFString: Any] = [
|
||||||
|
kCGImageDestinationLossyCompressionQuality: quality
|
||||||
|
]
|
||||||
|
CGImageDestinationAddImage(destination, cgImage, options as CFDictionary)
|
||||||
|
guard CGImageDestinationFinalize(destination) else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return data as Data
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
private static func makeOutputURL() throws -> URL {
|
||||||
|
let formatter = DateFormatter()
|
||||||
|
formatter.dateFormat = "yyyyMMdd_HHmmss"
|
||||||
|
let fileName = "img_\(formatter.string(from: Date())).jpg"
|
||||||
|
|
||||||
|
let directory = try applicationFilesDirectory().appendingPathComponent("images/outgoing", isDirectory: true)
|
||||||
|
try FileManager.default.createDirectory(at: directory, withIntermediateDirectories: true, attributes: nil)
|
||||||
|
return directory.appendingPathComponent(fileName)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func applicationFilesDirectory() throws -> URL {
|
||||||
|
let base = try FileManager.default.url(for: .applicationSupportDirectory, in: .userDomainMask, appropriateFor: nil, create: true)
|
||||||
|
return base.appendingPathComponent("files", isDirectory: true)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,193 @@
|
|||||||
|
import Foundation
|
||||||
|
import AVFoundation
|
||||||
|
import BitLogger
|
||||||
|
|
||||||
|
/// Controls playback for a single voice note and coordinates exclusive playback across the app.
|
||||||
|
final class VoiceNotePlaybackController: NSObject, ObservableObject, AVAudioPlayerDelegate {
|
||||||
|
@Published private(set) var isPlaying: Bool = false
|
||||||
|
@Published private(set) var currentTime: TimeInterval = 0
|
||||||
|
@Published private(set) var duration: TimeInterval = 0
|
||||||
|
@Published private(set) var progress: Double = 0
|
||||||
|
|
||||||
|
private var player: AVAudioPlayer?
|
||||||
|
private var timer: Timer?
|
||||||
|
private var url: URL
|
||||||
|
|
||||||
|
init(url: URL) {
|
||||||
|
self.url = url
|
||||||
|
super.init()
|
||||||
|
// Don't load anything eagerly - wait until user interaction or view is fully displayed
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadDuration() {
|
||||||
|
guard duration == 0 else { return }
|
||||||
|
|
||||||
|
DispatchQueue.global(qos: .utility).async { [weak self] in
|
||||||
|
guard let self = self else { return }
|
||||||
|
do {
|
||||||
|
let player = try AVAudioPlayer(contentsOf: self.url)
|
||||||
|
let loadedDuration = player.duration
|
||||||
|
DispatchQueue.main.async { [weak self] in
|
||||||
|
guard let self = self, self.duration == 0 else { return }
|
||||||
|
self.duration = loadedDuration
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
SecureLogger.error("Failed to load audio duration: \(error)", category: .session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
deinit {
|
||||||
|
timer?.invalidate()
|
||||||
|
}
|
||||||
|
|
||||||
|
func replaceURL(_ url: URL) {
|
||||||
|
guard url != self.url else { return }
|
||||||
|
stop()
|
||||||
|
self.url = url
|
||||||
|
player = nil
|
||||||
|
duration = 0
|
||||||
|
// Duration will be loaded on demand when needed
|
||||||
|
}
|
||||||
|
|
||||||
|
func togglePlayback() {
|
||||||
|
isPlaying ? pause() : play()
|
||||||
|
}
|
||||||
|
|
||||||
|
func play() {
|
||||||
|
guard ensurePlayerReady() else { return }
|
||||||
|
VoiceNotePlaybackCoordinator.shared.activate(self)
|
||||||
|
player?.play()
|
||||||
|
startTimer()
|
||||||
|
updateProgress()
|
||||||
|
isPlaying = true
|
||||||
|
}
|
||||||
|
|
||||||
|
func pause() {
|
||||||
|
player?.pause()
|
||||||
|
stopTimer()
|
||||||
|
updateProgress()
|
||||||
|
isPlaying = false
|
||||||
|
}
|
||||||
|
|
||||||
|
func stop() {
|
||||||
|
player?.stop()
|
||||||
|
player?.currentTime = 0
|
||||||
|
stopTimer()
|
||||||
|
updateProgress()
|
||||||
|
isPlaying = false
|
||||||
|
VoiceNotePlaybackCoordinator.shared.deactivate(self)
|
||||||
|
}
|
||||||
|
|
||||||
|
func seek(to fraction: Double) {
|
||||||
|
guard ensurePlayerReady() else { return }
|
||||||
|
let clamped = max(0, min(1, fraction))
|
||||||
|
if let player = player {
|
||||||
|
player.currentTime = clamped * player.duration
|
||||||
|
if isPlaying {
|
||||||
|
player.play()
|
||||||
|
}
|
||||||
|
updateProgress()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - AVAudioPlayerDelegate
|
||||||
|
|
||||||
|
func audioPlayerDidFinishPlaying(_ player: AVAudioPlayer, successfully flag: Bool) {
|
||||||
|
// Delegate callback may be on background thread - ensure main thread for UI updates
|
||||||
|
DispatchQueue.main.async { [weak self] in
|
||||||
|
guard let self = self else { return }
|
||||||
|
self.stopTimer()
|
||||||
|
self.updateProgress()
|
||||||
|
self.isPlaying = false
|
||||||
|
VoiceNotePlaybackCoordinator.shared.deactivate(self)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Private Helpers
|
||||||
|
|
||||||
|
private func preparePlayer(for url: URL) {
|
||||||
|
// Prepare player synchronously (only called when playback is requested)
|
||||||
|
do {
|
||||||
|
let player = try AVAudioPlayer(contentsOf: url)
|
||||||
|
player.delegate = self
|
||||||
|
player.prepareToPlay()
|
||||||
|
self.player = player
|
||||||
|
duration = player.duration
|
||||||
|
currentTime = player.currentTime
|
||||||
|
progress = duration > 0 ? currentTime / duration : 0
|
||||||
|
} catch {
|
||||||
|
SecureLogger.error("Voice note playback failed for \(url.lastPathComponent): \(error)", category: .session)
|
||||||
|
player = nil
|
||||||
|
duration = 0
|
||||||
|
currentTime = 0
|
||||||
|
progress = 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func ensurePlayerReady() -> Bool {
|
||||||
|
if player == nil {
|
||||||
|
preparePlayer(for: url)
|
||||||
|
}
|
||||||
|
#if os(iOS)
|
||||||
|
let session = AVAudioSession.sharedInstance()
|
||||||
|
do {
|
||||||
|
try session.setCategory(.playback, mode: .spokenAudio, options: [.mixWithOthers])
|
||||||
|
try session.setActive(true, options: [])
|
||||||
|
} catch {
|
||||||
|
SecureLogger.error("Failed to activate audio session: \(error)", category: .session)
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
return player != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
private func startTimer() {
|
||||||
|
if timer != nil { return }
|
||||||
|
timer = Timer.scheduledTimer(withTimeInterval: 0.05, repeats: true) { [weak self] _ in
|
||||||
|
self?.updateProgress()
|
||||||
|
}
|
||||||
|
if let timer = timer {
|
||||||
|
RunLoop.main.add(timer, forMode: .common)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func stopTimer() {
|
||||||
|
timer?.invalidate()
|
||||||
|
timer = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
private func updateProgress() {
|
||||||
|
guard let player = player else {
|
||||||
|
currentTime = 0
|
||||||
|
duration = 0
|
||||||
|
progress = 0
|
||||||
|
return
|
||||||
|
}
|
||||||
|
currentTime = player.currentTime
|
||||||
|
duration = player.duration
|
||||||
|
progress = duration > 0 ? currentTime / duration : 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ensures only one voice note plays at a time.
|
||||||
|
final class VoiceNotePlaybackCoordinator {
|
||||||
|
static let shared = VoiceNotePlaybackCoordinator()
|
||||||
|
|
||||||
|
private weak var activeController: VoiceNotePlaybackController?
|
||||||
|
|
||||||
|
private init() {}
|
||||||
|
|
||||||
|
func activate(_ controller: VoiceNotePlaybackController) {
|
||||||
|
if activeController === controller {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
activeController?.pause()
|
||||||
|
activeController = controller
|
||||||
|
}
|
||||||
|
|
||||||
|
func deactivate(_ controller: VoiceNotePlaybackController) {
|
||||||
|
if activeController === controller {
|
||||||
|
activeController = nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,179 @@
|
|||||||
|
import Foundation
|
||||||
|
import AVFoundation
|
||||||
|
|
||||||
|
/// Manages audio capture for mesh voice notes with predictable encoding settings.
|
||||||
|
/// Recording runs on an internal serial queue to avoid AVAudioSession contention.
|
||||||
|
final class VoiceRecorder: NSObject, AVAudioRecorderDelegate {
|
||||||
|
enum RecorderError: Error {
|
||||||
|
case microphoneAccessDenied
|
||||||
|
case recorderInitializationFailed
|
||||||
|
case recordingInProgress
|
||||||
|
}
|
||||||
|
|
||||||
|
static let shared = VoiceRecorder()
|
||||||
|
|
||||||
|
private let queue = DispatchQueue(label: "com.bitchat.voice-recorder")
|
||||||
|
private let paddingInterval: TimeInterval = 0.5
|
||||||
|
private let maxRecordingDuration: TimeInterval = 120
|
||||||
|
|
||||||
|
private var recorder: AVAudioRecorder?
|
||||||
|
private var currentURL: URL?
|
||||||
|
private var stopWorkItem: DispatchWorkItem?
|
||||||
|
|
||||||
|
private override init() {
|
||||||
|
super.init()
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Permissions
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
func requestPermission() async -> Bool {
|
||||||
|
#if os(iOS)
|
||||||
|
return await withCheckedContinuation { continuation in
|
||||||
|
AVAudioSession.sharedInstance().requestRecordPermission { granted in
|
||||||
|
continuation.resume(returning: granted)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#elseif os(macOS)
|
||||||
|
return await withCheckedContinuation { continuation in
|
||||||
|
AVCaptureDevice.requestAccess(for: .audio) { granted in
|
||||||
|
continuation.resume(returning: granted)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
return true
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Recording Lifecycle
|
||||||
|
|
||||||
|
func startRecording() throws -> URL {
|
||||||
|
try queue.sync {
|
||||||
|
if recorder?.isRecording == true {
|
||||||
|
throw RecorderError.recordingInProgress
|
||||||
|
}
|
||||||
|
|
||||||
|
#if os(iOS)
|
||||||
|
let session = AVAudioSession.sharedInstance()
|
||||||
|
guard session.recordPermission == .granted else {
|
||||||
|
throw RecorderError.microphoneAccessDenied
|
||||||
|
}
|
||||||
|
#if targetEnvironment(simulator)
|
||||||
|
// allowBluetoothHFP is not available on iOS Simulator
|
||||||
|
try session.setCategory(
|
||||||
|
.playAndRecord,
|
||||||
|
mode: .default,
|
||||||
|
options: [.defaultToSpeaker, .allowBluetoothA2DP]
|
||||||
|
)
|
||||||
|
#else
|
||||||
|
try session.setCategory(
|
||||||
|
.playAndRecord,
|
||||||
|
mode: .default,
|
||||||
|
options: [.defaultToSpeaker, .allowBluetoothA2DP, .allowBluetoothHFP]
|
||||||
|
)
|
||||||
|
#endif
|
||||||
|
try session.setActive(true, options: .notifyOthersOnDeactivation)
|
||||||
|
#endif
|
||||||
|
#if os(macOS)
|
||||||
|
guard AVCaptureDevice.authorizationStatus(for: .audio) == .authorized else {
|
||||||
|
throw RecorderError.microphoneAccessDenied
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
let outputURL = try makeOutputURL()
|
||||||
|
let settings: [String: Any] = [
|
||||||
|
AVFormatIDKey: kAudioFormatMPEG4AAC,
|
||||||
|
AVSampleRateKey: 16_000,
|
||||||
|
AVNumberOfChannelsKey: 1,
|
||||||
|
AVEncoderBitRateKey: 16_000
|
||||||
|
]
|
||||||
|
|
||||||
|
let audioRecorder = try AVAudioRecorder(url: outputURL, settings: settings)
|
||||||
|
audioRecorder.delegate = self
|
||||||
|
audioRecorder.isMeteringEnabled = true
|
||||||
|
audioRecorder.prepareToRecord()
|
||||||
|
audioRecorder.record(forDuration: maxRecordingDuration)
|
||||||
|
|
||||||
|
recorder = audioRecorder
|
||||||
|
currentURL = outputURL
|
||||||
|
stopWorkItem?.cancel()
|
||||||
|
stopWorkItem = nil
|
||||||
|
return outputURL
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func stopRecording(completion: @escaping (URL?) -> Void) {
|
||||||
|
queue.async { [weak self] in
|
||||||
|
guard let self = self, let recorder = self.recorder, recorder.isRecording else {
|
||||||
|
completion(self?.currentURL)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let item = DispatchWorkItem { [weak self] in
|
||||||
|
guard let self = self else { return }
|
||||||
|
recorder.stop()
|
||||||
|
self.cleanupSession()
|
||||||
|
let url = self.currentURL
|
||||||
|
self.recorder = nil
|
||||||
|
self.currentURL = url
|
||||||
|
completion(url)
|
||||||
|
}
|
||||||
|
self.stopWorkItem = item
|
||||||
|
self.queue.asyncAfter(deadline: .now() + self.paddingInterval, execute: item)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func cancelRecording() {
|
||||||
|
queue.async { [weak self] in
|
||||||
|
guard let self = self else { return }
|
||||||
|
self.stopWorkItem?.cancel()
|
||||||
|
self.stopWorkItem = nil
|
||||||
|
if let recorder = self.recorder, recorder.isRecording {
|
||||||
|
recorder.stop()
|
||||||
|
}
|
||||||
|
self.cleanupSession()
|
||||||
|
if let url = self.currentURL {
|
||||||
|
try? FileManager.default.removeItem(at: url)
|
||||||
|
}
|
||||||
|
self.recorder = nil
|
||||||
|
self.currentURL = nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Metering
|
||||||
|
|
||||||
|
func currentAveragePower() -> Float {
|
||||||
|
queue.sync {
|
||||||
|
recorder?.updateMeters()
|
||||||
|
return recorder?.averagePower(forChannel: 0) ?? -160
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Helpers
|
||||||
|
|
||||||
|
private func makeOutputURL() throws -> URL {
|
||||||
|
let formatter = DateFormatter()
|
||||||
|
formatter.dateFormat = "yyyyMMdd_HHmmss"
|
||||||
|
let fileName = "voice_\(formatter.string(from: Date())).m4a"
|
||||||
|
|
||||||
|
let baseDirectory = try applicationFilesDirectory().appendingPathComponent("voicenotes/outgoing", isDirectory: true)
|
||||||
|
try FileManager.default.createDirectory(at: baseDirectory, withIntermediateDirectories: true, attributes: nil)
|
||||||
|
return baseDirectory.appendingPathComponent(fileName)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func applicationFilesDirectory() throws -> URL {
|
||||||
|
#if os(iOS)
|
||||||
|
return try FileManager.default.url(for: .applicationSupportDirectory, in: .userDomainMask, appropriateFor: nil, create: true)
|
||||||
|
.appendingPathComponent("files", isDirectory: true)
|
||||||
|
#else
|
||||||
|
let base = try FileManager.default.url(for: .applicationSupportDirectory, in: .userDomainMask, appropriateFor: nil, create: true)
|
||||||
|
return base.appendingPathComponent("files", isDirectory: true)
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
private func cleanupSession() {
|
||||||
|
#if os(iOS)
|
||||||
|
try? AVAudioSession.sharedInstance().setActive(false, options: .notifyOthersOnDeactivation)
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,113 @@
|
|||||||
|
import AVFoundation
|
||||||
|
import Foundation
|
||||||
|
import BitLogger
|
||||||
|
|
||||||
|
/// Generates and caches downsampled waveforms for audio files so UI rendering is cheap.
|
||||||
|
final class WaveformCache {
|
||||||
|
static let shared = WaveformCache()
|
||||||
|
|
||||||
|
private let queue = DispatchQueue(label: "com.bitchat.waveform-cache", attributes: .concurrent)
|
||||||
|
private var cache: [URL: (waveform: [Float], lastAccess: Date)] = [:]
|
||||||
|
private let maxCacheSize = 20 // Limit cache to prevent unbounded memory growth
|
||||||
|
|
||||||
|
private init() {}
|
||||||
|
|
||||||
|
func cachedWaveform(for url: URL) -> [Float]? {
|
||||||
|
queue.sync {
|
||||||
|
guard let entry = cache[url] else { return nil }
|
||||||
|
return entry.waveform
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func waveform(for url: URL, bins: Int = 120, completion: @escaping ([Float]) -> Void) {
|
||||||
|
queue.async { [weak self] in
|
||||||
|
guard let self = self else { return }
|
||||||
|
|
||||||
|
// Check cache (read-only, no update needed on cache hit for performance)
|
||||||
|
if let entry = self.cache[url] {
|
||||||
|
DispatchQueue.main.async { completion(entry.waveform) }
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
guard let computed = self.computeWaveform(url: url, bins: bins) else {
|
||||||
|
DispatchQueue.main.async { completion([]) }
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
self.queue.async(flags: .barrier) { [weak self] in
|
||||||
|
guard let self = self else { return }
|
||||||
|
|
||||||
|
// Evict oldest entry if cache is full
|
||||||
|
if self.cache.count >= self.maxCacheSize {
|
||||||
|
if let oldest = self.cache.min(by: { $0.value.lastAccess < $1.value.lastAccess }) {
|
||||||
|
self.cache.removeValue(forKey: oldest.key)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
self.cache[url] = (computed, Date())
|
||||||
|
}
|
||||||
|
DispatchQueue.main.async { completion(computed) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func purge(url: URL) {
|
||||||
|
queue.async(flags: .barrier) { [weak self] in
|
||||||
|
self?.cache.removeValue(forKey: url)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func purgeAll() {
|
||||||
|
queue.async(flags: .barrier) { [weak self] in
|
||||||
|
self?.cache.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func computeWaveform(url: URL, bins: Int) -> [Float]? {
|
||||||
|
guard bins > 0 else { return nil }
|
||||||
|
// Use autoreleasepool to manage memory from audio buffer allocations
|
||||||
|
return autoreleasepool {
|
||||||
|
do {
|
||||||
|
let audioFile = try AVAudioFile(forReading: url)
|
||||||
|
let length = Int(audioFile.length)
|
||||||
|
guard length > 0 else { return nil }
|
||||||
|
|
||||||
|
guard let buffer = AVAudioPCMBuffer(pcmFormat: audioFile.processingFormat, frameCapacity: AVAudioFrameCount(length)) else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
try audioFile.read(into: buffer, frameCount: AVAudioFrameCount(length))
|
||||||
|
guard let channelData = buffer.floatChannelData else { return nil }
|
||||||
|
|
||||||
|
let channelCount = Int(audioFile.processingFormat.channelCount)
|
||||||
|
let frameLength = Int(buffer.frameLength)
|
||||||
|
let samplesPerBin = max(1, frameLength / bins)
|
||||||
|
|
||||||
|
var magnitudes: [Float] = Array(repeating: 0, count: bins)
|
||||||
|
for bin in 0..<bins {
|
||||||
|
let start = bin * samplesPerBin
|
||||||
|
let end = min(frameLength, start + samplesPerBin)
|
||||||
|
if start >= end { break }
|
||||||
|
|
||||||
|
var sum: Float = 0
|
||||||
|
var sampleCount = 0
|
||||||
|
for frame in start..<end {
|
||||||
|
var sampleValue: Float = 0
|
||||||
|
for channel in 0..<channelCount {
|
||||||
|
sampleValue += fabsf(channelData[channel][frame])
|
||||||
|
}
|
||||||
|
sum += sampleValue / Float(channelCount)
|
||||||
|
sampleCount += 1
|
||||||
|
}
|
||||||
|
magnitudes[bin] = sampleCount > 0 ? sum / Float(sampleCount) : 0
|
||||||
|
}
|
||||||
|
|
||||||
|
if let maxMagnitude = magnitudes.max(), maxMagnitude > 0 {
|
||||||
|
magnitudes = magnitudes.map { min($0 / maxMagnitude, 1.0) }
|
||||||
|
}
|
||||||
|
return magnitudes
|
||||||
|
} catch {
|
||||||
|
SecureLogger.error("Waveform extraction failed for \(url.lastPathComponent): \(error)", category: .session)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -37,6 +37,10 @@
|
|||||||
<string>bitchat uses Bluetooth to discover and connect with other bitchat users nearby.</string>
|
<string>bitchat uses Bluetooth to discover and connect with other bitchat users nearby.</string>
|
||||||
<key>NSCameraUsageDescription</key>
|
<key>NSCameraUsageDescription</key>
|
||||||
<string>bitchat uses the camera to scan QR codes to verify peers.</string>
|
<string>bitchat uses the camera to scan QR codes to verify peers.</string>
|
||||||
|
<key>NSPhotoLibraryUsageDescription</key>
|
||||||
|
<string>bitchat lets you pick images from your photo library to share with nearby peers.</string>
|
||||||
|
<key>NSMicrophoneUsageDescription</key>
|
||||||
|
<string>bitchat uses the microphone to record voice notes that relay across the mesh.</string>
|
||||||
<key>NSLocationWhenInUseUsageDescription</key>
|
<key>NSLocationWhenInUseUsageDescription</key>
|
||||||
<string>bitchat uses your approximate location to compute local geohash channels for optional public chats. Exact GPS is never shared.</string>
|
<string>bitchat uses your approximate location to compute local geohash channels for optional public chats. Exact GPS is never shared.</string>
|
||||||
<key>UIBackgroundModes</key>
|
<key>UIBackgroundModes</key>
|
||||||
|
|||||||
+15690
-14470
File diff suppressed because it is too large
Load Diff
@@ -21,9 +21,10 @@ struct BitchatPacket: Codable {
|
|||||||
let payload: Data
|
let payload: Data
|
||||||
var signature: Data?
|
var signature: Data?
|
||||||
var ttl: UInt8
|
var ttl: UInt8
|
||||||
|
var route: [Data]?
|
||||||
|
|
||||||
init(type: UInt8, senderID: Data, recipientID: Data?, timestamp: UInt64, payload: Data, signature: Data?, ttl: UInt8) {
|
init(type: UInt8, senderID: Data, recipientID: Data?, timestamp: UInt64, payload: Data, signature: Data?, ttl: UInt8, version: UInt8 = 1, route: [Data]? = nil) {
|
||||||
self.version = 1
|
self.version = version
|
||||||
self.type = type
|
self.type = type
|
||||||
self.senderID = senderID
|
self.senderID = senderID
|
||||||
self.recipientID = recipientID
|
self.recipientID = recipientID
|
||||||
@@ -31,6 +32,7 @@ struct BitchatPacket: Codable {
|
|||||||
self.payload = payload
|
self.payload = payload
|
||||||
self.signature = signature
|
self.signature = signature
|
||||||
self.ttl = ttl
|
self.ttl = ttl
|
||||||
|
self.route = route
|
||||||
}
|
}
|
||||||
|
|
||||||
// Convenience initializer for new binary format
|
// Convenience initializer for new binary format
|
||||||
@@ -53,6 +55,7 @@ struct BitchatPacket: Codable {
|
|||||||
self.payload = payload
|
self.payload = payload
|
||||||
self.signature = nil
|
self.signature = nil
|
||||||
self.ttl = ttl
|
self.ttl = ttl
|
||||||
|
self.route = nil
|
||||||
}
|
}
|
||||||
|
|
||||||
var data: Data? {
|
var data: Data? {
|
||||||
@@ -80,7 +83,9 @@ struct BitchatPacket: Codable {
|
|||||||
timestamp: timestamp,
|
timestamp: timestamp,
|
||||||
payload: payload,
|
payload: payload,
|
||||||
signature: nil, // Remove signature for signing
|
signature: nil, // Remove signature for signing
|
||||||
ttl: 0 // Use fixed TTL=0 for signing to ensure relay compatibility
|
ttl: 0, // Use fixed TTL=0 for signing to ensure relay compatibility
|
||||||
|
version: version,
|
||||||
|
route: route
|
||||||
)
|
)
|
||||||
return BinaryProtocol.encode(unsignedPacket)
|
return BinaryProtocol.encode(unsignedPacket)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,58 @@
|
|||||||
|
//
|
||||||
|
// CommandsInfo.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
// For more information, see <https://unlicense.org>
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
// MARK: - CommandInfo Enum
|
||||||
|
|
||||||
|
enum CommandInfo: String, Identifiable {
|
||||||
|
case block
|
||||||
|
case clear
|
||||||
|
case hug
|
||||||
|
case message = "dm"
|
||||||
|
case slap
|
||||||
|
case unblock
|
||||||
|
case who
|
||||||
|
case favorite
|
||||||
|
case unfavorite
|
||||||
|
|
||||||
|
var id: String { rawValue }
|
||||||
|
|
||||||
|
var alias: String { "/" + rawValue }
|
||||||
|
|
||||||
|
var placeholder: String? {
|
||||||
|
switch self {
|
||||||
|
case .block, .hug, .message, .slap, .unblock, .favorite, .unfavorite:
|
||||||
|
return "<" + String(localized: "content.input.nickname_placeholder") + ">"
|
||||||
|
case .clear, .who:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var description: String {
|
||||||
|
switch self {
|
||||||
|
case .block: String(localized: "content.commands.block")
|
||||||
|
case .clear: String(localized: "content.commands.clear")
|
||||||
|
case .hug: String(localized: "content.commands.hug")
|
||||||
|
case .message: String(localized: "content.commands.message")
|
||||||
|
case .slap: String(localized: "content.commands.slap")
|
||||||
|
case .unblock: String(localized: "content.commands.unblock")
|
||||||
|
case .who: String(localized: "content.commands.who")
|
||||||
|
case .favorite: String(localized: "content.commands.favorite")
|
||||||
|
case .unfavorite: String(localized: "content.commands.unfavorite")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static func all(isGeoPublic: Bool, isGeoDM: Bool) -> [CommandInfo] {
|
||||||
|
let baseCommands: [CommandInfo] = [.block, .unblock, .clear, .hug, .message, .slap, .who]
|
||||||
|
if isGeoPublic || isGeoDM {
|
||||||
|
return baseCommands + [.favorite, .unfavorite]
|
||||||
|
}
|
||||||
|
return baseCommands
|
||||||
|
}
|
||||||
|
}
|
||||||
+22
-18
@@ -35,7 +35,7 @@ struct PeerID: Equatable, Hashable {
|
|||||||
// Private so the callers have to go through a convenience init
|
// Private so the callers have to go through a convenience init
|
||||||
private init(prefix: Prefix, bare: any StringProtocol) {
|
private init(prefix: Prefix, bare: any StringProtocol) {
|
||||||
self.prefix = prefix
|
self.prefix = prefix
|
||||||
self.bare = String(bare)
|
self.bare = String(bare).lowercased()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -76,6 +76,12 @@ extension PeerID {
|
|||||||
init(hexData: Data) {
|
init(hexData: Data) {
|
||||||
self.init(str: hexData.hexEncodedString())
|
self.init(str: hexData.hexEncodedString())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Convenience init to "hide" hex-encoding implementation detail
|
||||||
|
init?(hexData: Data?) {
|
||||||
|
guard let hexData else { return nil }
|
||||||
|
self.init(hexData: hexData)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Noise Public Key Helpers
|
// MARK: - Noise Public Key Helpers
|
||||||
@@ -130,6 +136,20 @@ extension PeerID {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
extension PeerID {
|
||||||
|
var routingData: Data? {
|
||||||
|
if let direct = Data(hexString: id), direct.count == 8 { return direct }
|
||||||
|
if let bareData = Data(hexString: bare), bareData.count == 8 { return bareData }
|
||||||
|
let short = toShort()
|
||||||
|
return Data(hexString: short.id)
|
||||||
|
}
|
||||||
|
|
||||||
|
init?(routingData: Data) {
|
||||||
|
guard routingData.count == 8 else { return nil }
|
||||||
|
self.init(hexData: routingData)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// MARK: - Validation
|
// MARK: - Validation
|
||||||
|
|
||||||
extension PeerID {
|
extension PeerID {
|
||||||
@@ -191,9 +211,7 @@ extension PeerID: Comparable {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - String Interop Helpers
|
// MARK: - CustomStringConvertible
|
||||||
|
|
||||||
// MARK: CustomStringConvertible
|
|
||||||
|
|
||||||
extension PeerID: CustomStringConvertible {
|
extension PeerID: CustomStringConvertible {
|
||||||
/// So it returns the actual `id` like before even inside another String
|
/// So it returns the actual `id` like before even inside another String
|
||||||
@@ -201,17 +219,3 @@ extension PeerID: CustomStringConvertible {
|
|||||||
id
|
id
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: Custom Equatable w/ String & Optionality
|
|
||||||
|
|
||||||
// PeerID <> String
|
|
||||||
extension Optional where Wrapped == PeerID {
|
|
||||||
static func ==(lhs: Optional<Wrapped>, rhs: Optional<String>) -> Bool { lhs?.id == rhs }
|
|
||||||
static func !=(lhs: Optional<Wrapped>, rhs: Optional<String>) -> Bool { lhs?.id != rhs }
|
|
||||||
}
|
|
||||||
|
|
||||||
// String <> PeerID
|
|
||||||
extension Optional where Wrapped == String {
|
|
||||||
static func ==(lhs: Optional<Wrapped>, rhs: Optional<PeerID>) -> Bool { lhs == rhs?.id }
|
|
||||||
static func !=(lhs: Optional<Wrapped>, rhs: Optional<PeerID>) -> Bool { lhs != rhs?.id }
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -11,11 +11,11 @@ import Foundation
|
|||||||
struct ReadReceipt: Codable {
|
struct ReadReceipt: Codable {
|
||||||
let originalMessageID: String
|
let originalMessageID: String
|
||||||
let receiptID: String
|
let receiptID: String
|
||||||
var readerID: String // Who read it
|
var readerID: PeerID // Who read it
|
||||||
let readerNickname: String
|
let readerNickname: String
|
||||||
let timestamp: Date
|
let timestamp: Date
|
||||||
|
|
||||||
init(originalMessageID: String, readerID: String, readerNickname: String) {
|
init(originalMessageID: String, readerID: PeerID, readerNickname: String) {
|
||||||
self.originalMessageID = originalMessageID
|
self.originalMessageID = originalMessageID
|
||||||
self.receiptID = UUID().uuidString
|
self.receiptID = UUID().uuidString
|
||||||
self.readerID = readerID
|
self.readerID = readerID
|
||||||
@@ -24,7 +24,7 @@ struct ReadReceipt: Codable {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// For binary decoding
|
// For binary decoding
|
||||||
private init(originalMessageID: String, receiptID: String, readerID: String, readerNickname: String, timestamp: Date) {
|
private init(originalMessageID: String, receiptID: String, readerID: PeerID, readerNickname: String, timestamp: Date) {
|
||||||
self.originalMessageID = originalMessageID
|
self.originalMessageID = originalMessageID
|
||||||
self.receiptID = receiptID
|
self.receiptID = receiptID
|
||||||
self.readerID = readerID
|
self.readerID = readerID
|
||||||
@@ -48,7 +48,7 @@ struct ReadReceipt: Codable {
|
|||||||
data.appendUUID(receiptID)
|
data.appendUUID(receiptID)
|
||||||
// ReaderID as 8-byte hex string
|
// ReaderID as 8-byte hex string
|
||||||
var readerData = Data()
|
var readerData = Data()
|
||||||
var tempID = readerID
|
var tempID = readerID.id
|
||||||
while tempID.count >= 2 && readerData.count < 8 {
|
while tempID.count >= 2 && readerData.count < 8 {
|
||||||
let hexByte = String(tempID.prefix(2))
|
let hexByte = String(tempID.prefix(2))
|
||||||
if let byte = UInt8(hexByte, radix: 16) {
|
if let byte = UInt8(hexByte, radix: 16) {
|
||||||
@@ -78,8 +78,8 @@ struct ReadReceipt: Codable {
|
|||||||
let receiptID = dataCopy.readUUID(at: &offset) else { return nil }
|
let receiptID = dataCopy.readUUID(at: &offset) else { return nil }
|
||||||
|
|
||||||
guard let readerIDData = dataCopy.readFixedBytes(at: &offset, count: 8) else { return nil }
|
guard let readerIDData = dataCopy.readFixedBytes(at: &offset, count: 8) else { return nil }
|
||||||
let readerID = readerIDData.hexEncodedString()
|
let readerID = PeerID(hexData: readerIDData)
|
||||||
guard PeerID(str: readerID).isValid else { return nil }
|
guard readerID.isValid else { return nil }
|
||||||
|
|
||||||
guard let timestamp = dataCopy.readDate(at: &offset),
|
guard let timestamp = dataCopy.readDate(at: &offset),
|
||||||
InputValidator.validateTimestamp(timestamp),
|
InputValidator.validateTimestamp(timestamp),
|
||||||
|
|||||||
@@ -8,6 +8,14 @@ struct RequestSyncPacket {
|
|||||||
let p: Int
|
let p: Int
|
||||||
let m: UInt32
|
let m: UInt32
|
||||||
let data: Data
|
let data: Data
|
||||||
|
let types: SyncTypeFlags?
|
||||||
|
|
||||||
|
init(p: Int, m: UInt32, data: Data, types: SyncTypeFlags? = nil) {
|
||||||
|
self.p = p
|
||||||
|
self.m = m
|
||||||
|
self.data = data
|
||||||
|
self.types = types
|
||||||
|
}
|
||||||
|
|
||||||
func encode() -> Data {
|
func encode() -> Data {
|
||||||
var out = Data()
|
var out = Data()
|
||||||
@@ -25,6 +33,9 @@ struct RequestSyncPacket {
|
|||||||
putTLV(0x02, withUnsafeBytes(of: &mBE) { Data($0) })
|
putTLV(0x02, withUnsafeBytes(of: &mBE) { Data($0) })
|
||||||
// data
|
// data
|
||||||
putTLV(0x03, data)
|
putTLV(0x03, data)
|
||||||
|
if let typesData = types?.toData() {
|
||||||
|
putTLV(0x04, typesData)
|
||||||
|
}
|
||||||
return out
|
return out
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -33,6 +44,7 @@ struct RequestSyncPacket {
|
|||||||
var p: Int? = nil
|
var p: Int? = nil
|
||||||
var m: UInt32? = nil
|
var m: UInt32? = nil
|
||||||
var payload: Data? = nil
|
var payload: Data? = nil
|
||||||
|
var types: SyncTypeFlags? = nil
|
||||||
|
|
||||||
while off + 3 <= data.count {
|
while off + 3 <= data.count {
|
||||||
let t = Int(data[off]); off += 1
|
let t = Int(data[off]); off += 1
|
||||||
@@ -52,12 +64,16 @@ struct RequestSyncPacket {
|
|||||||
case 0x03:
|
case 0x03:
|
||||||
if v.count > maxAcceptBytes { return nil }
|
if v.count > maxAcceptBytes { return nil }
|
||||||
payload = v
|
payload = v
|
||||||
|
case 0x04:
|
||||||
|
if let decoded = SyncTypeFlags.decode(v) {
|
||||||
|
types = decoded
|
||||||
|
}
|
||||||
default:
|
default:
|
||||||
break // forward compatible; ignore unknown TLVs
|
break // forward compatible; ignore unknown TLVs
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
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, mm > 0 else { return nil }
|
||||||
return RequestSyncPacket(p: pp, m: mm, data: dd)
|
return RequestSyncPacket(p: pp, m: mm, data: dd, types: types)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -165,8 +165,12 @@ final class NoiseCipherState {
|
|||||||
// MARK: - Sliding Window Replay Protection
|
// MARK: - Sliding Window Replay Protection
|
||||||
|
|
||||||
/// Check if nonce is valid for replay protection
|
/// Check if nonce is valid for replay protection
|
||||||
|
/// BCH-01-010: Use safe arithmetic to prevent integer overflow
|
||||||
private func isValidNonce(_ receivedNonce: UInt64) -> Bool {
|
private func isValidNonce(_ receivedNonce: UInt64) -> Bool {
|
||||||
if receivedNonce + UInt64(Self.REPLAY_WINDOW_SIZE) <= highestReceivedNonce {
|
// Safe overflow check: instead of (receivedNonce + WINDOW_SIZE <= highest)
|
||||||
|
// use (highest >= WINDOW_SIZE && receivedNonce <= highest - WINDOW_SIZE)
|
||||||
|
let windowSize = UInt64(Self.REPLAY_WINDOW_SIZE)
|
||||||
|
if highestReceivedNonce >= windowSize && receivedNonce <= highestReceivedNonce - windowSize {
|
||||||
return false // Too old, outside window
|
return false // Too old, outside window
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -348,15 +352,19 @@ final class NoiseCipherState {
|
|||||||
do {
|
do {
|
||||||
let plaintext = try ChaChaPoly.open(sealedBox, using: key, authenticating: associatedData)
|
let plaintext = try ChaChaPoly.open(sealedBox, using: key, authenticating: associatedData)
|
||||||
|
|
||||||
|
// BCH-01-010: Atomic nonce state update
|
||||||
|
// Both replay window marking and nonce increment must complete together
|
||||||
|
// to prevent state desynchronization. We perform both after successful
|
||||||
|
// decryption only, ensuring state consistency on any failure path.
|
||||||
if useExtractedNonce {
|
if useExtractedNonce {
|
||||||
// Mark nonce as seen after successful decryption
|
|
||||||
markNonceAsSeen(decryptionNonce)
|
markNonceAsSeen(decryptionNonce)
|
||||||
}
|
}
|
||||||
nonce += 1
|
nonce += 1
|
||||||
|
|
||||||
return plaintext
|
return plaintext
|
||||||
} catch {
|
} catch {
|
||||||
|
// Decryption failed - nonce state remains unchanged (atomic rollback)
|
||||||
SecureLogger.debug("Decrypt failed: \(error) for nonce \(decryptionNonce)")
|
SecureLogger.debug("Decrypt failed: \(error) for nonce \(decryptionNonce)")
|
||||||
// Log authentication failures with nonce info
|
|
||||||
SecureLogger.error("Decryption failed at nonce \(decryptionNonce)", category: .encryption)
|
SecureLogger.error("Decryption failed at nonce \(decryptionNonce)", category: .encryption)
|
||||||
throw error
|
throw error
|
||||||
}
|
}
|
||||||
@@ -451,17 +459,40 @@ final class NoiseSymmetricState {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func split() -> (NoiseCipherState, NoiseCipherState) {
|
func split(useExtractedNonce: Bool) -> (NoiseCipherState, NoiseCipherState) {
|
||||||
let output = hkdf(chainingKey: chainingKey, inputKeyMaterial: Data(), numOutputs: 2)
|
let output = hkdf(chainingKey: chainingKey, inputKeyMaterial: Data(), numOutputs: 2)
|
||||||
let tempKey1 = SymmetricKey(data: output[0])
|
let tempKey1 = SymmetricKey(data: output[0])
|
||||||
let tempKey2 = SymmetricKey(data: output[1])
|
let tempKey2 = SymmetricKey(data: output[1])
|
||||||
|
|
||||||
let c1 = NoiseCipherState(key: tempKey1, useExtractedNonce: true)
|
let c1 = NoiseCipherState(key: tempKey1, useExtractedNonce: useExtractedNonce)
|
||||||
let c2 = NoiseCipherState(key: tempKey2, useExtractedNonce: true)
|
let c2 = NoiseCipherState(key: tempKey2, useExtractedNonce: useExtractedNonce)
|
||||||
|
|
||||||
|
// BCH-01-010: Clear symmetric state after split per Noise spec
|
||||||
|
// The chaining key and hash should not be retained after handshake completes
|
||||||
|
clearSensitiveData()
|
||||||
|
|
||||||
return (c1, c2)
|
return (c1, c2)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// BCH-01-010: Securely clear sensitive cryptographic state
|
||||||
|
/// Called after split() to clear chaining key and hash per Noise spec
|
||||||
|
func clearSensitiveData() {
|
||||||
|
// Clear chaining key by overwriting with zeros
|
||||||
|
let chainingKeyCount = chainingKey.count
|
||||||
|
chainingKey = Data(repeating: 0, count: chainingKeyCount)
|
||||||
|
|
||||||
|
// Clear hash by overwriting with zeros
|
||||||
|
let hashCount = hash.count
|
||||||
|
hash = Data(repeating: 0, count: hashCount)
|
||||||
|
|
||||||
|
// Clear the internal cipher state
|
||||||
|
cipherState.clearSensitiveData()
|
||||||
|
}
|
||||||
|
|
||||||
|
deinit {
|
||||||
|
clearSensitiveData()
|
||||||
|
}
|
||||||
|
|
||||||
// HKDF implementation
|
// HKDF implementation
|
||||||
private func hkdf(chainingKey: Data, inputKeyMaterial: Data, numOutputs: Int) -> [Data] {
|
private func hkdf(chainingKey: Data, inputKeyMaterial: Data, numOutputs: Int) -> [Data] {
|
||||||
let tempKey = HMAC<SHA256>.authenticationCode(for: inputKeyMaterial, using: SymmetricKey(data: chainingKey))
|
let tempKey = HMAC<SHA256>.authenticationCode(for: inputKeyMaterial, using: SymmetricKey(data: chainingKey))
|
||||||
@@ -507,16 +538,24 @@ final class NoiseHandshakeState {
|
|||||||
private var messagePatterns: [[NoiseMessagePattern]] = []
|
private var messagePatterns: [[NoiseMessagePattern]] = []
|
||||||
private var currentPattern = 0
|
private var currentPattern = 0
|
||||||
|
|
||||||
|
// Test support: predetermined ephemeral keys for test vectors
|
||||||
|
private var predeterminedEphemeralKey: Curve25519.KeyAgreement.PrivateKey?
|
||||||
|
private var prologueData: Data
|
||||||
|
|
||||||
init(
|
init(
|
||||||
role: NoiseRole,
|
role: NoiseRole,
|
||||||
pattern: NoisePattern,
|
pattern: NoisePattern,
|
||||||
keychain: KeychainManagerProtocol,
|
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(),
|
||||||
|
predeterminedEphemeralKey: Curve25519.KeyAgreement.PrivateKey? = nil
|
||||||
) {
|
) {
|
||||||
self.role = role
|
self.role = role
|
||||||
self.pattern = pattern
|
self.pattern = pattern
|
||||||
self.keychain = keychain
|
self.keychain = keychain
|
||||||
|
self.prologueData = prologue
|
||||||
|
self.predeterminedEphemeralKey = predeterminedEphemeralKey
|
||||||
|
|
||||||
// Initialize static keys
|
// Initialize static keys
|
||||||
if let localKey = localStaticKey {
|
if let localKey = localStaticKey {
|
||||||
@@ -537,8 +576,8 @@ final class NoiseHandshakeState {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private func mixPreMessageKeys() {
|
private func mixPreMessageKeys() {
|
||||||
// Mix prologue (empty for XX pattern normally)
|
// Mix prologue
|
||||||
symmetricState.mixHash(Data()) // Empty prologue for XX pattern
|
symmetricState.mixHash(self.prologueData)
|
||||||
// For XX pattern, no pre-message keys
|
// For XX pattern, no pre-message keys
|
||||||
// For IK/NK patterns, we'd mix the responder's static key here
|
// For IK/NK patterns, we'd mix the responder's static key here
|
||||||
switch pattern {
|
switch pattern {
|
||||||
@@ -563,8 +602,13 @@ final class NoiseHandshakeState {
|
|||||||
for pattern in patterns {
|
for pattern in patterns {
|
||||||
switch pattern {
|
switch pattern {
|
||||||
case .e:
|
case .e:
|
||||||
// Generate ephemeral key
|
// Generate ephemeral key (or use predetermined key for tests)
|
||||||
|
if let predetermined = predeterminedEphemeralKey {
|
||||||
|
localEphemeralPrivate = predetermined
|
||||||
|
predeterminedEphemeralKey = nil
|
||||||
|
} else {
|
||||||
localEphemeralPrivate = Curve25519.KeyAgreement.PrivateKey()
|
localEphemeralPrivate = Curve25519.KeyAgreement.PrivateKey()
|
||||||
|
}
|
||||||
localEphemeralPublic = localEphemeralPrivate!.publicKey
|
localEphemeralPublic = localEphemeralPrivate!.publicKey
|
||||||
messageBuffer.append(localEphemeralPublic!.rawRepresentation)
|
messageBuffer.append(localEphemeralPublic!.rawRepresentation)
|
||||||
symmetricState.mixHash(localEphemeralPublic!.rawRepresentation)
|
symmetricState.mixHash(localEphemeralPublic!.rawRepresentation)
|
||||||
@@ -597,14 +641,20 @@ final class NoiseHandshakeState {
|
|||||||
throw NoiseError.missingKeys
|
throw NoiseError.missingKeys
|
||||||
}
|
}
|
||||||
let shared = try localEphemeral.sharedSecretFromKeyAgreement(with: remoteStatic)
|
let shared = try localEphemeral.sharedSecretFromKeyAgreement(with: remoteStatic)
|
||||||
symmetricState.mixKey(shared.withUnsafeBytes { Data($0) })
|
var sharedData = shared.withUnsafeBytes { Data($0) }
|
||||||
|
symmetricState.mixKey(sharedData)
|
||||||
|
// Clear sensitive shared secret
|
||||||
|
keychain.secureClear(&sharedData)
|
||||||
} else {
|
} else {
|
||||||
guard let localStatic = localStaticPrivate,
|
guard let localStatic = localStaticPrivate,
|
||||||
let remoteEphemeral = remoteEphemeralPublic else {
|
let remoteEphemeral = remoteEphemeralPublic else {
|
||||||
throw NoiseError.missingKeys
|
throw NoiseError.missingKeys
|
||||||
}
|
}
|
||||||
let shared = try localStatic.sharedSecretFromKeyAgreement(with: remoteEphemeral)
|
let shared = try localStatic.sharedSecretFromKeyAgreement(with: remoteEphemeral)
|
||||||
symmetricState.mixKey(shared.withUnsafeBytes { Data($0) })
|
var sharedData = shared.withUnsafeBytes { Data($0) }
|
||||||
|
symmetricState.mixKey(sharedData)
|
||||||
|
// Clear sensitive shared secret
|
||||||
|
keychain.secureClear(&sharedData)
|
||||||
}
|
}
|
||||||
|
|
||||||
case .se:
|
case .se:
|
||||||
@@ -615,14 +665,20 @@ final class NoiseHandshakeState {
|
|||||||
throw NoiseError.missingKeys
|
throw NoiseError.missingKeys
|
||||||
}
|
}
|
||||||
let shared = try localStatic.sharedSecretFromKeyAgreement(with: remoteEphemeral)
|
let shared = try localStatic.sharedSecretFromKeyAgreement(with: remoteEphemeral)
|
||||||
symmetricState.mixKey(shared.withUnsafeBytes { Data($0) })
|
var sharedData = shared.withUnsafeBytes { Data($0) }
|
||||||
|
symmetricState.mixKey(sharedData)
|
||||||
|
// Clear sensitive shared secret
|
||||||
|
keychain.secureClear(&sharedData)
|
||||||
} else {
|
} else {
|
||||||
guard let localEphemeral = localEphemeralPrivate,
|
guard let localEphemeral = localEphemeralPrivate,
|
||||||
let remoteStatic = remoteStaticPublic else {
|
let remoteStatic = remoteStaticPublic else {
|
||||||
throw NoiseError.missingKeys
|
throw NoiseError.missingKeys
|
||||||
}
|
}
|
||||||
let shared = try localEphemeral.sharedSecretFromKeyAgreement(with: remoteStatic)
|
let shared = try localEphemeral.sharedSecretFromKeyAgreement(with: remoteStatic)
|
||||||
symmetricState.mixKey(shared.withUnsafeBytes { Data($0) })
|
var sharedData = shared.withUnsafeBytes { Data($0) }
|
||||||
|
symmetricState.mixKey(sharedData)
|
||||||
|
// Clear sensitive shared secret
|
||||||
|
keychain.secureClear(&sharedData)
|
||||||
}
|
}
|
||||||
|
|
||||||
case .ss:
|
case .ss:
|
||||||
@@ -711,7 +767,10 @@ final class NoiseHandshakeState {
|
|||||||
throw NoiseError.missingKeys
|
throw NoiseError.missingKeys
|
||||||
}
|
}
|
||||||
let shared = try localEphemeral.sharedSecretFromKeyAgreement(with: remoteEphemeral)
|
let shared = try localEphemeral.sharedSecretFromKeyAgreement(with: remoteEphemeral)
|
||||||
symmetricState.mixKey(shared.withUnsafeBytes { Data($0) })
|
var sharedData = shared.withUnsafeBytes { Data($0) }
|
||||||
|
symmetricState.mixKey(sharedData)
|
||||||
|
// Clear sensitive shared secret
|
||||||
|
keychain.secureClear(&sharedData)
|
||||||
|
|
||||||
case .es:
|
case .es:
|
||||||
if role == .initiator {
|
if role == .initiator {
|
||||||
@@ -765,9 +824,12 @@ final class NoiseHandshakeState {
|
|||||||
throw NoiseError.missingKeys
|
throw NoiseError.missingKeys
|
||||||
}
|
}
|
||||||
let shared = try localStatic.sharedSecretFromKeyAgreement(with: remoteStatic)
|
let shared = try localStatic.sharedSecretFromKeyAgreement(with: remoteStatic)
|
||||||
symmetricState.mixKey(shared.withUnsafeBytes { Data($0) })
|
var sharedData = shared.withUnsafeBytes { Data($0) }
|
||||||
|
symmetricState.mixKey(sharedData)
|
||||||
|
// Clear sensitive shared secret
|
||||||
|
keychain.secureClear(&sharedData)
|
||||||
|
|
||||||
default:
|
case .e, .s:
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -776,16 +838,20 @@ final class NoiseHandshakeState {
|
|||||||
return currentPattern >= messagePatterns.count
|
return currentPattern >= messagePatterns.count
|
||||||
}
|
}
|
||||||
|
|
||||||
func getTransportCiphers() throws -> (send: NoiseCipherState, receive: NoiseCipherState) {
|
func getTransportCiphers(useExtractedNonce: Bool) throws -> (send: NoiseCipherState, receive: NoiseCipherState, handshakeHash: Data) {
|
||||||
guard isHandshakeComplete() else {
|
guard isHandshakeComplete() else {
|
||||||
throw NoiseError.handshakeNotComplete
|
throw NoiseError.handshakeNotComplete
|
||||||
}
|
}
|
||||||
|
|
||||||
let (c1, c2) = symmetricState.split()
|
// BCH-01-010: Capture handshake hash BEFORE split() clears symmetric state
|
||||||
|
let finalHandshakeHash = symmetricState.getHandshakeHash()
|
||||||
|
|
||||||
|
let (c1, c2) = symmetricState.split(useExtractedNonce: useExtractedNonce)
|
||||||
|
|
||||||
// Initiator uses c1 for sending, c2 for receiving
|
// Initiator uses c1 for sending, c2 for receiving
|
||||||
// Responder uses c2 for sending, c1 for receiving
|
// Responder uses c2 for sending, c1 for receiving
|
||||||
return role == .initiator ? (c1, c2) : (c2, c1)
|
let ciphers = role == .initiator ? (c1, c2) : (c2, c1)
|
||||||
|
return (send: ciphers.0, receive: ciphers.1, handshakeHash: finalHandshakeHash)
|
||||||
}
|
}
|
||||||
|
|
||||||
func getRemoteStaticPublicKey() -> Curve25519.KeyAgreement.PublicKey? {
|
func getRemoteStaticPublicKey() -> Curve25519.KeyAgreement.PublicKey? {
|
||||||
@@ -846,19 +912,44 @@ enum NoiseError: Error {
|
|||||||
case nonceExceeded
|
case nonceExceeded
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// MARK: - Constant-Time Operations
|
||||||
|
|
||||||
|
/// BCH-01-010: Constant-time comparison to prevent timing side-channel attacks
|
||||||
|
/// This function compares two Data objects in constant time, preventing
|
||||||
|
/// information leakage via timing analysis.
|
||||||
|
private func constantTimeCompare(_ a: Data, _ b: Data) -> Bool {
|
||||||
|
guard a.count == b.count else { return false }
|
||||||
|
|
||||||
|
var result: UInt8 = 0
|
||||||
|
for i in 0..<a.count {
|
||||||
|
result |= a[a.startIndex.advanced(by: i)] ^ b[b.startIndex.advanced(by: i)]
|
||||||
|
}
|
||||||
|
return result == 0
|
||||||
|
}
|
||||||
|
|
||||||
|
/// BCH-01-010: Constant-time check if all bytes are zero
|
||||||
|
private func constantTimeIsZero(_ data: Data) -> Bool {
|
||||||
|
var result: UInt8 = 0
|
||||||
|
for byte in data {
|
||||||
|
result |= byte
|
||||||
|
}
|
||||||
|
return result == 0
|
||||||
|
}
|
||||||
|
|
||||||
// MARK: - Key Validation
|
// MARK: - Key Validation
|
||||||
|
|
||||||
extension NoiseHandshakeState {
|
extension NoiseHandshakeState {
|
||||||
/// Validate a Curve25519 public key
|
/// Validate a Curve25519 public key
|
||||||
/// Checks for weak/invalid keys that could compromise security
|
/// Checks for weak/invalid keys that could compromise security
|
||||||
|
/// BCH-01-010: Uses constant-time operations to prevent timing side-channels
|
||||||
static func validatePublicKey(_ keyData: Data) throws -> Curve25519.KeyAgreement.PublicKey {
|
static func validatePublicKey(_ keyData: Data) throws -> Curve25519.KeyAgreement.PublicKey {
|
||||||
// Check key length
|
// Check key length
|
||||||
guard keyData.count == 32 else {
|
guard keyData.count == 32 else {
|
||||||
throw NoiseError.invalidPublicKey
|
throw NoiseError.invalidPublicKey
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check for all-zero key (point at infinity)
|
// BCH-01-010: Constant-time check for all-zero key (point at infinity)
|
||||||
if keyData.allSatisfy({ $0 == 0 }) {
|
if constantTimeIsZero(keyData) {
|
||||||
throw NoiseError.invalidPublicKey
|
throw NoiseError.invalidPublicKey
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -883,8 +974,16 @@ extension NoiseHandshakeState {
|
|||||||
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]) // Another bad point
|
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]) // Another bad point
|
||||||
]
|
]
|
||||||
|
|
||||||
// Check against known bad points
|
// BCH-01-010: Constant-time check against known bad points
|
||||||
if lowOrderPoints.contains(keyData) {
|
// We check all points and accumulate matches to avoid early exit timing leaks
|
||||||
|
var foundBadPoint = false
|
||||||
|
for badPoint in lowOrderPoints {
|
||||||
|
if constantTimeCompare(keyData, badPoint) {
|
||||||
|
foundBadPoint = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if foundBadPoint {
|
||||||
SecureLogger.warning("Low-order point detected", category: .security)
|
SecureLogger.warning("Low-order point detected", category: .security)
|
||||||
throw NoiseError.invalidPublicKey
|
throw NoiseError.invalidPublicKey
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -102,8 +102,8 @@ class NoiseSession {
|
|||||||
|
|
||||||
// Check if handshake is complete
|
// Check if handshake is complete
|
||||||
if handshake.isHandshakeComplete() {
|
if handshake.isHandshakeComplete() {
|
||||||
// Get transport ciphers
|
// Get transport ciphers and handshake hash (hash captured before split clears state)
|
||||||
let (send, receive) = try handshake.getTransportCiphers()
|
let (send, receive, hash) = try handshake.getTransportCiphers(useExtractedNonce: true)
|
||||||
sendCipher = send
|
sendCipher = send
|
||||||
receiveCipher = receive
|
receiveCipher = receive
|
||||||
|
|
||||||
@@ -111,7 +111,7 @@ class NoiseSession {
|
|||||||
remoteStaticPublicKey = handshake.getRemoteStaticPublicKey()
|
remoteStaticPublicKey = handshake.getRemoteStaticPublicKey()
|
||||||
|
|
||||||
// Store handshake hash for channel binding
|
// Store handshake hash for channel binding
|
||||||
handshakeHash = handshake.getHandshakeHash()
|
handshakeHash = hash
|
||||||
|
|
||||||
state = .established
|
state = .established
|
||||||
handshakeState = nil // Clear handshake state
|
handshakeState = nil // Clear handshake state
|
||||||
@@ -128,8 +128,8 @@ class NoiseSession {
|
|||||||
|
|
||||||
// Check if handshake is complete after writing
|
// Check if handshake is complete after writing
|
||||||
if handshake.isHandshakeComplete() {
|
if handshake.isHandshakeComplete() {
|
||||||
// Get transport ciphers
|
// Get transport ciphers and handshake hash (hash captured before split clears state)
|
||||||
let (send, receive) = try handshake.getTransportCiphers()
|
let (send, receive, hash) = try handshake.getTransportCiphers(useExtractedNonce: true)
|
||||||
sendCipher = send
|
sendCipher = send
|
||||||
receiveCipher = receive
|
receiveCipher = receive
|
||||||
|
|
||||||
@@ -137,7 +137,7 @@ class NoiseSession {
|
|||||||
remoteStaticPublicKey = handshake.getRemoteStaticPublicKey()
|
remoteStaticPublicKey = handshake.getRemoteStaticPublicKey()
|
||||||
|
|
||||||
// Store handshake hash for channel binding
|
// Store handshake hash for channel binding
|
||||||
handshakeHash = handshake.getHandshakeHash()
|
handshakeHash = hash
|
||||||
|
|
||||||
state = .established
|
state = .established
|
||||||
handshakeState = nil // Clear handshake state
|
handshakeState = nil // Clear handshake state
|
||||||
|
|||||||
@@ -1,6 +1,11 @@
|
|||||||
import BitLogger
|
import BitLogger
|
||||||
import Foundation
|
import Foundation
|
||||||
import Tor
|
import Tor
|
||||||
|
#if os(iOS)
|
||||||
|
import UIKit
|
||||||
|
#elseif os(macOS)
|
||||||
|
import AppKit
|
||||||
|
#endif
|
||||||
|
|
||||||
/// Directory of online Nostr relays with approximate GPS locations, used for geohash routing.
|
/// Directory of online Nostr relays with approximate GPS locations, used for geohash routing.
|
||||||
@MainActor
|
@MainActor
|
||||||
@@ -12,19 +17,32 @@ final class GeoRelayDirectory {
|
|||||||
}
|
}
|
||||||
|
|
||||||
static let shared = GeoRelayDirectory()
|
static let shared = GeoRelayDirectory()
|
||||||
|
|
||||||
private(set) var entries: [Entry] = []
|
private(set) var entries: [Entry] = []
|
||||||
private let cacheFileName = "georelays_cache.csv"
|
private let cacheFileName = "georelays_cache.csv"
|
||||||
private let lastFetchKey = "georelay.lastFetchAt"
|
private let lastFetchKey = "georelay.lastFetchAt"
|
||||||
private let remoteURL = URL(string: "https://raw.githubusercontent.com/permissionlesstech/georelays/refs/heads/main/nostr_relays.csv")!
|
private let remoteURL = URL(string: "https://raw.githubusercontent.com/permissionlesstech/georelays/refs/heads/main/nostr_relays.csv")!
|
||||||
private let fetchInterval: TimeInterval = TransportConfig.geoRelayFetchIntervalSeconds // 24h
|
private let fetchInterval: TimeInterval = TransportConfig.geoRelayFetchIntervalSeconds
|
||||||
|
|
||||||
|
private var refreshTimer: Timer?
|
||||||
|
private var retryTask: Task<Void, Never>?
|
||||||
|
private var retryAttempt: Int = 0
|
||||||
|
private var isFetching: Bool = false
|
||||||
|
private var observers: [NSObjectProtocol] = []
|
||||||
|
|
||||||
private init() {
|
private init() {
|
||||||
// Load cached or bundled data synchronously
|
entries = loadLocalEntries()
|
||||||
self.entries = self.loadLocalEntries()
|
registerObservers()
|
||||||
// Fire-and-forget remote refresh if stale
|
startRefreshTimer()
|
||||||
prefetchIfNeeded()
|
prefetchIfNeeded()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
deinit {
|
||||||
|
observers.forEach { NotificationCenter.default.removeObserver($0) }
|
||||||
|
refreshTimer?.invalidate()
|
||||||
|
retryTask?.cancel()
|
||||||
|
}
|
||||||
|
|
||||||
/// Returns up to `count` relay URLs (wss://) closest to the geohash center.
|
/// Returns up to `count` relay URLs (wss://) closest to the geohash center.
|
||||||
func closestRelays(toGeohash geohash: String, count: Int = 5) -> [String] {
|
func closestRelays(toGeohash geohash: String, count: Int = 5) -> [String] {
|
||||||
let center = Geohash.decodeCenter(geohash)
|
let center = Geohash.decodeCenter(geohash)
|
||||||
@@ -62,40 +80,117 @@ final class GeoRelayDirectory {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Remote Fetch
|
// MARK: - Remote Fetch
|
||||||
func prefetchIfNeeded() {
|
func prefetchIfNeeded(force: Bool = false) {
|
||||||
|
guard !isFetching else { return }
|
||||||
|
|
||||||
let now = Date()
|
let now = Date()
|
||||||
let last = UserDefaults.standard.object(forKey: lastFetchKey) as? Date ?? .distantPast
|
let last = UserDefaults.standard.object(forKey: lastFetchKey) as? Date ?? .distantPast
|
||||||
|
|
||||||
|
if !force {
|
||||||
guard now.timeIntervalSince(last) >= fetchInterval else { return }
|
guard now.timeIntervalSince(last) >= fetchInterval else { return }
|
||||||
|
} else if last != .distantPast,
|
||||||
|
now.timeIntervalSince(last) < TransportConfig.geoRelayRetryInitialSeconds {
|
||||||
|
// Skip forced fetches if we just refreshed moments ago.
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cancelRetry()
|
||||||
fetchRemote()
|
fetchRemote()
|
||||||
}
|
}
|
||||||
|
|
||||||
private func fetchRemote() {
|
private func fetchRemote() {
|
||||||
let req = URLRequest(url: remoteURL, cachePolicy: .reloadIgnoringLocalCacheData, timeoutInterval: 15)
|
guard !isFetching else { return }
|
||||||
// Ensure Tor readiness before fetching (fail-closed by default)
|
isFetching = true
|
||||||
Task.detached {
|
|
||||||
|
let request = URLRequest(
|
||||||
|
url: remoteURL,
|
||||||
|
cachePolicy: .reloadIgnoringLocalCacheData,
|
||||||
|
timeoutInterval: 15
|
||||||
|
)
|
||||||
|
|
||||||
|
Task.detached { [weak self] in
|
||||||
|
guard let self else { return }
|
||||||
|
|
||||||
let ready = await TorManager.shared.awaitReady()
|
let ready = await TorManager.shared.awaitReady()
|
||||||
if !ready {
|
if !ready {
|
||||||
SecureLogger.warning("GeoRelayDirectory: Tor not ready; skipping remote fetch (fail-closed)", category: .session)
|
await self.handleFetchFailure(.torNotReady)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
let task = TorURLSession.shared.session.dataTask(with: req) { [weak self] data, _, error in
|
|
||||||
guard let self = self else { return }
|
do {
|
||||||
if let data = data, error == nil, let text = String(data: data, encoding: .utf8) {
|
let (data, _) = try await TorURLSession.shared.session.data(for: request)
|
||||||
|
guard let text = String(data: data, encoding: .utf8) else {
|
||||||
|
await self.handleFetchFailure(.invalidData)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
let parsed = GeoRelayDirectory.parseCSV(text)
|
let parsed = GeoRelayDirectory.parseCSV(text)
|
||||||
if !parsed.isEmpty {
|
guard !parsed.isEmpty else {
|
||||||
Task { @MainActor in
|
await self.handleFetchFailure(.invalidData)
|
||||||
self.entries = parsed
|
|
||||||
self.persistCache(text)
|
|
||||||
UserDefaults.standard.set(Date(), forKey: self.lastFetchKey)
|
|
||||||
SecureLogger.info("GeoRelayDirectory: refreshed \(parsed.count) relays from remote", category: .session)
|
|
||||||
}
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
await self.handleFetchSuccess(entries: parsed, csv: text)
|
||||||
|
} catch {
|
||||||
|
await self.handleFetchFailure(.network(error))
|
||||||
}
|
}
|
||||||
SecureLogger.warning("GeoRelayDirectory: remote fetch failed; keeping local entries", category: .session)
|
|
||||||
}
|
}
|
||||||
task.resume()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private enum FetchFailure {
|
||||||
|
case torNotReady
|
||||||
|
case invalidData
|
||||||
|
case network(Error)
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
private func handleFetchSuccess(entries parsed: [Entry], csv: String) {
|
||||||
|
entries = parsed
|
||||||
|
persistCache(csv)
|
||||||
|
UserDefaults.standard.set(Date(), forKey: lastFetchKey)
|
||||||
|
SecureLogger.info("GeoRelayDirectory: refreshed \(parsed.count) relays from remote", category: .session)
|
||||||
|
isFetching = false
|
||||||
|
retryAttempt = 0
|
||||||
|
cancelRetry()
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
private func handleFetchFailure(_ reason: FetchFailure) {
|
||||||
|
switch reason {
|
||||||
|
case .torNotReady:
|
||||||
|
SecureLogger.warning("GeoRelayDirectory: Tor not ready; scheduling retry", category: .session)
|
||||||
|
case .invalidData:
|
||||||
|
SecureLogger.warning("GeoRelayDirectory: remote fetch returned invalid data; scheduling retry", category: .session)
|
||||||
|
case .network(let error):
|
||||||
|
SecureLogger.warning("GeoRelayDirectory: remote fetch failed with error: \(error.localizedDescription)", category: .session)
|
||||||
|
}
|
||||||
|
isFetching = false
|
||||||
|
scheduleRetry()
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
private func scheduleRetry() {
|
||||||
|
retryAttempt = min(retryAttempt + 1, 10)
|
||||||
|
let base = TransportConfig.geoRelayRetryInitialSeconds
|
||||||
|
let maxDelay = TransportConfig.geoRelayRetryMaxSeconds
|
||||||
|
let multiplier = pow(2.0, Double(max(retryAttempt - 1, 0)))
|
||||||
|
let calculated = base * multiplier
|
||||||
|
let delay = min(maxDelay, max(base, calculated))
|
||||||
|
|
||||||
|
cancelRetry()
|
||||||
|
retryTask = Task { [weak self] in
|
||||||
|
let nanoseconds = UInt64(delay * 1_000_000_000)
|
||||||
|
try? await Task.sleep(nanoseconds: nanoseconds)
|
||||||
|
await MainActor.run {
|
||||||
|
self?.prefetchIfNeeded(force: true)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
private func cancelRetry() {
|
||||||
|
retryTask?.cancel()
|
||||||
|
retryTask = nil
|
||||||
}
|
}
|
||||||
|
|
||||||
private func persistCache(_ text: String) {
|
private func persistCache(_ text: String) {
|
||||||
@@ -110,30 +205,35 @@ final class GeoRelayDirectory {
|
|||||||
// MARK: - Loading
|
// MARK: - Loading
|
||||||
private func loadLocalEntries() -> [Entry] {
|
private func loadLocalEntries() -> [Entry] {
|
||||||
// Prefer cached file if present
|
// Prefer cached file if present
|
||||||
if let cache = self.cacheURL(),
|
if let cache = cacheURL(),
|
||||||
let data = try? Data(contentsOf: cache),
|
let data = try? Data(contentsOf: cache),
|
||||||
let text = String(data: data, encoding: .utf8) {
|
let text = String(data: data, encoding: .utf8) {
|
||||||
let arr = Self.parseCSV(text)
|
let arr = Self.parseCSV(text)
|
||||||
if !arr.isEmpty { return arr }
|
if !arr.isEmpty { return arr }
|
||||||
}
|
}
|
||||||
|
|
||||||
// Try bundled resource(s)
|
// Try bundled resource(s)
|
||||||
let bundleCandidates = [
|
let bundleCandidates = [
|
||||||
Bundle.main.url(forResource: "nostr_relays", withExtension: "csv"),
|
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"),
|
||||||
Bundle.main.url(forResource: "online_relays_gps", withExtension: "csv", subdirectory: "relays")
|
Bundle.main.url(forResource: "online_relays_gps", withExtension: "csv", subdirectory: "relays")
|
||||||
].compactMap { $0 }
|
].compactMap { $0 }
|
||||||
|
|
||||||
for url in bundleCandidates {
|
for url in bundleCandidates {
|
||||||
if let data = try? Data(contentsOf: url), let text = String(data: data, encoding: .utf8) {
|
if let data = try? Data(contentsOf: url),
|
||||||
|
let text = String(data: data, encoding: .utf8) {
|
||||||
let arr = Self.parseCSV(text)
|
let arr = Self.parseCSV(text)
|
||||||
if !arr.isEmpty { return arr }
|
if !arr.isEmpty { return arr }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Try filesystem path (development/test)
|
// Try filesystem path (development/test)
|
||||||
if let cwd = FileManager.default.currentDirectoryPath as String?,
|
if let cwd = FileManager.default.currentDirectoryPath as String?,
|
||||||
let data = try? Data(contentsOf: URL(fileURLWithPath: cwd).appendingPathComponent("relays/online_relays_gps.csv")),
|
let data = try? Data(contentsOf: URL(fileURLWithPath: cwd).appendingPathComponent("relays/online_relays_gps.csv")),
|
||||||
let text = String(data: data, encoding: .utf8) {
|
let text = String(data: data, encoding: .utf8) {
|
||||||
return Self.parseCSV(text)
|
return Self.parseCSV(text)
|
||||||
}
|
}
|
||||||
|
|
||||||
SecureLogger.warning("GeoRelayDirectory: no local CSV found; entries empty", category: .session)
|
SecureLogger.warning("GeoRelayDirectory: no local CSV found; entries empty", category: .session)
|
||||||
return []
|
return []
|
||||||
}
|
}
|
||||||
@@ -141,7 +241,6 @@ final class GeoRelayDirectory {
|
|||||||
nonisolated 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 })
|
||||||
// Skip header if present
|
|
||||||
for (idx, raw) in lines.enumerated() {
|
for (idx, raw) in lines.enumerated() {
|
||||||
let line = raw.trimmingCharacters(in: .whitespacesAndNewlines)
|
let line = raw.trimmingCharacters(in: .whitespacesAndNewlines)
|
||||||
if line.isEmpty { continue }
|
if line.isEmpty { continue }
|
||||||
@@ -162,11 +261,76 @@ final class GeoRelayDirectory {
|
|||||||
|
|
||||||
private func cacheURL() -> URL? {
|
private func cacheURL() -> URL? {
|
||||||
do {
|
do {
|
||||||
let base = try FileManager.default.url(for: .applicationSupportDirectory, in: .userDomainMask, appropriateFor: nil, create: true)
|
let base = try FileManager.default.url(
|
||||||
|
for: .applicationSupportDirectory,
|
||||||
|
in: .userDomainMask,
|
||||||
|
appropriateFor: nil,
|
||||||
|
create: true
|
||||||
|
)
|
||||||
let dir = base.appendingPathComponent("bitchat", isDirectory: true)
|
let dir = base.appendingPathComponent("bitchat", isDirectory: true)
|
||||||
try? FileManager.default.createDirectory(at: dir, withIntermediateDirectories: true)
|
try? FileManager.default.createDirectory(at: dir, withIntermediateDirectories: true)
|
||||||
return dir.appendingPathComponent(cacheFileName)
|
return dir.appendingPathComponent(cacheFileName)
|
||||||
} catch { return nil }
|
} catch {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Observers & Timers
|
||||||
|
private func registerObservers() {
|
||||||
|
let center = NotificationCenter.default
|
||||||
|
|
||||||
|
let torReady = center.addObserver(
|
||||||
|
forName: .TorDidBecomeReady,
|
||||||
|
object: nil,
|
||||||
|
queue: .main
|
||||||
|
) { [weak self] _ in
|
||||||
|
guard let self else { return }
|
||||||
|
Task { @MainActor in
|
||||||
|
self.prefetchIfNeeded(force: true)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
observers.append(torReady)
|
||||||
|
|
||||||
|
#if os(iOS)
|
||||||
|
let didBecomeActive = center.addObserver(
|
||||||
|
forName: UIApplication.didBecomeActiveNotification,
|
||||||
|
object: nil,
|
||||||
|
queue: .main
|
||||||
|
) { [weak self] _ in
|
||||||
|
guard let self else { return }
|
||||||
|
Task { @MainActor in
|
||||||
|
self.prefetchIfNeeded()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
observers.append(didBecomeActive)
|
||||||
|
#elseif os(macOS)
|
||||||
|
let didBecomeActive = center.addObserver(
|
||||||
|
forName: NSApplication.didBecomeActiveNotification,
|
||||||
|
object: nil,
|
||||||
|
queue: .main
|
||||||
|
) { [weak self] _ in
|
||||||
|
guard let self else { return }
|
||||||
|
Task { @MainActor in
|
||||||
|
self.prefetchIfNeeded()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
observers.append(didBecomeActive)
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
private func startRefreshTimer() {
|
||||||
|
refreshTimer?.invalidate()
|
||||||
|
let interval = TransportConfig.geoRelayRefreshCheckIntervalSeconds
|
||||||
|
guard interval > 0 else { return }
|
||||||
|
|
||||||
|
let timer = Timer.scheduledTimer(withTimeInterval: interval, repeats: true) { [weak self] _ in
|
||||||
|
guard let self else { return }
|
||||||
|
Task { @MainActor in
|
||||||
|
self.prefetchIfNeeded()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
refreshTimer = timer
|
||||||
|
RunLoop.main.add(timer, forMode: .common)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,50 +0,0 @@
|
|||||||
import Foundation
|
|
||||||
|
|
||||||
protocol KeychainHelperProtocol {
|
|
||||||
func save(key: String, data: Data, service: String, accessible: CFString?)
|
|
||||||
func load(key: String, service: String) -> Data?
|
|
||||||
func delete(key: String, service: String)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Keychain helper for secure storage
|
|
||||||
struct KeychainHelper: KeychainHelperProtocol {
|
|
||||||
func save(key: String, data: Data, service: String, accessible: CFString? = nil) {
|
|
||||||
var query: [String: Any] = [
|
|
||||||
kSecClass as String: kSecClassGenericPassword,
|
|
||||||
kSecAttrService as String: service,
|
|
||||||
kSecAttrAccount as String: key,
|
|
||||||
kSecValueData as String: data
|
|
||||||
]
|
|
||||||
if let accessible = accessible {
|
|
||||||
query[kSecAttrAccessible as String] = accessible
|
|
||||||
}
|
|
||||||
|
|
||||||
SecItemDelete(query as CFDictionary)
|
|
||||||
SecItemAdd(query as CFDictionary, nil)
|
|
||||||
}
|
|
||||||
|
|
||||||
func load(key: String, service: String) -> Data? {
|
|
||||||
let query: [String: Any] = [
|
|
||||||
kSecClass as String: kSecClassGenericPassword,
|
|
||||||
kSecAttrService as String: service,
|
|
||||||
kSecAttrAccount as String: key,
|
|
||||||
kSecReturnData as String: true
|
|
||||||
]
|
|
||||||
|
|
||||||
var result: AnyObject?
|
|
||||||
let status = SecItemCopyMatching(query as CFDictionary, &result)
|
|
||||||
|
|
||||||
guard status == errSecSuccess else { return nil }
|
|
||||||
return result as? Data
|
|
||||||
}
|
|
||||||
|
|
||||||
func delete(key: String, service: String) {
|
|
||||||
let query: [String: Any] = [
|
|
||||||
kSecClass as String: kSecClassGenericPassword,
|
|
||||||
kSecAttrService as String: service,
|
|
||||||
kSecAttrAccount as String: key
|
|
||||||
]
|
|
||||||
|
|
||||||
SecItemDelete(query as CFDictionary)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -4,7 +4,7 @@ import Foundation
|
|||||||
|
|
||||||
struct NostrEmbeddedBitChat {
|
struct NostrEmbeddedBitChat {
|
||||||
/// Build a `bitchat1:` base64url-encoded BitChat packet carrying a private message for Nostr DMs.
|
/// Build a `bitchat1:` base64url-encoded BitChat packet carrying a private message for Nostr DMs.
|
||||||
static func encodePMForNostr(content: String, messageID: String, recipientPeerID: String, senderPeerID: String) -> String? {
|
static func encodePMForNostr(content: String, messageID: String, recipientPeerID: PeerID, senderPeerID: PeerID) -> String? {
|
||||||
// TLV-encode the private message
|
// TLV-encode the private message
|
||||||
let pm = PrivateMessagePacket(messageID: messageID, content: content)
|
let pm = PrivateMessagePacket(messageID: messageID, content: content)
|
||||||
guard let tlv = pm.encode() else { return nil }
|
guard let tlv = pm.encode() else { return nil }
|
||||||
@@ -14,12 +14,12 @@ struct NostrEmbeddedBitChat {
|
|||||||
payload.append(tlv)
|
payload.append(tlv)
|
||||||
|
|
||||||
// Determine 8-byte recipient ID to embed
|
// Determine 8-byte recipient ID to embed
|
||||||
let recipientIDHex: String = normalizeRecipientPeerID(recipientPeerID)
|
let recipientID = normalizeRecipientPeerID(recipientPeerID)
|
||||||
|
|
||||||
let packet = BitchatPacket(
|
let packet = BitchatPacket(
|
||||||
type: MessageType.noiseEncrypted.rawValue,
|
type: MessageType.noiseEncrypted.rawValue,
|
||||||
senderID: Data(hexString: senderPeerID) ?? Data(),
|
senderID: Data(hexString: senderPeerID.id) ?? Data(),
|
||||||
recipientID: Data(hexString: recipientIDHex),
|
recipientID: Data(hexString: recipientID.id),
|
||||||
timestamp: UInt64(Date().timeIntervalSince1970 * 1000),
|
timestamp: UInt64(Date().timeIntervalSince1970 * 1000),
|
||||||
payload: payload,
|
payload: payload,
|
||||||
signature: nil,
|
signature: nil,
|
||||||
@@ -31,18 +31,18 @@ struct NostrEmbeddedBitChat {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Build a `bitchat1:` base64url-encoded BitChat packet carrying a delivery/read ack for Nostr DMs.
|
/// Build a `bitchat1:` base64url-encoded BitChat packet carrying a delivery/read ack for Nostr DMs.
|
||||||
static func encodeAckForNostr(type: NoisePayloadType, messageID: String, recipientPeerID: String, senderPeerID: String) -> String? {
|
static func encodeAckForNostr(type: NoisePayloadType, messageID: String, recipientPeerID: PeerID, senderPeerID: PeerID) -> String? {
|
||||||
guard type == .delivered || type == .readReceipt else { return nil }
|
guard type == .delivered || type == .readReceipt else { return nil }
|
||||||
|
|
||||||
var payload = Data([type.rawValue])
|
var payload = Data([type.rawValue])
|
||||||
payload.append(Data(messageID.utf8))
|
payload.append(Data(messageID.utf8))
|
||||||
|
|
||||||
let recipientIDHex: String = normalizeRecipientPeerID(recipientPeerID)
|
let recipientID = normalizeRecipientPeerID(recipientPeerID)
|
||||||
|
|
||||||
let packet = BitchatPacket(
|
let packet = BitchatPacket(
|
||||||
type: MessageType.noiseEncrypted.rawValue,
|
type: MessageType.noiseEncrypted.rawValue,
|
||||||
senderID: Data(hexString: senderPeerID) ?? Data(),
|
senderID: Data(hexString: senderPeerID.id) ?? Data(),
|
||||||
recipientID: Data(hexString: recipientIDHex),
|
recipientID: Data(hexString: recipientID.id),
|
||||||
timestamp: UInt64(Date().timeIntervalSince1970 * 1000),
|
timestamp: UInt64(Date().timeIntervalSince1970 * 1000),
|
||||||
payload: payload,
|
payload: payload,
|
||||||
signature: nil,
|
signature: nil,
|
||||||
@@ -54,7 +54,7 @@ struct NostrEmbeddedBitChat {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Build a `bitchat1:` ACK (delivered/read) without an embedded recipient peer ID (geohash DMs).
|
/// Build a `bitchat1:` ACK (delivered/read) without an embedded recipient peer ID (geohash DMs).
|
||||||
static func encodeAckForNostrNoRecipient(type: NoisePayloadType, messageID: String, senderPeerID: String) -> String? {
|
static func encodeAckForNostrNoRecipient(type: NoisePayloadType, messageID: String, senderPeerID: PeerID) -> String? {
|
||||||
guard type == .delivered || type == .readReceipt else { return nil }
|
guard type == .delivered || type == .readReceipt else { return nil }
|
||||||
|
|
||||||
var payload = Data([type.rawValue])
|
var payload = Data([type.rawValue])
|
||||||
@@ -62,7 +62,7 @@ struct NostrEmbeddedBitChat {
|
|||||||
|
|
||||||
let packet = BitchatPacket(
|
let packet = BitchatPacket(
|
||||||
type: MessageType.noiseEncrypted.rawValue,
|
type: MessageType.noiseEncrypted.rawValue,
|
||||||
senderID: Data(hexString: senderPeerID) ?? Data(),
|
senderID: Data(hexString: senderPeerID.id) ?? Data(),
|
||||||
recipientID: nil,
|
recipientID: nil,
|
||||||
timestamp: UInt64(Date().timeIntervalSince1970 * 1000),
|
timestamp: UInt64(Date().timeIntervalSince1970 * 1000),
|
||||||
payload: payload,
|
payload: payload,
|
||||||
@@ -75,7 +75,7 @@ struct NostrEmbeddedBitChat {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Build a `bitchat1:` payload without an embedded recipient peer ID (used for geohash DMs).
|
/// Build a `bitchat1:` payload without an embedded recipient peer ID (used for geohash DMs).
|
||||||
static func encodePMForNostrNoRecipient(content: String, messageID: String, senderPeerID: String) -> String? {
|
static func encodePMForNostrNoRecipient(content: String, messageID: String, senderPeerID: PeerID) -> String? {
|
||||||
let pm = PrivateMessagePacket(messageID: messageID, content: content)
|
let pm = PrivateMessagePacket(messageID: messageID, content: content)
|
||||||
guard let tlv = pm.encode() else { return nil }
|
guard let tlv = pm.encode() else { return nil }
|
||||||
|
|
||||||
@@ -84,7 +84,7 @@ struct NostrEmbeddedBitChat {
|
|||||||
|
|
||||||
let packet = BitchatPacket(
|
let packet = BitchatPacket(
|
||||||
type: MessageType.noiseEncrypted.rawValue,
|
type: MessageType.noiseEncrypted.rawValue,
|
||||||
senderID: Data(hexString: senderPeerID) ?? Data(),
|
senderID: Data(hexString: senderPeerID.id) ?? Data(),
|
||||||
recipientID: nil,
|
recipientID: nil,
|
||||||
timestamp: UInt64(Date().timeIntervalSince1970 * 1000),
|
timestamp: UInt64(Date().timeIntervalSince1970 * 1000),
|
||||||
payload: payload,
|
payload: payload,
|
||||||
@@ -96,11 +96,11 @@ struct NostrEmbeddedBitChat {
|
|||||||
return "bitchat1:" + base64URLEncode(data)
|
return "bitchat1:" + base64URLEncode(data)
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func normalizeRecipientPeerID(_ recipientPeerID: String) -> String {
|
private static func normalizeRecipientPeerID(_ recipientPeerID: PeerID) -> PeerID {
|
||||||
if let maybeData = Data(hexString: recipientPeerID) {
|
if let maybeData = Data(hexString: recipientPeerID.id) {
|
||||||
if maybeData.count == 32 {
|
if maybeData.count == 32 {
|
||||||
// Treat as Noise static public key; derive peerID from fingerprint
|
// Treat as Noise static public key; derive peerID from fingerprint
|
||||||
return PeerID(publicKey: maybeData).id
|
return PeerID(publicKey: maybeData)
|
||||||
} else if maybeData.count == 8 {
|
} else if maybeData.count == 8 {
|
||||||
// Already an 8-byte peer ID
|
// Already an 8-byte peer ID
|
||||||
return recipientPeerID
|
return recipientPeerID
|
||||||
|
|||||||
@@ -8,10 +8,13 @@ final class NostrIdentityBridge {
|
|||||||
private let deviceSeedKey = "nostr-device-seed"
|
private let deviceSeedKey = "nostr-device-seed"
|
||||||
// In-memory cache to avoid transient keychain access issues
|
// In-memory cache to avoid transient keychain access issues
|
||||||
private var deviceSeedCache: Data?
|
private var deviceSeedCache: Data?
|
||||||
|
// Cache derived identities to avoid repeated crypto during view rendering
|
||||||
|
private var derivedIdentityCache: [String: NostrIdentity] = [:]
|
||||||
|
private let cacheLock = NSLock()
|
||||||
|
|
||||||
private let keychain: KeychainHelperProtocol
|
private let keychain: KeychainManagerProtocol
|
||||||
|
|
||||||
init(keychain: KeychainHelperProtocol = KeychainHelper()) {
|
init(keychain: KeychainManagerProtocol = KeychainManager()) {
|
||||||
self.keychain = keychain
|
self.keychain = keychain
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -106,6 +109,14 @@ final class NostrIdentityBridge {
|
|||||||
/// Uses HMAC-SHA256(deviceSeed, geohash) as private key material, with fallback rehashing
|
/// Uses HMAC-SHA256(deviceSeed, geohash) as private key material, with fallback rehashing
|
||||||
/// if the candidate is not a valid secp256k1 private key.
|
/// if the candidate is not a valid secp256k1 private key.
|
||||||
func deriveIdentity(forGeohash geohash: String) throws -> NostrIdentity {
|
func deriveIdentity(forGeohash geohash: String) throws -> NostrIdentity {
|
||||||
|
// Check cache first to avoid repeated crypto + keychain I/O during view rendering
|
||||||
|
cacheLock.lock()
|
||||||
|
if let cached = derivedIdentityCache[geohash] {
|
||||||
|
cacheLock.unlock()
|
||||||
|
return cached
|
||||||
|
}
|
||||||
|
cacheLock.unlock()
|
||||||
|
|
||||||
let seed = getOrCreateDeviceSeed()
|
let seed = getOrCreateDeviceSeed()
|
||||||
guard let msg = geohash.data(using: .utf8) else {
|
guard let msg = geohash.data(using: .utf8) else {
|
||||||
throw NSError(domain: "NostrIdentity", code: -1, userInfo: [NSLocalizedDescriptionKey: "Invalid geohash string"])
|
throw NSError(domain: "NostrIdentity", code: -1, userInfo: [NSLocalizedDescriptionKey: "Invalid geohash string"])
|
||||||
@@ -125,11 +136,22 @@ final class NostrIdentityBridge {
|
|||||||
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()
|
||||||
|
derivedIdentityCache[geohash] = identity
|
||||||
|
cacheLock.unlock()
|
||||||
return identity
|
return identity
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// As a final fallback, hash the seed+msg and try again
|
// As a final fallback, hash the seed+msg and try again
|
||||||
let fallback = (seed + msg).sha256Hash()
|
let fallback = (seed + msg).sha256Hash()
|
||||||
return try NostrIdentity(privateKeyData: fallback)
|
let identity = try NostrIdentity(privateKeyData: fallback)
|
||||||
|
|
||||||
|
// Cache the result
|
||||||
|
cacheLock.lock()
|
||||||
|
derivedIdentityCache[geohash] = identity
|
||||||
|
cacheLock.unlock()
|
||||||
|
|
||||||
|
return identity
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -18,6 +18,7 @@ struct NostrProtocol {
|
|||||||
case seal = 13 // NIP-17 sealed event
|
case seal = 13 // NIP-17 sealed event
|
||||||
case giftWrap = 1059 // NIP-59 gift wrap
|
case giftWrap = 1059 // NIP-59 gift wrap
|
||||||
case ephemeralEvent = 20000
|
case ephemeralEvent = 20000
|
||||||
|
case geohashPresence = 20001
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Create a NIP-17 private message
|
/// Create a NIP-17 private message
|
||||||
@@ -125,6 +126,24 @@ struct NostrProtocol {
|
|||||||
return try event.sign(with: schnorrKey)
|
return try event.sign(with: schnorrKey)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Create a geohash presence heartbeat (kind 20001)
|
||||||
|
/// Must contain empty content and NO nickname tag
|
||||||
|
static func createGeohashPresenceEvent(
|
||||||
|
geohash: String,
|
||||||
|
senderIdentity: NostrIdentity
|
||||||
|
) throws -> NostrEvent {
|
||||||
|
let tags = [["g", geohash]]
|
||||||
|
let event = NostrEvent(
|
||||||
|
pubkey: senderIdentity.publicKeyHex,
|
||||||
|
createdAt: Date(),
|
||||||
|
kind: .geohashPresence,
|
||||||
|
tags: tags,
|
||||||
|
content: ""
|
||||||
|
)
|
||||||
|
let schnorrKey = try senderIdentity.schnorrSigningKey()
|
||||||
|
return try event.sign(with: schnorrKey)
|
||||||
|
}
|
||||||
|
|
||||||
/// 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.
|
||||||
static func createGeohashTextNote(
|
static func createGeohashTextNote(
|
||||||
content: String,
|
content: String,
|
||||||
|
|||||||
@@ -887,10 +887,10 @@ struct NostrFilter: Encodable {
|
|||||||
return filter
|
return filter
|
||||||
}
|
}
|
||||||
|
|
||||||
// For location channels: geohash-scoped ephemeral events (kind 20000)
|
// For location channels: geohash-scoped ephemeral events (kind 20000) and presence (kind 20001)
|
||||||
static func geohashEphemeral(_ geohash: String, since: Date? = nil, limit: Int = 200) -> NostrFilter {
|
static func geohashEphemeral(_ geohash: String, since: Date? = nil, limit: Int = 1000) -> NostrFilter {
|
||||||
var filter = NostrFilter()
|
var filter = NostrFilter()
|
||||||
filter.kinds = [20000]
|
filter.kinds = [20000, 20001]
|
||||||
filter.since = since?.timeIntervalSince1970.toInt()
|
filter.since = since?.timeIntervalSince1970.toInt()
|
||||||
filter.tagFilters = ["g": [geohash]]
|
filter.tagFilters = ["g": [geohash]]
|
||||||
filter.limit = limit
|
filter.limit = limit
|
||||||
@@ -906,6 +906,16 @@ struct NostrFilter: Encodable {
|
|||||||
filter.limit = limit
|
filter.limit = limit
|
||||||
return filter
|
return filter
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// For location notes with neighbors: subscribe to multiple geohashes (center + neighbors)
|
||||||
|
static func geohashNotes(_ geohashes: [String], since: Date? = nil, limit: Int = 200) -> NostrFilter {
|
||||||
|
var filter = NostrFilter()
|
||||||
|
filter.kinds = [1]
|
||||||
|
filter.since = since?.timeIntervalSince1970.toInt()
|
||||||
|
filter.tagFilters = ["g": geohashes]
|
||||||
|
filter.limit = limit
|
||||||
|
return filter
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Dynamic coding key for tag filters
|
// Dynamic coding key for tag filters
|
||||||
|
|||||||
@@ -5,16 +5,27 @@ import CryptoKit
|
|||||||
/// Implements HChaCha20 to derive a subkey and reduces the 24-byte nonce to a 12-byte nonce
|
/// Implements HChaCha20 to derive a subkey and reduces the 24-byte nonce to a 12-byte nonce
|
||||||
/// as per XChaCha20 construction.
|
/// as per XChaCha20 construction.
|
||||||
enum XChaCha20Poly1305Compat {
|
enum XChaCha20Poly1305Compat {
|
||||||
|
|
||||||
|
/// Errors that can occur during XChaCha20-Poly1305 operations
|
||||||
|
enum Error: Swift.Error {
|
||||||
|
case invalidKeyLength(expected: Int, got: Int)
|
||||||
|
case invalidNonceLength(expected: Int, got: Int)
|
||||||
|
}
|
||||||
|
|
||||||
struct SealBox {
|
struct SealBox {
|
||||||
let ciphertext: Data
|
let ciphertext: Data
|
||||||
let tag: Data
|
let tag: Data
|
||||||
}
|
}
|
||||||
|
|
||||||
static func seal(plaintext: Data, key: Data, nonce24: Data, aad: Data? = nil) throws -> SealBox {
|
static func seal(plaintext: Data, key: Data, nonce24: Data, aad: Data? = nil) throws -> SealBox {
|
||||||
precondition(key.count == 32, "XChaCha20 key must be 32 bytes")
|
guard key.count == 32 else {
|
||||||
precondition(nonce24.count == 24, "XChaCha20 nonce must be 24 bytes")
|
throw Error.invalidKeyLength(expected: 32, got: key.count)
|
||||||
|
}
|
||||||
|
guard nonce24.count == 24 else {
|
||||||
|
throw Error.invalidNonceLength(expected: 24, got: nonce24.count)
|
||||||
|
}
|
||||||
|
|
||||||
let subkey = hchacha20(key: key, nonce16: nonce24.prefix(16))
|
let subkey = try hchacha20(key: key, nonce16: Data(nonce24.prefix(16)))
|
||||||
let nonce12 = derive12ByteNonce(from24: nonce24)
|
let nonce12 = derive12ByteNonce(from24: nonce24)
|
||||||
let chachaKey = SymmetricKey(data: subkey)
|
let chachaKey = SymmetricKey(data: subkey)
|
||||||
let nonce = try ChaChaPoly.Nonce(data: nonce12)
|
let nonce = try ChaChaPoly.Nonce(data: nonce12)
|
||||||
@@ -23,10 +34,14 @@ enum XChaCha20Poly1305Compat {
|
|||||||
}
|
}
|
||||||
|
|
||||||
static func open(ciphertext: Data, tag: Data, key: Data, nonce24: Data, aad: Data? = nil) throws -> Data {
|
static func open(ciphertext: Data, tag: Data, key: Data, nonce24: Data, aad: Data? = nil) throws -> Data {
|
||||||
precondition(key.count == 32, "XChaCha20 key must be 32 bytes")
|
guard key.count == 32 else {
|
||||||
precondition(nonce24.count == 24, "XChaCha20 nonce must be 24 bytes")
|
throw Error.invalidKeyLength(expected: 32, got: key.count)
|
||||||
|
}
|
||||||
|
guard nonce24.count == 24 else {
|
||||||
|
throw Error.invalidNonceLength(expected: 24, got: nonce24.count)
|
||||||
|
}
|
||||||
|
|
||||||
let subkey = hchacha20(key: key, nonce16: nonce24.prefix(16))
|
let subkey = try hchacha20(key: key, nonce16: Data(nonce24.prefix(16)))
|
||||||
let nonce12 = derive12ByteNonce(from24: nonce24)
|
let nonce12 = derive12ByteNonce(from24: nonce24)
|
||||||
let chachaKey = SymmetricKey(data: subkey)
|
let chachaKey = SymmetricKey(data: subkey)
|
||||||
let box = try ChaChaPoly.SealedBox(nonce: ChaChaPoly.Nonce(data: nonce12), ciphertext: ciphertext, tag: tag)
|
let box = try ChaChaPoly.SealedBox(nonce: ChaChaPoly.Nonce(data: nonce12), ciphertext: ciphertext, tag: tag)
|
||||||
@@ -43,10 +58,14 @@ enum XChaCha20Poly1305Compat {
|
|||||||
return out
|
return out
|
||||||
}
|
}
|
||||||
|
|
||||||
private static func hchacha20(key: Data, nonce16: Data) -> Data {
|
private static func hchacha20(key: Data, nonce16: Data) throws -> Data {
|
||||||
// HChaCha20 based on the original ChaCha20 core with a 16-byte nonce.
|
// HChaCha20 based on the original ChaCha20 core with a 16-byte nonce.
|
||||||
precondition(key.count == 32)
|
guard key.count == 32 else {
|
||||||
precondition(nonce16.count == 16)
|
throw Error.invalidKeyLength(expected: 32, got: key.count)
|
||||||
|
}
|
||||||
|
guard nonce16.count == 16 else {
|
||||||
|
throw Error.invalidNonceLength(expected: 16, got: nonce16.count)
|
||||||
|
}
|
||||||
|
|
||||||
// Constants "expand 32-byte k"
|
// Constants "expand 32-byte k"
|
||||||
var state: [UInt32] = [
|
var state: [UInt32] = [
|
||||||
|
|||||||
@@ -6,6 +6,7 @@
|
|||||||
//
|
//
|
||||||
|
|
||||||
import Foundation
|
import Foundation
|
||||||
|
import CryptoKit
|
||||||
|
|
||||||
// MARK: - Hex Encoding/Decoding
|
// MARK: - Hex Encoding/Decoding
|
||||||
|
|
||||||
@@ -17,14 +18,41 @@ extension Data {
|
|||||||
return self.map { String(format: "%02x", $0) }.joined()
|
return self.map { String(format: "%02x", $0) }.joined()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func sha256Hex() -> String {
|
||||||
|
let digest = SHA256.hash(data: self)
|
||||||
|
return digest.map { String(format: "%02x", $0) }.joined()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Initialize Data from a hex string.
|
||||||
|
/// - Parameter hexString: A hex string, optionally prefixed with "0x" or "0X".
|
||||||
|
/// Whitespace is trimmed. Must have even length after prefix removal.
|
||||||
|
/// - Returns: nil if the string has odd length or contains invalid hex characters.
|
||||||
init?(hexString: String) {
|
init?(hexString: String) {
|
||||||
let len = hexString.count / 2
|
var hex = hexString.trimmingCharacters(in: .whitespaces)
|
||||||
|
|
||||||
|
// Remove optional 0x prefix
|
||||||
|
if hex.hasPrefix("0x") || hex.hasPrefix("0X") {
|
||||||
|
hex = String(hex.dropFirst(2))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reject odd-length strings
|
||||||
|
guard hex.count % 2 == 0 else {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reject empty strings
|
||||||
|
guard !hex.isEmpty else {
|
||||||
|
self = Data()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let len = hex.count / 2
|
||||||
var data = Data(capacity: len)
|
var data = Data(capacity: len)
|
||||||
var index = hexString.startIndex
|
var index = hex.startIndex
|
||||||
|
|
||||||
for _ in 0..<len {
|
for _ in 0..<len {
|
||||||
let nextIndex = hexString.index(index, offsetBy: 2)
|
let nextIndex = hex.index(index, offsetBy: 2)
|
||||||
guard let byte = UInt8(String(hexString[index..<nextIndex]), radix: 16) else {
|
guard let byte = UInt8(String(hex[index..<nextIndex]), radix: 16) else {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
data.append(byte)
|
data.append(byte)
|
||||||
|
|||||||
@@ -22,11 +22,11 @@
|
|||||||
///
|
///
|
||||||
/// ## Wire Format
|
/// ## Wire Format
|
||||||
/// ```
|
/// ```
|
||||||
/// Header (Fixed 13 bytes):
|
/// Header (Fixed 14 bytes for v1, 16 bytes for v2):
|
||||||
/// +--------+------+-----+-----------+-------+----------------+
|
/// +--------+------+-----+-----------+-------+------------------+
|
||||||
/// |Version | Type | TTL | Timestamp | Flags | PayloadLength |
|
/// |Version | Type | TTL | Timestamp | Flags | PayloadLength |
|
||||||
/// |1 byte |1 byte|1byte| 8 bytes | 1 byte| 2 bytes |
|
/// |1 byte |1 byte|1byte| 8 bytes | 1 byte| 2 or 4 bytes |
|
||||||
/// +--------+------+-----+-----------+-------+----------------+
|
/// +--------+------+-----+-----------+-------+------------------+
|
||||||
///
|
///
|
||||||
/// Variable sections:
|
/// Variable sections:
|
||||||
/// +----------+-------------+---------+------------+
|
/// +----------+-------------+---------+------------+
|
||||||
@@ -52,7 +52,7 @@
|
|||||||
/// ## Flag Bits
|
/// ## Flag Bits
|
||||||
/// - Bit 0: Has recipient ID (directed message)
|
/// - Bit 0: Has recipient ID (directed message)
|
||||||
/// - Bit 1: Has signature (authenticated message)
|
/// - Bit 1: Has signature (authenticated message)
|
||||||
/// - Bit 2: Is compressed (LZ4 compression applied)
|
/// - Bit 2: Is compressed (zlib compression applied)
|
||||||
/// - Bits 3-7: Reserved for future use
|
/// - Bits 3-7: Reserved for future use
|
||||||
///
|
///
|
||||||
/// ## Size Constraints
|
/// ## Size Constraints
|
||||||
@@ -89,6 +89,7 @@
|
|||||||
///
|
///
|
||||||
|
|
||||||
import Foundation
|
import Foundation
|
||||||
|
import BitLogger
|
||||||
|
|
||||||
extension Data {
|
extension Data {
|
||||||
func trimmingNullBytes() -> Data {
|
func trimmingNullBytes() -> Data {
|
||||||
@@ -105,83 +106,123 @@ extension Data {
|
|||||||
/// their binary wire format representation.
|
/// their binary wire format representation.
|
||||||
/// - Note: All multi-byte values use network byte order (big-endian)
|
/// - Note: All multi-byte values use network byte order (big-endian)
|
||||||
struct BinaryProtocol {
|
struct BinaryProtocol {
|
||||||
static let headerSize = 13
|
static let v1HeaderSize = 14
|
||||||
|
static let v2HeaderSize = 16
|
||||||
static let senderIDSize = 8
|
static let senderIDSize = 8
|
||||||
static let recipientIDSize = 8
|
static let recipientIDSize = 8
|
||||||
static let signatureSize = 64
|
static let signatureSize = 64
|
||||||
|
|
||||||
|
// Field offsets within packet header
|
||||||
|
struct Offsets {
|
||||||
|
static let version = 0
|
||||||
|
static let type = 1
|
||||||
|
static let ttl = 2
|
||||||
|
static let timestamp = 3
|
||||||
|
static let flags = 11 // After version(1) + type(1) + ttl(1) + timestamp(8)
|
||||||
|
}
|
||||||
|
|
||||||
|
static func headerSize(for version: UInt8) -> Int? {
|
||||||
|
switch version {
|
||||||
|
case 1: return v1HeaderSize
|
||||||
|
case 2: return v2HeaderSize
|
||||||
|
default: return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func lengthFieldSize(for version: UInt8) -> Int {
|
||||||
|
return version == 2 ? 4 : 2
|
||||||
|
}
|
||||||
|
|
||||||
struct Flags {
|
struct Flags {
|
||||||
static let hasRecipient: UInt8 = 0x01
|
static let hasRecipient: UInt8 = 0x01
|
||||||
static let hasSignature: UInt8 = 0x02
|
static let hasSignature: UInt8 = 0x02
|
||||||
static let isCompressed: UInt8 = 0x04
|
static let isCompressed: UInt8 = 0x04
|
||||||
|
static let hasRoute: UInt8 = 0x08
|
||||||
}
|
}
|
||||||
|
|
||||||
// Encode BitchatPacket to binary format
|
// Encode BitchatPacket to binary format
|
||||||
static func encode(_ packet: BitchatPacket, padding: Bool = true) -> Data? {
|
static func encode(_ packet: BitchatPacket, padding: Bool = true) -> Data? {
|
||||||
var data = Data()
|
let version = packet.version
|
||||||
|
guard version == 1 || version == 2 else { return nil }
|
||||||
|
|
||||||
|
// Try to compress payload when beneficial, keeping original size for later decoding
|
||||||
// Try to compress payload if beneficial
|
|
||||||
var payload = packet.payload
|
var payload = packet.payload
|
||||||
var originalPayloadSize: UInt16? = nil
|
|
||||||
var isCompressed = false
|
var isCompressed = false
|
||||||
|
var originalPayloadSize: Int?
|
||||||
if CompressionUtil.shouldCompress(payload) {
|
if CompressionUtil.shouldCompress(payload) {
|
||||||
if let compressedPayload = CompressionUtil.compress(payload) {
|
// Only compress when we can represent the original length in the outbound frame
|
||||||
// Store original size for decompression (2 bytes after payload)
|
let maxRepresentable = version == 2 ? Int(UInt32.max) : Int(UInt16.max)
|
||||||
originalPayloadSize = UInt16(payload.count)
|
if payload.count <= maxRepresentable,
|
||||||
|
let compressedPayload = CompressionUtil.compress(payload) {
|
||||||
|
originalPayloadSize = payload.count
|
||||||
payload = compressedPayload
|
payload = compressedPayload
|
||||||
isCompressed = true
|
isCompressed = true
|
||||||
|
|
||||||
} else {
|
|
||||||
}
|
}
|
||||||
} else {
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Header
|
let lengthFieldBytes = lengthFieldSize(for: version)
|
||||||
// Reserve capacity to reduce reallocations. Estimate base size conservatively.
|
|
||||||
// header(13) + sender(8) + opt recipient(8) + opt originalSize(2) + payload + opt signature(64) + up to 255 pad
|
// Route is only supported for v2+ packets (per SOURCE_ROUTING.md spec)
|
||||||
let estimatedPayload = payload.count + (isCompressed ? 2 : 0)
|
let originalRoute = (version >= 2) ? (packet.route ?? []) : []
|
||||||
let estimated = headerSize + senderIDSize + (packet.recipientID == nil ? 0 : recipientIDSize) + estimatedPayload + (packet.signature == nil ? 0 : signatureSize) + 255
|
if originalRoute.contains(where: { $0.isEmpty }) { return nil }
|
||||||
data.reserveCapacity(estimated)
|
let sanitizedRoute: [Data] = originalRoute.map { hop in
|
||||||
data.append(packet.version)
|
if hop.count == senderIDSize { return hop }
|
||||||
|
if hop.count > senderIDSize { return Data(hop.prefix(senderIDSize)) }
|
||||||
|
var padded = hop
|
||||||
|
padded.append(Data(repeating: 0, count: senderIDSize - hop.count))
|
||||||
|
return padded
|
||||||
|
}
|
||||||
|
guard sanitizedRoute.count <= 255 else { return nil }
|
||||||
|
|
||||||
|
let hasRoute = !sanitizedRoute.isEmpty
|
||||||
|
let routeLength = hasRoute ? 1 + sanitizedRoute.count * senderIDSize : 0
|
||||||
|
let originalSizeFieldBytes = isCompressed ? lengthFieldBytes : 0
|
||||||
|
// payloadLength in header is payload-only (does NOT include route bytes)
|
||||||
|
let payloadDataSize = payload.count + originalSizeFieldBytes
|
||||||
|
|
||||||
|
if version == 1 && payloadDataSize > Int(UInt16.max) { return nil }
|
||||||
|
if version == 2 && payloadDataSize > Int(UInt32.max) { return nil }
|
||||||
|
|
||||||
|
guard let headerSize = headerSize(for: version) else { return nil }
|
||||||
|
let estimatedHeader = headerSize + senderIDSize + (packet.recipientID == nil ? 0 : recipientIDSize) + routeLength
|
||||||
|
let estimatedPayload = payloadDataSize
|
||||||
|
let estimatedSignature = (packet.signature == nil ? 0 : signatureSize)
|
||||||
|
var data = Data()
|
||||||
|
data.reserveCapacity(estimatedHeader + estimatedPayload + estimatedSignature + 255)
|
||||||
|
|
||||||
|
data.append(version)
|
||||||
data.append(packet.type)
|
data.append(packet.type)
|
||||||
data.append(packet.ttl)
|
data.append(packet.ttl)
|
||||||
|
|
||||||
// Timestamp (8 bytes, big-endian)
|
for shift in stride(from: 56, through: 0, by: -8) {
|
||||||
for i in (0..<8).reversed() {
|
data.append(UInt8((packet.timestamp >> UInt64(shift)) & 0xFF))
|
||||||
data.append(UInt8((packet.timestamp >> (i * 8)) & 0xFF))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Flags
|
|
||||||
var flags: UInt8 = 0
|
var flags: UInt8 = 0
|
||||||
if packet.recipientID != nil {
|
if packet.recipientID != nil { flags |= Flags.hasRecipient }
|
||||||
flags |= Flags.hasRecipient
|
if packet.signature != nil { flags |= Flags.hasSignature }
|
||||||
}
|
if isCompressed { flags |= Flags.isCompressed }
|
||||||
if packet.signature != nil {
|
// HAS_ROUTE is only valid for v2+ packets
|
||||||
flags |= Flags.hasSignature
|
if hasRoute && version >= 2 { flags |= Flags.hasRoute }
|
||||||
}
|
|
||||||
if isCompressed {
|
|
||||||
flags |= Flags.isCompressed
|
|
||||||
}
|
|
||||||
data.append(flags)
|
data.append(flags)
|
||||||
|
|
||||||
// Payload length (2 bytes, big-endian) - includes original size if compressed
|
if version == 2 {
|
||||||
let payloadDataSize = payload.count + (isCompressed ? 2 : 0)
|
let length = UInt32(payloadDataSize)
|
||||||
let payloadLength = UInt16(payloadDataSize)
|
for shift in stride(from: 24, through: 0, by: -8) {
|
||||||
|
data.append(UInt8((length >> UInt32(shift)) & 0xFF))
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
let length = UInt16(payloadDataSize)
|
||||||
|
data.append(UInt8((length >> 8) & 0xFF))
|
||||||
|
data.append(UInt8(length & 0xFF))
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
data.append(UInt8((payloadLength >> 8) & 0xFF))
|
|
||||||
data.append(UInt8(payloadLength & 0xFF))
|
|
||||||
|
|
||||||
// SenderID (exactly 8 bytes)
|
|
||||||
let senderBytes = packet.senderID.prefix(senderIDSize)
|
let senderBytes = packet.senderID.prefix(senderIDSize)
|
||||||
data.append(senderBytes)
|
data.append(senderBytes)
|
||||||
if senderBytes.count < senderIDSize {
|
if senderBytes.count < senderIDSize {
|
||||||
data.append(Data(repeating: 0, count: senderIDSize - senderBytes.count))
|
data.append(Data(repeating: 0, count: senderIDSize - senderBytes.count))
|
||||||
}
|
}
|
||||||
|
|
||||||
// RecipientID (if present)
|
|
||||||
if let recipientID = packet.recipientID {
|
if let recipientID = packet.recipientID {
|
||||||
let recipientBytes = recipientID.prefix(recipientIDSize)
|
let recipientBytes = recipientID.prefix(recipientIDSize)
|
||||||
data.append(recipientBytes)
|
data.append(recipientBytes)
|
||||||
@@ -190,29 +231,36 @@ struct BinaryProtocol {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Payload (with original size prepended if compressed)
|
if hasRoute {
|
||||||
|
data.append(UInt8(sanitizedRoute.count))
|
||||||
|
for hop in sanitizedRoute {
|
||||||
|
data.append(hop)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if isCompressed, let originalSize = originalPayloadSize {
|
if isCompressed, let originalSize = originalPayloadSize {
|
||||||
// Prepend original size (2 bytes, big-endian)
|
if version == 2 {
|
||||||
data.append(UInt8((originalSize >> 8) & 0xFF))
|
let value = UInt32(originalSize)
|
||||||
data.append(UInt8(originalSize & 0xFF))
|
for shift in stride(from: 24, through: 0, by: -8) {
|
||||||
|
data.append(UInt8((value >> UInt32(shift)) & 0xFF))
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
let value = UInt16(originalSize)
|
||||||
|
data.append(UInt8((value >> 8) & 0xFF))
|
||||||
|
data.append(UInt8(value & 0xFF))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
data.append(payload)
|
data.append(payload)
|
||||||
|
|
||||||
// Signature (if present)
|
|
||||||
if let signature = packet.signature {
|
if let signature = packet.signature {
|
||||||
data.append(signature.prefix(signatureSize))
|
data.append(signature.prefix(signatureSize))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// Apply padding to standard block sizes for traffic analysis resistance
|
|
||||||
if padding {
|
if padding {
|
||||||
let optimalSize = MessagePadding.optimalBlockSize(for: data.count)
|
let optimalSize = MessagePadding.optimalBlockSize(for: data.count)
|
||||||
let paddedData = MessagePadding.pad(data, toSize: optimalSize)
|
return MessagePadding.pad(data, toSize: optimalSize)
|
||||||
return paddedData
|
|
||||||
} else {
|
|
||||||
// Caller explicitly requested no padding (e.g., BLE write path)
|
|
||||||
return data
|
|
||||||
}
|
}
|
||||||
|
return data
|
||||||
}
|
}
|
||||||
|
|
||||||
// Decode binary data to BitchatPacket
|
// Decode binary data to BitchatPacket
|
||||||
@@ -227,87 +275,124 @@ struct BinaryProtocol {
|
|||||||
|
|
||||||
// Core decoding implementation used by decode(_:) with and without padding removal
|
// Core decoding implementation used by decode(_:) with and without padding removal
|
||||||
private static func decodeCore(_ raw: Data) -> BitchatPacket? {
|
private static func decodeCore(_ raw: Data) -> BitchatPacket? {
|
||||||
// Minimum size: header + senderID
|
guard raw.count >= v1HeaderSize + senderIDSize else { return nil }
|
||||||
guard raw.count >= headerSize + senderIDSize else { return nil }
|
|
||||||
|
|
||||||
return raw.withUnsafeBytes { (buf: UnsafeRawBufferPointer) -> BitchatPacket? in
|
return raw.withUnsafeBytes { (buf: UnsafeRawBufferPointer) -> BitchatPacket? in
|
||||||
guard let base = buf.baseAddress else { return nil }
|
guard let base = buf.baseAddress else { return nil }
|
||||||
var offset = 0
|
var offset = 0
|
||||||
func require(_ n: Int) -> Bool { offset + n <= buf.count }
|
func require(_ n: Int) -> Bool { offset + n <= buf.count }
|
||||||
// Read single byte
|
|
||||||
func read8() -> UInt8? {
|
func read8() -> UInt8? {
|
||||||
guard require(1) else { return nil }
|
guard require(1) else { return nil }
|
||||||
let v = base.advanced(by: offset).assumingMemoryBound(to: UInt8.self).pointee
|
let value = base.advanced(by: offset).assumingMemoryBound(to: UInt8.self).pointee
|
||||||
offset += 1
|
offset += 1
|
||||||
return v
|
return value
|
||||||
}
|
}
|
||||||
// Read big-endian 16-bit
|
|
||||||
func read16() -> UInt16? {
|
func read16() -> UInt16? {
|
||||||
guard require(2) else { return nil }
|
guard require(2) else { return nil }
|
||||||
let p = base.advanced(by: offset).assumingMemoryBound(to: UInt8.self)
|
let ptr = base.advanced(by: offset).assumingMemoryBound(to: UInt8.self)
|
||||||
let v = (UInt16(p[0]) << 8) | UInt16(p[1])
|
let value = (UInt16(ptr[0]) << 8) | UInt16(ptr[1])
|
||||||
offset += 2
|
offset += 2
|
||||||
return v
|
return value
|
||||||
|
}
|
||||||
|
func read32() -> UInt32? {
|
||||||
|
guard require(4) else { return nil }
|
||||||
|
let ptr = base.advanced(by: offset).assumingMemoryBound(to: UInt8.self)
|
||||||
|
let value = (UInt32(ptr[0]) << 24) | (UInt32(ptr[1]) << 16) | (UInt32(ptr[2]) << 8) | UInt32(ptr[3])
|
||||||
|
offset += 4
|
||||||
|
return value
|
||||||
}
|
}
|
||||||
// Copy N bytes into Data
|
|
||||||
func readData(_ n: Int) -> Data? {
|
func readData(_ n: Int) -> Data? {
|
||||||
guard require(n) else { return nil }
|
guard require(n) else { return nil }
|
||||||
let ptr = base.advanced(by: offset)
|
let ptr = base.advanced(by: offset)
|
||||||
let d = Data(bytes: ptr, count: n)
|
let data = Data(bytes: ptr, count: n)
|
||||||
offset += n
|
offset += n
|
||||||
return d
|
return data
|
||||||
}
|
}
|
||||||
|
|
||||||
// Version
|
guard let version = read8(), version == 1 || version == 2 else { return nil }
|
||||||
guard let version = read8(), version == 1 else { return nil }
|
let lengthFieldBytes = lengthFieldSize(for: version)
|
||||||
guard let type = read8() else { return nil }
|
guard let headerSize = headerSize(for: version) else { return nil }
|
||||||
guard let ttl = read8() else { return nil }
|
let minimumRequired = headerSize + senderIDSize
|
||||||
|
guard raw.count >= minimumRequired else { return nil }
|
||||||
|
|
||||||
// Timestamp 8 bytes BE
|
guard let type = read8(), let ttl = read8() else { return nil }
|
||||||
guard require(8) else { return nil }
|
|
||||||
var ts: UInt64 = 0
|
var timestamp: UInt64 = 0
|
||||||
for _ in 0..<8 {
|
for _ in 0..<8 {
|
||||||
guard let b = read8() else { return nil }
|
guard let byte = read8() else { return nil }
|
||||||
ts = (ts << 8) | UInt64(b)
|
timestamp = (timestamp << 8) | UInt64(byte)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Flags
|
|
||||||
guard let flags = read8() else { return nil }
|
guard let flags = read8() else { return nil }
|
||||||
let hasRecipient = (flags & Flags.hasRecipient) != 0
|
let hasRecipient = (flags & Flags.hasRecipient) != 0
|
||||||
let hasSignature = (flags & Flags.hasSignature) != 0
|
let hasSignature = (flags & Flags.hasSignature) != 0
|
||||||
let isCompressed = (flags & Flags.isCompressed) != 0
|
let isCompressed = (flags & Flags.isCompressed) != 0
|
||||||
|
// HAS_ROUTE is only valid for v2+ packets; ignore the flag for v1
|
||||||
|
let hasRoute = (version >= 2) && (flags & Flags.hasRoute) != 0
|
||||||
|
|
||||||
// Payload length
|
let payloadLength: Int
|
||||||
guard let payloadLen = read16(), payloadLen <= 65535 else { return nil }
|
if version == 2 {
|
||||||
|
guard let len = read32() else { return nil }
|
||||||
|
payloadLength = Int(len)
|
||||||
|
} else {
|
||||||
|
guard let len = read16() else { return nil }
|
||||||
|
payloadLength = Int(len)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard payloadLength >= 0 else { return nil }
|
||||||
|
|
||||||
// SenderID
|
|
||||||
guard let senderID = readData(senderIDSize) else { return nil }
|
guard let senderID = readData(senderIDSize) else { return nil }
|
||||||
|
|
||||||
// Recipient
|
|
||||||
var recipientID: Data? = nil
|
var recipientID: Data? = nil
|
||||||
if hasRecipient {
|
if hasRecipient {
|
||||||
recipientID = readData(recipientIDSize)
|
recipientID = readData(recipientIDSize)
|
||||||
if recipientID == nil { return nil }
|
if recipientID == nil { return nil }
|
||||||
}
|
}
|
||||||
|
|
||||||
// Payload
|
// Route (optional, v2+ only): route bytes are NOT included in payloadLength
|
||||||
|
var route: [Data]? = nil
|
||||||
|
if hasRoute {
|
||||||
|
guard let routeCount = read8() else { return nil }
|
||||||
|
if routeCount > 0 {
|
||||||
|
var hops: [Data] = []
|
||||||
|
for _ in 0..<Int(routeCount) {
|
||||||
|
guard let hop = readData(senderIDSize) else { return nil }
|
||||||
|
hops.append(hop)
|
||||||
|
}
|
||||||
|
route = hops
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Payload: payloadLength is exactly the payload size (+ compression preamble if compressed)
|
||||||
let payload: Data
|
let payload: Data
|
||||||
if isCompressed {
|
if isCompressed {
|
||||||
// Need original size (2 bytes)
|
guard payloadLength >= lengthFieldBytes else { return nil }
|
||||||
guard let origSize16 = read16() else { return nil }
|
let originalSize: Int
|
||||||
let originalSize = Int(origSize16)
|
if version == 2 {
|
||||||
guard originalSize >= 0 && originalSize <= 1_048_576 else { return nil }
|
guard let rawSize = read32() else { return nil }
|
||||||
let compSize = Int(payloadLen) - 2
|
originalSize = Int(rawSize)
|
||||||
guard compSize >= 0, let compressed = readData(compSize) else { return nil }
|
} else {
|
||||||
|
guard let rawSize = read16() else { return nil }
|
||||||
|
originalSize = Int(rawSize)
|
||||||
|
}
|
||||||
|
guard originalSize >= 0 && originalSize <= FileTransferLimits.maxFramedFileBytes else { return nil }
|
||||||
|
let compressedSize = payloadLength - lengthFieldBytes
|
||||||
|
guard compressedSize > 0, let compressed = readData(compressedSize) else { return nil }
|
||||||
|
|
||||||
|
let compressionRatio = Double(originalSize) / Double(compressedSize)
|
||||||
|
guard compressionRatio <= 50_000.0 else {
|
||||||
|
SecureLogger.warning("🚫 Suspicious compression ratio: \(String(format: "%.0f", compressionRatio)):1", category: .security)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
guard let decompressed = CompressionUtil.decompress(compressed, originalSize: originalSize),
|
guard let decompressed = CompressionUtil.decompress(compressed, originalSize: originalSize),
|
||||||
decompressed.count == originalSize else { return nil }
|
decompressed.count == originalSize else { return nil }
|
||||||
payload = decompressed
|
payload = decompressed
|
||||||
} else {
|
} else {
|
||||||
guard let p = readData(Int(payloadLen)) else { return nil }
|
guard let rawPayload = readData(payloadLength) else { return nil }
|
||||||
payload = p
|
payload = rawPayload
|
||||||
}
|
}
|
||||||
|
|
||||||
// Signature
|
|
||||||
var signature: Data? = nil
|
var signature: Data? = nil
|
||||||
if hasSignature {
|
if hasSignature {
|
||||||
signature = readData(signatureSize)
|
signature = readData(signatureSize)
|
||||||
@@ -320,10 +405,12 @@ struct BinaryProtocol {
|
|||||||
type: type,
|
type: type,
|
||||||
senderID: senderID,
|
senderID: senderID,
|
||||||
recipientID: recipientID,
|
recipientID: recipientID,
|
||||||
timestamp: ts,
|
timestamp: timestamp,
|
||||||
payload: payload,
|
payload: payload,
|
||||||
signature: signature,
|
signature: signature,
|
||||||
ttl: ttl
|
ttl: ttl,
|
||||||
|
version: version,
|
||||||
|
route: route
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,155 @@
|
|||||||
|
//
|
||||||
|
// BitchatFilePacket.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
// For more information, see <https://unlicense.org>
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
import BitLogger
|
||||||
|
|
||||||
|
/// TLV payload for Bluetooth mesh file transfers (voice notes, images, generic files).
|
||||||
|
/// Mirrors the Android client specification to ensure cross-platform interoperability.
|
||||||
|
struct BitchatFilePacket {
|
||||||
|
var fileName: String?
|
||||||
|
var fileSize: UInt64?
|
||||||
|
var mimeType: String?
|
||||||
|
var content: Data
|
||||||
|
|
||||||
|
/// Canonical TLV tags defined by the Android implementation.
|
||||||
|
private enum TLVType: UInt8 {
|
||||||
|
case fileName = 0x01
|
||||||
|
case fileSize = 0x02
|
||||||
|
case mimeType = 0x03
|
||||||
|
case content = 0x04
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Encodes the packet using v2 canonical TLVs (4-byte FILE_SIZE, 4-byte CONTENT length).
|
||||||
|
/// Returns `nil` when fields exceed protocol limits (e.g., content > UInt32.max).
|
||||||
|
func encode() -> Data? {
|
||||||
|
let resolvedSize = fileSize ?? UInt64(content.count)
|
||||||
|
guard resolvedSize <= UInt64(UInt32.max) else { return nil }
|
||||||
|
guard resolvedSize <= UInt64(FileTransferLimits.maxPayloadBytes) else { return nil }
|
||||||
|
guard content.count <= Int(UInt32.max) else { return nil }
|
||||||
|
guard FileTransferLimits.isValidPayload(content.count) else { return nil }
|
||||||
|
|
||||||
|
func appendBE<T: FixedWidthInteger>(_ value: T, into data: inout Data) {
|
||||||
|
var big = value.bigEndian
|
||||||
|
withUnsafeBytes(of: &big) { data.append(contentsOf: $0) }
|
||||||
|
}
|
||||||
|
|
||||||
|
var encoded = Data()
|
||||||
|
|
||||||
|
if let name = fileName, let nameData = name.data(using: .utf8), nameData.count <= Int(UInt16.max) {
|
||||||
|
encoded.append(TLVType.fileName.rawValue)
|
||||||
|
appendBE(UInt16(nameData.count), into: &encoded)
|
||||||
|
encoded.append(nameData)
|
||||||
|
}
|
||||||
|
|
||||||
|
encoded.append(TLVType.fileSize.rawValue)
|
||||||
|
appendBE(UInt16(4), into: &encoded)
|
||||||
|
appendBE(UInt32(resolvedSize), into: &encoded)
|
||||||
|
|
||||||
|
if let mime = mimeType, let mimeData = mime.data(using: .utf8), mimeData.count <= Int(UInt16.max) {
|
||||||
|
encoded.append(TLVType.mimeType.rawValue)
|
||||||
|
appendBE(UInt16(mimeData.count), into: &encoded)
|
||||||
|
encoded.append(mimeData)
|
||||||
|
}
|
||||||
|
|
||||||
|
encoded.append(TLVType.content.rawValue)
|
||||||
|
appendBE(UInt32(content.count), into: &encoded)
|
||||||
|
encoded.append(content)
|
||||||
|
|
||||||
|
return encoded
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Decodes TLV payloads, tolerating legacy encodings (FILE_SIZE len=8, CONTENT len=2) when possible.
|
||||||
|
static func decode(_ data: Data) -> BitchatFilePacket? {
|
||||||
|
var cursor = data.startIndex
|
||||||
|
let end = data.endIndex
|
||||||
|
|
||||||
|
var fileName: String?
|
||||||
|
var fileSize: UInt64?
|
||||||
|
var mimeType: String?
|
||||||
|
var content = Data()
|
||||||
|
|
||||||
|
while cursor < end {
|
||||||
|
let typeRaw = data[cursor]
|
||||||
|
cursor = data.index(after: cursor)
|
||||||
|
|
||||||
|
guard cursor <= end else { return nil }
|
||||||
|
let tlvType = TLVType(rawValue: typeRaw)
|
||||||
|
|
||||||
|
func readBigEndianLength(bytes: Int) -> Int? {
|
||||||
|
guard data.distance(from: cursor, to: end) >= bytes else { return nil }
|
||||||
|
// Use UInt64 to prevent integer overflow during shift operations
|
||||||
|
var result: UInt64 = 0
|
||||||
|
for _ in 0..<bytes {
|
||||||
|
result = (result << 8) | UInt64(data[cursor])
|
||||||
|
cursor = data.index(after: cursor)
|
||||||
|
}
|
||||||
|
// Safely convert to Int with overflow check
|
||||||
|
guard result <= Int.max else { return nil }
|
||||||
|
return Int(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
let length: Int?
|
||||||
|
if tlvType == .content {
|
||||||
|
let snapshot = cursor
|
||||||
|
let canonical = readBigEndianLength(bytes: 4)
|
||||||
|
if let canonical = canonical,
|
||||||
|
canonical <= data.distance(from: cursor, to: end) {
|
||||||
|
length = canonical
|
||||||
|
} else {
|
||||||
|
cursor = snapshot
|
||||||
|
length = readBigEndianLength(bytes: 2)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
length = readBigEndianLength(bytes: 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
guard let tlvLength = length, tlvLength >= 0 else { return nil }
|
||||||
|
guard data.distance(from: cursor, to: end) >= tlvLength else { return nil }
|
||||||
|
|
||||||
|
let valueStart = cursor
|
||||||
|
cursor = data.index(cursor, offsetBy: tlvLength)
|
||||||
|
let value = data[valueStart..<cursor]
|
||||||
|
|
||||||
|
switch tlvType {
|
||||||
|
case .fileName:
|
||||||
|
fileName = String(data: Data(value), encoding: .utf8)
|
||||||
|
case .fileSize:
|
||||||
|
if tlvLength == 4 || tlvLength == 8 {
|
||||||
|
var size: UInt64 = 0
|
||||||
|
for byte in value {
|
||||||
|
size = (size << 8) | UInt64(byte)
|
||||||
|
}
|
||||||
|
if size > UInt64(FileTransferLimits.maxPayloadBytes) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
fileSize = size
|
||||||
|
}
|
||||||
|
case .mimeType:
|
||||||
|
mimeType = String(data: Data(value), encoding: .utf8)
|
||||||
|
case .content:
|
||||||
|
let proposedSize = content.count + value.count
|
||||||
|
if proposedSize > FileTransferLimits.maxPayloadBytes {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
content.append(contentsOf: value)
|
||||||
|
case nil:
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
guard !content.isEmpty else { return nil }
|
||||||
|
guard FileTransferLimits.isValidPayload(content.count) else { return nil }
|
||||||
|
return BitchatFilePacket(
|
||||||
|
fileName: fileName,
|
||||||
|
fileSize: fileSize ?? UInt64(content.count),
|
||||||
|
mimeType: mimeType,
|
||||||
|
content: content
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -79,6 +79,7 @@ enum MessageType: UInt8 {
|
|||||||
|
|
||||||
// Fragmentation (simplified)
|
// Fragmentation (simplified)
|
||||||
case fragment = 0x20 // Single fragment type for large messages
|
case fragment = 0x20 // Single fragment type for large messages
|
||||||
|
case fileTransfer = 0x22 // Binary file/audio/image payloads
|
||||||
|
|
||||||
var description: String {
|
var description: String {
|
||||||
switch self {
|
switch self {
|
||||||
@@ -89,6 +90,7 @@ enum MessageType: UInt8 {
|
|||||||
case .noiseHandshake: return "noiseHandshake"
|
case .noiseHandshake: return "noiseHandshake"
|
||||||
case .noiseEncrypted: return "noiseEncrypted"
|
case .noiseEncrypted: return "noiseEncrypted"
|
||||||
case .fragment: return "fragment"
|
case .fragment: return "fragment"
|
||||||
|
case .fileTransfer: return "fileTransfer"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -176,7 +178,7 @@ protocol BitchatDelegate: AnyObject {
|
|||||||
|
|
||||||
// Bluetooth state updates for user notifications
|
// Bluetooth state updates for user notifications
|
||||||
func didUpdateBluetoothState(_ state: CBManagerState)
|
func didUpdateBluetoothState(_ state: CBManagerState)
|
||||||
func didReceivePublicMessage(from peerID: PeerID, nickname: String, content: String, timestamp: Date)
|
func didReceivePublicMessage(from peerID: PeerID, nickname: String, content: String, timestamp: Date, messageID: String?)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Provide default implementation to make it effectively optional
|
// Provide default implementation to make it effectively optional
|
||||||
@@ -193,7 +195,7 @@ extension BitchatDelegate {
|
|||||||
// Default empty implementation
|
// Default empty implementation
|
||||||
}
|
}
|
||||||
|
|
||||||
func didReceivePublicMessage(from peerID: PeerID, nickname: String, content: String, timestamp: Date) {
|
func didReceivePublicMessage(from peerID: PeerID, nickname: String, content: String, timestamp: Date, messageID: String?) {
|
||||||
// Default empty implementation
|
// Default empty implementation
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -119,4 +119,57 @@ enum Geohash {
|
|||||||
}
|
}
|
||||||
return (latInterval.0, latInterval.1, lonInterval.0, lonInterval.1)
|
return (latInterval.0, latInterval.1, lonInterval.0, lonInterval.1)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Returns all 8 neighboring geohash cells at the same precision.
|
||||||
|
/// - Parameter geohash: Base32 geohash string.
|
||||||
|
/// - Returns: Array of 8 neighboring geohashes (N, NE, E, SE, S, SW, W, NW order).
|
||||||
|
static func neighbors(of geohash: String) -> [String] {
|
||||||
|
guard !geohash.isEmpty else { return [] }
|
||||||
|
|
||||||
|
let precision = geohash.count
|
||||||
|
let bounds = decodeBounds(geohash)
|
||||||
|
let center = decodeCenter(geohash)
|
||||||
|
|
||||||
|
// Calculate cell dimensions
|
||||||
|
let latHeight = bounds.latMax - bounds.latMin
|
||||||
|
let lonWidth = bounds.lonMax - bounds.lonMin
|
||||||
|
|
||||||
|
// Helper to wrap longitude around ±180
|
||||||
|
func wrapLongitude(_ lon: Double) -> Double {
|
||||||
|
var wrapped = lon
|
||||||
|
while wrapped > 180.0 { wrapped -= 360.0 }
|
||||||
|
while wrapped < -180.0 { wrapped += 360.0 }
|
||||||
|
return wrapped
|
||||||
|
}
|
||||||
|
|
||||||
|
// Helper to clamp latitude to ±90
|
||||||
|
func clampLatitude(_ lat: Double) -> Double {
|
||||||
|
return max(-90.0, min(90.0, lat))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Calculate 8 neighbor centers
|
||||||
|
let neighbors: [(lat: Double, lon: Double)] = [
|
||||||
|
(center.lat + latHeight, center.lon), // N
|
||||||
|
(center.lat + latHeight, center.lon + lonWidth), // NE
|
||||||
|
(center.lat, center.lon + lonWidth), // E
|
||||||
|
(center.lat - latHeight, center.lon + lonWidth), // SE
|
||||||
|
(center.lat - latHeight, center.lon), // S
|
||||||
|
(center.lat - latHeight, center.lon - lonWidth), // SW
|
||||||
|
(center.lat, center.lon - lonWidth), // W
|
||||||
|
(center.lat + latHeight, center.lon - lonWidth) // NW
|
||||||
|
]
|
||||||
|
|
||||||
|
// Encode each neighbor, handling boundary conditions
|
||||||
|
return neighbors.compactMap { neighbor in
|
||||||
|
let lat = clampLatitude(neighbor.lat)
|
||||||
|
let lon = wrapLongitude(neighbor.lon)
|
||||||
|
|
||||||
|
// Skip if we've crossed a pole (latitude clamped to boundary)
|
||||||
|
if (neighbor.lat > 90.0 || neighbor.lat < -90.0) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return encode(latitude: lat, longitude: lon, precision: precision)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -116,4 +116,18 @@ enum ChannelID: Equatable, Codable {
|
|||||||
case .location(let ch): return ch.geohash
|
case .location(let ch): return ch.geohash
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var isMesh: Bool {
|
||||||
|
switch self {
|
||||||
|
case .mesh: true
|
||||||
|
case .location: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var isLocation: Bool {
|
||||||
|
switch self {
|
||||||
|
case .mesh: false
|
||||||
|
case .location: true
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,11 +6,13 @@ struct AnnouncementPacket {
|
|||||||
let nickname: String
|
let nickname: String
|
||||||
let noisePublicKey: Data // Noise static public key (Curve25519.KeyAgreement)
|
let noisePublicKey: Data // Noise static public key (Curve25519.KeyAgreement)
|
||||||
let signingPublicKey: Data // Ed25519 public key for signing
|
let signingPublicKey: Data // Ed25519 public key for signing
|
||||||
|
let directNeighbors: [Data]? // 8-byte peer IDs
|
||||||
|
|
||||||
private enum TLVType: UInt8 {
|
private enum TLVType: UInt8 {
|
||||||
case nickname = 0x01
|
case nickname = 0x01
|
||||||
case noisePublicKey = 0x02
|
case noisePublicKey = 0x02
|
||||||
case signingPublicKey = 0x03
|
case signingPublicKey = 0x03
|
||||||
|
case directNeighbors = 0x04
|
||||||
}
|
}
|
||||||
|
|
||||||
func encode() -> Data? {
|
func encode() -> Data? {
|
||||||
@@ -36,6 +38,16 @@ struct AnnouncementPacket {
|
|||||||
data.append(UInt8(signingPublicKey.count))
|
data.append(UInt8(signingPublicKey.count))
|
||||||
data.append(signingPublicKey)
|
data.append(signingPublicKey)
|
||||||
|
|
||||||
|
// TLV for direct neighbors (optional)
|
||||||
|
if let neighbors = directNeighbors, !neighbors.isEmpty {
|
||||||
|
let neighborsData = neighbors.prefix(10).reduce(Data()) { $0 + $1 }
|
||||||
|
if !neighborsData.isEmpty && neighborsData.count % 8 == 0 {
|
||||||
|
data.append(TLVType.directNeighbors.rawValue)
|
||||||
|
data.append(UInt8(neighborsData.count))
|
||||||
|
data.append(neighborsData)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return data
|
return data
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -44,6 +56,7 @@ struct AnnouncementPacket {
|
|||||||
var nickname: String?
|
var nickname: String?
|
||||||
var noisePublicKey: Data?
|
var noisePublicKey: Data?
|
||||||
var signingPublicKey: Data?
|
var signingPublicKey: Data?
|
||||||
|
var directNeighbors: [Data]?
|
||||||
|
|
||||||
while offset + 2 <= data.count {
|
while offset + 2 <= data.count {
|
||||||
let typeRaw = data[offset]
|
let typeRaw = data[offset]
|
||||||
@@ -63,6 +76,17 @@ struct AnnouncementPacket {
|
|||||||
noisePublicKey = Data(value)
|
noisePublicKey = Data(value)
|
||||||
case .signingPublicKey:
|
case .signingPublicKey:
|
||||||
signingPublicKey = Data(value)
|
signingPublicKey = Data(value)
|
||||||
|
case .directNeighbors:
|
||||||
|
if length > 0 && length % 8 == 0 {
|
||||||
|
var neighbors = [Data]()
|
||||||
|
let count = length / 8
|
||||||
|
for i in 0..<count {
|
||||||
|
let start = value.startIndex + i * 8
|
||||||
|
let end = start + 8
|
||||||
|
neighbors.append(Data(value[start..<end]))
|
||||||
|
}
|
||||||
|
directNeighbors = neighbors
|
||||||
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// Unknown TLV; skip (tolerant decoder for forward compatibility)
|
// Unknown TLV; skip (tolerant decoder for forward compatibility)
|
||||||
@@ -74,7 +98,8 @@ struct AnnouncementPacket {
|
|||||||
return AnnouncementPacket(
|
return AnnouncementPacket(
|
||||||
nickname: nickname,
|
nickname: nickname,
|
||||||
noisePublicKey: noisePublicKey,
|
noisePublicKey: noisePublicKey,
|
||||||
signingPublicKey: signingPublicKey
|
signingPublicKey: signingPublicKey,
|
||||||
|
directNeighbors: directNeighbors
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,195 @@
|
|||||||
|
//
|
||||||
|
// MimeType.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
// For more information, see <https://unlicense.org>
|
||||||
|
//
|
||||||
|
|
||||||
|
import UniformTypeIdentifiers
|
||||||
|
|
||||||
|
// MARK: - Extensions for missing UTTypes
|
||||||
|
|
||||||
|
extension UTType {
|
||||||
|
static let webP = UTType(importedAs: "image/webp")
|
||||||
|
static let aac = UTType(importedAs: "audio/aac")
|
||||||
|
static let m4a = UTType(importedAs: "audio/m4a")
|
||||||
|
static let ogg = UTType(importedAs: "audio/ogg")
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - MimeType Enum
|
||||||
|
|
||||||
|
enum MimeType: CaseIterable, Hashable {
|
||||||
|
case jpeg
|
||||||
|
case jpg
|
||||||
|
case png
|
||||||
|
case gif
|
||||||
|
case webp
|
||||||
|
case mp4Audio
|
||||||
|
case m4a
|
||||||
|
case aac
|
||||||
|
case mpeg
|
||||||
|
case mp3
|
||||||
|
case wav
|
||||||
|
case xWav
|
||||||
|
case ogg
|
||||||
|
case pdf
|
||||||
|
case octetStream
|
||||||
|
|
||||||
|
var utType: UTType {
|
||||||
|
switch self {
|
||||||
|
case .jpeg, .jpg: .jpeg
|
||||||
|
case .png: .png
|
||||||
|
case .gif: .gif
|
||||||
|
case .webp: .webP
|
||||||
|
case .aac: .aac
|
||||||
|
case .m4a: .m4a
|
||||||
|
case .mp4Audio: .mpeg4Audio
|
||||||
|
case .mp3, .mpeg: .mp3
|
||||||
|
case .wav, .xWav: .wav
|
||||||
|
case .ogg: .ogg
|
||||||
|
case .pdf: .pdf
|
||||||
|
case .octetStream: .data
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var category: Category {
|
||||||
|
switch self {
|
||||||
|
case .jpeg, .jpg, .png, .gif, .webp:
|
||||||
|
return .image
|
||||||
|
case .aac, .m4a, .mp4Audio, .mpeg, .mp3, .wav, .xWav, .ogg:
|
||||||
|
return .audio
|
||||||
|
case .pdf, .octetStream:
|
||||||
|
return .file
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
var mimeString: String {
|
||||||
|
switch self {
|
||||||
|
case .jpeg, .jpg: "image/jpeg"
|
||||||
|
case .png: "image/png"
|
||||||
|
case .gif: "image/gif"
|
||||||
|
case .webp: "image/webp"
|
||||||
|
case .mp4Audio: "audio/mp4"
|
||||||
|
case .m4a: "audio/m4a"
|
||||||
|
case .aac: "audio/aac"
|
||||||
|
case .mpeg: "audio/mpeg"
|
||||||
|
case .mp3: "audio/mp3"
|
||||||
|
case .wav: "audio/wav"
|
||||||
|
case .xWav: "audio/x-wav"
|
||||||
|
case .ogg: "audio/ogg"
|
||||||
|
case .pdf: "application/pdf"
|
||||||
|
case .octetStream: "application/octet-stream"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var defaultExtension: String {
|
||||||
|
switch self {
|
||||||
|
case .jpeg, .jpg: "jpg"
|
||||||
|
case .png: "png"
|
||||||
|
case .webp: "webp"
|
||||||
|
case .gif: "gif"
|
||||||
|
case .mp4Audio, .m4a, .aac: "m4a"
|
||||||
|
case .mpeg, .mp3: "mp3"
|
||||||
|
case .wav, .xWav: "wav"
|
||||||
|
case .ogg: "ogg"
|
||||||
|
case .pdf: "pdf"
|
||||||
|
case .octetStream: "bin"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static var allowed: Set<MimeType> = [
|
||||||
|
.jpeg, .jpg, .png, .gif, .webp,
|
||||||
|
.mp4Audio, .m4a, .aac, .mpeg, .mp3,
|
||||||
|
.wav, .xWav, .ogg,
|
||||||
|
.pdf, .octetStream
|
||||||
|
]
|
||||||
|
|
||||||
|
var isAllowed: Bool {
|
||||||
|
Self.allowed.contains(self)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Byte signature validation
|
||||||
|
func matches(data: Data) -> Bool {
|
||||||
|
guard !data.isEmpty else { return false }
|
||||||
|
|
||||||
|
// Generic type → skip validation
|
||||||
|
if self == .octetStream { return true }
|
||||||
|
|
||||||
|
switch self {
|
||||||
|
case .jpeg, .jpg:
|
||||||
|
return data.count >= 3 && data[0] == 0xFF && data[1] == 0xD8 && data[2] == 0xFF
|
||||||
|
|
||||||
|
case .png:
|
||||||
|
return data.count >= 8 &&
|
||||||
|
data[0] == 0x89 && data[1] == 0x50 && data[2] == 0x4E && data[3] == 0x47 &&
|
||||||
|
data[4] == 0x0D && data[5] == 0x0A && data[6] == 0x1A && data[7] == 0x0A
|
||||||
|
|
||||||
|
case .gif:
|
||||||
|
return data.count >= 6 && data[0] == 0x47 && data[1] == 0x49 && data[2] == 0x46 &&
|
||||||
|
data[3] == 0x38 && (data[4] == 0x37 || data[4] == 0x39) && data[5] == 0x61
|
||||||
|
|
||||||
|
case .webp:
|
||||||
|
return data.count >= 12 &&
|
||||||
|
data[0] == 0x52 && data[1] == 0x49 && data[2] == 0x46 && data[3] == 0x46 &&
|
||||||
|
data[8] == 0x57 && data[9] == 0x45 && data[10] == 0x42 && data[11] == 0x50
|
||||||
|
|
||||||
|
case .m4a, .mp4Audio, .aac:
|
||||||
|
// AVAudioRecorder output varies by platform - be lenient
|
||||||
|
// Security: size already capped + sandboxed execution
|
||||||
|
return data.count > 100
|
||||||
|
|
||||||
|
case .mpeg, .mp3:
|
||||||
|
if data.count >= 3 && data[0] == 0x49 && data[1] == 0x44 && data[2] == 0x33 {
|
||||||
|
return true // ID3 header
|
||||||
|
}
|
||||||
|
return data.count >= 2 && data[0] == 0xFF && (data[1] & 0xE0) == 0xE0
|
||||||
|
|
||||||
|
case .wav, .xWav:
|
||||||
|
return data.count >= 12 &&
|
||||||
|
data[0] == 0x52 && data[1] == 0x49 && data[2] == 0x46 && data[3] == 0x46 &&
|
||||||
|
data[8] == 0x57 && data[9] == 0x41 && data[10] == 0x56 && data[11] == 0x45
|
||||||
|
|
||||||
|
case .ogg:
|
||||||
|
return data.count >= 4 &&
|
||||||
|
data[0] == 0x4F && data[1] == 0x67 && data[2] == 0x67 && data[3] == 0x53
|
||||||
|
|
||||||
|
case .pdf:
|
||||||
|
return data.count >= 4 &&
|
||||||
|
data[0] == 0x25 && data[1] == 0x50 && data[2] == 0x44 && data[3] == 0x46
|
||||||
|
|
||||||
|
default:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Convenience Initializers
|
||||||
|
|
||||||
|
init?(_ mimeString: String?) {
|
||||||
|
guard let mimeString else { return nil }
|
||||||
|
|
||||||
|
let normalized = mimeString.lowercased()
|
||||||
|
|
||||||
|
// Direct match with our canonical list
|
||||||
|
if let match = MimeType.allCases.first(where: { $0.mimeString == normalized }) {
|
||||||
|
self = match
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Let UTType normalize aliases like "image/jpg", "audio/x-wav", etc.
|
||||||
|
if let type = UTType(mimeType: normalized),
|
||||||
|
let match = MimeType.allCases.first(where: { type.conforms(to: $0.utType) }) {
|
||||||
|
self = match
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
extension MimeType {
|
||||||
|
enum Category: String {
|
||||||
|
case audio, image, file
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -15,19 +15,67 @@ enum CommandResult {
|
|||||||
case handled // Command handled, no message needed
|
case handled // Command handled, no message needed
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Simple struct for geo participant info used by CommandProcessor
|
||||||
|
struct CommandGeoParticipant {
|
||||||
|
let id: String // pubkey hex (lowercased)
|
||||||
|
let displayName: String
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Protocol defining what CommandProcessor needs from its context.
|
||||||
|
/// This breaks the circular dependency between CommandProcessor and ChatViewModel.
|
||||||
|
@MainActor
|
||||||
|
protocol CommandContextProvider: AnyObject {
|
||||||
|
// MARK: - State Properties
|
||||||
|
var nickname: String { get }
|
||||||
|
var selectedPrivateChatPeer: PeerID? { get }
|
||||||
|
var blockedUsers: Set<String> { get }
|
||||||
|
var privateChats: [PeerID: [BitchatMessage]] { get set }
|
||||||
|
var idBridge: NostrIdentityBridge { get }
|
||||||
|
|
||||||
|
// MARK: - Peer Lookup
|
||||||
|
func getPeerIDForNickname(_ nickname: String) -> PeerID?
|
||||||
|
func getVisibleGeoParticipants() -> [CommandGeoParticipant]
|
||||||
|
func nostrPubkeyForDisplayName(_ displayName: String) -> String?
|
||||||
|
|
||||||
|
// MARK: - Chat Actions
|
||||||
|
func startPrivateChat(with peerID: PeerID)
|
||||||
|
func sendPrivateMessage(_ content: String, to peerID: PeerID)
|
||||||
|
func clearCurrentPublicTimeline()
|
||||||
|
func sendPublicRaw(_ content: String)
|
||||||
|
|
||||||
|
// MARK: - System Messages
|
||||||
|
func addLocalPrivateSystemMessage(_ content: String, to peerID: PeerID)
|
||||||
|
func addPublicSystemMessage(_ content: String)
|
||||||
|
|
||||||
|
// MARK: - Favorites
|
||||||
|
func toggleFavorite(peerID: PeerID)
|
||||||
|
func sendFavoriteNotification(to peerID: PeerID, isFavorite: Bool)
|
||||||
|
}
|
||||||
|
|
||||||
/// Processes chat commands in a focused, efficient way
|
/// Processes chat commands in a focused, efficient way
|
||||||
@MainActor
|
@MainActor
|
||||||
final class CommandProcessor {
|
final class CommandProcessor {
|
||||||
weak var chatViewModel: ChatViewModel?
|
weak var contextProvider: CommandContextProvider?
|
||||||
weak var meshService: Transport?
|
weak var meshService: Transport?
|
||||||
private let identityManager: SecureIdentityStateManagerProtocol
|
private let identityManager: SecureIdentityStateManagerProtocol
|
||||||
|
|
||||||
init(chatViewModel: ChatViewModel? = nil, meshService: Transport? = nil, identityManager: SecureIdentityStateManagerProtocol) {
|
/// Backward-compatible property for existing code
|
||||||
self.chatViewModel = chatViewModel
|
weak var chatViewModel: CommandContextProvider? {
|
||||||
|
get { contextProvider }
|
||||||
|
set { contextProvider = newValue }
|
||||||
|
}
|
||||||
|
|
||||||
|
init(contextProvider: CommandContextProvider? = nil, meshService: Transport? = nil, identityManager: SecureIdentityStateManagerProtocol) {
|
||||||
|
self.contextProvider = contextProvider
|
||||||
self.meshService = meshService
|
self.meshService = meshService
|
||||||
self.identityManager = identityManager
|
self.identityManager = identityManager
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Backward-compatible initializer
|
||||||
|
convenience init(chatViewModel: ChatViewModel? = nil, meshService: Transport? = nil, identityManager: SecureIdentityStateManagerProtocol) {
|
||||||
|
self.init(contextProvider: chatViewModel, meshService: meshService, identityManager: identityManager)
|
||||||
|
}
|
||||||
|
|
||||||
/// Process a command string
|
/// Process a command string
|
||||||
@MainActor
|
@MainActor
|
||||||
func process(_ command: String) -> CommandResult {
|
func process(_ command: String) -> CommandResult {
|
||||||
@@ -42,7 +90,7 @@ final class CommandProcessor {
|
|||||||
case .location: return true
|
case .location: return true
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
let inGeoDM = chatViewModel?.selectedPrivateChatPeer?.isGeoDM == true
|
let inGeoDM = contextProvider?.selectedPrivateChatPeer?.isGeoDM == true
|
||||||
|
|
||||||
switch cmd {
|
switch cmd {
|
||||||
case "/m", "/msg":
|
case "/m", "/msg":
|
||||||
@@ -81,15 +129,15 @@ final class CommandProcessor {
|
|||||||
let targetName = String(parts[0])
|
let targetName = String(parts[0])
|
||||||
let nickname = targetName.hasPrefix("@") ? String(targetName.dropFirst()) : targetName
|
let nickname = targetName.hasPrefix("@") ? String(targetName.dropFirst()) : targetName
|
||||||
|
|
||||||
guard let peerID = chatViewModel?.getPeerIDForNickname(nickname) else {
|
guard let peerID = contextProvider?.getPeerIDForNickname(nickname) else {
|
||||||
return .error(message: "'\(nickname)' not found")
|
return .error(message: "'\(nickname)' not found")
|
||||||
}
|
}
|
||||||
|
|
||||||
chatViewModel?.startPrivateChat(with: peerID)
|
contextProvider?.startPrivateChat(with: peerID)
|
||||||
|
|
||||||
if parts.count > 1 {
|
if parts.count > 1 {
|
||||||
let message = String(parts[1])
|
let message = String(parts[1])
|
||||||
chatViewModel?.sendPrivateMessage(message, to: peerID)
|
contextProvider?.sendPrivateMessage(message, to: peerID)
|
||||||
}
|
}
|
||||||
|
|
||||||
return .success(message: "started private chat with \(nickname)")
|
return .success(message: "started private chat with \(nickname)")
|
||||||
@@ -100,9 +148,9 @@ final class CommandProcessor {
|
|||||||
switch LocationChannelManager.shared.selectedChannel {
|
switch LocationChannelManager.shared.selectedChannel {
|
||||||
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 = chatViewModel else { return .success(message: "nobody around") }
|
guard let vm = contextProvider else { return .success(message: "nobody around") }
|
||||||
let myHex = (try? chatViewModel?.idBridge.deriveIdentity(forGeohash: ch.geohash))?.publicKeyHex.lowercased()
|
let myHex = (try? vm.idBridge.deriveIdentity(forGeohash: ch.geohash))?.publicKeyHex.lowercased()
|
||||||
let people = vm.visibleGeohashPeople().filter { person in
|
let people = vm.getVisibleGeoParticipants().filter { person in
|
||||||
if let me = myHex { return person.id.lowercased() != me }
|
if let me = myHex { return person.id.lowercased() != me }
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
@@ -120,10 +168,10 @@ final class CommandProcessor {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private func handleClear() -> CommandResult {
|
private func handleClear() -> CommandResult {
|
||||||
if let peerID = chatViewModel?.selectedPrivateChatPeer {
|
if let peerID = contextProvider?.selectedPrivateChatPeer {
|
||||||
chatViewModel?.privateChats[peerID]?.removeAll()
|
contextProvider?.privateChats[peerID]?.removeAll()
|
||||||
} else {
|
} else {
|
||||||
chatViewModel?.clearCurrentPublicTimeline()
|
contextProvider?.clearCurrentPublicTimeline()
|
||||||
}
|
}
|
||||||
return .handled
|
return .handled
|
||||||
}
|
}
|
||||||
@@ -136,14 +184,14 @@ final class CommandProcessor {
|
|||||||
|
|
||||||
let nickname = targetName.hasPrefix("@") ? String(targetName.dropFirst()) : targetName
|
let nickname = targetName.hasPrefix("@") ? String(targetName.dropFirst()) : targetName
|
||||||
|
|
||||||
guard let targetPeerID = chatViewModel?.getPeerIDForNickname(nickname),
|
guard let targetPeerID = contextProvider?.getPeerIDForNickname(nickname),
|
||||||
let myNickname = chatViewModel?.nickname else {
|
let myNickname = contextProvider?.nickname else {
|
||||||
return .error(message: "cannot \(command) \(nickname): not found")
|
return .error(message: "cannot \(command) \(nickname): not found")
|
||||||
}
|
}
|
||||||
|
|
||||||
let emoteContent = "* \(emoji) \(myNickname) \(action) \(nickname)\(suffix) *"
|
let emoteContent = "* \(emoji) \(myNickname) \(action) \(nickname)\(suffix) *"
|
||||||
|
|
||||||
if chatViewModel?.selectedPrivateChatPeer != nil {
|
if contextProvider?.selectedPrivateChatPeer != nil {
|
||||||
// In private chat
|
// In private chat
|
||||||
if let peerNickname = meshService?.peerNickname(peerID: targetPeerID) {
|
if let peerNickname = meshService?.peerNickname(peerID: targetPeerID) {
|
||||||
let personalMessage = "* \(emoji) \(myNickname) \(action) you\(suffix) *"
|
let personalMessage = "* \(emoji) \(myNickname) \(action) you\(suffix) *"
|
||||||
@@ -159,13 +207,13 @@ final class CommandProcessor {
|
|||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
let localText = "\(emoji) you \(pastAction) \(nickname)\(suffix)"
|
let localText = "\(emoji) you \(pastAction) \(nickname)\(suffix)"
|
||||||
chatViewModel?.addLocalPrivateSystemMessage(localText, to: targetPeerID)
|
contextProvider?.addLocalPrivateSystemMessage(localText, to: targetPeerID)
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// In public chat: send to active public channel (mesh or geohash)
|
// In public chat: send to active public channel (mesh or geohash)
|
||||||
chatViewModel?.sendPublicRaw(emoteContent)
|
contextProvider?.sendPublicRaw(emoteContent)
|
||||||
let publicEcho = "\(emoji) \(myNickname) \(action) \(nickname)\(suffix)"
|
let publicEcho = "\(emoji) \(myNickname) \(action) \(nickname)\(suffix)"
|
||||||
chatViewModel?.addPublicSystemMessage(publicEcho)
|
contextProvider?.addPublicSystemMessage(publicEcho)
|
||||||
}
|
}
|
||||||
|
|
||||||
return .handled
|
return .handled
|
||||||
@@ -176,7 +224,7 @@ final class CommandProcessor {
|
|||||||
|
|
||||||
if targetName.isEmpty {
|
if targetName.isEmpty {
|
||||||
// List blocked users (mesh) and geohash (Nostr) blocks
|
// List blocked users (mesh) and geohash (Nostr) blocks
|
||||||
let meshBlocked = chatViewModel?.blockedUsers ?? []
|
let meshBlocked = contextProvider?.blockedUsers ?? []
|
||||||
var blockedNicknames: [String] = []
|
var blockedNicknames: [String] = []
|
||||||
if let peers = meshService?.getPeerNicknames() {
|
if let peers = meshService?.getPeerNicknames() {
|
||||||
for (peerID, nickname) in peers {
|
for (peerID, nickname) in peers {
|
||||||
@@ -190,8 +238,8 @@ final class CommandProcessor {
|
|||||||
// Geohash blocked names (prefer visible display names; fallback to #suffix)
|
// Geohash blocked names (prefer visible display names; fallback to #suffix)
|
||||||
let geoBlocked = Array(identityManager.getBlockedNostrPubkeys())
|
let geoBlocked = Array(identityManager.getBlockedNostrPubkeys())
|
||||||
var geoNames: [String] = []
|
var geoNames: [String] = []
|
||||||
if let vm = chatViewModel {
|
if let vm = contextProvider {
|
||||||
let visible = vm.visibleGeohashPeople()
|
let visible = vm.getVisibleGeoParticipants()
|
||||||
let visibleIndex = Dictionary(uniqueKeysWithValues: visible.map { ($0.id.lowercased(), $0.displayName) })
|
let visibleIndex = Dictionary(uniqueKeysWithValues: visible.map { ($0.id.lowercased(), $0.displayName) })
|
||||||
for pk in geoBlocked {
|
for pk in geoBlocked {
|
||||||
if let name = visibleIndex[pk.lowercased()] {
|
if let name = visibleIndex[pk.lowercased()] {
|
||||||
@@ -210,7 +258,7 @@ final class CommandProcessor {
|
|||||||
|
|
||||||
let nickname = targetName.hasPrefix("@") ? String(targetName.dropFirst()) : targetName
|
let nickname = targetName.hasPrefix("@") ? String(targetName.dropFirst()) : targetName
|
||||||
|
|
||||||
if let peerID = chatViewModel?.getPeerIDForNickname(nickname),
|
if let peerID = contextProvider?.getPeerIDForNickname(nickname),
|
||||||
let fingerprint = meshService?.getFingerprint(for: peerID) {
|
let fingerprint = meshService?.getFingerprint(for: peerID) {
|
||||||
if identityManager.isBlocked(fingerprint: fingerprint) {
|
if identityManager.isBlocked(fingerprint: fingerprint) {
|
||||||
return .success(message: "\(nickname) is already blocked")
|
return .success(message: "\(nickname) is already blocked")
|
||||||
@@ -235,7 +283,7 @@ final class CommandProcessor {
|
|||||||
return .success(message: "blocked \(nickname). you will no longer receive messages from them")
|
return .success(message: "blocked \(nickname). you will no longer receive messages from them")
|
||||||
}
|
}
|
||||||
// Mesh lookup failed; try geohash (Nostr) participant by display name
|
// Mesh lookup failed; try geohash (Nostr) participant by display name
|
||||||
if let pub = chatViewModel?.nostrPubkeyForDisplayName(nickname) {
|
if let pub = contextProvider?.nostrPubkeyForDisplayName(nickname) {
|
||||||
if identityManager.isNostrBlocked(pubkeyHexLowercased: pub) {
|
if identityManager.isNostrBlocked(pubkeyHexLowercased: pub) {
|
||||||
return .success(message: "\(nickname) is already blocked")
|
return .success(message: "\(nickname) is already blocked")
|
||||||
}
|
}
|
||||||
@@ -254,7 +302,7 @@ final class CommandProcessor {
|
|||||||
|
|
||||||
let nickname = targetName.hasPrefix("@") ? String(targetName.dropFirst()) : targetName
|
let nickname = targetName.hasPrefix("@") ? String(targetName.dropFirst()) : targetName
|
||||||
|
|
||||||
if let peerID = chatViewModel?.getPeerIDForNickname(nickname),
|
if let peerID = contextProvider?.getPeerIDForNickname(nickname),
|
||||||
let fingerprint = meshService?.getFingerprint(for: peerID) {
|
let fingerprint = meshService?.getFingerprint(for: peerID) {
|
||||||
if !identityManager.isBlocked(fingerprint: fingerprint) {
|
if !identityManager.isBlocked(fingerprint: fingerprint) {
|
||||||
return .success(message: "\(nickname) is not blocked")
|
return .success(message: "\(nickname) is not blocked")
|
||||||
@@ -263,7 +311,7 @@ final class CommandProcessor {
|
|||||||
return .success(message: "unblocked \(nickname)")
|
return .success(message: "unblocked \(nickname)")
|
||||||
}
|
}
|
||||||
// Try geohash unblock
|
// Try geohash unblock
|
||||||
if let pub = chatViewModel?.nostrPubkeyForDisplayName(nickname) {
|
if let pub = contextProvider?.nostrPubkeyForDisplayName(nickname) {
|
||||||
if !identityManager.isNostrBlocked(pubkeyHexLowercased: pub) {
|
if !identityManager.isNostrBlocked(pubkeyHexLowercased: pub) {
|
||||||
return .success(message: "\(nickname) is not blocked")
|
return .success(message: "\(nickname) is not blocked")
|
||||||
}
|
}
|
||||||
@@ -281,7 +329,7 @@ final class CommandProcessor {
|
|||||||
|
|
||||||
let nickname = targetName.hasPrefix("@") ? String(targetName.dropFirst()) : targetName
|
let nickname = targetName.hasPrefix("@") ? String(targetName.dropFirst()) : targetName
|
||||||
|
|
||||||
guard let peerID = chatViewModel?.getPeerIDForNickname(nickname),
|
guard let peerID = contextProvider?.getPeerIDForNickname(nickname),
|
||||||
let noisePublicKey = Data(hexString: peerID.id) else {
|
let noisePublicKey = Data(hexString: peerID.id) else {
|
||||||
return .error(message: "can't find peer: \(nickname)")
|
return .error(message: "can't find peer: \(nickname)")
|
||||||
}
|
}
|
||||||
@@ -294,15 +342,15 @@ final class CommandProcessor {
|
|||||||
peerNickname: nickname
|
peerNickname: nickname
|
||||||
)
|
)
|
||||||
|
|
||||||
chatViewModel?.toggleFavorite(peerID: peerID)
|
contextProvider?.toggleFavorite(peerID: peerID)
|
||||||
chatViewModel?.sendFavoriteNotification(to: peerID, isFavorite: true)
|
contextProvider?.sendFavoriteNotification(to: peerID, isFavorite: true)
|
||||||
|
|
||||||
return .success(message: "added \(nickname) to favorites")
|
return .success(message: "added \(nickname) to favorites")
|
||||||
} else {
|
} else {
|
||||||
FavoritesPersistenceService.shared.removeFavorite(peerNoisePublicKey: noisePublicKey)
|
FavoritesPersistenceService.shared.removeFavorite(peerNoisePublicKey: noisePublicKey)
|
||||||
|
|
||||||
chatViewModel?.toggleFavorite(peerID: peerID)
|
contextProvider?.toggleFavorite(peerID: peerID)
|
||||||
chatViewModel?.sendFavoriteNotification(to: peerID, isFavorite: false)
|
contextProvider?.sendFavoriteNotification(to: peerID, isFavorite: false)
|
||||||
|
|
||||||
return .success(message: "removed \(nickname) from favorites")
|
return .success(message: "removed \(nickname) from favorites")
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ final class FavoritesPersistenceService: ObservableObject {
|
|||||||
|
|
||||||
private static let storageKey = "chat.bitchat.favorites"
|
private static let storageKey = "chat.bitchat.favorites"
|
||||||
private static let keychainService = "chat.bitchat.favorites"
|
private static let keychainService = "chat.bitchat.favorites"
|
||||||
private let keychain: KeychainHelperProtocol
|
private let keychain: KeychainManagerProtocol
|
||||||
|
|
||||||
@Published private(set) var favorites: [Data: FavoriteRelationship] = [:] // Noise pubkey -> relationship
|
@Published private(set) var favorites: [Data: FavoriteRelationship] = [:] // Noise pubkey -> relationship
|
||||||
@Published private(set) var mutualFavorites: Set<Data> = []
|
@Published private(set) var mutualFavorites: Set<Data> = []
|
||||||
@@ -36,7 +36,7 @@ final class FavoritesPersistenceService: ObservableObject {
|
|||||||
|
|
||||||
static let shared = FavoritesPersistenceService()
|
static let shared = FavoritesPersistenceService()
|
||||||
|
|
||||||
init(keychain: KeychainHelperProtocol = KeychainHelper()) {
|
init(keychain: KeychainManagerProtocol = KeychainManager()) {
|
||||||
self.keychain = keychain
|
self.keychain = keychain
|
||||||
loadFavorites()
|
loadFavorites()
|
||||||
|
|
||||||
|
|||||||
@@ -1,219 +0,0 @@
|
|||||||
import Foundation
|
|
||||||
import Combine
|
|
||||||
#if os(iOS) || os(macOS)
|
|
||||||
import CoreLocation
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/// Stores a user-maintained list of bookmarked geohash channels.
|
|
||||||
/// - Persistence: UserDefaults (JSON string array)
|
|
||||||
/// - Semantics: geohashes are normalized to lowercase base32 and de-duplicated
|
|
||||||
final class GeohashBookmarksStore: ObservableObject {
|
|
||||||
static let shared = GeohashBookmarksStore()
|
|
||||||
|
|
||||||
@Published private(set) var bookmarks: [String] = []
|
|
||||||
@Published private(set) var bookmarkNames: [String: String] = [:] // geohash -> friendly name
|
|
||||||
|
|
||||||
private let storeKey = "locationChannel.bookmarks"
|
|
||||||
private let namesStoreKey = "locationChannel.bookmarkNames"
|
|
||||||
private var membership: Set<String> = []
|
|
||||||
#if os(iOS) || os(macOS)
|
|
||||||
private let geocoder = CLGeocoder()
|
|
||||||
private var resolving: Set<String> = []
|
|
||||||
#endif
|
|
||||||
|
|
||||||
private let storage: UserDefaults
|
|
||||||
|
|
||||||
init(storage: UserDefaults = .standard) {
|
|
||||||
self.storage = storage
|
|
||||||
load()
|
|
||||||
}
|
|
||||||
|
|
||||||
// MARK: - Public API
|
|
||||||
func isBookmarked(_ geohash: String) -> Bool {
|
|
||||||
return membership.contains(Self.normalize(geohash))
|
|
||||||
}
|
|
||||||
|
|
||||||
func toggle(_ geohash: String) {
|
|
||||||
let gh = Self.normalize(geohash)
|
|
||||||
if membership.contains(gh) {
|
|
||||||
remove(gh)
|
|
||||||
} else {
|
|
||||||
add(gh)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func add(_ geohash: String) {
|
|
||||||
let gh = Self.normalize(geohash)
|
|
||||||
guard !gh.isEmpty else { return }
|
|
||||||
guard !membership.contains(gh) else { return }
|
|
||||||
bookmarks.insert(gh, at: 0)
|
|
||||||
membership.insert(gh)
|
|
||||||
persist()
|
|
||||||
// Resolve and persist a friendly name once when added
|
|
||||||
resolveNameIfNeeded(for: gh)
|
|
||||||
}
|
|
||||||
|
|
||||||
func remove(_ geohash: String) {
|
|
||||||
let gh = Self.normalize(geohash)
|
|
||||||
guard membership.contains(gh) else { return }
|
|
||||||
if let idx = bookmarks.firstIndex(of: gh) { bookmarks.remove(at: idx) }
|
|
||||||
membership.remove(gh)
|
|
||||||
// Clean up stored name to avoid stale cache growth
|
|
||||||
if bookmarkNames.removeValue(forKey: gh) != nil {
|
|
||||||
persistNames()
|
|
||||||
}
|
|
||||||
persist()
|
|
||||||
}
|
|
||||||
|
|
||||||
// MARK: - Persistence
|
|
||||||
private func load() {
|
|
||||||
guard let data = storage.data(forKey: storeKey) else { return }
|
|
||||||
if let arr = try? JSONDecoder().decode([String].self, from: data) {
|
|
||||||
// Sanitize, normalize, dedupe while preserving order (first occurrence wins)
|
|
||||||
var seen = Set<String>()
|
|
||||||
var list: [String] = []
|
|
||||||
for raw in arr {
|
|
||||||
let gh = Self.normalize(raw)
|
|
||||||
guard !gh.isEmpty else { continue }
|
|
||||||
if !seen.contains(gh) {
|
|
||||||
seen.insert(gh)
|
|
||||||
list.append(gh)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
bookmarks = list
|
|
||||||
membership = seen
|
|
||||||
}
|
|
||||||
// Load any saved names
|
|
||||||
if let namesData = storage.data(forKey: namesStoreKey),
|
|
||||||
let dict = try? JSONDecoder().decode([String: String].self, from: namesData) {
|
|
||||||
bookmarkNames = dict
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private func persist() {
|
|
||||||
if let data = try? JSONEncoder().encode(bookmarks) {
|
|
||||||
storage.set(data, forKey: storeKey)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private func persistNames() {
|
|
||||||
if let data = try? JSONEncoder().encode(bookmarkNames) {
|
|
||||||
storage.set(data, forKey: namesStoreKey)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// MARK: - Helpers
|
|
||||||
private static func normalize(_ s: String) -> String {
|
|
||||||
let allowed = Set("0123456789bcdefghjkmnpqrstuvwxyz")
|
|
||||||
return s
|
|
||||||
.trimmingCharacters(in: .whitespacesAndNewlines)
|
|
||||||
.lowercased()
|
|
||||||
.replacingOccurrences(of: "#", with: "")
|
|
||||||
.filter { allowed.contains($0) }
|
|
||||||
}
|
|
||||||
|
|
||||||
// MARK: - Name Resolution
|
|
||||||
/// Attempt to resolve and persist a friendly place name for a bookmarked geohash.
|
|
||||||
func resolveNameIfNeeded(for geohash: String) {
|
|
||||||
let gh = Self.normalize(geohash)
|
|
||||||
guard !gh.isEmpty else { return }
|
|
||||||
if bookmarkNames[gh] != nil { return }
|
|
||||||
#if os(iOS) || os(macOS)
|
|
||||||
if resolving.contains(gh) { return }
|
|
||||||
resolving.insert(gh)
|
|
||||||
// For very coarse geohashes, sample multiple points to capture multiple admin areas
|
|
||||||
if gh.count <= 2 {
|
|
||||||
let b = Geohash.decodeBounds(gh)
|
|
||||||
let pts: [CLLocation] = [
|
|
||||||
CLLocation(latitude: (b.latMin + b.latMax) / 2, longitude: (b.lonMin + b.lonMax) / 2), // center
|
|
||||||
CLLocation(latitude: b.latMin, longitude: b.lonMin),
|
|
||||||
CLLocation(latitude: b.latMin, longitude: b.lonMax),
|
|
||||||
CLLocation(latitude: b.latMax, longitude: b.lonMin),
|
|
||||||
CLLocation(latitude: b.latMax, longitude: b.lonMax)
|
|
||||||
]
|
|
||||||
resolveCompositeAdminName(geohash: gh, points: pts)
|
|
||||||
} else {
|
|
||||||
let center = Geohash.decodeCenter(gh)
|
|
||||||
let loc = CLLocation(latitude: center.lat, longitude: center.lon)
|
|
||||||
geocoder.reverseGeocodeLocation(loc) { [weak self] placemarks, _ in
|
|
||||||
guard let self = self else { return }
|
|
||||||
defer { self.resolving.remove(gh) }
|
|
||||||
if let pm = placemarks?.first {
|
|
||||||
let name = Self.nameForGeohashLength(gh.count, from: pm)
|
|
||||||
if let name = name, !name.isEmpty {
|
|
||||||
DispatchQueue.main.async {
|
|
||||||
self.bookmarkNames[gh] = name
|
|
||||||
self.persistNames()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
#if os(iOS) || os(macOS)
|
|
||||||
private func resolveCompositeAdminName(geohash gh: String, points: [CLLocation]) {
|
|
||||||
var uniqueAdmins = OrderedSet<String>()
|
|
||||||
var idx = 0
|
|
||||||
func step() {
|
|
||||||
if idx >= points.count {
|
|
||||||
// Compose up to 2 names joined by ' and '
|
|
||||||
let finalName: String? = {
|
|
||||||
let names = uniqueAdmins.array
|
|
||||||
if names.count >= 2 { return names[0] + " and " + names[1] }
|
|
||||||
return names.first
|
|
||||||
}()
|
|
||||||
if let finalName = finalName, !finalName.isEmpty {
|
|
||||||
DispatchQueue.main.async {
|
|
||||||
self.bookmarkNames[gh] = finalName
|
|
||||||
self.persistNames()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
self.resolving.remove(gh)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
let loc = points[idx]
|
|
||||||
idx += 1
|
|
||||||
geocoder.reverseGeocodeLocation(loc) { [weak self] placemarks, _ in
|
|
||||||
guard self != nil else { return }
|
|
||||||
if let pm = placemarks?.first {
|
|
||||||
if let admin = pm.administrativeArea, !admin.isEmpty {
|
|
||||||
uniqueAdmins.insert(admin)
|
|
||||||
} else if let country = pm.country, !country.isEmpty {
|
|
||||||
uniqueAdmins.insert(country)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// Proceed to next point
|
|
||||||
step()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
step()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Minimal ordered-set for stable joining
|
|
||||||
private struct OrderedSet<Element: Hashable> {
|
|
||||||
private var set: Set<Element> = []
|
|
||||||
private(set) var array: [Element] = []
|
|
||||||
mutating func insert(_ element: Element) {
|
|
||||||
if set.insert(element).inserted { array.append(element) }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private static func nameForGeohashLength(_ len: Int, from pm: CLPlacemark) -> String? {
|
|
||||||
switch len {
|
|
||||||
case 0...2:
|
|
||||||
// Prefer administrative area if available at this coarse level
|
|
||||||
return pm.administrativeArea ?? pm.country
|
|
||||||
case 3...4:
|
|
||||||
return pm.administrativeArea ?? pm.subAdministrativeArea ?? pm.country
|
|
||||||
case 5:
|
|
||||||
return pm.locality ?? pm.subAdministrativeArea ?? pm.administrativeArea
|
|
||||||
case 6...7:
|
|
||||||
return pm.subLocality ?? pm.locality ?? pm.administrativeArea
|
|
||||||
default:
|
|
||||||
return pm.subLocality ?? pm.locality ?? pm.administrativeArea ?? pm.country
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,162 @@
|
|||||||
|
//
|
||||||
|
// GeohashParticipantTracker.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Tracks participants in geohash-based location channels.
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
/// Represents a participant in a geohash channel
|
||||||
|
public struct GeoPerson: Identifiable, Equatable, Sendable {
|
||||||
|
public let id: String // pubkey hex (lowercased)
|
||||||
|
public let displayName: String
|
||||||
|
public let lastSeen: Date
|
||||||
|
|
||||||
|
public init(id: String, displayName: String, lastSeen: Date) {
|
||||||
|
self.id = id
|
||||||
|
self.displayName = displayName
|
||||||
|
self.lastSeen = lastSeen
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Protocol for resolving display names and checking block status
|
||||||
|
@MainActor
|
||||||
|
public protocol GeohashParticipantContext: AnyObject {
|
||||||
|
/// Returns display name for a Nostr pubkey (e.g., "alice#a1b2" or "anon#c3d4")
|
||||||
|
func displayNameForPubkey(_ pubkeyHex: String) -> String
|
||||||
|
/// Returns true if the pubkey is blocked
|
||||||
|
func isBlocked(_ pubkeyHexLowercased: String) -> Bool
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Tracks participants across multiple geohash channels
|
||||||
|
@MainActor
|
||||||
|
public final class GeohashParticipantTracker: ObservableObject {
|
||||||
|
|
||||||
|
/// Activity cutoff duration (defaults to 5 minutes)
|
||||||
|
public let activityCutoff: TimeInterval
|
||||||
|
|
||||||
|
/// Per-geohash participant map: [geohash: [pubkeyHex: lastSeen]]
|
||||||
|
private var participants: [String: [String: Date]] = [:]
|
||||||
|
|
||||||
|
/// Currently visible people for the active geohash
|
||||||
|
@Published public private(set) var visiblePeople: [GeoPerson] = []
|
||||||
|
|
||||||
|
/// The currently active geohash (if any)
|
||||||
|
private var activeGeohash: String?
|
||||||
|
|
||||||
|
/// Context for display name resolution and block checking
|
||||||
|
private weak var context: GeohashParticipantContext?
|
||||||
|
|
||||||
|
/// Timer for periodic refresh
|
||||||
|
private var refreshTimer: Timer?
|
||||||
|
|
||||||
|
public init(activityCutoff: TimeInterval = -300) { // default 5 minutes
|
||||||
|
self.activityCutoff = activityCutoff
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Configure with a context provider
|
||||||
|
public func configure(context: GeohashParticipantContext) {
|
||||||
|
self.context = context
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Set the currently active geohash
|
||||||
|
public func setActiveGeohash(_ geohash: String?) {
|
||||||
|
activeGeohash = geohash
|
||||||
|
if geohash == nil {
|
||||||
|
visiblePeople = []
|
||||||
|
} else {
|
||||||
|
refresh()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Record activity from a participant in the current active geohash
|
||||||
|
public func recordParticipant(pubkeyHex: String) {
|
||||||
|
guard let gh = activeGeohash else { return }
|
||||||
|
recordParticipant(pubkeyHex: pubkeyHex, geohash: gh)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Record activity from a participant in a specific geohash
|
||||||
|
public func recordParticipant(pubkeyHex: String, geohash: String) {
|
||||||
|
let key = pubkeyHex.lowercased()
|
||||||
|
var map = participants[geohash] ?? [:]
|
||||||
|
map[key] = Date()
|
||||||
|
participants[geohash] = map
|
||||||
|
|
||||||
|
// Always notify observers that state has changed so counts in UI update
|
||||||
|
objectWillChange.send()
|
||||||
|
|
||||||
|
// Only refresh visible list if this geohash is currently active
|
||||||
|
if activeGeohash == geohash {
|
||||||
|
refresh()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Remove a participant from all geohashes (used when blocking)
|
||||||
|
public func removeParticipant(pubkeyHex: String) {
|
||||||
|
let key = pubkeyHex.lowercased()
|
||||||
|
for (gh, var map) in participants {
|
||||||
|
map.removeValue(forKey: key)
|
||||||
|
participants[gh] = map
|
||||||
|
}
|
||||||
|
refresh()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get participant count for a specific geohash
|
||||||
|
public func participantCount(for geohash: String) -> Int {
|
||||||
|
let cutoff = Date().addingTimeInterval(activityCutoff)
|
||||||
|
let map = participants[geohash] ?? [:]
|
||||||
|
return map.values.filter { $0 >= cutoff }.count
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the visible people list for the active geohash (read-only query)
|
||||||
|
public func getVisiblePeople() -> [GeoPerson] {
|
||||||
|
guard let gh = activeGeohash, let context = context else { return [] }
|
||||||
|
let cutoff = Date().addingTimeInterval(activityCutoff)
|
||||||
|
let map = (participants[gh] ?? [:])
|
||||||
|
.filter { $0.value >= cutoff }
|
||||||
|
.filter { !context.isBlocked($0.key) }
|
||||||
|
|
||||||
|
return map
|
||||||
|
.map { (pub, seen) in
|
||||||
|
GeoPerson(id: pub, displayName: context.displayNameForPubkey(pub), lastSeen: seen)
|
||||||
|
}
|
||||||
|
.sorted { $0.lastSeen > $1.lastSeen }
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Refresh the visible people list
|
||||||
|
public func refresh() {
|
||||||
|
visiblePeople = getVisiblePeople()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Start the periodic refresh timer
|
||||||
|
public func startRefreshTimer(interval: TimeInterval = 30.0) {
|
||||||
|
stopRefreshTimer()
|
||||||
|
refreshTimer = Timer.scheduledTimer(withTimeInterval: interval, repeats: true) { [weak self] _ in
|
||||||
|
Task { @MainActor in
|
||||||
|
self?.refresh()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Stop the periodic refresh timer
|
||||||
|
public func stopRefreshTimer() {
|
||||||
|
refreshTimer?.invalidate()
|
||||||
|
refreshTimer = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Clear all participant data
|
||||||
|
public func clear() {
|
||||||
|
participants.removeAll()
|
||||||
|
visiblePeople = []
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Clear participant data for a specific geohash
|
||||||
|
public func clear(geohash: String) {
|
||||||
|
participants.removeValue(forKey: geohash)
|
||||||
|
if activeGeohash == geohash {
|
||||||
|
visiblePeople = []
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,171 @@
|
|||||||
|
//
|
||||||
|
// GeohashPresenceService.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Manages the broadcasting of ephemeral presence heartbeats (Kind 20001)
|
||||||
|
// to geohash location channels.
|
||||||
|
//
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
import Combine
|
||||||
|
import BitLogger
|
||||||
|
import Tor
|
||||||
|
|
||||||
|
/// Service that coordinates the broadcasting of presence heartbeats.
|
||||||
|
///
|
||||||
|
/// Behavior:
|
||||||
|
/// - Monitors location changes via LocationStateManager
|
||||||
|
/// - Broadcasts Kind 20001 events to low-precision geohash channels
|
||||||
|
/// - Uses randomized timing (40-80s loop) and decorrelated bursts
|
||||||
|
/// - Respects privacy by NOT broadcasting to Neighborhood/Block/Building levels
|
||||||
|
@MainActor
|
||||||
|
final class GeohashPresenceService: ObservableObject {
|
||||||
|
static let shared = GeohashPresenceService()
|
||||||
|
|
||||||
|
private var subscriptions = Set<AnyCancellable>()
|
||||||
|
private var heartbeatTimer: Timer?
|
||||||
|
private let idBridge = NostrIdentityBridge()
|
||||||
|
|
||||||
|
// MARK: - Constants
|
||||||
|
|
||||||
|
// Loop interval range in seconds
|
||||||
|
private let loopMinInterval: TimeInterval = 40.0
|
||||||
|
private let loopMaxInterval: TimeInterval = 80.0
|
||||||
|
|
||||||
|
// Per-broadcast decorrelation delay range in seconds
|
||||||
|
private let burstMinDelay: TimeInterval = 2.0
|
||||||
|
private let burstMaxDelay: TimeInterval = 5.0
|
||||||
|
|
||||||
|
// Privacy: Only broadcast to these levels
|
||||||
|
private let allowedPrecisions: Set<Int> = [
|
||||||
|
GeohashChannelLevel.region.precision, // 2
|
||||||
|
GeohashChannelLevel.province.precision, // 4
|
||||||
|
GeohashChannelLevel.city.precision // 5
|
||||||
|
]
|
||||||
|
|
||||||
|
private init() {
|
||||||
|
setupObservers()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Start the service (safe to call multiple times)
|
||||||
|
func start() {
|
||||||
|
SecureLogger.info("Presence: service starting...", category: .session)
|
||||||
|
scheduleNextHeartbeat()
|
||||||
|
}
|
||||||
|
|
||||||
|
private func setupObservers() {
|
||||||
|
// Monitor location channel changes
|
||||||
|
LocationStateManager.shared.$availableChannels
|
||||||
|
.dropFirst()
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.handleLocationChange()
|
||||||
|
}
|
||||||
|
.store(in: &subscriptions)
|
||||||
|
|
||||||
|
// Monitor Tor readiness to kick off heartbeat if it was stalled
|
||||||
|
NotificationCenter.default.publisher(for: .TorDidBecomeReady)
|
||||||
|
.sink { [weak self] _ in
|
||||||
|
self?.handleConnectivityChange()
|
||||||
|
}
|
||||||
|
.store(in: &subscriptions)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func handleLocationChange() {
|
||||||
|
// When location changes, we trigger an immediate (but slightly delayed) heartbeat
|
||||||
|
// to announce presence in the new zone, then reset the loop.
|
||||||
|
SecureLogger.debug("Presence: location changed, scheduling update", category: .session)
|
||||||
|
heartbeatTimer?.invalidate()
|
||||||
|
|
||||||
|
// Small delay to allow location state to settle
|
||||||
|
heartbeatTimer = Timer.scheduledTimer(withTimeInterval: 5.0, repeats: false) { [weak self] _ in
|
||||||
|
Task { @MainActor [weak self] in
|
||||||
|
self?.performHeartbeat()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func handleConnectivityChange() {
|
||||||
|
SecureLogger.debug("Presence: connectivity restored, triggering heartbeat", category: .session)
|
||||||
|
// If we were waiting for network, do it now
|
||||||
|
if heartbeatTimer == nil || !heartbeatTimer!.isValid {
|
||||||
|
scheduleNextHeartbeat()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func scheduleNextHeartbeat() {
|
||||||
|
heartbeatTimer?.invalidate()
|
||||||
|
let interval = TimeInterval.random(in: loopMinInterval...loopMaxInterval)
|
||||||
|
heartbeatTimer = Timer.scheduledTimer(withTimeInterval: interval, repeats: false) { [weak self] _ in
|
||||||
|
Task { @MainActor [weak self] in
|
||||||
|
self?.performHeartbeat()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func performHeartbeat() {
|
||||||
|
// Always schedule next loop first ensures continuity even if this one fails/skips
|
||||||
|
defer { scheduleNextHeartbeat() }
|
||||||
|
|
||||||
|
// 1. Check preconditions
|
||||||
|
guard TorManager.shared.isReady else {
|
||||||
|
SecureLogger.debug("Presence: skipping heartbeat (Tor not ready)", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// App must be active (or at least we shouldn't broadcast if in background, usually)
|
||||||
|
if !TorManager.shared.isForeground() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Get channels
|
||||||
|
let channels = LocationStateManager.shared.availableChannels
|
||||||
|
guard !channels.isEmpty else { return }
|
||||||
|
|
||||||
|
// 3. Filter and broadcast
|
||||||
|
// We use Task + sleep for decorrelation to allow the main runloop to proceed
|
||||||
|
for channel in channels {
|
||||||
|
// Check privacy restriction
|
||||||
|
if !self.allowedPrecisions.contains(channel.geohash.count) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Launch independent task for each channel's delay
|
||||||
|
Task { @MainActor in
|
||||||
|
// Random delay for decorrelation
|
||||||
|
let delay = TimeInterval.random(in: self.burstMinDelay...self.burstMaxDelay)
|
||||||
|
let nanoseconds = UInt64(delay * 1_000_000_000)
|
||||||
|
try? await Task.sleep(nanoseconds: nanoseconds)
|
||||||
|
|
||||||
|
self.broadcastPresence(for: channel.geohash)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func broadcastPresence(for geohash: String) {
|
||||||
|
do {
|
||||||
|
guard let identity = try? idBridge.deriveIdentity(forGeohash: geohash) else {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let event = try NostrProtocol.createGeohashPresenceEvent(
|
||||||
|
geohash: geohash,
|
||||||
|
senderIdentity: identity
|
||||||
|
)
|
||||||
|
|
||||||
|
// Send via RelayManager
|
||||||
|
let targetRelays = GeoRelayDirectory.shared.closestRelays(
|
||||||
|
toGeohash: geohash,
|
||||||
|
count: TransportConfig.nostrGeoRelayCount
|
||||||
|
)
|
||||||
|
|
||||||
|
if !targetRelays.isEmpty {
|
||||||
|
NostrRelayManager.shared.sendEvent(event, to: targetRelays)
|
||||||
|
SecureLogger.debug("Presence: sent heartbeat for \(geohash) (pub=\(identity.publicKeyHex.prefix(6))...)", category: .session)
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
SecureLogger.error("Presence: failed to create event for \(geohash): \(error)", category: .session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -10,6 +10,47 @@ import BitLogger
|
|||||||
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 {
|
protocol KeychainManagerProtocol {
|
||||||
func saveIdentityKey(_ keyData: Data, forKey key: String) -> Bool
|
func saveIdentityKey(_ keyData: Data, forKey key: String) -> Bool
|
||||||
func getIdentityKey(forKey key: String) -> Data?
|
func getIdentityKey(forKey key: String) -> Data?
|
||||||
@@ -20,6 +61,20 @@ protocol KeychainManagerProtocol {
|
|||||||
func secureClear(_ string: inout String)
|
func secureClear(_ string: inout String)
|
||||||
|
|
||||||
func verifyIdentityKeyExists() -> Bool
|
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 {
|
||||||
@@ -47,6 +102,180 @@ final class KeychainManager: KeychainManagerProtocol {
|
|||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// MARK: - BCH-01-009: Methods with Proper Error Classification
|
||||||
|
|
||||||
|
/// Get identity key with detailed result for proper error handling
|
||||||
|
/// Distinguishes between missing keys (expected) and critical failures
|
||||||
|
func getIdentityKeyWithResult(forKey key: String) -> KeychainReadResult {
|
||||||
|
let fullKey = "identity_\(key)"
|
||||||
|
return retrieveDataWithResult(forKey: fullKey)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Save identity key with detailed result and retry logic for transient errors
|
||||||
|
func saveIdentityKeyWithResult(_ keyData: Data, forKey key: String) -> KeychainSaveResult {
|
||||||
|
let fullKey = "identity_\(key)"
|
||||||
|
return saveDataWithResult(keyData, forKey: fullKey)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Internal method to save data with detailed result and retry for transient errors
|
||||||
|
private func saveDataWithResult(_ data: Data, forKey key: String, retryCount: Int = 2) -> KeychainSaveResult {
|
||||||
|
// Delete any existing item first to ensure clean state
|
||||||
|
_ = delete(forKey: key)
|
||||||
|
|
||||||
|
// Build base query
|
||||||
|
var base: [String: Any] = [
|
||||||
|
kSecClass as String: kSecClassGenericPassword,
|
||||||
|
kSecAttrAccount as String: key,
|
||||||
|
kSecValueData as String: data,
|
||||||
|
kSecAttrService as String: service,
|
||||||
|
kSecAttrAccessible as String: kSecAttrAccessibleWhenUnlocked,
|
||||||
|
kSecAttrLabel as String: "bitchat-\(key)"
|
||||||
|
]
|
||||||
|
#if os(macOS)
|
||||||
|
base[kSecAttrSynchronizable as String] = false
|
||||||
|
#endif
|
||||||
|
|
||||||
|
func attempt(addAccessGroup: Bool) -> OSStatus {
|
||||||
|
var query = base
|
||||||
|
if addAccessGroup { query[kSecAttrAccessGroup as String] = appGroup }
|
||||||
|
return SecItemAdd(query as CFDictionary, nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
#if os(iOS)
|
||||||
|
var status = attempt(addAccessGroup: true)
|
||||||
|
if status == -34018 { // Missing entitlement, retry without access group
|
||||||
|
status = attempt(addAccessGroup: false)
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
let status = attempt(addAccessGroup: false)
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Classify the result
|
||||||
|
let result = classifySaveStatus(status)
|
||||||
|
|
||||||
|
// Log all outcomes consistently
|
||||||
|
switch result {
|
||||||
|
case .success:
|
||||||
|
SecureLogger.debug("Keychain save succeeded for key: \(key)", category: .keychain)
|
||||||
|
case .duplicateItem:
|
||||||
|
SecureLogger.warning("Keychain save found duplicate for key: \(key)", category: .keychain)
|
||||||
|
case .accessDenied:
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: Int(status)),
|
||||||
|
context: "Keychain access denied for key: \(key)", category: .keychain)
|
||||||
|
case .deviceLocked:
|
||||||
|
SecureLogger.warning("Device locked during keychain save for key: \(key)", category: .keychain)
|
||||||
|
case .storageFull:
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: Int(status)),
|
||||||
|
context: "Keychain storage full for key: \(key)", category: .keychain)
|
||||||
|
case .otherError(let code):
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: Int(code)),
|
||||||
|
context: "Keychain save failed for key: \(key)", category: .keychain)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Retry transient errors with exponential backoff
|
||||||
|
if result.isRecoverableError && retryCount > 0 {
|
||||||
|
let delayMs = UInt32((3 - retryCount) * 100) // 100ms, 200ms backoff
|
||||||
|
usleep(delayMs * 1000)
|
||||||
|
SecureLogger.debug("Retrying keychain save for key: \(key), attempts remaining: \(retryCount)", category: .keychain)
|
||||||
|
return saveDataWithResult(data, forKey: key, retryCount: retryCount - 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Internal method to retrieve data with detailed result
|
||||||
|
private func retrieveDataWithResult(forKey key: String) -> KeychainReadResult {
|
||||||
|
let base: [String: Any] = [
|
||||||
|
kSecClass as String: kSecClassGenericPassword,
|
||||||
|
kSecAttrAccount as String: key,
|
||||||
|
kSecAttrService as String: service,
|
||||||
|
kSecReturnData as String: true,
|
||||||
|
kSecMatchLimit as String: kSecMatchLimitOne
|
||||||
|
]
|
||||||
|
|
||||||
|
var result: AnyObject?
|
||||||
|
func attempt(withAccessGroup: Bool) -> OSStatus {
|
||||||
|
var q = base
|
||||||
|
if withAccessGroup { q[kSecAttrAccessGroup as String] = appGroup }
|
||||||
|
return SecItemCopyMatching(q as CFDictionary, &result)
|
||||||
|
}
|
||||||
|
|
||||||
|
#if os(iOS)
|
||||||
|
var status = attempt(withAccessGroup: true)
|
||||||
|
if status == -34018 { status = attempt(withAccessGroup: false) }
|
||||||
|
#else
|
||||||
|
let status = attempt(withAccessGroup: false)
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Classify the result
|
||||||
|
let readResult = classifyReadStatus(status, data: result as? Data)
|
||||||
|
|
||||||
|
// Log all outcomes consistently
|
||||||
|
switch readResult {
|
||||||
|
case .success:
|
||||||
|
SecureLogger.debug("Keychain read succeeded for key: \(key)", category: .keychain)
|
||||||
|
case .itemNotFound:
|
||||||
|
// Expected case - no logging needed for missing keys
|
||||||
|
break
|
||||||
|
case .accessDenied:
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: Int(status)),
|
||||||
|
context: "Keychain access denied for key: \(key)", category: .keychain)
|
||||||
|
case .deviceLocked:
|
||||||
|
SecureLogger.warning("Device locked during keychain read for key: \(key)", category: .keychain)
|
||||||
|
case .authenticationFailed:
|
||||||
|
SecureLogger.warning("Authentication failed for keychain read of key: \(key)", category: .keychain)
|
||||||
|
case .otherError(let code):
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: Int(code)),
|
||||||
|
context: "Keychain read failed for key: \(key)", category: .keychain)
|
||||||
|
}
|
||||||
|
|
||||||
|
return readResult
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Classify keychain read status into meaningful categories
|
||||||
|
private func classifyReadStatus(_ status: OSStatus, data: Data?) -> KeychainReadResult {
|
||||||
|
switch status {
|
||||||
|
case errSecSuccess:
|
||||||
|
if let data = data {
|
||||||
|
return .success(data)
|
||||||
|
}
|
||||||
|
return .otherError(status)
|
||||||
|
case errSecItemNotFound:
|
||||||
|
return .itemNotFound
|
||||||
|
case errSecInteractionNotAllowed:
|
||||||
|
// Device is locked or in a state that doesn't allow keychain access
|
||||||
|
return .deviceLocked
|
||||||
|
case errSecAuthFailed:
|
||||||
|
return .authenticationFailed
|
||||||
|
case -34018: // errSecMissingEntitlement
|
||||||
|
return .accessDenied
|
||||||
|
case errSecNotAvailable:
|
||||||
|
return .accessDenied
|
||||||
|
default:
|
||||||
|
return .otherError(status)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Classify keychain save status into meaningful categories
|
||||||
|
private func classifySaveStatus(_ status: OSStatus) -> KeychainSaveResult {
|
||||||
|
switch status {
|
||||||
|
case errSecSuccess:
|
||||||
|
return .success
|
||||||
|
case errSecDuplicateItem:
|
||||||
|
return .duplicateItem
|
||||||
|
case errSecInteractionNotAllowed:
|
||||||
|
return .deviceLocked
|
||||||
|
case -34018: // errSecMissingEntitlement
|
||||||
|
return .accessDenied
|
||||||
|
case errSecNotAvailable:
|
||||||
|
return .accessDenied
|
||||||
|
case errSecDiskFull:
|
||||||
|
return .storageFull
|
||||||
|
default:
|
||||||
|
return .otherError(status)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// MARK: - Generic Operations
|
// MARK: - Generic Operations
|
||||||
|
|
||||||
private func save(_ value: String, forKey key: String) -> Bool {
|
private func save(_ value: String, forKey key: String) -> Bool {
|
||||||
@@ -314,4 +543,49 @@ final class KeychainManager: KeychainManagerProtocol {
|
|||||||
let key = "identity_noiseStaticKey"
|
let key = "identity_noiseStaticKey"
|
||||||
return retrieveData(forKey: key) != nil
|
return retrieveData(forKey: key) != nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// MARK: - Generic Data Storage (consolidated from KeychainHelper)
|
||||||
|
|
||||||
|
/// Save data with a custom service name
|
||||||
|
func save(key: String, data: Data, service customService: String, accessible: CFString?) {
|
||||||
|
var query: [String: Any] = [
|
||||||
|
kSecClass as String: kSecClassGenericPassword,
|
||||||
|
kSecAttrService as String: customService,
|
||||||
|
kSecAttrAccount as String: key,
|
||||||
|
kSecValueData as String: data
|
||||||
|
]
|
||||||
|
if let accessible = accessible {
|
||||||
|
query[kSecAttrAccessible as String] = accessible
|
||||||
|
}
|
||||||
|
|
||||||
|
SecItemDelete(query as CFDictionary)
|
||||||
|
SecItemAdd(query as CFDictionary, nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Load data from a custom service
|
||||||
|
func load(key: String, service customService: String) -> Data? {
|
||||||
|
let query: [String: Any] = [
|
||||||
|
kSecClass as String: kSecClassGenericPassword,
|
||||||
|
kSecAttrService as String: customService,
|
||||||
|
kSecAttrAccount as String: key,
|
||||||
|
kSecReturnData as String: true
|
||||||
|
]
|
||||||
|
|
||||||
|
var result: AnyObject?
|
||||||
|
let status = SecItemCopyMatching(query as CFDictionary, &result)
|
||||||
|
|
||||||
|
guard status == errSecSuccess else { return nil }
|
||||||
|
return result as? Data
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Delete data from a custom service
|
||||||
|
func delete(key: String, service customService: String) {
|
||||||
|
let query: [String: Any] = [
|
||||||
|
kSecClass as String: kSecClassGenericPassword,
|
||||||
|
kSecAttrService as String: customService,
|
||||||
|
kSecAttrAccount as String: key
|
||||||
|
]
|
||||||
|
|
||||||
|
SecItemDelete(query as CFDictionary)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,304 +0,0 @@
|
|||||||
import BitLogger
|
|
||||||
import Foundation
|
|
||||||
import Combine
|
|
||||||
|
|
||||||
#if os(iOS) || os(macOS)
|
|
||||||
import CoreLocation
|
|
||||||
|
|
||||||
/// Manages location permissions, one-shot location retrieval, and computing geohash channels.
|
|
||||||
/// Not main-actor isolated to satisfy CLLocationManagerDelegate in Swift 6; state updates hop to MainActor.
|
|
||||||
final class LocationChannelManager: NSObject, CLLocationManagerDelegate, ObservableObject {
|
|
||||||
static let shared = LocationChannelManager()
|
|
||||||
|
|
||||||
enum PermissionState: Equatable {
|
|
||||||
case notDetermined
|
|
||||||
case denied
|
|
||||||
case restricted
|
|
||||||
case authorized
|
|
||||||
}
|
|
||||||
|
|
||||||
private let cl = CLLocationManager()
|
|
||||||
private let geocoder = CLGeocoder()
|
|
||||||
private var lastLocation: CLLocation?
|
|
||||||
private var refreshTimer: Timer?
|
|
||||||
private let userDefaultsKey = "locationChannel.selected"
|
|
||||||
private let teleportedStoreKey = "locationChannel.teleportedSet"
|
|
||||||
private var isGeocoding: Bool = false
|
|
||||||
|
|
||||||
// Published state for UI bindings
|
|
||||||
@Published private(set) var permissionState: PermissionState = .notDetermined
|
|
||||||
@Published private(set) var availableChannels: [GeohashChannel] = []
|
|
||||||
@Published private(set) var selectedChannel: ChannelID = .mesh
|
|
||||||
// True when the current location channel was selected via manual teleport
|
|
||||||
@Published var teleported: Bool = false
|
|
||||||
@Published private(set) var locationNames: [GeohashChannelLevel: String] = [:]
|
|
||||||
|
|
||||||
// Persisted set of geohashes that were selected via teleport
|
|
||||||
private var teleportedSet: Set<String> = []
|
|
||||||
|
|
||||||
private override init() {
|
|
||||||
super.init()
|
|
||||||
cl.delegate = self
|
|
||||||
cl.desiredAccuracy = kCLLocationAccuracyHundredMeters
|
|
||||||
cl.distanceFilter = TransportConfig.locationDistanceFilterMeters // meters; we're not tracking continuously
|
|
||||||
// Load selection
|
|
||||||
if let data = UserDefaults.standard.data(forKey: userDefaultsKey),
|
|
||||||
let channel = try? JSONDecoder().decode(ChannelID.self, from: data) {
|
|
||||||
selectedChannel = channel
|
|
||||||
}
|
|
||||||
// Load persisted teleported set
|
|
||||||
if let data = UserDefaults.standard.data(forKey: teleportedStoreKey),
|
|
||||||
let arr = try? JSONDecoder().decode([String].self, from: data) {
|
|
||||||
teleportedSet = Set(arr)
|
|
||||||
}
|
|
||||||
// Do not eagerly mark teleported on startup; wait for location to compute regional set.
|
|
||||||
// This avoids showing teleported for in-region channels during cold start.
|
|
||||||
let status: CLAuthorizationStatus
|
|
||||||
if #available(iOS 14.0, macOS 11.0, *) {
|
|
||||||
status = cl.authorizationStatus
|
|
||||||
} else {
|
|
||||||
status = CLLocationManager.authorizationStatus()
|
|
||||||
}
|
|
||||||
updatePermissionState(from: status)
|
|
||||||
// If we don't have location authorization at startup, fall back to persisted teleport state
|
|
||||||
switch status {
|
|
||||||
case .authorizedAlways, .authorizedWhenInUse, .authorized:
|
|
||||||
break // will compute from location
|
|
||||||
default:
|
|
||||||
if case .location(let ch) = selectedChannel {
|
|
||||||
teleported = teleportedSet.contains(ch.geohash)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// MARK: - Public API
|
|
||||||
func enableLocationChannels() {
|
|
||||||
let status: CLAuthorizationStatus
|
|
||||||
if #available(iOS 14.0, macOS 11.0, *) {
|
|
||||||
status = cl.authorizationStatus
|
|
||||||
} else {
|
|
||||||
status = CLLocationManager.authorizationStatus()
|
|
||||||
}
|
|
||||||
switch status {
|
|
||||||
case .notDetermined:
|
|
||||||
cl.requestWhenInUseAuthorization()
|
|
||||||
case .restricted:
|
|
||||||
Task { @MainActor in self.permissionState = .restricted }
|
|
||||||
case .denied:
|
|
||||||
Task { @MainActor in self.permissionState = .denied }
|
|
||||||
case .authorizedAlways, .authorizedWhenInUse, .authorized:
|
|
||||||
Task { @MainActor in self.permissionState = .authorized }
|
|
||||||
requestOneShotLocation()
|
|
||||||
@unknown default:
|
|
||||||
Task { @MainActor in self.permissionState = .restricted }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func refreshChannels() {
|
|
||||||
if permissionState == .authorized {
|
|
||||||
requestOneShotLocation()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Begin continuous, distance-filtered updates while the channel sheet is visible.
|
|
||||||
/// Uses a 21m filter (configurable) to only refresh on meaningful movement.
|
|
||||||
func beginLiveRefresh(interval: TimeInterval = TransportConfig.locationLiveRefreshInterval) {
|
|
||||||
guard permissionState == .authorized else { return }
|
|
||||||
// Stop any previous polling timer
|
|
||||||
refreshTimer?.invalidate()
|
|
||||||
refreshTimer = nil
|
|
||||||
// Tighten accuracy and distance filter for live view
|
|
||||||
cl.desiredAccuracy = kCLLocationAccuracyNearestTenMeters
|
|
||||||
cl.distanceFilter = TransportConfig.locationDistanceFilterLiveMeters
|
|
||||||
// Start continuous updates
|
|
||||||
cl.startUpdatingLocation()
|
|
||||||
// Request an immediate fix to populate UI without waiting for movement
|
|
||||||
requestOneShotLocation()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Stop continuous refreshes when selector UI is dismissed.
|
|
||||||
func endLiveRefresh() {
|
|
||||||
refreshTimer?.invalidate()
|
|
||||||
refreshTimer = nil
|
|
||||||
cl.stopUpdatingLocation()
|
|
||||||
// Restore more relaxed defaults for background/idle state
|
|
||||||
cl.desiredAccuracy = kCLLocationAccuracyHundredMeters
|
|
||||||
cl.distanceFilter = TransportConfig.locationDistanceFilterMeters
|
|
||||||
}
|
|
||||||
|
|
||||||
func select(_ channel: ChannelID) {
|
|
||||||
Task { @MainActor in
|
|
||||||
self.selectedChannel = channel
|
|
||||||
if let data = try? JSONEncoder().encode(channel) {
|
|
||||||
UserDefaults.standard.set(data, forKey: self.userDefaultsKey)
|
|
||||||
}
|
|
||||||
// Update teleported flag based on persisted state for immediate UI behavior
|
|
||||||
switch channel {
|
|
||||||
case .mesh:
|
|
||||||
self.teleported = false
|
|
||||||
case .location(let ch):
|
|
||||||
// If this geohash is in our current regional set, do NOT mark teleported.
|
|
||||||
let inRegional = self.availableChannels.contains { $0.geohash == ch.geohash }
|
|
||||||
if inRegional {
|
|
||||||
self.teleported = false
|
|
||||||
// Clear persisted teleport for this geohash to keep future selections clean
|
|
||||||
if self.teleportedSet.contains(ch.geohash) {
|
|
||||||
self.teleportedSet.remove(ch.geohash)
|
|
||||||
if let data = try? JSONEncoder().encode(Array(self.teleportedSet)) {
|
|
||||||
UserDefaults.standard.set(data, forKey: self.teleportedStoreKey)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// Fall back to persisted mark (set by deep link or manual teleport)
|
|
||||||
self.teleported = self.teleportedSet.contains(ch.geohash)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Mark or unmark a geohash as teleported in persistence and update current flag if relevant
|
|
||||||
func markTeleported(for geohash: String, _ flag: Bool) {
|
|
||||||
if flag { teleportedSet.insert(geohash) } else { teleportedSet.remove(geohash) }
|
|
||||||
if let data = try? JSONEncoder().encode(Array(teleportedSet)) {
|
|
||||||
UserDefaults.standard.set(data, forKey: teleportedStoreKey)
|
|
||||||
}
|
|
||||||
if case .location(let ch) = selectedChannel, ch.geohash == geohash {
|
|
||||||
Task { @MainActor in self.teleported = flag }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// MARK: - CoreLocation
|
|
||||||
private func requestOneShotLocation() {
|
|
||||||
cl.requestLocation()
|
|
||||||
}
|
|
||||||
|
|
||||||
// iOS < 14
|
|
||||||
func locationManager(_ manager: CLLocationManager, didChangeAuthorization status: CLAuthorizationStatus) {
|
|
||||||
updatePermissionState(from: status)
|
|
||||||
if case .authorized = permissionState {
|
|
||||||
requestOneShotLocation()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// iOS 14+ / macOS 11+
|
|
||||||
@available(iOS 14.0, macOS 11.0, *)
|
|
||||||
func locationManagerDidChangeAuthorization(_ manager: CLLocationManager) {
|
|
||||||
updatePermissionState(from: manager.authorizationStatus)
|
|
||||||
if case .authorized = permissionState {
|
|
||||||
requestOneShotLocation()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func locationManager(_ manager: CLLocationManager, didUpdateLocations locations: [CLLocation]) {
|
|
||||||
guard let loc = locations.last else { return }
|
|
||||||
lastLocation = loc
|
|
||||||
computeChannels(from: loc.coordinate)
|
|
||||||
reverseGeocodeIfNeeded(location: loc)
|
|
||||||
}
|
|
||||||
|
|
||||||
func locationManager(_ manager: CLLocationManager, didFailWithError error: Error) {
|
|
||||||
// Surface as denied/restricted if relevant; otherwise keep previous state
|
|
||||||
SecureLogger.error("LocationChannelManager: location error: \(error.localizedDescription)", category: .session)
|
|
||||||
}
|
|
||||||
|
|
||||||
// MARK: - Helpers
|
|
||||||
private func updatePermissionState(from status: CLAuthorizationStatus) {
|
|
||||||
let newState: PermissionState
|
|
||||||
switch status {
|
|
||||||
case .notDetermined: newState = .notDetermined
|
|
||||||
case .restricted: newState = .restricted
|
|
||||||
case .denied: newState = .denied
|
|
||||||
case .authorizedAlways, .authorizedWhenInUse, .authorized: newState = .authorized
|
|
||||||
@unknown default: newState = .restricted
|
|
||||||
}
|
|
||||||
Task { @MainActor in self.permissionState = newState }
|
|
||||||
}
|
|
||||||
|
|
||||||
private func computeChannels(from coord: CLLocationCoordinate2D) {
|
|
||||||
let levels = GeohashChannelLevel.allCases
|
|
||||||
var result: [GeohashChannel] = []
|
|
||||||
for level in levels {
|
|
||||||
let gh = Geohash.encode(latitude: coord.latitude, longitude: coord.longitude, precision: level.precision)
|
|
||||||
result.append(GeohashChannel(level: level, geohash: gh))
|
|
||||||
}
|
|
||||||
Task { @MainActor in
|
|
||||||
self.availableChannels = result
|
|
||||||
// Recompute teleported status based on whether the selected geohash is in our regional set
|
|
||||||
switch self.selectedChannel {
|
|
||||||
case .mesh:
|
|
||||||
self.teleported = false
|
|
||||||
case .location(let ch):
|
|
||||||
// Membership check using freshly computed regional channels; avoids precision/rename drift
|
|
||||||
let inRegional = result.contains { $0.geohash == ch.geohash }
|
|
||||||
if inRegional {
|
|
||||||
self.teleported = false
|
|
||||||
// Clear persisted teleport flag if present
|
|
||||||
if self.teleportedSet.contains(ch.geohash) {
|
|
||||||
self.teleportedSet.remove(ch.geohash)
|
|
||||||
if let data = try? JSONEncoder().encode(Array(self.teleportedSet)) {
|
|
||||||
UserDefaults.standard.set(data, forKey: self.teleportedStoreKey)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
self.teleported = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private func reverseGeocodeIfNeeded(location: CLLocation) {
|
|
||||||
// Always cancel previous to keep latest fresh while user moves
|
|
||||||
geocoder.cancelGeocode()
|
|
||||||
isGeocoding = true
|
|
||||||
geocoder.reverseGeocodeLocation(location) { [weak self] placemarks, error in
|
|
||||||
guard let self = self else { return }
|
|
||||||
self.isGeocoding = false
|
|
||||||
if let pm = placemarks?.first {
|
|
||||||
let names = self.namesByLevel(from: pm)
|
|
||||||
Task { @MainActor in self.locationNames = names }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private func namesByLevel(from pm: CLPlacemark) -> [GeohashChannelLevel: String] {
|
|
||||||
var dict: [GeohashChannelLevel: String] = [:]
|
|
||||||
// Region (country)
|
|
||||||
if let country = pm.country, !country.isEmpty {
|
|
||||||
dict[.region] = country
|
|
||||||
}
|
|
||||||
// Province (state/province or county)
|
|
||||||
if let admin = pm.administrativeArea, !admin.isEmpty {
|
|
||||||
dict[.province] = admin
|
|
||||||
} else if let subAdmin = pm.subAdministrativeArea, !subAdmin.isEmpty {
|
|
||||||
dict[.province] = subAdmin
|
|
||||||
}
|
|
||||||
// City (locality)
|
|
||||||
if let locality = pm.locality, !locality.isEmpty {
|
|
||||||
dict[.city] = locality
|
|
||||||
} else if let subAdmin = pm.subAdministrativeArea, !subAdmin.isEmpty {
|
|
||||||
dict[.city] = subAdmin
|
|
||||||
} else if let admin = pm.administrativeArea, !admin.isEmpty {
|
|
||||||
dict[.city] = admin
|
|
||||||
}
|
|
||||||
// Neighborhood
|
|
||||||
if let subLocality = pm.subLocality, !subLocality.isEmpty {
|
|
||||||
dict[.neighborhood] = subLocality
|
|
||||||
} else if let locality = pm.locality, !locality.isEmpty {
|
|
||||||
dict[.neighborhood] = locality
|
|
||||||
}
|
|
||||||
// Block: reuse neighborhood/locality granularity
|
|
||||||
if let subLocality = pm.subLocality, !subLocality.isEmpty {
|
|
||||||
dict[.block] = subLocality
|
|
||||||
} else if let locality = pm.locality, !locality.isEmpty {
|
|
||||||
dict[.block] = locality
|
|
||||||
}
|
|
||||||
// Building: prefer place name/street/venue when available
|
|
||||||
if let name = pm.name, !name.isEmpty {
|
|
||||||
dict[.building] = name
|
|
||||||
} else if let thoroughfare = pm.thoroughfare, !thoroughfare.isEmpty {
|
|
||||||
dict[.building] = thoroughfare
|
|
||||||
}
|
|
||||||
return dict
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
@@ -1,104 +0,0 @@
|
|||||||
import BitLogger
|
|
||||||
import Foundation
|
|
||||||
|
|
||||||
struct LocationNotesCounterDependencies {
|
|
||||||
typealias RelayLookup = @MainActor (_ geohash: String, _ count: Int) -> [String]
|
|
||||||
typealias Subscribe = @MainActor (_ filter: NostrFilter, _ id: String, _ relays: [String], _ handler: @escaping (NostrEvent) -> Void, _ onEOSE: (() -> Void)?) -> Void
|
|
||||||
typealias Unsubscribe = @MainActor (_ id: String) -> Void
|
|
||||||
|
|
||||||
var relayLookup: RelayLookup
|
|
||||||
var subscribe: Subscribe
|
|
||||||
var unsubscribe: Unsubscribe
|
|
||||||
|
|
||||||
static let live = LocationNotesCounterDependencies(
|
|
||||||
relayLookup: { geohash, count in
|
|
||||||
GeoRelayDirectory.shared.closestRelays(toGeohash: geohash, count: count)
|
|
||||||
},
|
|
||||||
subscribe: { filter, id, relays, handler, onEOSE in
|
|
||||||
NostrRelayManager.shared.subscribe(
|
|
||||||
filter: filter,
|
|
||||||
id: id,
|
|
||||||
relayUrls: relays,
|
|
||||||
handler: handler,
|
|
||||||
onEOSE: onEOSE
|
|
||||||
)
|
|
||||||
},
|
|
||||||
unsubscribe: { id in
|
|
||||||
NostrRelayManager.shared.unsubscribe(id: id)
|
|
||||||
}
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Lightweight background counter for location notes (kind 1) at building-level geohash (8 chars).
|
|
||||||
@MainActor
|
|
||||||
final class LocationNotesCounter: ObservableObject {
|
|
||||||
static let shared = LocationNotesCounter()
|
|
||||||
|
|
||||||
@Published private(set) var geohash: String? = nil
|
|
||||||
@Published private(set) var count: Int? = 0
|
|
||||||
@Published private(set) var initialLoadComplete: Bool = false
|
|
||||||
@Published private(set) var relayAvailable: Bool = true
|
|
||||||
|
|
||||||
private var subscriptionID: String? = nil
|
|
||||||
private var noteIDs = Set<String>()
|
|
||||||
private let dependencies: LocationNotesCounterDependencies
|
|
||||||
|
|
||||||
private init(dependencies: LocationNotesCounterDependencies = .live) {
|
|
||||||
self.dependencies = dependencies
|
|
||||||
}
|
|
||||||
|
|
||||||
init(testDependencies: LocationNotesCounterDependencies) {
|
|
||||||
self.dependencies = testDependencies
|
|
||||||
}
|
|
||||||
|
|
||||||
func subscribe(geohash gh: String) {
|
|
||||||
let norm = gh.lowercased()
|
|
||||||
if geohash == norm, subscriptionID != nil { return }
|
|
||||||
// Validate geohash (building-level precision: 8 chars)
|
|
||||||
guard Geohash.isValidBuildingGeohash(norm) else {
|
|
||||||
SecureLogger.warning("LocationNotesCounter: rejecting invalid geohash '\(norm)' (expected 8 valid base32 chars)", category: .session)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
// Unsubscribe previous without clearing count to avoid flicker
|
|
||||||
if let sub = subscriptionID { dependencies.unsubscribe(sub) }
|
|
||||||
subscriptionID = nil
|
|
||||||
geohash = norm
|
|
||||||
noteIDs.removeAll()
|
|
||||||
initialLoadComplete = false
|
|
||||||
relayAvailable = true
|
|
||||||
|
|
||||||
// Subscribe only to the building geohash (precision 8)
|
|
||||||
let subID = "locnotes-count-\(norm)-\(UUID().uuidString.prefix(6))"
|
|
||||||
let relays = dependencies.relayLookup(norm, TransportConfig.nostrGeoRelayCount)
|
|
||||||
guard !relays.isEmpty else {
|
|
||||||
relayAvailable = false
|
|
||||||
initialLoadComplete = true
|
|
||||||
count = 0
|
|
||||||
SecureLogger.warning("LocationNotesCounter: no geo relays for geohash=\(norm)", category: .session)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
subscriptionID = subID
|
|
||||||
let filter = NostrFilter.geohashNotes(norm, since: nil, limit: 200)
|
|
||||||
dependencies.subscribe(filter, subID, relays, { [weak self] event in
|
|
||||||
guard let self = self else { return }
|
|
||||||
guard event.kind == NostrProtocol.EventKind.textNote.rawValue else { return }
|
|
||||||
guard event.tags.contains(where: { $0.count >= 2 && $0[0].lowercased() == "g" && $0[1].lowercased() == norm }) else { return }
|
|
||||||
if !self.noteIDs.contains(event.id) {
|
|
||||||
self.noteIDs.insert(event.id)
|
|
||||||
self.count = self.noteIDs.count
|
|
||||||
}
|
|
||||||
}, { [weak self] in
|
|
||||||
self?.initialLoadComplete = true
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func cancel() {
|
|
||||||
if let sub = subscriptionID { dependencies.unsubscribe(sub) }
|
|
||||||
subscriptionID = nil
|
|
||||||
geohash = nil
|
|
||||||
count = 0
|
|
||||||
noteIDs.removeAll()
|
|
||||||
relayAvailable = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -163,14 +163,22 @@ final class LocationNotesManager: ObservableObject {
|
|||||||
|
|
||||||
subscriptionID = subID
|
subscriptionID = subID
|
||||||
initialLoadComplete = false
|
initialLoadComplete = false
|
||||||
// For persistent notes, allow relays to return recent history without an aggressive time cutoff
|
|
||||||
let filter = NostrFilter.geohashNotes(geohash, since: nil, limit: 200)
|
// Subscribe to center + 8 neighbors (± 1 grid)
|
||||||
|
let neighbors = Geohash.neighbors(of: geohash)
|
||||||
|
let allGeohashes = [geohash] + neighbors
|
||||||
|
let filter = NostrFilter.geohashNotes(allGeohashes, since: nil, limit: 200)
|
||||||
|
|
||||||
|
// Build a set of valid geohashes for tag matching (includes all 9 cells)
|
||||||
|
let validGeohashes = Set(allGeohashes.map { $0.lowercased() })
|
||||||
|
|
||||||
dependencies.subscribe(filter, subID, relays, { [weak self] event in
|
dependencies.subscribe(filter, subID, relays, { [weak self] event in
|
||||||
guard let self = self else { return }
|
guard let self = self else { return }
|
||||||
guard event.kind == NostrProtocol.EventKind.textNote.rawValue else { return }
|
guard event.kind == NostrProtocol.EventKind.textNote.rawValue else { return }
|
||||||
// Ensure matching tag
|
// Ensure matching tag - accept any of our 9 geohashes
|
||||||
guard event.tags.contains(where: { $0.count >= 2 && $0[0].lowercased() == "g" && $0[1].lowercased() == self.geohash }) else { return }
|
guard event.tags.contains(where: { tag in
|
||||||
|
tag.count >= 2 && tag[0].lowercased() == "g" && validGeohashes.contains(tag[1].lowercased())
|
||||||
|
}) else { return }
|
||||||
guard !self.noteIDs.contains(event.id) else { return }
|
guard !self.noteIDs.contains(event.id) else { return }
|
||||||
self.noteIDs.insert(event.id)
|
self.noteIDs.insert(event.id)
|
||||||
let nick = event.tags.first(where: { $0.first?.lowercased() == "n" && $0.count >= 2 })?.dropFirst().first
|
let nick = event.tags.first(where: { $0.first?.lowercased() == "n" && $0.count >= 2 })?.dropFirst().first
|
||||||
|
|||||||
@@ -0,0 +1,545 @@
|
|||||||
|
import BitLogger
|
||||||
|
import Foundation
|
||||||
|
import Combine
|
||||||
|
|
||||||
|
#if os(iOS) || os(macOS)
|
||||||
|
import CoreLocation
|
||||||
|
|
||||||
|
/// Unified manager for location-based channel state including:
|
||||||
|
/// - CoreLocation permissions and one-shot location retrieval
|
||||||
|
/// - Geohash channel computation from coordinates
|
||||||
|
/// - Channel selection and teleport state
|
||||||
|
/// - Bookmark persistence and friendly name resolution
|
||||||
|
///
|
||||||
|
/// Consolidates LocationChannelManager + GeohashBookmarksStore into a single source of truth.
|
||||||
|
final class LocationStateManager: NSObject, CLLocationManagerDelegate, ObservableObject {
|
||||||
|
static let shared = LocationStateManager()
|
||||||
|
|
||||||
|
// MARK: - Permission State
|
||||||
|
|
||||||
|
enum PermissionState: Equatable {
|
||||||
|
case notDetermined
|
||||||
|
case denied
|
||||||
|
case restricted
|
||||||
|
case authorized
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Private Properties (CoreLocation)
|
||||||
|
|
||||||
|
private let cl = CLLocationManager()
|
||||||
|
private let geocoder = CLGeocoder()
|
||||||
|
private var lastLocation: CLLocation?
|
||||||
|
private var refreshTimer: Timer?
|
||||||
|
private var isGeocoding: Bool = false
|
||||||
|
|
||||||
|
// MARK: - Persistence Keys
|
||||||
|
|
||||||
|
private let selectedChannelKey = "locationChannel.selected"
|
||||||
|
private let teleportedStoreKey = "locationChannel.teleportedSet"
|
||||||
|
private let bookmarksKey = "locationChannel.bookmarks"
|
||||||
|
private let bookmarkNamesKey = "locationChannel.bookmarkNames"
|
||||||
|
|
||||||
|
// MARK: - Published State (Channel)
|
||||||
|
|
||||||
|
@Published private(set) var permissionState: PermissionState = .notDetermined
|
||||||
|
@Published private(set) var availableChannels: [GeohashChannel] = []
|
||||||
|
@Published private(set) var selectedChannel: ChannelID = .mesh
|
||||||
|
@Published var teleported: Bool = false
|
||||||
|
@Published private(set) var locationNames: [GeohashChannelLevel: String] = [:]
|
||||||
|
|
||||||
|
// MARK: - Published State (Bookmarks)
|
||||||
|
|
||||||
|
@Published private(set) var bookmarks: [String] = []
|
||||||
|
@Published private(set) var bookmarkNames: [String: String] = [:]
|
||||||
|
|
||||||
|
// MARK: - Private State
|
||||||
|
|
||||||
|
private var teleportedSet: Set<String> = []
|
||||||
|
private var bookmarkMembership: Set<String> = []
|
||||||
|
private var resolvingNames: Set<String> = []
|
||||||
|
private let storage: UserDefaults
|
||||||
|
|
||||||
|
/// Returns true if running in test environment
|
||||||
|
private static var isRunningTests: Bool {
|
||||||
|
let env = ProcessInfo.processInfo.environment
|
||||||
|
return NSClassFromString("XCTestCase") != nil ||
|
||||||
|
env["XCTestConfigurationFilePath"] != nil ||
|
||||||
|
env["XCTestBundlePath"] != nil ||
|
||||||
|
env["GITHUB_ACTIONS"] != nil ||
|
||||||
|
env["CI"] != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Initialization
|
||||||
|
|
||||||
|
private override init() {
|
||||||
|
self.storage = .standard
|
||||||
|
super.init()
|
||||||
|
|
||||||
|
// Skip CoreLocation setup in test environments
|
||||||
|
guard !Self.isRunningTests else {
|
||||||
|
loadPersistedState()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cl.delegate = self
|
||||||
|
cl.desiredAccuracy = kCLLocationAccuracyHundredMeters
|
||||||
|
cl.distanceFilter = TransportConfig.locationDistanceFilterMeters
|
||||||
|
|
||||||
|
loadPersistedState()
|
||||||
|
initializePermissionState()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Internal initializer for testing with custom storage
|
||||||
|
init(storage: UserDefaults) {
|
||||||
|
self.storage = storage
|
||||||
|
super.init()
|
||||||
|
loadPersistedState()
|
||||||
|
}
|
||||||
|
|
||||||
|
private func loadPersistedState() {
|
||||||
|
// Load selected channel
|
||||||
|
if let data = storage.data(forKey: selectedChannelKey),
|
||||||
|
let channel = try? JSONDecoder().decode(ChannelID.self, from: data) {
|
||||||
|
selectedChannel = channel
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load teleported set
|
||||||
|
if let data = storage.data(forKey: teleportedStoreKey),
|
||||||
|
let arr = try? JSONDecoder().decode([String].self, from: data) {
|
||||||
|
teleportedSet = Set(arr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load bookmarks
|
||||||
|
if let data = storage.data(forKey: bookmarksKey),
|
||||||
|
let arr = try? JSONDecoder().decode([String].self, from: data) {
|
||||||
|
var seen = Set<String>()
|
||||||
|
var list: [String] = []
|
||||||
|
for raw in arr {
|
||||||
|
let gh = Self.normalizeGeohash(raw)
|
||||||
|
guard !gh.isEmpty, !seen.contains(gh) else { continue }
|
||||||
|
seen.insert(gh)
|
||||||
|
list.append(gh)
|
||||||
|
}
|
||||||
|
bookmarks = list
|
||||||
|
bookmarkMembership = seen
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load bookmark names
|
||||||
|
if let data = storage.data(forKey: bookmarkNamesKey),
|
||||||
|
let dict = try? JSONDecoder().decode([String: String].self, from: data) {
|
||||||
|
bookmarkNames = dict
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func initializePermissionState() {
|
||||||
|
let status: CLAuthorizationStatus
|
||||||
|
if #available(iOS 14.0, macOS 11.0, *) {
|
||||||
|
status = cl.authorizationStatus
|
||||||
|
} else {
|
||||||
|
status = CLLocationManager.authorizationStatus()
|
||||||
|
}
|
||||||
|
updatePermissionState(from: status)
|
||||||
|
|
||||||
|
// Fall back to persisted teleport state if no location authorization
|
||||||
|
switch status {
|
||||||
|
case .authorizedAlways, .authorizedWhenInUse, .authorized:
|
||||||
|
break
|
||||||
|
case .notDetermined, .restricted, .denied:
|
||||||
|
fallthrough
|
||||||
|
@unknown default:
|
||||||
|
if case .location(let ch) = selectedChannel {
|
||||||
|
teleported = teleportedSet.contains(ch.geohash)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Public API (Permissions & Location)
|
||||||
|
|
||||||
|
func enableLocationChannels() {
|
||||||
|
let status: CLAuthorizationStatus
|
||||||
|
if #available(iOS 14.0, macOS 11.0, *) {
|
||||||
|
status = cl.authorizationStatus
|
||||||
|
} else {
|
||||||
|
status = CLLocationManager.authorizationStatus()
|
||||||
|
}
|
||||||
|
switch status {
|
||||||
|
case .notDetermined:
|
||||||
|
cl.requestWhenInUseAuthorization()
|
||||||
|
case .restricted:
|
||||||
|
Task { @MainActor in self.permissionState = .restricted }
|
||||||
|
case .denied:
|
||||||
|
Task { @MainActor in self.permissionState = .denied }
|
||||||
|
case .authorizedAlways, .authorizedWhenInUse, .authorized:
|
||||||
|
Task { @MainActor in self.permissionState = .authorized }
|
||||||
|
requestOneShotLocation()
|
||||||
|
@unknown default:
|
||||||
|
Task { @MainActor in self.permissionState = .restricted }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func refreshChannels() {
|
||||||
|
if permissionState == .authorized {
|
||||||
|
requestOneShotLocation()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func beginLiveRefresh(interval: TimeInterval = TransportConfig.locationLiveRefreshInterval) {
|
||||||
|
guard permissionState == .authorized else { return }
|
||||||
|
refreshTimer?.invalidate()
|
||||||
|
refreshTimer = nil
|
||||||
|
cl.desiredAccuracy = kCLLocationAccuracyNearestTenMeters
|
||||||
|
cl.distanceFilter = TransportConfig.locationDistanceFilterLiveMeters
|
||||||
|
cl.startUpdatingLocation()
|
||||||
|
requestOneShotLocation()
|
||||||
|
}
|
||||||
|
|
||||||
|
func endLiveRefresh() {
|
||||||
|
refreshTimer?.invalidate()
|
||||||
|
refreshTimer = nil
|
||||||
|
cl.stopUpdatingLocation()
|
||||||
|
cl.desiredAccuracy = kCLLocationAccuracyHundredMeters
|
||||||
|
cl.distanceFilter = TransportConfig.locationDistanceFilterMeters
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Public API (Channel Selection)
|
||||||
|
|
||||||
|
func select(_ channel: ChannelID) {
|
||||||
|
Task { @MainActor in
|
||||||
|
self.selectedChannel = channel
|
||||||
|
if let data = try? JSONEncoder().encode(channel) {
|
||||||
|
self.storage.set(data, forKey: self.selectedChannelKey)
|
||||||
|
}
|
||||||
|
|
||||||
|
switch channel {
|
||||||
|
case .mesh:
|
||||||
|
self.teleported = false
|
||||||
|
case .location(let ch):
|
||||||
|
let inRegional = self.availableChannels.contains { $0.geohash == ch.geohash }
|
||||||
|
if inRegional {
|
||||||
|
self.teleported = false
|
||||||
|
if self.teleportedSet.contains(ch.geohash) {
|
||||||
|
self.teleportedSet.remove(ch.geohash)
|
||||||
|
self.persistTeleportedSet()
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
self.teleported = self.teleportedSet.contains(ch.geohash)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func markTeleported(for geohash: String, _ flag: Bool) {
|
||||||
|
if flag {
|
||||||
|
teleportedSet.insert(geohash)
|
||||||
|
} else {
|
||||||
|
teleportedSet.remove(geohash)
|
||||||
|
}
|
||||||
|
persistTeleportedSet()
|
||||||
|
if case .location(let ch) = selectedChannel, ch.geohash == geohash {
|
||||||
|
Task { @MainActor in self.teleported = flag }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Public API (Bookmarks)
|
||||||
|
|
||||||
|
func isBookmarked(_ geohash: String) -> Bool {
|
||||||
|
bookmarkMembership.contains(Self.normalizeGeohash(geohash))
|
||||||
|
}
|
||||||
|
|
||||||
|
func toggleBookmark(_ geohash: String) {
|
||||||
|
let gh = Self.normalizeGeohash(geohash)
|
||||||
|
if bookmarkMembership.contains(gh) {
|
||||||
|
removeBookmark(gh)
|
||||||
|
} else {
|
||||||
|
addBookmark(gh)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func addBookmark(_ geohash: String) {
|
||||||
|
let gh = Self.normalizeGeohash(geohash)
|
||||||
|
guard !gh.isEmpty, !bookmarkMembership.contains(gh) else { return }
|
||||||
|
bookmarks.insert(gh, at: 0)
|
||||||
|
bookmarkMembership.insert(gh)
|
||||||
|
persistBookmarks()
|
||||||
|
resolveBookmarkNameIfNeeded(for: gh)
|
||||||
|
}
|
||||||
|
|
||||||
|
func removeBookmark(_ geohash: String) {
|
||||||
|
let gh = Self.normalizeGeohash(geohash)
|
||||||
|
guard bookmarkMembership.contains(gh) else { return }
|
||||||
|
if let idx = bookmarks.firstIndex(of: gh) {
|
||||||
|
bookmarks.remove(at: idx)
|
||||||
|
}
|
||||||
|
bookmarkMembership.remove(gh)
|
||||||
|
if bookmarkNames.removeValue(forKey: gh) != nil {
|
||||||
|
persistBookmarkNames()
|
||||||
|
}
|
||||||
|
persistBookmarks()
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - CLLocationManagerDelegate
|
||||||
|
|
||||||
|
private func requestOneShotLocation() {
|
||||||
|
cl.requestLocation()
|
||||||
|
}
|
||||||
|
|
||||||
|
func locationManager(_ manager: CLLocationManager, didChangeAuthorization status: CLAuthorizationStatus) {
|
||||||
|
updatePermissionState(from: status)
|
||||||
|
if case .authorized = permissionState {
|
||||||
|
requestOneShotLocation()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@available(iOS 14.0, macOS 11.0, *)
|
||||||
|
func locationManagerDidChangeAuthorization(_ manager: CLLocationManager) {
|
||||||
|
updatePermissionState(from: manager.authorizationStatus)
|
||||||
|
if case .authorized = permissionState {
|
||||||
|
requestOneShotLocation()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func locationManager(_ manager: CLLocationManager, didUpdateLocations locations: [CLLocation]) {
|
||||||
|
guard let loc = locations.last else { return }
|
||||||
|
lastLocation = loc
|
||||||
|
computeChannels(from: loc.coordinate)
|
||||||
|
reverseGeocodeLocation(loc)
|
||||||
|
}
|
||||||
|
|
||||||
|
func locationManager(_ manager: CLLocationManager, didFailWithError error: Error) {
|
||||||
|
SecureLogger.error("LocationStateManager: location error: \(error.localizedDescription)", category: .session)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Private Helpers (Permission)
|
||||||
|
|
||||||
|
private func updatePermissionState(from status: CLAuthorizationStatus) {
|
||||||
|
let newState: PermissionState
|
||||||
|
switch status {
|
||||||
|
case .notDetermined: newState = .notDetermined
|
||||||
|
case .restricted: newState = .restricted
|
||||||
|
case .denied: newState = .denied
|
||||||
|
case .authorizedAlways, .authorizedWhenInUse, .authorized: newState = .authorized
|
||||||
|
@unknown default: newState = .restricted
|
||||||
|
}
|
||||||
|
Task { @MainActor in self.permissionState = newState }
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Private Helpers (Channel Computation)
|
||||||
|
|
||||||
|
private func computeChannels(from coord: CLLocationCoordinate2D) {
|
||||||
|
let levels = GeohashChannelLevel.allCases
|
||||||
|
var result: [GeohashChannel] = []
|
||||||
|
for level in levels {
|
||||||
|
let gh = Geohash.encode(latitude: coord.latitude, longitude: coord.longitude, precision: level.precision)
|
||||||
|
result.append(GeohashChannel(level: level, geohash: gh))
|
||||||
|
}
|
||||||
|
Task { @MainActor in
|
||||||
|
self.availableChannels = result
|
||||||
|
switch self.selectedChannel {
|
||||||
|
case .mesh:
|
||||||
|
self.teleported = false
|
||||||
|
case .location(let ch):
|
||||||
|
let inRegional = result.contains { $0.geohash == ch.geohash }
|
||||||
|
if inRegional {
|
||||||
|
self.teleported = false
|
||||||
|
if self.teleportedSet.contains(ch.geohash) {
|
||||||
|
self.teleportedSet.remove(ch.geohash)
|
||||||
|
self.persistTeleportedSet()
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
self.teleported = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Private Helpers (Geocoding)
|
||||||
|
|
||||||
|
private func reverseGeocodeLocation(_ location: CLLocation) {
|
||||||
|
geocoder.cancelGeocode()
|
||||||
|
isGeocoding = true
|
||||||
|
geocoder.reverseGeocodeLocation(location) { [weak self] placemarks, _ in
|
||||||
|
guard let self = self else { return }
|
||||||
|
self.isGeocoding = false
|
||||||
|
if let pm = placemarks?.first {
|
||||||
|
let names = self.locationNamesByLevel(from: pm)
|
||||||
|
Task { @MainActor in self.locationNames = names }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func locationNamesByLevel(from pm: CLPlacemark) -> [GeohashChannelLevel: String] {
|
||||||
|
var dict: [GeohashChannelLevel: String] = [:]
|
||||||
|
if let country = pm.country, !country.isEmpty {
|
||||||
|
dict[.region] = country
|
||||||
|
}
|
||||||
|
if let admin = pm.administrativeArea, !admin.isEmpty {
|
||||||
|
dict[.province] = admin
|
||||||
|
} else if let subAdmin = pm.subAdministrativeArea, !subAdmin.isEmpty {
|
||||||
|
dict[.province] = subAdmin
|
||||||
|
}
|
||||||
|
if let locality = pm.locality, !locality.isEmpty {
|
||||||
|
dict[.city] = locality
|
||||||
|
} else if let subAdmin = pm.subAdministrativeArea, !subAdmin.isEmpty {
|
||||||
|
dict[.city] = subAdmin
|
||||||
|
} else if let admin = pm.administrativeArea, !admin.isEmpty {
|
||||||
|
dict[.city] = admin
|
||||||
|
}
|
||||||
|
if let subLocality = pm.subLocality, !subLocality.isEmpty {
|
||||||
|
dict[.neighborhood] = subLocality
|
||||||
|
} else if let locality = pm.locality, !locality.isEmpty {
|
||||||
|
dict[.neighborhood] = locality
|
||||||
|
}
|
||||||
|
if let subLocality = pm.subLocality, !subLocality.isEmpty {
|
||||||
|
dict[.block] = subLocality
|
||||||
|
} else if let locality = pm.locality, !locality.isEmpty {
|
||||||
|
dict[.block] = locality
|
||||||
|
}
|
||||||
|
if let name = pm.name, !name.isEmpty {
|
||||||
|
dict[.building] = name
|
||||||
|
} else if let thoroughfare = pm.thoroughfare, !thoroughfare.isEmpty {
|
||||||
|
dict[.building] = thoroughfare
|
||||||
|
}
|
||||||
|
return dict
|
||||||
|
}
|
||||||
|
|
||||||
|
func resolveBookmarkNameIfNeeded(for geohash: String) {
|
||||||
|
let gh = Self.normalizeGeohash(geohash)
|
||||||
|
guard !gh.isEmpty, bookmarkNames[gh] == nil, !resolvingNames.contains(gh) else { return }
|
||||||
|
resolvingNames.insert(gh)
|
||||||
|
|
||||||
|
if gh.count <= 2 {
|
||||||
|
let b = Geohash.decodeBounds(gh)
|
||||||
|
let pts: [CLLocation] = [
|
||||||
|
CLLocation(latitude: (b.latMin + b.latMax) / 2, longitude: (b.lonMin + b.lonMax) / 2),
|
||||||
|
CLLocation(latitude: b.latMin, longitude: b.lonMin),
|
||||||
|
CLLocation(latitude: b.latMin, longitude: b.lonMax),
|
||||||
|
CLLocation(latitude: b.latMax, longitude: b.lonMin),
|
||||||
|
CLLocation(latitude: b.latMax, longitude: b.lonMax)
|
||||||
|
]
|
||||||
|
resolveCompositeAdminName(geohash: gh, points: pts)
|
||||||
|
} else {
|
||||||
|
let center = Geohash.decodeCenter(gh)
|
||||||
|
let loc = CLLocation(latitude: center.lat, longitude: center.lon)
|
||||||
|
geocoder.reverseGeocodeLocation(loc) { [weak self] placemarks, _ in
|
||||||
|
guard let self = self else { return }
|
||||||
|
defer { self.resolvingNames.remove(gh) }
|
||||||
|
if let pm = placemarks?.first,
|
||||||
|
let name = Self.nameForGeohashLength(gh.count, from: pm),
|
||||||
|
!name.isEmpty {
|
||||||
|
DispatchQueue.main.async {
|
||||||
|
self.bookmarkNames[gh] = name
|
||||||
|
self.persistBookmarkNames()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func resolveCompositeAdminName(geohash gh: String, points: [CLLocation]) {
|
||||||
|
var uniqueAdmins: [String] = []
|
||||||
|
var seenAdmins = Set<String>()
|
||||||
|
var idx = 0
|
||||||
|
|
||||||
|
func step() {
|
||||||
|
if idx >= points.count {
|
||||||
|
let finalName: String? = {
|
||||||
|
if uniqueAdmins.count >= 2 { return uniqueAdmins[0] + " and " + uniqueAdmins[1] }
|
||||||
|
return uniqueAdmins.first
|
||||||
|
}()
|
||||||
|
if let finalName = finalName, !finalName.isEmpty {
|
||||||
|
DispatchQueue.main.async {
|
||||||
|
self.bookmarkNames[gh] = finalName
|
||||||
|
self.persistBookmarkNames()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
self.resolvingNames.remove(gh)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
let loc = points[idx]
|
||||||
|
idx += 1
|
||||||
|
geocoder.reverseGeocodeLocation(loc) { [weak self] placemarks, _ in
|
||||||
|
guard self != nil else { return }
|
||||||
|
if let pm = placemarks?.first {
|
||||||
|
if let admin = pm.administrativeArea, !admin.isEmpty, !seenAdmins.contains(admin) {
|
||||||
|
seenAdmins.insert(admin)
|
||||||
|
uniqueAdmins.append(admin)
|
||||||
|
} else if let country = pm.country, !country.isEmpty, !seenAdmins.contains(country) {
|
||||||
|
seenAdmins.insert(country)
|
||||||
|
uniqueAdmins.append(country)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
step()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
step()
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func nameForGeohashLength(_ len: Int, from pm: CLPlacemark) -> String? {
|
||||||
|
switch len {
|
||||||
|
case 0...2:
|
||||||
|
return pm.administrativeArea ?? pm.country
|
||||||
|
case 3...4:
|
||||||
|
return pm.administrativeArea ?? pm.subAdministrativeArea ?? pm.country
|
||||||
|
case 5:
|
||||||
|
return pm.locality ?? pm.subAdministrativeArea ?? pm.administrativeArea
|
||||||
|
case 6...7:
|
||||||
|
return pm.subLocality ?? pm.locality ?? pm.administrativeArea
|
||||||
|
default:
|
||||||
|
return pm.subLocality ?? pm.locality ?? pm.administrativeArea ?? pm.country
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Private Helpers (Persistence)
|
||||||
|
|
||||||
|
private func persistTeleportedSet() {
|
||||||
|
if let data = try? JSONEncoder().encode(Array(teleportedSet)) {
|
||||||
|
storage.set(data, forKey: teleportedStoreKey)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func persistBookmarks() {
|
||||||
|
if let data = try? JSONEncoder().encode(bookmarks) {
|
||||||
|
storage.set(data, forKey: bookmarksKey)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func persistBookmarkNames() {
|
||||||
|
if let data = try? JSONEncoder().encode(bookmarkNames) {
|
||||||
|
storage.set(data, forKey: bookmarkNamesKey)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func normalizeGeohash(_ s: String) -> String {
|
||||||
|
let allowed = Set("0123456789bcdefghjkmnpqrstuvwxyz")
|
||||||
|
return s
|
||||||
|
.trimmingCharacters(in: .whitespacesAndNewlines)
|
||||||
|
.lowercased()
|
||||||
|
.replacingOccurrences(of: "#", with: "")
|
||||||
|
.filter { allowed.contains($0) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Backward Compatibility Typealiases
|
||||||
|
|
||||||
|
typealias LocationChannelManager = LocationStateManager
|
||||||
|
typealias GeohashBookmarksStore = LocationStateManager
|
||||||
|
|
||||||
|
// MARK: - Backward Compatibility Extensions
|
||||||
|
|
||||||
|
extension LocationStateManager {
|
||||||
|
/// Backward compatibility: toggle bookmark (was GeohashBookmarksStore.toggle)
|
||||||
|
func toggle(_ geohash: String) {
|
||||||
|
toggleBookmark(geohash)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Backward compatibility: add bookmark (was GeohashBookmarksStore.add)
|
||||||
|
func add(_ geohash: String) {
|
||||||
|
addBookmark(geohash)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Backward compatibility: remove bookmark (was GeohashBookmarksStore.remove)
|
||||||
|
func remove(_ geohash: String) {
|
||||||
|
removeBookmark(geohash)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
@@ -0,0 +1,113 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
/// Tracks observed mesh topology and computes hop-by-hop routes.
|
||||||
|
final class MeshTopologyTracker {
|
||||||
|
private typealias RoutingID = Data
|
||||||
|
|
||||||
|
private let queue = DispatchQueue(label: "mesh.topology", attributes: .concurrent)
|
||||||
|
private let hopSize = 8
|
||||||
|
// Directed claims: Key claims to see Value (neighbors)
|
||||||
|
private var claims: [RoutingID: Set<RoutingID>] = [:]
|
||||||
|
// Last time we received an update from a node
|
||||||
|
private var lastSeen: [RoutingID: Date] = [:]
|
||||||
|
|
||||||
|
func reset() {
|
||||||
|
queue.sync(flags: .barrier) {
|
||||||
|
self.claims.removeAll()
|
||||||
|
self.lastSeen.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Update the topology with a node's self-reported neighbor list
|
||||||
|
func updateNeighbors(for sourceData: Data?, neighbors: [Data]) {
|
||||||
|
guard let source = sanitize(sourceData) else { return }
|
||||||
|
// Sanitize neighbors and exclude self-loops
|
||||||
|
let validNeighbors = Set(neighbors.compactMap { sanitize($0) }).subtracting([source])
|
||||||
|
|
||||||
|
queue.sync(flags: .barrier) {
|
||||||
|
self.claims[source] = validNeighbors
|
||||||
|
self.lastSeen[source] = Date()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func removePeer(_ data: Data?) {
|
||||||
|
guard let peer = sanitize(data) else { return }
|
||||||
|
queue.sync(flags: .barrier) {
|
||||||
|
self.claims.removeValue(forKey: peer)
|
||||||
|
self.lastSeen.removeValue(forKey: peer)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Prune nodes that haven't updated their topology in `age` seconds
|
||||||
|
func prune(olderThan age: TimeInterval) {
|
||||||
|
let deadline = Date().addingTimeInterval(-age)
|
||||||
|
queue.sync(flags: .barrier) {
|
||||||
|
let stale = self.lastSeen.filter { $0.value < deadline }
|
||||||
|
for (peer, _) in stale {
|
||||||
|
self.claims.removeValue(forKey: peer)
|
||||||
|
self.lastSeen.removeValue(forKey: peer)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func computeRoute(from start: Data?, to goal: Data?, maxHops: Int = 10) -> [Data]? {
|
||||||
|
guard let source = sanitize(start), let target = sanitize(goal) else { return nil }
|
||||||
|
if source == target { return [] } // Direct connection, no intermediate hops
|
||||||
|
|
||||||
|
return queue.sync {
|
||||||
|
// BFS
|
||||||
|
var visited: Set<RoutingID> = [source]
|
||||||
|
// Queue stores paths: [Start, Hop1, Hop2, ..., Current]
|
||||||
|
var queuePaths: [[RoutingID]] = [[source]]
|
||||||
|
|
||||||
|
while !queuePaths.isEmpty {
|
||||||
|
let path = queuePaths.removeFirst()
|
||||||
|
// Limit path length (path contains source + maxHops + target) -> maxHops intermediate
|
||||||
|
// If maxHops = 10, max edges = 11, max nodes = 12.
|
||||||
|
if path.count > maxHops + 1 { continue }
|
||||||
|
|
||||||
|
guard let last = path.last else { continue }
|
||||||
|
|
||||||
|
// Get neighbors that 'last' claims to see
|
||||||
|
guard let neighbors = claims[last] else { continue }
|
||||||
|
|
||||||
|
for neighbor in neighbors {
|
||||||
|
if visited.contains(neighbor) { continue }
|
||||||
|
|
||||||
|
// CONFIRMED EDGE CHECK:
|
||||||
|
// 'last' claims 'neighbor' (checked above)
|
||||||
|
// Does 'neighbor' claim 'last'?
|
||||||
|
guard let neighborClaims = claims[neighbor],
|
||||||
|
neighborClaims.contains(last) else {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
var nextPath = path
|
||||||
|
nextPath.append(neighbor)
|
||||||
|
|
||||||
|
if neighbor == target {
|
||||||
|
// Return only intermediate hops
|
||||||
|
// Path: [Source, I1, I2, Target] -> [I1, I2]
|
||||||
|
return Array(nextPath.dropFirst().dropLast())
|
||||||
|
}
|
||||||
|
|
||||||
|
visited.insert(neighbor)
|
||||||
|
queuePaths.append(nextPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Helpers
|
||||||
|
|
||||||
|
private func sanitize(_ data: Data?) -> Data? {
|
||||||
|
guard var value = data, !value.isEmpty else { return nil }
|
||||||
|
if value.count > hopSize {
|
||||||
|
value = Data(value.prefix(hopSize))
|
||||||
|
} else if value.count < hopSize {
|
||||||
|
value.append(Data(repeating: 0, count: hopSize - value.count))
|
||||||
|
}
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,278 @@
|
|||||||
|
//
|
||||||
|
// MessageDeduplicationService.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Handles message deduplication using LRU caches.
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
// MARK: - LRU Deduplication Cache
|
||||||
|
|
||||||
|
/// Generic LRU (Least Recently Used) cache for deduplication.
|
||||||
|
/// Uses an efficient O(1) lookup with periodic compaction.
|
||||||
|
/// Thread-safe via @MainActor - all callers are already on main actor.
|
||||||
|
@MainActor
|
||||||
|
final class LRUDeduplicationCache<Value> {
|
||||||
|
private var map: [String: Value] = [:]
|
||||||
|
private var order: [String] = []
|
||||||
|
private var head: Int = 0
|
||||||
|
private let capacity: Int
|
||||||
|
|
||||||
|
/// Creates a new LRU cache with the specified capacity.
|
||||||
|
/// - Parameter capacity: Maximum number of entries before eviction
|
||||||
|
init(capacity: Int) {
|
||||||
|
precondition(capacity > 0, "LRU cache capacity must be positive")
|
||||||
|
self.capacity = capacity
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Number of active entries in the cache
|
||||||
|
var count: Int {
|
||||||
|
order.count - head
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Checks if a key exists in the cache
|
||||||
|
func contains(_ key: String) -> Bool {
|
||||||
|
map[key] != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value for a key, or nil if not present
|
||||||
|
func value(for key: String) -> Value? {
|
||||||
|
map[key]
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Records a key-value pair, updating if exists or inserting if new
|
||||||
|
func record(_ key: String, value: Value) {
|
||||||
|
if map[key] == nil {
|
||||||
|
order.append(key)
|
||||||
|
}
|
||||||
|
map[key] = value
|
||||||
|
trimIfNeeded()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Removes a specific key from the cache
|
||||||
|
func remove(_ key: String) {
|
||||||
|
map.removeValue(forKey: key)
|
||||||
|
// Note: key remains in order array but will be skipped during eviction
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Clears all entries from the cache
|
||||||
|
func clear() {
|
||||||
|
map.removeAll()
|
||||||
|
order.removeAll()
|
||||||
|
head = 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Private
|
||||||
|
|
||||||
|
private func trimIfNeeded() {
|
||||||
|
let activeCount = order.count - head
|
||||||
|
guard activeCount > capacity else { return }
|
||||||
|
|
||||||
|
let overflow = activeCount - capacity
|
||||||
|
for _ in 0..<overflow {
|
||||||
|
guard let victim = popOldest() else { break }
|
||||||
|
map.removeValue(forKey: victim)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func popOldest() -> String? {
|
||||||
|
// Skip keys that were already removed from map
|
||||||
|
while head < order.count {
|
||||||
|
let key = order[head]
|
||||||
|
head += 1
|
||||||
|
|
||||||
|
// Periodically compact the backing storage
|
||||||
|
if head >= 32 && head * 2 >= order.count {
|
||||||
|
order.removeFirst(head)
|
||||||
|
head = 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// Only return if key is still in map
|
||||||
|
if map[key] != nil {
|
||||||
|
return key
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Content Normalizer
|
||||||
|
|
||||||
|
/// Normalizes message content for near-duplicate detection.
|
||||||
|
enum ContentNormalizer {
|
||||||
|
|
||||||
|
/// Regex to simplify HTTP URLs by stripping query strings and fragments
|
||||||
|
private static let simplifyHTTPURL: NSRegularExpression = {
|
||||||
|
try! NSRegularExpression(
|
||||||
|
pattern: "https?://[^\\s?#]+(?:[?#][^\\s]*)?",
|
||||||
|
options: [.caseInsensitive]
|
||||||
|
)
|
||||||
|
}()
|
||||||
|
|
||||||
|
/// Normalizes content for deduplication comparison.
|
||||||
|
/// - Parameters:
|
||||||
|
/// - content: The raw message content
|
||||||
|
/// - prefixLength: Maximum characters to consider (default from TransportConfig)
|
||||||
|
/// - Returns: A hash-based key for comparison
|
||||||
|
static func normalizedKey(
|
||||||
|
_ content: String,
|
||||||
|
prefixLength: Int = TransportConfig.contentKeyPrefixLength
|
||||||
|
) -> String {
|
||||||
|
// Lowercase for case-insensitive comparison
|
||||||
|
let lowered = content.lowercased()
|
||||||
|
let ns = lowered as NSString
|
||||||
|
let range = NSRange(location: 0, length: ns.length)
|
||||||
|
|
||||||
|
// Simplify URLs by stripping query/fragment
|
||||||
|
var simplified = ""
|
||||||
|
var last = 0
|
||||||
|
for match in simplifyHTTPURL.matches(in: lowered, options: [], range: range) {
|
||||||
|
if match.range.location > last {
|
||||||
|
simplified += ns.substring(with: NSRange(location: last, length: match.range.location - last))
|
||||||
|
}
|
||||||
|
let url = ns.substring(with: match.range)
|
||||||
|
if let queryIndex = url.firstIndex(where: { $0 == "?" || $0 == "#" }) {
|
||||||
|
simplified += String(url[..<queryIndex])
|
||||||
|
} else {
|
||||||
|
simplified += url
|
||||||
|
}
|
||||||
|
last = match.range.location + match.range.length
|
||||||
|
}
|
||||||
|
if last < ns.length {
|
||||||
|
simplified += ns.substring(with: NSRange(location: last, length: ns.length - last))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Trim and collapse whitespace
|
||||||
|
let trimmed = simplified.trimmingCharacters(in: .whitespacesAndNewlines)
|
||||||
|
let collapsed = trimmed.replacingOccurrences(of: "\\s+", with: " ", options: .regularExpression)
|
||||||
|
|
||||||
|
// Take prefix and hash
|
||||||
|
let prefix = String(collapsed.prefix(prefixLength))
|
||||||
|
let hash = prefix.djb2()
|
||||||
|
return String(format: "h:%016llx", hash)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Message Deduplication Service
|
||||||
|
|
||||||
|
/// Service that manages message deduplication using LRU caches.
|
||||||
|
/// Provides separate caches for content-based dedup and Nostr event ID dedup.
|
||||||
|
/// Thread-safe via @MainActor - all callers are already on main actor.
|
||||||
|
@MainActor
|
||||||
|
final class MessageDeduplicationService {
|
||||||
|
|
||||||
|
/// Cache for content-based near-duplicate detection
|
||||||
|
private let contentCache: LRUDeduplicationCache<Date>
|
||||||
|
|
||||||
|
/// Cache for Nostr event ID deduplication
|
||||||
|
private let nostrEventCache: LRUDeduplicationCache<Bool>
|
||||||
|
|
||||||
|
/// Cache for Nostr ACK deduplication (messageId:ackType:senderPubkey format)
|
||||||
|
private let nostrAckCache: LRUDeduplicationCache<Bool>
|
||||||
|
|
||||||
|
/// Creates a new deduplication service with specified capacities.
|
||||||
|
/// - Parameters:
|
||||||
|
/// - contentCapacity: Max entries for content cache
|
||||||
|
/// - nostrEventCapacity: Max entries for Nostr event cache
|
||||||
|
init(
|
||||||
|
contentCapacity: Int = TransportConfig.contentLRUCap,
|
||||||
|
nostrEventCapacity: Int = TransportConfig.uiProcessedNostrEventsCap
|
||||||
|
) {
|
||||||
|
self.contentCache = LRUDeduplicationCache(capacity: contentCapacity)
|
||||||
|
self.nostrEventCache = LRUDeduplicationCache(capacity: nostrEventCapacity)
|
||||||
|
self.nostrAckCache = LRUDeduplicationCache(capacity: nostrEventCapacity)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Content Deduplication
|
||||||
|
|
||||||
|
/// Records content with its timestamp for near-duplicate detection.
|
||||||
|
/// - Parameters:
|
||||||
|
/// - content: The message content
|
||||||
|
/// - timestamp: When the content was received
|
||||||
|
func recordContent(_ content: String, timestamp: Date) {
|
||||||
|
let key = ContentNormalizer.normalizedKey(content)
|
||||||
|
contentCache.record(key, value: timestamp)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Records a pre-normalized content key with its timestamp.
|
||||||
|
/// - Parameters:
|
||||||
|
/// - key: The normalized content key
|
||||||
|
/// - timestamp: When the content was received
|
||||||
|
func recordContentKey(_ key: String, timestamp: Date) {
|
||||||
|
contentCache.record(key, value: timestamp)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the timestamp for previously seen content.
|
||||||
|
/// - Parameter content: The message content
|
||||||
|
/// - Returns: The timestamp when first seen, or nil if not seen
|
||||||
|
func contentTimestamp(for content: String) -> Date? {
|
||||||
|
let key = ContentNormalizer.normalizedKey(content)
|
||||||
|
return contentCache.value(for: key)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the timestamp for a pre-normalized content key.
|
||||||
|
/// - Parameter key: The normalized content key
|
||||||
|
/// - Returns: The timestamp when first seen, or nil if not seen
|
||||||
|
func contentTimestamp(forKey key: String) -> Date? {
|
||||||
|
contentCache.value(for: key)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Normalizes content to a deduplication key.
|
||||||
|
/// - Parameter content: The raw content
|
||||||
|
/// - Returns: A normalized hash key
|
||||||
|
func normalizedContentKey(_ content: String) -> String {
|
||||||
|
ContentNormalizer.normalizedKey(content)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Nostr Event Deduplication
|
||||||
|
|
||||||
|
/// Checks if a Nostr event has already been processed.
|
||||||
|
/// - Parameter eventId: The event ID
|
||||||
|
/// - Returns: true if already processed
|
||||||
|
func hasProcessedNostrEvent(_ eventId: String) -> Bool {
|
||||||
|
nostrEventCache.contains(eventId)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Records a Nostr event as processed.
|
||||||
|
/// - Parameter eventId: The event ID
|
||||||
|
func recordNostrEvent(_ eventId: String) {
|
||||||
|
nostrEventCache.record(eventId, value: true)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Nostr ACK Deduplication
|
||||||
|
|
||||||
|
/// Checks if a Nostr ACK has already been processed.
|
||||||
|
/// - Parameter ackKey: The ACK key in format "messageId:ackType:senderPubkey"
|
||||||
|
/// - Returns: true if already processed
|
||||||
|
func hasProcessedNostrAck(_ ackKey: String) -> Bool {
|
||||||
|
nostrAckCache.contains(ackKey)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Records a Nostr ACK as processed.
|
||||||
|
/// - Parameter ackKey: The ACK key in format "messageId:ackType:senderPubkey"
|
||||||
|
func recordNostrAck(_ ackKey: String) {
|
||||||
|
nostrAckCache.record(ackKey, value: true)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Creates an ACK key from components.
|
||||||
|
static func ackKey(messageId: String, ackType: String, senderPubkey: String) -> String {
|
||||||
|
"\(messageId):\(ackType):\(senderPubkey)"
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Clear
|
||||||
|
|
||||||
|
/// Clears all caches
|
||||||
|
func clearAll() {
|
||||||
|
contentCache.clear()
|
||||||
|
nostrEventCache.clear()
|
||||||
|
nostrAckCache.clear()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Clears only the Nostr caches (events and ACKs)
|
||||||
|
func clearNostrCaches() {
|
||||||
|
nostrEventCache.clear()
|
||||||
|
nostrAckCache.clear()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,471 @@
|
|||||||
|
//
|
||||||
|
// MessageFormattingEngine.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Handles message text formatting, including mentions, hashtags, URLs, and tokens.
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
import SwiftUI
|
||||||
|
|
||||||
|
// MARK: - Formatting Context Protocol
|
||||||
|
|
||||||
|
/// Protocol defining the context needed for message formatting.
|
||||||
|
/// Implemented by ChatViewModel to provide runtime state.
|
||||||
|
@MainActor
|
||||||
|
protocol MessageFormattingContext: AnyObject {
|
||||||
|
/// The user's current nickname
|
||||||
|
var nickname: String { get }
|
||||||
|
|
||||||
|
/// Determines if a message was sent by the current user
|
||||||
|
func isSelfMessage(_ message: BitchatMessage) -> Bool
|
||||||
|
|
||||||
|
/// Gets the color for a message's sender
|
||||||
|
func senderColor(for message: BitchatMessage, isDark: Bool) -> Color
|
||||||
|
|
||||||
|
/// Resolves a peer ID to a clickable URL
|
||||||
|
func peerURL(for peerID: PeerID) -> URL?
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Formatting Engine
|
||||||
|
|
||||||
|
/// Handles rich text formatting for chat messages.
|
||||||
|
/// Extracts mentions, hashtags, URLs, Lightning invoices, and Cashu tokens.
|
||||||
|
final class MessageFormattingEngine {
|
||||||
|
|
||||||
|
// MARK: - Precompiled Regexes
|
||||||
|
|
||||||
|
/// Precompiled regex patterns for message content parsing
|
||||||
|
enum Patterns {
|
||||||
|
static let hashtag: NSRegularExpression = {
|
||||||
|
try! NSRegularExpression(pattern: "#([a-zA-Z0-9_]+)", options: [])
|
||||||
|
}()
|
||||||
|
|
||||||
|
static let mention: NSRegularExpression = {
|
||||||
|
try! NSRegularExpression(pattern: "@([\\p{L}0-9_]+(?:#[a-fA-F0-9]{4})?)", options: [])
|
||||||
|
}()
|
||||||
|
|
||||||
|
static let cashu: NSRegularExpression = {
|
||||||
|
try! NSRegularExpression(pattern: "\\bcashu[AB][A-Za-z0-9._-]{40,}\\b", options: [])
|
||||||
|
}()
|
||||||
|
|
||||||
|
static let bolt11: NSRegularExpression = {
|
||||||
|
try! NSRegularExpression(pattern: "(?i)\\bln(bc|tb|bcrt)[0-9][a-z0-9]{50,}\\b", options: [])
|
||||||
|
}()
|
||||||
|
|
||||||
|
static let lnurl: NSRegularExpression = {
|
||||||
|
try! NSRegularExpression(pattern: "(?i)\\blnurl1[a-z0-9]{20,}\\b", options: [])
|
||||||
|
}()
|
||||||
|
|
||||||
|
static let lightningScheme: NSRegularExpression = {
|
||||||
|
try! NSRegularExpression(pattern: "(?i)\\blightning:[^\\s]+", options: [])
|
||||||
|
}()
|
||||||
|
|
||||||
|
static let linkDetector: NSDataDetector? = {
|
||||||
|
try? NSDataDetector(types: NSTextCheckingResult.CheckingType.link.rawValue)
|
||||||
|
}()
|
||||||
|
|
||||||
|
static let quickCashuPresence: NSRegularExpression = {
|
||||||
|
try! NSRegularExpression(pattern: "\\bcashu[AB][A-Za-z0-9._-]{40,}\\b", options: [])
|
||||||
|
}()
|
||||||
|
|
||||||
|
static let simplifyHTTPURL: NSRegularExpression = {
|
||||||
|
try! NSRegularExpression(pattern: "https?://[^\\s?#]+(?:[?#][^\\s]*)?", options: [.caseInsensitive])
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Match Types
|
||||||
|
|
||||||
|
/// Types of matches found in message content
|
||||||
|
enum MatchType: String {
|
||||||
|
case hashtag
|
||||||
|
case mention
|
||||||
|
case url
|
||||||
|
case cashu
|
||||||
|
case lightning
|
||||||
|
case bolt11
|
||||||
|
case lnurl
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A match found in message content
|
||||||
|
struct ContentMatch {
|
||||||
|
let range: NSRange
|
||||||
|
let type: MatchType
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Public API
|
||||||
|
|
||||||
|
/// Formats a message with rich text styling
|
||||||
|
@MainActor
|
||||||
|
static func formatMessage(
|
||||||
|
_ message: BitchatMessage,
|
||||||
|
context: MessageFormattingContext,
|
||||||
|
colorScheme: ColorScheme
|
||||||
|
) -> AttributedString {
|
||||||
|
let isDark = colorScheme == .dark
|
||||||
|
let isSelf = context.isSelfMessage(message)
|
||||||
|
|
||||||
|
// Check cache first
|
||||||
|
if let cached = message.getCachedFormattedText(isDark: isDark, isSelf: isSelf) {
|
||||||
|
return cached
|
||||||
|
}
|
||||||
|
|
||||||
|
var result = AttributedString()
|
||||||
|
let baseColor: Color = isSelf ? .orange : context.senderColor(for: message, isDark: isDark)
|
||||||
|
|
||||||
|
// Format system messages differently
|
||||||
|
if message.sender == "system" {
|
||||||
|
result = formatSystemMessage(message, isDark: isDark)
|
||||||
|
} else {
|
||||||
|
// Format sender header
|
||||||
|
result = formatSenderHeader(
|
||||||
|
message: message,
|
||||||
|
baseColor: baseColor,
|
||||||
|
isSelf: isSelf,
|
||||||
|
context: context
|
||||||
|
)
|
||||||
|
|
||||||
|
// Format content
|
||||||
|
let contentResult = formatContent(
|
||||||
|
message.content,
|
||||||
|
baseColor: baseColor,
|
||||||
|
isSelf: isSelf,
|
||||||
|
isMentioned: message.mentions?.contains(context.nickname) ?? false
|
||||||
|
)
|
||||||
|
result.append(contentResult)
|
||||||
|
|
||||||
|
// Add timestamp
|
||||||
|
result.append(formatTimestamp(message.formattedTimestamp))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Cache the result
|
||||||
|
message.setCachedFormattedText(result, isDark: isDark, isSelf: isSelf)
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Formats just the message header (sender portion)
|
||||||
|
@MainActor
|
||||||
|
static func formatHeader(
|
||||||
|
_ message: BitchatMessage,
|
||||||
|
context: MessageFormattingContext,
|
||||||
|
colorScheme: ColorScheme
|
||||||
|
) -> AttributedString {
|
||||||
|
let isDark = colorScheme == .dark
|
||||||
|
let isSelf = context.isSelfMessage(message)
|
||||||
|
let baseColor: Color = isSelf ? .orange : context.senderColor(for: message, isDark: isDark)
|
||||||
|
|
||||||
|
if message.sender == "system" {
|
||||||
|
var style = AttributeContainer()
|
||||||
|
style.foregroundColor = baseColor
|
||||||
|
style.font = .bitchatSystem(size: 14, weight: .medium, design: .monospaced)
|
||||||
|
return AttributedString(message.sender).mergingAttributes(style)
|
||||||
|
}
|
||||||
|
|
||||||
|
return formatSenderHeader(
|
||||||
|
message: message,
|
||||||
|
baseColor: baseColor,
|
||||||
|
isSelf: isSelf,
|
||||||
|
context: context
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Extracts mentions from message content
|
||||||
|
static func extractMentions(from content: String) -> [String] {
|
||||||
|
let nsContent = content as NSString
|
||||||
|
let range = NSRange(location: 0, length: nsContent.length)
|
||||||
|
let matches = Patterns.mention.matches(in: content, options: [], range: range)
|
||||||
|
|
||||||
|
return matches.compactMap { match -> String? in
|
||||||
|
guard match.numberOfRanges > 1 else { return nil }
|
||||||
|
let captureRange = match.range(at: 1)
|
||||||
|
guard let swiftRange = Range(captureRange, in: content) else { return nil }
|
||||||
|
return String(content[swiftRange])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Checks if content contains a Cashu token
|
||||||
|
static func containsCashuToken(_ content: String) -> Bool {
|
||||||
|
let nsContent = content as NSString
|
||||||
|
let range = NSRange(location: 0, length: nsContent.length)
|
||||||
|
return Patterns.quickCashuPresence.numberOfMatches(in: content, options: [], range: range) > 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Private Helpers
|
||||||
|
|
||||||
|
private static func formatSystemMessage(_ message: BitchatMessage, isDark: Bool) -> AttributedString {
|
||||||
|
var result = AttributedString()
|
||||||
|
|
||||||
|
let content = AttributedString("* \(message.content) *")
|
||||||
|
var contentStyle = AttributeContainer()
|
||||||
|
contentStyle.foregroundColor = Color.gray
|
||||||
|
contentStyle.font = .bitchatSystem(size: 12, design: .monospaced).italic()
|
||||||
|
result.append(content.mergingAttributes(contentStyle))
|
||||||
|
|
||||||
|
// Add timestamp
|
||||||
|
let timestamp = AttributedString(" [\(message.formattedTimestamp)]")
|
||||||
|
var timestampStyle = AttributeContainer()
|
||||||
|
timestampStyle.foregroundColor = Color.gray.opacity(0.5)
|
||||||
|
timestampStyle.font = .bitchatSystem(size: 10, design: .monospaced)
|
||||||
|
result.append(timestamp.mergingAttributes(timestampStyle))
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
private static func formatSenderHeader(
|
||||||
|
message: BitchatMessage,
|
||||||
|
baseColor: Color,
|
||||||
|
isSelf: Bool,
|
||||||
|
context: MessageFormattingContext
|
||||||
|
) -> AttributedString {
|
||||||
|
var result = AttributedString()
|
||||||
|
|
||||||
|
let (baseName, suffix) = message.sender.splitSuffix()
|
||||||
|
var senderStyle = AttributeContainer()
|
||||||
|
senderStyle.foregroundColor = baseColor
|
||||||
|
let fontWeight: Font.Weight = isSelf ? .bold : .medium
|
||||||
|
senderStyle.font = .bitchatSystem(size: 14, weight: fontWeight, design: .monospaced)
|
||||||
|
|
||||||
|
// Make sender clickable
|
||||||
|
if let spid = message.senderPeerID, let url = context.peerURL(for: spid) {
|
||||||
|
senderStyle.link = url
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build: "<@baseName#suffix> "
|
||||||
|
result.append(AttributedString("<@").mergingAttributes(senderStyle))
|
||||||
|
result.append(AttributedString(baseName).mergingAttributes(senderStyle))
|
||||||
|
|
||||||
|
if !suffix.isEmpty {
|
||||||
|
var suffixStyle = senderStyle
|
||||||
|
suffixStyle.foregroundColor = baseColor.opacity(0.6)
|
||||||
|
result.append(AttributedString(suffix).mergingAttributes(suffixStyle))
|
||||||
|
}
|
||||||
|
|
||||||
|
result.append(AttributedString("> ").mergingAttributes(senderStyle))
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func formatContent(
|
||||||
|
_ content: String,
|
||||||
|
baseColor: Color,
|
||||||
|
isSelf: Bool,
|
||||||
|
isMentioned: Bool
|
||||||
|
) -> AttributedString {
|
||||||
|
// For very long content without special tokens, use plain formatting
|
||||||
|
let containsCashu = containsCashuToken(content)
|
||||||
|
if (content.count > 4000 || content.hasVeryLongToken(threshold: 1024)) && !containsCashu {
|
||||||
|
return formatPlainContent(content, baseColor: baseColor, isSelf: isSelf)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Find all matches
|
||||||
|
let matches = findAllMatches(in: content)
|
||||||
|
|
||||||
|
// Build formatted content
|
||||||
|
var result = AttributedString()
|
||||||
|
var lastEnd = content.startIndex
|
||||||
|
|
||||||
|
for match in matches {
|
||||||
|
guard let swiftRange = Range(match.range, in: content) else { continue }
|
||||||
|
|
||||||
|
// Add text before match
|
||||||
|
if lastEnd < swiftRange.lowerBound {
|
||||||
|
let beforeText = String(content[lastEnd..<swiftRange.lowerBound])
|
||||||
|
result.append(formatPlainText(beforeText, baseColor: baseColor, isSelf: isSelf, isMentioned: isMentioned))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add styled match
|
||||||
|
let matchText = String(content[swiftRange])
|
||||||
|
result.append(formatMatch(matchText, type: match.type, baseColor: baseColor, isSelf: isSelf))
|
||||||
|
|
||||||
|
lastEnd = swiftRange.upperBound
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add remaining text
|
||||||
|
if lastEnd < content.endIndex {
|
||||||
|
let remainingText = String(content[lastEnd...])
|
||||||
|
result.append(formatPlainText(remainingText, baseColor: baseColor, isSelf: isSelf, isMentioned: isMentioned))
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func findAllMatches(in content: String) -> [ContentMatch] {
|
||||||
|
let nsContent = content as NSString
|
||||||
|
let nsLen = nsContent.length
|
||||||
|
let fullRange = NSRange(location: 0, length: nsLen)
|
||||||
|
|
||||||
|
// Quick hints to avoid unnecessary regex work
|
||||||
|
let hasMentions = content.contains("@")
|
||||||
|
let hasHashtags = content.contains("#")
|
||||||
|
let hasURLs = content.contains("://") || content.contains("www.") || content.contains("http")
|
||||||
|
let hasLightning = content.lowercased().contains("ln") || content.lowercased().contains("lightning:")
|
||||||
|
let hasCashu = content.lowercased().contains("cashu")
|
||||||
|
|
||||||
|
// Collect matches
|
||||||
|
let mentionMatches = hasMentions ? Patterns.mention.matches(in: content, options: [], range: fullRange) : []
|
||||||
|
let hashtagMatches = hasHashtags ? Patterns.hashtag.matches(in: content, options: [], range: fullRange) : []
|
||||||
|
let urlMatches = hasURLs ? (Patterns.linkDetector?.matches(in: content, options: [], range: fullRange) ?? []) : []
|
||||||
|
let cashuMatches = hasCashu ? Patterns.cashu.matches(in: content, options: [], range: fullRange) : []
|
||||||
|
let lightningMatches = hasLightning ? Patterns.lightningScheme.matches(in: content, options: [], range: fullRange) : []
|
||||||
|
let bolt11Matches = hasLightning ? Patterns.bolt11.matches(in: content, options: [], range: fullRange) : []
|
||||||
|
let lnurlMatches = hasLightning ? Patterns.lnurl.matches(in: content, options: [], range: fullRange) : []
|
||||||
|
|
||||||
|
// Build mention ranges for overlap checking
|
||||||
|
let mentionRanges = mentionMatches.map { $0.range(at: 0) }
|
||||||
|
|
||||||
|
func overlapsMention(_ r: NSRange) -> Bool {
|
||||||
|
mentionRanges.contains { NSIntersectionRange(r, $0).length > 0 }
|
||||||
|
}
|
||||||
|
|
||||||
|
func isStandaloneHashtag(_ r: NSRange) -> Bool {
|
||||||
|
guard let swiftRange = Range(r, in: content) else { return false }
|
||||||
|
if swiftRange.lowerBound == content.startIndex { return true }
|
||||||
|
let prev = content.index(before: swiftRange.lowerBound)
|
||||||
|
return content[prev].isWhitespace || content[prev].isNewline
|
||||||
|
}
|
||||||
|
|
||||||
|
func attachedToMention(_ r: NSRange) -> Bool {
|
||||||
|
guard let swiftRange = Range(r, in: content), swiftRange.lowerBound > content.startIndex else { return false }
|
||||||
|
var i = content.index(before: swiftRange.lowerBound)
|
||||||
|
while true {
|
||||||
|
let ch = content[i]
|
||||||
|
if ch.isWhitespace || ch.isNewline { break }
|
||||||
|
if ch == "@" { return true }
|
||||||
|
if i == content.startIndex { break }
|
||||||
|
i = content.index(before: i)
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
var allMatches: [ContentMatch] = []
|
||||||
|
|
||||||
|
// Add hashtags (excluding those attached to mentions)
|
||||||
|
for match in hashtagMatches {
|
||||||
|
let range = match.range(at: 0)
|
||||||
|
if !overlapsMention(range) && !attachedToMention(range) && isStandaloneHashtag(range) {
|
||||||
|
allMatches.append(ContentMatch(range: range, type: .hashtag))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add mentions
|
||||||
|
for match in mentionMatches {
|
||||||
|
allMatches.append(ContentMatch(range: match.range(at: 0), type: .mention))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add URLs
|
||||||
|
for match in urlMatches where !overlapsMention(match.range) {
|
||||||
|
allMatches.append(ContentMatch(range: match.range, type: .url))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add Cashu tokens
|
||||||
|
for match in cashuMatches where !overlapsMention(match.range(at: 0)) {
|
||||||
|
allMatches.append(ContentMatch(range: match.range(at: 0), type: .cashu))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add Lightning scheme URLs
|
||||||
|
for match in lightningMatches where !overlapsMention(match.range(at: 0)) {
|
||||||
|
allMatches.append(ContentMatch(range: match.range(at: 0), type: .lightning))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add bolt11/lnurl (avoiding overlaps with lightning scheme and URLs)
|
||||||
|
let occupied = urlMatches.map { $0.range } + lightningMatches.map { $0.range(at: 0) }
|
||||||
|
func overlapsOccupied(_ r: NSRange) -> Bool {
|
||||||
|
occupied.contains { NSIntersectionRange(r, $0).length > 0 }
|
||||||
|
}
|
||||||
|
|
||||||
|
for match in bolt11Matches where !overlapsMention(match.range(at: 0)) && !overlapsOccupied(match.range(at: 0)) {
|
||||||
|
allMatches.append(ContentMatch(range: match.range(at: 0), type: .bolt11))
|
||||||
|
}
|
||||||
|
|
||||||
|
for match in lnurlMatches where !overlapsMention(match.range(at: 0)) && !overlapsOccupied(match.range(at: 0)) {
|
||||||
|
allMatches.append(ContentMatch(range: match.range(at: 0), type: .lnurl))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sort by position
|
||||||
|
return allMatches.sorted { $0.range.location < $1.range.location }
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func formatPlainContent(_ content: String, baseColor: Color, isSelf: Bool) -> AttributedString {
|
||||||
|
var style = AttributeContainer()
|
||||||
|
style.foregroundColor = baseColor
|
||||||
|
style.font = isSelf
|
||||||
|
? .bitchatSystem(size: 14, weight: .bold, design: .monospaced)
|
||||||
|
: .bitchatSystem(size: 14, design: .monospaced)
|
||||||
|
return AttributedString(content).mergingAttributes(style)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func formatPlainText(_ text: String, baseColor: Color, isSelf: Bool, isMentioned: Bool) -> AttributedString {
|
||||||
|
guard !text.isEmpty else { return AttributedString() }
|
||||||
|
|
||||||
|
var style = AttributeContainer()
|
||||||
|
style.foregroundColor = baseColor
|
||||||
|
style.font = isSelf
|
||||||
|
? .bitchatSystem(size: 14, weight: .bold, design: .monospaced)
|
||||||
|
: .bitchatSystem(size: 14, design: .monospaced)
|
||||||
|
|
||||||
|
if isMentioned {
|
||||||
|
style.font = style.font?.bold()
|
||||||
|
}
|
||||||
|
|
||||||
|
return AttributedString(text).mergingAttributes(style)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func formatMatch(_ text: String, type: MatchType, baseColor: Color, isSelf: Bool) -> AttributedString {
|
||||||
|
var style = AttributeContainer()
|
||||||
|
|
||||||
|
switch type {
|
||||||
|
case .mention:
|
||||||
|
// Split optional '#abcd' suffix
|
||||||
|
let (baseName, suffix) = text.splitSuffix()
|
||||||
|
var result = AttributedString()
|
||||||
|
|
||||||
|
var mentionStyle = AttributeContainer()
|
||||||
|
mentionStyle.foregroundColor = .blue
|
||||||
|
mentionStyle.font = .bitchatSystem(size: 14, weight: .semibold, design: .monospaced)
|
||||||
|
result.append(AttributedString(baseName).mergingAttributes(mentionStyle))
|
||||||
|
|
||||||
|
if !suffix.isEmpty {
|
||||||
|
var suffixStyle = mentionStyle
|
||||||
|
suffixStyle.foregroundColor = Color.gray.opacity(0.7)
|
||||||
|
result.append(AttributedString(suffix).mergingAttributes(suffixStyle))
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
case .hashtag:
|
||||||
|
style.foregroundColor = .purple
|
||||||
|
style.font = .bitchatSystem(size: 14, weight: .medium, design: .monospaced)
|
||||||
|
|
||||||
|
case .url:
|
||||||
|
style.foregroundColor = .blue
|
||||||
|
style.font = .bitchatSystem(size: 14, design: .monospaced)
|
||||||
|
style.underlineStyle = .single
|
||||||
|
if let url = URL(string: text) {
|
||||||
|
style.link = url
|
||||||
|
}
|
||||||
|
|
||||||
|
case .cashu:
|
||||||
|
style.foregroundColor = .green
|
||||||
|
style.font = .bitchatSystem(size: 14, weight: .medium, design: .monospaced)
|
||||||
|
style.backgroundColor = Color.green.opacity(0.1)
|
||||||
|
|
||||||
|
case .lightning, .bolt11, .lnurl:
|
||||||
|
style.foregroundColor = .yellow
|
||||||
|
style.font = .bitchatSystem(size: 14, weight: .medium, design: .monospaced)
|
||||||
|
style.backgroundColor = Color.yellow.opacity(0.1)
|
||||||
|
}
|
||||||
|
|
||||||
|
return AttributedString(text).mergingAttributes(style)
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func formatTimestamp(_ timestamp: String) -> AttributedString {
|
||||||
|
let text = AttributedString(" [\(timestamp)]")
|
||||||
|
var style = AttributeContainer()
|
||||||
|
style.foregroundColor = Color.gray.opacity(0.5)
|
||||||
|
style.font = .bitchatSystem(size: 10, design: .monospaced)
|
||||||
|
return text.mergingAttributes(style)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,17 +1,14 @@
|
|||||||
import BitLogger
|
import BitLogger
|
||||||
import Foundation
|
import Foundation
|
||||||
|
|
||||||
/// Routes messages between BLE and Nostr transports
|
/// Routes messages using available transports (Mesh, Nostr, etc.)
|
||||||
@MainActor
|
@MainActor
|
||||||
final class MessageRouter {
|
final class MessageRouter {
|
||||||
private let mesh: Transport
|
private let transports: [Transport]
|
||||||
private let nostr: NostrTransport
|
|
||||||
private var outbox: [PeerID: [(content: String, nickname: String, messageID: String)]] = [:] // peerID -> queued messages
|
private var outbox: [PeerID: [(content: String, nickname: String, messageID: String)]] = [:] // peerID -> queued messages
|
||||||
|
|
||||||
init(mesh: Transport, nostr: NostrTransport) {
|
init(transports: [Transport]) {
|
||||||
self.mesh = mesh
|
self.transports = transports
|
||||||
self.nostr = nostr
|
|
||||||
self.nostr.senderPeerID = mesh.myPeerID
|
|
||||||
|
|
||||||
// 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(
|
||||||
@@ -38,88 +35,70 @@ final class MessageRouter {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func sendPrivate(_ content: String, to peerID: PeerID, recipientNickname: String, messageID: String) {
|
func sendPrivate(_ content: String, to peerID: PeerID, recipientNickname: String, messageID: String) {
|
||||||
let reachableMesh = mesh.isPeerReachable(peerID)
|
// Try to find a reachable transport
|
||||||
if reachableMesh {
|
if let transport = transports.first(where: { $0.isPeerReachable(peerID) }) {
|
||||||
SecureLogger.debug("Routing PM via mesh (reachable) to \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
SecureLogger.debug("Routing PM via \(type(of: transport)) to \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
||||||
// BLEService will initiate a handshake if needed and queue the message
|
transport.sendPrivateMessage(content, to: peerID, recipientNickname: recipientNickname, messageID: messageID)
|
||||||
mesh.sendPrivateMessage(content, to: peerID, recipientNickname: recipientNickname, messageID: messageID)
|
|
||||||
} else if canSendViaNostr(peerID: peerID) {
|
|
||||||
SecureLogger.debug("Routing PM via Nostr to \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
|
||||||
nostr.sendPrivateMessage(content, to: peerID, recipientNickname: recipientNickname, messageID: messageID)
|
|
||||||
} else {
|
} else {
|
||||||
// Queue for later (when mesh connects or Nostr mapping appears)
|
// Queue for later
|
||||||
if outbox[peerID] == nil { outbox[peerID] = [] }
|
if outbox[peerID] == nil { outbox[peerID] = [] }
|
||||||
outbox[peerID]?.append((content, recipientNickname, messageID))
|
outbox[peerID]?.append((content, recipientNickname, messageID))
|
||||||
SecureLogger.debug("Queued PM for \(peerID.id.prefix(8))… (no mesh, no Nostr mapping) id=\(messageID.prefix(8))…", category: .session)
|
SecureLogger.debug("Queued PM for \(peerID.id.prefix(8))… (no reachable transport) id=\(messageID.prefix(8))…", category: .session)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func sendReadReceipt(_ receipt: ReadReceipt, to peerID: PeerID) {
|
func sendReadReceipt(_ receipt: ReadReceipt, to peerID: PeerID) {
|
||||||
// Prefer mesh for reachable peers; BLE will queue if handshake is needed
|
if let transport = transports.first(where: { $0.isPeerReachable(peerID) }) {
|
||||||
if mesh.isPeerReachable(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 mesh (reachable) to \(peerID.id.prefix(8))… id=\(receipt.originalMessageID.prefix(8))…", category: .session)
|
transport.sendReadReceipt(receipt, to: peerID)
|
||||||
mesh.sendReadReceipt(receipt, to: peerID)
|
} else if !transports.isEmpty {
|
||||||
} else {
|
// Fallback to last transport (usually Nostr) if neither is explicitly reachable?
|
||||||
SecureLogger.debug("Routing READ ack via Nostr to \(peerID.id.prefix(8))… id=\(receipt.originalMessageID.prefix(8))…", category: .session)
|
// Or better: just try the first one that supports it?
|
||||||
nostr.sendReadReceipt(receipt, to: peerID)
|
// Existing logic preferred mesh, then nostr.
|
||||||
|
// If neither reachable, existing logic queued it (via mesh usually) or sent via nostr.
|
||||||
|
// Let's stick to "try reachable". If none, maybe pick the first one to queue?
|
||||||
|
// Actually, for READ receipts, we might want to just fire-and-forget on the "best effort" transport.
|
||||||
|
// But let's stick to the reachable check.
|
||||||
|
SecureLogger.debug("No reachable transport for READ ack to \(peerID.id.prefix(8))…", category: .session)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func sendDeliveryAck(_ messageID: String, to peerID: PeerID) {
|
func sendDeliveryAck(_ messageID: String, to peerID: PeerID) {
|
||||||
if mesh.isPeerReachable(peerID) {
|
if let transport = transports.first(where: { $0.isPeerReachable(peerID) }) {
|
||||||
SecureLogger.debug("Routing DELIVERED ack via mesh (reachable) to \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
SecureLogger.debug("Routing DELIVERED ack via \(type(of: transport)) to \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
||||||
mesh.sendDeliveryAck(for: messageID, to: peerID)
|
transport.sendDeliveryAck(for: messageID, to: peerID)
|
||||||
} else {
|
|
||||||
nostr.sendDeliveryAck(for: messageID, to: peerID)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func sendFavoriteNotification(to peerID: PeerID, isFavorite: Bool) {
|
func sendFavoriteNotification(to peerID: PeerID, isFavorite: Bool) {
|
||||||
// Route via mesh when connected; else use Nostr
|
if let transport = transports.first(where: { $0.isPeerConnected(peerID) }) {
|
||||||
if mesh.isPeerConnected(peerID) {
|
transport.sendFavoriteNotification(to: peerID, isFavorite: isFavorite)
|
||||||
mesh.sendFavoriteNotification(to: peerID, isFavorite: isFavorite)
|
} else if let transport = transports.first(where: { $0.isPeerReachable(peerID) }) {
|
||||||
|
transport.sendFavoriteNotification(to: peerID, isFavorite: isFavorite)
|
||||||
} else {
|
} else {
|
||||||
nostr.sendFavoriteNotification(to: peerID, isFavorite: isFavorite)
|
// Fallback: try all? or just the last one?
|
||||||
|
// Old logic: if mesh connected, mesh. Else nostr.
|
||||||
|
// Note: NostrTransport.isPeerReachable now returns true if mapped.
|
||||||
|
// If not mapped, we can't send via Nostr anyway.
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Outbox Management
|
// MARK: - Outbox Management
|
||||||
private func canSendViaNostr(peerID: PeerID) -> Bool {
|
|
||||||
// Two forms are supported:
|
|
||||||
// - 64-hex Noise public key (32 bytes)
|
|
||||||
// - 16-hex short peer ID (derived from Noise pubkey)
|
|
||||||
if let noiseKey = peerID.noiseKey {
|
|
||||||
if let fav = FavoritesPersistenceService.shared.getFavoriteStatus(for: noiseKey),
|
|
||||||
fav.peerNostrPublicKey != nil {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
} else if peerID.isShort {
|
|
||||||
if let fav = FavoritesPersistenceService.shared.getFavoriteStatus(forPeerID: peerID),
|
|
||||||
fav.peerNostrPublicKey != nil {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func flushOutbox(for peerID: PeerID) {
|
func flushOutbox(for peerID: PeerID) {
|
||||||
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)
|
||||||
var remaining: [(content: String, nickname: String, messageID: String)] = []
|
var remaining: [(content: String, nickname: String, messageID: String)] = []
|
||||||
// Prefer mesh if connected; else try Nostr if mapping exists
|
|
||||||
for (content, nickname, messageID) in queued {
|
for (content, nickname, messageID) in queued {
|
||||||
if mesh.isPeerReachable(peerID) {
|
if let transport = transports.first(where: { $0.isPeerReachable(peerID) }) {
|
||||||
SecureLogger.debug("Outbox -> mesh for \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
SecureLogger.debug("Outbox -> \(type(of: transport)) for \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
||||||
mesh.sendPrivateMessage(content, to: peerID, recipientNickname: nickname, messageID: messageID)
|
transport.sendPrivateMessage(content, to: peerID, recipientNickname: nickname, messageID: messageID)
|
||||||
} else if canSendViaNostr(peerID: peerID) {
|
|
||||||
SecureLogger.debug("Outbox -> Nostr for \(peerID.id.prefix(8))… id=\(messageID.prefix(8))…", category: .session)
|
|
||||||
nostr.sendPrivateMessage(content, to: peerID, recipientNickname: nickname, messageID: messageID)
|
|
||||||
} else {
|
} else {
|
||||||
// Keep unsent items queued
|
|
||||||
remaining.append((content, nickname, messageID))
|
remaining.append((content, nickname, messageID))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// Persist only items we could not send
|
|
||||||
if remaining.isEmpty {
|
if remaining.isEmpty {
|
||||||
outbox.removeValue(forKey: peerID)
|
outbox.removeValue(forKey: peerID)
|
||||||
} else {
|
} else {
|
||||||
|
|||||||
@@ -177,18 +177,18 @@ final class NoiseEncryptionService {
|
|||||||
private let rekeyCheckInterval: TimeInterval = 60.0 // Check every minute
|
private let rekeyCheckInterval: TimeInterval = 60.0 // Check every minute
|
||||||
|
|
||||||
// Callbacks
|
// Callbacks
|
||||||
private var onPeerAuthenticatedHandlers: [((String, String) -> Void)] = [] // Array of handlers for peer authentication
|
private var onPeerAuthenticatedHandlers: [((PeerID, String) -> Void)] = [] // Array of handlers for peer authentication
|
||||||
var onHandshakeRequired: ((PeerID) -> Void)? // peerID needs handshake
|
var onHandshakeRequired: ((PeerID) -> Void)? // peerID needs handshake
|
||||||
|
|
||||||
// Add a handler for peer authentication
|
// Add a handler for peer authentication
|
||||||
func addOnPeerAuthenticatedHandler(_ handler: @escaping (String, String) -> Void) {
|
func addOnPeerAuthenticatedHandler(_ handler: @escaping (PeerID, String) -> Void) {
|
||||||
serviceQueue.async(flags: .barrier) { [weak self] in
|
serviceQueue.async(flags: .barrier) { [weak self] in
|
||||||
self?.onPeerAuthenticatedHandlers.append(handler)
|
self?.onPeerAuthenticatedHandlers.append(handler)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Legacy support - setting this will add to the handlers array
|
// Legacy support - setting this will add to the handlers array
|
||||||
var onPeerAuthenticated: ((String, String) -> Void)? {
|
var onPeerAuthenticated: ((PeerID, String) -> Void)? {
|
||||||
get { nil } // Always return nil for backward compatibility
|
get { nil } // Always return nil for backward compatibility
|
||||||
set {
|
set {
|
||||||
if let handler = newValue {
|
if let handler = newValue {
|
||||||
@@ -200,46 +200,85 @@ final class NoiseEncryptionService {
|
|||||||
init(keychain: KeychainManagerProtocol) {
|
init(keychain: KeychainManagerProtocol) {
|
||||||
self.keychain = keychain
|
self.keychain = keychain
|
||||||
|
|
||||||
// Load or create static identity key (ONLY from keychain)
|
// BCH-01-009: Load or create static identity key with proper error handling
|
||||||
let loadedKey: Curve25519.KeyAgreement.PrivateKey
|
let loadedKey: Curve25519.KeyAgreement.PrivateKey
|
||||||
|
|
||||||
// Try to load from keychain
|
// Try to load from keychain with proper error classification
|
||||||
if let identityData = keychain.getIdentityKey(forKey: "noiseStaticKey"),
|
let noiseKeyResult = keychain.getIdentityKeyWithResult(forKey: "noiseStaticKey")
|
||||||
let key = try? Curve25519.KeyAgreement.PrivateKey(rawRepresentation: identityData) {
|
|
||||||
|
switch noiseKeyResult {
|
||||||
|
case .success(let identityData):
|
||||||
|
if let key = try? Curve25519.KeyAgreement.PrivateKey(rawRepresentation: identityData) {
|
||||||
loadedKey = key
|
loadedKey = key
|
||||||
SecureLogger.logKeyOperation(.load, keyType: "noiseStaticKey", success: true)
|
SecureLogger.logKeyOperation(.load, keyType: "noiseStaticKey", success: true)
|
||||||
|
} else {
|
||||||
|
// Data corrupted, regenerate
|
||||||
|
SecureLogger.warning("Noise static key data corrupted, regenerating", category: .keychain)
|
||||||
|
loadedKey = Self.generateAndSaveNoiseKey(keychain: keychain)
|
||||||
}
|
}
|
||||||
// If no identity exists, create new one
|
|
||||||
else {
|
|
||||||
loadedKey = Curve25519.KeyAgreement.PrivateKey()
|
|
||||||
let keyData = loadedKey.rawRepresentation
|
|
||||||
|
|
||||||
// Save to keychain
|
case .itemNotFound:
|
||||||
let saved = keychain.saveIdentityKey(keyData, forKey: "noiseStaticKey")
|
// Expected case: no key exists yet, create new one
|
||||||
SecureLogger.logKeyOperation(.create, keyType: "noiseStaticKey", success: saved)
|
loadedKey = Self.generateAndSaveNoiseKey(keychain: keychain)
|
||||||
|
|
||||||
|
case .accessDenied:
|
||||||
|
// Critical error - log but proceed with ephemeral key (will be lost on restart)
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: -1),
|
||||||
|
context: "Keychain access denied - using ephemeral identity", category: .keychain)
|
||||||
|
loadedKey = Curve25519.KeyAgreement.PrivateKey()
|
||||||
|
|
||||||
|
case .deviceLocked, .authenticationFailed:
|
||||||
|
// Recoverable error - use ephemeral key and warn
|
||||||
|
SecureLogger.warning("Device locked or auth failed - using ephemeral identity until unlocked", category: .keychain)
|
||||||
|
loadedKey = Curve25519.KeyAgreement.PrivateKey()
|
||||||
|
|
||||||
|
case .otherError(let status):
|
||||||
|
// Unexpected error - log and use ephemeral key
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: Int(status)),
|
||||||
|
context: "Unexpected keychain error - using ephemeral identity", category: .keychain)
|
||||||
|
loadedKey = Curve25519.KeyAgreement.PrivateKey()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Now assign the final value
|
// Now assign the final value
|
||||||
self.staticIdentityKey = loadedKey
|
self.staticIdentityKey = loadedKey
|
||||||
self.staticIdentityPublicKey = staticIdentityKey.publicKey
|
self.staticIdentityPublicKey = staticIdentityKey.publicKey
|
||||||
|
|
||||||
// Load or create signing key pair
|
// BCH-01-009: Load or create signing key pair with proper error handling
|
||||||
let loadedSigningKey: Curve25519.Signing.PrivateKey
|
let loadedSigningKey: Curve25519.Signing.PrivateKey
|
||||||
|
|
||||||
// Try to load from keychain
|
let signingKeyResult = keychain.getIdentityKeyWithResult(forKey: "ed25519SigningKey")
|
||||||
if let signingData = keychain.getIdentityKey(forKey: "ed25519SigningKey"),
|
|
||||||
let key = try? Curve25519.Signing.PrivateKey(rawRepresentation: signingData) {
|
switch signingKeyResult {
|
||||||
|
case .success(let signingData):
|
||||||
|
if let key = try? Curve25519.Signing.PrivateKey(rawRepresentation: signingData) {
|
||||||
loadedSigningKey = key
|
loadedSigningKey = key
|
||||||
SecureLogger.logKeyOperation(.load, keyType: "ed25519SigningKey", success: true)
|
SecureLogger.logKeyOperation(.load, keyType: "ed25519SigningKey", success: true)
|
||||||
|
} else {
|
||||||
|
// Data corrupted, regenerate
|
||||||
|
SecureLogger.warning("Ed25519 signing key data corrupted, regenerating", category: .keychain)
|
||||||
|
loadedSigningKey = Self.generateAndSaveSigningKey(keychain: keychain)
|
||||||
}
|
}
|
||||||
// If no signing key exists, create new one
|
|
||||||
else {
|
|
||||||
loadedSigningKey = Curve25519.Signing.PrivateKey()
|
|
||||||
let keyData = loadedSigningKey.rawRepresentation
|
|
||||||
|
|
||||||
// Save to keychain
|
case .itemNotFound:
|
||||||
let saved = keychain.saveIdentityKey(keyData, forKey: "ed25519SigningKey")
|
// Expected case: no key exists yet, create new one
|
||||||
SecureLogger.logKeyOperation(.create, keyType: "ed25519SigningKey", success: saved)
|
loadedSigningKey = Self.generateAndSaveSigningKey(keychain: keychain)
|
||||||
|
|
||||||
|
case .accessDenied:
|
||||||
|
// Critical error - log but proceed with ephemeral key
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: -1),
|
||||||
|
context: "Keychain access denied - using ephemeral signing key", category: .keychain)
|
||||||
|
loadedSigningKey = Curve25519.Signing.PrivateKey()
|
||||||
|
|
||||||
|
case .deviceLocked, .authenticationFailed:
|
||||||
|
// Recoverable error - use ephemeral key and warn
|
||||||
|
SecureLogger.warning("Device locked or auth failed - using ephemeral signing key until unlocked", category: .keychain)
|
||||||
|
loadedSigningKey = Curve25519.Signing.PrivateKey()
|
||||||
|
|
||||||
|
case .otherError(let status):
|
||||||
|
// Unexpected error - log and use ephemeral key
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: Int(status)),
|
||||||
|
context: "Unexpected keychain error - using ephemeral signing key", category: .keychain)
|
||||||
|
loadedSigningKey = Curve25519.Signing.PrivateKey()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Now assign the signing keys
|
// Now assign the signing keys
|
||||||
@@ -258,6 +297,56 @@ final class NoiseEncryptionService {
|
|||||||
startRekeyTimer()
|
startRekeyTimer()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// MARK: - BCH-01-009: Key Generation Helpers with Save Verification
|
||||||
|
|
||||||
|
/// Generate and save a new Noise static key, verifying the save succeeds
|
||||||
|
private static func generateAndSaveNoiseKey(keychain: KeychainManagerProtocol) -> Curve25519.KeyAgreement.PrivateKey {
|
||||||
|
let newKey = Curve25519.KeyAgreement.PrivateKey()
|
||||||
|
let keyData = newKey.rawRepresentation
|
||||||
|
|
||||||
|
// Save to keychain and verify success
|
||||||
|
let saveResult = keychain.saveIdentityKeyWithResult(keyData, forKey: "noiseStaticKey")
|
||||||
|
|
||||||
|
switch saveResult {
|
||||||
|
case .success:
|
||||||
|
SecureLogger.logKeyOperation(.create, keyType: "noiseStaticKey", success: true)
|
||||||
|
case .duplicateItem:
|
||||||
|
// This shouldn't happen since we just tried to load, but handle it
|
||||||
|
SecureLogger.warning("Noise key already exists (race condition?)", category: .keychain)
|
||||||
|
default:
|
||||||
|
// Save failed - log but continue with the key (it will be ephemeral)
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: -1),
|
||||||
|
context: "Failed to persist noise static key - identity will be lost on restart",
|
||||||
|
category: .keychain)
|
||||||
|
}
|
||||||
|
|
||||||
|
return newKey
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Generate and save a new Ed25519 signing key, verifying the save succeeds
|
||||||
|
private static func generateAndSaveSigningKey(keychain: KeychainManagerProtocol) -> Curve25519.Signing.PrivateKey {
|
||||||
|
let newKey = Curve25519.Signing.PrivateKey()
|
||||||
|
let keyData = newKey.rawRepresentation
|
||||||
|
|
||||||
|
// Save to keychain and verify success
|
||||||
|
let saveResult = keychain.saveIdentityKeyWithResult(keyData, forKey: "ed25519SigningKey")
|
||||||
|
|
||||||
|
switch saveResult {
|
||||||
|
case .success:
|
||||||
|
SecureLogger.logKeyOperation(.create, keyType: "ed25519SigningKey", success: true)
|
||||||
|
case .duplicateItem:
|
||||||
|
// This shouldn't happen since we just tried to load, but handle it
|
||||||
|
SecureLogger.warning("Signing key already exists (race condition?)", category: .keychain)
|
||||||
|
default:
|
||||||
|
// Save failed - log but continue with the key (it will be ephemeral)
|
||||||
|
SecureLogger.error(NSError(domain: "Keychain", code: -1),
|
||||||
|
context: "Failed to persist signing key - identity will be lost on restart",
|
||||||
|
category: .keychain)
|
||||||
|
}
|
||||||
|
|
||||||
|
return newKey
|
||||||
|
}
|
||||||
|
|
||||||
// MARK: - Public Interface
|
// MARK: - Public Interface
|
||||||
|
|
||||||
/// Get our static public key for sharing
|
/// Get our static public key for sharing
|
||||||
@@ -546,7 +635,7 @@ final class NoiseEncryptionService {
|
|||||||
// Notify all handlers about authentication
|
// Notify all handlers about authentication
|
||||||
serviceQueue.async { [weak self] in
|
serviceQueue.async { [weak self] in
|
||||||
self?.onPeerAuthenticatedHandlers.forEach { handler in
|
self?.onPeerAuthenticatedHandlers.forEach { handler in
|
||||||
handler(peerID.id, fingerprint)
|
handler(peerID, fingerprint)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ import Foundation
|
|||||||
import Combine
|
import Combine
|
||||||
|
|
||||||
// Minimal Nostr transport conforming to Transport for offline sending
|
// Minimal Nostr transport conforming to Transport for offline sending
|
||||||
final class NostrTransport: Transport {
|
final class NostrTransport: Transport, @unchecked Sendable {
|
||||||
// Provide BLE short peer ID for BitChat embedding
|
// Provide BLE short peer ID for BitChat embedding
|
||||||
var senderPeerID = PeerID(str: "")
|
var senderPeerID = PeerID(str: "")
|
||||||
|
|
||||||
@@ -18,9 +18,49 @@ final class NostrTransport: Transport {
|
|||||||
private let keychain: KeychainManagerProtocol
|
private let keychain: KeychainManagerProtocol
|
||||||
private let idBridge: NostrIdentityBridge
|
private let idBridge: NostrIdentityBridge
|
||||||
|
|
||||||
|
// Reachability Cache (thread-safe)
|
||||||
|
private var reachablePeers: Set<PeerID> = []
|
||||||
|
private let queue = DispatchQueue(label: "nostr.transport.state", attributes: .concurrent)
|
||||||
|
|
||||||
|
@MainActor
|
||||||
init(keychain: KeychainManagerProtocol, idBridge: NostrIdentityBridge) {
|
init(keychain: KeychainManagerProtocol, idBridge: NostrIdentityBridge) {
|
||||||
self.keychain = keychain
|
self.keychain = keychain
|
||||||
self.idBridge = idBridge
|
self.idBridge = idBridge
|
||||||
|
|
||||||
|
setupObservers()
|
||||||
|
|
||||||
|
// Synchronously warm the cache to avoid startup race
|
||||||
|
let favorites = FavoritesPersistenceService.shared.favorites
|
||||||
|
let reachable = favorites.values
|
||||||
|
.filter { $0.peerNostrPublicKey != nil }
|
||||||
|
.map { PeerID(publicKey: $0.peerNoisePublicKey) }
|
||||||
|
|
||||||
|
queue.sync(flags: .barrier) {
|
||||||
|
self.reachablePeers = Set(reachable)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func setupObservers() {
|
||||||
|
NotificationCenter.default.addObserver(
|
||||||
|
forName: .favoriteStatusChanged,
|
||||||
|
object: nil,
|
||||||
|
queue: nil
|
||||||
|
) { [weak self] _ in
|
||||||
|
self?.refreshReachablePeers()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private func refreshReachablePeers() {
|
||||||
|
Task { @MainActor in
|
||||||
|
let favorites = FavoritesPersistenceService.shared.favorites
|
||||||
|
let reachable = favorites.values
|
||||||
|
.filter { $0.peerNostrPublicKey != nil }
|
||||||
|
.map { PeerID(publicKey: $0.peerNoisePublicKey) }
|
||||||
|
|
||||||
|
self.queue.async(flags: .barrier) { [weak self] in
|
||||||
|
self?.reachablePeers = Set(reachable)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// MARK: - Transport Protocol Conformance
|
// MARK: - Transport Protocol Conformance
|
||||||
@@ -42,7 +82,19 @@ final class NostrTransport: Transport {
|
|||||||
func emergencyDisconnectAll() { /* no-op */ }
|
func emergencyDisconnectAll() { /* no-op */ }
|
||||||
|
|
||||||
func isPeerConnected(_ peerID: PeerID) -> Bool { false }
|
func isPeerConnected(_ peerID: PeerID) -> Bool { false }
|
||||||
func isPeerReachable(_ peerID: PeerID) -> Bool { false }
|
|
||||||
|
func isPeerReachable(_ peerID: PeerID) -> Bool {
|
||||||
|
queue.sync {
|
||||||
|
// Check if exact match
|
||||||
|
if reachablePeers.contains(peerID) { return true }
|
||||||
|
// Check for short ID match
|
||||||
|
if peerID.isShort {
|
||||||
|
return reachablePeers.contains(where: { $0.toShort() == peerID })
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func peerNickname(peerID: PeerID) -> String? { nil }
|
func peerNickname(peerID: PeerID) -> String? { nil }
|
||||||
func getPeerNicknames() -> [PeerID : String] { [:] }
|
func getPeerNicknames() -> [PeerID : String] { [:] }
|
||||||
|
|
||||||
@@ -82,7 +134,7 @@ final class NostrTransport: Transport {
|
|||||||
SecureLogger.error("NostrTransport: failed to decode npub -> hex: \(error)", category: .session)
|
SecureLogger.error("NostrTransport: failed to decode npub -> hex: \(error)", category: .session)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
guard let embedded = NostrEmbeddedBitChat.encodePMForNostr(content: content, messageID: messageID, recipientPeerID: peerID.id, senderPeerID: senderPeerID.id) else {
|
guard let embedded = NostrEmbeddedBitChat.encodePMForNostr(content: content, messageID: messageID, recipientPeerID: peerID, senderPeerID: senderPeerID) else {
|
||||||
SecureLogger.error("NostrTransport: failed to embed PM packet", category: .session)
|
SecureLogger.error("NostrTransport: failed to embed PM packet", category: .session)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -97,8 +149,11 @@ final class NostrTransport: Transport {
|
|||||||
|
|
||||||
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
|
||||||
readQueue.append(QueuedRead(receipt: receipt, peerID: peerID))
|
// Use barrier to synchronize access to readQueue
|
||||||
processReadQueueIfNeeded()
|
queue.async(flags: .barrier) { [weak self] in
|
||||||
|
self?.readQueue.append(QueuedRead(receipt: receipt, peerID: peerID))
|
||||||
|
self?.processReadQueueIfNeeded()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func sendFavoriteNotification(to peerID: PeerID, isFavorite: Bool) {
|
func sendFavoriteNotification(to peerID: PeerID, isFavorite: Bool) {
|
||||||
@@ -113,8 +168,11 @@ final class NostrTransport: Transport {
|
|||||||
let (hrp, data) = try Bech32.decode(recipientNpub)
|
let (hrp, data) = try Bech32.decode(recipientNpub)
|
||||||
guard hrp == "npub" else { return }
|
guard hrp == "npub" else { return }
|
||||||
recipientHex = data.hexEncodedString()
|
recipientHex = data.hexEncodedString()
|
||||||
} catch { return }
|
} catch {
|
||||||
guard let embedded = NostrEmbeddedBitChat.encodePMForNostr(content: content, messageID: UUID().uuidString, recipientPeerID: peerID.id, senderPeerID: senderPeerID.id) else {
|
SecureLogger.error("NostrTransport: failed to decode recipient npub for favorite notification: \(error.localizedDescription)", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
guard let embedded = NostrEmbeddedBitChat.encodePMForNostr(content: content, messageID: UUID().uuidString, recipientPeerID: peerID, senderPeerID: senderPeerID) else {
|
||||||
SecureLogger.error("NostrTransport: failed to embed favorite notification", category: .session)
|
SecureLogger.error("NostrTransport: failed to embed favorite notification", category: .session)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -138,8 +196,11 @@ final class NostrTransport: Transport {
|
|||||||
let (hrp, data) = try Bech32.decode(recipientNpub)
|
let (hrp, data) = try Bech32.decode(recipientNpub)
|
||||||
guard hrp == "npub" else { return }
|
guard hrp == "npub" else { return }
|
||||||
recipientHex = data.hexEncodedString()
|
recipientHex = data.hexEncodedString()
|
||||||
} catch { return }
|
} catch {
|
||||||
guard let ack = NostrEmbeddedBitChat.encodeAckForNostr(type: .delivered, messageID: messageID, recipientPeerID: peerID.id, senderPeerID: senderPeerID.id) else {
|
SecureLogger.error("NostrTransport: failed to decode recipient npub for delivery ack: \(error.localizedDescription)", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
guard let ack = NostrEmbeddedBitChat.encodeAckForNostr(type: .delivered, messageID: messageID, recipientPeerID: peerID, senderPeerID: senderPeerID) else {
|
||||||
SecureLogger.error("NostrTransport: failed to embed DELIVERED ack", category: .session)
|
SecureLogger.error("NostrTransport: failed to embed DELIVERED ack", category: .session)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -161,7 +222,7 @@ extension NostrTransport {
|
|||||||
func sendDeliveryAckGeohash(for messageID: String, toRecipientHex recipientHex: String, from identity: NostrIdentity) {
|
func sendDeliveryAckGeohash(for messageID: String, toRecipientHex recipientHex: String, from identity: NostrIdentity) {
|
||||||
Task { @MainActor in
|
Task { @MainActor in
|
||||||
SecureLogger.debug("GeoDM: send DELIVERED -> recip=\(recipientHex.prefix(8))… mid=\(messageID.prefix(8))… from=\(identity.publicKeyHex.prefix(8))…", category: .session)
|
SecureLogger.debug("GeoDM: send DELIVERED -> recip=\(recipientHex.prefix(8))… mid=\(messageID.prefix(8))… from=\(identity.publicKeyHex.prefix(8))…", category: .session)
|
||||||
guard let embedded = NostrEmbeddedBitChat.encodeAckForNostrNoRecipient(type: .delivered, messageID: messageID, senderPeerID: senderPeerID.id) else { return }
|
guard let embedded = NostrEmbeddedBitChat.encodeAckForNostrNoRecipient(type: .delivered, messageID: messageID, senderPeerID: senderPeerID) else { return }
|
||||||
guard let event = try? NostrProtocol.createPrivateMessage(content: embedded, recipientPubkey: recipientHex, senderIdentity: identity) else { return }
|
guard let event = try? NostrProtocol.createPrivateMessage(content: embedded, recipientPubkey: recipientHex, senderIdentity: identity) else { return }
|
||||||
NostrRelayManager.registerPendingGiftWrap(id: event.id)
|
NostrRelayManager.registerPendingGiftWrap(id: event.id)
|
||||||
NostrRelayManager.shared.sendEvent(event)
|
NostrRelayManager.shared.sendEvent(event)
|
||||||
@@ -171,7 +232,7 @@ extension NostrTransport {
|
|||||||
func sendReadReceiptGeohash(_ messageID: String, toRecipientHex recipientHex: String, from identity: NostrIdentity) {
|
func sendReadReceiptGeohash(_ messageID: String, toRecipientHex recipientHex: String, from identity: NostrIdentity) {
|
||||||
Task { @MainActor in
|
Task { @MainActor in
|
||||||
SecureLogger.debug("GeoDM: send READ -> recip=\(recipientHex.prefix(8))… mid=\(messageID.prefix(8))… from=\(identity.publicKeyHex.prefix(8))…", category: .session)
|
SecureLogger.debug("GeoDM: send READ -> recip=\(recipientHex.prefix(8))… mid=\(messageID.prefix(8))… from=\(identity.publicKeyHex.prefix(8))…", category: .session)
|
||||||
guard let embedded = NostrEmbeddedBitChat.encodeAckForNostrNoRecipient(type: .readReceipt, messageID: messageID, senderPeerID: senderPeerID.id) else { return }
|
guard let embedded = NostrEmbeddedBitChat.encodeAckForNostrNoRecipient(type: .readReceipt, messageID: messageID, senderPeerID: senderPeerID) else { return }
|
||||||
guard let event = try? NostrProtocol.createPrivateMessage(content: embedded, recipientPubkey: recipientHex, senderIdentity: identity) else { return }
|
guard let event = try? NostrProtocol.createPrivateMessage(content: embedded, recipientPubkey: recipientHex, senderIdentity: identity) else { return }
|
||||||
NostrRelayManager.registerPendingGiftWrap(id: event.id)
|
NostrRelayManager.registerPendingGiftWrap(id: event.id)
|
||||||
NostrRelayManager.shared.sendEvent(event)
|
NostrRelayManager.shared.sendEvent(event)
|
||||||
@@ -184,7 +245,7 @@ extension NostrTransport {
|
|||||||
guard !recipientHex.isEmpty else { return }
|
guard !recipientHex.isEmpty else { return }
|
||||||
SecureLogger.debug("GeoDM: send PM -> recip=\(recipientHex.prefix(8))… mid=\(messageID.prefix(8))… from=\(identity.publicKeyHex.prefix(8))…", category: .session)
|
SecureLogger.debug("GeoDM: send PM -> recip=\(recipientHex.prefix(8))… mid=\(messageID.prefix(8))… from=\(identity.publicKeyHex.prefix(8))…", category: .session)
|
||||||
// Build embedded BitChat packet without recipient peer ID
|
// Build embedded BitChat packet without recipient peer ID
|
||||||
guard let embedded = NostrEmbeddedBitChat.encodePMForNostrNoRecipient(content: content, messageID: messageID, senderPeerID: senderPeerID.id) else {
|
guard let embedded = NostrEmbeddedBitChat.encodePMForNostrNoRecipient(content: content, messageID: messageID, senderPeerID: senderPeerID) else {
|
||||||
SecureLogger.error("NostrTransport: failed to embed geohash PM packet", category: .session)
|
SecureLogger.error("NostrTransport: failed to embed geohash PM packet", category: .session)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -202,16 +263,17 @@ extension NostrTransport {
|
|||||||
// MARK: - Private Helpers
|
// MARK: - Private Helpers
|
||||||
|
|
||||||
extension NostrTransport {
|
extension NostrTransport {
|
||||||
|
/// Must be called within a barrier on `queue`
|
||||||
private func processReadQueueIfNeeded() {
|
private func processReadQueueIfNeeded() {
|
||||||
guard !isSendingReadAcks else { return }
|
guard !isSendingReadAcks else { return }
|
||||||
guard !readQueue.isEmpty else { return }
|
guard !readQueue.isEmpty else { return }
|
||||||
isSendingReadAcks = true
|
isSendingReadAcks = true
|
||||||
sendNextReadAck()
|
let item = readQueue.removeFirst()
|
||||||
|
sendReadAckItem(item)
|
||||||
}
|
}
|
||||||
|
|
||||||
private func sendNextReadAck() {
|
/// Sends a single read ack item (called after extraction from queue within barrier)
|
||||||
guard !readQueue.isEmpty else { isSendingReadAcks = false; return }
|
private func sendReadAckItem(_ item: QueuedRead) {
|
||||||
let item = readQueue.removeFirst()
|
|
||||||
Task { @MainActor in
|
Task { @MainActor in
|
||||||
guard let recipientNpub = resolveRecipientNpub(for: item.peerID) else { scheduleNextReadAck(); return }
|
guard let recipientNpub = resolveRecipientNpub(for: item.peerID) else { scheduleNextReadAck(); return }
|
||||||
guard let senderIdentity = try? idBridge.getCurrentNostrIdentity() else { scheduleNextReadAck(); return }
|
guard let senderIdentity = try? idBridge.getCurrentNostrIdentity() else { scheduleNextReadAck(); return }
|
||||||
@@ -222,8 +284,12 @@ extension NostrTransport {
|
|||||||
let (hrp, data) = try Bech32.decode(recipientNpub)
|
let (hrp, data) = try Bech32.decode(recipientNpub)
|
||||||
guard hrp == "npub" else { scheduleNextReadAck(); return }
|
guard hrp == "npub" else { scheduleNextReadAck(); return }
|
||||||
recipientHex = data.hexEncodedString()
|
recipientHex = data.hexEncodedString()
|
||||||
} catch { scheduleNextReadAck(); return }
|
} catch {
|
||||||
guard let ack = NostrEmbeddedBitChat.encodeAckForNostr(type: .readReceipt, messageID: item.receipt.originalMessageID, recipientPeerID: item.peerID.id, senderPeerID: senderPeerID.id) else {
|
SecureLogger.error("NostrTransport: failed to decode recipient npub for read ack: \(error.localizedDescription)", category: .session)
|
||||||
|
scheduleNextReadAck()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
guard let ack = NostrEmbeddedBitChat.encodeAckForNostr(type: .readReceipt, messageID: item.receipt.originalMessageID, recipientPeerID: item.peerID, senderPeerID: senderPeerID) else {
|
||||||
SecureLogger.error("NostrTransport: failed to embed READ ack", category: .session)
|
SecureLogger.error("NostrTransport: failed to embed READ ack", category: .session)
|
||||||
scheduleNextReadAck(); return
|
scheduleNextReadAck(); return
|
||||||
}
|
}
|
||||||
@@ -239,9 +305,10 @@ extension NostrTransport {
|
|||||||
|
|
||||||
private func scheduleNextReadAck() {
|
private func scheduleNextReadAck() {
|
||||||
DispatchQueue.main.asyncAfter(deadline: .now() + readAckInterval) { [weak self] in
|
DispatchQueue.main.asyncAfter(deadline: .now() + readAckInterval) { [weak self] in
|
||||||
guard let self = self else { return }
|
self?.queue.async(flags: .barrier) { [weak self] in
|
||||||
self.isSendingReadAcks = false
|
self?.isSendingReadAcks = false
|
||||||
self.processReadQueueIfNeeded()
|
self?.processReadQueueIfNeeded()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -17,9 +17,20 @@ import AppKit
|
|||||||
final class NotificationService {
|
final class NotificationService {
|
||||||
static let shared = NotificationService()
|
static let shared = NotificationService()
|
||||||
|
|
||||||
|
/// Returns true if running in test environment (XCTest, Swift Testing, or CI)
|
||||||
|
private var isRunningTests: Bool {
|
||||||
|
let env = ProcessInfo.processInfo.environment
|
||||||
|
return NSClassFromString("XCTestCase") != nil ||
|
||||||
|
env["XCTestConfigurationFilePath"] != nil ||
|
||||||
|
env["XCTestBundlePath"] != nil ||
|
||||||
|
env["GITHUB_ACTIONS"] != nil ||
|
||||||
|
env["CI"] != nil
|
||||||
|
}
|
||||||
|
|
||||||
private init() {}
|
private init() {}
|
||||||
|
|
||||||
func requestAuthorization() {
|
func requestAuthorization() {
|
||||||
|
guard !isRunningTests else { return }
|
||||||
UNUserNotificationCenter.current().requestAuthorization(options: [.alert, .sound, .badge]) { granted, error in
|
UNUserNotificationCenter.current().requestAuthorization(options: [.alert, .sound, .badge]) { granted, error in
|
||||||
if granted {
|
if granted {
|
||||||
// Permission granted
|
// Permission granted
|
||||||
@@ -29,14 +40,20 @@ final class NotificationService {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func sendLocalNotification(title: String, body: String, identifier: String, userInfo: [String: Any]? = nil) {
|
func sendLocalNotification(
|
||||||
// For now, skip app state check entirely to avoid thread issues
|
title: String,
|
||||||
// The NotificationDelegate will handle foreground presentation
|
body: String,
|
||||||
DispatchQueue.main.async {
|
identifier: String,
|
||||||
|
userInfo: [String: Any]? = nil,
|
||||||
|
interruptionLevel: UNNotificationInterruptionLevel = .active
|
||||||
|
) {
|
||||||
|
guard !isRunningTests else { return }
|
||||||
let content = UNMutableNotificationContent()
|
let content = UNMutableNotificationContent()
|
||||||
content.title = title
|
content.title = title
|
||||||
content.body = body
|
content.body = body
|
||||||
content.sound = .default
|
content.sound = .default
|
||||||
|
content.interruptionLevel = interruptionLevel
|
||||||
|
|
||||||
if let userInfo = userInfo {
|
if let userInfo = userInfo {
|
||||||
content.userInfo = userInfo
|
content.userInfo = userInfo
|
||||||
}
|
}
|
||||||
@@ -47,10 +64,7 @@ final class NotificationService {
|
|||||||
trigger: nil // Deliver immediately
|
trigger: nil // Deliver immediately
|
||||||
)
|
)
|
||||||
|
|
||||||
UNUserNotificationCenter.current().add(request) { _ in
|
UNUserNotificationCenter.current().add(request)
|
||||||
// Notification added
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func sendMentionNotification(from sender: String, message: String) {
|
func sendMentionNotification(from sender: String, message: String) {
|
||||||
@@ -61,11 +75,11 @@ final class NotificationService {
|
|||||||
sendLocalNotification(title: title, body: body, identifier: identifier)
|
sendLocalNotification(title: title, body: body, identifier: identifier)
|
||||||
}
|
}
|
||||||
|
|
||||||
func sendPrivateMessageNotification(from sender: String, message: String, peerID: String) {
|
func sendPrivateMessageNotification(from sender: String, message: String, peerID: PeerID) {
|
||||||
let title = "🔒 DM from \(sender)"
|
let title = "🔒 DM from \(sender)"
|
||||||
let body = message
|
let body = message
|
||||||
let identifier = "private-\(UUID().uuidString)"
|
let identifier = "private-\(UUID().uuidString)"
|
||||||
let userInfo = ["peerID": peerID, "senderName": sender]
|
let userInfo = ["peerID": peerID.id, "senderName": sender]
|
||||||
|
|
||||||
sendLocalNotification(title: title, body: body, identifier: identifier, userInfo: userInfo)
|
sendLocalNotification(title: title, body: body, identifier: identifier, userInfo: userInfo)
|
||||||
}
|
}
|
||||||
@@ -84,24 +98,11 @@ final class NotificationService {
|
|||||||
let body = peerCount == 1 ? "1 person around" : "\(peerCount) people around"
|
let body = peerCount == 1 ? "1 person around" : "\(peerCount) people around"
|
||||||
let identifier = "network-available-\(Date().timeIntervalSince1970)"
|
let identifier = "network-available-\(Date().timeIntervalSince1970)"
|
||||||
|
|
||||||
// For network notifications, we want to show them even in foreground
|
sendLocalNotification(
|
||||||
// No app state check - let the notification delegate handle presentation
|
title: title,
|
||||||
DispatchQueue.main.async {
|
body: body,
|
||||||
let content = UNMutableNotificationContent()
|
|
||||||
content.title = title
|
|
||||||
content.body = body
|
|
||||||
content.sound = .default
|
|
||||||
content.interruptionLevel = .timeSensitive // Make it more prominent
|
|
||||||
|
|
||||||
let request = UNNotificationRequest(
|
|
||||||
identifier: identifier,
|
identifier: identifier,
|
||||||
content: content,
|
interruptionLevel: .timeSensitive
|
||||||
trigger: nil // Deliver immediately
|
|
||||||
)
|
)
|
||||||
|
|
||||||
UNUserNotificationCenter.current().add(request) { _ in
|
|
||||||
// Notification added
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,10 +6,19 @@
|
|||||||
// For more information, see <https://unlicense.org>
|
// For more information, see <https://unlicense.org>
|
||||||
//
|
//
|
||||||
|
|
||||||
|
import BitLogger
|
||||||
import Foundation
|
import Foundation
|
||||||
|
|
||||||
struct NotificationStreamAssembler {
|
struct NotificationStreamAssembler {
|
||||||
private var buffer = Data()
|
private var buffer = Data()
|
||||||
|
private var pendingFrameStartedAt: DispatchTime?
|
||||||
|
private var pendingFrameExpectedLength: Int = 0
|
||||||
|
|
||||||
|
private mutating func resetState() {
|
||||||
|
buffer.removeAll(keepingCapacity: false)
|
||||||
|
pendingFrameStartedAt = nil
|
||||||
|
pendingFrameExpectedLength = 0
|
||||||
|
}
|
||||||
|
|
||||||
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) }
|
||||||
@@ -18,64 +27,117 @@ struct NotificationStreamAssembler {
|
|||||||
|
|
||||||
var frames: [Data] = []
|
var frames: [Data] = []
|
||||||
var dropped: [UInt8] = []
|
var dropped: [UInt8] = []
|
||||||
var reset = false
|
var didReset = false
|
||||||
let maxFrameLength = TransportConfig.blePendingWriteBufferCapBytes
|
let now = DispatchTime.now()
|
||||||
|
let maxFrameLength = TransportConfig.bleNotificationAssemblerHardCapBytes
|
||||||
|
let minimumFramePrefix = BinaryProtocol.v1HeaderSize + BinaryProtocol.senderIDSize
|
||||||
|
|
||||||
let minHeaderBytes = 14 // version + type + ttl + timestamp(8) + flags + length(2)
|
if buffer.count > TransportConfig.bleNotificationAssemblerHardCapBytes {
|
||||||
let minFramePrefix = minHeaderBytes + BinaryProtocol.senderIDSize
|
SecureLogger.error("❌ Notification assembler overflow (\(buffer.count) bytes); dropping partial frame", category: .session)
|
||||||
|
resetState()
|
||||||
|
return ([], [], true)
|
||||||
|
}
|
||||||
|
|
||||||
while buffer.count >= minFramePrefix {
|
while buffer.count >= minimumFramePrefix {
|
||||||
guard let first = buffer.first else { break }
|
guard let version = buffer.first else { break }
|
||||||
if first != 1 {
|
guard version == 1 || version == 2 else {
|
||||||
dropped.append(buffer.removeFirst())
|
dropped.append(buffer.removeFirst())
|
||||||
|
pendingFrameStartedAt = nil
|
||||||
|
pendingFrameExpectedLength = 0
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
guard buffer.count >= minHeaderBytes else { break }
|
guard let headerSize = BinaryProtocol.headerSize(for: version) else {
|
||||||
|
dropped.append(buffer.removeFirst())
|
||||||
|
pendingFrameStartedAt = nil
|
||||||
|
pendingFrameExpectedLength = 0
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
let framePrefix = headerSize + BinaryProtocol.senderIDSize
|
||||||
|
guard buffer.count >= framePrefix else { break }
|
||||||
|
|
||||||
let headerBytes = Array(buffer.prefix(minFramePrefix))
|
let flagsIndex = buffer.startIndex + BinaryProtocol.Offsets.flags
|
||||||
guard headerBytes.count == minFramePrefix else { break }
|
guard flagsIndex < buffer.endIndex else { break }
|
||||||
|
let flags = buffer[flagsIndex]
|
||||||
let flags = headerBytes[11]
|
|
||||||
let hasRecipient = (flags & BinaryProtocol.Flags.hasRecipient) != 0
|
let hasRecipient = (flags & BinaryProtocol.Flags.hasRecipient) != 0
|
||||||
let hasSignature = (flags & BinaryProtocol.Flags.hasSignature) != 0
|
let hasSignature = (flags & BinaryProtocol.Flags.hasSignature) != 0
|
||||||
let payloadLen = (Int(headerBytes[12]) << 8) | Int(headerBytes[13])
|
let isCompressed = (flags & BinaryProtocol.Flags.isCompressed) != 0
|
||||||
|
let hasRoute = (version >= 2) && (flags & BinaryProtocol.Flags.hasRoute) != 0
|
||||||
|
|
||||||
var frameLength = minFramePrefix + payloadLen
|
let lengthOffset = 12
|
||||||
|
let payloadLength: Int
|
||||||
|
if version == 2 {
|
||||||
|
let lengthIndex = buffer.startIndex + lengthOffset
|
||||||
|
payloadLength =
|
||||||
|
(Int(buffer[lengthIndex]) << 24) |
|
||||||
|
(Int(buffer[lengthIndex + 1]) << 16) |
|
||||||
|
(Int(buffer[lengthIndex + 2]) << 8) |
|
||||||
|
Int(buffer[lengthIndex + 3])
|
||||||
|
} else {
|
||||||
|
let lengthIndex = buffer.startIndex + lengthOffset
|
||||||
|
payloadLength = (Int(buffer[lengthIndex]) << 8) | Int(buffer[lengthIndex + 1])
|
||||||
|
}
|
||||||
|
|
||||||
|
var frameLength = framePrefix + payloadLength
|
||||||
if hasRecipient { frameLength += BinaryProtocol.recipientIDSize }
|
if hasRecipient { frameLength += BinaryProtocol.recipientIDSize }
|
||||||
if hasSignature { frameLength += BinaryProtocol.signatureSize }
|
if hasSignature { frameLength += BinaryProtocol.signatureSize }
|
||||||
|
|
||||||
|
if hasRoute {
|
||||||
|
let routeCountOffset = framePrefix + (hasRecipient ? BinaryProtocol.recipientIDSize : 0)
|
||||||
|
let routeCountIndex = buffer.startIndex + routeCountOffset
|
||||||
|
guard buffer.count > routeCountOffset else { break }
|
||||||
|
let routeCount = Int(buffer[routeCountIndex])
|
||||||
|
frameLength += 1 + (routeCount * BinaryProtocol.senderIDSize)
|
||||||
|
}
|
||||||
|
|
||||||
|
if isCompressed {
|
||||||
|
let rawLengthFieldBytes = (version == 2) ? 4 : 2
|
||||||
|
if payloadLength < rawLengthFieldBytes {
|
||||||
|
SecureLogger.error("❌ Invalid compressed payload length (\(payloadLength))", category: .session)
|
||||||
|
resetState()
|
||||||
|
didReset = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
guard frameLength > 0, frameLength <= maxFrameLength else {
|
guard frameLength > 0, frameLength <= maxFrameLength else {
|
||||||
buffer.removeAll()
|
SecureLogger.error("❌ Notification frame length \(frameLength) invalid (cap=\(maxFrameLength)); resetting stream", category: .session)
|
||||||
reset = true
|
resetState()
|
||||||
|
didReset = true
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
if buffer.count < frameLength {
|
if buffer.count < frameLength {
|
||||||
// Check if a new frame start exists within the incomplete buffer; if so, drop leading partial bytes.
|
let remaining = frameLength - buffer.count
|
||||||
if let nextStart = buffer.dropFirst().firstIndex(of: 1) {
|
if pendingFrameStartedAt == nil || frameLength != pendingFrameExpectedLength {
|
||||||
let dropCount = buffer.distance(from: buffer.startIndex, to: nextStart)
|
pendingFrameStartedAt = now
|
||||||
if dropCount > 0 {
|
pendingFrameExpectedLength = frameLength
|
||||||
buffer.removeFirst(dropCount)
|
} else if let started = pendingFrameStartedAt {
|
||||||
dropped.append(1) // treat as dropped partial start
|
let elapsed = now.uptimeNanoseconds - started.uptimeNanoseconds
|
||||||
|
let threshold = UInt64(TransportConfig.bleAssemblerStallResetMs) * 1_000_000
|
||||||
|
if elapsed >= threshold {
|
||||||
|
SecureLogger.debug("📉 Resetting notification assembler after waiting \(remaining)B for \(TransportConfig.bleAssemblerStallResetMs)ms", category: .session)
|
||||||
|
resetState()
|
||||||
|
didReset = true
|
||||||
|
} else {
|
||||||
|
SecureLogger.debug("⌛ Waiting for remaining \(remaining)B to complete BLE frame", category: .session)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pendingFrameStartedAt = nil
|
||||||
|
pendingFrameExpectedLength = 0
|
||||||
|
|
||||||
let frame = Data(buffer.prefix(frameLength))
|
let frame = Data(buffer.prefix(frameLength))
|
||||||
frames.append(frame)
|
frames.append(frame)
|
||||||
buffer.removeFirst(frameLength)
|
buffer.removeFirst(frameLength)
|
||||||
}
|
}
|
||||||
|
|
||||||
if !buffer.isEmpty, buffer.allSatisfy({ $0 == 0 }) {
|
if !buffer.isEmpty, buffer.allSatisfy({ $0 == 0 }) {
|
||||||
buffer.removeAll(keepingCapacity: false)
|
resetState()
|
||||||
}
|
}
|
||||||
|
|
||||||
return (frames, dropped, reset)
|
return (frames, dropped, didReset)
|
||||||
}
|
|
||||||
|
|
||||||
mutating func reset() {
|
|
||||||
buffer.removeAll(keepingCapacity: false)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -22,6 +22,8 @@ final class PrivateChatManager: ObservableObject {
|
|||||||
weak var meshService: Transport?
|
weak var meshService: Transport?
|
||||||
// Route acks/receipts via MessageRouter (chooses mesh or Nostr)
|
// Route acks/receipts via MessageRouter (chooses mesh or Nostr)
|
||||||
weak var messageRouter: MessageRouter?
|
weak var messageRouter: MessageRouter?
|
||||||
|
// Peer service for looking up peer info during consolidation
|
||||||
|
weak var unifiedPeerService: UnifiedPeerService?
|
||||||
|
|
||||||
init(meshService: Transport? = nil) {
|
init(meshService: Transport? = nil) {
|
||||||
self.meshService = meshService
|
self.meshService = meshService
|
||||||
@@ -30,6 +32,158 @@ final class PrivateChatManager: ObservableObject {
|
|||||||
// Cap for messages stored per private chat
|
// Cap for messages stored per private chat
|
||||||
private let privateChatCap = TransportConfig.privateChatCap
|
private let privateChatCap = TransportConfig.privateChatCap
|
||||||
|
|
||||||
|
// MARK: - Message Consolidation
|
||||||
|
|
||||||
|
/// Consolidates messages from different peer ID representations into a single chat.
|
||||||
|
/// This ensures messages from stable Noise keys and temporary Nostr peer IDs are merged.
|
||||||
|
/// - Parameters:
|
||||||
|
/// - peerID: The target peer ID to consolidate messages into
|
||||||
|
/// - peerNickname: The peer's display name (lowercased for matching)
|
||||||
|
/// - persistedReadReceipts: The persisted read receipts set from ChatViewModel (UserDefaults-backed)
|
||||||
|
/// - Returns: True if any unread messages were found during consolidation
|
||||||
|
@MainActor
|
||||||
|
func consolidateMessages(for peerID: PeerID, peerNickname: String, persistedReadReceipts: Set<String>) -> Bool {
|
||||||
|
guard let meshService = meshService else { return false }
|
||||||
|
var hasUnreadMessages = false
|
||||||
|
|
||||||
|
// 1. Consolidate from stable Noise key (64-char hex)
|
||||||
|
if let peer = unifiedPeerService?.getPeer(by: peerID) {
|
||||||
|
let noiseKeyHex = PeerID(hexData: peer.noisePublicKey)
|
||||||
|
|
||||||
|
if noiseKeyHex != peerID, let nostrMessages = privateChats[noiseKeyHex], !nostrMessages.isEmpty {
|
||||||
|
if privateChats[peerID] == nil {
|
||||||
|
privateChats[peerID] = []
|
||||||
|
}
|
||||||
|
|
||||||
|
let existingMessageIds = Set(privateChats[peerID]?.map { $0.id } ?? [])
|
||||||
|
for message in nostrMessages {
|
||||||
|
if !existingMessageIds.contains(message.id) {
|
||||||
|
// Update senderPeerID for correct read receipts
|
||||||
|
let updatedMessage = BitchatMessage(
|
||||||
|
id: message.id,
|
||||||
|
sender: message.sender,
|
||||||
|
content: message.content,
|
||||||
|
timestamp: message.timestamp,
|
||||||
|
isRelay: message.isRelay,
|
||||||
|
originalSender: message.originalSender,
|
||||||
|
isPrivate: message.isPrivate,
|
||||||
|
recipientNickname: message.recipientNickname,
|
||||||
|
senderPeerID: message.senderPeerID == meshService.myPeerID ? meshService.myPeerID : peerID,
|
||||||
|
mentions: message.mentions,
|
||||||
|
deliveryStatus: message.deliveryStatus
|
||||||
|
)
|
||||||
|
privateChats[peerID]?.append(updatedMessage)
|
||||||
|
|
||||||
|
// Check for recent unread messages (< 60s, not sent by us, not already read)
|
||||||
|
// Use persistedReadReceipts to correctly identify already-read messages after app restart
|
||||||
|
if message.senderPeerID != meshService.myPeerID {
|
||||||
|
let messageAge = Date().timeIntervalSince(message.timestamp)
|
||||||
|
if messageAge < 60 && !persistedReadReceipts.contains(message.id) {
|
||||||
|
hasUnreadMessages = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
privateChats[peerID]?.sort { $0.timestamp < $1.timestamp }
|
||||||
|
|
||||||
|
if hasUnreadMessages {
|
||||||
|
unreadMessages.insert(peerID)
|
||||||
|
} else if unreadMessages.contains(noiseKeyHex) {
|
||||||
|
unreadMessages.remove(noiseKeyHex)
|
||||||
|
}
|
||||||
|
|
||||||
|
privateChats.removeValue(forKey: noiseKeyHex)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Consolidate from temporary Nostr peer IDs (nostr_* prefixed)
|
||||||
|
let normalizedNickname = peerNickname.lowercased()
|
||||||
|
var tempPeerIDsToConsolidate: [PeerID] = []
|
||||||
|
|
||||||
|
for (storedPeerID, messages) in privateChats {
|
||||||
|
if storedPeerID.isGeoDM && storedPeerID != peerID {
|
||||||
|
let nicknamesMatch = messages.allSatisfy { $0.sender.lowercased() == normalizedNickname }
|
||||||
|
if nicknamesMatch && !messages.isEmpty {
|
||||||
|
tempPeerIDsToConsolidate.append(storedPeerID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !tempPeerIDsToConsolidate.isEmpty {
|
||||||
|
if privateChats[peerID] == nil {
|
||||||
|
privateChats[peerID] = []
|
||||||
|
}
|
||||||
|
|
||||||
|
let existingMessageIds = Set(privateChats[peerID]?.map { $0.id } ?? [])
|
||||||
|
var consolidatedCount = 0
|
||||||
|
var hadUnreadTemp = false
|
||||||
|
|
||||||
|
for tempPeerID in tempPeerIDsToConsolidate {
|
||||||
|
if unreadMessages.contains(tempPeerID) {
|
||||||
|
hadUnreadTemp = true
|
||||||
|
}
|
||||||
|
|
||||||
|
if let tempMessages = privateChats[tempPeerID] {
|
||||||
|
for message in tempMessages {
|
||||||
|
if !existingMessageIds.contains(message.id) {
|
||||||
|
let updatedMessage = BitchatMessage(
|
||||||
|
id: message.id,
|
||||||
|
sender: message.sender,
|
||||||
|
content: message.content,
|
||||||
|
timestamp: message.timestamp,
|
||||||
|
isRelay: message.isRelay,
|
||||||
|
originalSender: message.originalSender,
|
||||||
|
isPrivate: message.isPrivate,
|
||||||
|
recipientNickname: message.recipientNickname,
|
||||||
|
senderPeerID: peerID,
|
||||||
|
mentions: message.mentions,
|
||||||
|
deliveryStatus: message.deliveryStatus
|
||||||
|
)
|
||||||
|
privateChats[peerID]?.append(updatedMessage)
|
||||||
|
consolidatedCount += 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
privateChats.removeValue(forKey: tempPeerID)
|
||||||
|
unreadMessages.remove(tempPeerID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if hadUnreadTemp {
|
||||||
|
unreadMessages.insert(peerID)
|
||||||
|
hasUnreadMessages = true
|
||||||
|
SecureLogger.debug("📬 Transferred unread status from temp peer IDs to \(peerID)", category: .session)
|
||||||
|
}
|
||||||
|
|
||||||
|
if consolidatedCount > 0 {
|
||||||
|
privateChats[peerID]?.sort { $0.timestamp < $1.timestamp }
|
||||||
|
SecureLogger.info("📥 Consolidated \(consolidatedCount) Nostr messages from temporary peer IDs to \(peerNickname)", category: .session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return hasUnreadMessages
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Syncs the read receipt tracking between manager and view model for sent messages
|
||||||
|
@MainActor
|
||||||
|
func syncReadReceiptsForSentMessages(peerID: PeerID, nickname: String, externalReceipts: inout Set<String>) {
|
||||||
|
guard let messages = privateChats[peerID] else { return }
|
||||||
|
|
||||||
|
for message in messages {
|
||||||
|
if message.sender == nickname {
|
||||||
|
if let status = message.deliveryStatus {
|
||||||
|
switch status {
|
||||||
|
case .read, .delivered:
|
||||||
|
externalReceipts.insert(message.id)
|
||||||
|
sentReadReceipts.insert(message.id)
|
||||||
|
case .failed, .partiallyDelivered, .sending, .sent:
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Start a private chat with a peer
|
/// Start a private chat with a peer
|
||||||
func startChat(with peerID: PeerID) {
|
func startChat(with peerID: PeerID) {
|
||||||
selectedPeer = peerID
|
selectedPeer = peerID
|
||||||
@@ -105,7 +259,7 @@ final class PrivateChatManager: ObservableObject {
|
|||||||
// 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.id ?? "",
|
readerID: meshService?.myPeerID ?? PeerID(str: ""),
|
||||||
readerNickname: meshService?.myNickname ?? ""
|
readerNickname: meshService?.myNickname ?? ""
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -13,6 +13,7 @@ struct RelayController {
|
|||||||
senderIsSelf: Bool,
|
senderIsSelf: Bool,
|
||||||
isEncrypted: Bool,
|
isEncrypted: Bool,
|
||||||
isDirectedEncrypted: Bool,
|
isDirectedEncrypted: Bool,
|
||||||
|
isFragment: Bool,
|
||||||
isDirectedFragment: Bool,
|
isDirectedFragment: Bool,
|
||||||
isHandshake: Bool,
|
isHandshake: Bool,
|
||||||
isAnnounce: Bool,
|
isAnnounce: Bool,
|
||||||
@@ -36,6 +37,16 @@ struct RelayController {
|
|||||||
return RelayDecision(shouldRelay: true, newTTL: newTTL, delayMs: delayMs)
|
return RelayDecision(shouldRelay: true, newTTL: newTTL, delayMs: delayMs)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if isFragment {
|
||||||
|
let ttlLimit = min(ttlCap, TransportConfig.bleFragmentRelayTtlCap)
|
||||||
|
guard ttlLimit > 1 else {
|
||||||
|
return RelayDecision(shouldRelay: false, newTTL: ttlLimit, delayMs: 0)
|
||||||
|
}
|
||||||
|
let newTTL = ttlLimit &- 1
|
||||||
|
let delayMs = Int.random(in: TransportConfig.bleFragmentRelayMinDelayMs...TransportConfig.bleFragmentRelayMaxDelayMs)
|
||||||
|
return RelayDecision(shouldRelay: true, newTTL: newTTL, delayMs: delayMs)
|
||||||
|
}
|
||||||
|
|
||||||
// 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
|
||||||
// - Announces get a bit more headroom
|
// - Announces get a bit more headroom
|
||||||
|
|||||||
@@ -0,0 +1,65 @@
|
|||||||
|
import Foundation
|
||||||
|
import Combine
|
||||||
|
|
||||||
|
/// Centralized progress bus for Bluetooth file transfers.
|
||||||
|
/// Emits Combine events consumed by ChatViewModel to update UI progress indicators.
|
||||||
|
final class TransferProgressManager {
|
||||||
|
static let shared = TransferProgressManager()
|
||||||
|
|
||||||
|
enum Event {
|
||||||
|
case started(id: String, totalFragments: Int)
|
||||||
|
case updated(id: String, sentFragments: Int, totalFragments: Int)
|
||||||
|
case completed(id: String, totalFragments: Int)
|
||||||
|
case cancelled(id: String, sentFragments: Int, totalFragments: Int)
|
||||||
|
}
|
||||||
|
|
||||||
|
private let subject = PassthroughSubject<Event, Never>()
|
||||||
|
private let queue = DispatchQueue(label: "com.bitchat.transfer-progress", attributes: .concurrent)
|
||||||
|
private var states: [String: (sent: Int, total: Int)] = [:]
|
||||||
|
|
||||||
|
var publisher: AnyPublisher<Event, Never> {
|
||||||
|
subject.eraseToAnyPublisher()
|
||||||
|
}
|
||||||
|
|
||||||
|
func start(id: String, totalFragments: Int) {
|
||||||
|
queue.async(flags: .barrier) { [weak self] in
|
||||||
|
guard let self = self else { return }
|
||||||
|
self.states[id] = (sent: 0, total: totalFragments)
|
||||||
|
self.subject.send(.started(id: id, totalFragments: totalFragments))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func recordFragmentSent(id: String) {
|
||||||
|
queue.async(flags: .barrier) { [weak self] in
|
||||||
|
guard let self = self, var state = self.states[id] else { return }
|
||||||
|
state.sent = min(state.sent + 1, state.total)
|
||||||
|
self.states[id] = state
|
||||||
|
self.subject.send(.updated(id: id, sentFragments: state.sent, totalFragments: state.total))
|
||||||
|
if state.sent >= state.total {
|
||||||
|
self.states.removeValue(forKey: id)
|
||||||
|
self.subject.send(.completed(id: id, totalFragments: state.total))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func cancel(id: String) {
|
||||||
|
queue.async(flags: .barrier) { [weak self] in
|
||||||
|
guard let self = self, let state = self.states.removeValue(forKey: id) else { return }
|
||||||
|
self.subject.send(.cancelled(id: id, sentFragments: state.sent, totalFragments: state.total))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func reset(id: String) {
|
||||||
|
queue.async(flags: .barrier) { [weak self] in
|
||||||
|
self?.states.removeValue(forKey: id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func snapshot(id: String) -> (sent: Int, total: Int)? {
|
||||||
|
var result: (sent: Int, total: Int)?
|
||||||
|
queue.sync {
|
||||||
|
result = states[id]
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -45,20 +45,38 @@ protocol Transport: AnyObject {
|
|||||||
|
|
||||||
// Messaging
|
// Messaging
|
||||||
func sendMessage(_ content: String, mentions: [String])
|
func sendMessage(_ content: String, mentions: [String])
|
||||||
|
func sendMessage(_ content: String, mentions: [String], messageID: String, timestamp: Date)
|
||||||
func sendPrivateMessage(_ content: String, to peerID: PeerID, recipientNickname: String, messageID: String)
|
func sendPrivateMessage(_ content: String, to peerID: PeerID, recipientNickname: String, messageID: String)
|
||||||
func sendReadReceipt(_ receipt: ReadReceipt, to peerID: PeerID)
|
func sendReadReceipt(_ receipt: ReadReceipt, to peerID: PeerID)
|
||||||
func sendFavoriteNotification(to peerID: PeerID, isFavorite: Bool)
|
func sendFavoriteNotification(to peerID: PeerID, isFavorite: Bool)
|
||||||
func sendBroadcastAnnounce()
|
func sendBroadcastAnnounce()
|
||||||
func sendDeliveryAck(for messageID: String, to peerID: PeerID)
|
func sendDeliveryAck(for messageID: String, to peerID: PeerID)
|
||||||
|
func sendFileBroadcast(_ packet: BitchatFilePacket, transferId: String)
|
||||||
|
func sendFilePrivate(_ packet: BitchatFilePacket, to peerID: PeerID, transferId: String)
|
||||||
|
func cancelTransfer(_ transferId: String)
|
||||||
|
|
||||||
// QR verification (optional for transports)
|
// QR verification (optional for transports)
|
||||||
func sendVerifyChallenge(to peerID: PeerID, noiseKeyHex: String, nonceA: Data)
|
func sendVerifyChallenge(to peerID: PeerID, noiseKeyHex: String, nonceA: Data)
|
||||||
func sendVerifyResponse(to peerID: PeerID, noiseKeyHex: String, nonceA: Data)
|
func sendVerifyResponse(to peerID: PeerID, noiseKeyHex: String, nonceA: Data)
|
||||||
|
|
||||||
|
// Pending file management (BCH-01-002: files held in memory until user accepts)
|
||||||
|
func acceptPendingFile(id: String) -> URL?
|
||||||
|
func declinePendingFile(id: String)
|
||||||
}
|
}
|
||||||
|
|
||||||
extension Transport {
|
extension Transport {
|
||||||
func sendVerifyChallenge(to peerID: PeerID, noiseKeyHex: String, nonceA: Data) {}
|
func sendVerifyChallenge(to peerID: PeerID, noiseKeyHex: String, nonceA: Data) {}
|
||||||
func sendVerifyResponse(to peerID: PeerID, noiseKeyHex: String, nonceA: Data) {}
|
func sendVerifyResponse(to peerID: PeerID, noiseKeyHex: String, nonceA: Data) {}
|
||||||
|
func sendFileBroadcast(_ packet: BitchatFilePacket, transferId: String) {}
|
||||||
|
func sendFilePrivate(_ packet: BitchatFilePacket, to peerID: PeerID, transferId: String) {}
|
||||||
|
func cancelTransfer(_ transferId: String) {}
|
||||||
|
|
||||||
|
func sendMessage(_ content: String, mentions: [String], messageID: String, timestamp: Date) {
|
||||||
|
sendMessage(content, mentions: mentions)
|
||||||
|
}
|
||||||
|
|
||||||
|
func acceptPendingFile(id: String) -> URL? { nil }
|
||||||
|
func declinePendingFile(id: String) {}
|
||||||
}
|
}
|
||||||
|
|
||||||
protocol TransportPeerEventsDelegate: AnyObject {
|
protocol TransportPeerEventsDelegate: AnyObject {
|
||||||
|
|||||||
@@ -8,6 +8,10 @@ enum TransportConfig {
|
|||||||
static let messageTTLDefault: UInt8 = 7 // Default TTL for mesh flooding
|
static let messageTTLDefault: UInt8 = 7 // Default TTL for mesh flooding
|
||||||
static let bleMaxInFlightAssemblies: Int = 128 // Cap concurrent fragment assemblies
|
static let bleMaxInFlightAssemblies: Int = 128 // Cap concurrent fragment assemblies
|
||||||
static let bleHighDegreeThreshold: Int = 6 // For adaptive TTL/probabilistic relays
|
static let bleHighDegreeThreshold: Int = 6 // For adaptive TTL/probabilistic relays
|
||||||
|
static let bleMaxConcurrentTransfers: Int = 2 // Limit simultaneous large media sends
|
||||||
|
static let bleFragmentRelayMinDelayMs: Int = 8 // Faster forwarding for media fragments
|
||||||
|
static let bleFragmentRelayMaxDelayMs: Int = 25 // Upper jitter bound for fragment relays
|
||||||
|
static let bleFragmentRelayTtlCap: UInt8 = 5 // Clamp fragment TTL to contain floods
|
||||||
|
|
||||||
// UI / Storage Caps
|
// UI / Storage Caps
|
||||||
static let privateChatCap: Int = 1337
|
static let privateChatCap: Int = 1337
|
||||||
@@ -30,7 +34,11 @@ enum TransportConfig {
|
|||||||
static let bleDynamicRSSIThresholdDefault: Int = -90
|
static let bleDynamicRSSIThresholdDefault: Int = -90
|
||||||
static let bleConnectionCandidatesMax: Int = 100
|
static let bleConnectionCandidatesMax: Int = 100
|
||||||
static let blePendingWriteBufferCapBytes: Int = 1_000_000
|
static let blePendingWriteBufferCapBytes: Int = 1_000_000
|
||||||
static let blePendingNotificationsCapCount: Int = 20
|
static let bleNotificationAssemblerHardCapBytes: Int = 8 * 1024 * 1024
|
||||||
|
static let bleAssemblerStallResetMs: Int = 250
|
||||||
|
static let blePendingNotificationsCapCount: Int = 128
|
||||||
|
static let bleNotificationRetryDelayMs: Int = 25
|
||||||
|
static let bleNotificationRetryMaxAttempts: Int = 80
|
||||||
|
|
||||||
// Nostr
|
// Nostr
|
||||||
static let nostrReadAckInterval: TimeInterval = 0.35 // ~3 per second
|
static let nostrReadAckInterval: TimeInterval = 0.35 // ~3 per second
|
||||||
@@ -62,6 +70,7 @@ enum TransportConfig {
|
|||||||
static let uiAnimationMediumSeconds: TimeInterval = 0.2
|
static let uiAnimationMediumSeconds: TimeInterval = 0.2
|
||||||
static let uiAnimationSidebarSeconds: TimeInterval = 0.25
|
static let uiAnimationSidebarSeconds: TimeInterval = 0.25
|
||||||
static let uiRecentCutoffFiveMinutesSeconds: TimeInterval = 5 * 60
|
static let uiRecentCutoffFiveMinutesSeconds: TimeInterval = 5 * 60
|
||||||
|
static let uiMeshEmptyConfirmationSeconds: TimeInterval = 30.0
|
||||||
|
|
||||||
// BLE maintenance & thresholds
|
// BLE maintenance & thresholds
|
||||||
static let bleMaintenanceInterval: TimeInterval = 5.0
|
static let bleMaintenanceInterval: TimeInterval = 5.0
|
||||||
@@ -141,6 +150,9 @@ enum TransportConfig {
|
|||||||
|
|
||||||
// Geo relay directory
|
// Geo relay directory
|
||||||
static let geoRelayFetchIntervalSeconds: TimeInterval = 60 * 60 * 24
|
static let geoRelayFetchIntervalSeconds: TimeInterval = 60 * 60 * 24
|
||||||
|
static let geoRelayRefreshCheckIntervalSeconds: TimeInterval = 60 * 60
|
||||||
|
static let geoRelayRetryInitialSeconds: TimeInterval = 60
|
||||||
|
static let geoRelayRetryMaxSeconds: TimeInterval = 60 * 60
|
||||||
|
|
||||||
// BLE operational delays
|
// BLE operational delays
|
||||||
static let bleInitialAnnounceDelaySeconds: TimeInterval = 0.6
|
static let bleInitialAnnounceDelaySeconds: TimeInterval = 0.6
|
||||||
@@ -150,6 +162,14 @@ enum TransportConfig {
|
|||||||
static let blePostAnnounceDelaySeconds: TimeInterval = 0.4
|
static let blePostAnnounceDelaySeconds: TimeInterval = 0.4
|
||||||
static let bleForceAnnounceMinIntervalSeconds: TimeInterval = 0.15
|
static let bleForceAnnounceMinIntervalSeconds: TimeInterval = 0.15
|
||||||
|
|
||||||
|
// BCH-01-004: Rate-limiting for subscription-triggered announces
|
||||||
|
// Prevents rapid enumeration attacks by rate-limiting announce responses
|
||||||
|
static let bleSubscriptionRateLimitMinSeconds: TimeInterval = 2.0 // Minimum interval between announces per central
|
||||||
|
static let bleSubscriptionRateLimitBackoffFactor: Double = 2.0 // Exponential backoff multiplier
|
||||||
|
static let bleSubscriptionRateLimitMaxBackoffSeconds: TimeInterval = 30.0 // Maximum backoff period
|
||||||
|
static let bleSubscriptionRateLimitWindowSeconds: TimeInterval = 60.0 // Window for tracking subscription attempts
|
||||||
|
static let bleSubscriptionRateLimitMaxAttempts: Int = 5 // Max attempts before extended cooldown
|
||||||
|
|
||||||
// Store-and-forward for directed packets at relays
|
// Store-and-forward for directed packets at relays
|
||||||
static let bleDirectedSpoolWindowSeconds: TimeInterval = 15.0
|
static let bleDirectedSpoolWindowSeconds: TimeInterval = 15.0
|
||||||
|
|
||||||
|
|||||||
@@ -8,6 +8,55 @@ final class GossipSyncManager {
|
|||||||
func signPacketForBroadcast(_ packet: BitchatPacket) -> BitchatPacket
|
func signPacketForBroadcast(_ packet: BitchatPacket) -> BitchatPacket
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private struct PacketStore {
|
||||||
|
private(set) var packets: [String: BitchatPacket] = [:]
|
||||||
|
private(set) var order: [String] = []
|
||||||
|
|
||||||
|
mutating func insert(idHex: String, packet: BitchatPacket, capacity: Int) {
|
||||||
|
guard capacity > 0 else { return }
|
||||||
|
if packets[idHex] != nil {
|
||||||
|
packets[idHex] = packet
|
||||||
|
return
|
||||||
|
}
|
||||||
|
packets[idHex] = packet
|
||||||
|
order.append(idHex)
|
||||||
|
while order.count > capacity {
|
||||||
|
let victim = order.removeFirst()
|
||||||
|
packets.removeValue(forKey: victim)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func allPackets(isFresh: (BitchatPacket) -> Bool) -> [BitchatPacket] {
|
||||||
|
order.compactMap { key in
|
||||||
|
guard let packet = packets[key], isFresh(packet) else { return nil }
|
||||||
|
return packet
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func remove(where shouldRemove: (BitchatPacket) -> Bool) {
|
||||||
|
var nextOrder: [String] = []
|
||||||
|
for key in order {
|
||||||
|
guard let packet = packets[key] else { continue }
|
||||||
|
if shouldRemove(packet) {
|
||||||
|
packets.removeValue(forKey: key)
|
||||||
|
} else {
|
||||||
|
nextOrder.append(key)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
order = nextOrder
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeExpired(isFresh: (BitchatPacket) -> Bool) {
|
||||||
|
remove { !isFresh($0) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private struct SyncSchedule {
|
||||||
|
let types: SyncTypeFlags
|
||||||
|
let interval: TimeInterval
|
||||||
|
var lastSent: Date
|
||||||
|
}
|
||||||
|
|
||||||
struct Config {
|
struct Config {
|
||||||
var seenCapacity: Int = 1000 // max packets per sync (cap across types)
|
var seenCapacity: Int = 1000 // max packets per sync (cap across types)
|
||||||
var gcsMaxBytes: Int = 400 // filter size budget (128..1024)
|
var gcsMaxBytes: Int = 400 // filter size budget (128..1024)
|
||||||
@@ -16,25 +65,43 @@ final class GossipSyncManager {
|
|||||||
var maintenanceIntervalSeconds: TimeInterval = 30.0
|
var maintenanceIntervalSeconds: TimeInterval = 30.0
|
||||||
var stalePeerCleanupIntervalSeconds: TimeInterval = 60.0
|
var stalePeerCleanupIntervalSeconds: TimeInterval = 60.0
|
||||||
var stalePeerTimeoutSeconds: TimeInterval = 60.0
|
var stalePeerTimeoutSeconds: TimeInterval = 60.0
|
||||||
|
var fragmentCapacity: Int = 600
|
||||||
|
var fileTransferCapacity: Int = 200
|
||||||
|
var fragmentSyncIntervalSeconds: TimeInterval = 30.0
|
||||||
|
var fileTransferSyncIntervalSeconds: TimeInterval = 60.0
|
||||||
|
var messageSyncIntervalSeconds: TimeInterval = 15.0
|
||||||
}
|
}
|
||||||
|
|
||||||
private let myPeerID: PeerID
|
private let myPeerID: PeerID
|
||||||
private let config: Config
|
private let config: Config
|
||||||
weak var delegate: Delegate?
|
weak var delegate: Delegate?
|
||||||
|
|
||||||
// Storage: broadcast messages (ordered by insert), and latest announce per sender
|
// Storage: broadcast packets by type, and latest announce per sender
|
||||||
private var messages: [String: BitchatPacket] = [:] // idHex -> packet
|
private var messages = PacketStore()
|
||||||
private var messageOrder: [String] = []
|
private var fragments = PacketStore()
|
||||||
private var latestAnnouncementByPeer: [String: (id: String, packet: BitchatPacket)] = [:]
|
private var fileTransfers = PacketStore()
|
||||||
|
private var latestAnnouncementByPeer: [PeerID: (id: String, packet: BitchatPacket)] = [:]
|
||||||
|
|
||||||
// Timer
|
// Timer
|
||||||
private var periodicTimer: DispatchSourceTimer?
|
private var periodicTimer: DispatchSourceTimer?
|
||||||
private let queue = DispatchQueue(label: "mesh.sync", qos: .utility)
|
private let queue = DispatchQueue(label: "mesh.sync", qos: .utility)
|
||||||
private var lastStalePeerCleanup: Date = .distantPast
|
private var lastStalePeerCleanup: Date = .distantPast
|
||||||
|
private var syncSchedules: [SyncSchedule] = []
|
||||||
|
|
||||||
init(myPeerID: PeerID, config: Config = Config()) {
|
init(myPeerID: PeerID, config: Config = Config()) {
|
||||||
self.myPeerID = myPeerID
|
self.myPeerID = myPeerID
|
||||||
self.config = config
|
self.config = config
|
||||||
|
var schedules: [SyncSchedule] = []
|
||||||
|
if config.seenCapacity > 0 && config.messageSyncIntervalSeconds > 0 {
|
||||||
|
schedules.append(SyncSchedule(types: .publicMessages, interval: config.messageSyncIntervalSeconds, lastSent: .distantPast))
|
||||||
|
}
|
||||||
|
if config.fragmentCapacity > 0 && config.fragmentSyncIntervalSeconds > 0 {
|
||||||
|
schedules.append(SyncSchedule(types: .fragment, interval: config.fragmentSyncIntervalSeconds, lastSent: .distantPast))
|
||||||
|
}
|
||||||
|
if config.fileTransferCapacity > 0 && config.fileTransferSyncIntervalSeconds > 0 {
|
||||||
|
schedules.append(SyncSchedule(types: .fileTransfer, interval: config.fileTransferSyncIntervalSeconds, lastSent: .distantPast))
|
||||||
|
}
|
||||||
|
syncSchedules = schedules
|
||||||
}
|
}
|
||||||
|
|
||||||
func start() {
|
func start() {
|
||||||
@@ -55,7 +122,18 @@ final class GossipSyncManager {
|
|||||||
|
|
||||||
func scheduleInitialSyncToPeer(_ peerID: PeerID, delaySeconds: TimeInterval = 5.0) {
|
func scheduleInitialSyncToPeer(_ peerID: PeerID, delaySeconds: TimeInterval = 5.0) {
|
||||||
queue.asyncAfter(deadline: .now() + delaySeconds) { [weak self] in
|
queue.asyncAfter(deadline: .now() + delaySeconds) { [weak self] in
|
||||||
self?.sendRequestSync(to: peerID)
|
guard let self = self else { return }
|
||||||
|
self.sendRequestSync(to: peerID, types: .publicMessages)
|
||||||
|
if self.config.fragmentCapacity > 0 && self.config.fragmentSyncIntervalSeconds > 0 {
|
||||||
|
self.queue.asyncAfter(deadline: .now() + 0.5) { [weak self] in
|
||||||
|
self?.sendRequestSync(to: peerID, types: .fragment)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if self.config.fileTransferCapacity > 0 && self.config.fileTransferSyncIntervalSeconds > 0 {
|
||||||
|
self.queue.asyncAfter(deadline: .now() + 1.0) { [weak self] in
|
||||||
|
self?.sendRequestSync(to: peerID, types: .fileTransfer)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -87,47 +165,45 @@ final class GossipSyncManager {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private func _onPublicPacketSeen(_ packet: BitchatPacket) {
|
private func _onPublicPacketSeen(_ packet: BitchatPacket) {
|
||||||
let mt = MessageType(rawValue: packet.type)
|
guard let messageType = MessageType(rawValue: packet.type) else { return }
|
||||||
let isBroadcastRecipient: Bool = {
|
let isBroadcastRecipient: Bool = {
|
||||||
guard let r = packet.recipientID else { return true }
|
guard let r = packet.recipientID else { return true }
|
||||||
return r.count == 8 && r.allSatisfy { $0 == 0xFF }
|
return r.count == 8 && r.allSatisfy { $0 == 0xFF }
|
||||||
}()
|
}()
|
||||||
let isBroadcastMessage = (mt == .message && isBroadcastRecipient)
|
|
||||||
let isAnnounce = (mt == .announce)
|
|
||||||
guard isBroadcastMessage || isAnnounce else { return }
|
|
||||||
|
|
||||||
// Reject expired packets to prevent ghost peers and old messages
|
switch messageType {
|
||||||
|
case .announce:
|
||||||
guard isPacketFresh(packet) else { return }
|
guard isPacketFresh(packet) else { return }
|
||||||
|
|
||||||
if isAnnounce {
|
|
||||||
guard isAnnouncementFresh(packet) else {
|
guard isAnnouncementFresh(packet) else {
|
||||||
let sender = packet.senderID.hexEncodedString().lowercased()
|
let sender = PeerID(hexData: packet.senderID)
|
||||||
removeState(forNormalizedPeerID: sender)
|
removeState(for: sender)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
let idHex = PacketIdUtil.computeId(packet).hexEncodedString()
|
let idHex = PacketIdUtil.computeId(packet).hexEncodedString()
|
||||||
|
let sender = PeerID(hexData: packet.senderID)
|
||||||
if isBroadcastMessage {
|
|
||||||
if messages[idHex] == nil {
|
|
||||||
messages[idHex] = packet
|
|
||||||
messageOrder.append(idHex)
|
|
||||||
// Enforce capacity
|
|
||||||
let cap = max(1, config.seenCapacity)
|
|
||||||
while messageOrder.count > cap {
|
|
||||||
let victim = messageOrder.removeFirst()
|
|
||||||
messages.removeValue(forKey: victim)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if isAnnounce {
|
|
||||||
let sender = packet.senderID.hexEncodedString().lowercased()
|
|
||||||
latestAnnouncementByPeer[sender] = (id: idHex, packet: packet)
|
latestAnnouncementByPeer[sender] = (id: idHex, packet: packet)
|
||||||
|
case .message:
|
||||||
|
guard isBroadcastRecipient else { return }
|
||||||
|
guard isPacketFresh(packet) else { return }
|
||||||
|
let idHex = PacketIdUtil.computeId(packet).hexEncodedString()
|
||||||
|
messages.insert(idHex: idHex, packet: packet, capacity: max(1, config.seenCapacity))
|
||||||
|
case .fragment:
|
||||||
|
guard isBroadcastRecipient else { return }
|
||||||
|
guard isPacketFresh(packet) else { return }
|
||||||
|
let idHex = PacketIdUtil.computeId(packet).hexEncodedString()
|
||||||
|
fragments.insert(idHex: idHex, packet: packet, capacity: max(1, config.fragmentCapacity))
|
||||||
|
case .fileTransfer:
|
||||||
|
guard isBroadcastRecipient else { return }
|
||||||
|
guard isPacketFresh(packet) else { return }
|
||||||
|
let idHex = PacketIdUtil.computeId(packet).hexEncodedString()
|
||||||
|
fileTransfers.insert(idHex: idHex, packet: packet, capacity: max(1, config.fileTransferCapacity))
|
||||||
|
default:
|
||||||
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func sendRequestSync() {
|
private func sendRequestSync(for types: SyncTypeFlags) {
|
||||||
let payload = buildGcsPayload()
|
let payload = buildGcsPayload(for: types)
|
||||||
let pkt = BitchatPacket(
|
let pkt = BitchatPacket(
|
||||||
type: MessageType.requestSync.rawValue,
|
type: MessageType.requestSync.rawValue,
|
||||||
senderID: Data(hexString: myPeerID.id) ?? Data(),
|
senderID: Data(hexString: myPeerID.id) ?? Data(),
|
||||||
@@ -141,8 +217,8 @@ final class GossipSyncManager {
|
|||||||
delegate?.sendPacket(signed)
|
delegate?.sendPacket(signed)
|
||||||
}
|
}
|
||||||
|
|
||||||
private func sendRequestSync(to peerID: PeerID) {
|
private func sendRequestSync(to peerID: PeerID, types: SyncTypeFlags) {
|
||||||
let payload = buildGcsPayload()
|
let payload = buildGcsPayload(for: types)
|
||||||
var recipient = Data()
|
var recipient = Data()
|
||||||
var temp = peerID.id
|
var temp = peerID.id
|
||||||
while temp.count >= 2 && recipient.count < 8 {
|
while temp.count >= 2 && recipient.count < 8 {
|
||||||
@@ -170,6 +246,7 @@ final class GossipSyncManager {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private func _handleRequestSync(from peerID: PeerID, request: RequestSyncPacket) {
|
private func _handleRequestSync(from peerID: PeerID, request: RequestSyncPacket) {
|
||||||
|
let requestedTypes = (request.types ?? .publicMessages)
|
||||||
// Decode GCS into sorted set and prepare membership checker
|
// Decode GCS into sorted set and prepare membership checker
|
||||||
let sorted = GCSFilter.decodeToSortedSet(p: request.p, m: request.m, data: request.data)
|
let sorted = GCSFilter.decodeToSortedSet(p: request.p, m: request.m, data: request.data)
|
||||||
func mightContain(_ id: Data) -> Bool {
|
func mightContain(_ id: Data) -> Bool {
|
||||||
@@ -177,7 +254,7 @@ final class GossipSyncManager {
|
|||||||
return GCSFilter.contains(sortedValues: sorted, candidate: bucket)
|
return GCSFilter.contains(sortedValues: sorted, candidate: bucket)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 1) Announcements: send latest per peer if requester lacks them (and not expired)
|
if requestedTypes.contains(.announce) {
|
||||||
for (_, pair) in latestAnnouncementByPeer {
|
for (_, pair) in latestAnnouncementByPeer {
|
||||||
let (idHex, pkt) = pair
|
let (idHex, pkt) = pair
|
||||||
guard isPacketFresh(pkt) else { continue }
|
guard isPacketFresh(pkt) else { continue }
|
||||||
@@ -188,11 +265,11 @@ final class GossipSyncManager {
|
|||||||
delegate?.sendPacket(to: peerID, packet: toSend)
|
delegate?.sendPacket(to: peerID, packet: toSend)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// 2) Broadcast messages: send all missing (and not expired)
|
if requestedTypes.contains(.message) {
|
||||||
let toSendMsgs = messageOrder.compactMap { messages[$0] }
|
let toSendMsgs = messages.allPackets(isFresh: isPacketFresh)
|
||||||
for pkt in toSendMsgs {
|
for pkt in toSendMsgs {
|
||||||
guard isPacketFresh(pkt) else { continue }
|
|
||||||
let idBytes = PacketIdUtil.computeId(pkt)
|
let idBytes = PacketIdUtil.computeId(pkt)
|
||||||
if !mightContain(idBytes) {
|
if !mightContain(idBytes) {
|
||||||
var toSend = pkt
|
var toSend = pkt
|
||||||
@@ -202,35 +279,75 @@ final class GossipSyncManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if requestedTypes.contains(.fragment) {
|
||||||
|
let frags = fragments.allPackets(isFresh: isPacketFresh)
|
||||||
|
for pkt in frags {
|
||||||
|
let idBytes = PacketIdUtil.computeId(pkt)
|
||||||
|
if !mightContain(idBytes) {
|
||||||
|
var toSend = pkt
|
||||||
|
toSend.ttl = 0
|
||||||
|
delegate?.sendPacket(to: peerID, packet: toSend)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if requestedTypes.contains(.fileTransfer) {
|
||||||
|
let files = fileTransfers.allPackets(isFresh: isPacketFresh)
|
||||||
|
for pkt in files {
|
||||||
|
let idBytes = PacketIdUtil.computeId(pkt)
|
||||||
|
if !mightContain(idBytes) {
|
||||||
|
var toSend = pkt
|
||||||
|
toSend.ttl = 0
|
||||||
|
delegate?.sendPacket(to: peerID, packet: toSend)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Build REQUEST_SYNC payload using current candidates and GCS params
|
// Build REQUEST_SYNC payload using current candidates and GCS params
|
||||||
private func buildGcsPayload() -> Data {
|
private func buildGcsPayload(for types: SyncTypeFlags) -> Data {
|
||||||
// Collect candidates: latest announce per peer + broadcast messages (only fresh)
|
|
||||||
var candidates: [BitchatPacket] = []
|
var candidates: [BitchatPacket] = []
|
||||||
candidates.reserveCapacity(latestAnnouncementByPeer.count + messageOrder.count)
|
if types.contains(.announce) {
|
||||||
for (_, pair) in latestAnnouncementByPeer {
|
for (_, pair) in latestAnnouncementByPeer where isPacketFresh(pair.packet) {
|
||||||
if isPacketFresh(pair.packet) {
|
|
||||||
candidates.append(pair.packet)
|
candidates.append(pair.packet)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
for id in messageOrder {
|
if types.contains(.message) {
|
||||||
if let p = messages[id], isPacketFresh(p) {
|
candidates.append(contentsOf: messages.allPackets(isFresh: isPacketFresh))
|
||||||
candidates.append(p)
|
|
||||||
}
|
}
|
||||||
|
if types.contains(.fragment) {
|
||||||
|
candidates.append(contentsOf: fragments.allPackets(isFresh: isPacketFresh))
|
||||||
}
|
}
|
||||||
|
if types.contains(.fileTransfer) {
|
||||||
|
candidates.append(contentsOf: fileTransfers.allPackets(isFresh: isPacketFresh))
|
||||||
|
}
|
||||||
|
if candidates.isEmpty {
|
||||||
|
let p = GCSFilter.deriveP(targetFpr: config.gcsTargetFpr)
|
||||||
|
let req = RequestSyncPacket(p: p, m: 1, data: Data(), types: types)
|
||||||
|
return req.encode()
|
||||||
|
}
|
||||||
|
|
||||||
// Sort by timestamp desc
|
// Sort by timestamp desc
|
||||||
candidates.sort { $0.timestamp > $1.timestamp }
|
candidates.sort { $0.timestamp > $1.timestamp }
|
||||||
|
|
||||||
let p = GCSFilter.deriveP(targetFpr: config.gcsTargetFpr)
|
let p = GCSFilter.deriveP(targetFpr: config.gcsTargetFpr)
|
||||||
let nMax = GCSFilter.estimateMaxElements(sizeBytes: config.gcsMaxBytes, p: p)
|
let nMax = GCSFilter.estimateMaxElements(sizeBytes: config.gcsMaxBytes, p: p)
|
||||||
let cap = max(1, config.seenCapacity)
|
let cap: Int
|
||||||
|
if types == .fragment {
|
||||||
|
cap = max(1, config.fragmentCapacity)
|
||||||
|
} else if types == .fileTransfer {
|
||||||
|
cap = max(1, config.fileTransferCapacity)
|
||||||
|
} else {
|
||||||
|
cap = max(1, config.seenCapacity)
|
||||||
|
}
|
||||||
let takeN = min(candidates.count, min(nMax, cap))
|
let takeN = min(candidates.count, min(nMax, cap))
|
||||||
if takeN <= 0 {
|
if takeN <= 0 {
|
||||||
let req = RequestSyncPacket(p: p, m: 1, data: Data())
|
let req = RequestSyncPacket(p: p, m: 1, data: Data(), types: types)
|
||||||
return req.encode()
|
return req.encode()
|
||||||
}
|
}
|
||||||
let ids: [Data] = candidates.prefix(takeN).map { PacketIdUtil.computeId($0) }
|
let ids: [Data] = candidates.prefix(takeN).map { PacketIdUtil.computeId($0) }
|
||||||
let params = GCSFilter.buildFilter(ids: ids, maxBytes: config.gcsMaxBytes, targetFpr: config.gcsTargetFpr)
|
let params = GCSFilter.buildFilter(ids: ids, maxBytes: config.gcsMaxBytes, targetFpr: config.gcsTargetFpr)
|
||||||
let req = RequestSyncPacket(p: params.p, m: params.m, data: params.data)
|
let req = RequestSyncPacket(p: params.p, m: params.m, data: params.data, types: types)
|
||||||
return req.encode()
|
return req.encode()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -241,20 +358,21 @@ final class GossipSyncManager {
|
|||||||
isPacketFresh(pair.packet)
|
isPacketFresh(pair.packet)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Remove expired messages
|
messages.removeExpired(isFresh: isPacketFresh)
|
||||||
let expiredMessageIds = messages.compactMap { id, pkt in
|
fragments.removeExpired(isFresh: isPacketFresh)
|
||||||
isPacketFresh(pkt) ? nil : id
|
fileTransfers.removeExpired(isFresh: isPacketFresh)
|
||||||
}
|
|
||||||
for id in expiredMessageIds {
|
|
||||||
messages.removeValue(forKey: id)
|
|
||||||
messageOrder.removeAll { $0 == id }
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private func performPeriodicMaintenance(now: Date = Date()) {
|
private func performPeriodicMaintenance(now: Date = Date()) {
|
||||||
cleanupExpiredMessages()
|
cleanupExpiredMessages()
|
||||||
cleanupStaleAnnouncementsIfNeeded(now: now)
|
cleanupStaleAnnouncementsIfNeeded(now: now)
|
||||||
sendRequestSync()
|
for index in syncSchedules.indices {
|
||||||
|
guard syncSchedules[index].interval > 0 else { continue }
|
||||||
|
if syncSchedules[index].lastSent == .distantPast || now.timeIntervalSince(syncSchedules[index].lastSent) >= syncSchedules[index].interval {
|
||||||
|
syncSchedules[index].lastSent = now
|
||||||
|
sendRequestSync(for: syncSchedules[index].types)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func cleanupStaleAnnouncementsIfNeeded(now: Date) {
|
private func cleanupStaleAnnouncementsIfNeeded(now: Date) {
|
||||||
@@ -270,40 +388,27 @@ final class GossipSyncManager {
|
|||||||
let nowMs = UInt64(now.timeIntervalSince1970 * 1000)
|
let nowMs = UInt64(now.timeIntervalSince1970 * 1000)
|
||||||
guard nowMs >= timeoutMs else { return }
|
guard nowMs >= timeoutMs else { return }
|
||||||
let cutoff = nowMs - timeoutMs
|
let cutoff = nowMs - timeoutMs
|
||||||
let stalePeerIDs = latestAnnouncementByPeer.compactMap { (peerHex, pair) -> String? in
|
let stalePeerIDs = latestAnnouncementByPeer.compactMap { peerID, pair in
|
||||||
pair.packet.timestamp < cutoff ? peerHex.lowercased() : nil
|
pair.packet.timestamp < cutoff ? peerID : nil
|
||||||
}
|
}
|
||||||
guard !stalePeerIDs.isEmpty else { return }
|
guard !stalePeerIDs.isEmpty else { return }
|
||||||
for peerKey in stalePeerIDs {
|
for peerKey in stalePeerIDs {
|
||||||
removeState(forNormalizedPeerID: peerKey)
|
removeState(for: peerKey)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Explicit removal hook for LEAVE/stale peer
|
// Explicit removal hook for LEAVE/stale peer
|
||||||
func removeAnnouncementForPeer(_ peerID: PeerID) {
|
func removeAnnouncementForPeer(_ peerID: PeerID) {
|
||||||
queue.async { [weak self] in
|
queue.async { [weak self] in
|
||||||
self?._removeAnnouncementForPeer(peerID)
|
self?.removeState(for: peerID)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func _removeAnnouncementForPeer(_ peerID: PeerID) {
|
private func removeState(for peerID: PeerID) {
|
||||||
let normalizedPeerID = peerID.id.lowercased()
|
_ = latestAnnouncementByPeer.removeValue(forKey: peerID)
|
||||||
removeState(forNormalizedPeerID: normalizedPeerID)
|
messages.remove { PeerID(hexData: $0.senderID) == peerID }
|
||||||
}
|
fragments.remove { PeerID(hexData: $0.senderID) == peerID }
|
||||||
|
fileTransfers.remove { PeerID(hexData: $0.senderID) == peerID }
|
||||||
private func removeState(forNormalizedPeerID normalizedPeerID: String) {
|
|
||||||
_ = latestAnnouncementByPeer.removeValue(forKey: normalizedPeerID)
|
|
||||||
// Remove messages from this peer
|
|
||||||
// Collect IDs to remove first to avoid concurrent modification
|
|
||||||
let messageIdsToRemove = messages.compactMap { (id, message) -> String? in
|
|
||||||
message.senderID.hexEncodedString().lowercased() == normalizedPeerID ? id : nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Remove messages and update messageOrder
|
|
||||||
for id in messageIdsToRemove {
|
|
||||||
messages.removeValue(forKey: id)
|
|
||||||
messageOrder.removeAll { $0 == id }
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -317,13 +422,13 @@ extension GossipSyncManager {
|
|||||||
|
|
||||||
func _hasAnnouncement(for peerID: PeerID) -> Bool {
|
func _hasAnnouncement(for peerID: PeerID) -> Bool {
|
||||||
queue.sync {
|
queue.sync {
|
||||||
latestAnnouncementByPeer[peerID.id.lowercased()] != nil
|
latestAnnouncementByPeer[peerID] != nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func _messageCount(for peerID: PeerID) -> Int {
|
func _messageCount(for peerID: PeerID) -> Int {
|
||||||
queue.sync {
|
queue.sync {
|
||||||
messages.values.filter { $0.senderID.hexEncodedString().lowercased() == peerID.id.lowercased() }.count
|
messages.allPackets { _ in true }.filter { PeerID(hexData: $0.senderID) == peerID }.count
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,104 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
/// Bitfield describing which message types are covered by a REQUEST_SYNC round.
|
||||||
|
/// Matches the Android mapping (bit index -> message type).
|
||||||
|
struct SyncTypeFlags: OptionSet {
|
||||||
|
let rawValue: UInt64
|
||||||
|
|
||||||
|
init(rawValue: UInt64) {
|
||||||
|
self.rawValue = rawValue & 0x00FF_FFFF_FFFF_FFFF // Trim to max 8 bytes
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func bitIndex(for type: MessageType) -> Int? {
|
||||||
|
switch type {
|
||||||
|
case .announce: return 0
|
||||||
|
case .message: return 1
|
||||||
|
case .leave: return 2
|
||||||
|
case .noiseHandshake: return 3
|
||||||
|
case .noiseEncrypted: return 4
|
||||||
|
case .fragment: return 5
|
||||||
|
case .requestSync: return 6
|
||||||
|
case .fileTransfer: return 7
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static func type(forBit index: Int) -> MessageType? {
|
||||||
|
switch index {
|
||||||
|
case 0: return .announce
|
||||||
|
case 1: return .message
|
||||||
|
case 2: return .leave
|
||||||
|
case 3: return .noiseHandshake
|
||||||
|
case 4: return .noiseEncrypted
|
||||||
|
case 5: return .fragment
|
||||||
|
case 6: return .requestSync
|
||||||
|
case 7: return .fileTransfer
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static let announce = SyncTypeFlags(messageTypes: [.announce])
|
||||||
|
static let message = SyncTypeFlags(messageTypes: [.message])
|
||||||
|
static let fragment = SyncTypeFlags(messageTypes: [.fragment])
|
||||||
|
static let fileTransfer = SyncTypeFlags(messageTypes: [.fileTransfer])
|
||||||
|
|
||||||
|
static let publicMessages = SyncTypeFlags(messageTypes: [.announce, .message])
|
||||||
|
|
||||||
|
init(messageTypes: [MessageType]) {
|
||||||
|
var raw: UInt64 = 0
|
||||||
|
for type in messageTypes {
|
||||||
|
guard let bit = SyncTypeFlags.bitIndex(for: type) else { continue }
|
||||||
|
raw |= (1 << UInt64(bit))
|
||||||
|
}
|
||||||
|
self.init(rawValue: raw)
|
||||||
|
}
|
||||||
|
|
||||||
|
func contains(_ type: MessageType) -> Bool {
|
||||||
|
guard let bit = SyncTypeFlags.bitIndex(for: type) else { return false }
|
||||||
|
return contains(SyncTypeFlags(rawValue: 1 << UInt64(bit)))
|
||||||
|
}
|
||||||
|
|
||||||
|
func union(_ other: SyncTypeFlags) -> SyncTypeFlags {
|
||||||
|
SyncTypeFlags(rawValue: rawValue | other.rawValue)
|
||||||
|
}
|
||||||
|
|
||||||
|
func intersection(_ other: SyncTypeFlags) -> SyncTypeFlags {
|
||||||
|
SyncTypeFlags(rawValue: rawValue & other.rawValue)
|
||||||
|
}
|
||||||
|
|
||||||
|
func toMessageTypes() -> [MessageType] {
|
||||||
|
guard rawValue != 0 else { return [] }
|
||||||
|
var types: [MessageType] = []
|
||||||
|
for bit in 0..<64 {
|
||||||
|
guard (rawValue & (1 << UInt64(bit))) != 0 else { continue }
|
||||||
|
if let type = SyncTypeFlags.type(forBit: bit) {
|
||||||
|
types.append(type)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return types
|
||||||
|
}
|
||||||
|
|
||||||
|
func toData() -> Data? {
|
||||||
|
guard rawValue != 0 else { return nil }
|
||||||
|
var value = rawValue
|
||||||
|
var bytes: [UInt8] = []
|
||||||
|
while value > 0 && bytes.count < 8 {
|
||||||
|
bytes.append(UInt8(value & 0xFF))
|
||||||
|
value >>= 8
|
||||||
|
}
|
||||||
|
while let last = bytes.last, last == 0 {
|
||||||
|
bytes.removeLast()
|
||||||
|
}
|
||||||
|
guard !bytes.isEmpty, bytes.count <= 8 else { return nil }
|
||||||
|
return Data(bytes)
|
||||||
|
}
|
||||||
|
|
||||||
|
static func decode(_ data: Data) -> SyncTypeFlags? {
|
||||||
|
guard (1...8).contains(data.count) else { return nil }
|
||||||
|
var raw: UInt64 = 0
|
||||||
|
for (index, byte) in data.enumerated() {
|
||||||
|
raw |= UInt64(byte) << UInt64(index * 8)
|
||||||
|
}
|
||||||
|
return SyncTypeFlags(rawValue: raw)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -62,14 +62,12 @@ struct CompressionUtil {
|
|||||||
// 2. Data appears to be already compressed (high entropy)
|
// 2. Data appears to be already compressed (high entropy)
|
||||||
guard data.count >= compressionThreshold else { return false }
|
guard data.count >= compressionThreshold else { return false }
|
||||||
|
|
||||||
// Simple entropy check - count unique bytes
|
// Quick uniqueness check — a high diversity of bytes usually means the
|
||||||
var byteFrequency = [UInt8: Int]()
|
// payload is already compressed. We only need to know how many unique
|
||||||
for byte in data {
|
// values exist rather than keeping full frequency counts.
|
||||||
byteFrequency[byte, default: 0] += 1
|
let uniqueByteCount = Set(data).count
|
||||||
}
|
let sampleSize = min(data.count, 256)
|
||||||
|
let uniqueByteRatio = Double(uniqueByteCount) / Double(sampleSize)
|
||||||
// If we have very high byte diversity, data is likely already compressed
|
|
||||||
let uniqueByteRatio = Double(byteFrequency.count) / Double(min(data.count, 256))
|
|
||||||
return uniqueByteRatio < 0.9 // Compress if less than 90% unique bytes
|
return uniqueByteRatio < 0.9 // Compress if less than 90% unique bytes
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,25 @@
|
|||||||
|
import Foundation
|
||||||
|
|
||||||
|
/// Centralized thresholds for Bluetooth file transfers to keep payload sizes sane on constrained radios.
|
||||||
|
enum FileTransferLimits {
|
||||||
|
/// Absolute ceiling enforced for any file payload (voice, image, other).
|
||||||
|
static let maxPayloadBytes: Int = 1 * 1024 * 1024 // 1 MiB
|
||||||
|
/// Voice notes stay small for low-latency relays.
|
||||||
|
static let maxVoiceNoteBytes: Int = 512 * 1024 // 512 KiB
|
||||||
|
/// Compressed images after downscaling should comfortably fit under this budget.
|
||||||
|
static let maxImageBytes: Int = 512 * 1024 // 512 KiB
|
||||||
|
/// Worst-case size once TLV metadata and binary packet framing are included for the largest payloads.
|
||||||
|
static let maxFramedFileBytes: Int = {
|
||||||
|
let maxMetadataBytes = Int(UInt16.max) * 2 // fileName + mimeType TLVs
|
||||||
|
let tlvEnvelopeOverhead = 18 + maxMetadataBytes // TLV tags + lengths + metadata bytes
|
||||||
|
let binaryEnvelopeOverhead = BinaryProtocol.v2HeaderSize
|
||||||
|
+ BinaryProtocol.senderIDSize
|
||||||
|
+ BinaryProtocol.recipientIDSize
|
||||||
|
+ BinaryProtocol.signatureSize
|
||||||
|
return maxPayloadBytes + tlvEnvelopeOverhead + binaryEnvelopeOverhead
|
||||||
|
}()
|
||||||
|
|
||||||
|
static func isValidPayload(_ size: Int) -> Bool {
|
||||||
|
size <= maxPayloadBytes
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,4 +1,5 @@
|
|||||||
import Foundation
|
import Foundation
|
||||||
|
import BitLogger
|
||||||
|
|
||||||
/// Comprehensive input validation for BitChat protocol
|
/// Comprehensive input validation for BitChat protocol
|
||||||
/// Prevents injection attacks, buffer overflows, and malformed data
|
/// Prevents injection attacks, buffer overflows, and malformed data
|
||||||
@@ -16,29 +17,28 @@ struct InputValidator {
|
|||||||
// MARK: - String Content Validation
|
// MARK: - String Content Validation
|
||||||
|
|
||||||
/// Validates and sanitizes user-provided strings used in UI
|
/// Validates and sanitizes user-provided strings used in UI
|
||||||
|
///
|
||||||
|
/// Rejects strings containing control characters to prevent potential security issues
|
||||||
|
/// and UI rendering problems. This strict approach ensures data integrity at input time.
|
||||||
static func validateUserString(_ string: String, maxLength: Int) -> String? {
|
static func validateUserString(_ string: String, maxLength: Int) -> String? {
|
||||||
// Check empty
|
|
||||||
guard !string.isEmpty else { return nil }
|
|
||||||
|
|
||||||
// Trim whitespace
|
|
||||||
let trimmed = string.trimmingCharacters(in: .whitespacesAndNewlines)
|
let trimmed = string.trimmingCharacters(in: .whitespacesAndNewlines)
|
||||||
guard !trimmed.isEmpty else { return nil }
|
guard !trimmed.isEmpty else { return nil }
|
||||||
|
|
||||||
// Check length
|
|
||||||
guard trimmed.count <= maxLength else { return nil }
|
guard trimmed.count <= maxLength else { return nil }
|
||||||
|
|
||||||
// Remove control characters
|
// Reject control characters outright instead of rewriting the string.
|
||||||
|
// This prevents injection attacks and ensures consistent UI rendering.
|
||||||
let controlChars = CharacterSet.controlCharacters
|
let controlChars = CharacterSet.controlCharacters
|
||||||
let cleaned = trimmed.components(separatedBy: controlChars).joined()
|
if !trimmed.unicodeScalars.allSatisfy({ !controlChars.contains($0) }) {
|
||||||
|
// Log rejection for monitoring, without exposing actual content for privacy
|
||||||
|
let controlCharCount = trimmed.unicodeScalars.filter { controlChars.contains($0) }.count
|
||||||
|
SecureLogger.debug(
|
||||||
|
"Input validation rejected string (length: \(trimmed.count), control chars: \(controlCharCount))",
|
||||||
|
category: .security
|
||||||
|
)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
// Ensure valid UTF-8 (should already be, but double-check)
|
return trimmed
|
||||||
guard cleaned.data(using: .utf8) != nil else { return nil }
|
|
||||||
|
|
||||||
// Prevent zero-width characters and other invisible unicode
|
|
||||||
let invisibleChars = CharacterSet(charactersIn: "\u{200B}\u{200C}\u{200D}\u{FEFF}")
|
|
||||||
let visible = cleaned.components(separatedBy: invisibleChars).joined()
|
|
||||||
|
|
||||||
return visible.isEmpty ? nil : visible
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Validates nickname
|
/// Validates nickname
|
||||||
@@ -52,11 +52,13 @@ struct InputValidator {
|
|||||||
// MessageType/NoisePayloadType enums; keeping validator free of stale lists.
|
// MessageType/NoisePayloadType enums; keeping validator free of stale lists.
|
||||||
|
|
||||||
/// Validates timestamp is reasonable (not too far in past or future)
|
/// Validates timestamp is reasonable (not too far in past or future)
|
||||||
|
/// BCH-01-011: Reduced from ±1 hour to ±5 minutes to limit replay attack window
|
||||||
static func validateTimestamp(_ timestamp: Date) -> Bool {
|
static func validateTimestamp(_ timestamp: Date) -> Bool {
|
||||||
let now = Date()
|
let now = Date()
|
||||||
let oneHourAgo = now.addingTimeInterval(-3600)
|
// 5 minutes = 300 seconds (industry standard for replay protection)
|
||||||
let oneHourFromNow = now.addingTimeInterval(3600)
|
let fiveMinutesAgo = now.addingTimeInterval(-300)
|
||||||
return timestamp >= oneHourAgo && timestamp <= oneHourFromNow
|
let fiveMinutesFromNow = now.addingTimeInterval(300)
|
||||||
|
return timestamp >= fiveMinutesAgo && timestamp <= fiveMinutesFromNow
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,66 +2,112 @@ import Foundation
|
|||||||
|
|
||||||
// MARK: - Message Deduplicator (shared)
|
// MARK: - Message Deduplicator (shared)
|
||||||
|
|
||||||
|
/// Thread-safe deduplicator with LRU eviction and time-based expiry.
|
||||||
|
/// Used for both message ID deduplication (network layer) and content key deduplication (UI layer).
|
||||||
final class MessageDeduplicator {
|
final class MessageDeduplicator {
|
||||||
private struct Entry {
|
private struct Entry: Equatable {
|
||||||
let messageID: String
|
let id: String
|
||||||
let timestamp: Date
|
let timestamp: Date
|
||||||
}
|
}
|
||||||
|
|
||||||
private var entries: [Entry] = []
|
private var entries: [Entry] = []
|
||||||
private var head: Int = 0
|
private var head: Int = 0
|
||||||
private var lookup = Set<String>()
|
private var lookup: [String: Date] = [:] // id -> timestamp for O(1) lookup
|
||||||
private let lock = NSLock()
|
private let lock = NSLock()
|
||||||
private let maxAge: TimeInterval = TransportConfig.messageDedupMaxAgeSeconds // 5 minutes
|
private let maxAge: TimeInterval
|
||||||
private let maxCount = TransportConfig.messageDedupMaxCount
|
private let maxCount: Int
|
||||||
|
|
||||||
/// Check if message is duplicate and add if not
|
/// Initialize with default config from TransportConfig
|
||||||
func isDuplicate(_ messageID: String) -> Bool {
|
convenience init() {
|
||||||
|
self.init(
|
||||||
|
maxAge: TransportConfig.messageDedupMaxAgeSeconds,
|
||||||
|
maxCount: TransportConfig.messageDedupMaxCount
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Initialize with custom config for content deduplication
|
||||||
|
init(maxAge: TimeInterval, maxCount: Int) {
|
||||||
|
self.maxAge = maxAge
|
||||||
|
self.maxCount = maxCount
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Check if message is duplicate and add if not.
|
||||||
|
/// - Parameter id: The message identifier to check.
|
||||||
|
/// - Returns: `true` if the message was already seen, `false` otherwise.
|
||||||
|
func isDuplicate(_ id: String) -> Bool {
|
||||||
lock.lock()
|
lock.lock()
|
||||||
defer { lock.unlock() }
|
defer { lock.unlock() }
|
||||||
|
|
||||||
cleanupOldEntries()
|
let now = Date()
|
||||||
|
cleanupOldEntries(before: now.addingTimeInterval(-maxAge))
|
||||||
|
|
||||||
if lookup.contains(messageID) {
|
if lookup[id] != nil {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
entries.append(Entry(messageID: messageID, timestamp: Date()))
|
entries.append(Entry(id: id, timestamp: now))
|
||||||
lookup.insert(messageID)
|
lookup[id] = now
|
||||||
|
trimIfNeeded()
|
||||||
// Soft-cap and advance head by a chunk to avoid O(n) shifting
|
|
||||||
if (entries.count - head) > maxCount {
|
|
||||||
let removeCount = min(100, entries.count - head)
|
|
||||||
for i in head..<(head + removeCount) {
|
|
||||||
lookup.remove(entries[i].messageID)
|
|
||||||
}
|
|
||||||
head += removeCount
|
|
||||||
// Periodically compact to reclaim memory
|
|
||||||
if head > entries.count / 2 {
|
|
||||||
entries.removeFirst(head)
|
|
||||||
head = 0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Add an ID without checking (for announce-back tracking)
|
/// Record an ID with a specific timestamp (for content key tracking)
|
||||||
func markProcessed(_ messageID: String) {
|
func record(_ id: String, timestamp: Date) {
|
||||||
lock.lock()
|
lock.lock()
|
||||||
defer { lock.unlock() }
|
defer { lock.unlock() }
|
||||||
|
|
||||||
if !lookup.contains(messageID) {
|
if lookup[id] == nil {
|
||||||
entries.append(Entry(messageID: messageID, timestamp: Date()))
|
entries.append(Entry(id: id, timestamp: timestamp))
|
||||||
lookup.insert(messageID)
|
}
|
||||||
|
lookup[id] = timestamp
|
||||||
|
trimIfNeeded()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Add an ID without checking (for announce-back tracking)
|
||||||
|
func markProcessed(_ id: String) {
|
||||||
|
lock.lock()
|
||||||
|
defer { lock.unlock() }
|
||||||
|
|
||||||
|
if lookup[id] == nil {
|
||||||
|
let now = Date()
|
||||||
|
entries.append(Entry(id: id, timestamp: now))
|
||||||
|
lookup[id] = now
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Check if ID exists without adding
|
/// Check if ID exists without adding
|
||||||
func contains(_ messageID: String) -> Bool {
|
func contains(_ id: String) -> Bool {
|
||||||
lock.lock()
|
lock.lock()
|
||||||
defer { lock.unlock() }
|
defer { lock.unlock() }
|
||||||
return lookup.contains(messageID)
|
return lookup[id] != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get timestamp for an ID (for content deduplication time-window checks)
|
||||||
|
func timestampFor(_ id: String) -> Date? {
|
||||||
|
lock.lock()
|
||||||
|
defer { lock.unlock() }
|
||||||
|
return lookup[id]
|
||||||
|
}
|
||||||
|
|
||||||
|
private func trimIfNeeded() {
|
||||||
|
let activeCount = entries.count - head
|
||||||
|
guard activeCount > maxCount else { return }
|
||||||
|
|
||||||
|
// Remove down to 75% of maxCount for better amortization
|
||||||
|
let targetCount = (maxCount * 3) / 4
|
||||||
|
let removeCount = activeCount - targetCount
|
||||||
|
|
||||||
|
for i in head..<(head + removeCount) {
|
||||||
|
lookup.removeValue(forKey: entries[i].id)
|
||||||
|
}
|
||||||
|
head += removeCount
|
||||||
|
|
||||||
|
// Compact when head exceeds half the array to reclaim memory
|
||||||
|
if head > entries.count / 2 {
|
||||||
|
entries.removeFirst(head)
|
||||||
|
head = 0
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Clear all entries
|
/// Clear all entries
|
||||||
@@ -74,24 +120,25 @@ final class MessageDeduplicator {
|
|||||||
lookup.removeAll()
|
lookup.removeAll()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Periodic cleanup
|
/// Periodic cleanup of expired entries and memory optimization.
|
||||||
func cleanup() {
|
func cleanup() {
|
||||||
lock.lock()
|
lock.lock()
|
||||||
defer { lock.unlock() }
|
defer { lock.unlock() }
|
||||||
|
|
||||||
cleanupOldEntries()
|
cleanupOldEntries(before: Date().addingTimeInterval(-maxAge))
|
||||||
|
|
||||||
if entries.capacity > maxCount * 2 {
|
// Shrink capacity if significantly oversized
|
||||||
|
if entries.capacity > maxCount * 2 && entries.count < maxCount {
|
||||||
entries.reserveCapacity(maxCount)
|
entries.reserveCapacity(maxCount)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private func cleanupOldEntries() {
|
private func cleanupOldEntries(before cutoff: Date) {
|
||||||
let cutoff = Date().addingTimeInterval(-maxAge)
|
|
||||||
while head < entries.count, entries[head].timestamp < cutoff {
|
while head < entries.count, entries[head].timestamp < cutoff {
|
||||||
lookup.remove(entries[head].messageID)
|
lookup.removeValue(forKey: entries[head].id)
|
||||||
head += 1
|
head += 1
|
||||||
}
|
}
|
||||||
|
// Compact when head exceeds half the array
|
||||||
if head > 0 && head > entries.count / 2 {
|
if head > 0 && head > entries.count / 2 {
|
||||||
entries.removeFirst(head)
|
entries.removeFirst(head)
|
||||||
head = 0
|
head = 0
|
||||||
|
|||||||
+745
-2834
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,836 @@
|
|||||||
|
//
|
||||||
|
// ChatViewModel+Nostr.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Geohash and Nostr logic for ChatViewModel
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
import Combine
|
||||||
|
import BitLogger
|
||||||
|
import SwiftUI
|
||||||
|
import Tor
|
||||||
|
|
||||||
|
extension ChatViewModel {
|
||||||
|
|
||||||
|
// MARK: - Geohash Subscription
|
||||||
|
|
||||||
|
// Resubscribe to the active geohash channel without clearing timeline
|
||||||
|
@MainActor
|
||||||
|
func resubscribeCurrentGeohash() {
|
||||||
|
guard case .location(let ch) = activeChannel else { return }
|
||||||
|
guard let subID = geoSubscriptionID else {
|
||||||
|
// No existing subscription; set it up
|
||||||
|
switchLocationChannel(to: activeChannel)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// Ensure participant decay timer is running
|
||||||
|
participantTracker.startRefreshTimer()
|
||||||
|
// Unsubscribe + resubscribe
|
||||||
|
NostrRelayManager.shared.unsubscribe(id: subID)
|
||||||
|
let filter = NostrFilter.geohashEphemeral(
|
||||||
|
ch.geohash,
|
||||||
|
since: Date().addingTimeInterval(-TransportConfig.nostrGeohashInitialLookbackSeconds),
|
||||||
|
limit: TransportConfig.nostrGeohashInitialLimit
|
||||||
|
)
|
||||||
|
let subRelays = GeoRelayDirectory.shared.closestRelays(
|
||||||
|
toGeohash: ch.geohash,
|
||||||
|
count: TransportConfig.nostrGeoRelayCount
|
||||||
|
)
|
||||||
|
NostrRelayManager.shared.subscribe(filter: filter, id: subID, relayUrls: subRelays) { [weak self] event in
|
||||||
|
self?.subscribeNostrEvent(event)
|
||||||
|
}
|
||||||
|
// Resubscribe geohash DMs for this identity
|
||||||
|
if let dmSub = geoDmSubscriptionID {
|
||||||
|
NostrRelayManager.shared.unsubscribe(id: dmSub); geoDmSubscriptionID = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if let id = try? idBridge.deriveIdentity(forGeohash: ch.geohash) {
|
||||||
|
let dmSub = "geo-dm-\(ch.geohash)"
|
||||||
|
geoDmSubscriptionID = dmSub
|
||||||
|
let dmFilter = NostrFilter.giftWrapsFor(pubkey: id.publicKeyHex, since: Date().addingTimeInterval(-TransportConfig.nostrDMSubscribeLookbackSeconds))
|
||||||
|
NostrRelayManager.shared.subscribe(filter: dmFilter, id: dmSub) { [weak self] giftWrap in
|
||||||
|
self?.subscribeGiftWrap(giftWrap, id: id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func subscribeNostrEvent(_ event: NostrEvent) {
|
||||||
|
guard (event.kind == NostrProtocol.EventKind.ephemeralEvent.rawValue ||
|
||||||
|
event.kind == NostrProtocol.EventKind.geohashPresence.rawValue),
|
||||||
|
!deduplicationService.hasProcessedNostrEvent(event.id)
|
||||||
|
else {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
deduplicationService.recordNostrEvent(event.id)
|
||||||
|
|
||||||
|
if let gh = currentGeohash,
|
||||||
|
let myGeoIdentity = try? idBridge.deriveIdentity(forGeohash: gh),
|
||||||
|
myGeoIdentity.publicKeyHex.lowercased() == event.pubkey.lowercased() {
|
||||||
|
// Skip very recent self-echo from relay, but allow older events (e.g., after app restart)
|
||||||
|
let eventTime = Date(timeIntervalSince1970: TimeInterval(event.created_at))
|
||||||
|
if Date().timeIntervalSince(eventTime) < 15 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let nickTag = event.tags.first(where: { $0.first == "n" }), nickTag.count >= 2 {
|
||||||
|
let nick = nickTag[1].trimmingCharacters(in: .whitespacesAndNewlines)
|
||||||
|
geoNicknames[event.pubkey.lowercased()] = nick
|
||||||
|
}
|
||||||
|
|
||||||
|
// Store mapping for geohash sender IDs used in messages (ensures consistent colors)
|
||||||
|
nostrKeyMapping[PeerID(nostr_: event.pubkey)] = event.pubkey
|
||||||
|
nostrKeyMapping[PeerID(nostr: event.pubkey)] = event.pubkey
|
||||||
|
|
||||||
|
// Update participants last-seen for this pubkey
|
||||||
|
participantTracker.recordParticipant(pubkeyHex: event.pubkey)
|
||||||
|
|
||||||
|
// If presence heartbeat (Kind 20001), stop here - no content to display
|
||||||
|
if event.kind == NostrProtocol.EventKind.geohashPresence.rawValue {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Track teleported tag (only our format ["t","teleport"]) for icon state
|
||||||
|
let hasTeleportTag = event.tags.contains(where: { tag in
|
||||||
|
tag.count >= 2 && tag[0].lowercased() == "t" && tag[1].lowercased() == "teleport"
|
||||||
|
})
|
||||||
|
|
||||||
|
if hasTeleportTag {
|
||||||
|
let key = event.pubkey.lowercased()
|
||||||
|
// Do not mark our own key from historical events; rely on manager.teleported for self
|
||||||
|
let isSelf: Bool = {
|
||||||
|
if let gh = currentGeohash, let my = try? idBridge.deriveIdentity(forGeohash: gh) {
|
||||||
|
return my.publicKeyHex.lowercased() == key
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}()
|
||||||
|
if !isSelf {
|
||||||
|
Task { @MainActor in
|
||||||
|
teleportedGeo = teleportedGeo.union([key])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let senderName = displayNameForNostrPubkey(event.pubkey)
|
||||||
|
let content = event.content.trimmingCharacters(in: .whitespacesAndNewlines)
|
||||||
|
|
||||||
|
// Clamp future timestamps to now to avoid future-dated messages skewing order
|
||||||
|
let rawTs = Date(timeIntervalSince1970: TimeInterval(event.created_at))
|
||||||
|
let timestamp = min(rawTs, Date())
|
||||||
|
let mentions = parseMentions(from: content)
|
||||||
|
let msg = BitchatMessage(
|
||||||
|
id: event.id,
|
||||||
|
sender: senderName,
|
||||||
|
content: content,
|
||||||
|
timestamp: timestamp,
|
||||||
|
isRelay: false,
|
||||||
|
senderPeerID: PeerID(nostr: event.pubkey),
|
||||||
|
mentions: mentions.isEmpty ? nil : mentions
|
||||||
|
)
|
||||||
|
Task { @MainActor in
|
||||||
|
// BCH-01-012: Check blocking before any notifications
|
||||||
|
// handlePublicMessage has its own blocking check but returns silently,
|
||||||
|
// so we must also guard checkForMentions to prevent notification bypass
|
||||||
|
let isBlocked = identityManager.isNostrBlocked(pubkeyHexLowercased: event.pubkey.lowercased())
|
||||||
|
|
||||||
|
handlePublicMessage(msg)
|
||||||
|
|
||||||
|
// Only check mentions and send haptic if sender is not blocked
|
||||||
|
if !isBlocked {
|
||||||
|
checkForMentions(msg)
|
||||||
|
sendHapticFeedback(for: msg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func subscribeGiftWrap(_ giftWrap: NostrEvent, id: NostrIdentity) {
|
||||||
|
guard !deduplicationService.hasProcessedNostrEvent(giftWrap.id) else { return }
|
||||||
|
deduplicationService.recordNostrEvent(giftWrap.id)
|
||||||
|
|
||||||
|
guard let (content, senderPubkey, rumorTs) = try? NostrProtocol.decryptPrivateMessage(giftWrap: giftWrap, recipientIdentity: id),
|
||||||
|
content.hasPrefix("bitchat1:"),
|
||||||
|
let packetData = Self.base64URLDecode(String(content.dropFirst("bitchat1:".count))),
|
||||||
|
let packet = BitchatPacket.from(packetData),
|
||||||
|
packet.type == MessageType.noiseEncrypted.rawValue,
|
||||||
|
let noisePayload = NoisePayload.decode(packet.payload)
|
||||||
|
else {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let messageTimestamp = Date(timeIntervalSince1970: TimeInterval(rumorTs))
|
||||||
|
let convKey = PeerID(nostr_: senderPubkey)
|
||||||
|
nostrKeyMapping[convKey] = senderPubkey
|
||||||
|
|
||||||
|
switch noisePayload.type {
|
||||||
|
case .privateMessage:
|
||||||
|
handlePrivateMessage(noisePayload, senderPubkey: senderPubkey, convKey: convKey, id: id, messageTimestamp: messageTimestamp)
|
||||||
|
case .delivered:
|
||||||
|
handleDelivered(noisePayload, senderPubkey: senderPubkey, convKey: convKey)
|
||||||
|
case .readReceipt:
|
||||||
|
handleReadReceipt(noisePayload, senderPubkey: senderPubkey, convKey: convKey)
|
||||||
|
case .verifyChallenge, .verifyResponse:
|
||||||
|
// QR verification payloads over Nostr are not supported; ignore in geohash DMs
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Geohash Channel Handling
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
func switchLocationChannel(to channel: ChannelID) {
|
||||||
|
// Reset pending public batches to avoid cross-channel bleed
|
||||||
|
publicMessagePipeline.reset()
|
||||||
|
|
||||||
|
activeChannel = channel
|
||||||
|
publicMessagePipeline.updateActiveChannel(channel)
|
||||||
|
|
||||||
|
// Reset deduplication set and optionally hydrate timeline for mesh
|
||||||
|
deduplicationService.clearNostrCaches()
|
||||||
|
switch channel {
|
||||||
|
case .mesh:
|
||||||
|
refreshVisibleMessages(from: .mesh)
|
||||||
|
// Debug: log if any empty messages are present
|
||||||
|
let emptyMesh = messages.filter { $0.content.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty }.count
|
||||||
|
if emptyMesh > 0 {
|
||||||
|
SecureLogger.debug("RenderGuard: mesh timeline contains \(emptyMesh) empty messages", category: .session)
|
||||||
|
}
|
||||||
|
participantTracker.stopRefreshTimer()
|
||||||
|
participantTracker.setActiveGeohash(nil)
|
||||||
|
teleportedGeo.removeAll()
|
||||||
|
case .location:
|
||||||
|
refreshVisibleMessages(from: channel)
|
||||||
|
}
|
||||||
|
// If switching to a location channel, flush any pending geohash-only system messages
|
||||||
|
if case .location = channel {
|
||||||
|
for content in timelineStore.drainPendingGeohashSystemMessages() {
|
||||||
|
addPublicSystemMessage(content)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Unsubscribe previous
|
||||||
|
if let sub = geoSubscriptionID {
|
||||||
|
NostrRelayManager.shared.unsubscribe(id: sub)
|
||||||
|
geoSubscriptionID = nil
|
||||||
|
}
|
||||||
|
if let dmSub = geoDmSubscriptionID {
|
||||||
|
NostrRelayManager.shared.unsubscribe(id: dmSub)
|
||||||
|
geoDmSubscriptionID = nil
|
||||||
|
}
|
||||||
|
currentGeohash = nil
|
||||||
|
participantTracker.setActiveGeohash(nil)
|
||||||
|
// Reset nickname cache for geochat participants
|
||||||
|
geoNicknames.removeAll()
|
||||||
|
|
||||||
|
guard case .location(let ch) = channel else { return }
|
||||||
|
currentGeohash = ch.geohash
|
||||||
|
participantTracker.setActiveGeohash(ch.geohash)
|
||||||
|
|
||||||
|
// Ensure self appears immediately in the people list; mark teleported state only when truly teleported
|
||||||
|
if let id = try? idBridge.deriveIdentity(forGeohash: ch.geohash) {
|
||||||
|
participantTracker.recordParticipant(pubkeyHex: id.publicKeyHex)
|
||||||
|
let hasRegional = !LocationChannelManager.shared.availableChannels.isEmpty
|
||||||
|
let inRegional = LocationChannelManager.shared.availableChannels.contains { $0.geohash == ch.geohash }
|
||||||
|
let key = id.publicKeyHex.lowercased()
|
||||||
|
if LocationChannelManager.shared.teleported && hasRegional && !inRegional {
|
||||||
|
teleportedGeo = teleportedGeo.union([key])
|
||||||
|
SecureLogger.info("GeoTeleport: channel switch mark self teleported key=\(key.prefix(8))… total=\(teleportedGeo.count)", category: .session)
|
||||||
|
} else {
|
||||||
|
teleportedGeo.remove(key)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let subID = "geo-\(ch.geohash)"
|
||||||
|
geoSubscriptionID = subID
|
||||||
|
participantTracker.startRefreshTimer()
|
||||||
|
let ts = Date().addingTimeInterval(-TransportConfig.nostrGeohashInitialLookbackSeconds)
|
||||||
|
let filter = NostrFilter.geohashEphemeral(ch.geohash, since: ts, limit: TransportConfig.nostrGeohashInitialLimit)
|
||||||
|
let subRelays = GeoRelayDirectory.shared.closestRelays(toGeohash: ch.geohash, count: 5)
|
||||||
|
NostrRelayManager.shared.subscribe(filter: filter, id: subID, relayUrls: subRelays) { [weak self] event in
|
||||||
|
self?.handleNostrEvent(event)
|
||||||
|
}
|
||||||
|
|
||||||
|
subscribeToGeoChat(ch)
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleNostrEvent(_ event: NostrEvent) {
|
||||||
|
// Only handle ephemeral kind 20000 or presence kind 20001 with matching tag
|
||||||
|
guard (event.kind == NostrProtocol.EventKind.ephemeralEvent.rawValue ||
|
||||||
|
event.kind == NostrProtocol.EventKind.geohashPresence.rawValue) else { return }
|
||||||
|
|
||||||
|
// Deduplicate
|
||||||
|
if deduplicationService.hasProcessedNostrEvent(event.id) { return }
|
||||||
|
deduplicationService.recordNostrEvent(event.id)
|
||||||
|
|
||||||
|
// Log incoming tags for diagnostics
|
||||||
|
let tagSummary = event.tags.map { "[" + $0.joined(separator: ",") + "]" }.joined(separator: ",")
|
||||||
|
SecureLogger.debug("GeoTeleport: recv pub=\(event.pubkey.prefix(8))… tags=\(tagSummary)", category: .session)
|
||||||
|
|
||||||
|
// If this pubkey is blocked, skip mapping, participants, and timeline
|
||||||
|
if identityManager.isNostrBlocked(pubkeyHexLowercased: event.pubkey) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Track teleport tag for participants – only our format ["t", "teleport"]
|
||||||
|
let hasTeleportTag: Bool = event.tags.contains { tag in
|
||||||
|
tag.count >= 2 && tag[0].lowercased() == "t" && tag[1].lowercased() == "teleport"
|
||||||
|
}
|
||||||
|
|
||||||
|
let isSelf: Bool = {
|
||||||
|
if let gh = currentGeohash, let my = try? idBridge.deriveIdentity(forGeohash: gh) {
|
||||||
|
return my.publicKeyHex.lowercased() == event.pubkey.lowercased()
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}()
|
||||||
|
|
||||||
|
if hasTeleportTag {
|
||||||
|
// Avoid marking our own key from historical events; rely on manager.teleported for self
|
||||||
|
if !isSelf {
|
||||||
|
let key = event.pubkey.lowercased()
|
||||||
|
Task { @MainActor in
|
||||||
|
teleportedGeo = teleportedGeo.union([key])
|
||||||
|
SecureLogger.info("GeoTeleport: mark peer teleported key=\(key.prefix(8))… total=\(teleportedGeo.count)", category: .session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update participants last-seen for this pubkey
|
||||||
|
participantTracker.recordParticipant(pubkeyHex: event.pubkey)
|
||||||
|
|
||||||
|
// Skip only very recent self-echo from relay; include older self events for hydration
|
||||||
|
if isSelf {
|
||||||
|
let eventTime = Date(timeIntervalSince1970: TimeInterval(event.created_at))
|
||||||
|
if Date().timeIntervalSince(eventTime) < 15 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Cache nickname from tag if present
|
||||||
|
if let nickTag = event.tags.first(where: { $0.first == "n" }), nickTag.count >= 2 {
|
||||||
|
let nick = nickTag[1].trimmingCharacters(in: .whitespacesAndNewlines)
|
||||||
|
geoNicknames[event.pubkey.lowercased()] = nick
|
||||||
|
}
|
||||||
|
|
||||||
|
// Store mapping for geohash DM initiation
|
||||||
|
nostrKeyMapping[PeerID(nostr_: event.pubkey)] = event.pubkey
|
||||||
|
nostrKeyMapping[PeerID(nostr: event.pubkey)] = event.pubkey
|
||||||
|
|
||||||
|
// If presence heartbeat (Kind 20001), stop here - no content to display
|
||||||
|
if event.kind == NostrProtocol.EventKind.geohashPresence.rawValue {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let senderName = displayNameForNostrPubkey(event.pubkey)
|
||||||
|
let content = event.content
|
||||||
|
|
||||||
|
// If this is a teleport presence event (no content), don't add to timeline
|
||||||
|
if let teleTag = event.tags.first(where: { $0.first == "t" }), teleTag.count >= 2, (teleTag[1] == "teleport"),
|
||||||
|
content.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Clamp future timestamps
|
||||||
|
let rawTs = Date(timeIntervalSince1970: TimeInterval(event.created_at))
|
||||||
|
let mentions = parseMentions(from: content)
|
||||||
|
let msg = BitchatMessage(
|
||||||
|
id: event.id,
|
||||||
|
sender: senderName,
|
||||||
|
content: content,
|
||||||
|
timestamp: min(rawTs, Date()),
|
||||||
|
isRelay: false,
|
||||||
|
senderPeerID: PeerID(nostr: event.pubkey),
|
||||||
|
mentions: mentions.isEmpty ? nil : mentions
|
||||||
|
)
|
||||||
|
|
||||||
|
Task { @MainActor in
|
||||||
|
handlePublicMessage(msg)
|
||||||
|
checkForMentions(msg)
|
||||||
|
sendHapticFeedback(for: msg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
func subscribeToGeoChat(_ ch: GeohashChannel) {
|
||||||
|
guard let id = try? idBridge.deriveIdentity(forGeohash: ch.geohash) else { return }
|
||||||
|
|
||||||
|
let dmSub = "geo-dm-\(ch.geohash)"
|
||||||
|
geoDmSubscriptionID = dmSub
|
||||||
|
// pared back logging: subscribe debug only
|
||||||
|
// Log GeoDM subscribe only when Tor is ready to avoid early noise
|
||||||
|
if TorManager.shared.isReady {
|
||||||
|
SecureLogger.debug("GeoDM: subscribing DMs pub=\(id.publicKeyHex.prefix(8))… sub=\(dmSub)", category: .session)
|
||||||
|
}
|
||||||
|
let dmFilter = NostrFilter.giftWrapsFor(pubkey: id.publicKeyHex, since: Date().addingTimeInterval(-TransportConfig.nostrDMSubscribeLookbackSeconds))
|
||||||
|
NostrRelayManager.shared.subscribe(filter: dmFilter, id: dmSub) { [weak self] giftWrap in
|
||||||
|
self?.handleGiftWrap(giftWrap, id: id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleGiftWrap(_ giftWrap: NostrEvent, id: NostrIdentity) {
|
||||||
|
if deduplicationService.hasProcessedNostrEvent(giftWrap.id) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
deduplicationService.recordNostrEvent(giftWrap.id)
|
||||||
|
|
||||||
|
// Decrypt with per-geohash identity
|
||||||
|
guard let (content, senderPubkey, rumorTs) = try? NostrProtocol.decryptPrivateMessage(giftWrap: giftWrap, recipientIdentity: id) else {
|
||||||
|
SecureLogger.warning("GeoDM: failed decrypt giftWrap id=\(giftWrap.id.prefix(8))…", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
SecureLogger.debug("GeoDM: decrypted gift-wrap id=\(giftWrap.id.prefix(16))... from=\(senderPubkey.prefix(8))...", category: .session)
|
||||||
|
|
||||||
|
guard content.hasPrefix("bitchat1:"),
|
||||||
|
let packetData = Self.base64URLDecode(String(content.dropFirst("bitchat1:".count))),
|
||||||
|
let packet = BitchatPacket.from(packetData),
|
||||||
|
packet.type == MessageType.noiseEncrypted.rawValue,
|
||||||
|
let payload = NoisePayload.decode(packet.payload)
|
||||||
|
else {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let convKey = PeerID(nostr_: senderPubkey)
|
||||||
|
nostrKeyMapping[convKey] = senderPubkey
|
||||||
|
|
||||||
|
switch payload.type {
|
||||||
|
case .privateMessage:
|
||||||
|
let messageTimestamp = Date(timeIntervalSince1970: TimeInterval(rumorTs))
|
||||||
|
handlePrivateMessage(payload, senderPubkey: senderPubkey, convKey: convKey, id: id, messageTimestamp: messageTimestamp)
|
||||||
|
case .delivered:
|
||||||
|
handleDelivered(payload, senderPubkey: senderPubkey, convKey: convKey)
|
||||||
|
case .readReceipt:
|
||||||
|
handleReadReceipt(payload, senderPubkey: senderPubkey, convKey: convKey)
|
||||||
|
|
||||||
|
// Explicitly list other cases so we get compile-time check if a new case is added in the future
|
||||||
|
case .verifyChallenge, .verifyResponse:
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
func sendGeohash(context: GeoOutgoingContext) {
|
||||||
|
let ch = context.channel
|
||||||
|
let event = context.event
|
||||||
|
let identity = context.identity
|
||||||
|
|
||||||
|
let targetRelays = GeoRelayDirectory.shared.closestRelays(
|
||||||
|
toGeohash: ch.geohash,
|
||||||
|
count: TransportConfig.nostrGeoRelayCount
|
||||||
|
)
|
||||||
|
|
||||||
|
if targetRelays.isEmpty {
|
||||||
|
SecureLogger.warning("Geo: no geohash relays available for \(ch.geohash); not sending", category: .session)
|
||||||
|
} else {
|
||||||
|
NostrRelayManager.shared.sendEvent(event, to: targetRelays)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Track ourselves as active participant
|
||||||
|
participantTracker.recordParticipant(pubkeyHex: identity.publicKeyHex)
|
||||||
|
nostrKeyMapping[PeerID(nostr: identity.publicKeyHex)] = identity.publicKeyHex
|
||||||
|
SecureLogger.debug("GeoTeleport: sent geo message pub=\(identity.publicKeyHex.prefix(8))… teleported=\(context.teleported)", category: .session)
|
||||||
|
|
||||||
|
// If we tagged this as teleported, also mark our pubkey in teleportedGeo for UI
|
||||||
|
// Only when not in our regional set (and regional list is known)
|
||||||
|
let hasRegional = !LocationChannelManager.shared.availableChannels.isEmpty
|
||||||
|
let inRegional = LocationChannelManager.shared.availableChannels.contains { $0.geohash == ch.geohash }
|
||||||
|
|
||||||
|
if context.teleported && hasRegional && !inRegional {
|
||||||
|
let key = identity.publicKeyHex.lowercased()
|
||||||
|
teleportedGeo = teleportedGeo.union([key])
|
||||||
|
SecureLogger.info("GeoTeleport: mark self teleported key=\(key.prefix(8))… total=\(teleportedGeo.count)", category: .session)
|
||||||
|
}
|
||||||
|
|
||||||
|
deduplicationService.recordNostrEvent(event.id)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Sampling
|
||||||
|
|
||||||
|
/// Begin sampling multiple geohashes (used by channel sheet) without changing active channel.
|
||||||
|
@MainActor
|
||||||
|
func beginGeohashSampling(for geohashes: [String]) {
|
||||||
|
// Disable sampling when app is backgrounded (Tor is stopped there)
|
||||||
|
if !TorManager.shared.isForeground() {
|
||||||
|
endGeohashSampling()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// Determine which to add and which to remove
|
||||||
|
let desired = Set(geohashes)
|
||||||
|
let current = Set(geoSamplingSubs.values)
|
||||||
|
let toAdd = desired.subtracting(current)
|
||||||
|
let toRemove = current.subtracting(desired)
|
||||||
|
|
||||||
|
for (subID, gh) in geoSamplingSubs where toRemove.contains(gh) {
|
||||||
|
NostrRelayManager.shared.unsubscribe(id: subID)
|
||||||
|
geoSamplingSubs.removeValue(forKey: subID)
|
||||||
|
}
|
||||||
|
|
||||||
|
for gh in toAdd {
|
||||||
|
subscribe(gh)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
func subscribe(_ gh: String) {
|
||||||
|
let subID = "geo-sample-\(gh)"
|
||||||
|
geoSamplingSubs[subID] = gh
|
||||||
|
let filter = NostrFilter.geohashEphemeral(
|
||||||
|
gh,
|
||||||
|
since: Date().addingTimeInterval(-TransportConfig.nostrGeohashSampleLookbackSeconds),
|
||||||
|
limit: TransportConfig.nostrGeohashSampleLimit
|
||||||
|
)
|
||||||
|
let subRelays = GeoRelayDirectory.shared.closestRelays(toGeohash: gh, count: 5)
|
||||||
|
NostrRelayManager.shared.subscribe(filter: filter, id: subID, relayUrls: subRelays) { [weak self] event in
|
||||||
|
self?.subscribeNostrEvent(event, gh: gh)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func subscribeNostrEvent(_ event: NostrEvent, gh: String) {
|
||||||
|
guard (event.kind == NostrProtocol.EventKind.ephemeralEvent.rawValue ||
|
||||||
|
event.kind == NostrProtocol.EventKind.geohashPresence.rawValue) else { return }
|
||||||
|
|
||||||
|
// Compute current participant count (5-minute window) BEFORE updating with this event
|
||||||
|
let existingCount = participantTracker.participantCount(for: gh)
|
||||||
|
|
||||||
|
// Update participants for this specific geohash
|
||||||
|
participantTracker.recordParticipant(pubkeyHex: event.pubkey, geohash: gh)
|
||||||
|
|
||||||
|
// Notify only on rising-edge: previously zero people, now someone sends a chat
|
||||||
|
let content = event.content.trimmingCharacters(in: .whitespacesAndNewlines)
|
||||||
|
guard !content.isEmpty else { return }
|
||||||
|
|
||||||
|
// Respect geohash blocks
|
||||||
|
if identityManager.isNostrBlocked(pubkeyHexLowercased: event.pubkey.lowercased()) { return }
|
||||||
|
|
||||||
|
// Skip self identity for this geohash
|
||||||
|
if let my = try? idBridge.deriveIdentity(forGeohash: gh), my.publicKeyHex.lowercased() == event.pubkey.lowercased() { return }
|
||||||
|
|
||||||
|
// Only trigger when there were zero participants in this geohash recently
|
||||||
|
guard existingCount == 0 else { return }
|
||||||
|
|
||||||
|
// Avoid notifications for old sampled events when launching or (re)subscribing
|
||||||
|
let eventTime = Date(timeIntervalSince1970: TimeInterval(event.created_at))
|
||||||
|
if Date().timeIntervalSince(eventTime) > 30 { return }
|
||||||
|
|
||||||
|
// Foreground-only notifications: app must be active, and not already viewing this geohash
|
||||||
|
#if os(iOS)
|
||||||
|
guard UIApplication.shared.applicationState == .active else { return }
|
||||||
|
if case .location(let ch) = activeChannel, ch.geohash == gh { return }
|
||||||
|
#elseif os(macOS)
|
||||||
|
guard NSApplication.shared.isActive else { return }
|
||||||
|
if case .location(let ch) = activeChannel, ch.geohash == gh { return }
|
||||||
|
#endif
|
||||||
|
|
||||||
|
cooldownPerGeohash(gh, content: content, event: event)
|
||||||
|
}
|
||||||
|
|
||||||
|
func cooldownPerGeohash(_ gh: String, content: String, event: NostrEvent) {
|
||||||
|
let now = Date()
|
||||||
|
let last = lastGeoNotificationAt[gh] ?? .distantPast
|
||||||
|
if now.timeIntervalSince(last) < TransportConfig.uiGeoNotifyCooldownSeconds { return }
|
||||||
|
|
||||||
|
// Compose a short preview
|
||||||
|
let preview: String = {
|
||||||
|
let maxLen = TransportConfig.uiGeoNotifySnippetMaxLen
|
||||||
|
if content.count <= maxLen { return content }
|
||||||
|
let idx = content.index(content.startIndex, offsetBy: maxLen)
|
||||||
|
return String(content[..<idx]) + "…"
|
||||||
|
}()
|
||||||
|
|
||||||
|
Task { @MainActor in
|
||||||
|
lastGeoNotificationAt[gh] = now
|
||||||
|
// Pre-populate the target geohash timeline so the triggering message appears when user opens it
|
||||||
|
let senderSuffix = String(event.pubkey.suffix(4))
|
||||||
|
let nick = geoNicknames[event.pubkey.lowercased()]
|
||||||
|
let senderName = (nick?.isEmpty == false ? nick! : "anon") + "#" + senderSuffix
|
||||||
|
|
||||||
|
// Clamp future timestamps
|
||||||
|
let rawTs = Date(timeIntervalSince1970: TimeInterval(event.created_at))
|
||||||
|
let ts = min(rawTs, Date())
|
||||||
|
let mentions = self.parseMentions(from: content)
|
||||||
|
let msg = BitchatMessage(
|
||||||
|
id: event.id,
|
||||||
|
sender: senderName,
|
||||||
|
content: content,
|
||||||
|
timestamp: ts,
|
||||||
|
isRelay: false,
|
||||||
|
senderPeerID: PeerID(nostr: event.pubkey),
|
||||||
|
mentions: mentions.isEmpty ? nil : mentions
|
||||||
|
)
|
||||||
|
if timelineStore.appendIfAbsent(msg, toGeohash: gh) {
|
||||||
|
NotificationService.shared.sendGeohashActivityNotification(geohash: gh, bodyPreview: preview)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Stop sampling all extra geohashes.
|
||||||
|
@MainActor
|
||||||
|
func endGeohashSampling() {
|
||||||
|
for subID in geoSamplingSubs.keys { NostrRelayManager.shared.unsubscribe(id: subID) }
|
||||||
|
geoSamplingSubs.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Nostr DM Handling
|
||||||
|
|
||||||
|
func setupNostrMessageHandling() {
|
||||||
|
guard let currentIdentity = try? idBridge.getCurrentNostrIdentity() else {
|
||||||
|
SecureLogger.warning("⚠️ No Nostr identity available for message handling", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
SecureLogger.debug("🔑 Setting up Nostr subscription for pubkey: \(currentIdentity.publicKeyHex.prefix(16))...", category: .session)
|
||||||
|
|
||||||
|
// Subscribe to Nostr messages
|
||||||
|
let filter = NostrFilter.giftWrapsFor(
|
||||||
|
pubkey: currentIdentity.publicKeyHex,
|
||||||
|
since: Date().addingTimeInterval(-TransportConfig.nostrDMSubscribeLookbackSeconds) // Last 24 hours
|
||||||
|
)
|
||||||
|
|
||||||
|
nostrRelayManager?.subscribe(filter: filter, id: "chat-messages") { [weak self] event in
|
||||||
|
self?.handleNostrMessage(event)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleNostrMessage(_ giftWrap: NostrEvent) {
|
||||||
|
// Deduplicate messages by ID
|
||||||
|
if deduplicationService.hasProcessedNostrEvent(giftWrap.id) { return }
|
||||||
|
deduplicationService.recordNostrEvent(giftWrap.id)
|
||||||
|
|
||||||
|
// Ensure we're on a background queue for decryption
|
||||||
|
Task.detached(priority: .userInitiated) { [weak self] in
|
||||||
|
await self?.processNostrMessage(giftWrap)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func processNostrMessage(_ giftWrap: NostrEvent) async {
|
||||||
|
guard let currentIdentity = try? idBridge.getCurrentNostrIdentity() else { return }
|
||||||
|
|
||||||
|
do {
|
||||||
|
let (content, senderPubkey, rumorTimestamp) = try NostrProtocol.decryptPrivateMessage(
|
||||||
|
giftWrap: giftWrap,
|
||||||
|
recipientIdentity: currentIdentity
|
||||||
|
)
|
||||||
|
|
||||||
|
// Handle verification payloads first
|
||||||
|
if content.hasPrefix("verify:") {
|
||||||
|
// Ignore verification payloads arriving via Nostr path for now
|
||||||
|
// Verification should ideally happen over mesh for security binding
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if it's a BitChat packet embedded in the content (bitchat1:...)
|
||||||
|
if content.hasPrefix("bitchat1:") {
|
||||||
|
guard let packetData = Self.base64URLDecode(String(content.dropFirst("bitchat1:".count))),
|
||||||
|
let packet = BitchatPacket.from(packetData) else {
|
||||||
|
SecureLogger.error("Failed to decode embedded BitChat packet from Nostr DM", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Map sender by Nostr pubkey to Noise key when possible
|
||||||
|
let actualSenderNoiseKey = findNoiseKey(for: senderPubkey)
|
||||||
|
|
||||||
|
// Stable target ID if we know Noise key; otherwise temporary Nostr-based peer
|
||||||
|
let targetPeerID = PeerID(str: actualSenderNoiseKey?.hexEncodedString()) ?? PeerID(nostr_: senderPubkey)
|
||||||
|
|
||||||
|
if packet.type == MessageType.noiseEncrypted.rawValue {
|
||||||
|
if let payload = NoisePayload.decode(packet.payload) {
|
||||||
|
let messageTimestamp = Date(timeIntervalSince1970: TimeInterval(rumorTimestamp))
|
||||||
|
// Store Nostr mapping
|
||||||
|
await MainActor.run {
|
||||||
|
nostrKeyMapping[targetPeerID] = senderPubkey
|
||||||
|
|
||||||
|
// Handle packet types
|
||||||
|
switch payload.type {
|
||||||
|
case .privateMessage:
|
||||||
|
handlePrivateMessage(payload, senderPubkey: senderPubkey, convKey: targetPeerID, id: currentIdentity, messageTimestamp: messageTimestamp)
|
||||||
|
case .delivered:
|
||||||
|
handleDelivered(payload, senderPubkey: senderPubkey, convKey: targetPeerID)
|
||||||
|
case .readReceipt:
|
||||||
|
handleReadReceipt(payload, senderPubkey: senderPubkey, convKey: targetPeerID)
|
||||||
|
case .verifyChallenge, .verifyResponse:
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
SecureLogger.debug("Ignoring non-embedded Nostr DM content", category: .session)
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
SecureLogger.error("Failed to decrypt Nostr message: \(error)", category: .session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func findNoiseKey(for nostrPubkey: String) -> Data? {
|
||||||
|
// Check favorites for this Nostr key
|
||||||
|
let favorites = FavoritesPersistenceService.shared.favorites.values
|
||||||
|
var npubToMatch = nostrPubkey
|
||||||
|
|
||||||
|
// Convert hex to npub if needed for comparison
|
||||||
|
if !nostrPubkey.hasPrefix("npub") {
|
||||||
|
if let pubkeyData = Data(hexString: nostrPubkey),
|
||||||
|
let encoded = try? Bech32.encode(hrp: "npub", data: pubkeyData) {
|
||||||
|
npubToMatch = encoded
|
||||||
|
} else {
|
||||||
|
SecureLogger.warning("⚠️ Invalid hex public key format or encoding failed: \(nostrPubkey.prefix(16))...", category: .session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for relationship in favorites {
|
||||||
|
// Search through favorites for matching Nostr pubkey
|
||||||
|
if let storedNostrKey = relationship.peerNostrPublicKey {
|
||||||
|
// Compare against stored key (could be hex or npub)
|
||||||
|
if storedNostrKey == npubToMatch {
|
||||||
|
// SecureLogger.debug("✅ Found Noise key for Nostr sender (npub match)", category: .session)
|
||||||
|
return relationship.peerNoisePublicKey
|
||||||
|
}
|
||||||
|
|
||||||
|
// Also try comparing raw hex if stored key is hex
|
||||||
|
if !storedNostrKey.hasPrefix("npub") && storedNostrKey == nostrPubkey {
|
||||||
|
SecureLogger.debug("✅ Found Noise key for Nostr sender (hex match)", category: .session)
|
||||||
|
return relationship.peerNoisePublicKey
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
SecureLogger.debug("⚠️ No matching Noise key found for Nostr pubkey: \(nostrPubkey.prefix(16))... (tried npub: \(npubToMatch.prefix(16))...)", category: .session)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func sendDeliveryAckViaNostrEmbedded(_ message: BitchatMessage, wasReadBefore: Bool, senderPubkey: String, key: Data?) {
|
||||||
|
// If we have a Noise key, try to route securely if possible, otherwise fallback to direct
|
||||||
|
if let _ = key {
|
||||||
|
// Ideally we would use MessageRouter here, but for simplicity in this direct callback:
|
||||||
|
// check if we have an identity
|
||||||
|
if let id = try? idBridge.getCurrentNostrIdentity() {
|
||||||
|
let nt = NostrTransport(keychain: keychain, idBridge: idBridge)
|
||||||
|
nt.senderPeerID = meshService.myPeerID
|
||||||
|
nt.sendDeliveryAckGeohash(for: message.id, toRecipientHex: senderPubkey, from: id)
|
||||||
|
}
|
||||||
|
} else if let id = try? idBridge.getCurrentNostrIdentity() {
|
||||||
|
// Fallback: no Noise mapping yet — send directly to sender's Nostr pubkey
|
||||||
|
let nt = NostrTransport(keychain: keychain, idBridge: idBridge)
|
||||||
|
nt.senderPeerID = meshService.myPeerID
|
||||||
|
nt.sendDeliveryAckGeohash(for: message.id, toRecipientHex: senderPubkey, from: id)
|
||||||
|
SecureLogger.debug("Sent DELIVERED ack directly to Nostr pub=\(senderPubkey.prefix(8))… for mid=\(message.id.prefix(8))…", category: .session)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Same for READ receipt if viewing
|
||||||
|
if !wasReadBefore && selectedPrivateChatPeer == message.senderPeerID {
|
||||||
|
if let _ = key {
|
||||||
|
if let id = try? idBridge.getCurrentNostrIdentity() {
|
||||||
|
let nt = NostrTransport(keychain: keychain, idBridge: idBridge)
|
||||||
|
nt.senderPeerID = meshService.myPeerID
|
||||||
|
nt.sendReadReceiptGeohash(message.id, toRecipientHex: senderPubkey, from: id)
|
||||||
|
}
|
||||||
|
} else if let id = try? idBridge.getCurrentNostrIdentity() {
|
||||||
|
let nt = NostrTransport(keychain: keychain, idBridge: idBridge)
|
||||||
|
nt.senderPeerID = meshService.myPeerID
|
||||||
|
nt.sendReadReceiptGeohash(message.id, toRecipientHex: senderPubkey, from: id)
|
||||||
|
SecureLogger.debug("Viewing chat; sent READ ack directly to Nostr pub=\(senderPubkey.prefix(8))… for mid=\(message.id.prefix(8))…", category: .session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleFavoriteNotification(content: String, from nostrPubkey: String) {
|
||||||
|
// Try to find Noise key associated with this Nostr pubkey
|
||||||
|
guard let senderNoiseKey = findNoiseKey(for: nostrPubkey) else { return }
|
||||||
|
|
||||||
|
let isFavorite = content.contains("FAVORITE:TRUE")
|
||||||
|
let senderNickname = content.components(separatedBy: "|").last ?? "Unknown"
|
||||||
|
|
||||||
|
// Update favorite status
|
||||||
|
if isFavorite {
|
||||||
|
FavoritesPersistenceService.shared.addFavorite(
|
||||||
|
peerNoisePublicKey: senderNoiseKey,
|
||||||
|
peerNostrPublicKey: nostrPubkey,
|
||||||
|
peerNickname: senderNickname
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
// Only remove if we don't have it set locally
|
||||||
|
// Logic handled by persistence service usually, here we just update remote state
|
||||||
|
// Actually for now we just process the notification
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extract Nostr public key if included
|
||||||
|
var extractedNostrPubkey: String? = nil
|
||||||
|
if let range = content.range(of: "NPUB:") {
|
||||||
|
let suffix = content[range.upperBound...]
|
||||||
|
let parts = suffix.components(separatedBy: "|")
|
||||||
|
if let key = parts.first {
|
||||||
|
extractedNostrPubkey = String(key)
|
||||||
|
}
|
||||||
|
} else if content.contains(":") {
|
||||||
|
// Fallback: simple format FAVORITE:TRUE:npub...
|
||||||
|
let parts = content.components(separatedBy: ":")
|
||||||
|
if parts.count >= 3 {
|
||||||
|
extractedNostrPubkey = String(parts[2])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
SecureLogger.info("📝 Received favorite notification from \(senderNickname): \(isFavorite)", category: .session)
|
||||||
|
|
||||||
|
// If they favorited us and provided their Nostr key, ensure it's stored
|
||||||
|
if isFavorite && extractedNostrPubkey != nil {
|
||||||
|
SecureLogger.info("💾 Storing Nostr key association for \(senderNickname): \(extractedNostrPubkey!.prefix(16))...", category: .session)
|
||||||
|
FavoritesPersistenceService.shared.addFavorite(
|
||||||
|
peerNoisePublicKey: senderNoiseKey,
|
||||||
|
peerNostrPublicKey: extractedNostrPubkey,
|
||||||
|
peerNickname: senderNickname
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Show notification
|
||||||
|
NotificationService.shared.sendLocalNotification(
|
||||||
|
title: isFavorite ? "New Favorite" : "Favorite Removed",
|
||||||
|
body: "\(senderNickname) \(isFavorite ? "favorited" : "unfavorited") you",
|
||||||
|
identifier: "fav-\(UUID().uuidString)"
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
func sendFavoriteNotificationViaNostr(noisePublicKey: Data, isFavorite: Bool) {
|
||||||
|
// Find peer Nostr key
|
||||||
|
guard let relationship = FavoritesPersistenceService.shared.getFavoriteStatus(for: noisePublicKey),
|
||||||
|
relationship.peerNostrPublicKey != nil else {
|
||||||
|
SecureLogger.warning("⚠️ Cannot send favorite notification - no Nostr key for peer", category: .session)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let peerID = PeerID(hexData: noisePublicKey)
|
||||||
|
|
||||||
|
// Route via message router
|
||||||
|
messageRouter.sendFavoriteNotification(to: peerID, isFavorite: isFavorite)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Geohash Nickname Resolution (for /block in geohash)
|
||||||
|
|
||||||
|
func nostrPubkeyForDisplayName(_ name: String) -> String? {
|
||||||
|
// Look up current visible geohash participants for an exact displayName match
|
||||||
|
for p in visibleGeohashPeople() {
|
||||||
|
if p.displayName == name {
|
||||||
|
return p.id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Also check nickname cache directly
|
||||||
|
for (pub, nick) in geoNicknames {
|
||||||
|
if nick == name { return pub }
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func startGeohashDM(withPubkeyHex hex: String) {
|
||||||
|
let convKey = PeerID(nostr_: hex)
|
||||||
|
nostrKeyMapping[convKey] = hex
|
||||||
|
startPrivateChat(with: convKey)
|
||||||
|
}
|
||||||
|
|
||||||
|
func fullNostrHex(forSenderPeerID senderID: PeerID) -> String? {
|
||||||
|
return nostrKeyMapping[senderID]
|
||||||
|
}
|
||||||
|
|
||||||
|
func geohashDisplayName(for convKey: PeerID) -> String {
|
||||||
|
guard let full = nostrKeyMapping[convKey] else {
|
||||||
|
return convKey.bare
|
||||||
|
}
|
||||||
|
return displayNameForNostrPubkey(full)
|
||||||
|
}
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,64 @@
|
|||||||
|
//
|
||||||
|
// ChatViewModel+Tor.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Tor lifecycle handling for ChatViewModel
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
import Combine
|
||||||
|
import Tor
|
||||||
|
|
||||||
|
extension ChatViewModel {
|
||||||
|
|
||||||
|
// MARK: - Tor notifications
|
||||||
|
|
||||||
|
@objc func handleTorWillStart() {
|
||||||
|
Task { @MainActor in
|
||||||
|
if !self.torStatusAnnounced && TorManager.shared.torEnforced {
|
||||||
|
self.torStatusAnnounced = true
|
||||||
|
// Post only in geohash channels (queue if not active)
|
||||||
|
self.addGeohashOnlySystemMessage(
|
||||||
|
String(localized: "system.tor.starting", comment: "System message when Tor is starting")
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@objc func handleTorWillRestart() {
|
||||||
|
Task { @MainActor in
|
||||||
|
self.torRestartPending = true
|
||||||
|
// Post only in geohash channels (queue if not active)
|
||||||
|
self.addGeohashOnlySystemMessage(
|
||||||
|
String(localized: "system.tor.restarting", comment: "System message when Tor is restarting")
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@objc func handleTorDidBecomeReady() {
|
||||||
|
Task { @MainActor in
|
||||||
|
// Only announce "restarted" if we actually restarted this session
|
||||||
|
if self.torRestartPending {
|
||||||
|
// Post only in geohash channels (queue if not active)
|
||||||
|
self.addGeohashOnlySystemMessage(
|
||||||
|
String(localized: "system.tor.restarted", comment: "System message when Tor has restarted")
|
||||||
|
)
|
||||||
|
self.torRestartPending = false
|
||||||
|
} else if TorManager.shared.torEnforced && !self.torInitialReadyAnnounced {
|
||||||
|
// Initial start completed
|
||||||
|
self.addGeohashOnlySystemMessage(
|
||||||
|
String(localized: "system.tor.started", comment: "System message when Tor has started")
|
||||||
|
)
|
||||||
|
self.torInitialReadyAnnounced = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@objc func handleTorPreferenceChanged(_ notification: Notification) {
|
||||||
|
Task { @MainActor in
|
||||||
|
self.torStatusAnnounced = false
|
||||||
|
self.torInitialReadyAnnounced = false
|
||||||
|
self.torRestartPending = false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,9 @@
|
|||||||
|
# ChatViewModel Extensions
|
||||||
|
|
||||||
|
This directory contains extensions to `ChatViewModel` to modularize its functionality.
|
||||||
|
|
||||||
|
- `ChatViewModel+Tor.swift`: Handles Tor lifecycle events and notifications.
|
||||||
|
- `ChatViewModel+PrivateChat.swift`: Manages private chat logic, media transfers (images, voice notes), and file handling.
|
||||||
|
- `ChatViewModel+Nostr.swift`: Contains all logic related to Nostr integration, Geohash channels, and Nostr identity management.
|
||||||
|
|
||||||
|
The main `ChatViewModel.swift` retains core state, initialization, and coordination logic.
|
||||||
@@ -0,0 +1,118 @@
|
|||||||
|
//
|
||||||
|
// GeoChannelCoordinator.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Centralizes Combine wiring for location channel selection and sampling.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Combine
|
||||||
|
import Foundation
|
||||||
|
import Tor
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class GeoChannelCoordinator {
|
||||||
|
private let locationManager: LocationChannelManager
|
||||||
|
private let bookmarksStore: GeohashBookmarksStore
|
||||||
|
private let torManager: TorManager
|
||||||
|
|
||||||
|
private let onChannelSwitch: (ChannelID) -> Void
|
||||||
|
private let beginSampling: ([String]) -> Void
|
||||||
|
private let endSampling: () -> Void
|
||||||
|
|
||||||
|
private var cancellables = Set<AnyCancellable>()
|
||||||
|
private var regionalGeohashes: [String] = []
|
||||||
|
private var bookmarkedGeohashes: [String] = []
|
||||||
|
|
||||||
|
init(
|
||||||
|
locationManager: LocationChannelManager? = nil,
|
||||||
|
bookmarksStore: GeohashBookmarksStore? = nil,
|
||||||
|
torManager: TorManager? = nil,
|
||||||
|
onChannelSwitch: @escaping (ChannelID) -> Void,
|
||||||
|
beginSampling: @escaping ([String]) -> Void,
|
||||||
|
endSampling: @escaping () -> Void
|
||||||
|
) {
|
||||||
|
self.locationManager = locationManager ?? Self.defaultLocationManager()
|
||||||
|
self.bookmarksStore = bookmarksStore ?? GeohashBookmarksStore.shared
|
||||||
|
self.torManager = torManager ?? Self.defaultTorManager()
|
||||||
|
self.onChannelSwitch = onChannelSwitch
|
||||||
|
self.beginSampling = beginSampling
|
||||||
|
self.endSampling = endSampling
|
||||||
|
|
||||||
|
start()
|
||||||
|
}
|
||||||
|
|
||||||
|
func start() {
|
||||||
|
regionalGeohashes = locationManager.availableChannels.map { $0.geohash }
|
||||||
|
bookmarkedGeohashes = bookmarksStore.bookmarks
|
||||||
|
|
||||||
|
locationManager.$selectedChannel
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] channel in
|
||||||
|
guard let self else { return }
|
||||||
|
Task { @MainActor in
|
||||||
|
self.onChannelSwitch(channel)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
locationManager.$availableChannels
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] channels in
|
||||||
|
guard let self else { return }
|
||||||
|
self.regionalGeohashes = channels.map { $0.geohash }
|
||||||
|
self.updateSampling()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
bookmarksStore.$bookmarks
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] bookmarks in
|
||||||
|
guard let self else { return }
|
||||||
|
self.bookmarkedGeohashes = bookmarks
|
||||||
|
self.updateSampling()
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
locationManager.$permissionState
|
||||||
|
.receive(on: DispatchQueue.main)
|
||||||
|
.sink { [weak self] state in
|
||||||
|
guard let self, state == .authorized else { return }
|
||||||
|
Task { @MainActor [weak self] in
|
||||||
|
self?.locationManager.refreshChannels()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.store(in: &cancellables)
|
||||||
|
|
||||||
|
Task { @MainActor in
|
||||||
|
self.onChannelSwitch(self.locationManager.selectedChannel)
|
||||||
|
}
|
||||||
|
updateSampling()
|
||||||
|
}
|
||||||
|
|
||||||
|
private func updateSampling() {
|
||||||
|
let union = Array(Set(regionalGeohashes).union(bookmarkedGeohashes))
|
||||||
|
Task { @MainActor in
|
||||||
|
guard !union.isEmpty else {
|
||||||
|
endSampling()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if torManager.isForeground() {
|
||||||
|
beginSampling(union)
|
||||||
|
} else {
|
||||||
|
endSampling()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func refreshSampling() {
|
||||||
|
updateSampling()
|
||||||
|
}
|
||||||
|
private static func defaultLocationManager() -> LocationChannelManager {
|
||||||
|
LocationChannelManager.shared
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
private static func defaultTorManager() -> TorManager {
|
||||||
|
TorManager.shared
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,77 @@
|
|||||||
|
//
|
||||||
|
// MessageRateLimiter.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Handles per-sender and per-content token buckets for public message intake.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct MessageRateLimiter {
|
||||||
|
private struct TokenBucket {
|
||||||
|
var capacity: Double
|
||||||
|
var tokens: Double
|
||||||
|
var refillPerSec: Double
|
||||||
|
var lastRefill: Date
|
||||||
|
|
||||||
|
mutating func allow(cost: Double = 1.0, now: Date = Date()) -> Bool {
|
||||||
|
let dt = now.timeIntervalSince(lastRefill)
|
||||||
|
if dt > 0 {
|
||||||
|
tokens = min(capacity, tokens + dt * refillPerSec)
|
||||||
|
lastRefill = now
|
||||||
|
}
|
||||||
|
if tokens >= cost {
|
||||||
|
tokens -= cost
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private var senderBuckets: [String: TokenBucket] = [:]
|
||||||
|
private var contentBuckets: [String: TokenBucket] = [:]
|
||||||
|
|
||||||
|
private let senderCapacity: Double
|
||||||
|
private let senderRefill: Double
|
||||||
|
private let contentCapacity: Double
|
||||||
|
private let contentRefill: Double
|
||||||
|
|
||||||
|
init(
|
||||||
|
senderCapacity: Double,
|
||||||
|
senderRefillPerSec: Double,
|
||||||
|
contentCapacity: Double,
|
||||||
|
contentRefillPerSec: Double
|
||||||
|
) {
|
||||||
|
self.senderCapacity = senderCapacity
|
||||||
|
self.senderRefill = senderRefillPerSec
|
||||||
|
self.contentCapacity = contentCapacity
|
||||||
|
self.contentRefill = contentRefillPerSec
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func allow(senderKey: String, contentKey: String, now: Date = Date()) -> Bool {
|
||||||
|
var senderBucket = senderBuckets[senderKey] ?? TokenBucket(
|
||||||
|
capacity: senderCapacity,
|
||||||
|
tokens: senderCapacity,
|
||||||
|
refillPerSec: senderRefill,
|
||||||
|
lastRefill: now
|
||||||
|
)
|
||||||
|
let senderAllowed = senderBucket.allow(now: now)
|
||||||
|
senderBuckets[senderKey] = senderBucket
|
||||||
|
|
||||||
|
var contentBucket = contentBuckets[contentKey] ?? TokenBucket(
|
||||||
|
capacity: contentCapacity,
|
||||||
|
tokens: contentCapacity,
|
||||||
|
refillPerSec: contentRefill,
|
||||||
|
lastRefill: now
|
||||||
|
)
|
||||||
|
let contentAllowed = contentBucket.allow(now: now)
|
||||||
|
contentBuckets[contentKey] = contentBucket
|
||||||
|
|
||||||
|
return senderAllowed && contentAllowed
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func reset() {
|
||||||
|
senderBuckets.removeAll()
|
||||||
|
contentBuckets.removeAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,210 @@
|
|||||||
|
//
|
||||||
|
// MinimalDistancePalette.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Lightweight palette generator that keeps peer colors evenly spaced.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
import SwiftUI
|
||||||
|
|
||||||
|
final class MinimalDistancePalette {
|
||||||
|
struct Config {
|
||||||
|
let slotCount: Int
|
||||||
|
let avoidCenterHue: Double
|
||||||
|
let avoidHueDelta: Double
|
||||||
|
let saturationLight: Double
|
||||||
|
let saturationDark: Double
|
||||||
|
let baseBrightnessLight: Double
|
||||||
|
let baseBrightnessDark: Double
|
||||||
|
let ringBrightnessDeltaLight: Double
|
||||||
|
let ringBrightnessDeltaDark: Double
|
||||||
|
let preferredBiasWeight: Double
|
||||||
|
let goldenStep: Int
|
||||||
|
|
||||||
|
init(
|
||||||
|
slotCount: Int,
|
||||||
|
avoidCenterHue: Double,
|
||||||
|
avoidHueDelta: Double,
|
||||||
|
saturationLight: Double,
|
||||||
|
saturationDark: Double,
|
||||||
|
baseBrightnessLight: Double,
|
||||||
|
baseBrightnessDark: Double,
|
||||||
|
ringBrightnessDeltaLight: Double,
|
||||||
|
ringBrightnessDeltaDark: Double,
|
||||||
|
preferredBiasWeight: Double = 0.05,
|
||||||
|
goldenStep: Int = 7
|
||||||
|
) {
|
||||||
|
self.slotCount = slotCount
|
||||||
|
self.avoidCenterHue = avoidCenterHue
|
||||||
|
self.avoidHueDelta = avoidHueDelta
|
||||||
|
self.saturationLight = saturationLight
|
||||||
|
self.saturationDark = saturationDark
|
||||||
|
self.baseBrightnessLight = baseBrightnessLight
|
||||||
|
self.baseBrightnessDark = baseBrightnessDark
|
||||||
|
self.ringBrightnessDeltaLight = ringBrightnessDeltaLight
|
||||||
|
self.ringBrightnessDeltaDark = ringBrightnessDeltaDark
|
||||||
|
self.preferredBiasWeight = preferredBiasWeight
|
||||||
|
self.goldenStep = goldenStep
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private struct Entry {
|
||||||
|
let slot: Int
|
||||||
|
let ring: Int
|
||||||
|
let hue: Double
|
||||||
|
}
|
||||||
|
|
||||||
|
private let config: Config
|
||||||
|
private var currentSeeds: [String: String] = [:]
|
||||||
|
private var entries: [String: Entry] = [:]
|
||||||
|
private var previousEntries: [String: Entry] = [:]
|
||||||
|
|
||||||
|
init(config: Config) {
|
||||||
|
self.config = config
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
func ensurePalette(for seeds: [String: String]) {
|
||||||
|
guard seeds != currentSeeds || entries.count != seeds.count else { return }
|
||||||
|
previousEntries = entries
|
||||||
|
currentSeeds = seeds
|
||||||
|
rebuildEntries()
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
func color(for identifier: String, isDark: Bool) -> Color? {
|
||||||
|
guard let entry = entries[identifier] else { return nil }
|
||||||
|
let saturation = isDark ? config.saturationDark : config.saturationLight
|
||||||
|
let baseBrightness = isDark ? config.baseBrightnessDark : config.baseBrightnessLight
|
||||||
|
let ringDelta = isDark ? config.ringBrightnessDeltaDark : config.ringBrightnessDeltaLight
|
||||||
|
let brightness = min(1.0, max(0.0, baseBrightness + ringDelta * Double(entry.ring)))
|
||||||
|
return Color(hue: entry.hue, saturation: saturation, brightness: brightness)
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
func reset() {
|
||||||
|
currentSeeds.removeAll()
|
||||||
|
entries.removeAll()
|
||||||
|
previousEntries.removeAll()
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
private func rebuildEntries() {
|
||||||
|
guard !currentSeeds.isEmpty else {
|
||||||
|
entries.removeAll()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
let slotCount = max(8, config.slotCount)
|
||||||
|
var slots: [Double] = []
|
||||||
|
for idx in 0..<slotCount {
|
||||||
|
let hue = Double(idx) / Double(slotCount)
|
||||||
|
if abs(hue - config.avoidCenterHue) < config.avoidHueDelta {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
slots.append(hue)
|
||||||
|
}
|
||||||
|
if slots.isEmpty {
|
||||||
|
for idx in 0..<slotCount {
|
||||||
|
slots.append(Double(idx) / Double(slotCount))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func circularDistance(_ a: Double, _ b: Double) -> Double {
|
||||||
|
let diff = abs(a - b)
|
||||||
|
return diff > 0.5 ? 1.0 - diff : diff
|
||||||
|
}
|
||||||
|
|
||||||
|
let peerIDs = currentSeeds.keys.sorted()
|
||||||
|
let preferredIndex: [String: Int] = Dictionary(uniqueKeysWithValues: peerIDs.map { id in
|
||||||
|
let seed = currentSeeds[id] ?? id
|
||||||
|
let hash = seed.djb2()
|
||||||
|
let index = Int(hash % UInt64(slots.count))
|
||||||
|
return (id, index)
|
||||||
|
})
|
||||||
|
|
||||||
|
var mapping: [String: Entry] = [:]
|
||||||
|
var usedSlots = Set<Int>()
|
||||||
|
var usedHues: [Double] = []
|
||||||
|
|
||||||
|
let prior = entries.isEmpty ? previousEntries : entries
|
||||||
|
for (id, entry) in prior {
|
||||||
|
guard currentSeeds.keys.contains(id), entry.slot < slots.count else { continue }
|
||||||
|
let hue = slots[entry.slot]
|
||||||
|
mapping[id] = Entry(slot: entry.slot, ring: entry.ring, hue: hue)
|
||||||
|
usedSlots.insert(entry.slot)
|
||||||
|
usedHues.append(hue)
|
||||||
|
}
|
||||||
|
|
||||||
|
let unassigned = peerIDs.filter { mapping[$0] == nil }
|
||||||
|
for id in unassigned {
|
||||||
|
let preferred = preferredIndex[id] ?? 0
|
||||||
|
if !usedSlots.contains(preferred), preferred < slots.count {
|
||||||
|
let hue = slots[preferred]
|
||||||
|
mapping[id] = Entry(slot: preferred, ring: 0, hue: hue)
|
||||||
|
usedSlots.insert(preferred)
|
||||||
|
usedHues.append(hue)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
var bestSlot: Int?
|
||||||
|
var bestScore = -Double.infinity
|
||||||
|
for slot in 0..<slots.count where !usedSlots.contains(slot) {
|
||||||
|
let hue = slots[slot]
|
||||||
|
let minDistance = usedHues.isEmpty ? 1.0 : usedHues.map { circularDistance(hue, $0) }.min() ?? 1.0
|
||||||
|
let bias = 1.0 - (Double((abs(slot - (preferredIndex[id] ?? 0)) % slots.count)) / Double(slots.count))
|
||||||
|
let score = minDistance + config.preferredBiasWeight * bias
|
||||||
|
if score > bestScore {
|
||||||
|
bestScore = score
|
||||||
|
bestSlot = slot
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let slot = bestSlot {
|
||||||
|
let hue = slots[slot]
|
||||||
|
mapping[id] = Entry(slot: slot, ring: 0, hue: hue)
|
||||||
|
usedSlots.insert(slot)
|
||||||
|
usedHues.append(hue)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let remaining = peerIDs.filter { mapping[$0] == nil }
|
||||||
|
if !remaining.isEmpty {
|
||||||
|
for (index, id) in remaining.enumerated() {
|
||||||
|
let preferred = preferredIndex[id] ?? 0
|
||||||
|
let slot = (preferred + index * config.goldenStep) % slots.count
|
||||||
|
let hue = slots[slot]
|
||||||
|
mapping[id] = Entry(slot: slot, ring: 1, hue: hue)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
entries = mapping
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
extension MinimalDistancePalette.Config {
|
||||||
|
static let mesh = MinimalDistancePalette.Config(
|
||||||
|
slotCount: TransportConfig.uiPeerPaletteSlots,
|
||||||
|
avoidCenterHue: 30.0 / 360.0,
|
||||||
|
avoidHueDelta: TransportConfig.uiColorHueAvoidanceDelta,
|
||||||
|
saturationLight: 0.70,
|
||||||
|
saturationDark: 0.80,
|
||||||
|
baseBrightnessLight: 0.45,
|
||||||
|
baseBrightnessDark: 0.75,
|
||||||
|
ringBrightnessDeltaLight: TransportConfig.uiPeerPaletteRingBrightnessDeltaLight,
|
||||||
|
ringBrightnessDeltaDark: TransportConfig.uiPeerPaletteRingBrightnessDeltaDark
|
||||||
|
)
|
||||||
|
|
||||||
|
static let nostr = MinimalDistancePalette.Config(
|
||||||
|
slotCount: TransportConfig.uiPeerPaletteSlots,
|
||||||
|
avoidCenterHue: 30.0 / 360.0,
|
||||||
|
avoidHueDelta: TransportConfig.uiColorHueAvoidanceDelta,
|
||||||
|
saturationLight: 0.70,
|
||||||
|
saturationDark: 0.80,
|
||||||
|
baseBrightnessLight: 0.45,
|
||||||
|
baseBrightnessDark: 0.75,
|
||||||
|
ringBrightnessDeltaLight: TransportConfig.uiPeerPaletteRingBrightnessDeltaLight,
|
||||||
|
ringBrightnessDeltaDark: TransportConfig.uiPeerPaletteRingBrightnessDeltaDark
|
||||||
|
)
|
||||||
|
}
|
||||||
@@ -0,0 +1,190 @@
|
|||||||
|
//
|
||||||
|
// PublicMessagePipeline.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Handles batching and deduplication of public chat messages before surfacing them to the UI.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
protocol PublicMessagePipelineDelegate: AnyObject {
|
||||||
|
func pipelineCurrentMessages(_ pipeline: PublicMessagePipeline) -> [BitchatMessage]
|
||||||
|
func pipeline(_ pipeline: PublicMessagePipeline, setMessages messages: [BitchatMessage])
|
||||||
|
func pipeline(_ pipeline: PublicMessagePipeline, normalizeContent content: String) -> String
|
||||||
|
func pipeline(_ pipeline: PublicMessagePipeline, contentTimestampForKey key: String) -> Date?
|
||||||
|
func pipeline(_ pipeline: PublicMessagePipeline, recordContentKey key: String, timestamp: Date)
|
||||||
|
func pipelineTrimMessages(_ pipeline: PublicMessagePipeline)
|
||||||
|
func pipelinePrewarmMessage(_ pipeline: PublicMessagePipeline, message: BitchatMessage)
|
||||||
|
func pipelineSetBatchingState(_ pipeline: PublicMessagePipeline, isBatching: Bool)
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
final class PublicMessagePipeline {
|
||||||
|
weak var delegate: PublicMessagePipelineDelegate?
|
||||||
|
|
||||||
|
private var buffer: [BitchatMessage] = []
|
||||||
|
private var timer: Timer?
|
||||||
|
private let baseFlushInterval: TimeInterval
|
||||||
|
private var dynamicFlushInterval: TimeInterval
|
||||||
|
private var recentBatchSizes: [Int] = []
|
||||||
|
private let maxRecentBatchSamples: Int
|
||||||
|
private let dedupWindow: TimeInterval
|
||||||
|
private var activeChannel: ChannelID = .mesh
|
||||||
|
|
||||||
|
init(
|
||||||
|
baseFlushInterval: TimeInterval = TransportConfig.basePublicFlushInterval,
|
||||||
|
maxRecentBatchSamples: Int = 10,
|
||||||
|
dedupWindow: TimeInterval = 1.0
|
||||||
|
) {
|
||||||
|
self.baseFlushInterval = baseFlushInterval
|
||||||
|
self.dynamicFlushInterval = baseFlushInterval
|
||||||
|
self.maxRecentBatchSamples = maxRecentBatchSamples
|
||||||
|
self.dedupWindow = dedupWindow
|
||||||
|
}
|
||||||
|
|
||||||
|
deinit {
|
||||||
|
timer?.invalidate()
|
||||||
|
}
|
||||||
|
|
||||||
|
func updateActiveChannel(_ channel: ChannelID) {
|
||||||
|
activeChannel = channel
|
||||||
|
}
|
||||||
|
|
||||||
|
func enqueue(_ message: BitchatMessage) {
|
||||||
|
buffer.append(message)
|
||||||
|
scheduleFlush()
|
||||||
|
}
|
||||||
|
|
||||||
|
func flushIfNeeded() {
|
||||||
|
flushBuffer()
|
||||||
|
}
|
||||||
|
|
||||||
|
func reset() {
|
||||||
|
timer?.invalidate()
|
||||||
|
timer = nil
|
||||||
|
buffer.removeAll(keepingCapacity: false)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private extension PublicMessagePipeline {
|
||||||
|
func scheduleFlush() {
|
||||||
|
guard timer == nil else { return }
|
||||||
|
timer = Timer.scheduledTimer(withTimeInterval: dynamicFlushInterval, repeats: false) { [weak self] _ in
|
||||||
|
guard let self else { return }
|
||||||
|
Task { @MainActor in
|
||||||
|
self.flushBuffer()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func flushBuffer() {
|
||||||
|
timer?.invalidate()
|
||||||
|
timer = nil
|
||||||
|
guard !buffer.isEmpty else { return }
|
||||||
|
guard let delegate = delegate else {
|
||||||
|
buffer.removeAll(keepingCapacity: false)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
delegate.pipelineSetBatchingState(self, isBatching: true)
|
||||||
|
|
||||||
|
var existingIDs = Set(delegate.pipelineCurrentMessages(self).map { $0.id })
|
||||||
|
var pending: [(message: BitchatMessage, contentKey: String)] = []
|
||||||
|
var batchContentLatest: [String: Date] = [:]
|
||||||
|
|
||||||
|
for message in buffer {
|
||||||
|
if existingIDs.contains(message.id) { continue }
|
||||||
|
let contentKey = delegate.pipeline(self, normalizeContent: message.content)
|
||||||
|
if let ts = delegate.pipeline(self, contentTimestampForKey: contentKey),
|
||||||
|
abs(ts.timeIntervalSince(message.timestamp)) < dedupWindow {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if let ts = batchContentLatest[contentKey],
|
||||||
|
abs(ts.timeIntervalSince(message.timestamp)) < dedupWindow {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
existingIDs.insert(message.id)
|
||||||
|
pending.append((message, contentKey))
|
||||||
|
batchContentLatest[contentKey] = message.timestamp
|
||||||
|
}
|
||||||
|
|
||||||
|
buffer.removeAll(keepingCapacity: true)
|
||||||
|
guard !pending.isEmpty else {
|
||||||
|
delegate.pipelineSetBatchingState(self, isBatching: false)
|
||||||
|
if !buffer.isEmpty { scheduleFlush() }
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
pending.sort { $0.message.timestamp < $1.message.timestamp }
|
||||||
|
|
||||||
|
var messages = delegate.pipelineCurrentMessages(self)
|
||||||
|
let threshold = lateInsertThreshold(for: activeChannel)
|
||||||
|
let lastTimestamp = messages.last?.timestamp ?? .distantPast
|
||||||
|
|
||||||
|
for item in pending {
|
||||||
|
let message = item.message
|
||||||
|
if threshold == 0 || message.timestamp < lastTimestamp.addingTimeInterval(-threshold) {
|
||||||
|
let index = insertionIndex(for: message.timestamp, in: messages)
|
||||||
|
if index >= messages.count {
|
||||||
|
messages.append(message)
|
||||||
|
} else {
|
||||||
|
messages.insert(message, at: index)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
messages.append(message)
|
||||||
|
}
|
||||||
|
delegate.pipeline(self, recordContentKey: item.contentKey, timestamp: message.timestamp)
|
||||||
|
}
|
||||||
|
|
||||||
|
delegate.pipeline(self, setMessages: messages)
|
||||||
|
delegate.pipelineTrimMessages(self)
|
||||||
|
|
||||||
|
updateFlushInterval(withBatchSize: pending.count)
|
||||||
|
|
||||||
|
for item in pending {
|
||||||
|
delegate.pipelinePrewarmMessage(self, message: item.message)
|
||||||
|
}
|
||||||
|
|
||||||
|
delegate.pipelineSetBatchingState(self, isBatching: false)
|
||||||
|
|
||||||
|
if !buffer.isEmpty {
|
||||||
|
scheduleFlush()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func updateFlushInterval(withBatchSize size: Int) {
|
||||||
|
recentBatchSizes.append(size)
|
||||||
|
if recentBatchSizes.count > maxRecentBatchSamples {
|
||||||
|
recentBatchSizes.removeFirst(recentBatchSizes.count - maxRecentBatchSamples)
|
||||||
|
}
|
||||||
|
let avg = recentBatchSizes.isEmpty
|
||||||
|
? 0.0
|
||||||
|
: Double(recentBatchSizes.reduce(0, +)) / Double(recentBatchSizes.count)
|
||||||
|
dynamicFlushInterval = avg > 100.0 ? 0.12 : baseFlushInterval
|
||||||
|
}
|
||||||
|
|
||||||
|
func lateInsertThreshold(for channel: ChannelID) -> TimeInterval {
|
||||||
|
switch channel {
|
||||||
|
case .mesh:
|
||||||
|
return TransportConfig.uiLateInsertThreshold
|
||||||
|
case .location:
|
||||||
|
return TransportConfig.uiLateInsertThresholdGeo
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func insertionIndex(for timestamp: Date, in messages: [BitchatMessage]) -> 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
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,124 @@
|
|||||||
|
//
|
||||||
|
// PublicTimelineStore.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Maintains mesh and geohash public timelines with simple caps and helpers.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
struct PublicTimelineStore {
|
||||||
|
private var meshTimeline: [BitchatMessage] = []
|
||||||
|
private var geohashTimelines: [String: [BitchatMessage]] = [:]
|
||||||
|
private var pendingGeohashSystemMessages: [String] = []
|
||||||
|
|
||||||
|
private let meshCap: Int
|
||||||
|
private let geohashCap: Int
|
||||||
|
|
||||||
|
init(meshCap: Int, geohashCap: Int) {
|
||||||
|
self.meshCap = meshCap
|
||||||
|
self.geohashCap = geohashCap
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func append(_ message: BitchatMessage, to channel: ChannelID) {
|
||||||
|
switch channel {
|
||||||
|
case .mesh:
|
||||||
|
guard !meshTimeline.contains(where: { $0.id == message.id }) else { return }
|
||||||
|
meshTimeline.append(message)
|
||||||
|
trimMeshTimelineIfNeeded()
|
||||||
|
case .location(let channel):
|
||||||
|
append(message, toGeohash: channel.geohash)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func append(_ message: BitchatMessage, toGeohash geohash: String) {
|
||||||
|
var timeline = geohashTimelines[geohash] ?? []
|
||||||
|
guard !timeline.contains(where: { $0.id == message.id }) else { return }
|
||||||
|
timeline.append(message)
|
||||||
|
trimGeohashTimelineIfNeeded(&timeline)
|
||||||
|
geohashTimelines[geohash] = timeline
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Append message if absent, returning true when stored.
|
||||||
|
mutating func appendIfAbsent(_ message: BitchatMessage, toGeohash geohash: String) -> Bool {
|
||||||
|
var timeline = geohashTimelines[geohash] ?? []
|
||||||
|
guard !timeline.contains(where: { $0.id == message.id }) else { return false }
|
||||||
|
timeline.append(message)
|
||||||
|
trimGeohashTimelineIfNeeded(&timeline)
|
||||||
|
geohashTimelines[geohash] = timeline
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func messages(for channel: ChannelID) -> [BitchatMessage] {
|
||||||
|
switch channel {
|
||||||
|
case .mesh:
|
||||||
|
return meshTimeline
|
||||||
|
case .location(let channel):
|
||||||
|
let cleaned = geohashTimelines[channel.geohash]?.cleanedAndDeduped() ?? []
|
||||||
|
geohashTimelines[channel.geohash] = cleaned
|
||||||
|
return cleaned
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func clear(channel: ChannelID) {
|
||||||
|
switch channel {
|
||||||
|
case .mesh:
|
||||||
|
meshTimeline.removeAll()
|
||||||
|
case .location(let channel):
|
||||||
|
geohashTimelines[channel.geohash] = []
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@discardableResult
|
||||||
|
mutating func removeMessage(withID id: String) -> BitchatMessage? {
|
||||||
|
if let index = meshTimeline.firstIndex(where: { $0.id == id }) {
|
||||||
|
return meshTimeline.remove(at: index)
|
||||||
|
}
|
||||||
|
|
||||||
|
for key in Array(geohashTimelines.keys) {
|
||||||
|
var timeline = geohashTimelines[key] ?? []
|
||||||
|
if let index = timeline.firstIndex(where: { $0.id == id }) {
|
||||||
|
let removed = timeline.remove(at: index)
|
||||||
|
geohashTimelines[key] = timeline.isEmpty ? nil : timeline
|
||||||
|
return removed
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func removeMessages(in geohash: String, where predicate: (BitchatMessage) -> Bool) {
|
||||||
|
var timeline = geohashTimelines[geohash] ?? []
|
||||||
|
timeline.removeAll(where: predicate)
|
||||||
|
geohashTimelines[geohash] = timeline.isEmpty ? nil : timeline
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func mutateGeohash(_ geohash: String, _ transform: (inout [BitchatMessage]) -> Void) {
|
||||||
|
var timeline = geohashTimelines[geohash] ?? []
|
||||||
|
transform(&timeline)
|
||||||
|
geohashTimelines[geohash] = timeline.isEmpty ? nil : timeline
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func queueGeohashSystemMessage(_ content: String) {
|
||||||
|
pendingGeohashSystemMessages.append(content)
|
||||||
|
}
|
||||||
|
|
||||||
|
mutating func drainPendingGeohashSystemMessages() -> [String] {
|
||||||
|
defer { pendingGeohashSystemMessages.removeAll(keepingCapacity: false) }
|
||||||
|
return pendingGeohashSystemMessages
|
||||||
|
}
|
||||||
|
|
||||||
|
func geohashKeys() -> [String] {
|
||||||
|
Array(geohashTimelines.keys)
|
||||||
|
}
|
||||||
|
|
||||||
|
private mutating func trimMeshTimelineIfNeeded() {
|
||||||
|
guard meshTimeline.count > meshCap else { return }
|
||||||
|
meshTimeline = Array(meshTimeline.suffix(meshCap))
|
||||||
|
}
|
||||||
|
|
||||||
|
private func trimGeohashTimelineIfNeeded(_ timeline: inout [BitchatMessage]) {
|
||||||
|
guard timeline.count > geohashCap else { return }
|
||||||
|
timeline = Array(timeline.suffix(geohashCap))
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -86,10 +86,6 @@ struct AppInfoView: View {
|
|||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
|
||||||
enum Warning {
|
|
||||||
static let title: LocalizedStringKey = "app_info.warning.title"
|
|
||||||
static let message: LocalizedStringKey = "app_info.warning.message"
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var body: some View {
|
var body: some View {
|
||||||
@@ -192,24 +188,6 @@ struct AppInfoView: View {
|
|||||||
|
|
||||||
FeatureRow(info: Strings.Privacy.panic)
|
FeatureRow(info: Strings.Privacy.panic)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Warning
|
|
||||||
VStack(alignment: .leading, spacing: 6) {
|
|
||||||
SectionHeader(Strings.Warning.title)
|
|
||||||
.foregroundColor(Color.red)
|
|
||||||
|
|
||||||
Text(Strings.Warning.message)
|
|
||||||
.font(.bitchatSystem(size: 14, design: .monospaced))
|
|
||||||
.foregroundColor(Color.red)
|
|
||||||
.fixedSize(horizontal: false, vertical: true)
|
|
||||||
}
|
|
||||||
.padding(.top, 6)
|
|
||||||
.padding(.bottom, 16)
|
|
||||||
.padding(.horizontal)
|
|
||||||
.background(Color.red.opacity(0.1))
|
|
||||||
.cornerRadius(8)
|
|
||||||
|
|
||||||
.padding(.top)
|
|
||||||
}
|
}
|
||||||
.padding()
|
.padding()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,90 @@
|
|||||||
|
//
|
||||||
|
// CommandSuggestionsView.swift
|
||||||
|
// bitchat
|
||||||
|
//
|
||||||
|
// Created by Islam on 29/10/2025.
|
||||||
|
//
|
||||||
|
|
||||||
|
import SwiftUI
|
||||||
|
|
||||||
|
struct CommandSuggestionsView: View {
|
||||||
|
@EnvironmentObject private var viewModel: ChatViewModel
|
||||||
|
@ObservedObject private var locationManager = LocationChannelManager.shared
|
||||||
|
|
||||||
|
@Binding var messageText: String
|
||||||
|
|
||||||
|
let textColor: Color
|
||||||
|
let backgroundColor: Color
|
||||||
|
let secondaryTextColor: Color
|
||||||
|
|
||||||
|
private var filteredCommands: [CommandInfo] {
|
||||||
|
guard messageText.hasPrefix("/") && !messageText.contains(" ") else { return [] }
|
||||||
|
let isGeoPublic = locationManager.selectedChannel.isLocation
|
||||||
|
let isGeoDM = viewModel.selectedPrivateChatPeer?.isGeoDM == true
|
||||||
|
return CommandInfo.all(isGeoPublic: isGeoPublic, isGeoDM: isGeoDM).filter { command in
|
||||||
|
command.alias.starts(with: messageText.lowercased())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var body: some View {
|
||||||
|
VStack(alignment: .leading, spacing: 0) {
|
||||||
|
ForEach(filteredCommands) { command in
|
||||||
|
Button {
|
||||||
|
messageText = command.alias + " "
|
||||||
|
} label: {
|
||||||
|
buttonRow(for: command)
|
||||||
|
}
|
||||||
|
.buttonStyle(.plain)
|
||||||
|
.background(Color.gray.opacity(0.1))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.background(backgroundColor)
|
||||||
|
.overlay(
|
||||||
|
RoundedRectangle(cornerRadius: 4)
|
||||||
|
.stroke(secondaryTextColor.opacity(0.3), lineWidth: 1)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func buttonRow(for command: CommandInfo) -> some View {
|
||||||
|
HStack {
|
||||||
|
Text(command.alias)
|
||||||
|
.font(.bitchatSystem(size: 11, design: .monospaced))
|
||||||
|
.foregroundColor(textColor)
|
||||||
|
.fontWeight(.medium)
|
||||||
|
|
||||||
|
if let placeholder = command.placeholder {
|
||||||
|
Text(placeholder)
|
||||||
|
.font(.bitchatSystem(size: 10, design: .monospaced))
|
||||||
|
.foregroundColor(secondaryTextColor.opacity(0.8))
|
||||||
|
}
|
||||||
|
|
||||||
|
Spacer()
|
||||||
|
|
||||||
|
Text(command.description)
|
||||||
|
.font(.bitchatSystem(size: 10, design: .monospaced))
|
||||||
|
.foregroundColor(secondaryTextColor)
|
||||||
|
}
|
||||||
|
.padding(.horizontal, 12)
|
||||||
|
.padding(.vertical, 3)
|
||||||
|
.frame(maxWidth: .infinity, alignment: .leading)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@available(iOS 17, macOS 14, *)
|
||||||
|
#Preview {
|
||||||
|
@Previewable @State var messageText: String = "/"
|
||||||
|
let keychain = KeychainManager()
|
||||||
|
let viewModel = ChatViewModel(
|
||||||
|
keychain: keychain,
|
||||||
|
idBridge: NostrIdentityBridge(),
|
||||||
|
identityManager: SecureIdentityStateManager(keychain)
|
||||||
|
)
|
||||||
|
|
||||||
|
CommandSuggestionsView(
|
||||||
|
messageText: $messageText,
|
||||||
|
textColor: .green,
|
||||||
|
backgroundColor: .primary,
|
||||||
|
secondaryTextColor: .secondary
|
||||||
|
)
|
||||||
|
.environmentObject(viewModel)
|
||||||
|
}
|
||||||
+1110
-420
File diff suppressed because it is too large
Load Diff
@@ -10,7 +10,7 @@ import SwiftUI
|
|||||||
|
|
||||||
struct FingerprintView: View {
|
struct FingerprintView: View {
|
||||||
@ObservedObject var viewModel: ChatViewModel
|
@ObservedObject var viewModel: ChatViewModel
|
||||||
let peerID: String
|
let peerID: PeerID
|
||||||
@Environment(\.dismiss) var dismiss
|
@Environment(\.dismiss) var dismiss
|
||||||
@Environment(\.colorScheme) var colorScheme
|
@Environment(\.colorScheme) var colorScheme
|
||||||
|
|
||||||
@@ -65,15 +65,12 @@ struct FingerprintView: View {
|
|||||||
|
|
||||||
VStack(alignment: .leading, spacing: 16) {
|
VStack(alignment: .leading, spacing: 16) {
|
||||||
// Prefer short mesh ID for session/encryption status
|
// Prefer short mesh ID for session/encryption status
|
||||||
let statusPeerID: String = {
|
let statusPeerID = viewModel.getShortIDForNoiseKey(peerID)
|
||||||
if peerID.count == 64, let short = viewModel.getShortIDForNoiseKey(peerID) { return short.id }
|
|
||||||
return peerID
|
|
||||||
}()
|
|
||||||
// Resolve a friendly name
|
// Resolve a friendly name
|
||||||
let peerNickname: String = {
|
let peerNickname: String = {
|
||||||
if let p = viewModel.getPeer(byID: PeerID(str: statusPeerID)) { return p.displayName }
|
if let p = viewModel.getPeer(byID: statusPeerID) { return p.displayName }
|
||||||
if let name = viewModel.meshService.peerNickname(peerID: PeerID(str: statusPeerID)) { return name }
|
if let name = viewModel.meshService.peerNickname(peerID: statusPeerID) { return name }
|
||||||
if peerID.count == 64, let data = Data(hexString: peerID) {
|
if let data = peerID.noiseKey {
|
||||||
if let fav = FavoritesPersistenceService.shared.getFavoriteStatus(for: data), !fav.peerNickname.isEmpty { return fav.peerNickname }
|
if let fav = FavoritesPersistenceService.shared.getFavoriteStatus(for: data), !fav.peerNickname.isEmpty { return fav.peerNickname }
|
||||||
let fp = data.sha256Fingerprint()
|
let fp = data.sha256Fingerprint()
|
||||||
if let social = viewModel.identityManager.getSocialIdentity(for: fp) {
|
if let social = viewModel.identityManager.getSocialIdentity(for: fp) {
|
||||||
@@ -84,7 +81,7 @@ struct FingerprintView: View {
|
|||||||
return Strings.unknownPeer()
|
return Strings.unknownPeer()
|
||||||
}()
|
}()
|
||||||
// Accurate encryption state based on short ID session
|
// Accurate encryption state based on short ID session
|
||||||
let encryptionStatus = viewModel.getEncryptionStatus(for: PeerID(str: statusPeerID))
|
let encryptionStatus = viewModel.getEncryptionStatus(for: statusPeerID)
|
||||||
|
|
||||||
HStack {
|
HStack {
|
||||||
if let icon = encryptionStatus.icon {
|
if let icon = encryptionStatus.icon {
|
||||||
@@ -115,7 +112,7 @@ struct FingerprintView: View {
|
|||||||
.font(.bitchatSystem(size: 12, weight: .bold, design: .monospaced))
|
.font(.bitchatSystem(size: 12, weight: .bold, design: .monospaced))
|
||||||
.foregroundColor(textColor.opacity(0.7))
|
.foregroundColor(textColor.opacity(0.7))
|
||||||
|
|
||||||
if let fingerprint = viewModel.getFingerprint(for: PeerID(str: statusPeerID)) {
|
if let fingerprint = viewModel.getFingerprint(for: statusPeerID) {
|
||||||
Text(formatFingerprint(fingerprint))
|
Text(formatFingerprint(fingerprint))
|
||||||
.font(.bitchatSystem(size: 14, design: .monospaced))
|
.font(.bitchatSystem(size: 14, design: .monospaced))
|
||||||
.foregroundColor(textColor)
|
.foregroundColor(textColor)
|
||||||
@@ -176,7 +173,6 @@ struct FingerprintView: View {
|
|||||||
// Verification status
|
// Verification status
|
||||||
if encryptionStatus == .noiseSecured || encryptionStatus == .noiseVerified {
|
if encryptionStatus == .noiseSecured || encryptionStatus == .noiseVerified {
|
||||||
let isVerified = encryptionStatus == .noiseVerified
|
let isVerified = encryptionStatus == .noiseVerified
|
||||||
let peerID = PeerID(str: peerID)
|
|
||||||
|
|
||||||
VStack(spacing: 12) {
|
VStack(spacing: 12) {
|
||||||
Text(isVerified ? Strings.verifiedBadge : Strings.notVerifiedBadge)
|
Text(isVerified ? Strings.verifiedBadge : Strings.notVerifiedBadge)
|
||||||
@@ -240,8 +236,6 @@ struct FingerprintView: View {
|
|||||||
.padding()
|
.padding()
|
||||||
.frame(maxWidth: .infinity, maxHeight: .infinity)
|
.frame(maxWidth: .infinity, maxHeight: .infinity)
|
||||||
.background(backgroundColor)
|
.background(backgroundColor)
|
||||||
.presentationDetents([.large])
|
|
||||||
.presentationDragIndicator(.visible)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private func formatFingerprint(_ fingerprint: String) -> String {
|
private func formatFingerprint(_ fingerprint: String) -> String {
|
||||||
|
|||||||
@@ -40,10 +40,31 @@ struct LocationChannelsSheet: View {
|
|||||||
}
|
}
|
||||||
|
|
||||||
static func levelTitle(for level: GeohashChannelLevel, count: Int) -> String {
|
static func levelTitle(for level: GeohashChannelLevel, count: Int) -> String {
|
||||||
|
// High-precision uncertainty: if count is 0 for high-precision levels,
|
||||||
|
// show "?" because presence broadcasting is disabled for privacy.
|
||||||
|
let isHighPrecision = (level == .neighborhood || level == .block || level == .building)
|
||||||
|
if isHighPrecision && count == 0 {
|
||||||
|
return String(
|
||||||
|
format: String(localized: "location_channels.row_title_unknown", defaultValue: "%@ [? people]"),
|
||||||
|
locale: .current,
|
||||||
|
level.displayName
|
||||||
|
)
|
||||||
|
}
|
||||||
return rowTitle(label: level.displayName, count: count)
|
return rowTitle(label: level.displayName, count: count)
|
||||||
}
|
}
|
||||||
|
|
||||||
static func bookmarkTitle(geohash: String, count: Int) -> String {
|
static func bookmarkTitle(geohash: String, count: Int) -> String {
|
||||||
|
// Check precision for bookmarks too
|
||||||
|
let len = geohash.count
|
||||||
|
// Neighborhood=6, Block=7, Building=8+
|
||||||
|
let isHighPrecision = (len >= 6)
|
||||||
|
if isHighPrecision && count == 0 {
|
||||||
|
return String(
|
||||||
|
format: String(localized: "location_channels.row_title_unknown", defaultValue: "%@ [? people]"),
|
||||||
|
locale: .current,
|
||||||
|
"#\(geohash)"
|
||||||
|
)
|
||||||
|
}
|
||||||
return rowTitle(label: "#\(geohash)", count: count)
|
return rowTitle(label: "#\(geohash)", count: count)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -125,9 +146,6 @@ struct LocationChannelsSheet: View {
|
|||||||
.navigationTitle("")
|
.navigationTitle("")
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
#if os(iOS)
|
|
||||||
.presentationDetents([.large])
|
|
||||||
#endif
|
|
||||||
#if os(macOS)
|
#if os(macOS)
|
||||||
.frame(minWidth: 420, minHeight: 520)
|
.frame(minWidth: 420, minHeight: 520)
|
||||||
#endif
|
#endif
|
||||||
@@ -360,7 +378,7 @@ struct LocationChannelsSheet: View {
|
|||||||
isPresented = false
|
isPresented = false
|
||||||
}
|
}
|
||||||
.padding(.vertical, 6)
|
.padding(.vertical, 6)
|
||||||
.onAppear { bookmarks.resolveNameIfNeeded(for: gh) }
|
.onAppear { bookmarks.resolveBookmarkNameIfNeeded(for: gh) }
|
||||||
|
|
||||||
if index < entries.count - 1 {
|
if index < entries.count - 1 {
|
||||||
sectionDivider
|
sectionDivider
|
||||||
@@ -599,7 +617,7 @@ extension LocationChannelsSheet {
|
|||||||
switch level {
|
switch level {
|
||||||
case .region:
|
case .region:
|
||||||
return ""
|
return ""
|
||||||
default:
|
case .building, .block, .neighborhood, .city, .province:
|
||||||
return "~"
|
return "~"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -78,9 +78,6 @@ struct LocationNotesView: View {
|
|||||||
.navigationTitle("")
|
.navigationTitle("")
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
#if os(iOS)
|
|
||||||
.presentationDetents([.large])
|
|
||||||
#endif
|
|
||||||
.background(backgroundColor)
|
.background(backgroundColor)
|
||||||
.onDisappear { manager.cancel() }
|
.onDisappear { manager.cancel() }
|
||||||
.onChange(of: geohash) { newValue in
|
.onChange(of: geohash) { newValue in
|
||||||
@@ -141,7 +138,7 @@ struct LocationNotesView: View {
|
|||||||
String(
|
String(
|
||||||
format: String(localized: "location_notes.header", comment: "Header displaying the geohash and localized note count"),
|
format: String(localized: "location_notes.header", comment: "Header displaying the geohash and localized note count"),
|
||||||
locale: .current,
|
locale: .current,
|
||||||
geohash, count
|
"\(geohash) ± 1", count
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,191 @@
|
|||||||
|
import SwiftUI
|
||||||
|
|
||||||
|
#if os(iOS)
|
||||||
|
import UIKit
|
||||||
|
private typealias PlatformImage = UIImage
|
||||||
|
#else
|
||||||
|
import AppKit
|
||||||
|
private typealias PlatformImage = NSImage
|
||||||
|
#endif
|
||||||
|
|
||||||
|
struct BlockRevealImageView: View {
|
||||||
|
private let url: URL
|
||||||
|
private let revealProgress: Double?
|
||||||
|
private let isSending: Bool
|
||||||
|
private let onCancel: (() -> Void)?
|
||||||
|
private let initiallyBlurred: Bool
|
||||||
|
private let onOpen: (() -> Void)?
|
||||||
|
private let onDelete: (() -> Void)?
|
||||||
|
|
||||||
|
@State private var platformImage: PlatformImage?
|
||||||
|
@State private var aspectRatio: CGFloat = 1
|
||||||
|
@State private var isBlurred: Bool = false
|
||||||
|
|
||||||
|
init(
|
||||||
|
url: URL,
|
||||||
|
revealProgress: Double?,
|
||||||
|
isSending: Bool,
|
||||||
|
onCancel: (() -> Void)?,
|
||||||
|
initiallyBlurred: Bool = false,
|
||||||
|
onOpen: (() -> Void)? = nil,
|
||||||
|
onDelete: (() -> Void)? = nil
|
||||||
|
) {
|
||||||
|
self.url = url
|
||||||
|
self.revealProgress = revealProgress
|
||||||
|
self.isSending = isSending
|
||||||
|
self.onCancel = onCancel
|
||||||
|
self.initiallyBlurred = initiallyBlurred
|
||||||
|
self.onOpen = onOpen
|
||||||
|
self.onDelete = onDelete
|
||||||
|
}
|
||||||
|
|
||||||
|
private var fraction: Double {
|
||||||
|
guard let revealProgress = revealProgress else { return 1 }
|
||||||
|
return max(0, min(1, revealProgress))
|
||||||
|
}
|
||||||
|
|
||||||
|
var body: some View {
|
||||||
|
ZStack(alignment: .topTrailing) {
|
||||||
|
if let image = platformImage {
|
||||||
|
Image(platformImage: image)
|
||||||
|
.resizable()
|
||||||
|
.aspectRatio(aspectRatio, contentMode: .fit)
|
||||||
|
.clipShape(RoundedRectangle(cornerRadius: 16, style: .continuous))
|
||||||
|
.overlay(
|
||||||
|
RoundedRectangle(cornerRadius: 16, style: .continuous)
|
||||||
|
.stroke(Color.gray.opacity(0.2), lineWidth: 1)
|
||||||
|
)
|
||||||
|
.mask(
|
||||||
|
BlockRevealMask(
|
||||||
|
fraction: fraction,
|
||||||
|
columns: 24,
|
||||||
|
rows: 16
|
||||||
|
)
|
||||||
|
.animation(.easeOut(duration: 0.2), value: fraction)
|
||||||
|
)
|
||||||
|
.blur(radius: isBlurred ? 20 : 0)
|
||||||
|
.overlay {
|
||||||
|
if isBlurred {
|
||||||
|
RoundedRectangle(cornerRadius: 16, style: .continuous)
|
||||||
|
.fill(Color.black.opacity(0.35))
|
||||||
|
.overlay(
|
||||||
|
Image(systemName: "eye.slash.fill")
|
||||||
|
.font(.bitchatSystem(size: 24, weight: .semibold))
|
||||||
|
.foregroundColor(.white.opacity(0.85))
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
RoundedRectangle(cornerRadius: 16, style: .continuous)
|
||||||
|
.fill(Color.gray.opacity(0.2))
|
||||||
|
.frame(height: 200)
|
||||||
|
.overlay(
|
||||||
|
ProgressView()
|
||||||
|
.progressViewStyle(.circular)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
if let onCancel = onCancel, isSending {
|
||||||
|
Button(action: onCancel) {
|
||||||
|
Image(systemName: "xmark")
|
||||||
|
.font(.bitchatSystem(size: 12, weight: .bold))
|
||||||
|
.padding(8)
|
||||||
|
.background(Circle().fill(Color.black.opacity(0.7)))
|
||||||
|
.foregroundColor(.white)
|
||||||
|
.padding(8)
|
||||||
|
}
|
||||||
|
.buttonStyle(.plain)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.onAppear {
|
||||||
|
isBlurred = initiallyBlurred
|
||||||
|
loadImage()
|
||||||
|
}
|
||||||
|
.onChange(of: url) { _ in
|
||||||
|
isBlurred = initiallyBlurred
|
||||||
|
loadImage()
|
||||||
|
}
|
||||||
|
.gesture(mainGesture)
|
||||||
|
}
|
||||||
|
|
||||||
|
private func loadImage() {
|
||||||
|
DispatchQueue.global(qos: .userInitiated).async {
|
||||||
|
#if os(iOS)
|
||||||
|
guard let image = UIImage(contentsOfFile: url.path) else { return }
|
||||||
|
#else
|
||||||
|
guard let image = NSImage(contentsOf: url) else { return }
|
||||||
|
#endif
|
||||||
|
let ratio = image.size.height > 0 ? image.size.width / image.size.height : 1
|
||||||
|
DispatchQueue.main.async {
|
||||||
|
self.platformImage = image
|
||||||
|
self.aspectRatio = ratio
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private var mainGesture: some Gesture {
|
||||||
|
let doubleTap = TapGesture(count: 2).onEnded {
|
||||||
|
guard !isSending else { return }
|
||||||
|
onDelete?()
|
||||||
|
}
|
||||||
|
let singleTap = TapGesture().onEnded {
|
||||||
|
guard !isSending else { return }
|
||||||
|
if isBlurred {
|
||||||
|
withAnimation(.easeOut(duration: 0.2)) {
|
||||||
|
isBlurred = false
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
onOpen?()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let swipe = DragGesture(minimumDistance: 20, coordinateSpace: .local).onEnded { value in
|
||||||
|
guard !isSending else { return }
|
||||||
|
let horizontal = value.translation.width
|
||||||
|
let vertical = value.translation.height
|
||||||
|
guard abs(horizontal) > abs(vertical), abs(horizontal) > 40 else { return }
|
||||||
|
if !isBlurred {
|
||||||
|
withAnimation(.easeInOut(duration: 0.2)) {
|
||||||
|
isBlurred = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return doubleTap.exclusively(before: singleTap).simultaneously(with: swipe)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private struct BlockRevealMask: Shape {
|
||||||
|
let fraction: Double
|
||||||
|
let columns: Int
|
||||||
|
let rows: Int
|
||||||
|
|
||||||
|
func path(in rect: CGRect) -> Path {
|
||||||
|
var path = Path()
|
||||||
|
guard fraction > 0, columns > 0, rows > 0 else { return path }
|
||||||
|
let totalBlocks = columns * rows
|
||||||
|
let revealCount = max(0, min(totalBlocks, Int(ceil(fraction * Double(totalBlocks)))))
|
||||||
|
guard revealCount > 0 else { return path }
|
||||||
|
let blockWidth = rect.width / CGFloat(columns)
|
||||||
|
let blockHeight = rect.height / CGFloat(rows)
|
||||||
|
var remaining = revealCount
|
||||||
|
for row in 0..<rows {
|
||||||
|
for column in 0..<columns {
|
||||||
|
if remaining <= 0 { return path }
|
||||||
|
let x = CGFloat(column) * blockWidth
|
||||||
|
let y = CGFloat(row) * blockHeight
|
||||||
|
path.addRect(CGRect(x: x, y: y, width: blockWidth, height: blockHeight))
|
||||||
|
remaining -= 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return path
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private extension Image {
|
||||||
|
init(platformImage: PlatformImage) {
|
||||||
|
#if os(iOS)
|
||||||
|
self.init(uiImage: platformImage)
|
||||||
|
#else
|
||||||
|
self.init(nsImage: platformImage)
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,123 @@
|
|||||||
|
import SwiftUI
|
||||||
|
import AVFoundation
|
||||||
|
|
||||||
|
struct VoiceNoteView: View {
|
||||||
|
private let url: URL
|
||||||
|
private let isSending: Bool
|
||||||
|
private let sendProgress: Double?
|
||||||
|
private let onCancel: (() -> Void)?
|
||||||
|
|
||||||
|
@Environment(\.colorScheme) private var colorScheme
|
||||||
|
@StateObject private var playback: VoiceNotePlaybackController
|
||||||
|
@State private var waveform: [Float] = []
|
||||||
|
|
||||||
|
init(url: URL, isSending: Bool, sendProgress: Double?, onCancel: (() -> Void)?) {
|
||||||
|
self.url = url
|
||||||
|
self.isSending = isSending
|
||||||
|
self.sendProgress = sendProgress
|
||||||
|
self.onCancel = onCancel
|
||||||
|
_playback = StateObject(wrappedValue: VoiceNotePlaybackController(url: url))
|
||||||
|
}
|
||||||
|
|
||||||
|
private var samples: [Float] {
|
||||||
|
if waveform.isEmpty {
|
||||||
|
return Array(repeating: 0.25, count: 64)
|
||||||
|
}
|
||||||
|
return waveform
|
||||||
|
}
|
||||||
|
|
||||||
|
private var backgroundColor: Color {
|
||||||
|
colorScheme == .dark ? Color.black.opacity(0.6) : Color.white
|
||||||
|
}
|
||||||
|
|
||||||
|
private var borderColor: Color {
|
||||||
|
colorScheme == .dark ? Color.green.opacity(0.3) : Color.green.opacity(0.2)
|
||||||
|
}
|
||||||
|
|
||||||
|
private var durationText: String {
|
||||||
|
let duration = playback.duration
|
||||||
|
guard duration.isFinite, duration > 0 else { return "--:--" }
|
||||||
|
let minutes = Int(duration) / 60
|
||||||
|
let seconds = Int(duration) % 60
|
||||||
|
return String(format: "%02d:%02d", minutes, seconds)
|
||||||
|
}
|
||||||
|
|
||||||
|
private var currentText: String {
|
||||||
|
let current = playback.currentTime
|
||||||
|
guard current.isFinite, current > 0 else { return "00:00" }
|
||||||
|
let minutes = Int(current) / 60
|
||||||
|
let seconds = Int(current) % 60
|
||||||
|
return String(format: "%02d:%02d", minutes, seconds)
|
||||||
|
}
|
||||||
|
|
||||||
|
private var playbackLabel: String {
|
||||||
|
playback.isPlaying ? currentText + "/" + durationText : durationText
|
||||||
|
}
|
||||||
|
|
||||||
|
var body: some View {
|
||||||
|
HStack(spacing: 12) {
|
||||||
|
Button(action: playback.togglePlayback) {
|
||||||
|
Image(systemName: playback.isPlaying ? "pause.fill" : "play.fill")
|
||||||
|
.foregroundColor(.white)
|
||||||
|
.frame(width: 36, height: 36)
|
||||||
|
.background(Circle().fill(Color.green))
|
||||||
|
}
|
||||||
|
.buttonStyle(.plain)
|
||||||
|
|
||||||
|
WaveformView(
|
||||||
|
samples: samples,
|
||||||
|
playbackProgress: playback.progress,
|
||||||
|
sendProgress: sendProgress,
|
||||||
|
onSeek: { fraction in
|
||||||
|
playback.seek(to: fraction)
|
||||||
|
},
|
||||||
|
isInteractive: playback.isPlaying
|
||||||
|
)
|
||||||
|
|
||||||
|
Text(playbackLabel)
|
||||||
|
.font(.bitchatSystem(size: 13, design: .monospaced))
|
||||||
|
.foregroundColor(Color.secondary)
|
||||||
|
|
||||||
|
if let onCancel = onCancel, isSending {
|
||||||
|
Button(action: onCancel) {
|
||||||
|
Image(systemName: "xmark")
|
||||||
|
.font(.bitchatSystem(size: 12, weight: .bold))
|
||||||
|
.frame(width: 28, height: 28)
|
||||||
|
.background(Circle().fill(Color.red.opacity(0.9)))
|
||||||
|
.foregroundColor(.white)
|
||||||
|
}
|
||||||
|
.buttonStyle(.plain)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.padding(12)
|
||||||
|
.background(
|
||||||
|
RoundedRectangle(cornerRadius: 14)
|
||||||
|
.fill(backgroundColor)
|
||||||
|
.shadow(color: Color.black.opacity(colorScheme == .dark ? 0.3 : 0.1), radius: 6, x: 0, y: 2)
|
||||||
|
)
|
||||||
|
.overlay(
|
||||||
|
RoundedRectangle(cornerRadius: 14)
|
||||||
|
.stroke(borderColor, lineWidth: 1)
|
||||||
|
)
|
||||||
|
.task {
|
||||||
|
// Defer loading to let UI settle after view appears
|
||||||
|
try? await Task.sleep(nanoseconds: 100_000_000) // 0.1s
|
||||||
|
playback.loadDuration()
|
||||||
|
await withCheckedContinuation { continuation in
|
||||||
|
WaveformCache.shared.waveform(for: url, completion: { bins in
|
||||||
|
waveform = bins
|
||||||
|
continuation.resume()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.onChange(of: url) { newValue in
|
||||||
|
WaveformCache.shared.waveform(for: newValue, completion: { bins in
|
||||||
|
self.waveform = bins
|
||||||
|
})
|
||||||
|
playback.replaceURL(newValue)
|
||||||
|
}
|
||||||
|
.onDisappear {
|
||||||
|
playback.stop()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,67 @@
|
|||||||
|
import SwiftUI
|
||||||
|
|
||||||
|
struct WaveformView: View {
|
||||||
|
let samples: [Float]
|
||||||
|
let playbackProgress: Double
|
||||||
|
let sendProgress: Double?
|
||||||
|
let onSeek: ((Double) -> Void)?
|
||||||
|
let isInteractive: Bool
|
||||||
|
|
||||||
|
private var clampedPlayback: Double {
|
||||||
|
max(0, min(1, playbackProgress))
|
||||||
|
}
|
||||||
|
|
||||||
|
private var clampedSend: Double? {
|
||||||
|
guard let sendProgress = sendProgress else { return nil }
|
||||||
|
return max(0, min(1, sendProgress))
|
||||||
|
}
|
||||||
|
|
||||||
|
var body: some View {
|
||||||
|
GeometryReader { geometry in
|
||||||
|
ZStack {
|
||||||
|
Canvas { context, size in
|
||||||
|
guard !samples.isEmpty else { return }
|
||||||
|
let width = max(size.width, 1)
|
||||||
|
let height = max(size.height, 1)
|
||||||
|
let barWidth = max(width / CGFloat(samples.count), 1)
|
||||||
|
for (index, sample) in samples.enumerated() {
|
||||||
|
let normalized = max(0, min(sample, 1))
|
||||||
|
let barHeight = CGFloat(normalized) * height
|
||||||
|
let originX = CGFloat(index) * barWidth
|
||||||
|
let rect = CGRect(
|
||||||
|
x: originX,
|
||||||
|
y: (height - barHeight) / 2,
|
||||||
|
width: max(barWidth * 0.7, 1),
|
||||||
|
height: barHeight
|
||||||
|
)
|
||||||
|
let binPosition = Double(index) / Double(samples.count)
|
||||||
|
let color: Color
|
||||||
|
if binPosition <= clampedPlayback {
|
||||||
|
color = Color.green
|
||||||
|
} else if let send = clampedSend, binPosition <= send {
|
||||||
|
color = Color.blue
|
||||||
|
} else {
|
||||||
|
color = Color.gray.opacity(0.35)
|
||||||
|
}
|
||||||
|
context.fill(Path(rect), with: .color(color))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.frame(width: geometry.size.width, height: geometry.size.height)
|
||||||
|
|
||||||
|
if isInteractive, let onSeek = onSeek {
|
||||||
|
Color.clear
|
||||||
|
.contentShape(Rectangle())
|
||||||
|
.gesture(
|
||||||
|
DragGesture(minimumDistance: 0)
|
||||||
|
.onEnded { value in
|
||||||
|
guard geometry.size.width > 0 else { return }
|
||||||
|
let fraction = max(0, min(1, value.location.x / geometry.size.width))
|
||||||
|
onSeek(fraction)
|
||||||
|
}
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.frame(height: 48)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -4,9 +4,9 @@ struct MeshPeerList: View {
|
|||||||
@ObservedObject var viewModel: ChatViewModel
|
@ObservedObject var viewModel: ChatViewModel
|
||||||
let textColor: Color
|
let textColor: Color
|
||||||
let secondaryTextColor: Color
|
let secondaryTextColor: Color
|
||||||
let onTapPeer: (String) -> Void
|
let onTapPeer: (PeerID) -> Void
|
||||||
let onToggleFavorite: (String) -> Void
|
let onToggleFavorite: (PeerID) -> Void
|
||||||
let onShowFingerprint: (String) -> Void
|
let onShowFingerprint: (PeerID) -> Void
|
||||||
@Environment(\.colorScheme) var colorScheme
|
@Environment(\.colorScheme) var colorScheme
|
||||||
|
|
||||||
@State private var orderedIDs: [String] = []
|
@State private var orderedIDs: [String] = []
|
||||||
@@ -130,7 +130,7 @@ struct MeshPeerList: View {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if !isMe {
|
if !isMe {
|
||||||
Button(action: { onToggleFavorite(peer.peerID.id) }) {
|
Button(action: { onToggleFavorite(peer.peerID) }) {
|
||||||
Image(systemName: (peer.favoriteStatus?.isFavorite ?? false) ? "star.fill" : "star")
|
Image(systemName: (peer.favoriteStatus?.isFavorite ?? false) ? "star.fill" : "star")
|
||||||
.font(.bitchatSystem(size: 12))
|
.font(.bitchatSystem(size: 12))
|
||||||
.foregroundColor((peer.favoriteStatus?.isFavorite ?? false) ? .yellow : secondaryTextColor)
|
.foregroundColor((peer.favoriteStatus?.isFavorite ?? false) ? .yellow : secondaryTextColor)
|
||||||
@@ -142,8 +142,8 @@ struct MeshPeerList: View {
|
|||||||
.padding(.vertical, 4)
|
.padding(.vertical, 4)
|
||||||
.padding(.top, idx == 0 ? 10 : 0)
|
.padding(.top, idx == 0 ? 10 : 0)
|
||||||
.contentShape(Rectangle())
|
.contentShape(Rectangle())
|
||||||
.onTapGesture { if !isMe { onTapPeer(peer.peerID.id) } }
|
.onTapGesture { if !isMe { onTapPeer(peer.peerID) } }
|
||||||
.onTapGesture(count: 2) { if !isMe { onShowFingerprint(peer.peerID.id) } }
|
.onTapGesture(count: 2) { if !isMe { onShowFingerprint(peer.peerID) } }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// Seed and update order outside result builder
|
// Seed and update order outside result builder
|
||||||
|
|||||||
@@ -388,10 +388,6 @@ struct VerificationSheetView: View {
|
|||||||
.padding(.vertical, 14)
|
.padding(.vertical, 14)
|
||||||
}
|
}
|
||||||
.background(backgroundColor)
|
.background(backgroundColor)
|
||||||
#if os(iOS)
|
|
||||||
.presentationDetents([.large])
|
|
||||||
.presentationDragIndicator(.visible)
|
|
||||||
#endif
|
|
||||||
.onDisappear { showingScanner = false }
|
.onDisappear { showingScanner = false }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -10,6 +10,7 @@ import Foundation
|
|||||||
|
|
||||||
final class PreviewKeychainManager: KeychainManagerProtocol {
|
final class PreviewKeychainManager: KeychainManagerProtocol {
|
||||||
private var storage: [String: Data] = [:]
|
private var storage: [String: Data] = [:]
|
||||||
|
private var serviceStorage: [String: [String: Data]] = [:]
|
||||||
init() {}
|
init() {}
|
||||||
|
|
||||||
func saveIdentityKey(_ keyData: Data, forKey key: String) -> Bool {
|
func saveIdentityKey(_ keyData: Data, forKey key: String) -> Bool {
|
||||||
@@ -28,6 +29,7 @@ final class PreviewKeychainManager: KeychainManagerProtocol {
|
|||||||
|
|
||||||
func deleteAllKeychainData() -> Bool {
|
func deleteAllKeychainData() -> Bool {
|
||||||
storage.removeAll()
|
storage.removeAll()
|
||||||
|
serviceStorage.removeAll()
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -38,4 +40,34 @@ final class PreviewKeychainManager: KeychainManagerProtocol {
|
|||||||
func verifyIdentityKeyExists() -> Bool {
|
func verifyIdentityKeyExists() -> Bool {
|
||||||
storage["identity_noiseStaticKey"] != nil
|
storage["identity_noiseStaticKey"] != nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// BCH-01-009: New methods with proper error classification
|
||||||
|
func getIdentityKeyWithResult(forKey key: String) -> KeychainReadResult {
|
||||||
|
if let data = storage[key] {
|
||||||
|
return .success(data)
|
||||||
|
}
|
||||||
|
return .itemNotFound
|
||||||
|
}
|
||||||
|
|
||||||
|
func saveIdentityKeyWithResult(_ keyData: Data, forKey key: String) -> KeychainSaveResult {
|
||||||
|
storage[key] = keyData
|
||||||
|
return .success
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Generic Data Storage (consolidated from KeychainHelper)
|
||||||
|
|
||||||
|
func save(key: String, data: Data, service: String, accessible: CFString?) {
|
||||||
|
if serviceStorage[service] == nil {
|
||||||
|
serviceStorage[service] = [:]
|
||||||
|
}
|
||||||
|
serviceStorage[service]?[key] = data
|
||||||
|
}
|
||||||
|
|
||||||
|
func load(key: String, service: String) -> Data? {
|
||||||
|
serviceStorage[service]?[key]
|
||||||
|
}
|
||||||
|
|
||||||
|
func delete(key: String, service: String) {
|
||||||
|
serviceStorage[service]?.removeValue(forKey: key)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -10,11 +10,19 @@
|
|||||||
</array>
|
</array>
|
||||||
<key>com.apple.security.device.bluetooth</key>
|
<key>com.apple.security.device.bluetooth</key>
|
||||||
<true/>
|
<true/>
|
||||||
|
<key>com.apple.security.device.microphone</key>
|
||||||
|
<true/>
|
||||||
<key>com.apple.security.personal-information.location</key>
|
<key>com.apple.security.personal-information.location</key>
|
||||||
<true/>
|
<true/>
|
||||||
<key>com.apple.security.network.client</key>
|
<key>com.apple.security.network.client</key>
|
||||||
<true/>
|
<true/>
|
||||||
<key>com.apple.security.network.server</key>
|
<key>com.apple.security.network.server</key>
|
||||||
<true/>
|
<true/>
|
||||||
|
<key>com.apple.security.files.user-selected.read-only</key>
|
||||||
|
<true/>
|
||||||
|
<key>com.apple.security.files.user-selected.read-write</key>
|
||||||
|
<true/>
|
||||||
|
<key>com.apple.security.assets.pictures.read-only</key>
|
||||||
|
<true/>
|
||||||
</dict>
|
</dict>
|
||||||
</plist>
|
</plist>
|
||||||
|
|||||||
@@ -298,5 +298,5 @@ private final class MockBitchatDelegate: BitchatDelegate {
|
|||||||
func didUpdateMessageDeliveryStatus(_ messageID: String, status: DeliveryStatus) {}
|
func didUpdateMessageDeliveryStatus(_ messageID: String, status: DeliveryStatus) {}
|
||||||
func didReceiveNoisePayload(from peerID: PeerID, type: NoisePayloadType, payload: Data, timestamp: Date) {}
|
func didReceiveNoisePayload(from peerID: PeerID, type: NoisePayloadType, payload: Data, timestamp: Date) {}
|
||||||
func didUpdateBluetoothState(_ state: CBManagerState) {}
|
func didUpdateBluetoothState(_ state: CBManagerState) {}
|
||||||
func didReceivePublicMessage(from peerID: String, nickname: String, content: String, timestamp: Date) {}
|
func didReceivePublicMessage(from peerID: PeerID, nickname: String, content: String, timestamp: Date, messageID: String?) {}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,285 @@
|
|||||||
|
//
|
||||||
|
// ChatViewModelExtensionsTests.swift
|
||||||
|
// bitchatTests
|
||||||
|
//
|
||||||
|
// Tests for ChatViewModel extensions (PrivateChat, Nostr, Tor).
|
||||||
|
//
|
||||||
|
|
||||||
|
import Testing
|
||||||
|
import Foundation
|
||||||
|
import Combine
|
||||||
|
@testable import bitchat
|
||||||
|
|
||||||
|
// MARK: - Test Helpers
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
private func makeTestableViewModel() -> (viewModel: ChatViewModel, transport: MockTransport) {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
let keychainHelper = MockKeychainHelper()
|
||||||
|
let idBridge = NostrIdentityBridge(keychain: keychainHelper)
|
||||||
|
let identityManager = MockIdentityManager(keychain)
|
||||||
|
let transport = MockTransport()
|
||||||
|
|
||||||
|
let viewModel = ChatViewModel(
|
||||||
|
keychain: keychain,
|
||||||
|
idBridge: idBridge,
|
||||||
|
identityManager: identityManager,
|
||||||
|
transport: transport
|
||||||
|
)
|
||||||
|
|
||||||
|
return (viewModel, transport)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Private Chat Extension Tests
|
||||||
|
|
||||||
|
struct ChatViewModelPrivateChatExtensionTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func sendPrivateMessage_mesh_storesAndSends() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
// Use valid hex string for PeerID (32 bytes = 64 hex chars for Noise key usually, or just valid hex)
|
||||||
|
let validHex = "0102030405060708090a0b0c0d0e0f100102030405060708090a0b0c0d0e0f10"
|
||||||
|
let peerID = PeerID(str: validHex)
|
||||||
|
|
||||||
|
// Simulate connection
|
||||||
|
transport.connectedPeers.insert(peerID)
|
||||||
|
transport.peerNicknames[peerID] = "MeshUser"
|
||||||
|
|
||||||
|
viewModel.sendPrivateMessage("Hello Mesh", to: peerID)
|
||||||
|
|
||||||
|
// Verify transport was called
|
||||||
|
// Note: MockTransport stores sent messages
|
||||||
|
// Since sendPrivateMessage delegates to MessageRouter which delegates to Transport...
|
||||||
|
// We need to ensure MessageRouter is using our MockTransport.
|
||||||
|
// ChatViewModel init sets up MessageRouter with the passed transport.
|
||||||
|
|
||||||
|
// Wait for async processing
|
||||||
|
try? await Task.sleep(nanoseconds: 100_000_000)
|
||||||
|
|
||||||
|
// Verify message stored locally
|
||||||
|
#expect(viewModel.privateChats[peerID]?.count == 1)
|
||||||
|
#expect(viewModel.privateChats[peerID]?.first?.content == "Hello Mesh")
|
||||||
|
|
||||||
|
// Verify message sent to transport (MockTransport captures sendPrivateMessage)
|
||||||
|
// MockTransport.sendPrivateMessage is what MessageRouter calls for connected peers
|
||||||
|
// Check MockTransport implementation... it might need update or verification
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func handlePrivateMessage_storesMessage() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let peerID = PeerID(str: "SENDER_001")
|
||||||
|
|
||||||
|
let message = BitchatMessage(
|
||||||
|
id: "msg-1",
|
||||||
|
sender: "Sender",
|
||||||
|
content: "Private Content",
|
||||||
|
timestamp: Date(),
|
||||||
|
isRelay: false,
|
||||||
|
originalSender: nil,
|
||||||
|
isPrivate: true,
|
||||||
|
recipientNickname: "Me",
|
||||||
|
senderPeerID: peerID
|
||||||
|
)
|
||||||
|
|
||||||
|
// Simulate receiving a private message via the handlePrivateMessage extension method
|
||||||
|
viewModel.handlePrivateMessage(message)
|
||||||
|
|
||||||
|
// Verify stored
|
||||||
|
#expect(viewModel.privateChats[peerID]?.count == 1)
|
||||||
|
#expect(viewModel.privateChats[peerID]?.first?.content == "Private Content")
|
||||||
|
|
||||||
|
// Verify notification trigger (unread count should increase if not viewing)
|
||||||
|
#expect(viewModel.unreadPrivateMessages.contains(peerID))
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func handlePrivateMessage_deduplicates() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let peerID = PeerID(str: "SENDER_001")
|
||||||
|
|
||||||
|
let message = BitchatMessage(
|
||||||
|
id: "msg-1",
|
||||||
|
sender: "Sender",
|
||||||
|
content: "Content",
|
||||||
|
timestamp: Date(),
|
||||||
|
isRelay: false,
|
||||||
|
isPrivate: true,
|
||||||
|
senderPeerID: peerID
|
||||||
|
)
|
||||||
|
|
||||||
|
viewModel.handlePrivateMessage(message)
|
||||||
|
viewModel.handlePrivateMessage(message) // Duplicate
|
||||||
|
|
||||||
|
#expect(viewModel.privateChats[peerID]?.count == 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func handlePrivateMessage_sendsReadReceipt_whenViewing() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let peerID = PeerID(str: "SENDER_001")
|
||||||
|
|
||||||
|
// Set as currently viewing
|
||||||
|
viewModel.selectedPrivateChatPeer = peerID
|
||||||
|
|
||||||
|
let message = BitchatMessage(
|
||||||
|
id: "msg-1",
|
||||||
|
sender: "Sender",
|
||||||
|
content: "Content",
|
||||||
|
timestamp: Date(),
|
||||||
|
isRelay: false,
|
||||||
|
isPrivate: true,
|
||||||
|
senderPeerID: peerID
|
||||||
|
)
|
||||||
|
|
||||||
|
viewModel.handlePrivateMessage(message)
|
||||||
|
|
||||||
|
// Should NOT be marked unread
|
||||||
|
#expect(!viewModel.unreadPrivateMessages.contains(peerID))
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func migratePrivateChats_consolidatesHistory_onFingerprintMatch() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let oldPeerID = PeerID(str: "OLD_PEER")
|
||||||
|
let newPeerID = PeerID(str: "NEW_PEER")
|
||||||
|
let fingerprint = "fp_123"
|
||||||
|
|
||||||
|
// Setup old chat
|
||||||
|
let oldMessage = BitchatMessage(
|
||||||
|
id: "msg-old",
|
||||||
|
sender: "User",
|
||||||
|
content: "Old message",
|
||||||
|
timestamp: Date(),
|
||||||
|
isRelay: false,
|
||||||
|
isPrivate: true,
|
||||||
|
senderPeerID: oldPeerID
|
||||||
|
)
|
||||||
|
viewModel.privateChats[oldPeerID] = [oldMessage]
|
||||||
|
viewModel.peerIDToPublicKeyFingerprint[oldPeerID] = fingerprint
|
||||||
|
|
||||||
|
// Setup new peer fingerprint
|
||||||
|
viewModel.peerIDToPublicKeyFingerprint[newPeerID] = fingerprint
|
||||||
|
|
||||||
|
// Trigger migration
|
||||||
|
viewModel.migratePrivateChatsIfNeeded(for: newPeerID, senderNickname: "User")
|
||||||
|
|
||||||
|
// Verify migration
|
||||||
|
#expect(viewModel.privateChats[newPeerID]?.count == 1)
|
||||||
|
#expect(viewModel.privateChats[newPeerID]?.first?.content == "Old message")
|
||||||
|
#expect(viewModel.privateChats[oldPeerID] == nil) // Old chat removed
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func isMessageBlocked_filtersBlockedUsers() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let blockedPeerID = PeerID(str: "BLOCKED_PEER")
|
||||||
|
|
||||||
|
// Block the peer
|
||||||
|
// MockIdentityManager stores state based on fingerprint
|
||||||
|
// We need to map peerID to a fingerprint
|
||||||
|
viewModel.peerIDToPublicKeyFingerprint[blockedPeerID] = "fp_blocked"
|
||||||
|
viewModel.identityManager.setBlocked("fp_blocked", isBlocked: true)
|
||||||
|
|
||||||
|
// Also ensure UnifiedPeerService can resolve the fingerprint.
|
||||||
|
// UnifiedPeerService uses its own cache or delegates to meshService/Peer list.
|
||||||
|
// Since we are mocking, we can't easily inject into UnifiedPeerService's internal cache.
|
||||||
|
// However, ChatViewModel's isMessageBlocked uses:
|
||||||
|
// 1. isPeerBlocked(peerID) -> unifiedPeerService.isBlocked(peerID) -> getFingerprint -> identityManager.isBlocked
|
||||||
|
|
||||||
|
// We need UnifiedPeerService.getFingerprint(for: blockedPeerID) to return "fp_blocked"
|
||||||
|
// UnifiedPeerService tries: cache -> meshService -> getPeer
|
||||||
|
|
||||||
|
// Option 1: Mock the transport (meshService) to return the fingerprint
|
||||||
|
// (viewModel.transport is MockTransport, but UnifiedPeerService holds a reference to it)
|
||||||
|
// Check if MockTransport has `getFingerprint`
|
||||||
|
|
||||||
|
// If not, we might need to rely on the fallback: ChatViewModel.isMessageBlocked also checks Nostr blocks.
|
||||||
|
|
||||||
|
// Let's assume MockTransport needs `getFingerprint` implementation or update it.
|
||||||
|
// For now, let's try to verify if `MockTransport` supports `getFingerprint`.
|
||||||
|
|
||||||
|
// Actually, let's just use the Nostr block path which is simpler and also tested here.
|
||||||
|
// "Check geohash (Nostr) blocks using mapping to full pubkey"
|
||||||
|
|
||||||
|
let hexPubkey = "0000000000000000000000000000000000000000000000000000000000000001"
|
||||||
|
viewModel.nostrKeyMapping[blockedPeerID] = hexPubkey
|
||||||
|
viewModel.identityManager.setNostrBlocked(hexPubkey, isBlocked: true)
|
||||||
|
|
||||||
|
// Force isGeoChat/isGeoDM check to be true by setting prefix?
|
||||||
|
// Or ensure the logic covers it.
|
||||||
|
// The logic is:
|
||||||
|
// if peerID.isGeoChat || peerID.isGeoDM { check nostr }
|
||||||
|
// We need a peerID that looks like geo.
|
||||||
|
|
||||||
|
let geoPeerID = PeerID(nostr_: hexPubkey)
|
||||||
|
viewModel.nostrKeyMapping[geoPeerID] = hexPubkey
|
||||||
|
|
||||||
|
let geoMessage = BitchatMessage(
|
||||||
|
id: "msg-geo-blocked",
|
||||||
|
sender: "BlockedGeoUser",
|
||||||
|
content: "Spam",
|
||||||
|
timestamp: Date(),
|
||||||
|
isRelay: false,
|
||||||
|
isPrivate: true,
|
||||||
|
senderPeerID: geoPeerID
|
||||||
|
)
|
||||||
|
|
||||||
|
#expect(viewModel.isMessageBlocked(geoMessage))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Nostr Extension Tests
|
||||||
|
|
||||||
|
struct ChatViewModelNostrExtensionTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func switchLocationChannel_mesh_clearsGeo() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
|
||||||
|
// Setup some geo state
|
||||||
|
viewModel.switchLocationChannel(to: .location(GeohashChannel(level: .city, geohash: "u4pruydq")))
|
||||||
|
#expect(viewModel.currentGeohash == "u4pruydq")
|
||||||
|
|
||||||
|
// Switch to mesh
|
||||||
|
viewModel.switchLocationChannel(to: .mesh)
|
||||||
|
|
||||||
|
#expect(viewModel.activeChannel == .mesh)
|
||||||
|
#expect(viewModel.currentGeohash == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func subscribeNostrEvent_addsToTimeline_ifMatchesGeohash() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let geohash = "u4pruydq"
|
||||||
|
|
||||||
|
viewModel.switchLocationChannel(to: .location(GeohashChannel(level: .city, geohash: geohash)))
|
||||||
|
|
||||||
|
var event = NostrEvent(
|
||||||
|
pubkey: "pub1",
|
||||||
|
createdAt: Date(),
|
||||||
|
kind: .ephemeralEvent,
|
||||||
|
tags: [["g", geohash]],
|
||||||
|
content: "Hello Geo"
|
||||||
|
)
|
||||||
|
event.id = "evt1"
|
||||||
|
event.sig = "sig"
|
||||||
|
|
||||||
|
viewModel.handleNostrEvent(event)
|
||||||
|
|
||||||
|
// Allow async processing
|
||||||
|
try? await Task.sleep(nanoseconds: 100_000_000)
|
||||||
|
|
||||||
|
// Check timeline
|
||||||
|
// This depends on `handlePublicMessage` being called and updating `messages`
|
||||||
|
// Since `handlePublicMessage` delegates to `timelineStore` and updates `messages`...
|
||||||
|
// And we are in the correct channel...
|
||||||
|
|
||||||
|
// However, `handleNostrEvent` in the extension now calls `handlePublicMessage`.
|
||||||
|
// Let's verify if the message appears.
|
||||||
|
// Note: `handleNostrEvent` logic was refactored.
|
||||||
|
// The new logic in `ChatViewModel+Nostr.swift` calls `handlePublicMessage`.
|
||||||
|
|
||||||
|
// We need to ensure `deduplicationService` doesn't block it (new instance, so empty).
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,330 @@
|
|||||||
|
//
|
||||||
|
// ChatViewModelTests.swift
|
||||||
|
// bitchatTests
|
||||||
|
//
|
||||||
|
// Tests for ChatViewModel using MockTransport for isolation.
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Testing
|
||||||
|
import Foundation
|
||||||
|
@testable import bitchat
|
||||||
|
|
||||||
|
// MARK: - Test Helpers
|
||||||
|
|
||||||
|
/// Creates a ChatViewModel with mock dependencies for testing
|
||||||
|
@MainActor
|
||||||
|
private func makeTestableViewModel() -> (viewModel: ChatViewModel, transport: MockTransport) {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
let keychainHelper = MockKeychainHelper()
|
||||||
|
let idBridge = NostrIdentityBridge(keychain: keychainHelper)
|
||||||
|
let identityManager = MockIdentityManager(keychain)
|
||||||
|
let transport = MockTransport()
|
||||||
|
|
||||||
|
let viewModel = ChatViewModel(
|
||||||
|
keychain: keychain,
|
||||||
|
idBridge: idBridge,
|
||||||
|
identityManager: identityManager,
|
||||||
|
transport: transport
|
||||||
|
)
|
||||||
|
|
||||||
|
return (viewModel, transport)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Initialization Tests
|
||||||
|
|
||||||
|
struct ChatViewModelInitializationTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func initialization_setsDelegate() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
// The viewModel should set itself as the transport delegate
|
||||||
|
#expect(transport.delegate === viewModel)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func initialization_startsServices() async {
|
||||||
|
let (_, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
// Services should be started during init
|
||||||
|
#expect(transport.startServicesCallCount == 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func initialization_hasEmptyMessageList() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
|
||||||
|
// Initial messages may include system messages, but should be limited
|
||||||
|
#expect(viewModel.messages.count < 10)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func initialization_setsNickname() async {
|
||||||
|
let (_, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
// Nickname should be set during init
|
||||||
|
#expect(!transport.myNickname.isEmpty)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Message Sending Tests
|
||||||
|
|
||||||
|
struct ChatViewModelSendingTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func sendMessage_delegatesToTransport() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
viewModel.sendMessage("Hello World")
|
||||||
|
|
||||||
|
#expect(transport.sentMessages.count == 1)
|
||||||
|
#expect(transport.sentMessages.first?.content == "Hello World")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func sendMessage_emptyContent_ignored() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
viewModel.sendMessage("")
|
||||||
|
viewModel.sendMessage(" ")
|
||||||
|
viewModel.sendMessage("\n\t")
|
||||||
|
|
||||||
|
#expect(transport.sentMessages.isEmpty)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func sendMessage_withMentions_sendsContent() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
viewModel.sendMessage("Hello @alice")
|
||||||
|
|
||||||
|
#expect(transport.sentMessages.count == 1)
|
||||||
|
#expect(transport.sentMessages.first?.content == "Hello @alice")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func sendMessage_command_notSentToTransport() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
viewModel.sendMessage("/help")
|
||||||
|
|
||||||
|
// Commands are processed locally, not sent to transport
|
||||||
|
#expect(transport.sentMessages.isEmpty)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Message Receiving Tests
|
||||||
|
|
||||||
|
struct ChatViewModelReceivingTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func didReceiveMessage_callsDelegate() async {
|
||||||
|
let (_, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
let message = BitchatMessage(
|
||||||
|
id: "msg-001",
|
||||||
|
sender: "Alice",
|
||||||
|
content: "Hello from Alice",
|
||||||
|
timestamp: Date(),
|
||||||
|
isRelay: false,
|
||||||
|
originalSender: nil,
|
||||||
|
isPrivate: false,
|
||||||
|
recipientNickname: nil,
|
||||||
|
senderPeerID: PeerID(str: "PEER001"),
|
||||||
|
mentions: nil
|
||||||
|
)
|
||||||
|
|
||||||
|
transport.simulateIncomingMessage(message)
|
||||||
|
|
||||||
|
// Give time for Task and pipeline processing
|
||||||
|
try? await Task.sleep(nanoseconds: 200_000_000)
|
||||||
|
|
||||||
|
// Message may or may not appear due to rate limiting/pipeline batching
|
||||||
|
// The important thing is no crash and delegate was called
|
||||||
|
#expect(transport.delegate != nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func didReceivePublicMessage_addsToTimeline() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
transport.simulateIncomingPublicMessage(
|
||||||
|
from: PeerID(str: "PEER002"),
|
||||||
|
nickname: "Bob",
|
||||||
|
content: "Public hello from Bob",
|
||||||
|
timestamp: Date(),
|
||||||
|
messageID: "pub-001"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Give time for async Task and pipeline processing
|
||||||
|
try? await Task.sleep(nanoseconds: 500_000_000)
|
||||||
|
|
||||||
|
#expect(viewModel.messages.contains { $0.content == "Public hello from Bob" })
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Peer Connection Tests
|
||||||
|
|
||||||
|
struct ChatViewModelPeerTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func didConnectToPeer_notifiesDelegate() async {
|
||||||
|
let (_, transport) = makeTestableViewModel()
|
||||||
|
let peerID = PeerID(str: "NEWPEER")
|
||||||
|
|
||||||
|
transport.simulateConnect(peerID, nickname: "NewUser")
|
||||||
|
|
||||||
|
#expect(transport.connectedPeers.contains(peerID))
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func didDisconnectFromPeer_notifiesDelegate() async {
|
||||||
|
let (_, transport) = makeTestableViewModel()
|
||||||
|
let peerID = PeerID(str: "OLDPEER")
|
||||||
|
|
||||||
|
transport.simulateConnect(peerID, nickname: "OldUser")
|
||||||
|
transport.simulateDisconnect(peerID)
|
||||||
|
|
||||||
|
#expect(!transport.connectedPeers.contains(peerID))
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func isPeerConnected_delegatesToTransport() async {
|
||||||
|
let (_, transport) = makeTestableViewModel()
|
||||||
|
let peerID = PeerID(str: "TESTPEER")
|
||||||
|
|
||||||
|
// Not connected initially
|
||||||
|
#expect(!transport.isPeerConnected(peerID))
|
||||||
|
|
||||||
|
transport.connectedPeers.insert(peerID)
|
||||||
|
|
||||||
|
#expect(transport.isPeerConnected(peerID))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Deduplication Integration Tests
|
||||||
|
//
|
||||||
|
// Note: Detailed deduplication logic is tested in MessageDeduplicationServiceTests.
|
||||||
|
// These tests verify that ChatViewModel has a deduplication service configured.
|
||||||
|
|
||||||
|
struct ChatViewModelDeduplicationTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func deduplicationService_isConfigured() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
|
||||||
|
// Verify the deduplication service is available and functional
|
||||||
|
// by checking that we can record and query content
|
||||||
|
let testContent = "Test dedup content \(UUID().uuidString)"
|
||||||
|
let testDate = Date()
|
||||||
|
|
||||||
|
viewModel.deduplicationService.recordContent(testContent, timestamp: testDate)
|
||||||
|
|
||||||
|
let retrieved = viewModel.deduplicationService.contentTimestamp(for: testContent)
|
||||||
|
#expect(retrieved == testDate)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func deduplicationService_normalizedKey_consistent() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
|
||||||
|
let content = "Hello World"
|
||||||
|
let key1 = viewModel.deduplicationService.normalizedContentKey(content)
|
||||||
|
let key2 = viewModel.deduplicationService.normalizedContentKey(content)
|
||||||
|
|
||||||
|
#expect(key1 == key2)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Private Chat Tests
|
||||||
|
|
||||||
|
struct ChatViewModelPrivateChatTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func sendPrivateMessage_delegatesToTransport() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
let recipientID = PeerID(str: "RECIPIENT")
|
||||||
|
|
||||||
|
// Set up connected peer for routing
|
||||||
|
transport.connectedPeers.insert(recipientID)
|
||||||
|
transport.peerNicknames[recipientID] = "Recipient"
|
||||||
|
|
||||||
|
viewModel.sendPrivateMessage("Secret message", to: recipientID)
|
||||||
|
|
||||||
|
// The message routing depends on connection state and other factors
|
||||||
|
// At minimum, it should not crash
|
||||||
|
#expect(true) // If we get here without crash, the test passes
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Bluetooth State Tests
|
||||||
|
|
||||||
|
struct ChatViewModelBluetoothTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func didUpdateBluetoothState_poweredOn_noAlert() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
transport.simulateBluetoothStateChange(.poweredOn)
|
||||||
|
|
||||||
|
// Give time for async processing
|
||||||
|
try? await Task.sleep(nanoseconds: 100_000_000)
|
||||||
|
|
||||||
|
#expect(!viewModel.showBluetoothAlert)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func didUpdateBluetoothState_poweredOff_showsAlert() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
transport.simulateBluetoothStateChange(.poweredOff)
|
||||||
|
|
||||||
|
// Give time for async processing
|
||||||
|
try? await Task.sleep(nanoseconds: 100_000_000)
|
||||||
|
|
||||||
|
#expect(viewModel.showBluetoothAlert)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func didUpdateBluetoothState_unauthorized_showsAlert() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
transport.simulateBluetoothStateChange(.unauthorized)
|
||||||
|
|
||||||
|
// Give time for async processing
|
||||||
|
try? await Task.sleep(nanoseconds: 100_000_000)
|
||||||
|
|
||||||
|
#expect(viewModel.showBluetoothAlert)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Panic Clear Tests
|
||||||
|
|
||||||
|
struct ChatViewModelPanicTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func panicClearAllData_delegatesToTransport() async {
|
||||||
|
let (viewModel, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
// Set up some state
|
||||||
|
transport.connectedPeers.insert(PeerID(str: "PEER1"))
|
||||||
|
|
||||||
|
viewModel.panicClearAllData()
|
||||||
|
|
||||||
|
// After panic, emergency disconnect should be called
|
||||||
|
#expect(transport.emergencyDisconnectCallCount == 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Service Lifecycle Tests
|
||||||
|
|
||||||
|
struct ChatViewModelLifecycleTests {
|
||||||
|
|
||||||
|
@Test @MainActor
|
||||||
|
func startServices_calledOnInit() async {
|
||||||
|
let (_, transport) = makeTestableViewModel()
|
||||||
|
|
||||||
|
#expect(transport.startServicesCallCount == 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -186,11 +186,11 @@ struct PrivateChatE2ETests {
|
|||||||
// Bob relays private messages for Charlie
|
// Bob relays private messages for Charlie
|
||||||
bob.packetDeliveryHandler = { packet in
|
bob.packetDeliveryHandler = { packet in
|
||||||
if let recipientID = packet.recipientID,
|
if let recipientID = packet.recipientID,
|
||||||
String(data: recipientID, encoding: .utf8) == charlie.peerID {
|
PeerID(data: recipientID) == charlie.peerID {
|
||||||
// Relay to Charlie
|
// Relay to Charlie
|
||||||
var relayPacket = packet
|
var relayPacket = packet
|
||||||
relayPacket.ttl = packet.ttl - 1
|
relayPacket.ttl = packet.ttl - 1
|
||||||
self.charlie.simulateIncomingPacket(relayPacket)
|
charlie.simulateIncomingPacket(relayPacket)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -388,7 +388,7 @@ struct PublicChatE2ETests {
|
|||||||
|
|
||||||
if let message = BitchatMessage(packet.payload) {
|
if let message = BitchatMessage(packet.payload) {
|
||||||
// Don't relay own messages
|
// Don't relay own messages
|
||||||
guard message.senderPeerID?.id != node.peerID else { return }
|
guard message.senderPeerID != node.peerID else { return }
|
||||||
|
|
||||||
// Create relay message
|
// Create relay message
|
||||||
let relayMessage = BitchatMessage(
|
let relayMessage = BitchatMessage(
|
||||||
|
|||||||
@@ -43,17 +43,16 @@ struct FragmentationTests {
|
|||||||
// Shuffle fragments to simulate out-of-order arrival
|
// Shuffle fragments to simulate out-of-order arrival
|
||||||
let shuffled = fragments.shuffled()
|
let shuffled = fragments.shuffled()
|
||||||
|
|
||||||
// Inject fragments spaced out to avoid concurrent mutation inside BLEService
|
// Send fragments sequentially with small delays (no fire-and-forget Tasks)
|
||||||
for (i, fragment) in shuffled.enumerated() {
|
for (i, fragment) in shuffled.enumerated() {
|
||||||
let delay = 5 * Double(i) * 0.001
|
if i > 0 {
|
||||||
Task {
|
try await Task.sleep(for: .milliseconds(5))
|
||||||
try await sleep(delay)
|
}
|
||||||
ble._test_handlePacket(fragment, fromPeerID: remoteShortID)
|
ble._test_handlePacket(fragment, fromPeerID: remoteShortID)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// Allow async processing
|
// Wait for delegate callback with proper timeout
|
||||||
try await sleep(0.5)
|
try await capture.waitForPublicMessages(count: 1, timeout: .seconds(2))
|
||||||
|
|
||||||
#expect(capture.publicMessages.count == 1)
|
#expect(capture.publicMessages.count == 1)
|
||||||
#expect(capture.publicMessages.first?.content.count == 3_000)
|
#expect(capture.publicMessages.first?.content.count == 3_000)
|
||||||
@@ -78,21 +77,77 @@ struct FragmentationTests {
|
|||||||
frags.insert(dup, at: 1)
|
frags.insert(dup, at: 1)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Send fragments sequentially with small delays (no fire-and-forget Tasks)
|
||||||
for (i, fragment) in frags.enumerated() {
|
for (i, fragment) in frags.enumerated() {
|
||||||
let delay = 5 * Double(i) * 0.001
|
if i > 0 {
|
||||||
Task {
|
try await Task.sleep(for: .milliseconds(5))
|
||||||
try await sleep(delay)
|
}
|
||||||
ble._test_handlePacket(fragment, fromPeerID: remoteShortID)
|
ble._test_handlePacket(fragment, fromPeerID: remoteShortID)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// Allow async processing
|
// Wait for delegate callback with proper timeout
|
||||||
try await sleep(0.5)
|
try await capture.waitForPublicMessages(count: 1, timeout: .seconds(2))
|
||||||
|
|
||||||
#expect(capture.publicMessages.count == 1)
|
#expect(capture.publicMessages.count == 1)
|
||||||
#expect(capture.publicMessages.first?.content.count == 2048)
|
#expect(capture.publicMessages.first?.content.count == 2048)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test("Max-sized file transfer survives reassembly")
|
||||||
|
func maxSizedFileTransferSurvivesReassembly() async throws {
|
||||||
|
let ble = BLEService(
|
||||||
|
keychain: mockKeychain,
|
||||||
|
idBridge: idBridge,
|
||||||
|
identityManager: mockIdentityManager
|
||||||
|
)
|
||||||
|
let capture = CaptureDelegate()
|
||||||
|
ble.delegate = capture
|
||||||
|
|
||||||
|
let remoteID = PeerID(str: "CAFEBABECAFEBABE")
|
||||||
|
let fileContent = Data(repeating: 0x42, count: FileTransferLimits.maxPayloadBytes)
|
||||||
|
let filePacket = BitchatFilePacket(
|
||||||
|
fileName: "limit.bin",
|
||||||
|
fileSize: UInt64(fileContent.count),
|
||||||
|
mimeType: "application/octet-stream",
|
||||||
|
content: fileContent
|
||||||
|
)
|
||||||
|
let encoded = try #require(filePacket.encode(), "File packet encoding failed")
|
||||||
|
|
||||||
|
let packet = BitchatPacket(
|
||||||
|
type: MessageType.fileTransfer.rawValue,
|
||||||
|
senderID: Data(hexString: remoteID.id) ?? Data(),
|
||||||
|
recipientID: nil,
|
||||||
|
timestamp: UInt64(Date().timeIntervalSince1970 * 1000),
|
||||||
|
payload: encoded,
|
||||||
|
signature: nil,
|
||||||
|
ttl: 7,
|
||||||
|
version: 2
|
||||||
|
)
|
||||||
|
|
||||||
|
let fragments = fragmentPacket(packet, fragmentSize: 4096, pad: false)
|
||||||
|
#expect(!fragments.isEmpty)
|
||||||
|
|
||||||
|
for (i, fragment) in fragments.enumerated() {
|
||||||
|
let delay = 5 * Double(i) * 0.001
|
||||||
|
Task {
|
||||||
|
try await sleep(delay)
|
||||||
|
ble._test_handlePacket(fragment, fromPeerID: remoteID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
try await sleep(1.0)
|
||||||
|
|
||||||
|
let message = try #require(capture.receivedMessages.first, "Expected file transfer message")
|
||||||
|
#expect(message.content.hasPrefix("[file]"))
|
||||||
|
|
||||||
|
if let fileName = message.content.split(separator: " ").last {
|
||||||
|
let base = try FileManager.default.url(for: .applicationSupportDirectory, in: .userDomainMask, appropriateFor: nil, create: true)
|
||||||
|
let filesRoot = base.appendingPathComponent("files", isDirectory: true)
|
||||||
|
let incoming = filesRoot.appendingPathComponent("files/incoming", isDirectory: true)
|
||||||
|
let url = incoming.appendingPathComponent(String(fileName))
|
||||||
|
try? FileManager.default.removeItem(at: url)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@Test("Invalid fragment header is ignored")
|
@Test("Invalid fragment header is ignored")
|
||||||
func invalidFragmentHeaderIsIgnored() async throws {
|
func invalidFragmentHeaderIsIgnored() async throws {
|
||||||
let ble = BLEService(
|
let ble = BLEService(
|
||||||
@@ -140,9 +195,128 @@ struct FragmentationTests {
|
|||||||
}
|
}
|
||||||
|
|
||||||
extension FragmentationTests {
|
extension FragmentationTests {
|
||||||
private final class CaptureDelegate: BitchatDelegate {
|
/// Thread-safe delegate that supports awaiting message delivery
|
||||||
var publicMessages: [(peerID: PeerID, nickname: String, content: String)] = []
|
private final class CaptureDelegate: BitchatDelegate, @unchecked Sendable {
|
||||||
func didReceiveMessage(_ message: BitchatMessage) {}
|
private let lock = NSLock()
|
||||||
|
private var _publicMessages: [(peerID: PeerID, nickname: String, content: String)] = []
|
||||||
|
private var _receivedMessages: [BitchatMessage] = []
|
||||||
|
private var publicMessageContinuation: CheckedContinuation<Void, Never>?
|
||||||
|
private var receivedMessageContinuation: CheckedContinuation<Void, Never>?
|
||||||
|
private var expectedPublicMessageCount: Int = 0
|
||||||
|
private var expectedReceivedMessageCount: Int = 0
|
||||||
|
|
||||||
|
var publicMessages: [(peerID: PeerID, nickname: String, content: String)] {
|
||||||
|
lock.lock()
|
||||||
|
defer { lock.unlock() }
|
||||||
|
return _publicMessages
|
||||||
|
}
|
||||||
|
|
||||||
|
var receivedMessages: [BitchatMessage] {
|
||||||
|
lock.lock()
|
||||||
|
defer { lock.unlock() }
|
||||||
|
return _receivedMessages
|
||||||
|
}
|
||||||
|
|
||||||
|
func didReceiveMessage(_ message: BitchatMessage) {
|
||||||
|
lock.lock()
|
||||||
|
_receivedMessages.append(message)
|
||||||
|
let count = _receivedMessages.count
|
||||||
|
let expected = expectedReceivedMessageCount
|
||||||
|
let continuation = receivedMessageContinuation
|
||||||
|
lock.unlock()
|
||||||
|
|
||||||
|
if count >= expected, let cont = continuation {
|
||||||
|
lock.lock()
|
||||||
|
receivedMessageContinuation = nil
|
||||||
|
lock.unlock()
|
||||||
|
cont.resume()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func didReceivePublicMessage(from peerID: PeerID, nickname: String, content: String, timestamp: Date, messageID: String?) {
|
||||||
|
lock.lock()
|
||||||
|
_publicMessages.append((peerID, nickname, content))
|
||||||
|
let count = _publicMessages.count
|
||||||
|
let expected = expectedPublicMessageCount
|
||||||
|
let continuation = publicMessageContinuation
|
||||||
|
lock.unlock()
|
||||||
|
|
||||||
|
if count >= expected, let cont = continuation {
|
||||||
|
lock.lock()
|
||||||
|
publicMessageContinuation = nil
|
||||||
|
lock.unlock()
|
||||||
|
cont.resume()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Waits for the specified number of public messages to be received
|
||||||
|
func waitForPublicMessages(count: Int, timeout: Duration = .seconds(2)) async throws {
|
||||||
|
lock.lock()
|
||||||
|
if _publicMessages.count >= count {
|
||||||
|
lock.unlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
expectedPublicMessageCount = count
|
||||||
|
lock.unlock()
|
||||||
|
|
||||||
|
try await withThrowingTaskGroup(of: Void.self) { group in
|
||||||
|
group.addTask {
|
||||||
|
await withCheckedContinuation { continuation in
|
||||||
|
self.lock.lock()
|
||||||
|
// Recheck count after acquiring lock to avoid race condition
|
||||||
|
// where message arrives between initial check and continuation install
|
||||||
|
if self._publicMessages.count >= count {
|
||||||
|
self.lock.unlock()
|
||||||
|
continuation.resume()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
self.publicMessageContinuation = continuation
|
||||||
|
self.lock.unlock()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
group.addTask {
|
||||||
|
try await Task.sleep(for: timeout)
|
||||||
|
throw CancellationError()
|
||||||
|
}
|
||||||
|
try await group.next()
|
||||||
|
group.cancelAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Waits for the specified number of received messages
|
||||||
|
func waitForReceivedMessages(count: Int, timeout: Duration = .seconds(2)) async throws {
|
||||||
|
lock.lock()
|
||||||
|
if _receivedMessages.count >= count {
|
||||||
|
lock.unlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
expectedReceivedMessageCount = count
|
||||||
|
lock.unlock()
|
||||||
|
|
||||||
|
try await withThrowingTaskGroup(of: Void.self) { group in
|
||||||
|
group.addTask {
|
||||||
|
await withCheckedContinuation { continuation in
|
||||||
|
self.lock.lock()
|
||||||
|
// Recheck count after acquiring lock to avoid race condition
|
||||||
|
// where message arrives between initial check and continuation install
|
||||||
|
if self._receivedMessages.count >= count {
|
||||||
|
self.lock.unlock()
|
||||||
|
continuation.resume()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
self.receivedMessageContinuation = continuation
|
||||||
|
self.lock.unlock()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
group.addTask {
|
||||||
|
try await Task.sleep(for: timeout)
|
||||||
|
throw CancellationError()
|
||||||
|
}
|
||||||
|
try await group.next()
|
||||||
|
group.cancelAll()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func didConnectToPeer(_ peerID: PeerID) {}
|
func didConnectToPeer(_ peerID: PeerID) {}
|
||||||
func didDisconnectFromPeer(_ peerID: PeerID) {}
|
func didDisconnectFromPeer(_ peerID: PeerID) {}
|
||||||
func didUpdatePeerList(_ peers: [PeerID]) {}
|
func didUpdatePeerList(_ peers: [PeerID]) {}
|
||||||
@@ -150,9 +324,6 @@ extension FragmentationTests {
|
|||||||
func didUpdateMessageDeliveryStatus(_ messageID: String, status: DeliveryStatus) {}
|
func didUpdateMessageDeliveryStatus(_ messageID: String, status: DeliveryStatus) {}
|
||||||
func didReceiveNoisePayload(from peerID: PeerID, type: NoisePayloadType, payload: Data, timestamp: Date) {}
|
func didReceiveNoisePayload(from peerID: PeerID, type: NoisePayloadType, payload: Data, timestamp: Date) {}
|
||||||
func didUpdateBluetoothState(_ state: CBManagerState) {}
|
func didUpdateBluetoothState(_ state: CBManagerState) {}
|
||||||
func didReceivePublicMessage(from peerID: PeerID, nickname: String, content: String, timestamp: Date) {
|
|
||||||
publicMessages.append((peerID, nickname, content))
|
|
||||||
}
|
|
||||||
func didReceiveRegionalPublicMessage(from peerID: PeerID, nickname: String, content: String, timestamp: Date) {}
|
func didReceiveRegionalPublicMessage(from peerID: PeerID, nickname: String, content: String, timestamp: Date) {}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -173,8 +344,8 @@ extension FragmentationTests {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Helper: fragment a packet using the same header format BLEService expects
|
// Helper: fragment a packet using the same header format BLEService expects
|
||||||
private func fragmentPacket(_ packet: BitchatPacket, fragmentSize: Int, fragmentID: Data? = nil) -> [BitchatPacket] {
|
private func fragmentPacket(_ packet: BitchatPacket, fragmentSize: Int, fragmentID: Data? = nil, pad: Bool = true) -> [BitchatPacket] {
|
||||||
let fullData = packet.toBinaryData() ?? Data()
|
guard let fullData = packet.toBinaryData(padding: pad) else { return [] }
|
||||||
let fid = fragmentID ?? Data((0..<8).map { _ in UInt8.random(in: 0...255) })
|
let fid = fragmentID ?? Data((0..<8).map { _ in UInt8.random(in: 0...255) })
|
||||||
let chunks: [Data] = stride(from: 0, to: fullData.count, by: fragmentSize).map { off in
|
let chunks: [Data] = stride(from: 0, to: fullData.count, by: fragmentSize).map { off in
|
||||||
Data(fullData[off..<min(off + fragmentSize, fullData.count)])
|
Data(fullData[off..<min(off + fragmentSize, fullData.count)])
|
||||||
|
|||||||
@@ -0,0 +1,293 @@
|
|||||||
|
//
|
||||||
|
// GeohashParticipantTrackerTests.swift
|
||||||
|
// bitchatTests
|
||||||
|
//
|
||||||
|
// Tests for GeohashParticipantTracker.
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Testing
|
||||||
|
import Foundation
|
||||||
|
@testable import bitchat
|
||||||
|
|
||||||
|
/// Mock context for testing
|
||||||
|
@MainActor
|
||||||
|
final class MockParticipantContext: GeohashParticipantContext {
|
||||||
|
var blockedPubkeys: Set<String> = []
|
||||||
|
var nicknameMap: [String: String] = [:]
|
||||||
|
var selfPubkey: String?
|
||||||
|
|
||||||
|
func displayNameForPubkey(_ pubkeyHex: String) -> String {
|
||||||
|
let suffix = String(pubkeyHex.suffix(4))
|
||||||
|
if let self = selfPubkey, pubkeyHex.lowercased() == self.lowercased() {
|
||||||
|
return "me#\(suffix)"
|
||||||
|
}
|
||||||
|
if let nick = nicknameMap[pubkeyHex.lowercased()] {
|
||||||
|
return "\(nick)#\(suffix)"
|
||||||
|
}
|
||||||
|
return "anon#\(suffix)"
|
||||||
|
}
|
||||||
|
|
||||||
|
func isBlocked(_ pubkeyHexLowercased: String) -> Bool {
|
||||||
|
blockedPubkeys.contains(pubkeyHexLowercased.lowercased())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
struct GeohashParticipantTrackerTests {
|
||||||
|
|
||||||
|
// MARK: - Basic Recording Tests
|
||||||
|
|
||||||
|
@Test func recordParticipant_addsToActiveGeohash() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "deadbeef1234")
|
||||||
|
|
||||||
|
#expect(tracker.participantCount(for: "abc123") == 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func recordParticipant_noActiveGeohash_noOp() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
// No active geohash set
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "deadbeef1234")
|
||||||
|
|
||||||
|
// Should not throw or crash
|
||||||
|
#expect(tracker.participantCount(for: "abc123") == 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func recordParticipant_specificGeohash() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1", geohash: "geo1")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey2", geohash: "geo2")
|
||||||
|
|
||||||
|
#expect(tracker.participantCount(for: "geo1") == 1)
|
||||||
|
#expect(tracker.participantCount(for: "geo2") == 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func recordParticipant_updatesLastSeen() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1")
|
||||||
|
// Small delay and record again
|
||||||
|
try? await Task.sleep(nanoseconds: 10_000_000) // 10ms
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1")
|
||||||
|
|
||||||
|
// Should still count as 1 participant (updated, not duplicated)
|
||||||
|
#expect(tracker.participantCount(for: "abc123") == 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func recordParticipant_lowercasesPubkey() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "DEADBEEF")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "deadbeef")
|
||||||
|
|
||||||
|
// Should be treated as same participant
|
||||||
|
#expect(tracker.participantCount(for: "abc123") == 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Visible People Tests
|
||||||
|
|
||||||
|
@Test func getVisiblePeople_returnsActiveGeohashParticipants() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey2")
|
||||||
|
|
||||||
|
let people = tracker.getVisiblePeople()
|
||||||
|
#expect(people.count == 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func getVisiblePeople_excludesBlockedParticipants() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
context.blockedPubkeys = ["pubkey2"]
|
||||||
|
tracker.configure(context: context)
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey2")
|
||||||
|
|
||||||
|
let people = tracker.getVisiblePeople()
|
||||||
|
#expect(people.count == 1)
|
||||||
|
#expect(people.first?.id == "pubkey1")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func getVisiblePeople_usesDisplayNameFromContext() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
context.nicknameMap = ["pubkey1234": "alice"]
|
||||||
|
tracker.configure(context: context)
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1234")
|
||||||
|
|
||||||
|
let people = tracker.getVisiblePeople()
|
||||||
|
#expect(people.count == 1)
|
||||||
|
#expect(people.first?.displayName == "alice#1234")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func getVisiblePeople_sortedByLastSeen() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "older")
|
||||||
|
try? await Task.sleep(nanoseconds: 10_000_000) // 10ms
|
||||||
|
tracker.recordParticipant(pubkeyHex: "newer")
|
||||||
|
|
||||||
|
let people = tracker.getVisiblePeople()
|
||||||
|
#expect(people.count == 2)
|
||||||
|
#expect(people.first?.id == "newer")
|
||||||
|
#expect(people.last?.id == "older")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func getVisiblePeople_emptyWhenNoActiveGeohash() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1", geohash: "abc123")
|
||||||
|
|
||||||
|
let people = tracker.getVisiblePeople()
|
||||||
|
#expect(people.isEmpty)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Activity Cutoff Tests
|
||||||
|
|
||||||
|
@Test func participantCount_excludesExpiredEntries() async {
|
||||||
|
// Use a very short cutoff for testing
|
||||||
|
let tracker = GeohashParticipantTracker(activityCutoff: -0.05) // 50ms cutoff
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1")
|
||||||
|
|
||||||
|
// Should be counted immediately
|
||||||
|
#expect(tracker.participantCount(for: "abc123") == 1)
|
||||||
|
|
||||||
|
// Wait for expiry
|
||||||
|
try? await Task.sleep(nanoseconds: 100_000_000) // 100ms
|
||||||
|
|
||||||
|
// Should be expired now
|
||||||
|
#expect(tracker.participantCount(for: "abc123") == 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Remove Participant Tests
|
||||||
|
|
||||||
|
@Test func removeParticipant_removesFromAllGeohashes() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1", geohash: "geo1")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1", geohash: "geo2")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey2", geohash: "geo1")
|
||||||
|
|
||||||
|
tracker.removeParticipant(pubkeyHex: "pubkey1")
|
||||||
|
|
||||||
|
#expect(tracker.participantCount(for: "geo1") == 1)
|
||||||
|
#expect(tracker.participantCount(for: "geo2") == 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Clear Tests
|
||||||
|
|
||||||
|
@Test func clear_removesAllData() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey2", geohash: "other")
|
||||||
|
|
||||||
|
tracker.clear()
|
||||||
|
|
||||||
|
#expect(tracker.participantCount(for: "abc123") == 0)
|
||||||
|
#expect(tracker.participantCount(for: "other") == 0)
|
||||||
|
#expect(tracker.visiblePeople.isEmpty)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func clearGeohash_removesOnlySpecificGeohash() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1", geohash: "geo1")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey2", geohash: "geo2")
|
||||||
|
|
||||||
|
tracker.clear(geohash: "geo1")
|
||||||
|
|
||||||
|
#expect(tracker.participantCount(for: "geo1") == 0)
|
||||||
|
#expect(tracker.participantCount(for: "geo2") == 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Set Active Geohash Tests
|
||||||
|
|
||||||
|
@Test func setActiveGeohash_clearsVisiblePeopleWhenNil() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1")
|
||||||
|
|
||||||
|
#expect(!tracker.visiblePeople.isEmpty)
|
||||||
|
|
||||||
|
tracker.setActiveGeohash(nil)
|
||||||
|
|
||||||
|
#expect(tracker.visiblePeople.isEmpty)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func setActiveGeohash_refreshesVisiblePeople() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = MockParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
|
||||||
|
// Pre-populate a geohash
|
||||||
|
tracker.recordParticipant(pubkeyHex: "pubkey1", geohash: "abc123")
|
||||||
|
|
||||||
|
// Set it as active
|
||||||
|
tracker.setActiveGeohash("abc123")
|
||||||
|
|
||||||
|
#expect(tracker.visiblePeople.count == 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - GeoPerson Tests
|
||||||
|
|
||||||
|
@Test func geoPerson_identifiable() async {
|
||||||
|
let person1 = GeoPerson(id: "abc", displayName: "alice", lastSeen: Date())
|
||||||
|
let person2 = GeoPerson(id: "abc", displayName: "alice", lastSeen: Date())
|
||||||
|
let person3 = GeoPerson(id: "xyz", displayName: "bob", lastSeen: Date())
|
||||||
|
|
||||||
|
#expect(person1.id == person2.id)
|
||||||
|
#expect(person1.id != person3.id)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func geoPerson_equatable() async {
|
||||||
|
let date = Date()
|
||||||
|
let person1 = GeoPerson(id: "abc", displayName: "alice", lastSeen: date)
|
||||||
|
let person2 = GeoPerson(id: "abc", displayName: "alice", lastSeen: date)
|
||||||
|
|
||||||
|
#expect(person1 == person2)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,567 @@
|
|||||||
|
//
|
||||||
|
// GeohashPresenceTests.swift
|
||||||
|
// bitchatTests
|
||||||
|
//
|
||||||
|
// Tests for the Geohash Presence (Kind 20001) feature.
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Testing
|
||||||
|
import Foundation
|
||||||
|
import Combine
|
||||||
|
@testable import bitchat
|
||||||
|
|
||||||
|
// MARK: - NostrProtocol Presence Event Tests
|
||||||
|
|
||||||
|
struct NostrProtocolPresenceTests {
|
||||||
|
|
||||||
|
@Test func createGeohashPresenceEvent_hasCorrectKind() throws {
|
||||||
|
let identity = try makeTestIdentity()
|
||||||
|
let event = try NostrProtocol.createGeohashPresenceEvent(
|
||||||
|
geohash: "u4pruydq",
|
||||||
|
senderIdentity: identity
|
||||||
|
)
|
||||||
|
|
||||||
|
#expect(event.kind == NostrProtocol.EventKind.geohashPresence.rawValue)
|
||||||
|
#expect(event.kind == 20001)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func createGeohashPresenceEvent_hasEmptyContent() throws {
|
||||||
|
let identity = try makeTestIdentity()
|
||||||
|
let event = try NostrProtocol.createGeohashPresenceEvent(
|
||||||
|
geohash: "u4pruydq",
|
||||||
|
senderIdentity: identity
|
||||||
|
)
|
||||||
|
|
||||||
|
#expect(event.content == "")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func createGeohashPresenceEvent_hasOnlyGeohashTag() throws {
|
||||||
|
let identity = try makeTestIdentity()
|
||||||
|
let event = try NostrProtocol.createGeohashPresenceEvent(
|
||||||
|
geohash: "u4pruydq",
|
||||||
|
senderIdentity: identity
|
||||||
|
)
|
||||||
|
|
||||||
|
// Should have exactly one tag: ["g", geohash]
|
||||||
|
#expect(event.tags.count == 1)
|
||||||
|
#expect(event.tags[0] == ["g", "u4pruydq"])
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func createGeohashPresenceEvent_noNicknameTag() throws {
|
||||||
|
let identity = try makeTestIdentity()
|
||||||
|
let event = try NostrProtocol.createGeohashPresenceEvent(
|
||||||
|
geohash: "u4pruydq",
|
||||||
|
senderIdentity: identity
|
||||||
|
)
|
||||||
|
|
||||||
|
// Should NOT contain nickname tag
|
||||||
|
let hasNicknameTag = event.tags.contains { $0.first == "n" }
|
||||||
|
#expect(!hasNicknameTag)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func createGeohashPresenceEvent_usesSenderPubkey() throws {
|
||||||
|
let identity = try makeTestIdentity()
|
||||||
|
let event = try NostrProtocol.createGeohashPresenceEvent(
|
||||||
|
geohash: "u4pruydq",
|
||||||
|
senderIdentity: identity
|
||||||
|
)
|
||||||
|
|
||||||
|
#expect(event.pubkey == identity.publicKeyHex)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func createGeohashPresenceEvent_isSigned() throws {
|
||||||
|
let identity = try makeTestIdentity()
|
||||||
|
let event = try NostrProtocol.createGeohashPresenceEvent(
|
||||||
|
geohash: "u4pruydq",
|
||||||
|
senderIdentity: identity
|
||||||
|
)
|
||||||
|
|
||||||
|
#expect(event.sig != nil && !event.sig!.isEmpty)
|
||||||
|
#expect(!event.id.isEmpty)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func createGeohashPresenceEvent_differentGeohashes() throws {
|
||||||
|
let identity = try makeTestIdentity()
|
||||||
|
|
||||||
|
let event1 = try NostrProtocol.createGeohashPresenceEvent(geohash: "87", senderIdentity: identity)
|
||||||
|
let event2 = try NostrProtocol.createGeohashPresenceEvent(geohash: "87yw", senderIdentity: identity)
|
||||||
|
let event3 = try NostrProtocol.createGeohashPresenceEvent(geohash: "87yw7", senderIdentity: identity)
|
||||||
|
|
||||||
|
#expect(event1.tags[0][1] == "87")
|
||||||
|
#expect(event2.tags[0][1] == "87yw")
|
||||||
|
#expect(event3.tags[0][1] == "87yw7")
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Helper
|
||||||
|
|
||||||
|
private func makeTestIdentity() throws -> NostrIdentity {
|
||||||
|
// Generate a fresh test identity
|
||||||
|
return try NostrIdentity.generate()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - NostrFilter Presence Tests
|
||||||
|
|
||||||
|
struct NostrFilterPresenceTests {
|
||||||
|
|
||||||
|
@Test func geohashEphemeral_includesBothKinds() {
|
||||||
|
let filter = NostrFilter.geohashEphemeral("u4pruydq")
|
||||||
|
|
||||||
|
#expect(filter.kinds?.contains(20000) == true)
|
||||||
|
#expect(filter.kinds?.contains(20001) == true)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func geohashEphemeral_hasLimit1000() {
|
||||||
|
let filter = NostrFilter.geohashEphemeral("u4pruydq")
|
||||||
|
|
||||||
|
#expect(filter.limit == 1000)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func geohashEphemeral_respectsSinceParameter() {
|
||||||
|
let since = Date(timeIntervalSince1970: 1700000000)
|
||||||
|
let filter = NostrFilter.geohashEphemeral("u4pruydq", since: since)
|
||||||
|
|
||||||
|
#expect(filter.since == 1700000000)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - ChatViewModel Presence Handling Tests
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
struct ChatViewModelPresenceHandlingTests {
|
||||||
|
|
||||||
|
@Test func handleNostrEvent_presenceUpdatesParticipantTracker() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let geohash = "u4pruydq"
|
||||||
|
|
||||||
|
// Set up the channel
|
||||||
|
viewModel.switchLocationChannel(to: .location(GeohashChannel(level: .city, geohash: geohash)))
|
||||||
|
|
||||||
|
// Create a presence event (kind 20001)
|
||||||
|
var event = NostrEvent(
|
||||||
|
pubkey: "abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234",
|
||||||
|
createdAt: Date(),
|
||||||
|
kind: .geohashPresence,
|
||||||
|
tags: [["g", geohash]],
|
||||||
|
content: ""
|
||||||
|
)
|
||||||
|
event.id = "presence_evt_1"
|
||||||
|
event.sig = "sig"
|
||||||
|
|
||||||
|
// Handle the event
|
||||||
|
viewModel.handleNostrEvent(event)
|
||||||
|
|
||||||
|
// Allow async processing
|
||||||
|
try? await Task.sleep(nanoseconds: 50_000_000)
|
||||||
|
|
||||||
|
// Participant should be recorded
|
||||||
|
let count = viewModel.geohashParticipantCount(for: geohash)
|
||||||
|
#expect(count >= 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func handleNostrEvent_presenceDoesNotAddToTimeline() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let geohash = "u4pruydq"
|
||||||
|
|
||||||
|
viewModel.switchLocationChannel(to: .location(GeohashChannel(level: .city, geohash: geohash)))
|
||||||
|
|
||||||
|
let initialMessageCount = viewModel.messages.count
|
||||||
|
|
||||||
|
// Create a presence event (kind 20001)
|
||||||
|
var event = NostrEvent(
|
||||||
|
pubkey: "abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234",
|
||||||
|
createdAt: Date(),
|
||||||
|
kind: .geohashPresence,
|
||||||
|
tags: [["g", geohash]],
|
||||||
|
content: ""
|
||||||
|
)
|
||||||
|
event.id = "presence_evt_2"
|
||||||
|
event.sig = "sig"
|
||||||
|
|
||||||
|
viewModel.handleNostrEvent(event)
|
||||||
|
|
||||||
|
try? await Task.sleep(nanoseconds: 50_000_000)
|
||||||
|
|
||||||
|
// Message count should NOT increase
|
||||||
|
#expect(viewModel.messages.count == initialMessageCount)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func handleNostrEvent_chatMessageUpdatesParticipant() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let geohash = "u4pruydq"
|
||||||
|
|
||||||
|
viewModel.switchLocationChannel(to: .location(GeohashChannel(level: .city, geohash: geohash)))
|
||||||
|
|
||||||
|
// Create a chat event (kind 20000) - NOT presence
|
||||||
|
var event = NostrEvent(
|
||||||
|
pubkey: "abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234",
|
||||||
|
createdAt: Date(),
|
||||||
|
kind: .ephemeralEvent,
|
||||||
|
tags: [["g", geohash]],
|
||||||
|
content: "Hello world"
|
||||||
|
)
|
||||||
|
event.id = "chat_evt_1"
|
||||||
|
event.sig = "sig"
|
||||||
|
|
||||||
|
viewModel.handleNostrEvent(event)
|
||||||
|
|
||||||
|
try? await Task.sleep(nanoseconds: 50_000_000)
|
||||||
|
|
||||||
|
// Chat messages should also update participant count (not just presence)
|
||||||
|
let count = viewModel.geohashParticipantCount(for: geohash)
|
||||||
|
#expect(count >= 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func presenceEvent_hasDifferentKindThanChat() {
|
||||||
|
// Verify the two event kinds are distinct
|
||||||
|
let presenceKind = NostrProtocol.EventKind.geohashPresence.rawValue
|
||||||
|
let chatKind = NostrProtocol.EventKind.ephemeralEvent.rawValue
|
||||||
|
|
||||||
|
#expect(presenceKind != chatKind)
|
||||||
|
#expect(presenceKind == 20001)
|
||||||
|
#expect(chatKind == 20000)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func subscribeNostrEvent_acceptsPresenceKind() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let geohash = "u4pruydq"
|
||||||
|
|
||||||
|
viewModel.switchLocationChannel(to: .location(GeohashChannel(level: .city, geohash: geohash)))
|
||||||
|
|
||||||
|
// Create presence event
|
||||||
|
var event = NostrEvent(
|
||||||
|
pubkey: "test1234test1234test1234test1234test1234test1234test1234test1234",
|
||||||
|
createdAt: Date(),
|
||||||
|
kind: .geohashPresence,
|
||||||
|
tags: [["g", geohash]],
|
||||||
|
content: ""
|
||||||
|
)
|
||||||
|
event.id = "subscribe_presence_evt"
|
||||||
|
event.sig = "sig"
|
||||||
|
|
||||||
|
// subscribeNostrEvent should accept kind 20001
|
||||||
|
viewModel.subscribeNostrEvent(event)
|
||||||
|
|
||||||
|
try? await Task.sleep(nanoseconds: 50_000_000)
|
||||||
|
|
||||||
|
// Should record participant
|
||||||
|
let count = viewModel.geohashParticipantCount(for: geohash)
|
||||||
|
#expect(count >= 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func subscribeNostrEvent_presenceForNonActiveGeohash() async {
|
||||||
|
let (viewModel, _) = makeTestableViewModel()
|
||||||
|
let activeGeohash = "u4pruydq"
|
||||||
|
let otherGeohash = "87yw7"
|
||||||
|
|
||||||
|
viewModel.switchLocationChannel(to: .location(GeohashChannel(level: .city, geohash: activeGeohash)))
|
||||||
|
|
||||||
|
// Create presence event for a DIFFERENT geohash
|
||||||
|
var event = NostrEvent(
|
||||||
|
pubkey: "other1234other1234other1234other1234other1234other1234other1234",
|
||||||
|
createdAt: Date(),
|
||||||
|
kind: .geohashPresence,
|
||||||
|
tags: [["g", otherGeohash]],
|
||||||
|
content: ""
|
||||||
|
)
|
||||||
|
event.id = "other_geohash_presence"
|
||||||
|
event.sig = "sig"
|
||||||
|
|
||||||
|
// Use subscribeNostrEvent with geohash parameter
|
||||||
|
viewModel.subscribeNostrEvent(event, gh: otherGeohash)
|
||||||
|
|
||||||
|
try? await Task.sleep(nanoseconds: 50_000_000)
|
||||||
|
|
||||||
|
// Should record for the other geohash
|
||||||
|
let count = viewModel.geohashParticipantCount(for: otherGeohash)
|
||||||
|
#expect(count >= 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Test Helper
|
||||||
|
|
||||||
|
private func makeTestableViewModel() -> (viewModel: ChatViewModel, transport: MockTransport) {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
let keychainHelper = MockKeychainHelper()
|
||||||
|
let idBridge = NostrIdentityBridge(keychain: keychainHelper)
|
||||||
|
let identityManager = MockIdentityManager(keychain)
|
||||||
|
let transport = MockTransport()
|
||||||
|
|
||||||
|
let viewModel = ChatViewModel(
|
||||||
|
keychain: keychain,
|
||||||
|
idBridge: idBridge,
|
||||||
|
identityManager: identityManager,
|
||||||
|
transport: transport
|
||||||
|
)
|
||||||
|
|
||||||
|
return (viewModel, transport)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Presence Privacy Tests
|
||||||
|
|
||||||
|
struct GeohashPresencePrivacyTests {
|
||||||
|
|
||||||
|
@Test func allowedPrecisions_onlyLowPrecision() {
|
||||||
|
// The allowed precisions for presence broadcasting should be:
|
||||||
|
// Region (2), Province (4), City (5)
|
||||||
|
// NOT Neighborhood (6), Block (7), Building (8+)
|
||||||
|
|
||||||
|
let regionPrecision = GeohashChannelLevel.region.precision
|
||||||
|
let provincePrecision = GeohashChannelLevel.province.precision
|
||||||
|
let cityPrecision = GeohashChannelLevel.city.precision
|
||||||
|
let neighborhoodPrecision = GeohashChannelLevel.neighborhood.precision
|
||||||
|
let blockPrecision = GeohashChannelLevel.block.precision
|
||||||
|
let buildingPrecision = GeohashChannelLevel.building.precision
|
||||||
|
|
||||||
|
#expect(regionPrecision == 2)
|
||||||
|
#expect(provincePrecision == 4)
|
||||||
|
#expect(cityPrecision == 5)
|
||||||
|
#expect(neighborhoodPrecision == 6)
|
||||||
|
#expect(blockPrecision == 7)
|
||||||
|
#expect(buildingPrecision == 8)
|
||||||
|
|
||||||
|
// High precision channels should NOT receive presence broadcasts
|
||||||
|
#expect(neighborhoodPrecision > 5)
|
||||||
|
#expect(blockPrecision > 5)
|
||||||
|
#expect(buildingPrecision > 5)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func geohashLengthDeterminesPrecision() {
|
||||||
|
// Verify geohash length maps to expected precision
|
||||||
|
#expect("87".count == GeohashChannelLevel.region.precision)
|
||||||
|
#expect("87yw".count == GeohashChannelLevel.province.precision)
|
||||||
|
#expect("87yw7".count == GeohashChannelLevel.city.precision)
|
||||||
|
#expect("87yw7t".count == GeohashChannelLevel.neighborhood.precision)
|
||||||
|
#expect("87yw7tc".count == GeohashChannelLevel.block.precision)
|
||||||
|
#expect("87yw7tcx".count == GeohashChannelLevel.building.precision)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func highPrecisionGeohash_isPrivacySensitive() {
|
||||||
|
// Helper to check if a geohash is "high precision" (privacy sensitive)
|
||||||
|
func isHighPrecision(_ geohash: String) -> Bool {
|
||||||
|
geohash.count >= 6
|
||||||
|
}
|
||||||
|
|
||||||
|
// Low precision - OK to broadcast presence
|
||||||
|
#expect(!isHighPrecision("87")) // region
|
||||||
|
#expect(!isHighPrecision("87yw")) // province
|
||||||
|
#expect(!isHighPrecision("87yw7")) // city
|
||||||
|
|
||||||
|
// High precision - should NOT broadcast presence
|
||||||
|
#expect(isHighPrecision("87yw7t")) // neighborhood
|
||||||
|
#expect(isHighPrecision("87yw7tc")) // block
|
||||||
|
#expect(isHighPrecision("87yw7tcx")) // building
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Display Logic Tests
|
||||||
|
|
||||||
|
struct LocationChannelsDisplayLogicTests {
|
||||||
|
|
||||||
|
@Test func displayLogic_highPrecisionZeroCount_showsUnknown() {
|
||||||
|
// Test the logic that determines "?" vs actual count
|
||||||
|
// High precision + count 0 = "?"
|
||||||
|
|
||||||
|
let shouldShowUnknown = shouldShowUnknownCount(
|
||||||
|
level: .neighborhood,
|
||||||
|
count: 0
|
||||||
|
)
|
||||||
|
#expect(shouldShowUnknown)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func displayLogic_highPrecisionNonZeroCount_showsActual() {
|
||||||
|
// High precision + count > 0 = show actual
|
||||||
|
let shouldShowUnknown = shouldShowUnknownCount(
|
||||||
|
level: .neighborhood,
|
||||||
|
count: 5
|
||||||
|
)
|
||||||
|
#expect(!shouldShowUnknown)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func displayLogic_lowPrecisionZeroCount_showsActual() {
|
||||||
|
// Low precision + count 0 = show "0" (not "?")
|
||||||
|
let shouldShowUnknown = shouldShowUnknownCount(
|
||||||
|
level: .city,
|
||||||
|
count: 0
|
||||||
|
)
|
||||||
|
#expect(!shouldShowUnknown)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func displayLogic_lowPrecisionNonZeroCount_showsActual() {
|
||||||
|
// Low precision + count > 0 = show actual
|
||||||
|
let shouldShowUnknown = shouldShowUnknownCount(
|
||||||
|
level: .region,
|
||||||
|
count: 10
|
||||||
|
)
|
||||||
|
#expect(!shouldShowUnknown)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func displayLogic_allHighPrecisionLevels() {
|
||||||
|
// All high precision levels with 0 should show "?"
|
||||||
|
let highPrecisionLevels: [GeohashChannelLevel] = [.neighborhood, .block, .building]
|
||||||
|
|
||||||
|
for level in highPrecisionLevels {
|
||||||
|
let shouldShowUnknown = shouldShowUnknownCount(level: level, count: 0)
|
||||||
|
#expect(shouldShowUnknown, "Level \(level) with count 0 should show unknown")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func displayLogic_allLowPrecisionLevels() {
|
||||||
|
// All low precision levels with 0 should show actual count
|
||||||
|
let lowPrecisionLevels: [GeohashChannelLevel] = [.region, .province, .city]
|
||||||
|
|
||||||
|
for level in lowPrecisionLevels {
|
||||||
|
let shouldShowUnknown = shouldShowUnknownCount(level: level, count: 0)
|
||||||
|
#expect(!shouldShowUnknown, "Level \(level) with count 0 should show actual count")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func displayLogic_bookmarkHighPrecision() {
|
||||||
|
// Bookmarks use geohash length to determine precision
|
||||||
|
#expect(shouldShowUnknownForBookmark(geohash: "87yw7t", count: 0)) // len 6
|
||||||
|
#expect(shouldShowUnknownForBookmark(geohash: "87yw7tc", count: 0)) // len 7
|
||||||
|
#expect(shouldShowUnknownForBookmark(geohash: "87yw7tcx", count: 0)) // len 8
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func displayLogic_bookmarkLowPrecision() {
|
||||||
|
#expect(!shouldShowUnknownForBookmark(geohash: "87", count: 0)) // len 2
|
||||||
|
#expect(!shouldShowUnknownForBookmark(geohash: "87yw", count: 0)) // len 4
|
||||||
|
#expect(!shouldShowUnknownForBookmark(geohash: "87yw7", count: 0)) // len 5
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Helpers (mirror the logic from LocationChannelsSheet)
|
||||||
|
|
||||||
|
private func shouldShowUnknownCount(level: GeohashChannelLevel, count: Int) -> Bool {
|
||||||
|
let isHighPrecision = (level == .neighborhood || level == .block || level == .building)
|
||||||
|
return isHighPrecision && count == 0
|
||||||
|
}
|
||||||
|
|
||||||
|
private func shouldShowUnknownForBookmark(geohash: String, count: Int) -> Bool {
|
||||||
|
let isHighPrecision = (geohash.count >= 6)
|
||||||
|
return isHighPrecision && count == 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Event Kind Tests
|
||||||
|
|
||||||
|
struct NostrEventKindTests {
|
||||||
|
|
||||||
|
@Test func eventKind_geohashPresence_is20001() {
|
||||||
|
#expect(NostrProtocol.EventKind.geohashPresence.rawValue == 20001)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func eventKind_ephemeralEvent_is20000() {
|
||||||
|
#expect(NostrProtocol.EventKind.ephemeralEvent.rawValue == 20000)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func eventKind_presenceIsEphemeral() {
|
||||||
|
// Both 20000 and 20001 are in the ephemeral range (20000-29999)
|
||||||
|
let presenceKind = NostrProtocol.EventKind.geohashPresence.rawValue
|
||||||
|
let chatKind = NostrProtocol.EventKind.ephemeralEvent.rawValue
|
||||||
|
|
||||||
|
#expect(presenceKind >= 20000 && presenceKind < 30000)
|
||||||
|
#expect(chatKind >= 20000 && chatKind < 30000)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Participant Tracker Presence Integration Tests
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
struct ParticipantTrackerPresenceTests {
|
||||||
|
|
||||||
|
@Test func recordParticipant_fromPresenceEvent_countsParticipant() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = PresenceTestParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
|
||||||
|
let geohash = "87yw7"
|
||||||
|
tracker.setActiveGeohash(geohash)
|
||||||
|
|
||||||
|
// Simulate recording from a presence event
|
||||||
|
tracker.recordParticipant(pubkeyHex: "presence_user_1")
|
||||||
|
|
||||||
|
#expect(tracker.participantCount(for: geohash) == 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func recordParticipant_multiplePresenceEvents_countsUnique() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = PresenceTestParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
|
||||||
|
let geohash = "87yw7"
|
||||||
|
tracker.setActiveGeohash(geohash)
|
||||||
|
|
||||||
|
// Multiple presence events from same user = 1 participant
|
||||||
|
tracker.recordParticipant(pubkeyHex: "user_a")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "user_a")
|
||||||
|
tracker.recordParticipant(pubkeyHex: "user_a")
|
||||||
|
|
||||||
|
#expect(tracker.participantCount(for: geohash) == 1)
|
||||||
|
|
||||||
|
// Different user = 2 participants
|
||||||
|
tracker.recordParticipant(pubkeyHex: "user_b")
|
||||||
|
|
||||||
|
#expect(tracker.participantCount(for: geohash) == 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func recordParticipant_nonActiveGeohash_stillCounts() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = PresenceTestParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
|
||||||
|
// Active geohash is different from where we're recording
|
||||||
|
tracker.setActiveGeohash("active_gh")
|
||||||
|
|
||||||
|
// Record to a non-active geohash (like when sampling nearby channels)
|
||||||
|
tracker.recordParticipant(pubkeyHex: "nearby_user", geohash: "other_gh")
|
||||||
|
|
||||||
|
#expect(tracker.participantCount(for: "other_gh") == 1)
|
||||||
|
#expect(tracker.participantCount(for: "active_gh") == 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func objectWillChange_firesOnNonActiveGeohashUpdate() async {
|
||||||
|
let tracker = GeohashParticipantTracker()
|
||||||
|
let context = PresenceTestParticipantContext()
|
||||||
|
tracker.configure(context: context)
|
||||||
|
|
||||||
|
tracker.setActiveGeohash("active_gh")
|
||||||
|
|
||||||
|
var changeCount = 0
|
||||||
|
let cancellable = tracker.objectWillChange.sink { _ in
|
||||||
|
changeCount += 1
|
||||||
|
}
|
||||||
|
|
||||||
|
// Record to non-active geohash
|
||||||
|
tracker.recordParticipant(pubkeyHex: "user1", geohash: "other_gh")
|
||||||
|
|
||||||
|
// Should fire objectWillChange even for non-active geohash
|
||||||
|
#expect(changeCount >= 1)
|
||||||
|
|
||||||
|
_ = cancellable // Keep alive
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Mock for Participant Context (Presence Tests)
|
||||||
|
|
||||||
|
@MainActor
|
||||||
|
private final class PresenceTestParticipantContext: GeohashParticipantContext {
|
||||||
|
var blockedPubkeys: Set<String> = []
|
||||||
|
var nicknameMap: [String: String] = [:]
|
||||||
|
var selfPubkey: String?
|
||||||
|
|
||||||
|
func displayNameForPubkey(_ pubkeyHex: String) -> String {
|
||||||
|
let suffix = String(pubkeyHex.suffix(4))
|
||||||
|
if let s = selfPubkey, pubkeyHex.lowercased() == s.lowercased() {
|
||||||
|
return "me#\(suffix)"
|
||||||
|
}
|
||||||
|
if let nick = nicknameMap[pubkeyHex.lowercased()] {
|
||||||
|
return "\(nick)#\(suffix)"
|
||||||
|
}
|
||||||
|
return "anon#\(suffix)"
|
||||||
|
}
|
||||||
|
|
||||||
|
func isBlocked(_ pubkeyHexLowercased: String) -> Bool {
|
||||||
|
blockedPubkeys.contains(pubkeyHexLowercased.lowercased())
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -125,16 +125,138 @@ struct GossipSyncManagerTests {
|
|||||||
#expect(manager._hasAnnouncement(for: PeerID(str: peerHex)) == false)
|
#expect(manager._hasAnnouncement(for: PeerID(str: peerHex)) == false)
|
||||||
#expect(manager._messageCount(for: PeerID(str: peerHex)) == 0)
|
#expect(manager._messageCount(for: PeerID(str: peerHex)) == 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test func maintenanceEmitsTypedSyncRequests() throws {
|
||||||
|
var config = GossipSyncManager.Config()
|
||||||
|
config.seenCapacity = 10
|
||||||
|
config.fragmentCapacity = 5
|
||||||
|
config.fileTransferCapacity = 4
|
||||||
|
config.messageSyncIntervalSeconds = 1
|
||||||
|
config.fragmentSyncIntervalSeconds = 1
|
||||||
|
config.fileTransferSyncIntervalSeconds = 1
|
||||||
|
config.maintenanceIntervalSeconds = 0
|
||||||
|
|
||||||
|
let manager = GossipSyncManager(myPeerID: myPeerID, config: config)
|
||||||
|
let delegate = RecordingDelegate()
|
||||||
|
manager.delegate = delegate
|
||||||
|
|
||||||
|
let sender = try #require(Data(hexString: "1122334455667788"))
|
||||||
|
let now = UInt64(Date().timeIntervalSince1970 * 1000)
|
||||||
|
|
||||||
|
let announcePacket = BitchatPacket(
|
||||||
|
type: MessageType.announce.rawValue,
|
||||||
|
senderID: sender,
|
||||||
|
recipientID: nil,
|
||||||
|
timestamp: now,
|
||||||
|
payload: Data(),
|
||||||
|
signature: nil,
|
||||||
|
ttl: 1
|
||||||
|
)
|
||||||
|
let messagePacket = BitchatPacket(
|
||||||
|
type: MessageType.message.rawValue,
|
||||||
|
senderID: sender,
|
||||||
|
recipientID: nil,
|
||||||
|
timestamp: now,
|
||||||
|
payload: Data([0x01]),
|
||||||
|
signature: nil,
|
||||||
|
ttl: 1
|
||||||
|
)
|
||||||
|
let fragmentPacket = BitchatPacket(
|
||||||
|
type: MessageType.fragment.rawValue,
|
||||||
|
senderID: sender,
|
||||||
|
recipientID: nil,
|
||||||
|
timestamp: now,
|
||||||
|
payload: Data([0xAA]),
|
||||||
|
signature: nil,
|
||||||
|
ttl: 1
|
||||||
|
)
|
||||||
|
let filePacket = BitchatPacket(
|
||||||
|
type: MessageType.fileTransfer.rawValue,
|
||||||
|
senderID: sender,
|
||||||
|
recipientID: nil,
|
||||||
|
timestamp: now,
|
||||||
|
payload: Data([0xBB]),
|
||||||
|
signature: nil,
|
||||||
|
ttl: 1,
|
||||||
|
version: 2
|
||||||
|
)
|
||||||
|
|
||||||
|
manager.onPublicPacketSeen(announcePacket)
|
||||||
|
manager.onPublicPacketSeen(messagePacket)
|
||||||
|
manager.onPublicPacketSeen(fragmentPacket)
|
||||||
|
manager.onPublicPacketSeen(filePacket)
|
||||||
|
|
||||||
|
manager._performMaintenanceSynchronously(now: Date())
|
||||||
|
|
||||||
|
let sentPackets = delegate.packets
|
||||||
|
#expect(sentPackets.count == 3)
|
||||||
|
let decoded = sentPackets.compactMap { RequestSyncPacket.decode(from: $0.payload) }
|
||||||
|
#expect(decoded.count == 3)
|
||||||
|
#expect(decoded[0].types == .publicMessages)
|
||||||
|
#expect(decoded[1].types == .fragment)
|
||||||
|
#expect(decoded[2].types == .fileTransfer)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func handleRequestSyncHonorsTypeFilter() async throws {
|
||||||
|
var config = GossipSyncManager.Config()
|
||||||
|
config.seenCapacity = 5
|
||||||
|
config.fragmentCapacity = 5
|
||||||
|
config.fileTransferCapacity = 0
|
||||||
|
config.messageSyncIntervalSeconds = 0
|
||||||
|
config.fragmentSyncIntervalSeconds = 0
|
||||||
|
config.fileTransferSyncIntervalSeconds = 0
|
||||||
|
|
||||||
|
let manager = GossipSyncManager(myPeerID: myPeerID, config: config)
|
||||||
|
let delegate = RecordingDelegate()
|
||||||
|
manager.delegate = delegate
|
||||||
|
|
||||||
|
let sender = try #require(Data(hexString: "aabbccddeeff0011"))
|
||||||
|
let now = UInt64(Date().timeIntervalSince1970 * 1000)
|
||||||
|
|
||||||
|
let messagePacket = BitchatPacket(
|
||||||
|
type: MessageType.message.rawValue,
|
||||||
|
senderID: sender,
|
||||||
|
recipientID: nil,
|
||||||
|
timestamp: now,
|
||||||
|
payload: Data([0x10]),
|
||||||
|
signature: nil,
|
||||||
|
ttl: 1
|
||||||
|
)
|
||||||
|
|
||||||
|
let fragmentPacket = BitchatPacket(
|
||||||
|
type: MessageType.fragment.rawValue,
|
||||||
|
senderID: sender,
|
||||||
|
recipientID: nil,
|
||||||
|
timestamp: now,
|
||||||
|
payload: Data([0x20]),
|
||||||
|
signature: nil,
|
||||||
|
ttl: 1
|
||||||
|
)
|
||||||
|
|
||||||
|
manager.onPublicPacketSeen(messagePacket)
|
||||||
|
manager.onPublicPacketSeen(fragmentPacket)
|
||||||
|
|
||||||
|
let peer = PeerID(str: "FFFFFFFFFFFFFFFF")
|
||||||
|
let request = RequestSyncPacket(p: 4, m: 1, data: Data(), types: .fragment)
|
||||||
|
manager.handleRequestSync(from: peer, request: request)
|
||||||
|
|
||||||
|
try await sleep(0.01)
|
||||||
|
let sentPackets = delegate.packets
|
||||||
|
#expect(sentPackets.count == 1)
|
||||||
|
#expect(sentPackets[0].type == MessageType.fragment.rawValue)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private final class RecordingDelegate: GossipSyncManager.Delegate {
|
private final class RecordingDelegate: GossipSyncManager.Delegate {
|
||||||
var onSend: (() -> Void)?
|
var onSend: (() -> Void)?
|
||||||
private(set) var lastPacket: BitchatPacket?
|
private(set) var lastPacket: BitchatPacket?
|
||||||
|
private(set) var packets: [BitchatPacket] = []
|
||||||
private let lock = NSLock()
|
private let lock = NSLock()
|
||||||
|
|
||||||
func sendPacket(_ packet: BitchatPacket) {
|
func sendPacket(_ packet: BitchatPacket) {
|
||||||
lock.lock()
|
lock.lock()
|
||||||
lastPacket = packet
|
lastPacket = packet
|
||||||
|
packets.append(packet)
|
||||||
lock.unlock()
|
lock.unlock()
|
||||||
onSend?()
|
onSend?()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,210 @@
|
|||||||
|
//
|
||||||
|
// InputValidatorTests.swift
|
||||||
|
// bitchatTests
|
||||||
|
//
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
// For more information, see <https://unlicense.org>
|
||||||
|
//
|
||||||
|
|
||||||
|
import Testing
|
||||||
|
import Foundation
|
||||||
|
@testable import bitchat
|
||||||
|
|
||||||
|
struct InputValidatorTests {
|
||||||
|
|
||||||
|
// MARK: - Basic Validation Tests
|
||||||
|
|
||||||
|
@Test func validStringPassesValidation() throws {
|
||||||
|
let result = InputValidator.validateUserString("Hello World", maxLength: 100)
|
||||||
|
#expect(result == "Hello World")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func emptyStringReturnsNil() throws {
|
||||||
|
let result = InputValidator.validateUserString("", maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func whitespaceOnlyStringReturnsNil() throws {
|
||||||
|
let result = InputValidator.validateUserString(" \n\t ", maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func stringExceedingMaxLengthReturnsNil() throws {
|
||||||
|
let longString = String(repeating: "a", count: 101)
|
||||||
|
let result = InputValidator.validateUserString(longString, maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func stringAtMaxLengthIsAccepted() throws {
|
||||||
|
let exactString = String(repeating: "a", count: 100)
|
||||||
|
let result = InputValidator.validateUserString(exactString, maxLength: 100)
|
||||||
|
#expect(result == exactString)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func whitespaceIsTrimmed() throws {
|
||||||
|
let result = InputValidator.validateUserString(" Hello ", maxLength: 100)
|
||||||
|
#expect(result == "Hello")
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Control Character Tests
|
||||||
|
|
||||||
|
@Test func nullCharacterIsRejected() throws {
|
||||||
|
let stringWithNull = "Hello\u{0000}World"
|
||||||
|
let result = InputValidator.validateUserString(stringWithNull, maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func bellCharacterIsRejected() throws {
|
||||||
|
let stringWithBell = "Hello\u{0007}World"
|
||||||
|
let result = InputValidator.validateUserString(stringWithBell, maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func backspaceCharacterIsRejected() throws {
|
||||||
|
let stringWithBackspace = "Hello\u{0008}World"
|
||||||
|
let result = InputValidator.validateUserString(stringWithBackspace, maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func escapeCharacterIsRejected() throws {
|
||||||
|
let stringWithEscape = "Hello\u{001B}World"
|
||||||
|
let result = InputValidator.validateUserString(stringWithEscape, maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func deleteCharacterIsRejected() throws {
|
||||||
|
let stringWithDelete = "Hello\u{007F}World"
|
||||||
|
let result = InputValidator.validateUserString(stringWithDelete, maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func multipleControlCharactersAreRejected() throws {
|
||||||
|
let stringWithMultiple = "Hello\u{0000}\u{0007}\u{001B}World"
|
||||||
|
let result = InputValidator.validateUserString(stringWithMultiple, maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Unicode and Special Character Tests
|
||||||
|
|
||||||
|
@Test func emojiIsAccepted() throws {
|
||||||
|
let result = InputValidator.validateUserString("Hello 👋 World", maxLength: 100)
|
||||||
|
#expect(result == "Hello 👋 World")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func unicodeCharactersAreAccepted() throws {
|
||||||
|
let result = InputValidator.validateUserString("Hello 世界 مرحبا", maxLength: 100)
|
||||||
|
#expect(result == "Hello 世界 مرحبا")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func specialCharactersAreAccepted() throws {
|
||||||
|
let result = InputValidator.validateUserString("Hello!@#$%^&*()_+-=[]{}|;':\",./<>?", maxLength: 100)
|
||||||
|
#expect(result == "Hello!@#$%^&*()_+-=[]{}|;':\",./<>?")
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Nickname Validation Tests
|
||||||
|
|
||||||
|
@Test func validNicknameIsAccepted() throws {
|
||||||
|
let result = InputValidator.validateNickname("Alice")
|
||||||
|
#expect(result == "Alice")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func nicknameWithEmojiIsAccepted() throws {
|
||||||
|
let result = InputValidator.validateNickname("Alice 🚀")
|
||||||
|
#expect(result == "Alice 🚀")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func nicknameTooLongIsRejected() throws {
|
||||||
|
let longNickname = String(repeating: "a", count: 51)
|
||||||
|
let result = InputValidator.validateNickname(longNickname)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func nicknameAtMaxLengthIsAccepted() throws {
|
||||||
|
let exactNickname = String(repeating: "a", count: 50)
|
||||||
|
let result = InputValidator.validateNickname(exactNickname)
|
||||||
|
#expect(result == exactNickname)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func nicknameWithControlCharacterIsRejected() throws {
|
||||||
|
let result = InputValidator.validateNickname("Alice\u{0000}")
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Timestamp Validation Tests
|
||||||
|
// BCH-01-011: Window reduced from ±1 hour to ±5 minutes
|
||||||
|
|
||||||
|
@Test func currentTimestampIsValid() throws {
|
||||||
|
let now = Date()
|
||||||
|
let result = InputValidator.validateTimestamp(now)
|
||||||
|
#expect(result == true)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func timestampWithinFiveMinutesIsValid() throws {
|
||||||
|
// 2 minutes ago should be valid (within 5-minute window)
|
||||||
|
let twoMinutesAgo = Date().addingTimeInterval(-2 * 60)
|
||||||
|
let result = InputValidator.validateTimestamp(twoMinutesAgo)
|
||||||
|
#expect(result == true)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func timestampThirtyMinutesAgoIsInvalid() throws {
|
||||||
|
// BCH-01-011: 30 minutes is now outside the 5-minute window
|
||||||
|
let thirtyMinutesAgo = Date().addingTimeInterval(-30 * 60)
|
||||||
|
let result = InputValidator.validateTimestamp(thirtyMinutesAgo)
|
||||||
|
#expect(result == false)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func timestampTenMinutesAgoIsInvalid() throws {
|
||||||
|
// 10 minutes is outside the 5-minute window
|
||||||
|
let tenMinutesAgo = Date().addingTimeInterval(-10 * 60)
|
||||||
|
let result = InputValidator.validateTimestamp(tenMinutesAgo)
|
||||||
|
#expect(result == false)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func timestampTenMinutesInFutureIsInvalid() throws {
|
||||||
|
// 10 minutes in future is outside the 5-minute window
|
||||||
|
let tenMinutesFromNow = Date().addingTimeInterval(10 * 60)
|
||||||
|
let result = InputValidator.validateTimestamp(tenMinutesFromNow)
|
||||||
|
#expect(result == false)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func timestampAtFiveMinuteBoundaryIsValid() throws {
|
||||||
|
// Just slightly within the five-minute window (299 seconds)
|
||||||
|
let almostFiveMinutesAgo = Date().addingTimeInterval(-299)
|
||||||
|
let result = InputValidator.validateTimestamp(almostFiveMinutesAgo)
|
||||||
|
#expect(result == true)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func timestampJustOutsideFiveMinuteWindowIsInvalid() throws {
|
||||||
|
// Just outside the five-minute window (301 seconds)
|
||||||
|
let justOverFiveMinutesAgo = Date().addingTimeInterval(-301)
|
||||||
|
let result = InputValidator.validateTimestamp(justOverFiveMinutesAgo)
|
||||||
|
#expect(result == false)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Edge Cases
|
||||||
|
|
||||||
|
@Test func singleCharacterStringIsAccepted() throws {
|
||||||
|
let result = InputValidator.validateUserString("a", maxLength: 100)
|
||||||
|
#expect(result == "a")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func stringWithOnlyNewlinesIsRejected() throws {
|
||||||
|
let result = InputValidator.validateUserString("\n\n\n", maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func stringWithMixedWhitespaceIsTrimmed() throws {
|
||||||
|
let result = InputValidator.validateUserString(" \t\nHello\n\t ", maxLength: 100)
|
||||||
|
#expect(result == "Hello")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func stringWithLeadingControlCharacterIsRejected() throws {
|
||||||
|
let result = InputValidator.validateUserString("\u{0000}Hello", maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func stringWithTrailingControlCharacterIsRejected() throws {
|
||||||
|
let result = InputValidator.validateUserString("Hello\u{0000}", maxLength: 100)
|
||||||
|
#expect(result == nil)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,216 @@
|
|||||||
|
//
|
||||||
|
// KeychainErrorHandlingTests.swift
|
||||||
|
// bitchatTests
|
||||||
|
//
|
||||||
|
// This is free and unencumbered software released into the public domain.
|
||||||
|
// For more information, see <https://unlicense.org>
|
||||||
|
//
|
||||||
|
// BCH-01-009: Tests for proper keychain error classification and handling
|
||||||
|
|
||||||
|
import Testing
|
||||||
|
import Foundation
|
||||||
|
@testable import bitchat
|
||||||
|
|
||||||
|
struct KeychainErrorHandlingTests {
|
||||||
|
|
||||||
|
// MARK: - Error Classification Tests
|
||||||
|
|
||||||
|
@Test func keychainReadResult_successIsNotRecoverable() throws {
|
||||||
|
let result = KeychainReadResult.success(Data([1, 2, 3]))
|
||||||
|
#expect(result.isRecoverableError == false)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func keychainReadResult_itemNotFoundIsNotRecoverable() throws {
|
||||||
|
let result = KeychainReadResult.itemNotFound
|
||||||
|
#expect(result.isRecoverableError == false)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func keychainReadResult_deviceLockedIsRecoverable() throws {
|
||||||
|
let result = KeychainReadResult.deviceLocked
|
||||||
|
#expect(result.isRecoverableError == true)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func keychainReadResult_authenticationFailedIsRecoverable() throws {
|
||||||
|
let result = KeychainReadResult.authenticationFailed
|
||||||
|
#expect(result.isRecoverableError == true)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func keychainReadResult_accessDeniedIsNotRecoverable() throws {
|
||||||
|
let result = KeychainReadResult.accessDenied
|
||||||
|
#expect(result.isRecoverableError == false)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func keychainSaveResult_successIsNotRecoverable() throws {
|
||||||
|
let result = KeychainSaveResult.success
|
||||||
|
#expect(result.isRecoverableError == false)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func keychainSaveResult_duplicateItemIsRecoverable() throws {
|
||||||
|
let result = KeychainSaveResult.duplicateItem
|
||||||
|
#expect(result.isRecoverableError == true)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func keychainSaveResult_deviceLockedIsRecoverable() throws {
|
||||||
|
let result = KeychainSaveResult.deviceLocked
|
||||||
|
#expect(result.isRecoverableError == true)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func keychainSaveResult_storageFullIsNotRecoverable() throws {
|
||||||
|
let result = KeychainSaveResult.storageFull
|
||||||
|
#expect(result.isRecoverableError == false)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - Mock Keychain Error Simulation Tests
|
||||||
|
|
||||||
|
@Test func mockKeychain_canSimulateReadErrors() throws {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
|
||||||
|
// Simulate access denied error
|
||||||
|
keychain.simulatedReadError = .accessDenied
|
||||||
|
let result = keychain.getIdentityKeyWithResult(forKey: "testKey")
|
||||||
|
|
||||||
|
switch result {
|
||||||
|
case .accessDenied:
|
||||||
|
// Expected
|
||||||
|
break
|
||||||
|
default:
|
||||||
|
throw KeychainTestError("Expected accessDenied, got \(result)")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func mockKeychain_canSimulateSaveErrors() throws {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
|
||||||
|
// Simulate storage full error
|
||||||
|
keychain.simulatedSaveError = .storageFull
|
||||||
|
let result = keychain.saveIdentityKeyWithResult(Data([1, 2, 3]), forKey: "testKey")
|
||||||
|
|
||||||
|
switch result {
|
||||||
|
case .storageFull:
|
||||||
|
// Expected
|
||||||
|
break
|
||||||
|
default:
|
||||||
|
throw KeychainTestError("Expected storageFull, got \(result)")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func mockKeychain_returnsItemNotFoundForMissingKey() throws {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
let result = keychain.getIdentityKeyWithResult(forKey: "nonExistentKey")
|
||||||
|
|
||||||
|
switch result {
|
||||||
|
case .itemNotFound:
|
||||||
|
// Expected
|
||||||
|
break
|
||||||
|
default:
|
||||||
|
throw KeychainTestError("Expected itemNotFound, got \(result)")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func mockKeychain_returnsSuccessForExistingKey() throws {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
let testData = Data([1, 2, 3, 4, 5])
|
||||||
|
|
||||||
|
// First save the key
|
||||||
|
_ = keychain.saveIdentityKey(testData, forKey: "existingKey")
|
||||||
|
|
||||||
|
// Now read it back
|
||||||
|
let result = keychain.getIdentityKeyWithResult(forKey: "existingKey")
|
||||||
|
|
||||||
|
switch result {
|
||||||
|
case .success(let data):
|
||||||
|
#expect(data == testData)
|
||||||
|
default:
|
||||||
|
throw KeychainTestError("Expected success, got \(result)")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func mockKeychain_saveWithResultStoresData() throws {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
let testData = Data([10, 20, 30])
|
||||||
|
|
||||||
|
let saveResult = keychain.saveIdentityKeyWithResult(testData, forKey: "newKey")
|
||||||
|
|
||||||
|
switch saveResult {
|
||||||
|
case .success:
|
||||||
|
// Verify data was stored
|
||||||
|
let readResult = keychain.getIdentityKeyWithResult(forKey: "newKey")
|
||||||
|
switch readResult {
|
||||||
|
case .success(let data):
|
||||||
|
#expect(data == testData)
|
||||||
|
default:
|
||||||
|
throw KeychainTestError("Expected to read back saved data")
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
throw KeychainTestError("Expected save success, got \(saveResult)")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: - NoiseEncryptionService Integration Tests
|
||||||
|
|
||||||
|
@Test func noiseEncryptionService_generatesNewIdentityWhenMissing() throws {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
|
||||||
|
// Create service with empty keychain - should generate new identity
|
||||||
|
let service = NoiseEncryptionService(keychain: keychain)
|
||||||
|
|
||||||
|
// Should have generated and saved keys
|
||||||
|
#expect(service.getStaticPublicKeyData().count == 32)
|
||||||
|
#expect(service.getSigningPublicKeyData().count == 32)
|
||||||
|
|
||||||
|
// Keys should be persisted
|
||||||
|
let noiseKeyResult = keychain.getIdentityKeyWithResult(forKey: "noiseStaticKey")
|
||||||
|
switch noiseKeyResult {
|
||||||
|
case .success:
|
||||||
|
// Expected - key was saved
|
||||||
|
break
|
||||||
|
default:
|
||||||
|
throw KeychainTestError("Expected noise key to be saved")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func noiseEncryptionService_loadsExistingIdentity() throws {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
|
||||||
|
// Create first service to generate identity
|
||||||
|
let service1 = NoiseEncryptionService(keychain: keychain)
|
||||||
|
let originalPublicKey = service1.getStaticPublicKeyData()
|
||||||
|
let originalSigningKey = service1.getSigningPublicKeyData()
|
||||||
|
|
||||||
|
// Create second service - should load same identity
|
||||||
|
let service2 = NoiseEncryptionService(keychain: keychain)
|
||||||
|
|
||||||
|
#expect(service2.getStaticPublicKeyData() == originalPublicKey)
|
||||||
|
#expect(service2.getSigningPublicKeyData() == originalSigningKey)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func noiseEncryptionService_handlesAccessDeniedGracefully() throws {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
keychain.simulatedReadError = .accessDenied
|
||||||
|
|
||||||
|
// Service should still initialize with ephemeral key
|
||||||
|
let service = NoiseEncryptionService(keychain: keychain)
|
||||||
|
|
||||||
|
// Should have an identity (ephemeral)
|
||||||
|
#expect(service.getStaticPublicKeyData().count == 32)
|
||||||
|
#expect(service.getSigningPublicKeyData().count == 32)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test func noiseEncryptionService_handlesDeviceLockedGracefully() throws {
|
||||||
|
let keychain = MockKeychain()
|
||||||
|
keychain.simulatedReadError = .deviceLocked
|
||||||
|
|
||||||
|
// Service should still initialize with ephemeral key
|
||||||
|
let service = NoiseEncryptionService(keychain: keychain)
|
||||||
|
|
||||||
|
// Should have an identity (ephemeral)
|
||||||
|
#expect(service.getStaticPublicKeyData().count == 32)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Helper error type for tests
|
||||||
|
private struct KeychainTestError: Error, CustomStringConvertible {
|
||||||
|
let message: String
|
||||||
|
init(_ message: String) { self.message = message }
|
||||||
|
var description: String { message }
|
||||||
|
}
|
||||||
@@ -91,61 +91,3 @@ struct LocationNotesManagerTests {
|
|||||||
case shouldNotDerive
|
case shouldNotDerive
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@MainActor
|
|
||||||
struct LocationNotesCounterTests {
|
|
||||||
@Test func subscribeWithoutRelaysMarksUnavailable() {
|
|
||||||
var subscribeCalled = false
|
|
||||||
let deps = LocationNotesCounterDependencies(
|
|
||||||
relayLookup: { _, _ in [] },
|
|
||||||
subscribe: { _, _, _, _, _ in subscribeCalled = true },
|
|
||||||
unsubscribe: { _ in }
|
|
||||||
)
|
|
||||||
|
|
||||||
let counter = LocationNotesCounter(testDependencies: deps)
|
|
||||||
counter.subscribe(geohash: "u4pruydq")
|
|
||||||
|
|
||||||
#expect(!subscribeCalled)
|
|
||||||
#expect(!counter.relayAvailable)
|
|
||||||
#expect(counter.initialLoadComplete)
|
|
||||||
#expect(counter.count == 0)
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test func subscribeCountsUniqueNotes() {
|
|
||||||
var storedHandler: ((NostrEvent) -> Void)?
|
|
||||||
var storedEOSE: (() -> Void)?
|
|
||||||
let deps = LocationNotesCounterDependencies(
|
|
||||||
relayLookup: { _, _ in ["wss://relay.geo"] },
|
|
||||||
subscribe: { filter, id, relays, handler, eose in
|
|
||||||
#expect(relays == ["wss://relay.geo"])
|
|
||||||
#expect(filter.kinds == [1])
|
|
||||||
#expect(!id.isEmpty)
|
|
||||||
storedHandler = handler
|
|
||||||
storedEOSE = eose
|
|
||||||
},
|
|
||||||
unsubscribe: { _ in }
|
|
||||||
)
|
|
||||||
|
|
||||||
let counter = LocationNotesCounter(testDependencies: deps)
|
|
||||||
counter.subscribe(geohash: "u4pruydq")
|
|
||||||
|
|
||||||
var first = NostrEvent(
|
|
||||||
pubkey: "pub",
|
|
||||||
createdAt: Date(),
|
|
||||||
kind: .textNote,
|
|
||||||
tags: [["g", "u4pruydq"]],
|
|
||||||
content: "a"
|
|
||||||
)
|
|
||||||
first.id = "eventA"
|
|
||||||
storedHandler?(first)
|
|
||||||
|
|
||||||
let duplicate = first
|
|
||||||
storedHandler?(duplicate)
|
|
||||||
|
|
||||||
storedEOSE?()
|
|
||||||
|
|
||||||
#expect(counter.relayAvailable)
|
|
||||||
#expect(counter.count == 1)
|
|
||||||
#expect(counter.initialLoadComplete)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user