module yaml

import os

enum Role {
    worker
    manager
}

struct Address {
    city string
    zip  int
}

struct AppConfig {
    name    string
    role    Role @[json: 'role']
    enabled bool
    ports   []int
    address Address
    notes   string
}

fn test_parse_doc_queries_and_block_scalars() ! {
    doc := parse_text('title: Example
config: { enabled: true, retries: 3 }
servers:
  - host: api.local
    ports: [80, 443]
  - host: jobs.local
quoted:
  "a.b": 7
notes: |
  first line
  second line
folded: >
  hello
  world
')!

    assert doc.value('title').string() == 'Example'
    assert doc.value('config.enabled').bool()
    assert doc.value('config.retries').int() == 3
    assert doc.value('servers[0].host').string() == 'api.local'
    assert doc.value('servers[0].ports[1]').int() == 443
    assert doc.value('quoted."a.b"').int() == 7
    assert doc.value('notes').string() == 'first line\nsecond line\n'
    assert doc.value('folded').string() == 'hello world\n'
}

fn test_generic_encode_decode_with_json_attrs() ! {
    config := AppConfig{
        name:    'worker'
        role:    .manager
        enabled: true
        ports:   [8080, 9090]
        address: Address{
            city: 'Springwood'
            zip:  1428
        }
        notes:   'line one\nline two'
    }

    encoded := encode(config)
    assert encoded.contains('"role": "manager"')
    assert encoded.contains('"address":')
    assert encoded.contains('- 8080')

    decoded := decode[AppConfig](encoded)!
    assert decoded == config
}

fn test_file_helpers() ! {
    path := os.join_path(os.vtmp_dir(), 'yaml_test_${os.getpid()}.yml')
    defer {
        os.rm(path) or {}
    }
    config := AppConfig{
        name:    'batch11'
        role:    .worker
        enabled: false
        ports:   [7000]
        address: Address{
            city: 'Moscow'
            zip:  101000
        }
        notes:   'plain'
    }

    encode_file(path, config)!
    decoded := decode_file[AppConfig](path)!
    assert decoded == config

    doc := parse_file(path)!
    assert doc.value('name').string() == 'batch11'
    assert doc.value('role').string() == 'worker'
    assert doc.value('address.city').string() == 'Moscow'
}