import XsdManager from './XsdManager'
import XsdCompletion from './XsdCompletion'
import XsdValidation from './XsdValidation'
import { editor } from 'monaco-editor'
import { debounce } from 'ts-debounce'
import xsdGenerateTemplate from './XsdGenerateTemplate'
import prettier from 'prettier'
import parserHTML from 'prettier/plugins/html'
import IStandaloneCodeEditor = editor.IStandaloneCodeEditor

export default class XsdFeatures {
    private static readonly completionRegistry = new WeakMap<object, { dispose(): void }>()
    private readonly xsdCollection: XsdManager
    private monaco: any
    private oldDecorations: string[] = []
    private editor: editor.IStandaloneCodeEditor
    private xsdValidation: XsdValidation | undefined
    private xsdGenerateTemplate: xsdGenerateTemplate | undefined

    constructor(xsdCollection: XsdManager, monaco: any, editor: IStandaloneCodeEditor) {
        this.xsdCollection = xsdCollection
        this.monaco = monaco
        this.editor = editor

        this.editor.updateOptions({
            wordSeparators: '`~!@#$%^&*()-=+[{]}\\|;\'",.<>/?',
        })
    }

    public addCompletion = (): void => {
        XsdFeatures.completionRegistry.get(this.monaco)?.dispose()
        const xsdCompletion = new XsdCompletion(this.xsdCollection)
        const disposable = this.monaco.languages.registerCompletionItemProvider(
            'xml',
            xsdCompletion.provider(),
        )
        XsdFeatures.completionRegistry.set(this.monaco, disposable)
    }

    public doValidation = (): void => {
        this.xsdValidation = this.xsdValidation ?? new XsdValidation(this.xsdCollection)
        const model = this.editor.getModel()
        const newDecorations = this.xsdValidation.decorations(model)
        this.oldDecorations = this.editor.deltaDecorations(this.oldDecorations, newDecorations)
    }

    public addValidation = (): void => {
        this.xsdValidation = new XsdValidation(this.xsdCollection)
        const debouncedDoValidation = debounce(this.doValidation, 1000)
        this.editor.onKeyDown(() => debouncedDoValidation())
        this.doValidation()
    }

    public doReformatCode = async (): Promise<void> => {
        const model = this.editor.getModel()
        if (model) model.setValue(await this.prettier(model.getValue()))
    }

    public prettier = (xml: string): Promise<string> =>
        prettier.format(xml, {
            parser: 'html',
            plugins: [parserHTML],
            tabWidth: 4,
        })

    public doGenerate = async (levels: number, withAttributes: boolean): Promise<void> => {
        this.xsdGenerateTemplate =
            this.xsdGenerateTemplate ?? new xsdGenerateTemplate(this.xsdCollection)
        const model = this.editor.getModel()
        const template = this.xsdGenerateTemplate.getTemplate(model, levels, withAttributes)
        if (template) model?.setValue(await this.prettier(template))
    }

    public addGenerateAction = (): void => {
        this.xsdGenerateTemplate = new xsdGenerateTemplate(this.xsdCollection)
        this.editor.addAction({
            id: 'xsd-template-generate-without-attributes',
            label: 'Generate Template from XSD',
            keybindings: [
                this.monaco.KeyMod.CtrlCmd | this.monaco.KeyMod.Shift | this.monaco.KeyCode.KEY_G,
            ],
            contextMenuGroupId: '1_modification',
            contextMenuOrder: 2,
            run: () => this.doGenerate(parseInt(window.prompt('levels', '1') ?? ''), false),
        })
        this.editor.addAction({
            id: 'xsd-template-generate-with-attributes',
            label: 'Generate Template from XSD with required attributes',
            keybindings: [
                this.monaco.KeyMod.CtrlCmd |
                    this.monaco.KeyMod.Alt |
                    this.monaco.KeyMod.Shift |
                    this.monaco.KeyCode.KEY_G,
            ],
            contextMenuGroupId: '1_modification',
            contextMenuOrder: 3,
            run: () => this.doGenerate(parseInt(window.prompt('levels', '1') ?? ''), true),
        })
    }

    public addReformatAction = (): void => {
        this.editor.addAction({
            id: 'xsd-reformat-code',
            label: 'Reformat code',
            keybindings: [
                this.monaco.KeyMod.CtrlCmd | this.monaco.KeyMod.Shift | this.monaco.KeyCode.KEY_R,
            ],
            contextMenuGroupId: '1_modification',
            contextMenuOrder: 1.5,
            run: this.doReformatCode,
        })
    }
}