use rusqlite::Connection;

#[derive(Debug)]
struct Person {
    id: i32,
    name: String,
    age: i32,
}

fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Get input from CLI
    let args: Vec<String> = std::env::args().collect();
    let name = &args[1];
    let age = &args[2];

    let connection = Connection::open_in_memory()?;

    connection.execute(       // $ sql-sink
        "CREATE TABLE person (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name VARCHAR NOT NULL,
            age INT NOT NULL
        )",
        (),
    )?;

    let query = format!("INSERT INTO person (name, age) VALUES ('{}', '{}')", name, age);

    connection.execute(&query, ())?;    // $ sql-sink

    let person = connection.query_row(&query, (), |row| {    // $ sql-sink
        let row: &rusqlite::Row<'_> = row;
        Ok(Person {
            id: row.get(0)?,        // $ database-read
            name: row.get(1)?,      // $ database-read
            age: row.get(2)?,       // $ database-read
        })
    })?;

    let mut stmt = connection.prepare("SELECT id, name, age FROM person")?;      // $ sql-sink
    let people = stmt.query_map([], |row| {
        let row: &rusqlite::Row<'_> = row;
        Ok(Person {
            id: row.get_unwrap(0),      // $ database-read
            name: row.get_unwrap(1),    // $ database-read
            age: row.get_unwrap(2),     // $ database-read
        })
    })?;

    _ = connection.prepare_cached("SELECT id, name, age FROM person")?; // $ sql-sink
    _ = connection.prepare_with_flags("SELECT id, name, age FROM person", rusqlite::PrepFlags::empty())?; // $ sql-sink
    _ = connection.query_row_and_then("SELECT id, name, age FROM person", [], |row| { // $ sql-sink
        let row: &rusqlite::Row<'_> = row;
        let result: Result<i32, rusqlite::Error> = Ok(row.get(0)?); // $ database-read
        result
    })?;

    Ok(())
}