mirror of
https://github.com/outbackdingo/optimclaw.git
synced 2026-08-26 15:40:18 +00:00
Compare commits
3
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3ca8b1bf68 | ||
|
|
d887309208 | ||
|
|
0c119b5c1e |
@@ -2640,7 +2640,7 @@ function renderExtensionCard(ext) {
|
|||||||
pairingSection.className = 'ext-pairing';
|
pairingSection.className = 'ext-pairing';
|
||||||
pairingSection.setAttribute('data-channel', ext.name);
|
pairingSection.setAttribute('data-channel', ext.name);
|
||||||
card.appendChild(pairingSection);
|
card.appendChild(pairingSection);
|
||||||
loadPairingRequests(ext.name, pairingSection);
|
loadPairingRequests(ext.name, pairingSection, ext.activation_status);
|
||||||
}
|
}
|
||||||
|
|
||||||
return card;
|
return card;
|
||||||
@@ -3034,11 +3034,19 @@ function openOAuthUrl(url) {
|
|||||||
|
|
||||||
// --- Pairing ---
|
// --- Pairing ---
|
||||||
|
|
||||||
function loadPairingRequests(channel, container) {
|
function loadPairingRequests(channel, container, status) {
|
||||||
apiFetch('/api/pairing/' + encodeURIComponent(channel))
|
apiFetch('/api/pairing/' + encodeURIComponent(channel))
|
||||||
.then(data => {
|
.then(data => {
|
||||||
container.innerHTML = '';
|
container.innerHTML = '';
|
||||||
if (!data.requests || data.requests.length === 0) return;
|
if (!data.requests || data.requests.length === 0) {
|
||||||
|
if (status === 'pairing') {
|
||||||
|
const hint = document.createElement('p');
|
||||||
|
hint.className = 'pairing-hint';
|
||||||
|
hint.textContent = 'Send any message to your bot to receive a pairing request here.';
|
||||||
|
container.appendChild(hint);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
const heading = document.createElement('div');
|
const heading = document.createElement('div');
|
||||||
heading.className = 'pairing-heading';
|
heading.className = 'pairing-heading';
|
||||||
|
|||||||
@@ -2865,6 +2865,13 @@ body {
|
|||||||
flex: 1;
|
flex: 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.pairing-hint {
|
||||||
|
color: var(--text-secondary);
|
||||||
|
font-size: 13px;
|
||||||
|
margin: 4px 0 8px;
|
||||||
|
font-style: italic;
|
||||||
|
}
|
||||||
|
|
||||||
/* Configure modal */
|
/* Configure modal */
|
||||||
.configure-overlay {
|
.configure-overlay {
|
||||||
position: fixed;
|
position: fixed;
|
||||||
|
|||||||
@@ -3739,6 +3739,26 @@ impl ExtensionManager {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Credentials changed (new bot token) — clear pairing state so existing users
|
||||||
|
// must re-approve with the new bot identity.
|
||||||
|
if cred_count > 0 {
|
||||||
|
let pairing_store = crate::pairing::PairingStore::new();
|
||||||
|
if let Err(e) = pairing_store.clear_allow_from(name) {
|
||||||
|
tracing::warn!(
|
||||||
|
channel = %name,
|
||||||
|
error = %e,
|
||||||
|
"Failed to clear allow-from on credential refresh"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
if let Err(e) = pairing_store.clear_pending(name) {
|
||||||
|
tracing::warn!(
|
||||||
|
channel = %name,
|
||||||
|
error = %e,
|
||||||
|
"Failed to clear pending pairings on credential refresh"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Load capabilities file once to extract all secret names
|
// Load capabilities file once to extract all secret names
|
||||||
let cap_path = self
|
let cap_path = self
|
||||||
.wasm_channels_dir
|
.wasm_channels_dir
|
||||||
|
|||||||
@@ -440,6 +440,39 @@ impl PairingStore {
|
|||||||
Ok(file.allow_from)
|
Ok(file.allow_from)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Clear the allow-from list for a channel.
|
||||||
|
///
|
||||||
|
/// Called on credential refresh so that existing users must re-approve
|
||||||
|
/// after a bot token change.
|
||||||
|
pub fn clear_allow_from(&self, channel: &str) -> Result<(), PairingStoreError> {
|
||||||
|
let path = allow_from_path(&self.base_dir, channel)?;
|
||||||
|
if let Some(parent) = path.parent() {
|
||||||
|
fs::create_dir_all(parent)?;
|
||||||
|
}
|
||||||
|
let file = fs::OpenOptions::new()
|
||||||
|
.read(true)
|
||||||
|
.write(true)
|
||||||
|
.create(true)
|
||||||
|
.truncate(true)
|
||||||
|
.open(&path)?;
|
||||||
|
file.lock_exclusive()?;
|
||||||
|
let store = AllowFromStoreFile {
|
||||||
|
version: 1,
|
||||||
|
allow_from: Vec::new(),
|
||||||
|
};
|
||||||
|
let json = serde_json::to_string_pretty(&store)?;
|
||||||
|
fs::write(&path, json)?;
|
||||||
|
fs4::FileExt::unlock(&file)?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Clear all pending pairing requests for a channel.
|
||||||
|
///
|
||||||
|
/// Called on credential refresh so stale requests don't confuse users.
|
||||||
|
pub fn clear_pending(&self, channel: &str) -> Result<(), PairingStoreError> {
|
||||||
|
self.write_pairing_file(channel, &[])
|
||||||
|
}
|
||||||
|
|
||||||
/// Check if a sender is allowed (by id or username).
|
/// Check if a sender is allowed (by id or username).
|
||||||
pub fn is_sender_allowed(
|
pub fn is_sender_allowed(
|
||||||
&self,
|
&self,
|
||||||
@@ -517,6 +550,11 @@ impl PairingStore {
|
|||||||
requests: &[PairingRequest],
|
requests: &[PairingRequest],
|
||||||
) -> Result<(), PairingStoreError> {
|
) -> Result<(), PairingStoreError> {
|
||||||
let path = pairing_path(&self.base_dir, channel)?;
|
let path = pairing_path(&self.base_dir, channel)?;
|
||||||
|
let parent = path.parent().ok_or_else(|| {
|
||||||
|
PairingStoreError::InvalidPath(format!("path has no parent: {}", path.display()))
|
||||||
|
})?;
|
||||||
|
fs::create_dir_all(parent)?;
|
||||||
|
|
||||||
let mut file = fs::OpenOptions::new()
|
let mut file = fs::OpenOptions::new()
|
||||||
.write(true)
|
.write(true)
|
||||||
.create(true)
|
.create(true)
|
||||||
@@ -717,4 +755,117 @@ mod tests {
|
|||||||
store.list_pending("").unwrap_err();
|
store.list_pending("").unwrap_err();
|
||||||
store.upsert_request("", "u1", None).unwrap_err();
|
store.upsert_request("", "u1", None).unwrap_err();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_clear_allow_from_removes_all_entries() {
|
||||||
|
let (store, _) = test_store();
|
||||||
|
let r1 = store.upsert_request("telegram", "user1", None).unwrap();
|
||||||
|
store.approve("telegram", &r1.code).unwrap();
|
||||||
|
|
||||||
|
let list = store.read_allow_from("telegram").unwrap();
|
||||||
|
assert_eq!(list.len(), 1);
|
||||||
|
|
||||||
|
store.clear_allow_from("telegram").unwrap();
|
||||||
|
let list = store.read_allow_from("telegram").unwrap();
|
||||||
|
assert!(list.is_empty());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_clear_pending_removes_all_requests() {
|
||||||
|
let (store, _) = test_store();
|
||||||
|
store
|
||||||
|
.upsert_request("telegram", "user1", Some(serde_json::json!({"chat_id": 1})))
|
||||||
|
.unwrap();
|
||||||
|
store
|
||||||
|
.upsert_request("telegram", "user2", Some(serde_json::json!({"chat_id": 2})))
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let requests = store.list_pending("telegram").unwrap();
|
||||||
|
assert_eq!(requests.len(), 2);
|
||||||
|
|
||||||
|
store.clear_pending("telegram").unwrap();
|
||||||
|
let requests = store.list_pending("telegram").unwrap();
|
||||||
|
assert!(requests.is_empty());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_clear_allow_from_allows_new_approval() {
|
||||||
|
let (store, _) = test_store();
|
||||||
|
let r1 = store.upsert_request("telegram", "user1", None).unwrap();
|
||||||
|
store.approve("telegram", &r1.code).unwrap();
|
||||||
|
|
||||||
|
assert!(store.is_sender_allowed("telegram", "user1", None).unwrap());
|
||||||
|
|
||||||
|
store.clear_allow_from("telegram").unwrap();
|
||||||
|
|
||||||
|
assert!(!store.is_sender_allowed("telegram", "user1", None).unwrap());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_clear_allow_from_on_nonexistent_file() {
|
||||||
|
let (store, _) = test_store();
|
||||||
|
// No requests created, so allow_from file doesn't exist
|
||||||
|
let result = store.clear_allow_from("telegram");
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
// After clearing, should return empty list
|
||||||
|
let list = store.read_allow_from("telegram").unwrap();
|
||||||
|
assert!(list.is_empty());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_clear_pending_on_nonexistent_file() {
|
||||||
|
let (store, _) = test_store();
|
||||||
|
// No requests created, so pairing file doesn't exist
|
||||||
|
let result = store.clear_pending("telegram");
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
// After clearing, should return empty list
|
||||||
|
let requests = store.list_pending("telegram").unwrap();
|
||||||
|
assert!(requests.is_empty());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_clear_and_reapprove_workflow() {
|
||||||
|
let (store, _) = test_store();
|
||||||
|
|
||||||
|
// Step 1: Create and approve user1
|
||||||
|
let r1 = store.upsert_request("telegram", "user1", None).unwrap();
|
||||||
|
store.approve("telegram", &r1.code).unwrap();
|
||||||
|
assert!(store.is_sender_allowed("telegram", "user1", None).unwrap());
|
||||||
|
|
||||||
|
// Step 2: Simulate credential refresh by clearing pairing state
|
||||||
|
store.clear_allow_from("telegram").unwrap();
|
||||||
|
store.clear_pending("telegram").unwrap();
|
||||||
|
|
||||||
|
// Step 3: Verify user1 is no longer approved and no pending requests exist
|
||||||
|
assert!(!store.is_sender_allowed("telegram", "user1", None).unwrap());
|
||||||
|
let requests = store.list_pending("telegram").unwrap();
|
||||||
|
assert!(requests.is_empty());
|
||||||
|
|
||||||
|
// Step 4: Create new pairing request and approve user1 again
|
||||||
|
let r2 = store.upsert_request("telegram", "user1", None).unwrap();
|
||||||
|
assert!(r2.created); // Should be a new request
|
||||||
|
store.approve("telegram", &r2.code).unwrap();
|
||||||
|
assert!(store.is_sender_allowed("telegram", "user1", None).unwrap());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_clear_one_channel_doesnt_affect_other() {
|
||||||
|
let (store, _) = test_store();
|
||||||
|
|
||||||
|
// Approve users on two channels
|
||||||
|
let r1 = store.upsert_request("telegram", "user1", None).unwrap();
|
||||||
|
store.approve("telegram", &r1.code).unwrap();
|
||||||
|
|
||||||
|
let r2 = store.upsert_request("discord", "user2", None).unwrap();
|
||||||
|
store.approve("discord", &r2.code).unwrap();
|
||||||
|
|
||||||
|
// Clear only telegram
|
||||||
|
store.clear_allow_from("telegram").unwrap();
|
||||||
|
|
||||||
|
// Verify telegram is cleared but discord is not
|
||||||
|
assert!(!store.is_sender_allowed("telegram", "user1", None).unwrap());
|
||||||
|
assert!(store.is_sender_allowed("discord", "user2", None).unwrap());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user