iridium_server 0.1.12

use crate::tds::packet::PacketReader;
use std::io;

use super::types::DecodedValue;

#[derive(Debug, Clone)]
struct TvpColMeta {
    type_id: u8,
    precision: u8,
    scale: u8,
}

pub(super) fn read_typed_value(r: &mut PacketReader, type_id: u8) -> io::Result<DecodedValue> {
    match type_id {
        // INTNTYPE
        0x26 => {
            let max_len = r.read_u8()?;
            let actual_len = r.read_u8()?;
            if actual_len == 0 || r.remaining() < actual_len as usize {
                return Ok(scalar(int_type_name(max_len), "NULL".into()));
            }
            let bytes = r.read_bytes(actual_len as usize)?;
            Ok(scalar(
                int_type_name(max_len),
                read_le_int(bytes).to_string(),
            ))
        }
        // BITNTYPE
        0x68 => {
            let _max = r.read_u8()?;
            let actual = r.read_u8()?;
            if actual == 0 {
                return Ok(scalar("BIT".into(), "NULL".into()));
            }
            let b = r.read_u8()?;
            Ok(scalar(
                "BIT".into(),
                if b != 0 { "1".into() } else { "0".into() },
            ))
        }
        // FLTNTYPE
        0x6D => {
            let _max_len = r.read_u8()?;
            let actual_len = r.read_u8()?;
            if actual_len == 0 || r.remaining() < actual_len as usize {
                return Ok(scalar("FLOAT".into(), "NULL".into()));
            }
            let bytes = r.read_bytes(actual_len as usize)?;
            let val = if actual_len == 4 {
                f32::from_le_bytes([bytes[0], bytes[1], bytes[2], bytes[3]]) as f64
            } else {
                f64::from_le_bytes(bytes.try_into().unwrap_or([0u8; 8]))
            };
            Ok(scalar("FLOAT".into(), format!("{}", val)))
        }
        // NVARCHARTYPE
        0xE7 => {
            if r.remaining() < 7 {
                return Ok(scalar("NVARCHAR(MAX)".into(), "NULL".into()));
            }
            let max_len = r.read_u16_le()?;
            r.skip(5)?;
            let actual_len = r.read_u16_le()? as usize;
            if actual_len == 0xFFFF || r.remaining() < actual_len {
                return Ok(scalar("NVARCHAR(MAX)".into(), "NULL".into()));
            }
            let bytes = r.read_bytes(actual_len)?;
            let s = decode_utf16le(bytes);
            let ty = if max_len == 0xFFFF {
                "NVARCHAR(MAX)".to_string()
            } else {
                format!("NVARCHAR({})", max_len / 2)
            };
            Ok(scalar(ty, format!("N'{}'", s.replace('\'', "''"))))
        }
        // BIGVARCHARTYPE
        0xA7 => {
            if r.remaining() < 7 {
                return Ok(scalar("VARCHAR(MAX)".into(), "NULL".into()));
            }
            let max_len = r.read_u16_le()?;
            r.skip(5)?;
            let actual_len = r.read_u16_le()? as usize;
            if actual_len == 0xFFFF || r.remaining() < actual_len {
                return Ok(scalar("VARCHAR(MAX)".into(), "NULL".into()));
            }
            let bytes = r.read_bytes(actual_len)?;
            let s = String::from_utf8_lossy(bytes).to_string();
            let ty = if max_len == 0xFFFF {
                "VARCHAR(MAX)".to_string()
            } else {
                format!("VARCHAR({})", max_len)
            };
            Ok(scalar(ty, format!("'{}'", s.replace('\'', "''"))))
        }
        // NTEXTTYPE
        0x63 => {
            if r.remaining() < 13 {
                return Ok(scalar("NTEXT".into(), "NULL".into()));
            }
            let _max_len = r.read_u32_le()?;
            r.skip(5)?;
            let actual_len = r.read_u32_le()? as usize;
            if r.remaining() < actual_len {
                return Ok(scalar("NTEXT".into(), "NULL".into()));
            }
            let bytes = r.read_bytes(actual_len)?;
            let s = decode_utf16le(bytes);
            Ok(scalar(
                "NTEXT".into(),
                format!("N'{}'", s.replace('\'', "''")),
            ))
        }
        // TEXTTYPE
        0x23 => {
            if r.remaining() < 13 {
                return Ok(scalar("TEXT".into(), "NULL".into()));
            }
            let _max_len = r.read_u32_le()?;
            r.skip(5)?;
            let actual_len = r.read_u32_le()? as usize;
            if r.remaining() < actual_len {
                return Ok(scalar("TEXT".into(), "NULL".into()));
            }
            let bytes = r.read_bytes(actual_len)?;
            let s = String::from_utf8_lossy(bytes).to_string();
            Ok(scalar(
                "TEXT".into(),
                format!("'{}'", s.replace('\'', "''")),
            ))
        }
        // DECIMALNTYPE / NUMERICNTYPE
        0x6A | 0x6C => {
            let _max_len = r.read_u8()?;
            let precision = r.read_u8()?;
            let scale = r.read_u8()?;
            let actual_len = r.read_u8()?;
            let ty = format!("DECIMAL({},{})", precision, scale);
            if actual_len == 0 || r.remaining() < actual_len as usize {
                return Ok(scalar(ty, "NULL".into()));
            }
            let bytes = r.read_bytes(actual_len as usize)?;
            let sign = bytes[0];
            let mut val: i128 = 0;
            for i in (1..bytes.len()).rev() {
                val = (val << 8) | bytes[i] as i128;
            }
            if sign == 0 {
                val = -val;
            }
            Ok(scalar(ty, format_decimal(val, scale)))
        }
        // GUIDTYPE
        0x24 => {
            let actual_len = r.read_u8()?;
            if actual_len == 0 || r.remaining() < 16 {
                return Ok(scalar("UNIQUEIDENTIFIER".into(), "NULL".into()));
            }
            let b = r.read_bytes(16)?;
            let guid = format!(
                "{:02X}{:02X}{:02X}{:02X}-{:02X}{:02X}-{:02X}{:02X}-{:02X}{:02X}-{:02X}{:02X}{:02X}{:02X}{:02X}{:02X}",
                b[3],
                b[2],
                b[1],
                b[0],
                b[5],
                b[4],
                b[7],
                b[6],
                b[8],
                b[9],
                b[10],
                b[11],
                b[12],
                b[13],
                b[14],
                b[15]
            );
            Ok(scalar("UNIQUEIDENTIFIER".into(), format!("'{}'", guid)))
        }
        // TVPTYPE
        0xF3 => read_tvp_value(r),
        _ => Ok(scalar("NVARCHAR(MAX)".into(), "NULL".into())),
    }
}

