p6.p6

use nqp;

sub infix:<zo> (&f, &g) {
    my \ret = &f.count > 1
        ?? sub (|args) { f |g |args }
        !! sub (|args) { f  g |args }
    nqp::bindattr(ret, Code, '$!signature',
        nqp::getattr(nqp::decont(&g), Code, '$!signature'));
    ret;
}

sub one ($a, $b) { $a + $b };
sub two ($x) { $x² }

my &combined = &two zo &one;
say combined 2, 4;
say combined(2, 4)³;

sub three ($x) { $x³ }
my &co-combined = &three zo &combined;
say co-combined 2, 4;

say &one.count;
say &one.arity;

say &co-combined.count;
say &co-combined.arity;