// setup
use sea_orm::sea_query::{ForeignKeyCreateStatement, ColumnDef, TableCreateStatement};
use sea_orm::{Set, EntityTrait, error::*, sea_query, ConnectionTrait, DbConn, ExecResult};
pub mod comment;
pub mod post;
pub use comment::Entity as Comment;
pub use post::Entity as Post;
use chrono::{Local, Duration, DurationRound};
use sea_orm::prelude::Decimal;

// setup
async fn create_table(db: &DbConn, stmt: &TableCreateStatement) -> Result<ExecResult, DbErr> {
    let builder = db.get_database_backend();
    db.execute(builder.build(stmt)).await
}

pub async fn create_post_table(db: &DbConn) -> Result<ExecResult, DbErr> {
    let stmt = sea_query::Table::create()
        .table(post::Entity)
        .if_not_exists()
        .col(
            ColumnDef::new(post::Column::Id)
                .integer()
                .not_null()
                .auto_increment()
                .primary_key(),
        )
        .col(ColumnDef::new(post::Column::Title).string().not_null())
        .col(ColumnDef::new(post::Column::Text).string().not_null())
        .col(ColumnDef::new(post::Column::TeaMandatory).string().not_null())
        .col(ColumnDef::new(post::Column::TeaOptional).string())
        .col(ColumnDef::new(post::Column::InsertDate).date().not_null())
        .col(ColumnDef::new(post::Column::Attachment).string())
        .to_owned();

    let _result = create_table(db, &stmt).await;

    let stmt = sea_query::Table::create()
        .table(comment::Entity)
        .if_not_exists()
        .col(
            ColumnDef::new(post::Column::Id)
                .integer()
                .not_null()
                .auto_increment()
                .primary_key(),
        )
        .col(ColumnDef::new(comment::Column::Comment).string().not_null())
        .col(ColumnDef::new(comment::Column::User).string().not_null())
        .col(ColumnDef::new(comment::Column::InsertDate).date_time().not_null())
        .col(ColumnDef::new(comment::Column::IsVisible).boolean().not_null())
        .col(ColumnDef::new(comment::Column::MyDecimal).decimal().not_null())
        .col(ColumnDef::new(comment::Column::PostId).integer())
        .foreign_key(
            ForeignKeyCreateStatement::new()
                .name("fk-comment-post")
                .from_tbl(Comment)
                .from_col(comment::Column::PostId)
                .to_tbl(Post)
                .to_col(post::Column::Id),
        )
        .to_owned();

    let res = create_table(db, &stmt).await;

    for i in 1..1000 {
        let row = post::ActiveModel {
           title: Set(format!("Test {}", i)),
           text: Set("some content".to_string()),
           tea_mandatory: Set(post::Tea::EverydayTea),
           tea_optional: Set(None),
           insert_date: Set(Local::now().date_naive()),
           ..Default::default()
        };
        let _res = Post::insert(row).exec(db).await;
   }

   for i in 1..1000 {
       let row = comment::ActiveModel {
          comment: Set(format!("Test {}", i)),
          user: Set("me@home.com".to_string()),
          my_decimal: Set(Decimal::new(105, 0)),
          insert_date: Set(Local::now().naive_utc().duration_round(Duration::minutes(1)).unwrap()),
          is_visible: Set(i%2 == 0),
          ..Default::default()
       };
       let _res = Comment::insert(row).exec(db).await;
  }

    res
}