From 0577c1d6b502ebe84ae60f75b8a3554398dd6171 Mon Sep 17 00:00:00 2001
From: James Campbell <james.campbell@tanti.org.uk>
Date: Wed, 15 May 2024 14:29:47 +0200
Subject: [PATCH] Add first draft of InfractionCollector

Fixes #266
---
 .../coordination/InfractionCollector.sol      | 40 +++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 contracts/contracts/coordination/InfractionCollector.sol

diff --git a/contracts/contracts/coordination/InfractionCollector.sol b/contracts/contracts/coordination/InfractionCollector.sol
new file mode 100644
index 00000000..1c1d7e0d
--- /dev/null
+++ b/contracts/contracts/coordination/InfractionCollector.sol
@@ -0,0 +1,40 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.0;
+
+import "@openzeppelin-upgradeable/contracts/access/OwnableUpgradeable.sol";
+import "./Coordinator.sol";
+import "./TACoChildApplication.sol";
+
+contract InfractionCollector is OwnableUpgradeable {
+    // Reference to the Coordinator contract
+    Coordinator private coordinator;
+
+    // Reference to the TACoChildApplication contract
+    TACoChildApplication private tacoChildApplication;
+
+    // Mapping to keep track of reported infractions
+    mapping(bytes32 => mapping(address => bool)) private infractions;
+
+    function initialize(Coordinator _coordinator, TACoChildApplication _tacoChildApplication) public initializer {
+        __Ownable_init();
+
+        coordinator = _coordinator;
+        tacoChildApplication = _tacoChildApplication;
+    }
+
+    function reportMissingTranscript(bytes32 ritualId, address[] calldata stakingProviders) external {
+        // Ritual must have failed
+        require(coordinator.getRitualState(ritualId) == Coordinator.RitualState.DKG_TIMEOUT, "Ritual must have failed");
+
+        for (uint256 i = 0; i < stakingProviders.length; i++) {
+            // Check if the infraction has already been reported
+            require(!infractions[ritualId][stakingProviders[i]], "Infraction already reported");
+            participant = coordinator.getParticipantFromProvider(ritualId, stakingProviders[i]);
+            if (participant.transcript.length == 0) {
+                // Penalize the staking provider
+                tacoChildApplication.penalize(stakingProviders[i]);
+                infractions[ritualId][stakingProviders[i]] = true;
+            }
+        }
+    }
+}
\ No newline at end of file