use tandem_garble_interop::{
    check_program, compile_program, deserialize_output, serialize_input, Role,
};

#[test]
fn compute_score() -> Result<(), String> {
    let credit_scoring = include_str!("credit_scoring_setup/credit_scoring.garble.rs");
    println!("Parsing and type-checking...");
    let typed_prg = check_program(credit_scoring)?;

    println!("Compiling...");
    let circuit = compile_program(&typed_prg, "compute_score")?;

    println!("Running program...");
    let credit_scorer_input = serialize_input(
        Role::Contributor,
        &typed_prg,
        &circuit.fn_def,
        SCORING_ALGORITHM,
    )?;

    let user_input = serialize_input(Role::Evaluator, &typed_prg, &circuit.fn_def, USER)?;

    let result = tandem::simulate(&circuit.gates, &credit_scorer_input, &user_input).unwrap();

    let score = deserialize_output(&typed_prg, &circuit.fn_def, &result)?;

    assert_eq!(score.to_string(), "Score::Good(85u8)");

    Ok(())
}

const SCORING_ALGORITHM: &str = "
ScoringAlgorithm {
    age_score: [
        MatchClause::Range(
            Range {
                min: 0i64,
                max: 18i64,
            },
            Points { inc: 0i32 },
        ),
        MatchClause::Range(
            Range {
                min: 18i64,
                max: 35i64,
            },
            Points { inc: 50i32 },
        ),
        MatchClause::Range(
            Range {
                min: 35i64,
                max: 65i64,
            },
            Points { inc: 100i32 },
        ),
        MatchClause::Range(
            Range {
                min: 65i64,
                max: 120i64,
            },
            Points { inc: 50i32 },
        ),
    ],
    income_score: [
        MatchClause::Range(
            Range {
                min: 2000i64,
                max: 5000i64,
            },
            Points { inc: 50i32 },
        ),
        MatchClause::Range(
            Range {
                min: 5000i64,
                max: 10000i64,
            },
            Points { inc: 100i32 },
        ),
        MatchClause::Range(
            Range {
                min: 10000i64,
                max: 999999999i64,
            },
            Points { inc: 200i32 },
        ),
        MatchClause::None,
    ],
    account_balance_score: [
        MatchClause::Range(
            Range {
                min: -999999999i64,
                max: 0i64,
            },
            Points { inc: -100i32 },
        ),
        MatchClause::Range(
            Range {
                min: 0i64,
                max: 5000i64,
            },
            Points { inc: 0i32 },
        ),
        MatchClause::Range(
            Range {
                min: 5000i64,
                max: 10000i64,
            },
            Points { inc: 50i32 },
        ),
        MatchClause::Range(
            Range {
                min: 10000i64,
                max: 999999999i64,
            },
            Points { inc: 200i32 },
        ),
    ],
    current_loans_score: [
        MatchClause::Range(
            Range {
                min: 500000i64,
                max: 100000000i64,
            },
            Points { inc: 0i32 },
        ),
        MatchClause::Range(
            Range {
                min: 100000i64,
                max: 500000i64,
            },
            Points { inc: 150i32 },
        ),
        MatchClause::Range(
            Range {
                min: 0i64,
                max: 100000i64,
            },
            Points { inc: 300i32 },
        ),
        MatchClause::None,
    ],
    credit_card_score: [
        MatchClause::Range(
            Range {
                min: 0i64,
                max: 10000i64,
            },
            Points { inc: 100i32 },
        ),
        MatchClause::None,
        MatchClause::None,
        MatchClause::None,
    ],
    bankruptcy_score: [
        MatchClause::Bool(true, Points { inc: -100i32 }),
        MatchClause::Bool(false, Points { inc: 50i32 }),
    ],
    loan_payment_history_score: [
        MatchClause::Range(
            Range {
                min: 0i64,
                max: 3i64,
            },
            Points { inc: 0i32 },
        ),
        MatchClause::Range(
            Range {
                min: 3i64,
                max: 6i64,
            },
            Points { inc: -100i32 },
        ),
        MatchClause::None,
        MatchClause::None,
    ],
    credit_payment_history_score: [
        MatchClause::Range(
            Range {
                min: 0i64,
                max: 1i64,
            },
            Points { inc: 0i32 },
        ),
        MatchClause::Range(
            Range {
                min: 1i64,
                max: 3i64,
            },
            Points { inc: -100i32 },
        ),
        MatchClause::Range(
            Range {
                min: 3i64,
                max: 6i64,
            },
            Points { inc: -200i32 },
        ),
        MatchClause::None,
    ],
    surety_income_score: [
        MatchClause::Range(
            Range {
                min: 0i64,
                max: 1000i64,
            },
            Points { inc: -50i32 },
        ),
        MatchClause::Range(
            Range {
                min: 1000i64,
                max: 5000i64,
            },
            Points { inc: 0i32 },
        ),
        MatchClause::Range(
            Range {
                min: 5000i64,
                max: 10000i64,
            },
            Points { inc: 100i32 },
        ),
        MatchClause::None,
    ],
    score_limits: ScoreLimits {
        min: 0i32,
        max: 1000i32,
    },
}";

const USER: &str = "
User {
    age: 37u8,
    income: 5500u32,
    account_balance: 25000i64,
    current_loans: 60000u64,
    credit_card_limit: 1000u32,
    ever_bankrupt: false,
    loan_payment_failures: 0u8,
    credit_payment_failures: 2u8,
    surety_income: 5000u32,
}";