import db.sqlite

struct OrmBulkUser {
    id   int @[primary]
    name string
    age  int
}

struct OrmBulkDefaultRow {
    id int @[primary; sql: serial]
}

struct OrmBulkRenamedUser {
    id           int    @[primary]
    display_name string @[sql: 'display_name_text']
}

fn test_orm_bulk_insert_and_update() {
    mut db := sqlite.connect(':memory:')!
    defer {
        db.close() or { panic(err) }
    }

    sql db {
        create table OrmBulkUser
    }!

    users := [
        OrmBulkUser{
            id:   1
            name: 'Alice'
            age:  25
        },
        OrmBulkUser{
            id:   2
            name: 'Bob'
            age:  30
        },
    ]

    sql db {
        insert users into OrmBulkUser
    }!

    first := sql db {
        select from OrmBulkUser where id == 1
    }!
    second := sql db {
        select from OrmBulkUser where id == 2
    }!

    assert first.len == 1
    assert first[0].name == 'Alice'
    assert first[0].age == 25
    assert second.len == 1
    assert second[0].name == 'Bob'
    assert second[0].age == 30

    updates := [
        OrmBulkUser{
            id:   1
            name: 'Alicia'
            age:  26
        },
        OrmBulkUser{
            id:   2
            name: 'Robert'
            age:  31
        },
    ]

    sql db {
        update OrmBulkUser set name = updates.name, age = updates.age where id == updates.id
    }!

    updated_first := sql db {
        select from OrmBulkUser where id == 1
    }!
    updated_second := sql db {
        select from OrmBulkUser where id == 2
    }!

    assert updated_first.len == 1
    assert updated_first[0].name == 'Alicia'
    assert updated_first[0].age == 26
    assert updated_second.len == 1
    assert updated_second[0].name == 'Robert'
    assert updated_second[0].age == 31
}

fn test_orm_bulk_insert_preserves_all_default_rows() {
    mut db := sqlite.connect(':memory:')!
    defer {
        db.close() or { panic(err) }
    }

    sql db {
        create table OrmBulkDefaultRow
    }!

    rows := [OrmBulkDefaultRow{}, OrmBulkDefaultRow{}, OrmBulkDefaultRow{}]

    sql db {
        insert rows into OrmBulkDefaultRow
    }!

    inserted := sql db {
        select from OrmBulkDefaultRow order by id
    }!

    assert inserted.len == 3
    assert inserted[0].id == 1
    assert inserted[1].id == 2
    assert inserted[2].id == 3
}

fn test_orm_bulk_insert_with_mixed_serial_values_keeps_defaults() {
    mut db := sqlite.connect(':memory:')!
    defer {
        db.close() or { panic(err) }
    }

    sql db {
        create table OrmBulkDefaultRow
    }!

    rows := [
        OrmBulkDefaultRow{},
        OrmBulkDefaultRow{
            id: 5
        },
    ]

    sql db {
        insert rows into OrmBulkDefaultRow
    }!

    inserted := sql db {
        select from OrmBulkDefaultRow order by id
    }!

    assert inserted.len == 2
    assert inserted[0].id == 1
    assert inserted[1].id == 5
}

fn test_orm_bulk_update_with_renamed_column() {
    mut db := sqlite.connect(':memory:')!
    defer {
        db.close() or { panic(err) }
    }

    sql db {
        create table OrmBulkRenamedUser
    }!

    users := [
        OrmBulkRenamedUser{
            id:           1
            display_name: 'Alice'
        },
        OrmBulkRenamedUser{
            id:           2
            display_name: 'Bob'
        },
    ]

    sql db {
        insert users into OrmBulkRenamedUser
    }!

    updates := [
        OrmBulkRenamedUser{
            id:           1
            display_name: 'Alicia'
        },
        OrmBulkRenamedUser{
            id:           2
            display_name: 'Robert'
        },
    ]

    sql db {
        update OrmBulkRenamedUser set display_name = updates.display_name where id == updates.id
    }!

    updated := sql db {
        select from OrmBulkRenamedUser order by id
    }!

    assert updated.len == 2
    assert updated[0].display_name == 'Alicia'
    assert updated[1].display_name == 'Robert'
}