/// @file

#include "jets/q.h"
#include "jets/w.h"

#include "noun.h"

u3_noun
u3kdb_jab(u3_noun a,
          u3_noun key,
          u3_noun fun)
  {
    if ( u3_nul == a ) {
      return u3m_bail(c3__exit);
    }
    else {
      u3_noun n_a, lr_a;
      u3_noun pn_a, qn_a;
      u3_noun pro;
      u3x_cell(a, &n_a, &lr_a);
      u3x_cell(n_a, &pn_a, &qn_a);
      
      if ( (c3y == u3r_sing(key, pn_a)) ) {
        u3_noun value = u3n_slam_on(fun, u3k(qn_a));
        pro = u3nc(u3nc(u3k(u3h(u3h(a))), value), u3k(u3t(a)));
        u3z(a); u3z(key);
        return pro;
      }
      else {
        u3_noun l_a, r_a;
        u3x_cell(lr_a, &l_a, &r_a);
        
        u3k(n_a); u3k(l_a); u3k(r_a);
        if ( c3y == u3qc_gor(key, pn_a) ) {
          pro = u3nt(n_a, u3kdb_jab(l_a, key, fun), r_a);
        }
        else {
          pro = u3nt(n_a, l_a, u3kdb_jab(r_a, key, fun));
        }
        u3z(a);
        return pro;
    }
  }
}

u3_noun
u3qdb_jab(u3_noun a,
          u3_noun key,
          u3_noun fun)
{
  return u3kdb_jab(u3k(a), u3k(key), u3k(fun));
}

u3_noun
u3wdb_jab(u3_noun cor)
{
  u3_noun a, key, fun;
  u3x_mean(cor, u3x_sam_2,   &key,
                u3x_sam_3,   &fun,
                u3x_con_sam, &a, 0);

  return u3qdb_jab(a, key, fun);
}