buzzert/Sybil-2
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

feat(agent): add opt-in push notifications

Browse Source
This commit is contained in:
buzzert
2026-10-14 22:03:21 -07:00
parent 80748f764f
commit a1c0264d36
51 changed files with 4815 additions and 121 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ios/Tests/Fixtures/agent_server.py
+42 -3
View File
@@ -26,6 +26,12 @@ attachment_capability = True
next_upload_error = None
blocked_proxy = False
transcript_comparison = False
PUSH_SOURCE = '11111111-1111-4111-8111-111111111111'
push_installations = {}
push_retired = set()
push_requests = []
push_available = True
auth_required = False
CAPABILITY = dict(version=1, maxCount=8, maxTotalBytes=33554432, maxImageBytes=6291456,
maxFileBytes=8388608, imageMimeTypes=['image/png', 'image/jpeg', 'image/gif', 'image/webp'],
maxImagePixels=50000000, maxImageDimension=16384, quotaBytes=1073741824, usedBytes=0,
@@ -88,7 +94,7 @@ async def reply(content):
await broadcast(dict(type='activity', conversationId='fixture-root', activity=activity()))
async def route(request):
global read_cursor, cursor, busy, attachment_capability, next_upload_error, blocked_proxy, transcript_comparison
global read_cursor, cursor, busy, attachment_capability, next_upload_error, blocked_proxy, transcript_comparison, push_available, auth_required
path = request.path
log.append(dict(method=request.method, path=path, query=dict(request.query)))
if path == '/control':
@@ -100,6 +106,8 @@ async def route(request):
reply_tasks.clear()
messages.clear(); requests.clear(); attempts.clear(); cursor = 0; read_cursor = 0; busy = False
attachment_capability = True; next_upload_error = None; blocked_proxy = False; transcript_comparison = False
push_installations.clear(); push_retired.clear(); push_requests.clear(); log.clear()
push_available = True; auth_required = False
for i in range(body.get('count', 3)):
message(f'History message {i + 1}: local simulator fixture.')
elif action == 'transcript-comparison':
@@ -126,13 +134,21 @@ async def route(request):
await broadcast(dict(type='activity', conversationId='fixture-root', activity=activity()))
elif action == 'read':
read_cursor = cursor
await broadcast(dict(type='read_state', sourceId=PUSH_SOURCE, cursor=str(read_cursor), updatedAt=stamp()))
elif action == 'push-available':
push_available = body.get('enabled', True)
elif action == 'auth-required':
auth_required = body.get('enabled', True)
elif action == 'disconnect':
for socket in list(sockets):
await socket.close(code=1012)
elif action == 'malformed':
await broadcast(dict(type='messages', conversationId='fixture-root', messages=[{'id': 'broken'}], cursor='999999', beforeCursor=None, hasMore=False))
return web.json_response(dict(cursor=cursor, readCursor=read_cursor, messages=messages, requests=requests, log=log))
return web.json_response(dict(cursor=cursor, readCursor=read_cursor, messages=messages, requests=requests, log=log,
pushRequests=push_requests, pushActive=len(push_installations), pushRetired=len(push_retired)))
if path == '/api/v1/auth/session':
if auth_required and request.headers.get('Authorization') != 'Bearer fixture-token':
return web.json_response(dict(message='Fixture authentication required'), status=401)
return web.json_response(dict(authenticated=True, mode='token'))
comparison = dict(id='fade-comparison', title='Transcript fade comparison', createdAt=stamp(), updatedAt=stamp())
if path == '/api/v1/workspace-items':
@@ -157,6 +173,29 @@ async def route(request):
if attachment_capability:
descriptor['capabilities'] = {'attachments': CAPABILITY}
return web.json_response(descriptor)
if path == '/api/v1/agent/push/config':
if not push_available:
return web.json_response(dict(message='Old server fixture'), status=404)
return web.json_response(dict(version=1, enabled=True, apnsAvailable=True, webPushAvailable=False,
sourceId=PUSH_SOURCE, quarantined=False, policy={}))
if path.startswith('/api/v1/agent/push/installations/'):
installation = str(uuid.UUID(path.rsplit('/', 1)[1]))
if request.method == 'DELETE':
push_installations.pop(installation, None); push_retired.add(installation)
push_requests.append(dict(method='DELETE', installationId=installation))
return web.Response(status=204)
if request.method == 'PUT':
body = await request.json()
assert set(body) == {'transport', 'enabled', 'token', 'environment', 'topic'}
assert body['transport'] == 'apns' and body['enabled'] is True
assert body['environment'] == 'sandbox' and body['topic'] == 'net.buzzert.sybil2'
assert body['token'] == '00010aff' + '42' * 28 # explicitly injected synthetic token
if installation in push_retired:
return web.json_response(dict(message='Retired installation'), status=409)
version = push_installations.get(installation, 0) + 1
push_installations[installation] = version
push_requests.append(dict(method='PUT', installationId=installation, environment='sandbox'))
return web.json_response(dict(installationId=installation, transport='apns', enabled=True, version=version))
if path == '/api/v1/agent/attachments' and request.method == 'POST':
if not attachment_capability:
return attachment_error(503, 'attachments_unavailable', 'Attachments are unavailable.')
@@ -217,7 +256,7 @@ async def route(request):
if path == '/api/v1/agent/read-state':
if request.method == 'PUT':
read_cursor = max(read_cursor, int((await request.json())['cursor']))
return web.json_response(dict(cursor=str(read_cursor), updatedAt=stamp()))
return web.json_response(dict(cursor=str(read_cursor), updatedAt=stamp(), sourceId=PUSH_SOURCE))
if path == '/api/v1/agent/messages':
if request.method == 'POST':
body = await request.json()
ios/Tests/SybilUITests/AgentPushUITests.swift
+128
View File
@@ -0,0 +1,128 @@
import XCTest
/// Local fixture + fake notification center/registrar only. No APNs provider or
/// system permission dialog is involved in these deterministic UI scenarios.
@MainActor final class AgentPushUITests: XCTestCase {
let app = XCUIApplication()
override func setUp() async throws {
continueAfterFailure = false
try await control(["action": "reset", "count": 3])
app.launchArguments = ["-sybil.ios.apiBaseURL", "http://127.0.0.1:8799/api", "-sybil.ios.adminToken", "fixture-token"]
app.launchEnvironment = ["SYBIL_PUSH_FIXTURE": "1", "SYBIL_PUSH_PERMISSION": "authorized"]
}
override func tearDown() async throws { app.terminate() }
@discardableResult private func control(_ body: [String: Any]) async throws -> [String: Any] {
var request = URLRequest(url: URL(string: "http://127.0.0.1:8799/control")!)
request.httpMethod = "POST"; request.httpBody = try JSONSerialization.data(withJSONObject: body)
request.setValue("application/json", forHTTPHeaderField: "Content-Type")
let (data, _) = try await URLSession.shared.data(for: request)
return try XCTUnwrap(JSONSerialization.jsonObject(with: data) as? [String: Any])
}
private func settings() {
if !app.buttons["Settings"].isHittable {
XCTAssertTrue(app.buttons["Show sidebar"].waitForExistence(timeout: 20))
app.buttons["Show sidebar"].tap()
}
XCTAssertTrue(app.buttons["Settings"].waitForExistence(timeout: 5)); app.buttons["Settings"].tap()
XCTAssertTrue(app.buttons["agent-push-refresh"].waitForExistence(timeout: 5))
if !app.buttons["agent-push-refresh"].isHittable { app.scrollViews.firstMatch.swipeUp() }
}
private func capture(_ name: String) {
let attachment = XCTAttachment(screenshot: XCUIScreen.main.screenshot())
attachment.name = name; attachment.lifetime = .keepAlways; add(attachment)
}
private func openAgent() {
if !app.buttons["agent-sidebar"].waitForExistence(timeout: 2) || !app.buttons["agent-sidebar"].isHittable {
XCTAssertTrue(app.buttons["Show sidebar"].waitForExistence(timeout: 20)); app.buttons["Show sidebar"].tap()
}
app.buttons["agent-sidebar"].tap()
XCTAssertTrue(app.textFields["agent-composer"].waitForExistence(timeout: 5))
}
private func ensureDisabled() {
if app.buttons["agent-push-disable"].exists { app.buttons["agent-push-disable"].tap() }
XCTAssertTrue(app.buttons["agent-push-enable"].waitForExistence(timeout: 5))
}
func testExplicitEnableRefreshAndDisable() async throws {
app.launch(); settings(); ensureDisabled()
let before = try await control(["action": "state"])
let count = (before["pushRequests"] as? [[String: Any]])?.count ?? 0
app.buttons["agent-push-enable"].tap()
XCTAssertTrue(app.buttons["agent-push-disable"].waitForExistence(timeout: 5))
let status = app.staticTexts["agent-push-status"]
let enabled = NSPredicate(format: "label == %@", "Notifications are enabled.")
let enabledExpectation = expectation(for: enabled, evaluatedWith: status)
await fulfillment(of: [enabledExpectation], timeout: 8)
capture("push-enabled-generic-preview")
app.buttons["agent-push-refresh"].tap()
let state = try await control(["action": "state"])
let requests = try XCTUnwrap(state["pushRequests"] as? [[String: Any]])
XCTAssertEqual(requests.count, count + 1)
XCTAssertEqual(requests.last?["environment"] as? String, "sandbox")
app.buttons["agent-push-disable"].tap()
XCTAssertTrue(app.buttons["agent-push-enable"].waitForExistence(timeout: 5))
let revoked = try await control(["action": "state"])
XCTAssertEqual(revoked["pushActive"] as? Int, 0)
capture("push-disabled-revoked")
}
func testColdTapRoutesAgentAndPreservesAttachmentCaption() async throws {
app.launch()
openAgent()
let composer = app.textFields["agent-composer"]
XCTAssertTrue(composer.waitForExistence(timeout: 5))
composer.tap(); composer.typeText("Push fixture saved caption")
app.terminate()
app.launchEnvironment["SYBIL_PUSH_TAP"] = "1"
app.launch()
XCTAssertTrue(composer.waitForExistence(timeout: 20))
XCTAssertTrue((composer.value as? String ?? "").contains("Push fixture saved caption"))
XCTAssertTrue(app.buttons["agent-attach"].exists)
capture("push-cold-tap-agent-draft-preserved")
}
func testTapBeforeLoginWaitsThenRoutesAgent() async throws {
try await control(["action": "auth-required", "enabled": true])
app.launchArguments = ["-sybil.ios.apiBaseURL", "http://127.0.0.1:8799/api", "-sybil.ios.adminToken", ""]
app.launchEnvironment["SYBIL_PUSH_TAP"] = "1"
app.launch()
XCTAssertTrue(app.buttons["Connect"].waitForExistence(timeout: 15))
XCTAssertFalse(app.textFields["agent-composer"].exists)
let state = try await control(["action": "state"])
XCTAssertEqual(state["readCursor"] as? Int, 0)
app.secureTextFields.firstMatch.tap(); app.secureTextFields.firstMatch.typeText("fixture-token")
app.buttons["Connect"].tap()
XCTAssertTrue(app.textFields["agent-composer"].waitForExistence(timeout: 20))
capture("push-pending-auth-routed-after-login")
}
func testDeniedAndOldServerLeaveAgentUsable() async throws {
app.launchEnvironment["SYBIL_PUSH_PERMISSION"] = "denied"
app.launch(); settings()
XCTAssertTrue(app.buttons["Open iOS Notification Settings"].waitForExistence(timeout: 5))
capture("push-denied-settings")
try await control(["action": "push-available", "enabled": false])
app.buttons["agent-push-refresh"].tap()
XCTAssertTrue(app.buttons["Show sidebar"].exists); app.buttons["Show sidebar"].tap()
app.buttons["agent-sidebar"].tap()
XCTAssertTrue(app.textFields["agent-composer"].waitForExistence(timeout: 10))
XCTAssertTrue(app.staticTexts["History message 3: local simulator fixture."].waitForExistence(timeout: 5))
capture("push-old-server-agent-compatible")
}
func testWarmTapDismissesAttachmentPickerPreservingDraft() async throws {
app.launchEnvironment["SYBIL_PUSH_TAP_ON_FOREGROUND"] = "1"
app.launch(); openAgent()
let composer = app.textFields["agent-composer"]
composer.tap(); composer.typeText(" Warm push preserved draft")
app.buttons["agent-attach"].tap(); app.buttons["Files"].tap()
XCTAssertTrue(app.buttons["Cancel"].firstMatch.waitForExistence(timeout: 5))
XCUIDevice.shared.press(.home); app.activate()
XCTAssertTrue(composer.waitForExistence(timeout: 10))
let visible = expectation(for: NSPredicate(format: "hittable == true"), evaluatedWith: composer)
await fulfillment(of: [visible], timeout: 5)
XCTAssertFalse(app.buttons["Cancel"].firstMatch.exists)
XCTAssertTrue((composer.value as? String ?? "").contains("Warm push preserved draft"))
capture("push-warm-tap-dismissed-picker-draft-preserved")
}
}