From 07617dd083c5950b76806f19301f2a6d771bad1a Mon Sep 17 00:00:00 2001
From: Alex Vanin <alexey@nspcc.ru>
Date: Thu, 29 Apr 2021 16:07:08 +0300
Subject: [PATCH] [#74] audit: Support notary disabled work flow

Signed-off-by: Alex Vanin <alexey@nspcc.ru>
---
 audit/audit_contract.go | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/audit/audit_contract.go b/audit/audit_contract.go
index 47b7794..276a6a9 100644
--- a/audit/audit_contract.go
+++ b/audit/audit_contract.go
@@ -78,7 +78,16 @@ func Migrate(script []byte, manifest []byte) bool {
 
 func Put(rawAuditResult []byte) bool {
 	ctx := storage.GetContext()
-	innerRing := common.InnerRingNodes()
+	notaryDisabled := storage.Get(ctx, notaryDisabledKey).(bool)
+
+	var innerRing []common.IRNode
+
+	if notaryDisabled {
+		netmapContract := storage.Get(ctx, netmapContractKey).(interop.Hash160)
+		innerRing = common.InnerRingNodesFromNetmap(netmapContract)
+	} else {
+		innerRing = common.InnerRingNodes()
+	}
 
 	hdr := newAuditHeader(rawAuditResult)
 	presented := false
@@ -153,6 +162,7 @@ func list(it iterator.Iterator) [][]byte {
 	ignore := [][]byte{
 		[]byte(netmapContractKey),
 		[]byte(common.OwnerKey),
+		[]byte(notaryDisabledKey),
 	}
 
 loop: