sql-cli 1.67.2

SQL query tool for CSV/JSON with both interactive TUI and non-interactive CLI modes - perfect for exploration and automation
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

use anyhow::{anyhow, Result};

use crate::data::datatable::DataValue;
use crate::data::unit_converter::convert_units;
use crate::sql::functions::{ArgCount, FunctionCategory, FunctionSignature, SqlFunction};

/// CONVERT function - Convert values between different units
pub struct ConvertFunction;

impl SqlFunction for ConvertFunction {
    fn signature(&self) -> FunctionSignature {
        FunctionSignature {
            name: "CONVERT",
            category: FunctionCategory::Conversion,
            arg_count: ArgCount::Fixed(3),
            description: "Convert a value from one unit to another",
            returns: "FLOAT",
            examples: vec![
                "SELECT CONVERT(100, 'km', 'miles')",         // Distance
                "SELECT CONVERT(0, 'celsius', 'fahrenheit')", // Temperature
                "SELECT CONVERT(1, 'gallon', 'liters')",      // Volume
                "SELECT CONVERT(70, 'kg', 'pounds')",         // Weight
                "SELECT CONVERT(1, 'bar', 'psi')",            // Pressure
            ],
        }
    }

    fn evaluate(&self, args: &[DataValue]) -> Result<DataValue> {
        self.validate_args(args)?;

        // Extract numeric value
        let value = match &args[0] {
            DataValue::Integer(n) => *n as f64,
            DataValue::Float(f) => *f,
            DataValue::Null => return Ok(DataValue::Null),
            _ => return Err(anyhow!("CONVERT first argument must be numeric")),
        };

        // Extract from_unit
        let from_unit = match &args[1] {
            DataValue::String(s) => s.as_str(),
            DataValue::InternedString(s) => s.as_str(),
            DataValue::Null => return Ok(DataValue::Null),
            _ => {
                return Err(anyhow!(
                    "CONVERT second argument must be a string (from_unit)"
                ))
            }
        };

        // Extract to_unit
        let to_unit = match &args[2] {
            DataValue::String(s) => s.as_str(),
            DataValue::InternedString(s) => s.as_str(),
            DataValue::Null => return Ok(DataValue::Null),
            _ => return Err(anyhow!("CONVERT third argument must be a string (to_unit)")),
        };

        // Perform conversion
        match convert_units(value, from_unit, to_unit) {
            Ok(result) => Ok(DataValue::Float(result)),
            Err(e) => Err(anyhow!("Unit conversion failed: {}", e)),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_convert_distance() {
        let func = ConvertFunction;

        // Test km to miles
        let args = vec![
            DataValue::Float(100.0),
            DataValue::String("km".to_string()),
            DataValue::String("miles".to_string()),
        ];
        let result = func.evaluate(&args).unwrap();
        if let DataValue::Float(val) = result {
            assert!((val - 62.137).abs() < 0.01);
        } else {
            panic!("Expected Float result");
        }
    }

    #[test]
    fn test_convert_temperature() {
        let func = ConvertFunction;

        // Test Celsius to Fahrenheit (0°C = 32°F)
        let args = vec![
            DataValue::Integer(0),
            DataValue::String("celsius".to_string()),
            DataValue::String("fahrenheit".to_string()),
        ];
        let result = func.evaluate(&args).unwrap();
        if let DataValue::Float(val) = result {
            assert!((val - 32.0).abs() < 0.01);
        } else {
            panic!("Expected Float result");
        }
    }

    #[test]
    fn test_convert_with_null() {
        let func = ConvertFunction;

        // Test with NULL value
        let args = vec![
            DataValue::Null,
            DataValue::String("km".to_string()),
            DataValue::String("miles".to_string()),
        ];
        let result = func.evaluate(&args).unwrap();
        assert!(matches!(result, DataValue::Null));
    }

    #[test]
    fn test_convert_invalid_unit() {
        let func = ConvertFunction;

        // Test with invalid unit
        let args = vec![
            DataValue::Float(100.0),
            DataValue::String("invalid_unit".to_string()),
            DataValue::String("miles".to_string()),
        ];
        let result = func.evaluate(&args);
        assert!(result.is_err());
    }
}