/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { Injectable, OnDestroy, inject } from '@angular/core';
import { Store } from '@ngrx/store';
import { CanvasState } from '../../state';
import { CanvasUtils } from '../canvas-utils.service';
import { PositionBehavior } from '../behavior/position-behavior.service';
import { SelectableBehavior } from '../behavior/selectable-behavior.service';
import { EditableBehavior } from '../behavior/editable-behavior.service';
import * as d3 from 'd3';
import {
    selectFlowLoadingStatus,
    selectProcessors,
    selectAnySelectedComponentIds,
    selectTransitionRequired
} from '../../state/flow/flow.selectors';
import { QuickSelectBehavior } from '../behavior/quick-select-behavior.service';
import { ValidationErrorsTip } from '../../../../ui/common/tooltips/validation-errors-tip/validation-errors-tip.component';
import { ComponentType, TextTip, NiFiCommon } from '@nifi/shared';
import { Dimension } from '../../state/shared';
import { filter, Subject, switchMap, takeUntil } from 'rxjs';

@Injectable({
    providedIn: 'root'
})
export class ProcessorManager implements OnDestroy {
    private store = inject<Store<CanvasState>>(Store);
    private canvasUtils = inject(CanvasUtils);
    private nifiCommon = inject(NiFiCommon);
    private positionBehavior = inject(PositionBehavior);
    private selectableBehavior = inject(SelectableBehavior);
    private quickSelectBehavior = inject(QuickSelectBehavior);
    private editableBehavior = inject(EditableBehavior);

    private destroyed$: Subject<boolean> = new Subject();

    private dimensions: Dimension = {
        width: 352,
        height: 128
    };

    private static readonly PREVIEW_NAME_LENGTH: number = 25;

    private processors: [] = [];
    private processorContainer: any = null;
    private transitionRequired = false;

    private select() {
        return this.processorContainer.selectAll('g.processor').data(this.processors, function (d: any) {
            return d.id;
        });
    }

    private renderProcessors(entered: any) {
        if (entered.empty()) {
            return entered;
        }

        const processor = entered
            .append('g')
            .attr('id', function (d: any) {
                return 'id-' + d.id;
            })
            .attr('class', 'processor component');

        // processor border
        processor
            .append('rect')
            .attr('class', 'border')
            .attr('width', function (d: any) {
                return d.dimensions.width;
            })
            .attr('height', function (d: any) {
                return d.dimensions.height;
            })
            .attr('fill', 'transparent')
            .attr('stroke', 'transparent');

        // processor body
        processor
            .append('rect')
            .attr('class', 'body')
            .attr('width', function (d: any) {
                return d.dimensions.width;
            })
            .attr('height', function (d: any) {
                return d.dimensions.height;
            })
            .attr('filter', 'url(#component-drop-shadow)')
            .attr('stroke-width', 0);

        // processor name
        processor
            .append('text')
            .attr('x', 75)
            .attr('y', 18)
            .attr('width', 230)
            .attr('height', 14)
            .attr('class', 'processor-name');

        // processor icon container
        processor
            .append('rect')
            .attr('x', 0)
            .attr('y', 0)
            .attr('width', 50)
            .attr('height', 50)
            .attr('class', 'processor-icon-container');

        // processor icon
        processor.append('text').attr('x', 9).attr('y', 35).attr('class', 'processor-icon').text('\ue807');

        // restricted icon background
        processor.append('circle').attr('r', 9).attr('cx', 12).attr('cy', 12).attr('class', 'restricted-background');

        // restricted icon
        processor.append('text').attr('x', 7.75).attr('y', 17).attr('class', 'restricted').text('\uf132');

        // is primary icon background
        processor.append('circle').attr('r', 9).attr('cx', 38).attr('cy', 36).attr('class', 'is-primary-background');

        // is primary icon
        processor
            .append('text')
            .attr('x', 34.75)
            .attr('y', 40)
            .attr('class', 'is-primary')
            .text('P')
            .append('title')
            .text(function () {
                return 'This component is only scheduled to execute on the Primary Node';
            });

        this.selectableBehavior.activate(processor);
        this.quickSelectBehavior.activate(processor);

        return processor;
    }

