summaryrefslogtreecommitdiffstats
path: root/smartmatch.xs
blob: 36d7c51bf2470fe2ad1fb1f45c19a8e4824ad4c3 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#include "EXTERN.h"
#include "perl.h"
#include "XSUB.h"

#include "ppport.h"

#include "hook_op_check_smartmatch.h"

STATIC OP*
smartmatch_cb(pTHX_ OP *o, void *user_data)
{
    OP *left, *right, *cb_op, *list, *new;
    SV **cb;

    cb = hv_fetchs(GvHV(PL_hintgv), "smartmatch_cb", 0);
    if (!cb) {
        return o;
    }

    left = cBINOPo->op_first;
    right = left->op_sibling;

    o->op_flags &= ~OPf_KIDS;
    op_free(o);

    cb_op = newCVREF(0, newSVOP(OP_CONST, 0, newSVsv(*cb)));
    list = newLISTOP(OP_LIST, 0, left, right);
    new = newUNOP(OP_ENTERSUB, OPf_STACKED,
                  op_append_elem(OP_LIST, list, cb_op));

    return new;
}

UV
hook_op_check_smartmatch()
{
    return hook_op_check(OP_SMARTMATCH, smartmatch_cb, NULL);
}

MODULE = smartmatch  PACKAGE = smartmatch

PROTOTYPES: DISABLE

BOOT:
    hook_op_check_smartmatch();

void
register (cb)
    SV *cb;
    CODE:
        if (!SvROK(cb) || SvTYPE(SvRV(cb)) != SVt_PVCV) {
            croak("not a coderef");
        }

        PL_hints |= HINT_LOCALIZE_HH;
        gv_HVadd(PL_hintgv);

        SvREFCNT_inc(cb);
        if (!hv_stores(GvHV(PL_hintgv), "smartmatch_cb", cb)) {
            SvREFCNT_dec(cb);
            croak("couldn't store the callback");
        }

void
unregister ()
    CODE:
        PL_hints |= HINT_LOCALIZE_HH;
        gv_HVadd(PL_hintgv);

        hv_delete(GvHV(PL_hintgv), "smartmatch_cb", 13, G_DISCARD);