csv-writer/src/lib/csv-stringifiers/object.ts

Convert objects/arrays into a CSV string or write them into a CSV file

import {CsvStringifier} from './abstract';
import {FieldStringifier} from '../field-stringifier';
import {Field, ObjectHeaderItem, ObjectStringifierHeader} from '../record';
import {isObject, ObjectMap} from '../lang/object';

export class ObjectCsvStringifier extends CsvStringifier<ObjectMap<Field>> {

    constructor(fieldStringifier: FieldStringifier,
                private readonly header: ObjectStringifierHeader,
                recordDelimiter?: string,
                private readonly headerIdDelimiter?: string) {
        super(fieldStringifier, recordDelimiter);
    }

    protected getHeaderRecord(): string[] | null {
        if (!this.isObjectHeader) return null;
        return (this.header as ObjectHeaderItem[]).map(field => field.title);
    }

    protected getRecordAsArray(record: ObjectMap<Field>): Field[] {
        return this.fieldIds.map(fieldId => this.getNestedValue(record, fieldId));
    }

    private getNestedValue(obj: ObjectMap<Field>, key: string) {
        if (!this.headerIdDelimiter) return obj[key];
        return key.split(this.headerIdDelimiter).reduce((subObj, keyPart) => (subObj || {})[keyPart], obj);
    }

    private get fieldIds(): string[] {
        return this.isObjectHeader ? (this.header as ObjectHeaderItem[]).map(column => column.id) : (this.header as string[]);
    }

    private get isObjectHeader(): boolean {
        return isObject(this.header && this.header[0]);
    }
}