From 476771b6e4aba25edeed02323eff3d9d644718f1 Mon Sep 17 00:00:00 2001
From: "grafana-delivery-bot[bot]"
 <132647405+grafana-delivery-bot[bot]@users.noreply.github.com>
Date: Mon, 18 Sep 2023 20:52:32 +0300
Subject: [PATCH] [v10.1.x] Canvas: Avoid conflicting stylesheets when loading
 SVG icons (#75032)

---
 .../core/components/SVG/SanitizedSVG.test.tsx | 58 +++++++++++++++++++
 .../app/core/components/SVG/SanitizedSVG.tsx  | 25 +++++++-
 public/app/core/components/SVG/utils.ts       | 30 ++++++++++
 public/app/features/canvas/elements/icon.tsx  |  1 +
 4 files changed, 112 insertions(+), 2 deletions(-)
 create mode 100644 public/app/core/components/SVG/SanitizedSVG.test.tsx
 create mode 100644 public/app/core/components/SVG/utils.ts

diff --git a/public/app/core/components/SVG/SanitizedSVG.test.tsx b/public/app/core/components/SVG/SanitizedSVG.test.tsx
new file mode 100644
index 000000000000..16f2af2cd99c
--- /dev/null
+++ b/public/app/core/components/SVG/SanitizedSVG.test.tsx
@@ -0,0 +1,58 @@
+import { getSvgId, getSvgStyle, svgStyleCleanup } from './utils';
+
+const ID = 'TEST_ID';
+
+const svgNoId =
+  '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"><style type="text/css">.st0{fill:purple;}</style><circle cx="12" cy="12" r="10" class="st0"/></svg>';
+
+const svgWithId = `<svg id="${ID}" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"><style type="text/css">.st0{fill:blue;}</style><circle cx="12" cy="12" r="10" class="st0"/></svg>`;
+
+const svgWithWrongIdInStyle =
+  '<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"><style type="text/css">#WRONG .st0{fill:green;}</style><circle cx="12" cy="12" r="10" class="st0"/></svg>';
+
+describe('SanitizedSVG', () => {
+  it('should cleanup the style and generate an ID', () => {
+    const cleanStyle = svgStyleCleanup(svgNoId);
+    const updatedStyle = getSvgStyle(cleanStyle);
+    const svgId = getSvgId(cleanStyle);
+
+    expect(cleanStyle.indexOf('id="')).toBeGreaterThan(-1);
+    expect(svgId).toBeDefined();
+    expect(svgId?.startsWith('x')).toBeTruthy();
+    expect(updatedStyle?.indexOf(`#${svgId}`)).toBeGreaterThan(-1);
+
+    expect(cleanStyle).toEqual(
+      `<svg id="${svgId}" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"><style type="text/css">#${svgId} .st0{fill:purple;}</style><circle cx="12" cy="12" r="10" class="st0"/></svg>`
+    );
+  });
+
+  it('should cleanup the style and use the existing ID', () => {
+    const cleanStyle = svgStyleCleanup(svgWithId);
+    const updatedStyle = getSvgStyle(cleanStyle);
+    const svgId = getSvgId(cleanStyle);
+
+    expect(cleanStyle.indexOf('id="')).toBeGreaterThan(-1);
+    expect(svgId).toBeDefined();
+    expect(svgId).toEqual(ID);
+    expect(updatedStyle?.indexOf(`#${svgId}`)).toBeGreaterThan(-1);
+
+    expect(cleanStyle).toEqual(
+      `<svg id="${svgId}" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"><style type="text/css">#${svgId} .st0{fill:blue;}</style><circle cx="12" cy="12" r="10" class="st0"/></svg>`
+    );
+  });
+
+  it('should cleanup the style and replace the wrong ID', () => {
+    const cleanStyle = svgStyleCleanup(svgWithWrongIdInStyle);
+    const updatedStyle = getSvgStyle(cleanStyle);
+    const svgId = getSvgId(cleanStyle);
+
+    expect(cleanStyle.indexOf('id="')).toBeGreaterThan(-1);
+    expect(svgId).toBeDefined();
+    expect(svgId?.startsWith('x')).toBeTruthy();
+    expect(updatedStyle?.indexOf(`#${svgId}`)).toBeGreaterThan(-1);
+
+    expect(cleanStyle).toEqual(
+      `<svg id="${svgId}" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"><style type="text/css">#${svgId} .st0{fill:green;}</style><circle cx="12" cy="12" r="10" class="st0"/></svg>`
+    );
+  });
+});
diff --git a/public/app/core/components/SVG/SanitizedSVG.tsx b/public/app/core/components/SVG/SanitizedSVG.tsx
index 5f4f0a6f5aa6..b53bdccf5a10 100644
--- a/public/app/core/components/SVG/SanitizedSVG.tsx
+++ b/public/app/core/components/SVG/SanitizedSVG.tsx
@@ -3,8 +3,13 @@ import SVG, { Props } from 'react-inlinesvg';
 
 import { textUtil } from '@grafana/data';
 
-export const SanitizedSVG = (props: Props) => {
-  return <SVG {...props} cacheRequests={true} preProcessor={getCleanSVG} />;
+import { svgStyleCleanup } from './utils';
+
+type SanitizedSVGProps = Props & { cleanStyle?: boolean };
+
+export const SanitizedSVG = (props: SanitizedSVGProps) => {
+  const { cleanStyle, ...inlineSvgProps } = props;
+  return <SVG {...inlineSvgProps} cacheRequests={true} preProcessor={cleanStyle ? getCleanSVGAndStyle : getCleanSVG} />;
 };
 
 let cache = new Map<string, string>();
@@ -15,5 +20,21 @@ function getCleanSVG(code: string): string {
     clean = textUtil.sanitizeSVGContent(code);
     cache.set(code, clean);
   }
+
+  return clean;
+}
+
+function getCleanSVGAndStyle(code: string): string {
+  let clean = cache.get(code);
+  if (!clean) {
+    clean = textUtil.sanitizeSVGContent(code);
+
+    if (clean.indexOf('<style type="text/css">') > -1) {
+      clean = svgStyleCleanup(clean);
+    }
+
+    cache.set(code, clean);
+  }
+
   return clean;
 }
diff --git a/public/app/core/components/SVG/utils.ts b/public/app/core/components/SVG/utils.ts
new file mode 100644
index 000000000000..b6cfa52da667
--- /dev/null
+++ b/public/app/core/components/SVG/utils.ts
@@ -0,0 +1,30 @@
+import { v4 as uuidv4 } from 'uuid';
+
+const MATCH_ID_INDEX = 2;
+const SVG_ID_INSERT_POS = 5;
+
+export const getSvgStyle = (svgCode: string) => {
+  const svgStyle = svgCode.match(new RegExp('<style type="text/css">([\\s\\S]*?)<\\/style>'));
+  return svgStyle ? svgStyle[0] : null;
+};
+
+export const getSvgId = (svgCode: string) => {
+  return svgCode.match(new RegExp('<svg.*id\\s*=\\s*([\'"])(.*?)\\1'))?.[MATCH_ID_INDEX];
+};
+
+export const svgStyleCleanup = (svgCode: string) => {
+  let svgId = getSvgId(svgCode);
+  if (!svgId) {
+    svgId = `x${uuidv4()}`;
+    const pos = svgCode.indexOf('<svg') + SVG_ID_INSERT_POS;
+    svgCode = svgCode.substring(0, pos) + `id="${svgId}" ` + svgCode.substring(pos);
+  }
+
+  let svgStyle = getSvgStyle(svgCode);
+  if (svgStyle) {
+    let replacedId = svgStyle.replace(/(#(.*?))?\./g, `#${svgId} .`);
+    svgCode = svgCode.replace(svgStyle, replacedId);
+  }
+
+  return svgCode;
+};
diff --git a/public/app/features/canvas/elements/icon.tsx b/public/app/features/canvas/elements/icon.tsx
index 55b47a6024d8..3c5a860b0a4d 100644
--- a/public/app/features/canvas/elements/icon.tsx
+++ b/public/app/features/canvas/elements/icon.tsx
@@ -60,6 +60,7 @@ export function IconDisplay(props: CanvasElementProps) {
       src={data.path}
       style={svgStyle}
       className={svgStyle.strokeWidth ? svgStrokePathClass : undefined}
+      cleanStyle={true}
     />
   );
 }