import {
    CompilerOptions,
    CodegenResult,
    ParserOptions,
    RootNode,
    NodeTypes,
    ElementNode,
    SourceLocation,
    CompilerError,
    TextModes,
    BindingMetadata
} from '@vue/compiler-core'
import * as CompilerDom from '@vue/compiler-dom'
import { RawSourceMap, SourceMapGenerator } from 'source-map'
import { Statement } from '@babel/types'

/**
 * The following function is adapted from https://github.com/psalaets/vue-sfc-descriptor-to-string/blob/master/index.js
 */

/**
 * The MIT License (MIT)
 * Copyright (c) 2018 Paul Salaets
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

export function stringify(sfcDescriptor: SFCDescriptor) {
    const { template, script, styles, customBlocks } = sfcDescriptor

    return (
        (
            [template, script, ...styles, ...customBlocks]
                // discard blocks that don't exist
                .filter(block => block != null) as Array<NonNullable<SFCBlock>>
        )
            // sort blocks by source position
            .sort((a, b) => a.loc.start.offset - b.loc.start.offset)
            // figure out exact source positions of blocks
            .map(block => {
                const openTag = makeOpenTag(block)
                const closeTag = makeCloseTag(block)

                return Object.assign({}, block, {
                    openTag,
                    closeTag,
                })
            })
            // generate sfc source
            .reduce((sfcCode, block, index, array) => {
                return (
                    sfcCode +
                    block.openTag +
                    block.content +
                    block.closeTag
                )
            }, '')
    )
}

function makeOpenTag(block: SFCBlock) {
    let source = '<' + block.type

    source += Object.keys(block.attrs)
        .sort()
        .map(name => {
            const value = block.attrs[name]

            if (value === true) {
                return name
            } else {
                return `${name}="${value}"`
            }
        })
        .map(attr => ' ' + attr)
        .join('')

    return source + '>'
}

function makeCloseTag(block: SFCBlock) {
    return `</${block.type}>\n`
}

/**
 * The following content are modifed from https://github.com/vuejs/vue-next/blob/master/packages/compiler-sfc/src/parse.ts
 */

export interface TemplateCompiler {
    compile(template: string, options: CompilerOptions): CodegenResult
    parse(template: string, options: ParserOptions): RootNode
}

export interface SFCParseOptions {
    filename?: string
    sourceMap?: boolean
    sourceRoot?: string
    pad?: boolean | 'line' | 'space'
    compiler?: TemplateCompiler
}

export interface SFCBlock {
    type: string
    content: string
    attrs: Record<string, string | true>
    loc: SourceLocation
    map?: RawSourceMap
    lang?: string
    src?: string
}

export interface SFCTemplateBlock extends SFCBlock {
    type: 'template'
    ast: ElementNode
}

export interface SFCScriptBlock extends SFCBlock {
    type: 'script'
    setup?: string | boolean
    bindings?: BindingMetadata
    scriptAst?: Statement[]
    scriptSetupAst?: Statement[]
}

export interface SFCStyleBlock extends SFCBlock {
    type: 'style'
    scoped?: boolean
    module?: string | boolean
}

export interface SFCDescriptor {
    filename: string
    source: string
    template: SFCTemplateBlock | null
    script: SFCScriptBlock | null
    scriptSetup: SFCScriptBlock | null
    styles: SFCStyleBlock[]
    customBlocks: SFCBlock[]
}

export interface SFCParseResult {
    descriptor: SFCDescriptor
    errors: (CompilerError | SyntaxError)[]
}

const SFC_CACHE_MAX_SIZE = 500
const sourceToSFC = new (require('lru-cache'))({max: SFC_CACHE_MAX_SIZE}) as Map<
    string,
    SFCParseResult
    >

