From 2ec6942a6d95b7d9cdc0f64283fa5f0ef1683587 Mon Sep 17 00:00:00 2001
From: Iurii Golovinskii <iurii-golovinskii.qubership@gmail.com>
Date: Tue, 25 Feb 2025 14:22:02 +0300
Subject: [PATCH] Add label map

---
 packages/diff-block/src/DiffBlock.tsx |  4 ++--
 packages/diff-block/src/constants.ts  | 15 +++++++++++++++
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/packages/diff-block/src/DiffBlock.tsx b/packages/diff-block/src/DiffBlock.tsx
index 9586e45..4ce6fed 100644
--- a/packages/diff-block/src/DiffBlock.tsx
+++ b/packages/diff-block/src/DiffBlock.tsx
@@ -4,7 +4,7 @@ import { useAtom } from 'jotai';
 import React, { CSSProperties } from 'react';
 import ReactDOM from 'react-dom';
 
-import { DIFF_TYPE_COLOR_MAP } from './constants';
+import { CHANGE_SEVERITY_NAME_MAP, DIFF_TYPE_COLOR_MAP } from './constants'
 import { useDiffBlockIdPrefixContext } from './DiffBlockIdPrefixContext';
 import { useDiffContext } from './DiffContext';
 import { diffBlockHeight, diffBlockTop } from './state';
@@ -87,7 +87,7 @@ export const DiffBlock: React.FunctionComponent<{
     return (
       <Box
         className="diff-block-type"
-        data-diff-type={`${type === 'semi-breaking'? 'risky': type}${cause ? ', ' + cause : ''}`}
+        data-diff-type={`${CHANGE_SEVERITY_NAME_MAP[type]}${cause ? ', ' + cause : ''}`}
         style={{
           minHeight,
           background,
diff --git a/packages/diff-block/src/constants.ts b/packages/diff-block/src/constants.ts
index e2fccba..e81106b 100644
--- a/packages/diff-block/src/constants.ts
+++ b/packages/diff-block/src/constants.ts
@@ -9,6 +9,13 @@ export const syntheticTitleFlag = Symbol('syntheticTitleFlag');
 
 export const DIFF_TYPES: DiffType[] = keys(ClassifierType) as DiffType[];
 
+export const BREAKING_CHANGE_SEVERITY = 'breaking'
+export const NON_BREAKING_CHANGE_SEVERITY = 'non-breaking'
+export const SEMI_BREAKING_CHANGE_SEVERITY = 'semi-breaking'
+export const DEPRECATED_CHANGE_SEVERITY = 'deprecated'
+export const ANNOTATION_CHANGE_SEVERITY = 'annotation'
+export const UNCLASSIFIED_CHANGE_SEVERITY = 'unclassified'
+
 export const DIFF_TYPE_COLOR_MAP = {
   breaking: '#ED4A54',
   deprecated: '#F4B24D',
@@ -27,6 +34,14 @@ export const DIFF_BUDGES_COLOR_MAP = {
   unclassified: '#70A9EC',
 };
 
+export const CHANGE_SEVERITY_NAME_MAP: Record<DiffType, string> = {
+  [BREAKING_CHANGE_SEVERITY]: 'breaking',
+  [SEMI_BREAKING_CHANGE_SEVERITY]: 'risky',
+  [DEPRECATED_CHANGE_SEVERITY]: 'deprecated',
+  [NON_BREAKING_CHANGE_SEVERITY]: 'non-breaking',
+  [UNCLASSIFIED_CHANGE_SEVERITY]: 'unclassified',
+  [ANNOTATION_CHANGE_SEVERITY]: 'annotation',
+}
 export type WithDiffMetaKey<T> = T & {
   [diffMetaKey]: any;
 };