feature: remove unneeded series / categories

Author: singerla

src/helper/modify-helper.ts

@@ -72,6 +72,7 @@ export const setChartData = (data: ChartData) => (
   new ModifyChart(chart, workbook, data, slots).modify();
 
   // XmlHelper.dump(chart)
+  // XmlHelper.dump(workbook.table)
 };
 
 export const setChartVerticalLines = (data: ChartData) => (
@@ -139,6 +140,8 @@ export const setChartBubbles = (data: ChartData) => (
   });
 
   new ModifyChart(chart, workbook, data, slots).modify();
+
+  // XmlHelper.dump(chart)
 };
 
 export const setTableData = (data: TableData) => (

src/helper/modify-xml-helper.ts

@@ -16,6 +16,14 @@ export default class ModifyXmlHelper {
 
     for (const tag in tags) {
       const modifier = tags[tag] as Modification;
+
+      if (GeneralHelper.propertyExists(modifier, 'collection')) {
+        const modifies = GeneralHelper.arrayify(modifier.collection);
+        const collection = root.getElementsByTagName(tag);
+        Object.values(modifies).forEach((modifyXml) => modifyXml(collection));
+        return;
+      }
+
       const index = modifier.index || 0;
 
       this.assertNode(root.getElementsByTagName(tag), index, tag, modifier);

src/helper/xml-helper.ts

@@ -317,6 +317,16 @@ export class XmlHelper {
     referenceNode.parentNode.insertBefore(newNode, referenceNode.nextSibling);
   }
 
+  static sliceCollection(
+    collection: HTMLCollectionOf<Element>,
+    length: number,
+  ): void {
+    for (let i = collection.length; i > length; i--) {
+      const toRemove = collection[i - 1];
+      toRemove.parentNode.removeChild(toRemove);
+    }
+  }
+
   static dump(element: XMLDocument | Element): void {
     const s = new XMLSerializer();
     const xmlBuffer = s.serializeToString(element);

src/modify/modify-chart.ts

@@ -45,11 +45,15 @@ export class ModifyChart {
     this.width = this.columns.length;
   }
 
-  modify() {
+  modify(): void {
     this.setValues();
     this.setSeries();
+    this.sliceChartSpace();
+
     this.setWorkbook();
+    this.sliceWorkbook();
     this.setWorkbookTable();
+    this.sliceWorkbookTable();
   }
 
   setColumns(slot: ChartSlot[]): ChartColumn[] {
@@ -62,9 +66,10 @@ export class ModifyChart {
 
       const label = slot.label ? slot.label : series.label;
 
-      const mapData = slot.mapData ? slot.mapData : (point: number) => point;
+      const mapData =
+        slot.mapData !== undefined ? slot.mapData : (point: number) => point;
 
-      const isStrRef = slot.isStrRef ? slot.isStrRef : true;
+      const isStrRef = slot.isStrRef !== undefined ? slot.isStrRef : true;
 
       const worksheetCb = (
         point: number,
@@ -115,23 +120,22 @@ export class ModifyChart {
       this.columns
         .filter((col) => col.chart)
         .forEach((col) => {
-          if(category.values[col.series] === undefined) {
-            throw new Error(`No value for category "${category.label}" at series "${(col.label)}".`)
+          if (category.values[col.series] === undefined) {
+            throw new Error(
+              `No value for category "${category.label}" at series "${col.label}".`,
+            );
           }
 
-          this.chart.modify(
-            this.series(
-              col.series,
-              col.chart(category.values[col.series], c, category),
-            ),
-          );
+          col.modTags = col.chart(category.values[col.series], c, category);
+
+          this.chart.modify(this.series(col.series, col.modTags));
         });
     });
   }
 
   setSeries(): void {
     this.columns.forEach((column, colId) => {
-      if (column.isStrRef === undefined || column.isStrRef === true) {
+      if (column.isStrRef === true) {
         this.chart.modify(
           this.series(column.series, {
             ...this.seriesId(column.series),
@@ -142,6 +146,23 @@ export class ModifyChart {
     });
   }
 
+  sliceChartSpace(): void {
+    this.chart.modify({
+      'c:plotArea': this.slice('c:ser', this.data.series.length),
+    });
+
+    this.columns
+      .filter((column) => column.modTags)
+      .forEach((column) => {
+        const sliceMod = {};
+
+        Object.keys(column.modTags).forEach((tag) => {
+          sliceMod[tag] = this.slice('c:pt', this.height);
+        });
+        this.chart.modify(this.series(column.series, sliceMod));
+      });
+  }
+
   setWorkbook(): void {
     this.workbook.modify(this.spanString());
     this.workbook.modify(this.rowAttributes(0, 1));
@@ -161,6 +182,28 @@ export class ModifyChart {
     });
   }
 
+  sliceWorkbook(): void {
+    this.data.categories.forEach((category, c) => {
+      const r = c + 1;
+      this.workbook.modify({
+        row: {
+          index: r,
+          ...this.slice('c', this.width + 1),
+        },
+      });
+    });
+
+    this.workbook.modify({
+      row: {
+        ...this.slice('c', this.width + 1),
+      },
+    });
+
+    this.workbook.modify({
+      sheetData: this.slice('row', this.height + 1),
+    });
+  }
+
   series = (index: number, children: ModificationTags): ModificationTags => {
     return {
       'c:ser': {
@@ -311,6 +354,18 @@ export class ModifyChart {
     };
   }
 
+  slice(tag: string, length: number): Modification {
+    return {
+      children: {
+        [tag]: {
+          collection: (collection: HTMLCollectionOf<Element>) => {
+            XmlHelper.sliceCollection(collection, length);
+          },
+        },
+      },
+    };
+  }
+
   spanString(): ModificationTags {
     return {
       dimension: {
@@ -365,4 +420,10 @@ export class ModifyChart {
       },
     });
   }
+
+  sliceWorkbookTable(): void {
+    this.workbookTable.modify({
+      table: this.slice('tableColumn', this.width + 1),
+    });
+  }
 }

src/shapes/chart.ts

@@ -173,7 +173,7 @@ export class Chart extends Shape implements IChart {
     await this.editTargetWorksheetRel();
   }
 
-  async getWorksheetFilePrefix(targetRelFile: string) {
+  async getWorksheetFilePrefix(targetRelFile: string): Promise<string> {
     const relationTargets = await XmlHelper.getTargetsFromRelationships(
       this.sourceArchive,
       targetRelFile,

src/types/chart-types.ts

@@ -34,6 +34,7 @@ export type ChartColumn = {
     category: ChartCategory,
   ) => ModificationTags;
   isStrRef?: boolean;
+  modTags?: ModificationTags;
 };
 export type ChartData = {
   series: ChartSeries[];

src/types/modify-types.ts

@@ -1,8 +1,12 @@
 export type ModifyCallback = {
   (element: Element);
 };
+export type ModifyCollectionCallback = {
+  (collection: HTMLCollectionOf<Element>);
+};
 export type Modification = {
   index?: number;
+  collection?: ModifyCollectionCallback | ModifyCollectionCallback;
   children?: ModificationTags;
   modify?: ModifyCallback | ModifyCallback[];
 };