denoland-deno/src/handlers.rs

// Copyright 2018 Ryan Dahl <ry@tinyclouds.org>
// All rights reserved. MIT License.
extern crate libc;
#[macro_use]
extern crate log;
extern crate url;

use libc::c_char;
use std::ffi::CStr;
use std::ffi::CString;
use std::fs::File;
use std::io::Read;
use std::path::Path;
use url::Url;

mod binding;
use binding::{deno_reply_code_fetch, deno_reply_error, DenoC};

// TODO(ry) SRC_DIR is just a placeholder for future caching functionality.
static SRC_DIR: &str = "/Users/rld/.deno/src/";
const ASSET_PREFIX: &str = "/$asset$/";

#[test]
fn test_url() {
    let issue_list_url = Url::parse("https://github.com/rust-lang").unwrap();
    assert!(issue_list_url.scheme() == "https");
}

fn string_from_ptr(ptr: *const c_char) -> String {
    let cstr = unsafe { CStr::from_ptr(ptr as *const i8) };
    String::from(cstr.to_str().unwrap())
}

fn as_cstring(s: &String) -> CString {
    CString::new(s.as_str()).unwrap()
}

// Prototype: https://github.com/ry/deno/blob/golang/os.go#L56-L68
#[allow(dead_code)]
fn src_file_to_url<P: AsRef<Path>>(filename: P) -> String {
    assert!(SRC_DIR.len() > 0, "SRC_DIR shouldn't be empty");

    let filename = filename.as_ref().to_path_buf();
    let src = (SRC_DIR.as_ref() as &Path).to_path_buf();

    if filename.starts_with(&src) {
        let rest = filename.strip_prefix(&src).unwrap();
        "http://".to_string() + rest.to_str().unwrap()
    } else {
        String::from(filename.to_str().unwrap())
    }
}

#[test]
fn test_src_file_to_url() {
    assert_eq!("hello", src_file_to_url("hello"));
    assert_eq!("/hello", src_file_to_url("/hello"));
    let x = SRC_DIR.to_string() + "hello";
    assert_eq!("http://hello", src_file_to_url(&x));
    let x = SRC_DIR.to_string() + "/hello";
    assert_eq!("http://hello", src_file_to_url(&x));
}

// Prototype: https://github.com/ry/deno/blob/golang/os.go#L70-L98
// Returns (module name, local filename)
fn resolve_module(
    module_specifier: &String,
    containing_file: &String,
) -> Result<(String, String), url::ParseError> {
    info!(
        "resolve_module before module_specifier {} containing_file {}",
        module_specifier, containing_file
    );

    //let module_specifier = src_file_to_url(module_specifier);
    //let containing_file = src_file_to_url(containing_file);
    //let base_url = Url::parse(&containing_file)?;

    let j: Url = if containing_file.as_str().ends_with("/") {
        let base = Url::from_directory_path(&containing_file).unwrap();
        base.join(module_specifier)?
    } else if containing_file == "." {
        Url::from_file_path(module_specifier).unwrap()
    } else {
        let base = Url::from_file_path(&containing_file).unwrap();
        base.join(module_specifier)?
    };

    let p = j.to_file_path()
        .unwrap()
        .into_os_string()
        .into_string()
        .unwrap();

    let module_name = p.to_string();
    let filename = p.to_string();

    Ok((module_name, filename))
}