fn scalar(type_name: String, value_sql: String) -> DecodedValue {
    DecodedValue {
        type_name,
        value_sql,
        tvp_rows: None,
    }
}

fn read_tvp_value(r: &mut PacketReader) -> io::Result<DecodedValue> {
    let _db = read_b_varchar_utf16(r).unwrap_or_default();
    let schema = read_b_varchar_utf16(r).unwrap_or_else(|| "dbo".to_string());
    let type_name = read_b_varchar_utf16(r).unwrap_or_else(|| "TVP".to_string());
    let full_type = format!("{}.{}", schema, type_name);

    if r.remaining() < 2 {
        return Ok(DecodedValue {
            type_name: full_type,
            value_sql: "NULL".into(),
            tvp_rows: Some(vec![]),
        });
    }

    let col_count = r.read_u16_le()?;
    if col_count == 0xFFFF {
        return Ok(DecodedValue {
            type_name: full_type,
            value_sql: "NULL".into(),
            tvp_rows: Some(vec![]),
        });
    }

    let mut cols = Vec::new();
    for _ in 0..col_count {
        if r.remaining() < 7 {
            return Ok(DecodedValue {
                type_name: full_type,
                value_sql: "NULL".into(),
                tvp_rows: Some(vec![]),
            });
        }
        let _user_type = r.read_u32_le()?;
        let _flags = r.read_u16_le()?;
        let type_id = r.read_u8()?;
        cols.push(read_tvp_col_meta(r, type_id)?);

        if r.remaining() < 1 {
            return Ok(DecodedValue {
                type_name: full_type,
                value_sql: "NULL".into(),
                tvp_rows: Some(vec![]),
            });
        }
        let col_name_len = r.read_u8()? as usize;
        if col_name_len > 0 {
            if r.remaining() < col_name_len * 2 {
                return Ok(DecodedValue {
                    type_name: full_type,
                    value_sql: "NULL".into(),
                    tvp_rows: Some(vec![]),
                });
            }
            r.skip(col_name_len * 2)?;
        }
    }

    let mut rows: Vec<Vec<String>> = Vec::new();
    loop {
        if r.remaining() < 1 {
            break;
        }
        let token = r.read_u8()?;
        if token == 0x00 {
            break;
        }
        if token != 0x01 {
            break;
        }
        let mut row = Vec::with_capacity(cols.len());
        for col in &cols {
            row.push(read_tvp_cell_value(r, col)?);
        }
        rows.push(row);
    }

    Ok(DecodedValue {
        type_name: full_type,
        value_sql: "NULL".into(),
        tvp_rows: Some(rows),
    })
}

