diff --git a/src/Kompas.Mcp.Core/Assemblies/AssemblyService.cs b/src/Kompas.Mcp.Core/Assemblies/AssemblyService.cs
index 377fad6..5393bcb 100644
--- a/src/Kompas.Mcp.Core/Assemblies/AssemblyService.cs
+++ b/src/Kompas.Mcp.Core/Assemblies/AssemblyService.cs
@@ -39,11 +39,21 @@ public sealed class AssemblyService
var parts = top.Parts
?? throw new InvalidOperationException("Не удалось получить коллекцию компонентов сборки (Parts).");
+ // Фиксируем число компонентов ДО вставки — для верифицируемого отката (откатываем всё, что
+ // появилось сверх beforeCount), не полагаясь на возможный null-возврат AddFromFile.
+ int beforeCount = parts.Count;
+
// ExternalFile=true: компонент ссылается на внешний .m3d/.a3d. Redraw=true: вставить и
- // перерисовать в исходном положении (само перемещение применит UpdatePlacement ниже).
- var part = parts.AddFromFile(path, true, true) as IPart7
- ?? throw new InvalidOperationException(
+ // перерисовать в исходном положении (само перемещение применит SetOrigin+RebuildModel ниже).
+ var part = parts.AddFromFile(path, true, true) as IPart7;
+ if (part is null)
+ {
+ // AddFromFile мог вставить компонент и всё равно вернуть null/не-IPart7 — почистим, если
+ // число компонентов выросло, чтобы не оставить «осиротевший» компонент в сборке.
+ RollbackInsertion(top, parts, beforeCount);
+ throw new InvalidOperationException(
$"Не удалось вставить компонент из файла (AddFromFile вернул null или не IPart7): {path}");
+ }
try
{
@@ -56,18 +66,41 @@ public sealed class AssemblyService
// (без сопряжений/зависимостей) он может вернуть FALSE — это НЕ ошибка (нечего
// пересчитывать). Фактически положение становится видимым в геометрии после RebuildModel.
part.UpdatePlacement(true);
- top.RebuildModel(true);
+ if (!top.RebuildModel(true))
+ throw new InvalidOperationException("RebuildModel вернул FALSE — сборка не перестроена после вставки компонента.");
return new AddedComponent { Index = parts.Count - 1, Name = SafeName(part) };
}
catch
{
- // Best-effort откат осиротевшего компонента, если позиционирование не удалось.
- try { part.Owner?.Delete(); } catch { /* лучшее усилие */ }
+ // Верифицируемый откол: удаляем всё, что добавилось сверх beforeCount, и проверяем.
+ RollbackInsertion(top, parts, beforeCount);
throw;
}
}
+ ///
+ /// Откатить вставку: удалить компоненты, появившиеся сверх (обычно
+ /// один — только что вставленный). Best-effort, но с верификацией: компонент удаляется через
+ /// IFeature7.Delete() своего владельца, затем сборка перестраивается.
+ ///
+ private static void RollbackInsertion(IPart7 top, IParts7 parts, int beforeCount)
+ {
+ try
+ {
+ // Удаляем с конца, пока число не вернётся к исходному (защита от частичной вставки).
+ for (int guard = 0; parts.Count > beforeCount && guard < 8; guard++)
+ {
+ if (parts.Part[parts.Count - 1] is IPart7 last && last.Owner is { } owner)
+ owner.Delete();
+ else
+ break; // не удалось получить владельца — выходим, чтобы не зациклиться
+ }
+ top.RebuildModel(true);
+ }
+ catch { /* лучшее усилие: исходное исключение важнее ошибки отката */ }
+ }
+
///
/// Получить верхний компонент активной сборки, проверив тип документа. Раздельные ошибки:
/// нет документа / документ не сборка / не 3D / нет TopPart. Деталь — тоже IKompasDocument3D,
diff --git a/tests/Kompas.Mcp.Tests/Integration/AssemblyTests.cs b/tests/Kompas.Mcp.Tests/Integration/AssemblyTests.cs
index 694a68b..0c28a0b 100644
--- a/tests/Kompas.Mcp.Tests/Integration/AssemblyTests.cs
+++ b/tests/Kompas.Mcp.Tests/Integration/AssemblyTests.cs
@@ -97,6 +97,48 @@ public sealed class AssemblyTests
finally { await _docs.CloseAsync(save: false); }
}
+ [Fact]
+ public async Task AddComponent_into_non_empty_assembly_increments_and_indexes()
+ {
+ // Вставка ВТОРОГО компонента в непустую сборку: проверяет инкремент count и что
+ // возвращённый Index указывает именно на вставленный компонент (а не «последний по случаю»).
+ var partFile = await BuildAndSaveBoxAsync(30, 30, 10);
+
+ await _docs.CreateAsync(KompasDocumentType.Assembly);
+ try
+ {
+ await _assembly.AddComponentAsync(partFile, 0, 0, 0); // первый
+ var before = (await _query.ListComponentsAsync()).Count;
+ Assert.Equal(1, before);
+
+ var added = await _assembly.AddComponentAsync(partFile, 60, 0, 0); // второй, со сдвигом
+
+ var comps = await _query.ListComponentsAsync();
+ Assert.Equal(before + 1, comps.Count);
+ Assert.Equal(comps.Count - 1, added.Index);
+ // Компонент по возвращённому индексу — деталь нужного габарита.
+ Assert.InRange(comps[added.Index].SizeX, 30 * 0.95, 30 * 1.05);
+ }
+ finally { await _docs.CloseAsync(save: false); }
+ }
+
+ [Fact]
+ public async Task AddComponent_missing_file_throws_and_keeps_assembly_empty()
+ {
+ var missing = Path.Combine(TestPaths.Scratch, $"no_such_{Guid.NewGuid():N}.m3d");
+
+ await _docs.CreateAsync(KompasDocumentType.Assembly);
+ try
+ {
+ await Assert.ThrowsAsync(
+ () => _assembly.AddComponentAsync(missing, 0, 0, 0));
+
+ // Сборка не должна получить «осиротевший» компонент от неуспешной вставки.
+ Assert.Empty(await _query.ListComponentsAsync());
+ }
+ finally { await _docs.CloseAsync(save: false); }
+ }
+
[Fact]
public async Task AddComponent_rejects_non_assembly_document()
{