turbosql-impl 0.14.0

use super::Table;
use proc_macro_error::abort_call_site;
use quote::quote_spanned;

/// UPDATE tablename SET name1=?, name2=?... WHERE rowid=?
pub(super) fn update(table: &Table) -> proc_macro2::TokenStream {
	if table.columns[0].name != "rowid" {
		abort_call_site!("First field must be `rowid: Option<i64>`");
	}

	let sql = makesql_update(table);

	super::validate_sql_or_abort(&sql);

	let mut columns = table.columns.clone();
	columns.rotate_left(1);
	let columns = 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 update(&self) -> Result<usize, ::turbosql::Error> {
			assert!(self.rowid.is_some());
			::turbosql::__TURBOSQL_DB.with(|db| {
				let db = db.borrow_mut();
				let mut stmt = db.prepare_cached(#sql)?;
				Ok(stmt.execute(&[#( #columns ),*] as &[&dyn ::turbosql::ToSql])?)
			})
		}

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

fn makesql_update(table: &Table) -> String {
	format!(
		"UPDATE {} SET {} WHERE rowid=?",
		table.name,
		table.columns.iter().collect::<Vec<_>>()[1..]
			.iter()
			.map(|c| format!("{}=?", c.name.as_str()))
			.collect::<Vec<_>>()
			.join(", ")
	)
}