feat: VecU8

Author: bigearsenal

Sources/SolanaSwift/Models/BufferLayout/VecU8.swift

@@ -1,15 +1,21 @@
 import Foundation
 
-struct VecU8: BorshCodable, Codable, Equatable, Hashable {
-    let length: UInt16
-    let data: Data
-    init(from reader: inout BinaryReader) throws {
-        length = try UInt16(from: &reader)
+public struct VecU8<T: FixedWidthInteger & Codable>: BorshCodable, Codable, Equatable, Hashable {
+    public let length: T
+    public let data: Data
+
+    public init(from reader: inout BinaryReader) throws {
+        length = try T(from: &reader)
         data = try Data(reader.read(count: Int(length)))
     }
 
-    func serialize(to data: inout Data) throws {
-        try length.serialize(to: &data)
-        try data.serialize(to: &data)
+    public init(length: T, data: Data) {
+        self.length = length
+        self.data = data
+    }
+
+    public func serialize(to writer: inout Data) throws {
+        try length.serialize(to: &writer)
+        try data.serialize(to: &writer)
     }
 }

Sources/SolanaSwift/Programs/TokenPrograms/Token2022Program/ExtensionReader/Token2022ExtensionState.swift

@@ -4,4 +4,4 @@ public protocol Token2022ExtensionState: BorshCodable, Codable, Equatable, Hasha
     var length: UInt16 { get }
 }
 
-extension VecU8: Token2022ExtensionState {}
+extension VecU8<UInt16>: Token2022ExtensionState {}

Sources/SolanaSwift/Programs/TokenPrograms/Token2022Program/Token2022MintState.swift

@@ -12,7 +12,7 @@ public struct Token2022MintState: TokenMintState {
     public var extensions: [AnyToken2022ExtensionState]
 
     public func getParsedExtension<T: Token2022ExtensionState>(ofType _: T.Type) -> T? {
-        assert(T.self != VecU8.self)
+        assert(T.self != VecU8<UInt16>.self)
         return extensions.first(where: { $0.state is T })?.state as? T
     }
 }

Tests/SolanaSwiftUnitTests/Other/BufferLayout/BufferLayoutDecodingTests.swift

@@ -24,6 +24,23 @@ class BufferLayoutDecodingTests: XCTestCase {
         XCTAssertEqual(try Data(from: &binaryReader).bytes, expectedBytes)
     }
 
+    // MARK: - VecU8
+
+    func testDecodingVecU8() throws {
+        let string = "GQCn7dKsGcVAJhtFRDDBcRgD8i3I/WDk4Z2y"
+        let data = Data(base64Encoded: string)!
+        var binaryReader = BinaryReader(bytes: data.bytes)
+        let vecU8 = try VecU8<UInt16>(from: &binaryReader)
+
+        XCTAssertEqual(vecU8.length, 25)
+        XCTAssertEqual(vecU8.data.bytes, [
+            167, 237, 210, 172, 25, 197,
+            64, 38, 27, 69, 68, 48, 193,
+            113, 24, 3, 242, 45, 200, 253,
+            96, 228, 225, 157, 178,
+        ])
+    }
+
     // MARK: - Mint
 
     func testDecodingMint() throws {

Tests/SolanaSwiftUnitTests/Other/BufferLayout/BufferLayoutEncodingTests.swift

@@ -24,6 +24,23 @@ final class BufferLayoutEncodingTests: XCTestCase {
         )
     }
 
+    // MARK: - VecU8
+
+    func testEncodingVecU8() throws {
+        let length: UInt16 = 25
+        let data = Data([
+            167, 237, 210, 172, 25, 197,
+            64, 38, 27, 69, 68, 48, 193,
+            113, 24, 3, 242, 45, 200, 253,
+            96, 228, 225, 157, 178,
+        ])
+
+        var result = Data()
+        try VecU8(length: length, data: data).serialize(to: &result)
+
+        XCTAssertEqual(result.base64EncodedString(), "GQCn7dKsGcVAJhtFRDDBcRgD8i3I/WDk4Z2y")
+    }
+
     // MARK: - Account info
 
     func testEncodingAccountInfo() throws {