import option;
import str;
import io::reader_util;

import common::config;

export test_props;
export load_props;
export is_test_ignored;

type test_props = {
    // Lines that should be expected, in order, on standard out
    error_patterns: [str],
    // Extra flags to pass to the compiler
    compile_flags: option<str>,
    // If present, the name of a file that this test should match when
    // pretty-printed
    pp_exact: option<str>,
    // Modules from aux directory that should be compiled
    aux_builds: [str]
};

// Load any test directives embedded in the file
fn load_props(testfile: str) -> test_props {
    let mut error_patterns = [];
    let mut aux_builds = [];
    let mut compile_flags = option::none;
    let mut pp_exact = option::none;
    iter_header(testfile) {|ln|
        alt parse_error_pattern(ln) {
          option::some(ep) { error_patterns += [ep]; }
          option::none { }
        };

        if option::is_none(compile_flags) {
            compile_flags = parse_compile_flags(ln);
        }

        if option::is_none(pp_exact) {
            pp_exact = parse_pp_exact(ln, testfile);
        }

        option::may(parse_aux_build(ln)) {|ab|
            aux_builds += [ab];
        }
    };
    ret {
        error_patterns: error_patterns,
        compile_flags: compile_flags,
        pp_exact: pp_exact,
        aux_builds: aux_builds
    };
}

fn is_test_ignored(config: config, testfile: str) -> bool {
    let mut found = false;
    iter_header(testfile) {|ln|
        // FIXME: Can't return or break from iterator
        // (Fix when Issue #1619 is resolved)
        found = found || parse_name_directive(ln, "xfail-test");
        found = found || parse_name_directive(ln, xfail_target());
        if (config.mode == common::mode_pretty) {
            found = found || parse_name_directive(ln, "xfail-pretty");
        }
    };
    ret found;

    fn xfail_target() -> str {
        "xfail-" + os::sysname()
    }
}

fn iter_header(testfile: str, it: fn(str)) {
    let rdr = result::get(io::file_reader(testfile));
    while !rdr.eof() {
        let ln = rdr.read_line();

        // Assume that any directives will be found before the first
        // module or function. This doesn't seem to be an optimization
        // with a warm page cache. Maybe with a cold one.
        if str::starts_with(ln, "fn")
            || str::starts_with(ln, "mod") {
            break;
        } else { it(ln); }
    }
}

fn parse_error_pattern(line: str) -> option<str> {
    parse_name_value_directive(line, "error-pattern")
}

fn parse_aux_build(line: str) -> option<str> {
    parse_name_value_directive(line, "aux-build")
}

fn parse_compile_flags(line: str) -> option<str> {
    parse_name_value_directive(line, "compile-flags")
}

fn parse_pp_exact(line: str, testfile: str) -> option<str> {
    alt parse_name_value_directive(line, "pp-exact") {
      option::some(s) { option::some(s) }
      option::none {
        if parse_name_directive(line, "pp-exact") {
            option::some(path::basename(testfile))
        } else {
            option::none
        }
      }
    }
}

fn parse_name_directive(line: str, directive: str) -> bool {
    str::contains(line, directive)
}

fn parse_name_value_directive(line: str,
                              directive: str) -> option<str> unsafe {
    let keycolon = directive + ":";
    alt str::find_str(line, keycolon) {
        option::some(colon) {
            let value = str::slice(line, colon + str::len(keycolon),
                                   str::len(line));
            #debug("%s: %s", directive,  value);
            option::some(value)
        }
        option::none { option::none }
    }
}