feat(sunburst): 旭日图支持图例配置 & 单测 (#3011)

* feat(sunburst): 旭日图支持图例配置 & 单测

* chore: 增加包大小限制

Author: visiky

__tests__/unit/plots/sunburst/legend-spec.ts

@@ -0,0 +1,46 @@
+import { Sunburst } from '../../../../src';
+import { SUNBRUST_DATA } from '../../../data/sunburst';
+import { createDiv } from '../../../utils/dom';
+
+describe('plot legend', () => {
+  const div = createDiv();
+  const plot = new Sunburst(div, {
+    data: SUNBRUST_DATA,
+    colorField: 'label',
+  });
+  plot.render();
+
+  it('legend: default is false', () => {
+    expect(plot.chart.getOptions().legends).toEqual(false);
+    expect(plot.chart.getComponents().filter((co) => co.type === 'legend').length).toBe(0);
+  });
+
+  it('legend config', () => {
+    plot.update({
+      legend: {
+        position: 'top-left',
+      },
+    });
+    // @ts-ignore
+    expect(plot.chart.getOptions().legends.label).toEqual({ position: 'top-left' });
+    expect(plot.chart.getComponents().filter((co) => co.type === 'legend').length).toBe(1);
+
+    plot.update({
+      ...plot.options,
+      legend: {
+        flipPage: true,
+      },
+    });
+
+    // @ts-ignore
+    expect(plot.chart.getOptions().legends.label).toEqual({
+      position: 'top-left',
+      flipPage: true,
+    });
+    expect(plot.chart.getComponents().filter((co) => co.type === 'legend').length).toBe(1);
+  });
+
+  afterAll(() => {
+    plot.destroy();
+  });
+});

package.json

@@ -145,7 +145,7 @@
   "limit-size": [
     {
       "path": "dist/g2plot.min.js",
-      "limit": "995 Kb"
+      "limit": "998 Kb"
     },
     {
       "path": "dist/g2plot.min.js",

src/plots/sunburst/adaptor.ts

@@ -2,7 +2,15 @@ import { isFunction, get, uniq } from '@antv/util';
 import { Types } from '@antv/g2';
 import { Params } from '../../core/adaptor';
 import { polygon as polygonAdaptor } from '../../adaptor/geometries';
-import { interaction as baseInteraction, animation, theme, annotation, scale, pattern } from '../../adaptor/common';
+import {
+  interaction as baseInteraction,
+  animation,
+  theme,
+  annotation,
+  scale,
+  pattern,
+  legend,
+} from '../../adaptor/common';
 import { flow, findGeometry, transformLabel, deepAssign } from '../../utils';
 import { getAdjustAppendPadding } from '../../utils/padding';
 import { Datum } from '../../types';
@@ -63,18 +71,6 @@ export function axis(params: Params<SunburstOptions>): Params<SunburstOptions> {
   return params;
 }
 
-/**
- * legend 配置（旭日图暂时不支持图例，后续需要支持的话，得自定义数据筛选）
- * @param params
- * @returns
- */
-function legend(params: Params<SunburstOptions>): Params<SunburstOptions> {
-  const { chart } = params;
-
-  chart.legend(false);
-  return params;
-}
-
 /**
  * 数据标签
  * @param params

src/plots/sunburst/constant.ts

@@ -41,6 +41,7 @@ export const DEFAULT_OPTIONS: Partial<SunburstOptions> = deepAssign({}, Plot.get
     offset: 20,
     showTitle: false,
   },
+  legend: false,
 
   // 样式设置
   sunburstStyle: {

src/plots/sunburst/types.ts

@@ -2,7 +2,7 @@ import { ColorAttr, Options, StyleAttr } from '../../types';
 import { DrillDownCfg } from '../../types/drill-down';
 import { HierarchyOption } from '../../utils/hierarchy/types';
 
-export interface SunburstOptions extends Omit<Options, 'data' | 'legend' | 'slider' | 'scrollbar' | 'xAxis' | 'yAxis'> {
+export interface SunburstOptions extends Omit<Options, 'data' | 'slider' | 'scrollbar' | 'xAxis' | 'yAxis'> {
   /** 旭日图数据 */
   readonly data: any;
   /** 径向类型 */