    private updateProcessors(updated: d3.Selection<any, any, any, any>) {
        if (updated.empty()) {
            return;
        }

        // processor border authorization
        updated
            .select('rect.border')
            .classed('unauthorized', (d: any) => {
                return d.permissions.canRead === false;
            })
            .classed('ghost', (d: any) => {
                return d.permissions.canRead === true && d.component.extensionMissing === true;
            });

        // processor body authorization
        updated.select('rect.body').classed('unauthorized', (d: any) => {
            return d.permissions.canRead === false;
        });

        updated.each((processorData: any, i, nodes) => {
            const processor: d3.Selection<any, any, any, any> = d3.select(nodes[i]);
            let details: any = processor.select('g.processor-canvas-details');

            // update the component behavior as appropriate
            this.editableBehavior.editable(processor);

            // if this processor is visible, render everything
            if (processor.classed('visible')) {
                if (details.empty()) {
                    details = processor.append('g').attr('class', 'processor-canvas-details');

                    // run status icon
                    details
                        .append('text')
                        .attr('class', 'run-status-icon')
                        .attr('x', 55)
                        .attr('y', 23)
                        .attr('width', 14)
                        .attr('height', 14);

                    // processor type
                    details
                        .append('text')
                        .attr('class', 'processor-type')
                        .attr('x', 75)
                        .attr('y', 32)
                        .attr('width', 230)
                        .attr('height', 12);

                    // processor type
                    details
                        .append('text')
                        .attr('class', 'processor-bundle')
                        .attr('x', 75)
                        .attr('y', 45)
                        .attr('width', 200)
                        .attr('height', 12);

                    // -----
                    // stats
                    // -----

                    // draw the processor statistics table

                    // in
                    details
                        .append('rect')
                        .attr('class', 'processor-stats-in-out odd')
                        .attr('width', () => {
                            return processorData.dimensions.width;
                        })
                        .attr('height', 19)
                        .attr('x', 0)
                        .attr('y', 50);

                    // border
                    details
                        .append('rect')
                        .attr('class', 'processor-stats-border')
                        .attr('width', () => {
                            return processorData.dimensions.width;
                        })
                        .attr('height', 1)
                        .attr('x', 0)
                        .attr('y', 68);

                    // read/write
                    details
                        .append('rect')
                        .attr('class', 'processor-read-write-stats even')
                        .attr('width', () => {
                            return processorData.dimensions.width;
                        })
                        .attr('height', 19)
                        .attr('x', 0)
                        .attr('y', 69);

                    // border
                    details
                        .append('rect')
                        .attr('class', 'processor-stats-border')
                        .attr('width', () => {
                            return processorData.dimensions.width;
                        })
                        .attr('height', 1)
                        .attr('x', 0)
                        .attr('y', 87);

                    // out
                    details
                        .append('rect')
                        .attr('class', 'processor-stats-in-out odd')
                        .attr('width', () => {
                            return processorData.dimensions.width;
                        })
                        .attr('height', 20)
                        .attr('x', 0)
                        .attr('y', 88);

                    // border
                    details
                        .append('rect')
                        .attr('class', 'processor-stats-border')
                        .attr('width', () => {
                            return processorData.dimensions.width;
                        })
                        .attr('height', 1)
                        .attr('x', 0)
                        .attr('y', 106);

                    // tasks/time
                    details
                        .append('rect')
                        .attr('class', 'processor-read-write-stats even')
                        .attr('width', () => {
                            return processorData.dimensions.width;
                        })
                        .attr('height', 19)
                        .attr('x', 0)
                        .attr('y', 107);

                    // stats label container
                    const processorStatsLabel = details.append('g').attr('transform', 'translate(10, 55)');

                    // in label
                    processorStatsLabel
                        .append('text')
                        .attr('width', 73)
                        .attr('height', 10)
                        .attr('y', 9)
                        .attr('class', 'stats-label')
                        .text('In');

                    // read/write label
                    processorStatsLabel
                        .append('text')
                        .attr('width', 73)
                        .attr('height', 10)
                        .attr('y', 27)
                        .attr('class', 'stats-label')
                        .text('Read/Write');

                    // out label
                    processorStatsLabel
                        .append('text')
                        .attr('width', 73)
                        .attr('height', 10)
                        .attr('y', 46)
                        .attr('class', 'stats-label')
                        .text('Out');

                    // tasks/time label
                    processorStatsLabel
                        .append('text')
                        .attr('width', 73)
                        .attr('height', 10)
                        .attr('y', 65)
                        .attr('class', 'stats-label')
                        .text('Tasks/Time');

                    // stats value container
                    const processorStatsValue = details.append('g').attr('transform', 'translate(85, 55)');

                    // in value
                    const inText = processorStatsValue
                        .append('text')
                        .attr('width', 180)
                        .attr('height', 9)
                        .attr('y', 9)
                        .attr('class', 'processor-in stats-value');

                    // in count
                    inText.append('tspan').attr('class', 'count');

                    // in size
                    inText.append('tspan').attr('class', 'size');

                    // read/write value
                    processorStatsValue
                        .append('text')
                        .attr('width', 180)
                        .attr('height', 10)
                        .attr('y', 27)
                        .attr('class', 'processor-read-write stats-value');

                    // out value
                    const outText = processorStatsValue
                        .append('text')
                        .attr('width', 180)
                        .attr('height', 10)
                        .attr('y', 46)
                        .attr('class', 'processor-out stats-value');

                    // out count
                    outText.append('tspan').attr('class', 'count');

                    // out size
                    outText.append('tspan').attr('class', 'size');

                    // tasks/time value
                    processorStatsValue
                        .append('text')
                        .attr('width', 180)
                        .attr('height', 10)
                        .attr('y', 65)
                        .attr('class', 'processor-tasks-time stats-value');

                    // stats value container
                    const processorStatsInfo = details.append('g').attr('transform', 'translate(305, 55)');

                    // in info
                    processorStatsInfo
                        .append('text')
                        .attr('width', 25)
                        .attr('height', 10)
                        .attr('y', 9)
                        .attr('class', 'stats-info')
                        .text('5 min');

                    // read/write info
                    processorStatsInfo
                        .append('text')
                        .attr('width', 25)
                        .attr('height', 10)
                        .attr('y', 27)
                        .attr('class', 'stats-info')
                        .text('5 min');

                    // out info
                    processorStatsInfo
                        .append('text')
                        .attr('width', 25)
                        .attr('height', 10)
                        .attr('y', 46)
                        .attr('class', 'stats-info')
                        .text('5 min');

                    // tasks/time info
                    processorStatsInfo
                        .append('text')
                        .attr('width', 25)
                        .attr('height', 10)
                        .attr('y', 65)
                        .attr('class', 'stats-info')
                        .text('5 min');

                    // --------
                    // comments
                    // --------

                    details
                        .append('text')
                        .attr('class', 'component-comments')
                        .attr(
                            'transform',
                            'translate(' +
                                (processorData.dimensions.width - 11) +
                                ', ' +
                                (processorData.dimensions.height - 3) +
                                ')'
                        )
                        .text('\uf075');

                    // -------------------
                    // active thread count
                    // -------------------

                    // active thread count
                    details.append('text').attr('class', 'active-thread-count-icon').attr('y', 46).text('\ue83f');

                    // active thread background
                    details.append('text').attr('class', 'active-thread-count').attr('y', 46);

                    // ---------
                    // bulletins
                    // ---------

                    // bulletin background
                    details
                        .append('rect')
                        .attr('class', 'bulletin-background')
                        .attr('x', () => {
                            return processorData.dimensions.width - 24;
                        })
                        .attr('width', 24)
                        .attr('height', 24);

                    // bulletin icon
                    details
                        .append('text')
                        .attr('class', 'bulletin-icon')
                        .attr('x', () => {
                            return processorData.dimensions.width - 17;
                        })
                        .attr('y', 17)
                        .text('\uf24a');
                }

                if (processorData.permissions.canRead) {
                    // update the processor name
                    processor
                        .select('text.processor-name')
                        .each((d: any, i, nodes) => {
                            const processorName = d3.select(nodes[i]);

                            // reset the processor name to handle any previous state
                            processorName.text(null).selectAll('title').remove();

                            // apply ellipsis to the processor name as necessary
                            this.canvasUtils.ellipsis(processorName, d.component.name, 'processor-name');
                        })
                        .append('title')
                        .text((d: any) => {
                            return d.component.name;
                        });

                    // update the processor type
                    processor
                        .select('text.processor-type')
                        .each((d: any, i, nodes) => {
                            const processorType = d3.select(nodes[i]);

                            // reset the processor type to handle any previous state
                            processorType.text(null).selectAll('title').remove();

                            // apply ellipsis to the processor type as necessary
                            this.canvasUtils.ellipsis(
                                processorType,
                                this.nifiCommon.formatType(d.component),
                                'processor-type'
                            );
                        })
                        .append('title')
                        .text((d: any) => {
                            return this.nifiCommon.formatType(d.component);
                        });

                    // update the processor bundle
                    processor
                        .select('text.processor-bundle')
                        .each((d: any, i, nodes) => {
                            const processorBundle = d3.select(nodes[i]);

                            // reset the processor type to handle any previous state
                            processorBundle.text(null).selectAll('title').remove();

                            // apply ellipsis to the processor type as necessary
                            this.canvasUtils.ellipsis(
                                processorBundle,
                                this.nifiCommon.formatBundle(d.component.bundle),
                                'processor-bundle'
                            );
                        })
                        .append('title')
                        .text((d: any) => {
                            return this.nifiCommon.formatBundle(d.component.bundle);
                        });

                    // update the processor comments
                    processor
                        .select('text.component-comments')
                        .style(
                            'visibility',
                            this.nifiCommon.isBlank(processorData.component.config.comments) ? 'hidden' : 'visible'
                        )
                        .each((d: any, i, nodes) => {
                            if (!this.nifiCommon.isBlank(processorData.component.config.comments)) {
                                this.canvasUtils.canvasTooltip(
                                    TextTip,
                                    d3.select(nodes[i]),
                                    processorData.component.config.comments
                                );
                            } else {
                                this.canvasUtils.resetCanvasTooltip(d3.select(nodes[i]));
                            }
                        });
                } else {
                    // clear the processor name
                    processor.select('text.processor-name').text(null);

                    // clear the processor type
                    processor.select('text.processor-type').text(null);

                    // clear the processor bundle
                    processor.select('text.processor-bundle').text(null);

                    // clear the processor comments
                    processor.select('text.component-comments').style('visibility', 'hidden');
                }

                // populate the stats
                this.updateProcessorStatus(processor);
            } else {
                if (processorData.permissions.canRead) {
                    // update the processor name
                    processor.select('text.processor-name').text((d: any) => {
                        const name = d.component.name;
                        if (name.length > ProcessorManager.PREVIEW_NAME_LENGTH) {
                            return name.substring(0, ProcessorManager.PREVIEW_NAME_LENGTH) + String.fromCharCode(8230);
                        } else {
                            return name;
                        }
                    });
                } else {
                    // clear the processor name
                    processor.select('text.processor-name').text(null);
                }

                // remove the details if necessary
                if (!details.empty()) {
                    details.remove();
                }
            }

            // ---------------
            // processor color
            // ---------------

            //update the processor icon container
            processor
                .select('rect.processor-icon-container')
                .classed('unauthorized', !processorData.permissions.canRead);

            //update the processor icon
            processor
                .select('text.processor-icon')
                .classed('unauthorized tertiary-color', !processorData.permissions.canRead);

            //update the processor border
            processor.select('rect.border').classed('unauthorized', !processorData.permissions.canRead);

            // use the specified color if appropriate
            if (processorData.permissions.canRead) {
                if (processorData.component.style['background-color']) {
                    const color = processorData.component.style['background-color'];

                    //update the processor icon container
                    processor.select('rect.processor-icon-container').style('fill', () => {
                        return color;
                    });

                    //update the processor border
                    processor.select('rect.border').style('stroke', () => {
                        return color;
                    });

                    if (color) {
                        processor.select('text.processor-icon').attr('class', () => {
                            return 'processor-icon';
                        });

                        // update the processor color
                        processor.style('fill', (d: any) => {
                            let color = 'unset';

                            if (!d.permissions.canRead) {
                                return color;
                            }

                            // use the specified color if appropriate
                            if (d.component.style['background-color']) {
                                color = d.component.style['background-color'];

                                color = this.canvasUtils.determineContrastColor(
                                    this.nifiCommon.substringAfterLast(color, '#')
                                );
                            }

                            return color;
                        });
                    }
                } else {
                    // undo changes made above
                    processor.select('text.processor-icon').attr('class', () => {
                        return 'processor-icon tertiary-color';
                    });
                    processor.select('rect.processor-icon-container').style('fill', null);
                    processor.select('rect.border').style('stroke', null);
                }
            }

            // restricted component indicator
            processor.select('circle.restricted-background').style('visibility', (d: any) => {
                return this.showRestricted(d);
            });
            processor.select('text.restricted').style('visibility', (d: any) => {
                return this.showRestricted(d);
            });

            // is primary component indicator
            processor.select('circle.is-primary-background').style('visibility', (d: any) => {
                return this.showIsPrimary(d);
            });
            processor.select('text.is-primary').style('visibility', (d: any) => {
                return this.showIsPrimary(d);
            });
        });
    }

