module yaml

struct FlowParser {
    src string
mut:
    pos int
}

fn parse_flow_value(src string) !Any {
    mut parser := FlowParser{
        src: src
    }
    value := parser.parse_value()!
    parser.skip_space()
    if parser.pos != parser.src.len {
        return error('yaml: unexpected trailing flow content')
    }
    return value
}

fn (mut p FlowParser) parse_value() !Any {
    p.skip_space()
    if p.pos >= p.src.len {
        return error('yaml: unexpected end of flow value')
    }
    return match p.src[p.pos] {
        `[` { p.parse_array() }
        `{` { p.parse_object() }
        `"`, `'` { Any(parse_quoted_flow_string(mut p)!) }
        else { parse_scalar(p.parse_plain_token()) }
    }
}

fn (mut p FlowParser) parse_array() !Any {
    p.pos++
    mut items := []Any{}
    for {
        p.skip_space()
        if p.pos >= p.src.len {
            return error('yaml: unterminated flow array')
        }
        if p.src[p.pos] == `]` {
            p.pos++
            break
        }
        items << p.parse_value()!
        p.skip_space()
        if p.pos >= p.src.len {
            return error('yaml: unterminated flow array')
        }
        if p.src[p.pos] == `,` {
            p.pos++
            continue
        }
        if p.src[p.pos] == `]` {
            p.pos++
            break
        }
        return error('yaml: expected `,` or `]` in flow array')
    }
    return Any(items)
}

fn (mut p FlowParser) parse_object() !Any {
    p.pos++
    mut result := map[string]Any{}
    for {
        p.skip_space()
        if p.pos >= p.src.len {
            return error('yaml: unterminated flow object')
        }
        if p.src[p.pos] == `}` {
            p.pos++
            break
        }
        key := p.parse_key()!
        p.skip_space()
        if p.pos >= p.src.len || p.src[p.pos] != `:` {
            return error('yaml: expected `:` in flow object')
        }
        p.pos++
        result[key] = p.parse_value()!
        p.skip_space()
        if p.pos >= p.src.len {
            return error('yaml: unterminated flow object')
        }
        if p.src[p.pos] == `,` {
            p.pos++
            continue
        }
        if p.src[p.pos] == `}` {
            p.pos++
            break
        }
        return error('yaml: expected `,` or `}` in flow object')
    }
    return Any(result)
}

fn (mut p FlowParser) parse_key() !string {
    p.skip_space()
    if p.pos >= p.src.len {
        return error('yaml: unexpected end of flow key')
    }
    if p.src[p.pos] in [`"`, `'`] {
        return parse_quoted_flow_string(mut p)
    }
    start := p.pos
    for p.pos < p.src.len {
        ch := p.src[p.pos]
        if ch == `:` {
            break
        }
        p.pos++
    }
    return p.src[start..p.pos].trim_space()
}

fn (mut p FlowParser) parse_plain_token() string {
    start := p.pos
    mut bracket_depth := 0
    mut brace_depth := 0
    for p.pos < p.src.len {
        ch := p.src[p.pos]
        if ch == `[` {
            bracket_depth++
        } else if ch == `]` {
            if bracket_depth == 0 {
                break
            }
            bracket_depth--
        } else if ch == `{` {
            brace_depth++
        } else if ch == `}` {
            if brace_depth == 0 {
                break
            }
            brace_depth--
        } else if ch == `,` && bracket_depth == 0 && brace_depth == 0 {
            break
        }
        p.pos++
    }
    return p.src[start..p.pos].trim_space()
}

fn (mut p FlowParser) skip_space() {
    for p.pos < p.src.len && p.src[p.pos].is_space() {
        p.pos++
    }
}

fn parse_quoted_flow_string(mut p FlowParser) !string {
    start := p.pos
    quote := p.src[p.pos]
    p.pos++
    mut escape := false
    for p.pos < p.src.len {
        ch := p.src[p.pos]
        if quote == `"` {
            if escape {
                escape = false
            } else if ch == `\\` {
                escape = true
            } else if ch == `"` {
                p.pos++
                return parse_quoted_string(p.src[start..p.pos])
            }
        } else if ch == `'` {
            if p.pos + 1 < p.src.len && p.src[p.pos + 1] == `'` {
                p.pos += 2
                continue
            }
            p.pos++
            return parse_quoted_string(p.src[start..p.pos])
        }
        p.pos++
    }
    return error('yaml: unterminated quoted flow string')
}