export function parse(
    source: string,
    {
        sourceMap = true,
        filename = 'anonymous.vue',
        sourceRoot = '',
        pad = false,
        compiler = CompilerDom
    }: SFCParseOptions = {}
): SFCParseResult {
    const sourceKey =
        source + sourceMap + filename + sourceRoot + pad + compiler.parse
    const cache = sourceToSFC.get(sourceKey)
    if (cache) {
        return cache
    }

    const descriptor: SFCDescriptor = {
        filename,
        source,
        template: null,
        script: null,
        scriptSetup: null,
        styles: [],
        customBlocks: []
    }

    const errors: (CompilerError | SyntaxError)[] = []
    const ast = compiler.parse(source, {
        // there are no components at SFC parsing level
        isNativeTag: () => true,
        // preserve all whitespaces
        isPreTag: () => true,
        getTextMode: ({ tag, props }, parent) => {
            // all top level elements except <template> are parsed as raw text
            // containers
            if (
                (!parent && tag !== 'template') ||
                // <template lang="xxx"> should also be treated as raw text
                (tag === 'template' &&
                    props.some(
                        p =>
                            p.type === NodeTypes.ATTRIBUTE &&
                            p.name === 'lang' &&
                            p.value &&
                            p.value.content !== 'html'
                    ))
            ) {
                return TextModes.RAWTEXT
            } else {
                return TextModes.DATA
            }
        },
        onError: e => {
            errors.push(e)
        }
    })

    ast.children.forEach(node => {
        if (node.type !== NodeTypes.ELEMENT) {
            return
        }
        if (!node.children.length && !hasSrc(node) && node.tag !== 'template') {
            return
        }
        switch (node.tag) {
            case 'template':
                if (!descriptor.template) {
                    const templateBlock = (descriptor.template = createBlock(
                        node,
                        source,
                        false
                    ) as SFCTemplateBlock)
                    templateBlock.ast = node
                } else {
                    errors.push(createDuplicateBlockError(node))
                }
                break
            case 'script':
                const scriptBlock = createBlock(node, source, pad) as SFCScriptBlock
                const isSetup = !!scriptBlock.attrs.setup
                if (isSetup && !descriptor.scriptSetup) {
                    descriptor.scriptSetup = scriptBlock
                    break
                }
                if (!isSetup && !descriptor.script) {
                    descriptor.script = scriptBlock
                    break
                }
                errors.push(createDuplicateBlockError(node, isSetup))
                break
            case 'style':
                const styleBlock = createBlock(node, source, pad) as SFCStyleBlock
                if (styleBlock.attrs.vars) {
                    errors.push(
                        new SyntaxError(
                            `<style vars> has been replaced by a new proposal: ` +
                            `https://github.com/vuejs/rfcs/pull/231`
                        )
                    )
                }
                descriptor.styles.push(styleBlock)
                break
            default:
                descriptor.customBlocks.push(createBlock(node, source, pad))
                break
        }
    })

    if (descriptor.scriptSetup) {
        if (descriptor.scriptSetup.src) {
            errors.push(
                new SyntaxError(
                    `<script setup> cannot use the "src" attribute because ` +
                    `its syntax will be ambiguous outside of the component.`
                )
            )
            descriptor.scriptSetup = null
        }
        if (descriptor.script && descriptor.script.src) {
            errors.push(
                new SyntaxError(
                    `<script> cannot use the "src" attribute when <script setup> is ` +
                    `also present because they must be processed together.`
                )
            )
            descriptor.script = null
        }
    }

    if (sourceMap) {
        const genMap = (block: SFCBlock | null) => {
            if (block && !block.src) {
                block.map = generateSourceMap(
                    filename,
                    source,
                    block.content,
                    sourceRoot,
                    !pad || block.type === 'template' ? block.loc.start.line - 1 : 0
                )
            }
        }
        genMap(descriptor.template)
        genMap(descriptor.script)
        descriptor.styles.forEach(genMap)
        descriptor.customBlocks.forEach(genMap)
    }

    const result = {
        descriptor,
        errors
    }
    sourceToSFC.set(sourceKey, result)
    return result
}

function createDuplicateBlockError(
    node: ElementNode,
    isScriptSetup = false
): CompilerError {
    const err = new SyntaxError(
        `Single file component can contain only one <${node.tag}${
            isScriptSetup ? ` setup` : ``
        }> element`
    ) as CompilerError
    err.loc = node.loc
    return err
}

function createBlock(
    node: ElementNode,
    source: string,
    pad: SFCParseOptions['pad']
): SFCBlock {
    const type = node.tag
    let { start, end } = node.loc
    let content = ''
    if (node.children.length) {
        start = node.children[0].loc.start
        end = node.children[node.children.length - 1].loc.end
        content = source.slice(start.offset, end.offset)
    }
    const loc = {
        source: content,
        start,
        end
    }
    const attrs: Record<string, string | true> = {}
    const block: SFCBlock = {
        type,
        content,
        loc,
        attrs
    }
    if (pad) {
        block.content = padContent(source, block, pad) + block.content
    }
    node.props.forEach(p => {
        if (p.type === NodeTypes.ATTRIBUTE) {
            attrs[p.name] = p.value ? p.value.content || true : true
            if (p.name === 'lang') {
                block.lang = p.value && p.value.content
            } else if (p.name === 'src') {
                block.src = p.value && p.value.content
            } else if (type === 'style') {
                if (p.name === 'scoped') {
                    ;(block as SFCStyleBlock).scoped = true
                } else if (p.name === 'module') {
                    ;(block as SFCStyleBlock).module = attrs[p.name]
                }
            } else if (type === 'script' && p.name === 'setup') {
                ;(block as SFCScriptBlock).setup = attrs.setup
            }
        }
    })
    return block
}

const splitRE = /\r?\n/g
const emptyRE = /^(?:\/\/)?\s*$/
const replaceRE = /./g

function generateSourceMap(
    filename: string,
    source: string,
    generated: string,
    sourceRoot: string,
    lineOffset: number
): RawSourceMap {
    const map = new SourceMapGenerator({
        file: filename.replace(/\\/g, '/'),
        sourceRoot: sourceRoot.replace(/\\/g, '/')
    })
    map.setSourceContent(filename, source)
    generated.split(splitRE).forEach((line, index) => {
        if (!emptyRE.test(line)) {
            const originalLine = index + 1 + lineOffset
            const generatedLine = index + 1
            for (let i = 0; i < line.length; i++) {
                if (!/\s/.test(line[i])) {
                    map.addMapping({
                        source: filename,
                        original: {
                            line: originalLine,
                            column: i
                        },
                        generated: {
                            line: generatedLine,
                            column: i
                        }
                    })
                }
            }
        }
    })
    return JSON.parse(map.toString())
}

function padContent(
    content: string,
    block: SFCBlock,
    pad: SFCParseOptions['pad']
): string {
    content = content.slice(0, block.loc.start.offset)
    if (pad === 'space') {
        return content.replace(replaceRE, ' ')
    } else {
        const offset = content.split(splitRE).length
        const padChar = block.type === 'script' && !block.lang ? '//\n' : '\n'
        return Array(offset).join(padChar)
    }
}

function hasSrc(node: ElementNode) {
    return node.props.some(p => {
        if (p.type !== NodeTypes.ATTRIBUTE) {
            return false
        }
        return p.name === 'src'
    })
}