import EquationEditor from 'equation-editor-react';
import { action, reaction } from 'mobx';
import { observer } from 'mobx-react';
import * as React from 'react';
import { Width } from '../../../fields/DocSymbols';
import { Id } from '../../../fields/FieldSymbols';
import { NumCast, StrCast } from '../../../fields/Types';
import { TraceMobx } from '../../../fields/util';
import { Docs } from '../../documents/Documents';
import { undoBatch } from '../../util/UndoManager';
import { ViewBoxBaseComponent } from '../DocComponent';
import { LightboxView } from '../LightboxView';
import './EquationBox.scss';
import { FieldView, FieldViewProps } from './FieldView';

@observer
export class EquationBox extends ViewBoxBaseComponent<FieldViewProps>() {
    public static LayoutString(fieldKey: string) {
        return FieldView.LayoutString(EquationBox, fieldKey);
    }
    public static SelectOnLoad: string = '';
    _ref: React.RefObject<EquationEditor> = React.createRef();
    componentDidMount() {
        this.props.setContentView?.(this);
        if (EquationBox.SelectOnLoad === this.rootDoc[Id] && (!LightboxView.LightboxDoc || LightboxView.IsLightboxDocView(this.props.docViewPath()))) {
            this.props.select(false);

            this._ref.current!.mathField.focus();
            this._ref.current!.mathField.select();
        }
        reaction(
            () => StrCast(this.dataDoc.text),
            text => {
                if (text && text !== this._ref.current!.mathField.latex()) {
                    this._ref.current!.mathField.latex(text);
                }
            }
        );
        reaction(
            () => this.props.isSelected(),
            selected => {
                if (this._ref.current) {
                    if (selected) this._ref.current.element.current.children[0].addEventListener('keydown', this.keyPressed, true);
                    else this._ref.current.element.current.children[0].removeEventListener('keydown', this.keyPressed);
                }
            },
            { fireImmediately: true }
        );
    }

    @action
    keyPressed = (e: KeyboardEvent) => {
        const _height = Number(getComputedStyle(this._ref.current!.element.current).height.replace('px', ''));
        const _width = Number(getComputedStyle(this._ref.current!.element.current).width.replace('px', ''));
        if (e.key === 'Enter') {
            const nextEq = Docs.Create.EquationDocument({
                title: '# math',
                text: StrCast(this.dataDoc.text),
                _width,
                _height: 25,
                x: NumCast(this.layoutDoc.x),
                y: NumCast(this.layoutDoc.y) + _height + 10,
            });
            EquationBox.SelectOnLoad = nextEq[Id];
            this.props.addDocument?.(nextEq);
            e.stopPropagation();
        }
        if (e.key === 'Tab') {
            const graph = Docs.Create.FunctionPlotDocument([this.rootDoc], {
                x: NumCast(this.layoutDoc.x) + this.layoutDoc[Width](),
                y: NumCast(this.layoutDoc.y),
                _width: 400,
                _height: 300,
                backgroundColor: 'white',
            });
            this.props.addDocument?.(graph);
            e.stopPropagation();
        }
        if (e.key === 'Backspace' && !this.dataDoc.text) this.props.removeDocument?.(this.rootDoc);
    };
    @undoBatch
    onChange = (str: string) => (this.dataDoc.text = str);

    updateSize = () => {
        const style = this._ref.current && getComputedStyle(this._ref.current.element.current);
        if (style?.width.endsWith('px') && style?.height.endsWith('px')) {
            this.layoutDoc._width = Math.max(35, Number(style.width.replace('px', '')));
            this.layoutDoc._height = Math.max(25, Number(style.height.replace('px', '')));
        }
    };
    render() {
        TraceMobx();
        const scale = (this.props.NativeDimScaling?.() || 1) * NumCast(this.layoutDoc._freeform_scale, 1);
        return (
            <div
                ref={r => this.updateSize()}
                className="equationBox-cont"
                onPointerDown={e => !e.ctrlKey && e.stopPropagation()}
                style={{
                    transform: `scale(${scale})`,
                    width: 'fit-content', // `${100 / scale}%`,
                    height: `${100 / scale}%`,
                    pointerEvents: !this.props.isSelected() ? 'none' : undefined,
                    fontSize: StrCast(this.rootDoc._fontSize),
                }}
                onKeyDown={e => e.stopPropagation()}>
                <EquationEditor ref={this._ref} value={StrCast(this.dataDoc.text, 'x')} spaceBehavesLikeTab={true} onChange={this.onChange} autoCommands="pi theta sqrt sum prod alpha beta gamma rho" autoOperatorNames="sin cos tan" />
            </div>
        );
    }
}