rjango 0.1.1

A full-stack Rust backend framework inspired by Django
Documentation
use super::base::DatabaseBackend;

#[derive(Debug, Default, Clone, Copy)]
pub struct MySqlBackend;

#[must_use]
fn parse_optional_length(field_type: &str, base: &str) -> Option<Option<usize>> {
    let field_type = field_type.trim();
    if field_type == base {
        return Some(None);
    }

    let rest = field_type.strip_prefix(base)?;
    let length = rest.strip_prefix('(')?.strip_suffix(')')?.parse().ok()?;
    Some(Some(length))
}

impl DatabaseBackend for MySqlBackend {
    fn name(&self) -> &str {
        "mysql"
    }

    fn quote_name(&self, name: &str) -> String {
        format!("`{}`", name.replace('`', "``"))
    }

    fn supports_partial_indexes(&self) -> bool {
        false
    }

    fn db_type(&self, field_type: &str) -> String {
        if let Some(length) = parse_optional_length(field_type, "CharField") {
            let length = length.unwrap_or(255);
            return format!("varchar({length})");
        }

        match field_type.trim() {
            "AutoField" => "integer AUTO_INCREMENT".to_string(),
            "BigAutoField" => "bigint AUTO_INCREMENT".to_string(),
            "IntegerField" | "PositiveIntegerField" | "PositiveSmallIntegerField" => {
                "integer".to_string()
            }
            "BigIntegerField" | "PositiveBigIntegerField" => "bigint".to_string(),
            "SmallIntegerField" | "SmallAutoField" => "smallint".to_string(),
            "BooleanField" | "NullBooleanField" => "bool".to_string(),
            "FloatField" => "double precision".to_string(),
            "DateTimeField" => "datetime(6)".to_string(),
            "TextField" => "longtext".to_string(),
            "JSONField" => "json".to_string(),
            "UUIDField" => "char(32)".to_string(),
            "BinaryField" => "longblob".to_string(),
            other => other.to_string(),
        }
    }

    fn last_insert_id_sql(&self, _table: &str, _pk_name: &str) -> String {
        "SELECT LAST_INSERT_ID()".to_string()
    }
}

impl MySqlBackend {
    #[must_use]
    pub fn supports_json(&self) -> bool {
        self.supports_json_field()
    }

    #[must_use]
    pub fn max_query_params(&self) -> usize {
        65_535
    }

    #[must_use]
    pub fn supports_returning(&self) -> bool {
        false
    }
}

#[cfg(test)]
mod tests {
    use super::{DatabaseBackend, MySqlBackend};

    #[test]
    fn mysql_backend_reports_feature_flags() {
        let backend = MySqlBackend;

        assert_eq!(backend.name(), "mysql");
        assert!(backend.supports_transactions());
        assert!(backend.supports_json());
        assert_eq!(backend.max_query_params(), 65_535);
        assert!(!backend.supports_returning());
        assert!(!backend.supports_partial_indexes());
        assert!(!backend.supports_covering_indexes());
    }

    #[test]
    fn test_mysql_quote_name() {
        let backend = MySqlBackend;

        assert_eq!(backend.quote_name("table_name"), "`table_name`");
        assert_eq!(backend.quote_name("table`name"), "`table``name`");
    }

    #[test]
    fn test_mysql_db_type_mappings() {
        let backend = MySqlBackend;

        assert_eq!(backend.db_type("AutoField"), "integer AUTO_INCREMENT");
        assert_eq!(backend.db_type("CharField"), "varchar(255)");
        assert_eq!(backend.db_type("CharField(64)"), "varchar(64)");
        assert_eq!(backend.db_type("BooleanField"), "bool");
        assert_eq!(backend.db_type("DateTimeField"), "datetime(6)");
        assert_eq!(backend.db_type("JSONField"), "json");
        assert_eq!(backend.db_type("UUIDField"), "char(32)");
        assert_eq!(backend.db_type("BinaryField"), "longblob");
    }

    #[test]
    fn test_mysql_last_insert_id() {
        let backend = MySqlBackend;

        assert_eq!(
            backend.last_insert_id_sql("widgets", "id"),
            "SELECT LAST_INSERT_ID()"
        );
    }

    #[test]
    fn test_mysql_no_partial_indexes() {
        let backend = MySqlBackend;

        assert!(!backend.supports_partial_indexes());
    }
}