ngzorro 自定义指令操作 nz-table 列拖动缩放

创建指令

ng g d drap-table-column

import {AfterViewInit, Directive, ElementRef, Input} from '@angular/core';

@Directive({
    selector: '[appDrapTableColumn]'
})

export class DrapTableColumnDirective implements AfterViewInit {
    @Input('appDrapTableColumn') columnWidths: string[];
    @Input() usePixel = false;

    constructor(private $element: ElementRef<HTMLTableElement>) {
    }

    ngAfterViewInit() {
        const el = (() => {
            let result = this.$element.nativeElement;
            if (result.tagName === 'NZ-TABLE') {
                result = result.querySelector('table');
            }
            if (result.tagName !== 'TABLE') {
                throw new TypeError('Must be a TABLE element');
            }
            if (!result.tHead) {
                throw new TypeError('Must have a THEAD element');
            }
            return result;
        })();
        el.classList.add('table-resizable');
        const tr = el.tHead.rows[0];
        if (!tr) {
            throw new TypeError('Must have at least one TR element inside THEAD element');
        }

        setTimeout(() => {
            const ths = Array.from(tr.cells) as HTMLTableHeaderCellElement[];
            if (ths.length <= 1) { return; }
            --ths.length; // 最后一个方格总是自动列宽的（用于占满整行）

            if (!Array.isArray(this.columnWidths)) {
                this.columnWidths = new Array(ths.length).fill('');
            } else {
                this.columnWidths.length = ths.length;
                this.columnWidths.forEach((x, i, arr) => {
                    if (!x) {
                        arr[i] = '';
                    }
                });
            }

            ths.forEach(th => {
                if (th.classList.contains('no-resize')) {
                    return;
                }
                if (this.columnWidths[th.cellIndex]) {
                    th.width = this.columnWidths[th.cellIndex];
                }

                const i = document.createElement('i');
                i.classList.add('resize-indicator');
                th.appendChild(i);

                i.addEventListener('mousedown', e => {
                    if (e.button === 1) { // 鼠标中键
                        th.width = '';
                        return;
                    }
                    if (e.button !== 0) {
                        return;
                    }
                    const startX = e.pageX;
                    const startThWidth = th.clientWidth;
                    document.body.classList.add('table-resizing');

                    let mousemoveHandler;

                    document.body.addEventListener('mousemove', mousemoveHandler = e => {
                        if (e.button !== 0) {
                            return;
                        }
                        const pixel = e.pageX - startX + startThWidth;
                        if (this.usePixel) {
                            th.width = pixel + '';
                        } else {
                            th.width = pixel / tr.offsetWidth * 100 + '%';
                        }
                        this.columnWidths[th.cellIndex] = th.width;
                    });

                    document.body.addEventListener('mouseup', e => {
                        if (e.button !== 0) {
                            return;
                        }
                        document.body.removeEventListener('mousemove', mousemoveHandler);
                        document.body.classList.remove('table-resizing');
                    }, { once: true });
                });
            });
        });
    }
}

缩放样式

.table-resizable th {
    position: relative;
}

body.table-resizing {
    cursor: col-resize !important;
    user-select: none;
}

.resize-indicator {
    display: block;
    position: absolute;
    top: 0;
    right: -5px;
    width: 9px;
    height: 100%;
    cursor: col-resize;
}

在app.module.ts中引入自定义指令

import {NgModule} from '@angular/core';
import {DrapTableColumnDirective} from './drap-table-column.directive';

@NgModule({
    declarations: [DrapTableColumnDirective],
    exports: [DrapTableColumnDirective],
    providers: []
})
export class DirectiveModule { }

在app.module.ts中引入自定义指令

imports: [
	DirectiveModule
],

使用

<nz-table [appDrapTableColumn]></nz-table>