// https://github.com/ry/deno/blob/golang/os_test.go#L16-L87
#[test]
fn test_resolve_module() {
    let test_cases = [
        (
            "./subdir/print_hello.ts",
            "/Users/rld/go/src/github.com/ry/deno/testdata/006_url_imports.ts",
            "/Users/rld/go/src/github.com/ry/deno/testdata/subdir/print_hello.ts",
            "/Users/rld/go/src/github.com/ry/deno/testdata/subdir/print_hello.ts",
        ),
        (
            "testdata/001_hello.js",
            "/Users/rld/go/src/github.com/ry/deno/",
            "/Users/rld/go/src/github.com/ry/deno/testdata/001_hello.js",
            "/Users/rld/go/src/github.com/ry/deno/testdata/001_hello.js",
        ),
        (
            "/Users/rld/src/deno/hello.js",
            ".",
            "/Users/rld/src/deno/hello.js",
            "/Users/rld/src/deno/hello.js",
        ),
        /*
        (
            "http://localhost:4545/testdata/subdir/print_hello.ts",
            "/Users/rld/go/src/github.com/ry/deno/testdata/006_url_imports.ts",
            "http://localhost:4545/testdata/subdir/print_hello.ts",
            path.Join(SrcDir, "localhost:4545/testdata/subdir/print_hello.ts"),
        ),
        (
            path.Join(SrcDir, "unpkg.com/liltest@0.0.5/index.ts"),
            ".",
            "http://unpkg.com/liltest@0.0.5/index.ts",
            path.Join(SrcDir, "unpkg.com/liltest@0.0.5/index.ts"),
        ),
        (
            "./util",
            path.Join(SrcDir, "unpkg.com/liltest@0.0.5/index.ts"),
            "http://unpkg.com/liltest@0.0.5/util",
            path.Join(SrcDir, "unpkg.com/liltest@0.0.5/util"),
        ),
        */
    ];
    for &test in test_cases.iter() {
        let module_specifier = String::from(test.0);
        let containing_file = String::from(test.1);
        let (module_name, filename) = resolve_module(&module_specifier, &containing_file).unwrap();
        assert_eq!(module_name, test.2);
        assert_eq!(filename, test.3);
    }
}

// https://github.com/ry/deno/blob/golang/os.go#L100-L154
#[no_mangle]
pub extern "C" fn handle_code_fetch(
    d: *const DenoC,
    cmd_id: u32,
    module_specifier_: *const c_char,
    containing_file_: *const c_char,
) {
    let module_specifier = string_from_ptr(module_specifier_);
    let containing_file = string_from_ptr(containing_file_);

    let result = resolve_module(&module_specifier, &containing_file);
    if result.is_err() {
        let err = result.unwrap_err();
        let errmsg = format!("{} {} {}", err, module_specifier, containing_file);
        let errmsg_c = as_cstring(&errmsg);
        unsafe { deno_reply_error(d, cmd_id, errmsg_c.as_ptr()) };
        return;
    }
    let (module_name, filename) = result.unwrap();

    let mut source_code = String::new();

    debug!(
        "code_fetch. module_name = {} module_specifier = {} containing_file = {} filename = {}",
        module_name, module_specifier, containing_file, filename
    );

    if is_remote(&module_name) {
        unimplemented!();
    } else if module_name.starts_with(ASSET_PREFIX) {
        assert!(false, "Asset resolution should be done in JS, not Rust.");
    } else {
        assert!(
            module_name == filename,
            "if a module isn't remote, it should have the same filename"
        );
        let result = File::open(&filename);
        if result.is_err() {
            let err = result.unwrap_err();
            let errmsg = format!("{} {}", err, filename);
            let errmsg_c = as_cstring(&errmsg);
            unsafe { deno_reply_error(d, cmd_id, errmsg_c.as_ptr()) };
            return;
        }
        let mut f = result.unwrap();
        let result = f.read_to_string(&mut source_code);
        if result.is_err() {
            let err = result.unwrap_err();
            let errmsg = format!("{} {}", err, filename);
            let errmsg_c = as_cstring(&errmsg);
            unsafe { deno_reply_error(d, cmd_id, errmsg_c.as_ptr()) };
            return;
        }
    }

    let output_code = String::new(); //load_output_code_cache(filename, source_code);

    unsafe {
        deno_reply_code_fetch(
            d,
            cmd_id,
            as_cstring(&module_name).as_ptr(),
            as_cstring(&filename).as_ptr(),
            as_cstring(&source_code).as_ptr(),
            as_cstring(&output_code).as_ptr(),
        )
    }
}

fn is_remote(_module_name: &String) -> bool {
    false
}