extern crate libc; #[macro_use] extern crate log; extern crate flatbuffers; extern crate msg_rs as msg_generated; extern crate url; use libc::c_int; use std::env; use std::ffi::CStr; use std::ffi::CString; use std::mem; use std::ptr; mod handlers; pub use handlers::*; mod binding; use binding::{ deno_delete, deno_execute, deno_handle_msg_from_js, deno_init, deno_last_exception, deno_new, deno_set_flags, DenoC, }; // Returns args passed to V8, followed by args passed to JS fn parse_core_args(args: Vec) -> (Vec, Vec) { let mut rest = vec![]; // Filter out args that shouldn't be passed to V8 let mut args: Vec = args .into_iter() .filter(|arg| { if arg.as_str() == "--help" { rest.push(arg.clone()); return false; } true }) .collect(); // Replace args being sent to V8 for idx in 0..args.len() { if args[idx] == "--v8-options" { mem::swap(args.get_mut(idx).unwrap(), &mut String::from("--help")); } } (args, rest) } // Pass the command line arguments to v8. // Returns a vector of command line arguments that v8 did not understand. fn set_flags(args: Vec) -> Vec { // deno_set_flags(int* argc, char** argv) mutates argc and argv to remove // flags that v8 understands. // First parse core args, then converto to a vector of C strings. let (argv, rest) = parse_core_args(args); let mut argv = argv .iter() .map(|arg| CString::new(arg.as_str()).unwrap().into_bytes_with_nul()) .collect::>(); // Make a new array, that can be modified by V8::SetFlagsFromCommandLine(), // containing mutable raw pointers to the individual command line args. let mut c_argv = argv .iter_mut() .map(|arg| arg.as_mut_ptr() as *mut i8) .collect::>(); // Store the length of the argv array in a local variable. We'll pass a // pointer to this local variable to deno_set_flags(), which then // updates its value. let mut c_argc = c_argv.len() as c_int; // Let v8 parse the arguments it recognizes and remove them from c_argv. unsafe { deno_set_flags(&mut c_argc, c_argv.as_mut_ptr()); }; // If c_argc was updated we have to change the length of c_argv to match. c_argv.truncate(c_argc as usize); // Copy the modified arguments list into a proper rust vec and return it. c_argv .iter() .map(|ptr| unsafe { let cstr = CStr::from_ptr(*ptr as *const i8); let slice = cstr.to_str().unwrap(); slice.to_string() }) .chain(rest.into_iter()) .collect() } type DenoException<'a> = &'a str; struct Deno { ptr: *const DenoC, } impl Deno { fn new() -> Deno { let ptr = unsafe { deno_new(ptr::null(), deno_handle_msg_from_js) }; Deno { ptr: ptr } } fn execute( &mut self, js_filename: &str, js_source: &str, ) -> Result<(), DenoException> { let filename = CString::new(js_filename).unwrap(); let source = CString::new(js_source).unwrap(); let r = unsafe { deno_execute(self.ptr, filename.as_ptr(), source.as_ptr()) }; if r == 0 { let ptr = unsafe { deno_last_exception(self.ptr) }; let cstr = unsafe { CStr::from_ptr(ptr) }; return Err(cstr.to_str().unwrap()); } Ok(()) } } impl Drop for Deno { fn drop(&mut self) { unsafe { deno_delete(self.ptr) } } } #[test] fn test_parse_core_args_1() { let js_args = parse_core_args(vec!["deno".to_string(), "--v8-options".to_string()]); assert!(js_args == (vec!["deno".to_string(), "--help".to_string()], vec![])); } #[test] fn test_parse_core_args_2() { let js_args = parse_core_args(vec!["deno".to_string(), "--help".to_string()]); assert!(js_args == (vec!["deno".to_string()], vec!["--help".to_string()])); } fn main() { log::set_max_level(log::LevelFilter::Debug); unsafe { deno_init() }; let _js_args = set_flags(env::args().collect()); /* let v = unsafe { deno_v8_version() }; let c_str = unsafe { CStr::from_ptr(v) }; let version = c_str.to_str().unwrap(); println!("version: {}", version); */ let mut d = Deno::new(); d.execute("deno_main.js", "denoMain();") .unwrap_or_else(|err| { error!("{}", err); std::process::exit(1); }); }