You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							466 lines
						
					
					
						
							12 KiB
						
					
					
				
			
		
		
	
	
							466 lines
						
					
					
						
							12 KiB
						
					
					
				| /* eslint-disable max-classes-per-file */
 | |
| const colCache = require('../utils/col-cache');
 | |
| 
 | |
| class Column {
 | |
|   // wrapper around column model, allowing access and manipulation
 | |
|   constructor(table, column, index) {
 | |
|     this.table = table;
 | |
|     this.column = column;
 | |
|     this.index = index;
 | |
|   }
 | |
| 
 | |
|   _set(name, value) {
 | |
|     this.table.cacheState();
 | |
|     this.column[name] = value;
 | |
|   }
 | |
| 
 | |
|   /* eslint-disable lines-between-class-members */
 | |
|   get name() {
 | |
|     return this.column.name;
 | |
|   }
 | |
|   set name(value) {
 | |
|     this._set('name', value);
 | |
|   }
 | |
| 
 | |
|   get filterButton() {
 | |
|     return this.column.filterButton;
 | |
|   }
 | |
|   set filterButton(value) {
 | |
|     this.column.filterButton = value;
 | |
|   }
 | |
| 
 | |
|   get style() {
 | |
|     return this.column.style;
 | |
|   }
 | |
|   set style(value) {
 | |
|     this.column.style = value;
 | |
|   }
 | |
| 
 | |
|   get totalsRowLabel() {
 | |
|     return this.column.totalsRowLabel;
 | |
|   }
 | |
|   set totalsRowLabel(value) {
 | |
|     this._set('totalsRowLabel', value);
 | |
|   }
 | |
| 
 | |
|   get totalsRowFunction() {
 | |
|     return this.column.totalsRowFunction;
 | |
|   }
 | |
|   set totalsRowFunction(value) {
 | |
|     this._set('totalsRowFunction', value);
 | |
|   }
 | |
| 
 | |
|   get totalsRowResult() {
 | |
|     return this.column.totalsRowResult;
 | |
|   }
 | |
|   set totalsRowResult(value) {
 | |
|     this._set('totalsRowResult', value);
 | |
|   }
 | |
| 
 | |
|   get totalsRowFormula() {
 | |
|     return this.column.totalsRowFormula;
 | |
|   }
 | |
|   set totalsRowFormula(value) {
 | |
|     this._set('totalsRowFormula', value);
 | |
|   }
 | |
|   /* eslint-enable lines-between-class-members */
 | |
| }
 | |
| 
 | |
