Work-package submission (#290)

* update issue 278

* Work-package submission

* update networkmanager

* update more test

* update workpackage hash

* update networkmanager

* update workpackage

* update workpackpool

* update NetworkManagerTest

* update test

* update

Author: MacOMNI

Blockchain/Sources/Blockchain/RuntimeProtocols/RuntimeEvents.swift

@@ -54,7 +54,15 @@ public enum RuntimeEvents {
 
     // When a work package is recived via CE133
     public struct WorkPackagesReceived: Event {
-        public let item: WorkPackageRef
+        public let coreIndex: CoreIndex
+        public let workPackageRef: WorkPackageRef
+        public let extrinsics: [Data]
+
+        public init(coreIndex: CoreIndex, workPackageRef: WorkPackageRef, extrinsics: [Data]) {
+            self.coreIndex = coreIndex
+            self.workPackageRef = workPackageRef
+            self.extrinsics = extrinsics
+        }
     }
 
     // When a work package bundle is ready to shared via CE134

Blockchain/Sources/Blockchain/Types/RefinementContext.swift

@@ -4,7 +4,7 @@ import Utils
 // A refinement context, denoted by the set X, describes the context of the chain
 // at the point that the report’s corresponding work-package was evaluated.
 public struct RefinementContext: Comparable, Sendable, Equatable, Codable {
-    public struct Anchor: Comparable, Sendable, Equatable, Codable {
+    public struct Anchor: Comparable, Sendable, Equatable, Codable, Hashable {
         // a
         public var headerHash: Data32
         // s

Blockchain/Sources/Blockchain/Types/WorkItem.swift

@@ -2,9 +2,9 @@ import Foundation
 import Utils
 
 // I
-public struct WorkItem: Sendable, Equatable, Codable {
-    public struct ImportedDataSegment: Sendable, Equatable, Codable {
-        public enum DataSegmentRootKind: Sendable, Equatable {
+public struct WorkItem: Sendable, Equatable, Codable, Hashable {
+    public struct ImportedDataSegment: Sendable, Equatable, Codable, Hashable {
+        public enum DataSegmentRootKind: Sendable, Equatable, Hashable {
             case segmentRoot(Data32)
             case workPackageHash(Data32)
         }

Blockchain/Sources/Blockchain/Types/WorkPackage.swift

@@ -3,7 +3,7 @@ import Foundation
 import Utils
 
 // P
-public struct WorkPackage: Comparable, Sendable, Equatable, Codable {
+public struct WorkPackage: Comparable, Sendable, Equatable, Codable, Hashable {
     // j
     public var authorizationToken: Data
 

Blockchain/Sources/Blockchain/Validator/GuaranteeingService.swift

@@ -88,7 +88,7 @@ public final class GuaranteeingService: ServiceBase2, @unchecked Sendable {
     }
 
     private func on(workPackagesReceived event: RuntimeEvents.WorkPackagesReceived) async throws {
-        try await refine(package: event.item)
+        try await refine(package: event.workPackageRef)
     }
 
     private func refine(package: WorkPackageRef) async throws {

Node/Sources/Node/NetworkingProtocol/CommonEphemeral/CERequest.swift

@@ -7,6 +7,7 @@ public enum CERequest: Sendable, Equatable, Hashable {
     case blockRequest(BlockRequest)
     case safroleTicket1(SafroleTicketMessage)
     case safroleTicket2(SafroleTicketMessage)
+    case workPackageSubmission(WorkPackageMessage)
 }
 
 extension CERequest: RequestProtocol {
@@ -20,6 +21,8 @@ extension CERequest: RequestProtocol {
             try JamEncoder.encode(message)
         case let .safroleTicket2(message):
             try JamEncoder.encode(message)
+        case let .workPackageSubmission(message):
+            try JamEncoder.encode(message)
         }
     }
 
@@ -31,6 +34,8 @@ extension CERequest: RequestProtocol {
             .safroleTicket1
         case .safroleTicket2:
             .safroleTicket2
+        case .workPackageSubmission:
+            .workPackageSubmission
         }
     }
 
@@ -42,6 +47,8 @@ extension CERequest: RequestProtocol {
             SafroleTicketMessage.self
         case .safroleTicket2:
             SafroleTicketMessage.self
+        case .workPackageSubmission:
+            WorkPackageMessage.self
         default:
             fatalError("unimplemented")
         }
@@ -64,6 +71,9 @@ extension CERequest: RequestProtocol {
                 return nil
             }
             return .safroleTicket2(message)
+        case .workPackageSubmission:
+            guard let message = data as? WorkPackageMessage else { return nil }
+            return .workPackageSubmission(message)
         default:
             fatalError("unimplemented")
         }

Node/Sources/Node/NetworkingProtocol/CommonEphemeral/WorkPackage.swift

@@ -0,0 +1,20 @@
+import Blockchain
+import Codec
+import Foundation
+
+public struct WorkPackageMessage: Codable, Sendable, Equatable, Hashable {
+    /// The core index associated with the work-package.
+    public var coreIndex: CoreIndex
+
+    /// The work-package data.
+    public var workPackage: WorkPackage
+
+    /// The extrinsic data referenced by the work-package.
+    public var extrinsics: [Data]
+
+    public init(coreIndex: CoreIndex, workPackage: WorkPackage, extrinsics: [Data]) {
+        self.coreIndex = coreIndex
+        self.workPackage = workPackage
+        self.extrinsics = extrinsics
+    }
+}

Node/Sources/Node/NetworkingProtocol/NetworkManager.swift

@@ -226,6 +226,17 @@ struct HandlerImpl: NetworkProtocolHandler {
                 ]
             ))
             return []
+        case let .workPackageSubmission(message):
+            blockchain
+                .publish(
+                    event: RuntimeEvents
+                        .WorkPackagesReceived(
+                            coreIndex: message.coreIndex,
+                            workPackageRef: message.workPackage.asRef(),
+                            extrinsics: message.extrinsics
+                        )
+                )
+            return []
         }
     }
 

Node/Tests/NodeTests/NetworkManagerTests.swift

@@ -74,6 +74,31 @@ struct NetworkManagerTests {
         )
     }
 
+    @Test
+    func testWorkPackagesReceived() async throws {
+        // Create dummy work packages
+        let workPackage = WorkPackage.dummy(config: services.config).asRef()
+
+        // Publish WorkPackagesReceived event
+        await services.blockchain
+            .publish(event: RuntimeEvents.WorkPackagesReceived(coreIndex: 0, workPackageRef: workPackage, extrinsics: []))
+
+        // Wait for event processing
+        await storeMiddleware.wait()
+
+        // Verify network calls
+        #expect(
+            network.contain(calls: [
+                .init(function: "connect", parameters: ["address": devPeers.first!, "role": PeerRole.validator]),
+                .init(function: "sendToPeer", parameters: [
+                    "message": CERequest.workPackageSubmission(
+                        WorkPackageMessage(coreIndex: 0, workPackage: workPackage.value, extrinsics: [])
+                    ),
+                ]),
+            ])
+        )
+    }
+
     @Test
     func testBlockBroadcast() async throws {
         // Import a block

Utils/Sources/Utils/SaturatingNumber.swift

@@ -1,7 +1,7 @@
 import Codec
 import Numerics
 
-public struct SaturatingNumber<T: FixedWidthInteger & Sendable>: Sendable {
+public struct SaturatingNumber<T: FixedWidthInteger & Sendable>: Sendable, Hashable {
     public private(set) var value: T
 
     public static var max: SaturatingNumber {