refactor(Extensions): add ToBytes shortcut method

This commit is contained in:
tekh 2025-09-25 12:47:03 +02:00
parent b0b734ecfb
commit 7964cc44fb
2 changed files with 11 additions and 52 deletions

View File

@ -1,54 +1,17 @@
using iText.Kernel.Pdf;
using iText.Kernel.Pdf.Canvas;
#if NETFRAMEWORK
#if NETFRAMEWORK
using System.IO;
using System;
#endif
namespace EnvelopeGenerator.PdfEditor
#if NET
;
#elif NETFRAMEWORK
{
#endif
public static class Extensions
{
#region Edit PDF document
public static TStream Edit<TStream>(this TStream inputStream, Action<PdfDocument> edit)
where TStream : Stream, new()
{
using (var outputStream = new TStream())
{
using (var pdfDoc = new PdfDocument(new PdfReader(inputStream), new PdfWriter(outputStream)))
{
edit(pdfDoc);
}
return outputStream;
/// <summary>
/// Writes the stream contents to a byte array, regardless of the System.IO.MemoryStream.Position property.
/// </summary>
/// <typeparam name="TPdf"></typeparam>
/// <param name="pdf"></param>
/// <returns>A new byte array</returns>
public static byte[] ToBytes<TPdf>(this TPdf pdf) where TPdf : Pdf<MemoryStream, MemoryStream> => pdf.OutputStream.ToArray();
}
}
public static byte[] Edit(this byte[] pdfBytes, Action<PdfDocument> edit)
{
using (var inputStream = new MemoryStream(pdfBytes))
{
return inputStream.Edit(edit).ToArray();
}
}
#endregion
public static byte[] Design(this byte[] pdfBytes, params (int pageIndex, Action<PdfCanvas> design)[] pageDesign)
=> pdfBytes.Edit(doc =>
{
foreach((int pageIndex, Action<PdfCanvas> design) in pageDesign)
{
var page = doc.GetPage(pageIndex);
var canvas = new PdfCanvas(page);
design(canvas);
}
});
}
#if NETFRAMEWORK
}
#endif

View File

@ -224,7 +224,7 @@ public class EnvelopeController : ViewControllerBase
{
if (er.Envelope!.Documents?.FirstOrDefault() is DocumentDto doc && doc.ByteData is not null)
{
var edit = Pdf.FromMemory(doc.ByteData).Draw(1, canvas =>
using var pdf = Pdf.FromMemory(doc.ByteData).Design(1, canvas =>
{
canvas.SetStrokeColor(ColorConstants.RED);
canvas.SetFillColor(ColorConstants.CYAN);
@ -234,11 +234,7 @@ public class EnvelopeController : ViewControllerBase
canvas.FillStroke();
});
edit._doc.Close();
var foo = Convert.ToBase64String(edit.OutputStream.ToArray());
doc.ByteData = edit.OutputStream.ToArray();
doc.ByteData = pdf.ToBytes();
ViewData["DocumentBytes"] = doc.ByteData;
}