| class Table {
 | |
|   constructor(worksheet, table) {
 | |
|     this.worksheet = worksheet;
 | |
|     if (table) {
 | |
|       this.table = table;
 | |
|       // check things are ok first
 | |
|       this.validate();
 | |
| 
 | |
|       this.store();
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   getFormula(column) {
 | |
|     // get the correct formula to apply to the totals row
 | |
|     switch (column.totalsRowFunction) {
 | |
|       case 'none':
 | |
|         return null;
 | |
|       case 'average':
 | |
|         return `SUBTOTAL(101,${this.table.name}[${column.name}])`;
 | |
|       case 'countNums':
 | |
|         return `SUBTOTAL(102,${this.table.name}[${column.name}])`;
 | |
|       case 'count':
 | |
|         return `SUBTOTAL(103,${this.table.name}[${column.name}])`;
 | |
|       case 'max':
 | |
|         return `SUBTOTAL(104,${this.table.name}[${column.name}])`;
 | |
|       case 'min':
 | |
|         return `SUBTOTAL(105,${this.table.name}[${column.name}])`;
 | |
|       case 'stdDev':
 | |
|         return `SUBTOTAL(106,${this.table.name}[${column.name}])`;
 | |
|       case 'var':
 | |
|         return `SUBTOTAL(107,${this.table.name}[${column.name}])`;
 | |
|       case 'sum':
 | |
|         return `SUBTOTAL(109,${this.table.name}[${column.name}])`;
 | |
|       case 'custom':
 | |
|         return column.totalsRowFormula;
 | |
|       default:
 | |
|         throw new Error(`Invalid Totals Row Function: ${column.totalsRowFunction}`);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   get width() {
 | |
|     // width of the table
 | |
|     return this.table.columns.length;
 | |
|   }
 | |
| 
 | |
|   get height() {
 | |
|     // height of the table data
 | |
|     return this.table.rows.length;
 | |
|   }
 | |
| 
 | |
|   get filterHeight() {
 | |
|     // height of the table data plus optional header row
 | |
|     return this.height + (this.table.headerRow ? 1 : 0);
 | |
|   }
 | |
| 
 | |
|   get tableHeight() {
 | |
|     // full height of the table on the sheet
 | |
|     return this.filterHeight + (this.table.totalsRow ? 1 : 0);
 | |
|   }
 | |
| 
 | |
|   validate() {
 | |
|     const {table} = this;
 | |
|     // set defaults and check is valid
 | |
|     const assign = (o, name, dflt) => {
 | |
|       if (o[name] === undefined) {
 | |
|         o[name] = dflt;
 | |
|       }
 | |
|     };
 | |
|     assign(table, 'headerRow', true);
 | |
|     assign(table, 'totalsRow', false);
 | |
| 
 | |
|     assign(table, 'style', {});
 | |
|     assign(table.style, 'theme', 'TableStyleMedium2');
 | |
|     assign(table.style, 'showFirstColumn', false);
 | |
|     assign(table.style, 'showLastColumn', false);
 | |
|     assign(table.style, 'showRowStripes', false);
 | |
|     assign(table.style, 'showColumnStripes', false);
 | |
| 
 | |
|     const assert = (test, message) => {
 | |
|       if (!test) {
 | |
|         throw new Error(message);
 | |
|       }
 | |
|     };
 | |
|     assert(table.ref, 'Table must have ref');
 | |
|     assert(table.columns, 'Table must have column definitions');
 | |
|     assert(table.rows, 'Table must have row definitions');
 | |
| 
 | |
|     table.tl = colCache.decodeAddress(table.ref);
 | |
|     const {row, col} = table.tl;
 | |
|     assert(row > 0, 'Table must be on valid row');
 | |
|     assert(col > 0, 'Table must be on valid col');
 | |
| 
 | |
|     const {width, filterHeight, tableHeight} = this;
 | |
| 
 | |
|     // autoFilterRef is a range that includes optional headers only
 | |
|     table.autoFilterRef = colCache.encode(row, col, row + filterHeight - 1, col + width - 1);
 | |
| 
 | |
|     // tableRef is a range that includes optional headers and totals
 | |
|     table.tableRef = colCache.encode(row, col, row + tableHeight - 1, col + width - 1);
 | |
| 
 | |
|     table.columns.forEach((column, i) => {
 | |
|       assert(column.name, `Column ${i} must have a name`);
 | |
|       if (i === 0) {
 | |
|         assign(column, 'totalsRowLabel', 'Total');
 | |
|       } else {
 | |
|         assign(column, 'totalsRowFunction', 'none');
 | |
|         column.totalsRowFormula = this.getFormula(column);
 | |
|       }
 | |
|     });
 | |
|   }
 | |
| 
 | |
|   store() {
 | |
|     // where the table needs to store table data, headers, footers in
 | |
|     // the sheet...
 | |
|     const assignStyle = (cell, style) => {
 | |
|       if (style) {
 | |
|         Object.keys(style).forEach(key => {
 | |
|           cell[key] = style[key];
 | |
|         });
 | |
|       }
 | |
|     };
 | |
| 
 | |
|     const {worksheet, table} = this;
 | |
|     const {row, col} = table.tl;
 | |
|     let count = 0;
 | |
|     if (table.headerRow) {
 | |
|       const r = worksheet.getRow(row + count++);
 | |
|       table.columns.forEach((column, j) => {
 | |
|         const {style, name} = column;
 | |
|         const cell = r.getCell(col + j);
 | |
|         cell.value = name;
 | |
|         assignStyle(cell, style);
 | |
|       });
 | |
|     }
 | |
|     table.rows.forEach(data => {
 | |
|       const r = worksheet.getRow(row + count++);
 | |
|       data.forEach((value, j) => {
 | |
|         const cell = r.getCell(col + j);
 | |
|         cell.value = value;
 | |
| 
 | |
|         assignStyle(cell, table.columns[j].style);
 | |
|       });
 | |
|     });
 | |
| 
 | |
|     if (table.totalsRow) {
 | |
|       const r = worksheet.getRow(row + count++);
 | |
|       table.columns.forEach((column, j) => {
 | |
|         const cell = r.getCell(col + j);
 | |
|         if (j === 0) {
 | |
|           cell.value = column.totalsRowLabel;
 | |
|         } else {
 | |
|           const formula = this.getFormula(column);
 | |
|           if (formula) {
 | |
|             cell.value = {
 | |
|               formula: column.totalsRowFormula,
 | |
|               result: column.totalsRowResult,
 | |
|             };
 | |
|           } else {
 | |
|             cell.value = null;
 | |
|           }
 | |
|         }
 | |
| 
 | |
|         assignStyle(cell, column.style);
 | |
|       });
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   load(worksheet) {
 | |
|     // where the table will read necessary features from a loaded sheet
 | |
|     const {table} = this;
 | |
|     const {row, col} = table.tl;
 | |
|     let count = 0;
 | |
|     if (table.headerRow) {
 | |
|       const r = worksheet.getRow(row + count++);
 | |
|       table.columns.forEach((column, j) => {
 | |
|         const cell = r.getCell(col + j);
 | |
|         cell.value = column.name;
 | |
|       });
 | |
|     }
 | |
|     table.rows.forEach(data => {
 | |
|       const r = worksheet.getRow(row + count++);
 | |
|       data.forEach((value, j) => {
 | |
|         const cell = r.getCell(col + j);
 | |
|         cell.value = value;
 | |
|       });
 | |
|     });
 | |
| 
 | |
|     if (table.totalsRow) {
 | |
|       const r = worksheet.getRow(row + count++);
 | |
|       table.columns.forEach((column, j) => {
 | |
|         const cell = r.getCell(col + j);
 | |
|         if (j === 0) {
 | |
|           cell.value = column.totalsRowLabel;
 | |
|         } else {
 | |
|           const formula = this.getFormula(column);
 | |
|           if (formula) {
 | |
|             cell.value = {
 | |
|               formula: column.totalsRowFormula,
 | |
|               result: column.totalsRowResult,
 | |
|             };
 | |
|           }
 | |
|         }
 | |
|       });
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   get model() {
 | |
|     return this.table;
 | |
|   }
 | |
| 
 | |
|   set model(value) {
 | |
|     this.table = value;
 | |
|   }
 | |
| 
 | |
|   // ================================================================
 | |
|   // TODO: Mutating methods
 | |
|   cacheState() {
 | |
|     if (!this._cache) {
 | |
|       this._cache = {
 | |
|         ref: this.ref,
 | |
|         width: this.width,
 | |
|         tableHeight: this.tableHeight,
 | |
|       };
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   commit() {
 | |
|     // changes may have been made that might have on-sheet effects
 | |
|     if (!this._cache) {
 | |
|       return;
 | |
|     }
 | |
| 
 | |
|     // check things are ok first
 | |
|     this.validate();
 | |
| 
 | |
|     const ref = colCache.decodeAddress(this._cache.ref);
 | |
|     if (this.ref !== this._cache.ref) {
 | |
|       // wipe out whole table footprint at previous location
 | |
|       for (let i = 0; i < this._cache.tableHeight; i++) {
 | |
|         const row = this.worksheet.getRow(ref.row + i);
 | |
|         for (let j = 0; j < this._cache.width; j++) {
 | |
|           const cell = row.getCell(ref.col + j);
 | |
|           cell.value = null;
 | |
|         }
 | |
|       }
 | |
|     } else {
 | |
|       // clear out below table if it has shrunk
 | |
|       for (let i = this.tableHeight; i < this._cache.tableHeight; i++) {
 | |
|         const row = this.worksheet.getRow(ref.row + i);
 | |
|         for (let j = 0; j < this._cache.width; j++) {
 | |
|           const cell = row.getCell(ref.col + j);
 | |
|           cell.value = null;
 | |
|         }
 | |
|       }
 | |
| 
 | |
|       // clear out to right of table if it has lost columns
 | |
|       for (let i = 0; i < this.tableHeight; i++) {
 | |
|         const row = this.worksheet.getRow(ref.row + i);
 | |
|         for (let j = this.width; j < this._cache.width; j++) {
 | |
|           const cell = row.getCell(ref.col + j);
 | |
|           cell.value = null;
 | |
|         }
 | |
|       }
 | |
|     }
 | |
| 
 | |
|     this.store();
 | |
|   }
 | |
| 
 | |
|   addRow(values, rowNumber) {
 | |
|     // Add a row of data, either insert at rowNumber or append
 | |
|     this.cacheState();
 | |
| 
 | |
|     if (rowNumber === undefined) {
 | |
|       this.table.rows.push(values);
 | |
|     } else {
 | |
|       this.table.rows.splice(rowNumber, 0, values);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   removeRows(rowIndex, count = 1) {
 | |
|     // Remove a rows of data
 | |
|     this.cacheState();
 | |
|     this.table.rows.splice(rowIndex, count);
 | |
|   }
 | |
| 
 | |
|   getColumn(colIndex) {
 | |
|     const column = this.table.columns[colIndex];
 | |
|     return new Column(this, column, colIndex);
 | |
|   }
 | |
| 
 | |
|   addColumn(column, values, colIndex) {
 | |
|     // Add a new column, including column defn and values
 | |
|     // Inserts at colNumber or adds to the right
 | |
|     this.cacheState();
 | |
| 
 | |
|     if (colIndex === undefined) {
 | |
|       this.table.columns.push(column);
 | |
|       this.table.rows.forEach((row, i) => {
 | |
|         row.push(values[i]);
 | |
|       });
 | |
|     } else {
 | |
|       this.table.columns.splice(colIndex, 0, column);
 | |
|       this.table.rows.forEach((row, i) => {
 | |
|         row.splice(colIndex, 0, values[i]);
 | |
|       });
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   removeColumns(colIndex, count = 1) {
 | |
|     // Remove a column with data
 | |
|     this.cacheState();
 | |
| 
 | |
|     this.table.columns.splice(colIndex, count);
 | |
|     this.table.rows.forEach(row => {
 | |
|       row.splice(colIndex, count);
 | |
|     });
 | |
|   }
 | |
| 
 | |
|   _assign(target, prop, value) {
 | |
|     this.cacheState();
 | |
|     target[prop] = value;
 | |
|   }
 | |
| 
 | |
|   /* eslint-disable lines-between-class-members */
 | |
|   get ref() {
 | |
|     return this.table.ref;
 | |
|   }
 | |
|   set ref(value) {
 | |
|     this._assign(this.table, 'ref', value);
 | |
|   }
 | |
| 
 | |
|   get name() {
 | |
|     return this.table.name;
 | |
|   }
 | |
|   set name(value) {
 | |
|     this.table.name = value;
 | |
|   }
 | |
| 
 | |
|   get displayName() {
 | |
|     return this.table.displyName || this.table.name;
 | |
|   }
 | |
|   set displayNamename(value) {
 | |
|     this.table.displayName = value;
 | |
|   }
 | |
| 
 | |
|   get headerRow() {
 | |
|     return this.table.headerRow;
 | |
|   }
 | |
|   set headerRow(value) {
 | |
|     this._assign(this.table, 'headerRow', value);
 | |
|   }
 | |
| 
 | |
|   get totalsRow() {
 | |
|     return this.table.totalsRow;
 | |
|   }
 | |
|   set totalsRow(value) {
 | |
|     this._assign(this.table, 'totalsRow', value);
 | |
|   }
 | |
| 
 | |
|   get theme() {
 | |
|     return this.table.style.name;
 | |
|   }
 | |
|   set theme(value) {
 | |
|     this.table.style.name = value;
 | |
|   }
 | |
| 
 | |
|   get showFirstColumn() {
 | |
|     return this.table.style.showFirstColumn;
 | |
|   }
 | |
|   set showFirstColumn(value) {
 | |
|     this.table.style.showFirstColumn = value;
 | |
|   }
 | |
| 
 | |
|   get showLastColumn() {
 | |
|     return this.table.style.showLastColumn;
 | |
|   }
 | |
|   set showLastColumn(value) {
 | |
|     this.table.style.showLastColumn = value;
 | |
|   }
 | |
| 
 | |
|   get showRowStripes() {
 | |
|     return this.table.style.showRowStripes;
 | |
|   }
 | |
|   set showRowStripes(value) {
 | |
|     this.table.style.showRowStripes = value;
 | |
|   }
 | |
| 
 | |
|   get showColumnStripes() {
 | |
|     return this.table.style.showColumnStripes;
 | |
|   }
 | |
|   set showColumnStripes(value) {
 | |
|     this.table.style.showColumnStripes = value;
 | |
|   }
 | |
|   /* eslint-enable lines-between-class-members */
 | |
| }
 | |
| 
 | |
| module.exports = Table;
 |