turbosql-impl 0.14.0

use super::Table;
use quote::quote_spanned;

/// INSERT INTO tablename (name1, name2...) VALUES (?1, ?2...)
pub(super) fn insert(table: &Table) -> proc_macro2::TokenStream {
	let sql = makesql_insert(table);

	super::validate_sql_or_abort(&sql);

	let columns = table.columns.iter().map(|c| {
		let ident = &c.ident;
		if c.sql_type.starts_with("TEXT") && c.rust_type != "Option < String >" && c.rust_type != "String" {
			quote_spanned!(c.span => &::turbosql::serde_json::to_string(&self.#ident)? as &dyn ::turbosql::ToSql)
		} else {
			quote_spanned!(c.span => &self.#ident as &dyn ::turbosql::ToSql)
		}
	})
	.collect::<Vec<_>>();

	quote_spanned! { table.span =>
		fn insert(&self) -> Result<i64, ::turbosql::Error> {
			assert!(self.rowid.is_none());
			::turbosql::__TURBOSQL_DB.with(|db| {
				let db = db.borrow_mut();
				let mut stmt = db.prepare_cached(#sql)?;
				Ok(stmt.insert(&[#( #columns ),*] as &[&dyn ::turbosql::ToSql])?)
			})
		}

		fn insert_mut(&mut self) -> Result<i64, ::turbosql::Error> {
			assert!(self.rowid.is_none());
			::turbosql::__TURBOSQL_DB.with(|db| {
				let db = db.borrow_mut();
				let mut stmt = db.prepare_cached(#sql)?;
				let result = stmt.insert(&[#( #columns ),*] as &[&dyn ::turbosql::ToSql])?;
				self.rowid = Some(result);
				Ok(result)
			})
		}

		fn insert_batch<T: AsRef<#table>>(rows: &[T]) -> Result<(), ::turbosql::Error> {
			for row in rows {
				row.as_ref().insert()?;
			}
			Ok(())
		}
	}
}

fn makesql_insert(table: &Table) -> String {
	let mut sql = format!("INSERT INTO {} (", table.name);
	sql += table.columns.iter().map(|c| c.name.as_str()).collect::<Vec<_>>().join(", ").as_str();
	sql += ") VALUES (";
	sql += table.columns.iter().map(|_| "?").collect::<Vec<_>>().join(", ").as_str();
	sql += ")";

	sql
}