// Copyright 2013-2014 Simon Sapin.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.


use std::char;
use std::u32;
use std::path;
use super::{UrlParser, Url, RelativeSchemeData, NonRelativeSchemeData, Domain};
use super::UrlRelativeSchemeData;


#[test]
fn url_parsing() {
    for test in parse_test_data(include_str!("urltestdata.txt")).into_iter() {
        let Test {
            input: input,
            base: base,
            scheme: expected_scheme,
            username: expected_username,
            password: expected_password,
            host: expected_host,
            port: expected_port,
            path: expected_path,
            query: expected_query,
            fragment: expected_fragment,
            expected_failure: expected_failure,
        } = test;
        let base = match Url::parse(base.as_slice()) {
            Ok(base) => base,
            Err(message) => panic!("Error parsing base {}: {}", base, message)
        };
        let url = UrlParser::new().base_url(&base).parse(input.as_slice());
        if expected_scheme.is_none() {
            if url.is_ok() && !expected_failure {
                panic!("Expected a parse error for URL {}", input);
            }
            continue
        }
        let Url { scheme, scheme_data, query, fragment, .. } = match url {
            Ok(url) => url,
            Err(message) => {
                if expected_failure {
                    continue
                } else {
                    panic!("Error parsing URL {}: {}", input, message)
                }
            }
        };

        macro_rules! assert_eq {
            ($a: expr, $b: expr) => {
                {
                    let a = $a;
                    let b = $b;
                    if a != b {
                        if expected_failure {
                            continue
                        } else {
                            panic!("{} != {}", a, b)
                        }
                    }
                }
            }
        }

        assert_eq!(Some(scheme), expected_scheme);
        match scheme_data {
            RelativeSchemeData(UrlRelativeSchemeData {
                username, password, host, port, default_port: _, path,
            }) => {
                assert_eq!(username, expected_username);
                assert_eq!(password, expected_password);
                let host = host.serialize();
                assert_eq!(host, expected_host)
                assert_eq!(port, expected_port);
                assert_eq!(Some(format!("/{}", path.connect("/"))), expected_path);
            },
            NonRelativeSchemeData(scheme_data) => {
                assert_eq!(Some(scheme_data), expected_path);
                assert_eq!(String::new(), expected_username);
                assert_eq!(None, expected_password);
                assert_eq!(String::new(), expected_host);
                assert_eq!(None, expected_port);
            },
        }
        fn opt_prepend(prefix: &str, opt_s: Option<String>) -> Option<String> {
            opt_s.map(|s| format!("{}{}", prefix, s))
        }
        assert_eq!(opt_prepend("?", query), expected_query);
        assert_eq!(opt_prepend("#", fragment), expected_fragment);

        assert!(!expected_failure, "Unexpected success for {}", input);
    }
}

struct Test {
    input: String,
    base: String,
    scheme: Option<String>,
    username: String,
    password: Option<String>,
    host: String,
    port: Option<u16>,
    path: Option<String>,
    query: Option<String>,
    fragment: Option<String>,
    expected_failure: bool,
}

fn parse_test_data(input: &str) -> Vec<Test> {
    let mut tests: Vec<Test> = Vec::new();
    for line in input.lines() {
        if line == "" || line.starts_with("#") {
            continue
        }
        let mut pieces = line.split(' ').collect::<Vec<&str>>();
        let expected_failure = pieces[0] == "XFAIL";
        if expected_failure {
            pieces.remove(0);
        }
        let input = unescape(pieces.remove(0).unwrap());
        let mut test = Test {
            input: input,
            base: if pieces.is_empty() || pieces[0] == "" {
                tests.last().unwrap().base.clone()
            } else {
                unescape(pieces.remove(0).unwrap())
            },
            scheme: None,
            username: String::new(),
            password: None,
            host: String::new(),
            port: None,
            path: None,
            query: None,
            fragment: None,
            expected_failure: expected_failure,
        };
        for piece in pieces.into_iter() {
            if piece == "" || piece.starts_with("#") {
                continue
            }
            let colon = piece.find(':').unwrap();
            let value = unescape(piece.slice_from(colon + 1));
            match piece.slice_to(colon) {
                "s" => test.scheme = Some(value),
                "u" => test.username = value,
                "pass" => test.password = Some(value),
                "h" => test.host = value,
                "port" => test.port = Some(from_str(value.as_slice()).unwrap()),
                "p" => test.path = Some(value),
                "q" => test.query = Some(value),
                "f" => test.fragment = Some(value),
                _ => panic!("Invalid token")
            }
        }
        tests.push(test)
    }
    tests
}