    private updateProcessorStatus(updated: d3.Selection<any, any, any, any>) {
        if (updated.empty()) {
            return;
        }

        // update the run status
        updated
            .select('text.run-status-icon')
            .attr('class', (d: any) => {
                let clazz = 'primary-color';

                if (d.status.aggregateSnapshot.runStatus === 'Validating') {
                    clazz = 'validating neutral-color';
                } else if (d.status.aggregateSnapshot.runStatus === 'Invalid') {
                    clazz = 'invalid caution-color';
                } else if (d.status.aggregateSnapshot.runStatus === 'Running') {
                    clazz = 'running success-color-default';
                } else if (d.status.aggregateSnapshot.runStatus === 'Stopped') {
                    clazz = 'stopped error-color-variant';
                }

                return `run-status-icon ${clazz}`;
            })
            .attr('font-family', (d: any) => {
                let family = 'FontAwesome';
                if (d.status.aggregateSnapshot.runStatus === 'Disabled') {
                    family = 'flowfont';
                }
                return family;
            })
            .classed('fa-spin', (d: any) => {
                return d.status.aggregateSnapshot.runStatus === 'Validating';
            })
            .text((d: any) => {
                let img = '';
                if (d.status.aggregateSnapshot.runStatus === 'Disabled') {
                    img = '\ue802';
                } else if (d.status.aggregateSnapshot.runStatus === 'Validating') {
                    img = '\uf1ce';
                } else if (d.status.aggregateSnapshot.runStatus === 'Invalid') {
                    img = '\uf071';
                } else if (d.status.aggregateSnapshot.runStatus === 'Running') {
                    img = '\uf04b';
                } else if (d.status.aggregateSnapshot.runStatus === 'Stopped') {
                    img = '\uf04d';
                }
                return img;
            })
            .each((d: any, i, nodes) => {
                // if there are validation errors generate a tooltip
                if (this.needsTip(d)) {
                    this.canvasUtils.canvasTooltip(ValidationErrorsTip, d3.select(nodes[i]), {
                        isValidating: d.status.aggregateSnapshot.runStatus === 'Validating',
                        validationErrors: d.component.validationErrors
                    });
                } else {
                    this.canvasUtils.resetCanvasTooltip(d3.select(nodes[i]));
                }
            });

        // in count value
        updated.select('text.processor-in tspan.count').text((d: any) => {
            return this.nifiCommon.substringBeforeFirst(d.status.aggregateSnapshot.input, ' ');
        });

        // in size value
        updated.select('text.processor-in tspan.size').text((d: any) => {
            return ' ' + this.nifiCommon.substringAfterFirst(d.status.aggregateSnapshot.input, ' ');
        });

        // read/write value
        updated.select('text.processor-read-write').text((d: any) => {
            return d.status.aggregateSnapshot.read + ' / ' + d.status.aggregateSnapshot.written;
        });

        // out count value
        updated.select('text.processor-out tspan.count').text((d: any) => {
            return this.nifiCommon.substringBeforeFirst(d.status.aggregateSnapshot.output, ' ');
        });

        // out size value
        updated.select('text.processor-out tspan.size').text((d: any) => {
            return ' ' + this.nifiCommon.substringAfterFirst(d.status.aggregateSnapshot.output, ' ');
        });

        // tasks/time value
        updated.select('text.processor-tasks-time').text((d: any) => {
            return d.status.aggregateSnapshot.tasks + ' / ' + d.status.aggregateSnapshot.tasksDuration;
        });

        updated.each((d: any, i, nodes) => {
            const processor = d3.select(nodes[i]);

            // -------------------
            // active thread count
            // -------------------

            this.canvasUtils.activeThreadCount(processor, d);

            // ---------
            // bulletins
            // ---------

            this.canvasUtils.bulletins(processor, d.bulletins);
        });
    }

