| | | 1 | | using System.Runtime.Serialization; |
| | | 2 | | using NLightning.Domain.Protocol.Interfaces; |
| | | 3 | | using NLightning.Domain.Serialization.Interfaces; |
| | | 4 | | |
| | | 5 | | namespace NLightning.Infrastructure.Serialization.Tlv; |
| | | 6 | | |
| | | 7 | | using Domain.Protocol.Models; |
| | | 8 | | using Domain.Protocol.Tlv; |
| | | 9 | | using Interfaces; |
| | | 10 | | |
| | | 11 | | public class TlvStreamSerializer : ITlvStreamSerializer |
| | | 12 | | { |
| | | 13 | | private readonly ITlvSerializer _tlvSerializer; |
| | | 14 | | private readonly ITlvConverterFactory _tlvConverterFactory; |
| | | 15 | | |
| | 16 | 16 | | public TlvStreamSerializer(ITlvConverterFactory tlvConverterFactory, ITlvSerializer tlvSerializer) |
| | | 17 | | { |
| | 16 | 18 | | _tlvConverterFactory = tlvConverterFactory; |
| | 16 | 19 | | _tlvSerializer = tlvSerializer; |
| | 16 | 20 | | } |
| | | 21 | | |
| | | 22 | | public async Task SerializeAsync(TlvStream? tlvStream, Stream stream) |
| | | 23 | | { |
| | 76 | 24 | | if (tlvStream is null) |
| | 36 | 25 | | return; |
| | | 26 | | |
| | 216 | 27 | | foreach (var tlv in tlvStream.GetTlvs()) |
| | | 28 | | { |
| | 68 | 29 | | var baseTlv = tlv switch |
| | 68 | 30 | | { |
| | 4 | 31 | | BlindedPathTlv blindedPathTlv => _tlvConverterFactory |
| | 4 | 32 | | .GetConverter<BlindedPathTlv>()?.ConvertToBase(blindedPathTlv), |
| | 8 | 33 | | ChannelTypeTlv channelTypeTlv => _tlvConverterFactory |
| | 8 | 34 | | .GetConverter<ChannelTypeTlv>()?.ConvertToBase(channelTypeTlv), |
| | 4 | 35 | | FeeRangeTlv feeRangeTlv => _tlvConverterFactory |
| | 4 | 36 | | .GetConverter<FeeRangeTlv>()?.ConvertToBase(feeRangeTlv), |
| | 12 | 37 | | FundingOutputContributionTlv fundingOutputContributionTlv => _tlvConverterFactory |
| | 12 | 38 | | .GetConverter< |
| | 12 | 39 | | FundingOutputContributionTlv>() |
| | 12 | 40 | | ?.ConvertToBase( |
| | 12 | 41 | | fundingOutputContributionTlv), |
| | 4 | 42 | | NetworksTlv networksTlv => _tlvConverterFactory |
| | 4 | 43 | | .GetConverter<NetworksTlv>()?.ConvertToBase(networksTlv), |
| | 4 | 44 | | NextFundingTlv nextFundingTlv => _tlvConverterFactory |
| | 4 | 45 | | .GetConverter<NextFundingTlv>()?.ConvertToBase(nextFundingTlv), |
| | 20 | 46 | | RequireConfirmedInputsTlv requireConfirmedInputsTlv => _tlvConverterFactory |
| | 20 | 47 | | .GetConverter<RequireConfirmedInputsTlv>() |
| | 20 | 48 | | ?.ConvertToBase(requireConfirmedInputsTlv), |
| | 4 | 49 | | ShortChannelIdTlv shortChannelIdTlv => _tlvConverterFactory |
| | 4 | 50 | | .GetConverter<ShortChannelIdTlv>() |
| | 4 | 51 | | ?.ConvertToBase(shortChannelIdTlv), |
| | 8 | 52 | | UpfrontShutdownScriptTlv upfrontShutdownScriptTlv => _tlvConverterFactory |
| | 8 | 53 | | .GetConverter<UpfrontShutdownScriptTlv>() |
| | 8 | 54 | | ?.ConvertToBase(upfrontShutdownScriptTlv), |
| | 0 | 55 | | _ => null |
| | 68 | 56 | | } ?? throw new SerializationException($"No converter found for tlv type {tlv.GetType().Name}"); |
| | 68 | 57 | | await _tlvSerializer.SerializeAsync(baseTlv, stream); |
| | | 58 | | } |
| | 76 | 59 | | } |
| | | 60 | | |
| | | 61 | | public async Task<TlvStream?> DeserializeAsync(Stream stream) |
| | | 62 | | { |
| | 84 | 63 | | if (stream.Position == stream.Length) |
| | 4 | 64 | | return null; |
| | | 65 | | |
| | | 66 | | try |
| | | 67 | | { |
| | 80 | 68 | | var tlvStream = new TlvStream(); |
| | | 69 | | |
| | 148 | 70 | | while (stream.Position != stream.Length) |
| | | 71 | | { |
| | 108 | 72 | | var tlv = await _tlvSerializer.DeserializeAsync(stream); |
| | 68 | 73 | | if (tlv is not null) |
| | 68 | 74 | | tlvStream.Add(tlv); |
| | | 75 | | } |
| | | 76 | | |
| | 40 | 77 | | return tlvStream.Any() ? tlvStream : null; |
| | | 78 | | } |
| | 40 | 79 | | catch (Exception e) |
| | | 80 | | { |
| | 40 | 81 | | throw new SerializationException("Error deserializing TLVStream", e); |
| | | 82 | | } |
| | 44 | 83 | | } |
| | | 84 | | } |