fn unescape(input: &str) -> String {
    let mut output = String::new();
    let mut chars = input.chars();
    loop {
        match chars.next() {
            None => return output,
            Some(c) => output.push(
                if c == '\\' {
                    match chars.next().unwrap() {
                        '\\' => '\\',
                        'n' => '\n',
                        'r' => '\r',
                        's' => ' ',
                        't' => '\t',
                        'f' => '\x0C',
                        'u' => {
                            let mut hex = String::new();
                            hex.push(chars.next().unwrap());
                            hex.push(chars.next().unwrap());
                            hex.push(chars.next().unwrap());
                            hex.push(chars.next().unwrap());
                            u32::parse_bytes(hex.as_bytes(), 16)
                                .and_then(char::from_u32).unwrap()
                        }
                        _ => panic!("Invalid test data input"),
                    }
                } else {
                    c
                }
            )
        }
    }
}


#[test]
fn file_paths() {
    assert_eq!(Url::from_file_path(&path::posix::Path::new("relative")), Err(()));
    assert_eq!(Url::from_file_path(&path::posix::Path::new("../relative")), Err(()));
    assert_eq!(Url::from_file_path(&path::windows::Path::new("relative")), Err(()));
    assert_eq!(Url::from_file_path(&path::windows::Path::new(r"..\relative")), Err(()));
    assert_eq!(Url::from_file_path(&path::windows::Path::new(r"\drive-relative")), Err(()));
    assert_eq!(Url::from_file_path(&path::windows::Path::new(r"\\ucn\")), Err(()));

    let mut url = Url::from_file_path(&path::posix::Path::new("/foo/bar")).unwrap();
    assert_eq!(url.host(), Some(&Domain("".to_string())));
    assert_eq!(url.path(), Some(["foo".to_string(), "bar".to_string()].as_slice()));
    assert!(url.to_file_path() == Ok(path::posix::Path::new("/foo/bar")));

    *url.path_mut().unwrap().get_mut(1) = "ba\0r".to_string();
    assert!(url.to_file_path::<path::posix::Path>() == Err(()));

    *url.path_mut().unwrap().get_mut(1) = "ba%00r".to_string();
    assert!(url.to_file_path::<path::posix::Path>() == Err(()));

    // Invalid UTF-8
    *url.path_mut().unwrap().get_mut(1) = "ba%80r".to_string();
    assert!(url.to_file_path() == Ok(path::posix::Path::new(
        /* note: byte string, invalid UTF-8 */ b"/foo/ba\x80r")));

    let mut url = Url::from_file_path(&path::windows::Path::new(r"C:\foo\bar")).unwrap();
    assert_eq!(url.host(), Some(&Domain("".to_string())));
    assert_eq!(url.path(), Some(["C:".to_string(), "foo".to_string(), "bar".to_string()].as_slice()));
    assert!(url.to_file_path::<path::windows::Path>()
            == Ok(path::windows::Path::new(r"C:\foo\bar")));

    *url.path_mut().unwrap().get_mut(2) = "ba\0r".to_string();
    assert!(url.to_file_path::<path::windows::Path>() == Err(()));

    *url.path_mut().unwrap().get_mut(2) = "ba%00r".to_string();
    assert!(url.to_file_path::<path::windows::Path>() == Err(()));

    // Invalid UTF-8
    *url.path_mut().unwrap().get_mut(2) = "ba%80r".to_string();
    assert!(url.to_file_path::<path::windows::Path>() == Err(()));
}


#[test]
fn directory_paths() {
    assert_eq!(Url::from_directory_path(&path::posix::Path::new("relative")), Err(()));
    assert_eq!(Url::from_directory_path(&path::posix::Path::new("../relative")), Err(()));
    assert_eq!(Url::from_directory_path(&path::windows::Path::new("relative")), Err(()));
    assert_eq!(Url::from_directory_path(&path::windows::Path::new(r"..\relative")), Err(()));
    assert_eq!(Url::from_directory_path(&path::windows::Path::new(r"\drive-relative")), Err(()));
    assert_eq!(Url::from_directory_path(&path::windows::Path::new(r"\\ucn\")), Err(()));

    let url = Url::from_directory_path(&path::posix::Path::new("/foo/bar")).unwrap();
    assert_eq!(url.host(), Some(&Domain("".to_string())));
    assert_eq!(url.path(), Some(["foo".to_string(), "bar".to_string(), "".to_string()].as_slice()));

    let url = Url::from_directory_path(&path::windows::Path::new(r"C:\foo\bar")).unwrap();
    assert_eq!(url.host(), Some(&Domain("".to_string())));
    assert_eq!(url.path(), Some([
        "C:".to_string(), "foo".to_string(), "bar".to_string(), "".to_string()].as_slice()));
}