    private removeProcessors(removed: any) {
        removed.remove();
    }

    /**
     * Determines whether the specific component needs a tooltip.
     *
     * @param d
     * @return if a tip is required
     */
    private needsTip(d: any) {
        return (
            (d.permissions.canRead && !this.nifiCommon.isEmpty(d.component.validationErrors)) ||
            d.status.aggregateSnapshot.runStatus === 'Validating'
        );
    }

    /**
     * Returns whether the resticted indicator should be shown for a given component
     * @param d
     * @returns {*}
     */
    private showRestricted(d: any): string {
        if (!d.permissions.canRead) {
            return 'hidden';
        }

        return d.component.restricted ? 'visible' : 'hidden';
    }

    /**
     * Returns whether the is primary indicator should be shown for a given component
     * @param d
     * @returns {*}
     */
    private showIsPrimary(d: any): string {
        return d.status.aggregateSnapshot.executionNode === 'PRIMARY' ? 'visible' : 'hidden';
    }

    public init(): void {
        this.processorContainer = d3
            .select('#canvas')
            .append('g')
            .attr('pointer-events', 'all')
            .attr('class', 'processors');

        this.store
            .select(selectProcessors)
            .pipe(
                filter(() => this.processorContainer !== null),
                takeUntil(this.destroyed$)
            )
            .subscribe((processors) => {
                this.set(processors);
            });

        this.store
            .select(selectFlowLoadingStatus)
            .pipe(
                filter((status) => status === 'success'),
                filter(() => this.processorContainer !== null),
                switchMap(() => this.store.select(selectAnySelectedComponentIds)),
                takeUntil(this.destroyed$)
            )
            .subscribe((selected) => {
                this.processorContainer.selectAll('g.processor').classed('selected', function (d: any) {
                    return selected.includes(d.id);
                });
            });

        this.store
            .select(selectTransitionRequired)
            .pipe(takeUntil(this.destroyed$))
            .subscribe((transitionRequired) => {
                this.transitionRequired = transitionRequired;
            });
    }

    public destroy(): void {
        this.processorContainer = null;
        this.destroyed$.next(true);
    }

    ngOnDestroy(): void {
        this.destroyed$.complete();
    }

    private set(processors: any): void {
        // update the processors
        this.processors = processors.map((processor: any) => {
            return {
                ...processor,
                type: ComponentType.Processor,
                dimensions: this.dimensions
            };
        });

        // select
        const selection = this.select();

        // enter
        const entered = this.renderProcessors(selection.enter());

        // update
        const updated = selection.merge(entered);
        this.updateProcessors(updated);

        // position
        this.positionBehavior.position(updated, this.transitionRequired);

        // exit
        this.removeProcessors(selection.exit());
    }

    public selectAll(): any {
        return this.processorContainer.selectAll('g.processor');
    }

    public render(): void {
        this.updateProcessors(this.selectAll());
    }

    public pan(): void {
        this.updateProcessors(this.processorContainer.selectAll('g.processor.entering, g.processor.leaving'));
    }
}
