use elfmind::*;
use knowledge::*;

fn main() {
    let mut ctx = Context::default();
    let mut db = Database::default();
    db.add_fact(Relation::TimeIs(Hole::Val(6)));
    db.add_fact(Relation::TimeIs(Hole::Val(7)));
    db.add_fact(Relation::HasItem(Hole::Val(character::Character::Tyumici), Hole::Val(item::Item::Apple)));
    db.add_fact(Relation::FriendOf(Hole::Val(character::Character::Bobo), Hole::Val(character::Character::Naesten)));
    let vt = ctx.var("time");
    let vi = ctx.var("item");
    db.add_rule(Rule {
        head: Relation::UsedItemAt(
            Hole::Val(character::Character::Tyumici),
            Hole::Var(vi),
            Hole::Var(vt),
        ),
        body: vec![
            Relation::TimeIs(Hole::Var(vt)),
            Relation::HasItem(Hole::Val(character::Character::Tyumici), Hole::Var(vi)),
        ],
    });
    let vt = ctx.var("time");
    let vc = ctx.var("char");
    db.add_rule(Rule {
        head: Relation::FriendOf(
            Hole::Val(character::Character::Naesten),
            Hole::Var(vc),
        ),
        body: vec![
            Relation::TimeIs(Hole::Var(vt)),
            Relation::UsedItemAt(Hole::Var(vc), Hole::Val(item::Item::Apple), Hole::Var(vt)),
        ],
    });
    let vc1 = ctx.var("char");
    let vc2 = ctx.var("char");
    let vc3 = ctx.var("char");
    db.add_rule(Rule {
        head: Relation::FriendOf(
            Hole::Var(vc1),
            Hole::Var(vc3),
        ),
        body: vec![
            Relation::FriendOf(Hole::Var(vc1), Hole::Var(vc2)),
            Relation::FriendOf(Hole::Var(vc2), Hole::Var(vc3)),
        ],
    });
    db.apply_rules();
    println!("{:#?}", db);
}