PE-1136 Refactor ArConnect Signing

lib/src/models/data_item.dart

@@ -2,16 +2,23 @@ import 'dart:convert';
 import 'dart:typed_data';
 
 import 'package:arweave/src/utils/bundle_tag_parser.dart';
+import 'package:json_annotation/json_annotation.dart';
 
 import '../crypto/crypto.dart';
 import '../utils.dart';
 import 'models.dart';
 
+part 'data_item.g.dart';
+
 final MIN_BINARY_SIZE = 1044;
 
 /// ANS-104 [DataItem]
 /// Spec: https://github.com/joshbenaron/arweave-standards/blob/ans104/ans/ANS-104.md
+@JsonSerializable(explicitToJson: true)
 class DataItem implements TransactionBase {
+  @JsonKey(defaultValue: 1)
+  final int format = 1;
+
   @override
   String get id => _id;
   late String _id;
@@ -34,9 +41,15 @@ class DataItem implements TransactionBase {
   @override
   late Uint8List data;
 
+  @JsonKey(name: 'data_size')
+  String get dataSize => _dataSize;
+  String _dataSize = '0';
+
   @override
   String get signature => _signature;
   late String _signature;
+
+  @JsonKey(ignore: true)
   late ByteBuffer binary;
 
   /// This constructor is reserved for JSON serialisation.
@@ -49,13 +62,18 @@ class DataItem implements TransactionBase {
     List<Tag>? tags,
     String? data,
     Uint8List? dataBytes,
+    String? dataSize,
   })  : target = target ?? '',
         nonce = nonce ?? '',
         _owner = owner ?? '',
         data = data != null
             ? decodeBase64ToBytes(data)
             : (dataBytes ?? Uint8List(0)),
-        _tags = tags ?? [];
+        _tags = tags ?? [] {
+    if (dataSize != null) {
+      _dataSize = dataSize;
+    }
+  }
 
   /// Constructs a [DataItem] with the specified JSON data and appropriate Content-Type tag.
   factory DataItem.withJsonData({
@@ -87,6 +105,7 @@ class DataItem implements TransactionBase {
         nonce: nonce,
         tags: tags,
         dataBytes: data,
+        dataSize: data.lengthInBytes.toString(),
       );
 
   @override
@@ -106,7 +125,7 @@ class DataItem implements TransactionBase {
   Future<Uint8List> getSignatureData() => deepHash(
         [
           utf8.encode('dataitem'),
-          utf8.encode('1'), //Transaction format
+          utf8.encode(format.toString()), //Transaction format
           utf8.encode('1'), //Signature type
           decodeBase64ToBytes(owner),
           decodeBase64ToBytes(target),
@@ -119,8 +138,7 @@ class DataItem implements TransactionBase {
   /// Signs the [DataItem] using the specified wallet and sets the `id` and `signature` appropriately.
   @override
   Future<Uint8List> sign(Wallet wallet) async {
-    final signatureData = await getSignatureData();
-    final rawSignature = await wallet.sign(signatureData);
+    final rawSignature = await wallet.sign(this);
 
     _signature = encodeBytesToBase64(rawSignature);
 
@@ -287,4 +305,18 @@ class DataItem implements TransactionBase {
     bytesBuilder.add(data);
     return bytesBuilder;
   }
+
+  /// Encodes the [DataItem] as JSON with the `data` as the original unencoded [Uint8List].
+  @override
+  Map<String, dynamic> toJson() => _$DataItemToJson(this);
+
+  @override
+  Map<String, dynamic> toUnsignedJson() => <String, dynamic>{
+        'format': format,
+        'owner': owner,
+        'tags': tags.map((e) => e.toJson()).toList(),
+        'target': target,
+        'data': data,
+        'data_size': dataSize,
+      };
 }

lib/src/models/tag.dart

@@ -2,7 +2,7 @@ import 'package:json_annotation/json_annotation.dart';
 
 part 'tag.g.dart';
 
-@JsonSerializable()
+@JsonSerializable(explicitToJson: true)
 class Tag {
   /// The tag's name encoded as Base64.
   final String name;

lib/src/models/transaction-base.dart

@@ -27,4 +27,7 @@ abstract class TransactionBase {
   Future<void> sign(Wallet wallet);
 
   Future<bool> verify();
+
+  Map<String, dynamic> toJson();
+  Map<String, dynamic> toUnsignedJson();
 }

lib/src/models/transaction.dart

@@ -12,7 +12,7 @@ part 'transaction.g.dart';
 String _bigIntToString(BigInt v) => v.toString();
 BigInt _stringToBigInt(String v) => BigInt.parse(v);
 
-@JsonSerializable()
+@JsonSerializable(explicitToJson: true)
 class Transaction implements TransactionBase {
   @JsonKey(defaultValue: 1)
   final int format;
@@ -277,8 +277,7 @@ class Transaction implements TransactionBase {
 
   @override
   Future<void> sign(Wallet wallet) async {
-    final signatureData = await getSignatureData();
-    final rawSignature = await wallet.sign(signatureData);
+    final rawSignature = await wallet.sign(this);
 
     _signature = encodeBytesToBase64(rawSignature);
 
@@ -312,5 +311,20 @@ class Transaction implements TransactionBase {
       _$TransactionFromJson(json);
 
   /// Encodes the [Transaction] as JSON with the `data` as the original unencoded [Uint8List].
+  @override
   Map<String, dynamic> toJson() => _$TransactionToJson(this);
+
+  @override
+  Map<String, dynamic> toUnsignedJson() => <String, dynamic>{
+        'format': format,
+        'last_tx': lastTx,
+        'owner': owner,
+        'tags': tags.map((e) => e.toJson()).toList(),
+        'target': target,
+        'quantity': _bigIntToString(quantity),
+        'data': data,
+        'data_size': dataSize,
+        'data_root': dataRoot,
+        'reward': _bigIntToString(reward),
+      };
 }

lib/src/models/wallet.dart

@@ -3,6 +3,7 @@ import 'dart:core';
 import 'dart:math';
 import 'dart:typed_data';
 
+import 'package:arweave/arweave.dart';
 import 'package:cryptography/cryptography.dart';
 import 'package:jwk/jwk.dart';
 import 'package:pointycastle/export.dart';
@@ -54,7 +55,10 @@ class Wallet {
       await _keyPair!.extractPublicKey().then((res) => res.n));
   Future<String> getAddress() async => ownerToAddress(await getOwner());
 
-  Future<Uint8List> sign(Uint8List message) async =>
+  Future<Uint8List> sign(TransactionBase transaction) async => rsaPssSign(
+      message: await transaction.getSignatureData(), keyPair: _keyPair!);
+
+  Future<Uint8List> signMessage(Uint8List message) async =>
       rsaPssSign(message: message, keyPair: _keyPair!);
 
   factory Wallet.fromJwk(Map<String, dynamic> jwk) {

pubspec.yaml

@@ -7,7 +7,7 @@ environment:
 dependencies:
   cryptography: ^2.0.1
   http: ^0.13.3
-  json_annotation: ^4.0.1
+  json_annotation: 4.4.0
   jwk: ^0.1.0
   meta: ^1.7.0
   pointycastle: ^3.1.1

test/wallets_test.dart

@@ -3,7 +3,7 @@ import 'dart:io';
 import 'dart:typed_data';
 
 import 'package:arweave/arweave.dart';
-import 'package:arweave/utils.dart' as utils;
+import 'package:arweave/utils.dart';
 import 'package:test/test.dart';
 
 import 'utils.dart';
@@ -62,9 +62,9 @@ void main() {
       final wallet = await getTestWallet();
       final message = utf8.encode('<test message>');
 
-      final signature = await wallet.sign(message as Uint8List);
+      final signature = await wallet.signMessage(message as Uint8List);
       expect(
-        utils.encodeBytesToBase64(signature),
+        encodeBytesToBase64(signature),
         startsWith('II5LxGnPt4WTSz9P__wMAdjzXWlZE-wGbKU7wm4DbGuPXB5Vifs'),
       );
     }, onPlatform: {