use sqlx::migrate::MigrateDatabase;
use std::time::Instant;
use tokio;

mod graph;

mod geng;
use crate::geng::GengIterator;

mod compute;

#[tokio::main]
async fn main() -> Result<(), sqlx::Error> {
    dotenv::dotenv().ok();
    let database_url =
        std::env::var("DATABASE_URL").expect("Expected DATABASE_URL in env");
    if !sqlx::Sqlite::database_exists(&database_url).await? {
        sqlx::Sqlite::create_database(&database_url).await?;
    }
    let db = sqlx::SqlitePool::connect(&database_url).await?;
    let _ = sqlx::query!(
        "CREATE TABLE IF NOT EXISTS graphs (g6 VARCHAR NOT NULL, ind_dom INT NOT NULL, forced_geod INT NOT NULL);"
    )
    .execute(&db)
    .await;

    let gi = GengIterator::new(9);

    let start = Instant::now();

    println!("Started");
    const BATCH_SIZE: usize = 10000;
    let mut count = 0;
    loop {
        let graphs = gi.take(BATCH_SIZE);
        let batch = Instant::now();
        let tasks: Vec<_> = graphs
            .map(|g| {
                let db = db.clone();
                tokio::spawn(async move {
                    let (g6, ind_dom, fg) = compute::dominating_numbers(g).await;
                    let ind_dom = ind_dom.unwrap_or(0);
                    let fg = fg.unwrap_or(0);
                    let _ = sqlx::query!(
                        "INSERT INTO graphs (g6, ind_dom, forced_geod) VALUES (?, ?, ?);",
                        g6, ind_dom, fg
                    ).execute(&db).await;
                })
            })
            .collect();
        let part = futures::future::join_all(tasks).await;
        let sz = part.len();
        count += sz;
        println!(
            "Counted {count} in {}s",
            batch.elapsed().as_nanos() as f64 / 1e9
        );
        if sz < BATCH_SIZE {
            break;
        }
    }

    let elapsed = start.elapsed();
    let time = elapsed.as_nanos();
    println!("Time elapsed: {}s", time as f64 / 1e9);

    Ok(())
}