// vtest retry: 3
import db.sqlite

struct Person {
	id                int @[primary; sql: serial]
	age               int
	brothers          []Brother @[fkey: 'person_id']
	sisters           []Sister  @[fkey: 'person_id']
	field_after_fkeys string
}

struct Brother {
	id        int @[primary; sql: serial]
	person_id int
	name      string
}

struct Sister {
	id        int @[primary; sql: serial]
	person_id int
	name      string
}

fn test_field_after_fkeys() {
	db := sqlite.connect(':memory:') or { panic(err) }

	sql db {
		create table Brother
		create table Sister
		create table Person
	}!

	person := Person{
		age:               21
		brothers:          [Brother{
			name: 'aaa'
		}, Brother{
			name: 'bbb'
		}]
		sisters:           [Sister{
			name: 'ccc'
		}, Sister{
			name: 'ddd'
		}]
		field_after_fkeys: 'eee'
	}

	sql db {
		insert person into Person
	}!

	persons := sql db {
		select from Person
	}!
	assert persons[0].age == 21
	assert persons[0].field_after_fkeys == 'eee'
}

@[table: 'foos']
struct Foo {
	id       int @[primary; sql: serial]
	name     string
	children []Child @[fkey: 'parent_id']
}

struct Child {
	id        int @[primary; sql: serial]
	parent_id int
	name      string
}

fn test_fkey_insert_as_assignment_expr() {
	db := sqlite.connect(':memory:')!

	sql db {
		create table Foo
		create table Child
	}!

	foo := Foo{
		name:     'abc'
		children: [
			Child{
				name: 'abc'
			},
			Child{
				name: 'def'
			},
		]
	}
	// use insert as an assigment expr
	_ := sql db {
		insert foo into Foo
	}!

	result := sql db {
		select from Foo where id == 1
	}!
	assert result[0].children.len == 2

	res := sql db {
		select from Child
	}!
	assert res.len == 2
}

struct Foo2 {
	id       int @[primary; sql: serial]
	name     string
	children []Child2 @[fkey: 'parent_id']
}

struct Child2 {
	id        int @[primary; sql: serial]
	parent_id int
	name      string
	bar       ?Bar2 @[fkey: 'child_id']
}

struct Bar2 {
	id       int @[primary; sql: serial]
	child_id int
	name     string
}

fn test_double_fkey_insert() {
	db := sqlite.connect(':memory:')!

	sql db {
		create table Foo2
		create table Child2
		create table Bar2
	}!

	child_one := Child2{
		name: 'abc'
	}

	child_two := Child2{
		name: 'def'
	}

	bar_one := Bar2{
		id:   0
		name: 'name'
	}

	foo := Foo2{
		name:     'abc'
		children: [
			child_one,
			child_two,
		]
	}
	_ := sql db {
		insert foo into Foo2
	}!

	result := sql db {
		select from Foo2 where id == 1
	}!
	assert result[0].children.len == 2

	res := sql db {
		select from Child2
	}!
	assert res.len == 2
}