626 lines
20 KiB
C#
626 lines
20 KiB
C#
using QWERTYkez.WordProcessor.Builders;
|
||
|
||
namespace QWERTYkez.WordProcessor;
|
||
|
||
/// <summary>
|
||
/// Предоставляет потокобезопасный процессор для чтения и записи документов Word (DOCX) формата.
|
||
/// <para>Наследует от <see cref="WordReader"/> и добавляет операции изменения документа.</para>
|
||
/// </summary>
|
||
internal sealed class WordWriter : WordReader, IWordWriter
|
||
{
|
||
private bool _isModified = false;
|
||
|
||
internal WordWriter() { }
|
||
|
||
#region Factory Methods
|
||
|
||
internal static WordWriter? CreateFromData(ReadOnlyMemory<byte> data, string destinationPath)
|
||
{
|
||
if (data.IsEmpty || string.IsNullOrEmpty(destinationPath))
|
||
return null;
|
||
|
||
var ms = new MemoryStream();
|
||
try
|
||
{
|
||
// Копируем данные в MemoryStream
|
||
ms.Write(data.ToArray(), 0, data.Length);
|
||
ms.Position = 0;
|
||
|
||
var doc = WordprocessingDocument.Open(ms, true, new OpenSettings { AutoSave = false });
|
||
if (doc.MainDocumentPart?.Document?.Body is { } body)
|
||
{
|
||
return new WordWriter
|
||
{
|
||
_ms = ms,
|
||
_doc = doc,
|
||
_body = body,
|
||
FilePath = destinationPath,
|
||
_originalSourcePath = null // нет исходного файла
|
||
};
|
||
}
|
||
|
||
doc.Dispose();
|
||
}
|
||
catch { }
|
||
|
||
ms?.Dispose();
|
||
return null;
|
||
}
|
||
|
||
internal static new WordWriter? CreateFromData(ReadOnlyMemory<byte> data)
|
||
{
|
||
if (data.IsEmpty)
|
||
return null;
|
||
|
||
var ms = new MemoryStream();
|
||
try
|
||
{
|
||
ms.Write(data.ToArray(), 0, data.Length);
|
||
ms.Position = 0;
|
||
|
||
var doc = WordprocessingDocument.Open(ms, true, new OpenSettings { AutoSave = false });
|
||
if (doc.MainDocumentPart?.Document?.Body is { } body)
|
||
{
|
||
return new WordWriter
|
||
{
|
||
_ms = ms,
|
||
_doc = doc,
|
||
_body = body,
|
||
FilePath = null, // нет привязки к файлу
|
||
_originalSourcePath = null
|
||
};
|
||
}
|
||
|
||
doc.Dispose();
|
||
}
|
||
catch { }
|
||
|
||
ms.Dispose();
|
||
return null;
|
||
}
|
||
|
||
internal static WordWriter? CreateInternal(FileInfo sourceFile, string? destinationPath = null!)
|
||
{
|
||
if (sourceFile is null || !sourceFile.Exists) return null;
|
||
|
||
var ms = new MemoryStream();
|
||
try
|
||
{
|
||
using (var file = new FileStream(sourceFile.FullName,
|
||
FileMode.Open, FileAccess.Read, FileShare.ReadWrite | FileShare.Delete))
|
||
{
|
||
file.CopyTo(ms);
|
||
}
|
||
ms.Position = 0;
|
||
|
||
var doc = WordprocessingDocument.Open(ms, isEditable: true,
|
||
new OpenSettings { AutoSave = false });
|
||
|
||
if (doc.MainDocumentPart?.Document?.Body is { } body)
|
||
{
|
||
var processor = new WordWriter
|
||
{
|
||
_ms = ms,
|
||
_doc = doc,
|
||
_body = body,
|
||
_originalSourcePath = sourceFile.FullName,
|
||
FilePath = destinationPath ?? sourceFile.FullName
|
||
};
|
||
|
||
return processor;
|
||
}
|
||
|
||
doc?.Dispose();
|
||
}
|
||
catch { }
|
||
|
||
ms?.Dispose();
|
||
return null;
|
||
}
|
||
|
||
/// <summary>
|
||
/// Создаёт новый пустой документ Word.
|
||
/// </summary>
|
||
/// <param name="destinationPath">Путь, по которому будет сохранён документ (необязательный).</param>
|
||
/// <returns>Экземпляр <see cref="WordWriter"/> для редактирования нового документа.</returns>
|
||
internal static WordWriter CreateNew(string? destinationPath = null)
|
||
{
|
||
var ms = new MemoryStream();
|
||
try
|
||
{
|
||
// Создаём документ, НЕ используем using
|
||
var doc = WordprocessingDocument.Create(ms, WordprocessingDocumentType.Document);
|
||
var mainPart = doc.AddMainDocumentPart();
|
||
mainPart.Document = new Document();
|
||
var body = new Body();
|
||
mainPart.Document.AppendChild(body);
|
||
|
||
var writer = new WordWriter
|
||
{
|
||
_ms = ms,
|
||
_doc = doc,
|
||
_body = body,
|
||
FilePath = destinationPath,
|
||
_originalSourcePath = null
|
||
};
|
||
return writer;
|
||
}
|
||
catch
|
||
{
|
||
ms.Dispose();
|
||
throw;
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region Properties
|
||
|
||
/// <summary>
|
||
/// Получает значение, указывающее, будет ли документ сохранен с перезаписью исходного файла.
|
||
/// </summary>
|
||
public bool WillOverwriteSource => FilePath == _originalSourcePath;
|
||
|
||
#endregion
|
||
|
||
#region Replace text
|
||
|
||
public void ReplaceString(string oldValue, params string[] newValues) => ReplaceString(oldValue, (IEnumerable<string>)newValues);
|
||
public void ReplaceString(string oldValue, IEnumerable<string> newValues)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (string.IsNullOrEmpty(oldValue))
|
||
throw new ArgumentException("Old value cannot be null or empty", nameof(oldValue));
|
||
|
||
if (newValues is null) return;
|
||
|
||
#if DEBUG
|
||
Debug.WriteLine($"[DEBUG] [WordProcessor.Replace] START: '{oldValue}' -> [{string.Join(", ", newValues)}] (comparison: {StringComparison.OrdinalIgnoreCase})");
|
||
#endif
|
||
|
||
lock (_syncLock)
|
||
{
|
||
// 1. Основной текст
|
||
_body.Replace(oldValue, newValues, StringComparison.OrdinalIgnoreCase);
|
||
|
||
// 2. Таблицы в основном тексте
|
||
TableTextProcessor.ReplaceInTables(_body, oldValue, newValues, StringComparison.OrdinalIgnoreCase);
|
||
|
||
// 3. Колонтитулы
|
||
HeaderFooterProcessor.ReplaceInHeadersFooters(_doc, oldValue, newValues, StringComparison.OrdinalIgnoreCase);
|
||
|
||
_isModified = true;
|
||
}
|
||
}
|
||
|
||
public void ReplaceString(IDictionary<string, string> replacements) =>
|
||
/* */ReplaceString((IEnumerable<KeyValuePair<string, string>>)replacements);
|
||
public void ReplaceString(IEnumerable<KeyValuePair<string, string>> replacements)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (replacements is null) return;
|
||
|
||
lock (_syncLock)
|
||
{
|
||
// 1. Основной текст
|
||
_body.Replace(replacements, StringComparison.OrdinalIgnoreCase);
|
||
|
||
// 2. Таблицы в основном тексте
|
||
foreach (var kvp in replacements)
|
||
{
|
||
if (!string.IsNullOrEmpty(kvp.Key))
|
||
{
|
||
TableTextProcessor.ReplaceInTables(_body, kvp.Key, [kvp.Value ?? string.Empty], StringComparison.OrdinalIgnoreCase);
|
||
}
|
||
}
|
||
|
||
HeaderFooterProcessor.ReplaceInHeadersFooters(_doc, replacements, StringComparison.OrdinalIgnoreCase);
|
||
|
||
_isModified = true;
|
||
}
|
||
}
|
||
|
||
public void ReplaceString(IDictionary<string, IEnumerable<string>> replacements) =>
|
||
/* */ReplaceString((IEnumerable<KeyValuePair<string, IEnumerable<string>>>)replacements);
|
||
public void ReplaceString(IEnumerable<KeyValuePair<string, IEnumerable<string>>> replacements)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (replacements is null) return;
|
||
|
||
lock (_syncLock)
|
||
{
|
||
// 1. Основной текст
|
||
_body.Replace(replacements, StringComparison.OrdinalIgnoreCase);
|
||
|
||
// 2. Таблицы в основном тексте
|
||
TableTextProcessor.ReplaceInTables(_body, replacements, StringComparison.OrdinalIgnoreCase);
|
||
|
||
// 3. Колонтитулы
|
||
HeaderFooterProcessor.ReplaceInHeadersFooters(_doc, replacements, StringComparison.OrdinalIgnoreCase);
|
||
|
||
_isModified = true;
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region Replace ReplaceItem
|
||
public void ReplaceItem(string oldValue, params ReplaceItem[] newValues) => ReplaceItem(oldValue, (IEnumerable<ReplaceItem>)newValues);
|
||
public void ReplaceItem(string oldValue, IEnumerable<ReplaceItem> newValues)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (string.IsNullOrEmpty(oldValue))
|
||
throw new ArgumentException("Old value cannot be null or empty", nameof(oldValue));
|
||
|
||
if (newValues is null) return;
|
||
|
||
#if DEBUG
|
||
Debug.WriteLine($"[DEBUG] [WordProcessor.Replace] START: '{oldValue}' -> [{string.Join(", ", newValues)}] (comparison: {StringComparison.OrdinalIgnoreCase})");
|
||
#endif
|
||
|
||
lock (_syncLock)
|
||
{
|
||
// 1. Основной текст
|
||
_body.Replace(oldValue, newValues, StringComparison.OrdinalIgnoreCase);
|
||
|
||
var texts = newValues.Select(val => val.Text).ToArray();
|
||
|
||
// 2. Таблицы в основном тексте
|
||
TableTextProcessor.ReplaceInTables(_body, oldValue, texts, StringComparison.OrdinalIgnoreCase);
|
||
|
||
// 3. Колонтитулы
|
||
HeaderFooterProcessor.ReplaceInHeadersFooters(_doc, oldValue, texts, StringComparison.OrdinalIgnoreCase);
|
||
|
||
_isModified = true;
|
||
}
|
||
}
|
||
|
||
public void ReplaceItem(IDictionary<string, ReplaceItem> replacements) =>
|
||
/* */ReplaceItem((IEnumerable<KeyValuePair<string, ReplaceItem>>)replacements);
|
||
public void ReplaceItem(IEnumerable<KeyValuePair<string, ReplaceItem>> replacements)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (replacements is null) return;
|
||
|
||
lock (_syncLock)
|
||
{
|
||
// 1. Основной текст
|
||
_body.Replace(replacements, StringComparison.OrdinalIgnoreCase);
|
||
|
||
|
||
var texts = replacements.ToDictionary(val => val.Key, val => val.Value.Text);
|
||
|
||
// 2. Таблицы в основном тексте
|
||
foreach (var kvp in texts)
|
||
{
|
||
if (!string.IsNullOrEmpty(kvp.Key))
|
||
{
|
||
TableTextProcessor.ReplaceInTables(_body, kvp.Key, kvp.Value, StringComparison.OrdinalIgnoreCase);
|
||
}
|
||
}
|
||
|
||
// 3. Колонтитулы
|
||
HeaderFooterProcessor.ReplaceInHeadersFooters(_doc, texts, StringComparison.OrdinalIgnoreCase);
|
||
|
||
_isModified = true;
|
||
}
|
||
}
|
||
|
||
public void ReplaceItem(IDictionary<string, IEnumerable<ReplaceItem>> replacements) =>
|
||
/* */ReplaceItem((IEnumerable<KeyValuePair<string, IEnumerable<ReplaceItem>>>)replacements);
|
||
public void ReplaceItem(IEnumerable<KeyValuePair<string, IEnumerable<ReplaceItem>>> replacements)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (replacements is null) return;
|
||
|
||
lock (_syncLock)
|
||
{
|
||
// 1. Основной текст
|
||
_body.Replace(replacements, StringComparison.OrdinalIgnoreCase);
|
||
|
||
|
||
var texts = replacements.ToDictionary(val => val.Key, val => val.Value.Select(val => val.Text));
|
||
|
||
// 2. Таблицы в основном тексте
|
||
TableTextProcessor.ReplaceInTables(_body, texts, StringComparison.OrdinalIgnoreCase);
|
||
|
||
// 3. Колонтитулы
|
||
HeaderFooterProcessor.ReplaceInHeadersFooters(_doc, texts, StringComparison.OrdinalIgnoreCase);
|
||
|
||
_isModified = true;
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
|
||
/// <summary>
|
||
/// Добавляет новый параграф с указанным текстом в конец документа.
|
||
/// </summary>
|
||
public void AddParagraph(string text, bool preserveFormatting = true)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (_body is null)
|
||
return;
|
||
|
||
lock (_syncLock)
|
||
{
|
||
var paragraph = new Paragraph();
|
||
var run = new Run();
|
||
|
||
if (preserveFormatting && _body.Elements<Paragraph>().FirstOrDefault() is { } firstPara)
|
||
{
|
||
if (firstPara.ParagraphProperties is not null)
|
||
{
|
||
paragraph.ParagraphProperties = firstPara.ParagraphProperties.CloneNode(true) as ParagraphProperties;
|
||
}
|
||
|
||
if (firstPara.Elements<Run>().FirstOrDefault()?.RunProperties is not null)
|
||
{
|
||
run.RunProperties = firstPara.Elements<Run>().First().RunProperties?.CloneNode(true) as RunProperties;
|
||
}
|
||
}
|
||
|
||
string processedText = text.Replace(' ', '\u00A0');
|
||
run.AppendChild(new Text(processedText));
|
||
paragraph.AppendChild(run);
|
||
_body.AppendChild(paragraph);
|
||
_isModified = true;
|
||
}
|
||
}
|
||
|
||
#region Replace to table
|
||
|
||
/// <summary>
|
||
/// Заменяет параграф, содержащий указанный текст, на таблицу, созданную с помощью TableBuilder
|
||
/// </summary>
|
||
/// <param name="oldValue">Текст для поиска в параграфах</param>
|
||
/// <param name="buildTable">Действие для настройки таблицы через TableBuilder</param>
|
||
public void ReplaceToTable(string oldValue, Action<ITable> buildTable)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (string.IsNullOrEmpty(oldValue))
|
||
throw new ArgumentException("Old value cannot be null or empty", nameof(oldValue));
|
||
|
||
if (buildTable is null) return;
|
||
|
||
#if DEBUG
|
||
Debug.WriteLine($"[DEBUG] [WordProcessor.ReplaceToTable] Looking for '{oldValue}' to replace with custom table");
|
||
#endif
|
||
|
||
lock (_syncLock)
|
||
{
|
||
// Используем метод расширения для Body с TableBuilder
|
||
ReplaceToTableExt.ReplaceParagraphsContainingTextToTable(_body, oldValue, buildTable);
|
||
_isModified = true;
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region Replace to text
|
||
|
||
/// <summary>
|
||
/// Заменяет параграф, содержащий указанный текст, на таблицу, созданную с помощью TextBuilder
|
||
/// </summary>
|
||
/// <param name="oldValue">Текст для поиска в параграфах</param>
|
||
/// <param name="buildText">Действие для настройки таблицы через TextBuilder</param>
|
||
public void ReplaceToText(string oldValue, Action<IText> buildText)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (string.IsNullOrEmpty(oldValue))
|
||
throw new ArgumentException("Old value cannot be null or empty", nameof(oldValue));
|
||
|
||
if (buildText is null) return;
|
||
|
||
#if DEBUG
|
||
Debug.WriteLine($"[DEBUG] [WordProcessor.ReplaceToText] Looking for '{oldValue}' to replace with custom text");
|
||
#endif
|
||
|
||
lock (_syncLock)
|
||
{
|
||
// Используем метод расширения для Body с TextBuilder
|
||
ReplaceToTextExt.ReplaceParagraphsContainingTextToText(_body, oldValue, buildText);
|
||
_isModified = true;
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region Save Operations
|
||
|
||
/// <summary>
|
||
/// Сохраняет документ в файл, указанный при создании процессора.
|
||
/// </summary>
|
||
public void Save()
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (string.IsNullOrEmpty(FilePath))
|
||
throw new InvalidOperationException("Cannot save - no file path specified");
|
||
|
||
SaveTo(FilePath!);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Сохраняет документ в указанный файл.
|
||
/// </summary>
|
||
public void SaveTo(string path)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (string.IsNullOrEmpty(path))
|
||
throw new ArgumentException("Path cannot be null or empty", nameof(path));
|
||
|
||
lock (_syncLock)
|
||
{
|
||
try
|
||
{
|
||
if (_ms is not null)
|
||
{
|
||
_doc.Save();
|
||
_ms.Position = 0;
|
||
|
||
using var fileStream = new FileStream(
|
||
path,
|
||
FileMode.Create,
|
||
FileAccess.Write,
|
||
FileShare.None,
|
||
bufferSize: 65536);
|
||
|
||
_ms.CopyTo(fileStream);
|
||
_isModified = false;
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
throw new IOException($"Failed to save document to '{path}'", ex);
|
||
}
|
||
}
|
||
}
|
||
internal byte[] GetDocumentBytes()
|
||
{
|
||
lock (_syncLock)
|
||
{
|
||
_doc.Save();
|
||
_ms.Position = 0;
|
||
return _ms.ToArray();
|
||
}
|
||
}
|
||
|
||
public void SaveTo(Stream outputStream)
|
||
{
|
||
lock (_syncLock)
|
||
{
|
||
_doc.Save();
|
||
_ms.Position = 0;
|
||
_ms.CopyTo(outputStream);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// Пытается сохранить документ в указанный файл.
|
||
/// </summary>
|
||
public bool TrySaveTo(string path, out Exception? error)
|
||
{
|
||
error = null;
|
||
|
||
if (!IsValid || string.IsNullOrEmpty(path))
|
||
{
|
||
error = new InvalidOperationException("Processor is not valid or path is empty");
|
||
return false;
|
||
}
|
||
|
||
try
|
||
{
|
||
SaveTo(path);
|
||
return true;
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
error = ex;
|
||
#if DEBUG
|
||
Debug.WriteLine($"[DEBUG] Failed to save to '{path}': {ex.Message}");
|
||
#endif
|
||
return false;
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// Асинхронно сохраняет документ в указанный файл.
|
||
/// </summary>
|
||
public async Task SaveToAsync(string path, CancellationToken cancellationToken = default)
|
||
{
|
||
ThrowIfDisposed();
|
||
|
||
if (string.IsNullOrEmpty(path))
|
||
throw new ArgumentException("Path cannot be null or empty", nameof(path));
|
||
|
||
if (_ms is null)
|
||
throw new InvalidOperationException("Memory stream is not available");
|
||
|
||
byte[] buffer;
|
||
lock (_syncLock)
|
||
{
|
||
_doc.Save();
|
||
_ms.Position = 0;
|
||
buffer = _ms.ToArray();
|
||
}
|
||
|
||
using var fileStream = new FileStream(
|
||
path,
|
||
FileMode.Create,
|
||
FileAccess.Write,
|
||
FileShare.None,
|
||
bufferSize: 81920,
|
||
useAsync: true);
|
||
|
||
await fileStream.WriteAsync(buffer, 0, buffer.Length, cancellationToken);
|
||
}
|
||
|
||
/// <summary> Создаёт read-only процессор из текущего документа. </summary>
|
||
internal WordReader ToReader()
|
||
{
|
||
lock (_syncLock)
|
||
{
|
||
_doc.Save();
|
||
_ms.Position = 0;
|
||
var data = _ms.ToArray();
|
||
return WordReader.CreateFromData(data) ?? throw new InvalidOperationException("Failed to create reader");
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region Dispose Pattern
|
||
|
||
protected override void Dispose(bool disposing)
|
||
{
|
||
if (_disposed)
|
||
return;
|
||
|
||
lock (_syncLock)
|
||
{
|
||
if (disposing)
|
||
{
|
||
if (_isModified && !string.IsNullOrEmpty(FilePath))
|
||
{
|
||
try
|
||
{
|
||
_doc.Save();
|
||
|
||
Directory.CreateDirectory(Path.GetDirectoryName(FilePath));
|
||
|
||
_ms.Position = 0;
|
||
using var fileStream = new FileStream(
|
||
FilePath!,
|
||
FileMode.Create,
|
||
FileAccess.Write,
|
||
FileShare.None,
|
||
bufferSize: 81920);
|
||
|
||
_ms.CopyTo(fileStream);
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
#if DEBUG
|
||
Debug.WriteLine($"[DEBUG] Auto-save failed during Dispose: {ex.Message}");
|
||
#endif
|
||
}
|
||
}
|
||
}
|
||
|
||
base.Dispose(disposing);
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
} |