Refactor to remove code duplication

Signed-off-by: Nicolas Rol <nicolas.rol@rte-france.com>

Author: rolnico

network-area-diagram/src/main/java/com/powsybl/nad/svg/SvgWriter.java

@@ -219,25 +219,8 @@ private void insertSvgComponent(Injection injection, XMLStreamWriter writer) thr
         writer.writeStartElement(GROUP_ELEMENT_NAME);
         writer.writeAttribute(TRANSFORM_ATTRIBUTE, getTranslateString(injection.getIconOrigin(svgParameters.getInjectionCircleRadius())));
 
-        Result result = new SAXResult(new SvgContentHandlerToXMLStreamWriter(writer));
         String componentType = injection.getComponentType();
-        writeStyleClasses(writer, componentLibrary.getComponentStyleClass(componentType).map(List::of).orElse(List.of()));
-
-        try {
-            Transformer transformer = componentLibrary.getSvgTransformer();
-            Map<String, List<Element>> subComponents = componentLibrary.getSvgElements(componentType);
-            for (Map.Entry<String, List<Element>> scEntry : subComponents.entrySet()) {
-                List<String> edgeStyleClasses = componentLibrary.getSubComponentStyleClass(componentType, scEntry.getKey())
-                        .map(List::of).orElse(List.of());
-                writeStyleClasses(writer, edgeStyleClasses);
-                for (Element element : scEntry.getValue()) {
-                    transformer.transform(new DOMSource(element), result);
-                }
-            }
-        } catch (TransformerException e) {
-            throw new PowsyblException("Cannot insert SVG for injection of type " + injection.getType(), e);
-        }
-        writer.writeEndElement();
+        writeSvgComponent(writer, componentType, "Cannot insert SVG for injection of type " + injection.getType());
     }
 
     private void drawComponentOnBranchEdgeMiddle(XMLStreamWriter writer, String componentType) throws XMLStreamException {
@@ -250,6 +233,10 @@ private void drawComponentOnBranchEdgeMiddle(XMLStreamWriter writer, String comp
         Point trans = new Point(-componentSize.width() / 2, -componentSize.height() / 2);
         writer.writeAttribute(TRANSFORM_ATTRIBUTE, getTranslateString(trans));
 
+        writeSvgComponent(writer, componentType, "Cannot insert SVG component of type " + componentType + " on branch edge middle");
+    }
+
+    private void writeSvgComponent(XMLStreamWriter writer, String componentType, String errorMessage) throws XMLStreamException {
         Result result = new SAXResult(new SvgContentHandlerToXMLStreamWriter(writer));
         writeStyleClasses(writer, componentLibrary.getComponentStyleClass(componentType).map(List::of).orElse(List.of()));
 
@@ -259,15 +246,15 @@ private void drawComponentOnBranchEdgeMiddle(XMLStreamWriter writer, String comp
             if (subComponents != null) {
                 for (Map.Entry<String, List<Element>> scEntry : subComponents.entrySet()) {
                     List<String> edgeStyleClasses = componentLibrary.getSubComponentStyleClass(componentType, scEntry.getKey())
-                            .map(List::of).orElse(List.of());
+                        .map(List::of).orElse(List.of());
                     writeStyleClasses(writer, edgeStyleClasses);
                     for (Element element : scEntry.getValue()) {
                         transformer.transform(new DOMSource(element), result);
                     }
                 }
             }
         } catch (TransformerException e) {
-            throw new PowsyblException("Cannot insert SVG component of type " + componentType + " on branch edge middle", e);
+            throw new PowsyblException(errorMessage, e);
         }
         writer.writeEndElement();
     }