extension UI

Extension UI

Author: satyambnsal

README.md

@@ -128,3 +128,66 @@ See [LICENSE](LICENSE) file for details.
 ## Disclaimer
 
 This platform is for educational purposes and responsible red teaming. Use your powers for good, and happy hacking!
+
+## Development Setup
+
+### Prerequisites
+- Node.js and npm (for extension development)
+- Scarb (Cairo package manager)
+- jq (JSON processor for scripts)
+
+### Installation
+1. Install root dependencies:
+```bash
+npm install
+```
+
+2. Install jq if not already installed:
+```bash
+# macOS
+brew install jq
+
+# Ubuntu/Debian
+sudo apt-get install jq
+```
+
+3. Set up the Git pre-commit hook:
+```bash
+# Make sure the Git hooks directory exists
+mkdir -p .git/hooks
+
+# Copy the pre-commit hook
+cp scripts/pre-commit .git/hooks/
+chmod +x .git/hooks/pre-commit
+```
+
+### ABI Synchronization
+The project maintains automatic synchronization between the Cairo contract ABIs and the TypeScript interfaces used in the extension. This is handled through a Git pre-commit hook that:
+
+1. Builds the contracts using Scarb
+2. Extracts the ABIs from the contract class files using jq
+3. Updates the TypeScript ABI files in `extension/src/abis/`
+4. Verifies that all changes are committed
+
+If you see an error about ABI files being out of sync during commit, simply add the updated ABI files to your commit.
+
+### Manual ABI Sync
+To manually synchronize the ABIs:
+
+```bash
+./scripts/sync-abis.sh
+```
+
+## Project Structure
+
+### Contracts
+- `contracts/` - Cairo smart contracts
+- `contracts/target/release/*.contract_class.json` - Compiled contract files containing ABIs
+
+### Extension
+- `extension/` - Browser extension code
+- `extension/src/abis/` - TypeScript ABI definitions (auto-generated)
+
+### Scripts
+- `scripts/` - Development and maintenance scripts
+- `scripts/sync-abis.ts` - ABI synchronization script

contracts/tests/test_contract.cairo

@@ -82,6 +82,36 @@ fn deploy_test_token(token_holder: ContractAddress) -> ContractAddress {
     address.into()
 }
 
