fix(registry notifications): send state notice to the ant registry in…

…stead of the user on transfer
ar-io · Nov 24, 2024 · 856be78 · 856be78
1 parent 258a7d2
commit 856be78
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/src/common/utils.lua b/src/common/utils.lua
@@ -236,8 +236,8 @@ function utils.createHandler(tagName, tagValue, handler, position)
 
 			local hasNewOwner = Owner ~= prevOwner
 			local hasDifferentControllers = #utils.keys(Controllers) ~= #utils.keys(prevControllers)
-			if (hasNewOwner or hasDifferentControllers) and tagValue ~= "State" then
-				notices.notifyState(msg, msg.From)
+			if (hasNewOwner or hasDifferentControllers) and tagValue ~= "State" and AntRegistryId ~= nil then
+				notices.notifyState(msg, AntRegistryId)
 			end
 
 			return handlerRes

diff --git a/test/registry.test.mjs b/test/registry.test.mjs
@@ -5,6 +5,7 @@ import {
   AO_LOADER_HANDLER_ENV,
   DEFAULT_HANDLE_OPTIONS,
   STUB_ADDRESS,
+  STUB_ANT_REGISTRY_ID,
 } from '../tools/constants.mjs';
 
 describe('Registry Updates', async () => {
@@ -94,5 +95,6 @@ describe('Registry Updates', async () => {
     );
 
     assert(notifyMessage, 'State-Notice message not found');
+    assert.strictEqual(result.Messages[2]?.Target, STUB_ANT_REGISTRY_ID);
   });
 });
diff --git a/tools/constants.mjs b/tools/constants.mjs
@@ -7,14 +7,15 @@ const __dirname = path.dirname(__filename);
 
 const STUB_PROCESS_ID = 'process-id-'.padEnd(43, '1');
 const STUB_ADDRESS = 'arweave-address-'.padEnd(43, '1');
+export const STUB_ANT_REGISTRY_ID = 'ant-registry-'.padEnd(43, '1');
 /* ao READ-ONLY Env Variables */
 const AO_LOADER_HANDLER_ENV = {
   Process: {
     Id: STUB_PROCESS_ID,
     Owner: STUB_ADDRESS,
     Tags: [
       { name: 'Authority', value: 'XXXXXX' },
-      { name: 'ANT-Registry-Id', value: 'ant-registry-'.padEnd(43, '1') },
+      { name: 'ANT-Registry-Id', value: STUB_ANT_REGISTRY_ID },
     ],
   },
   Module: {