books/client/view/controls/table.js

const frappe = require('frappejs');
const BaseControl = require('./base');
const DataTable = require('frappe-datatable');
const controls = require('./index');
const Modal = require('frappejs/client/ui/modal');

class TableControl extends BaseControl {
    make() {
        if (!this.datatable) {
            this.wrapper = frappe.ui.add('div', 'table-wrapper', this.get_input_parent());
            this.wrapper.innerHTML =
            `<div class="datatable-wrapper"></div>
            <div class="table-toolbar">
                <button class="btn btn-sm btn-outline-secondary btn-add">${frappe._("Add")}</button>
                <button class="btn btn-sm btn-outline-danger btn-remove">${frappe._("Remove")}</button>
            </div>`;

            this.datatable = new DataTable(this.wrapper.querySelector('.datatable-wrapper'), {
                columns: this.get_columns(),
                data: this.get_table_data(),
                takeAvailableSpace: true,
                enableClusterize: true,
                addCheckboxColumn: true,
                editing: this.get_table_input.bind(this),
            });

            this.wrapper.querySelector('.btn-add').addEventListener('click', async (event) => {
                this.doc[this.fieldname].push({});
                await this.doc.commit();
                this.refresh();
            });

            this.wrapper.querySelector('.btn-remove').addEventListener('click', async (event) => {
                let checked = this.datatable.rowmanager.getCheckedRows();
                this.doc[this.fieldname] = this.doc[this.fieldname].filter(d => !checked.includes(d.idx));
                await this.doc.commit();
                this.refresh();
                this.datatable.rowmanager.checkAll(false);
            });
        }
    }

    get_input_value() {
        return this.doc[this.fieldname];
    }

    set_input_value(value) {
        this.datatable.refresh(this.get_table_data(value));
    }

    get_table_data(value) {
        return value || this.get_default_data();
    }

    get_table_input(colIndex, rowIndex, value, parent) {
        let field = this.datatable.getColumn(colIndex).field;

        if (field.fieldtype==='Text') {
            return this.get_text_control(field);
        } else {
            return this.get_control(field, parent);
        }
    }

    get_control(field, parent) {
        field.only_input = true;
        const control = controls.make_control({field: field, parent: parent});

        return {
            initValue: (value, rowIndex, column) => {
                control.parent_control = this;
                control.doc = this.doc[this.fieldname][rowIndex];
                control.set_focus();
                return control.set_input_value(value);
            },
            setValue: (value, rowIndex, column) => {
                return control.set_input_value(value);
            },
            getValue: () => {
                return control.get_input_value();
            }
        }

    }

    get_text_control(field, parent) {

        this.text_modal = new Modal({
            title: frappe._('Edit {0}', field.label),
            body: '',
            primary_label: frappe._('Submit'),
            primary_action: (modal) => {
                this.datatable.cellmanager.submitEditing();
                modal.hide();
            }
        });
        this.text_modal.$modal.on('hidden.bs.modal', () => {
            this.datatable.cellmanager.deactivateEditing();
        })

        return this.get_control(field, this.text_modal.get_body());
    }

    get_columns() {
        return this.get_child_fields().map(field => {
            return {
                id: field.fieldname,
                field: field,
                content: field.label,
                editable: true,
                width: 120,
                align: ['Int', 'Float', 'Currency'].includes(field.fieldtype) ? 'right' : 'left',
                format: (value) => frappe.format(value, field)
            }
        });
    }

    get_child_fields() {
        return frappe.get_meta(this.childtype).fields;
    }

    get_default_data() {
        // build flat table
        if (!this.doc) {
            return [];
        }
        if (!this.doc[this.fieldname]) {
            this.doc[this.fieldname] = [{}];
        }

        return this.doc[this.fieldname];
    }
};

module.exports = TableControl;