+fn deploy_registry_with_token(tee: ContractAddress) -> (ContractAddress, ContractAddress) {
+    let token_contract = declare("ERC20").unwrap();
+    let agent_contract = declare("Agent").unwrap();
+    let registry_contract = declare("AgentRegistry").unwrap();
+    
+    // Deploy token first
+    let mut token_calldata = ArrayTrait::new();
+    let name = ByteArray::from_string("Test Token");
+    let symbol = ByteArray::from_string("TST");
+    let initial_supply: u256 = 1000000;
+    let recipient = starknet::get_contract_address();
+    
+    name.serialize(ref token_calldata);
+    symbol.serialize(ref token_calldata);
+    initial_supply.serialize(ref token_calldata);
+    recipient.serialize(ref token_calldata);
+    
+    let (token_address, _) = token_contract.deploy(@token_calldata).unwrap();
+
+    // Deploy registry with token
+    let mut registry_calldata = ArrayTrait::new();
+    agent_contract.class_hash.serialize(ref registry_calldata);
+    tee.serialize(ref registry_calldata);
+    token_address.serialize(ref registry_calldata);
+    
+    let (registry_address, _) = registry_contract.deploy(@registry_calldata).unwrap();
+    
+    (registry_address, token_address)
+}
+
 #[test]
 fn test_register_agent() {
     let setup = setup();
@@ -1015,3 +1045,79 @@ fn test_withdraw_after_drain() {
     start_cheat_caller_address(agent_address, setup.creator);
     agent.withdraw(); // Should succeed
 }
+
+#[test]
+fn test_successful_agent_transfer() {
+    // Deploy registry with token
+    let tee = starknet::contract_address_const::<0x1>();
+    let (registry_address, token_address) = deploy_registry_with_token(tee);
+    let registry = IAgentRegistryDispatcher { contract_address: registry_address };
+    let token = IERC20Dispatcher { contract_address: token_address };
+
+    // Register an agent
+    registry.register_agent(
+        ByteArray::from_string("Test Agent"),
+        ByteArray::from_string("Test Prompt")
+    );
+    
+    let agents = registry.get_agents();
+    let agent_address = *agents.at(0);
+    
+    // Send some tokens to the agent
+    let agent_initial_balance: u256 = 100;
+    token.transfer(agent_address, agent_initial_balance);
+    
+    // Verify agent received the tokens
+    let balance_before = token.balance_of(agent_address);
+    assert(balance_before == agent_initial_balance, 'Wrong initial balance');
+    
+    // Create a new recipient address
+    let new_owner = starknet::contract_address_const::<0x123>();
+    
+    // Start spoofing the TEE address for the transfer
+    starknet::testing::set_caller_address(tee);
+    
+    // Transfer the agent to a new address
+    registry.transfer(agent_address, new_owner);
+    
+    // Verify tokens were transferred to new owner
+    let balance_after_agent = token.balance_of(agent_address);
+    let balance_after_new_owner = token.balance_of(new_owner);
+    
+    assert(balance_after_agent == 0, 'Agent should have 0 balance');
+    assert(balance_after_new_owner == agent_initial_balance, 'New owner wrong balance');
+}
+
+#[test]
+fn test_get_token_address() {
+    let tee = starknet::contract_address_const::<0x1>();
+    let (registry_address, token_address) = deploy_registry_with_token(tee);
+    let registry = IAgentRegistryDispatcher { contract_address: registry_address };
+    
+    let stored_token = registry.get_token();
+    assert(stored_token == token_address, 'Wrong token address');
+}
+
+#[test]
+#[should_panic(expected: ('Only tee can transfer',))]
+fn test_transfer_not_tee() {
+    let tee = starknet::contract_address_const::<0x1>();
+    let (registry_address, _) = deploy_registry_with_token(tee);
+    let registry = IAgentRegistryDispatcher { contract_address: registry_address };
+
+    // Register an agent
+    registry.register_agent(
+        ByteArray::from_string("Test Agent"),
+        ByteArray::from_string("Test Prompt")
+    );
+    
+    let agents = registry.get_agents();
+    let agent_address = *agents.at(0);
+    
+    // Try to transfer without being TEE
+    starknet::testing::set_caller_address(starknet::contract_address_const::<0x999>());
+    registry.transfer(
+        agent_address,
+        starknet::contract_address_const::<0x123>()
+    );
+}

extension/.env.development

@@ -0,0 +1 @@
+VITE_DEV_SERVER_URL=http://localhost:5173

extension/background.js

@@ -0,0 +1,6 @@
+console.log('Background script initiated')
+
+chrome.runtime.onInstalled.addListener(() => {
+  console.log('onInstalled listener handler 11')
+  // console.log('starknet', window.starknet_argentX)
+})

extension/manifest.json

@@ -5,19 +5,37 @@
     "description": "Pay-to-tweet mechanism for interacting with AI agents on Twitter/X",
     "permissions": [
         "activeTab",
-        "cookies"
+        "tabs",
+        "cookies",
+        "scripting",
+        "storage"
     ],
     "host_permissions": [
         "https://*.twitter.com/*"
     ],
     "content_scripts": [
         {
-            "matches": ["https://twitter.com/*", "https://x.com/*"],
-            "css": ["assets/contentStyle.css"],
-            "js": ["src/content-script/index.tsx"]
+            "matches": [
+                "https://twitter.com/*",
+                "https://x.com/*"
+            ],
+            "css": [
+                "assets/contentStyle.css"
+            ],
+            "js": [
+                "src/content-script/index.tsx"
+            ],
+            "run_at": "document_end",
+            "all_frames": true
         }
     ],
     "action": {
-      "default_popup": "index.html"
+        "default_popup": "index.html"
+    },
+    "background": {
+        "service_worker": "background.js"
+    },
+    "content_security_policy": {
+        "extension_pages": "script-src 'self' 'wasm-unsafe-eval'; object-src 'self'"
     }
-}
+}