fn read_tvp_col_meta(r: &mut PacketReader, type_id: u8) -> io::Result<TvpColMeta> {
    let mut meta = TvpColMeta {
        type_id,
        precision: 0,
        scale: 0,
    };

    match type_id {
        0x26 | 0x68 | 0x6D => {
            if r.remaining() >= 1 {
                let _ = r.read_u8()?;
            }
        }
        0xA7 | 0xE7 => {
            if r.remaining() >= 2 {
                let _ = r.read_u16_le()?;
            }
            if r.remaining() >= 5 {
                r.skip(5)?;
            }
        }
        0x6A | 0x6C => {
            if r.remaining() >= 1 {
                let _ = r.read_u8()?;
            }
            if r.remaining() >= 1 {
                meta.precision = r.read_u8()?;
            }
            if r.remaining() >= 1 {
                meta.scale = r.read_u8()?;
            }
        }
        _ => {}
    }

    Ok(meta)
}

fn read_tvp_cell_value(r: &mut PacketReader, col: &TvpColMeta) -> io::Result<String> {
    match col.type_id {
        0x26 => {
            if r.remaining() < 1 {
                return Ok("NULL".into());
            }
            let len = r.read_u8()? as usize;
            if len == 0 || r.remaining() < len {
                return Ok("NULL".into());
            }
            let bytes = r.read_bytes(len)?;
            Ok(read_le_int(bytes).to_string())
        }
        0x68 => {
            if r.remaining() < 1 {
                return Ok("NULL".into());
            }
            let len = r.read_u8()? as usize;
            if len == 0 || r.remaining() < len {
                return Ok("NULL".into());
            }
            let b = r.read_u8()?;
            Ok(if b == 0 { "0".into() } else { "1".into() })
        }
        0xA7 => {
            if r.remaining() < 2 {
                return Ok("NULL".into());
            }
            let len = r.read_u16_le()? as usize;
            if len == 0xFFFF || r.remaining() < len {
                return Ok("NULL".into());
            }
            let bytes = r.read_bytes(len)?;
            let s = String::from_utf8_lossy(bytes).to_string();
            Ok(format!("'{}'", s.replace('\'', "''")))
        }
        0xE7 => {
            if r.remaining() < 2 {
                return Ok("NULL".into());
            }
            let len = r.read_u16_le()? as usize;
            if len == 0xFFFF || r.remaining() < len {
                return Ok("NULL".into());
            }
            let bytes = r.read_bytes(len)?;
            let s = decode_utf16le(bytes);
            Ok(format!("N'{}'", s.replace('\'', "''")))
        }
        0x6A | 0x6C => {
            if r.remaining() < 1 {
                return Ok("NULL".into());
            }
            let len = r.read_u8()? as usize;
            if len == 0 || r.remaining() < len {
                return Ok("NULL".into());
            }
            let bytes = r.read_bytes(len)?;
            let sign = bytes[0];
            let mut val: i128 = 0;
            for i in (1..bytes.len()).rev() {
                val = (val << 8) | bytes[i] as i128;
            }
            if sign == 0 {
                val = -val;
            }
            Ok(format_decimal(val, col.scale))
        }
        _ => Ok("NULL".into()),
    }
}

fn read_b_varchar_utf16(r: &mut PacketReader) -> Option<String> {
    if r.remaining() < 1 {
        return None;
    }
    let len = r.read_u8().ok()? as usize;
    if len == 0 {
        return Some(String::new());
    }
    if r.remaining() < len * 2 {
        return None;
    }
    let bytes = r.read_bytes(len * 2).ok()?;
    Some(decode_utf16le(bytes))
}

fn int_type_name(max_len: u8) -> String {
    match max_len {
        1 => "TINYINT".into(),
        2 => "SMALLINT".into(),
        8 => "BIGINT".into(),
        _ => "INT".into(),
    }
}

fn read_le_int(bytes: &[u8]) -> i64 {
    match bytes.len() {
        1 => bytes[0] as i64,
        2 => i16::from_le_bytes([bytes[0], bytes[1]]) as i64,
        4 => i32::from_le_bytes([bytes[0], bytes[1], bytes[2], bytes[3]]) as i64,
        8 => i64::from_le_bytes(bytes.try_into().unwrap_or([0u8; 8])),
        _ => 0,
    }
}

pub(super) fn decode_utf16le(bytes: &[u8]) -> String {
    let mut u16s = Vec::with_capacity(bytes.len() / 2);
    for chunk in bytes.chunks_exact(2) {
        u16s.push(u16::from_le_bytes([chunk[0], chunk[1]]));
    }
    String::from_utf16_lossy(&u16s).to_string()
}

fn format_decimal(val: i128, scale: u8) -> String {
    if scale == 0 {
        return val.to_string();
    }
    let divisor = 10i128.pow(scale as u32);
    let int_part = val / divisor;
    let frac_part = (val % divisor).abs();
    format!(
        "{}.{:0>width$}",
        int_part,
        frac_part,
        width = scale as usize
    )
}