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

//! `list()` - builtin function
//!
//! In the future this will be replaced by a type object.
use std::borrow::Borrow;

use ::api::method::{GetItem,Iter};
use ::api::result::{ObjectResult};
use ::api::RtObject;
use ::modules::builtins::Type;
use ::modules::precondition::{check_args, check_kwargs, check_args_range};
use ::resources::strings;
use ::runtime::Runtime;
use ::runtime::traits::{ListProvider, DefaultListProvider, IntegerProvider};
use ::system::primitives as rs;
use ::system::primitives::{Func, FuncType, SignatureBuilder};


pub struct ListFn;


impl ListFn {
    pub fn create() -> rs::Func {
        trace!("create builtin"; "function" => "list");
        let callable: Box<rs::WrapperFn> = Box::new(rs_builtin_list);

        Func {
            name: String::from("list"),
            module: String::from(strings::BUILTINS_MODULE),
            callable: FuncType::Wrapper(callable),
            signature: ["iterable"].as_args()
        }
    }
}


fn rs_builtin_list(rt: &Runtime, pos_args: &RtObject, starargs: &RtObject, kwargs: &RtObject) -> ObjectResult {
    trace!("call"; "native_builtin" => "list");

    let arg_count = check_args_range(0..2, &pos_args)?;
    check_args(0, &starargs)?;
    check_kwargs(0, &kwargs)?;
    
    if arg_count == 0 {
        return Ok(rt.default_list())
    }

    let value = pos_args.op_getitem(&rt, &rt.int(0))?;
    let new_list  = value.op_iter(&rt)?.collect::<rs::List>();
    Ok(rt.list(new_list))
}


#[cfg(test)]
mod tests {
    use ::runtime::traits::{
        TupleProvider,
        DefaultTupleProvider,
        DefaultDictProvider
    };
    use super::*;

    fn setup() -> Runtime {
        Runtime::new()
    }

    #[test]
    fn from_tuple() {
        let rt = setup();
        let tuple = rt.tuple(vec![rt.int(1), rt.int(2), rt.int(3)]);

        let expected = rt.list(vec![rt.int(1), rt.int(2), rt.int(3)]);
        let list = rs_builtin_list(
            &rt, &rt.tuple(vec![tuple]),
            &rt.default_tuple(),
            &rt.default_dict()).unwrap();

        assert_eq!(list, expected);
    }

    #[test]
    #[should_panic]
    fn from_int() {
        let rt = setup();
        let int = rt.int(29345);

        rs_builtin_list(
            &rt, &rt.tuple(vec![int]),
            &rt.default_tuple(),
            &rt.default_dict()).unwrap();
    }
}