feature(chart): add setExtendedChartData and chartEx related modifiers (WIP)

Author: singerla

__tests__/pptx-templates/ChartWaterfall.pptx

@@ -419,27 +419,33 @@ export class Slide implements ISlide {
     sourceArchive: JSZip,
     slideNumber: number,
   ): Promise<AnalyzedElementType> {
-    vd('analyzeElement');
     const isChart = sourceElement.getElementsByTagName('c:chart');
     if (isChart.length) {
-      vd({
+      const target = await XmlHelper.getTargetByRelId(
+        sourceArchive,
+        slideNumber,
+        sourceElement,
+        'chart',
+      );
+
+      return {
         type: ElementType.Chart,
-        target: await XmlHelper.getTargetByRelId(
-          sourceArchive,
-          slideNumber,
-          sourceElement,
-          'chart',
-        ),
-      });
+        target: target,
+      } as AnalyzedElementType;
+    }
+
+    const isChartEx = sourceElement.getElementsByTagName('cx:chart');
+    if (isChartEx.length) {
+      const target = await XmlHelper.getTargetByRelId(
+        sourceArchive,
+        slideNumber,
+        sourceElement,
+        'chartEx',
+      );
 
       return {
         type: ElementType.Chart,
-        target: await XmlHelper.getTargetByRelId(
-          sourceArchive,
-          slideNumber,
-          sourceElement,
-          'chart',
-        ),
+        target: target,
       } as AnalyzedElementType;
     }
 

src/classes/slide.ts

@@ -6,6 +6,11 @@ export const TargetByRelIdMap = {
     relAttribute: 'r:id',
     prefix: '../charts/chart',
   } as TargetByRelIdMapParam,
+  chartEx: {
+    relRootTag: 'cx:chart',
+    relAttribute: 'r:id',
+    prefix: '../charts/chartEx',
+  } as TargetByRelIdMapParam,
   image: {
     relRootTag: 'a:blip',
     relAttribute: 'r:embed',

src/constants/constants.ts

@@ -10,18 +10,36 @@ const automizer = new Automizer({
 const run = async () => {
   const pres = automizer
     .loadRoot(`RootTemplate.pptx`)
-    .load(`ChartWaterfall.pptx`, 'charts');
+    .load(`EmptySlide.pptx`, 'EmptySlide')
+    .load(`ChartWaterfall.pptx`, 'ChartWaterfall')
+    .load(`ChartBarsStacked.pptx`, 'ChartBarsStacked');
 
   const result = await pres
-    .addSlide('charts', 1, (slide) => {
-      slide.addElement('charts', 1, 'Waterfall 1', [
-        modify.setChartData(<ChartData>{
+    .addSlide('EmptySlide', 1, (slide) => {
+      // slide.addElement('ChartBarsStacked', 1, 'BarsStacked', [
+      //   modify.setChartData(<ChartData>{
+      //     series: [{ label: 'series 1' }],
+      //     categories: [
+      //       { label: 'cat 2-1', values: [50] },
+      //       { label: 'cat 2-2', values: [14] },
+      //       { label: 'cat 2-3', values: [15] },
+      //       { label: 'cat 2-4', values: [26] },
+      //     ],
+      //   }),
+      // ]);
+
+      slide.addElement('ChartWaterfall', 1, 'Waterfall 1', [
+        modify.setExtendedChartData(<ChartData>{
           series: [{ label: 'series 1' }],
           categories: [
             { label: 'cat 2-1', values: [50] },
             { label: 'cat 2-2', values: [14] },
             { label: 'cat 2-3', values: [15] },
             { label: 'cat 2-4', values: [26] },
+            { label: 'cat 2-4', values: [26] },
+            { label: 'cat 2-4', values: [26] },
+            { label: 'cat 2-4', values: [26] },
+            { label: 'cat 2-4', values: [26] },
           ],
         }),
       ]);

src/dev.ts

@@ -29,9 +29,15 @@ export class FileHelper {
     return archive.files[file].async(type || 'string');
   }
 
+  static fileExistsInArchive(archive: JSZip, file: string): boolean {
+    if (archive === undefined || archive.files[file] === undefined) {
+      return false;
+    }
+  }
+
   static extractFileContent(file: Buffer): Promise<JSZip> {
     const zip = new JSZip();
-    return zip.loadAsync((file as unknown) as InputType);
+    return zip.loadAsync(file as unknown as InputType);
   }
 
   static getFileExtension(filename: string): string {

src/helper/file-helper.ts

@@ -206,6 +206,33 @@ export default class ModifyChartHelper {
       // XmlHelper.dump(chart)
     };
 
+  /**
+   * Set chart data to modify extended chart types.
+   * See `__tests__/modify-existing-extended-chart.test.js`
+   */
+  static setExtendedChartData =
+    (data: ChartData) =>
+    (
+      element: XMLDocument | Element,
+      chart?: Document,
+      workbook?: Workbook,
+    ): void => {
+      const slots = [] as ChartSlot[];
+      data.series.forEach((series: ChartSeries, s: number) => {
+        slots.push({
+          index: s,
+          series: series,
+          targetCol: s + 1,
+          type: 'extendedSeries',
+        });
+      });
+
+      new ModifyChart(chart, workbook, data, slots).modifyExtended();
+
+      XmlHelper.dump(chart);
+      // XmlHelper.dump(workbook.table)
+    };
+
   static setAxisRange =
     (range: ChartAxisRange) =>
     (chart: XMLDocument): void => {

src/helper/modify-chart-helper.ts

@@ -116,13 +116,23 @@ export default class ModifyXmlHelper {
   static value =
     (value: number | string, index?: number) =>
     (element: Element): void => {
-      element.getElementsByTagName('c:v')[0].firstChild.textContent =
-        String(value);
+      const valueElement = element.getElementsByTagName('c:v');
+      if (!valueElement.length) {
+        XmlHelper.dump(element);
+        throw 'Unable to set value @index: ' + index;
+      }
+
+      valueElement[0].firstChild.textContent = String(value);
       if (index !== undefined) {
         element.setAttribute('idx', String(index));
       }
     };
 
+  static textContent =
+    (value: number | string) =>
+    (element: Element): void => {
+      element.firstChild.textContent = String(value);
+    };
   static attribute =
     (attribute: string, value: string | number) =>
     (element: Element): void => {

src/helper/modify-xml-helper.ts

@@ -123,32 +123,49 @@ export class XmlHelper {
     return max;
   }
 
+  static pushRelTargets(element: Element, prefix: string, targets: Target[]) {
+    const type = element.getAttribute('Type');
+    const target = element.getAttribute('Target');
+    const rId = element.getAttribute('Id');
+
+    const subtype = _.last(prefix.split('/'));
+    const relType = _.last(type.split('/'));
+
+    const matchNumber = target.match(/(\d+)/);
+    const stripNumber =
+      matchNumber && matchNumber[1] ? Number(matchNumber[1]) : 0;
+
+    if (XmlHelper.targetMatchesRelationship(relType, subtype, target, prefix)) {
+      targets.push({
+        file: target,
+        rId: rId,
+        number: stripNumber,
+        type: type,
+        subtype: subtype,
+      } as Target);
+    }
+  }
+
+  static targetMatchesRelationship(relType, subtype, target, prefix) {
+    if (relType === 'package') return true;
+
+    return relType === subtype && target.indexOf(prefix) === 0;
+  }
+
   static async getTargetsFromRelationships(
     archive: JSZip,
     path: string,
     prefix: string | string[],
     suffix?: string | RegExp,
   ): Promise<Target[]> {
     const prefixes = typeof prefix === 'string' ? [prefix] : prefix;
+
     return XmlHelper.getRelationships(
       archive,
       path,
-      (element: Element, rels: Target[]) => {
-        const target = element.getAttribute('Target');
+      (element: Element, targets: Target[]) => {
         prefixes.forEach((prefix) => {
-          const stripNumber = target
-            .replace(prefix, '')
-            .replace(suffix || '.xml', '');
-
-          // vd(stripNumber);
-          if (target.indexOf(prefix) === 0) {
-            rels.push({
-              file: target,
-              rId: element.getAttribute('Id'),
-              number: Number(stripNumber),
-              subtype: _.last(prefix.split('/')),
-            } as Target);
-          }
+          XmlHelper.pushRelTargets(element, prefix, targets);
         });
       },
     );

src/helper/xml-helper.ts

@@ -20,6 +20,7 @@ const setTable = ModifyTableHelper.setTable;
 
 import ModifyChartHelper from './helper/modify-chart-helper';
 const setChartData = ModifyChartHelper.setChartData;
+const setExtendedChartData = ModifyChartHelper.setExtendedChartData;
 const setChartVerticalLines = ModifyChartHelper.setChartVerticalLines;
 const setChartScatter = ModifyChartHelper.setChartScatter;
 const setChartBubbles = ModifyChartHelper.setChartBubbles;
@@ -89,6 +90,7 @@ export const modify = {
   adjustWidth,
   setTable,
   setChartData,
+  setExtendedChartData,
   setChartVerticalLines,
   setChartScatter,
   setChartCombo,