initial working perl-space version
[p5sagit/Devel-Declare.git] / Declare.xs
CommitLineData
94caac6e 1#define PERL_CORE
2#define PERL_NO_GET_CONTEXT
94caac6e 3#include "EXTERN.h"
4#include "perl.h"
5#include "XSUB.h"
6#undef printf
e807ee50 7#include "stolen_chunk_of_toke.c"
94caac6e 8#include <stdio.h>
9#include <string.h>
10
9673d7ca 11#ifndef Newx
12# define Newx(v,n,t) New(0,v,n,t)
13#endif /* !Newx */
14
15#if 1
0f070758 16#define DD_HAS_TRAITS
9673d7ca 17#endif
18
53e3ab32 19#if 0
20#define DD_DEBUG
21#endif
c630715a 22
0ba8c7aa 23#define DD_HANDLE_NAME 1
24#define DD_HANDLE_PROTO 2
15d0d014 25#define DD_HANDLE_PACKAGE 8
0ba8c7aa 26
c630715a 27#ifdef DD_DEBUG
28#define DD_DEBUG_S printf("Buffer: %s\n", s);
29#else
30#define DD_DEBUG_S
31#endif
32
94caac6e 33#define LEX_NORMAL 10
34#define LEX_INTERPNORMAL 9
35
94caac6e 36/* flag to trigger removal of temporary declaree sub */
37
38static int in_declare = 0;
39
72f20f69 40/* thing that decides whether we're dealing with a declarator */
41
42int dd_is_declarator(pTHX_ char* name) {
43 HV* is_declarator;
44 SV** is_declarator_pack_ref;
45 HV* is_declarator_pack_hash;
46 SV** is_declarator_flag_ref;
47 int dd_flags;
48
49 is_declarator = get_hv("Devel::Declare::declarators", FALSE);
50
51 if (!is_declarator)
52 return -1;
53
569ac469 54 /* $declarators{$current_package_name} */
55
72f20f69 56 is_declarator_pack_ref = hv_fetch(is_declarator, HvNAME(PL_curstash),
57 strlen(HvNAME(PL_curstash)), FALSE);
58
59 if (!is_declarator_pack_ref || !SvROK(*is_declarator_pack_ref))
60 return -1; /* not a hashref */
61
62 is_declarator_pack_hash = (HV*) SvRV(*is_declarator_pack_ref);
63
569ac469 64 /* $declarators{$current_package_name}{$name} */
65
72f20f69 66 is_declarator_flag_ref = hv_fetch(
67 is_declarator_pack_hash, name,
68 strlen(name), FALSE
69 );
70
71 /* requires SvIOK as well as TRUE since flags not being an int is useless */
72
73 if (!is_declarator_flag_ref
74 || !SvIOK(*is_declarator_flag_ref)
75 || !SvTRUE(*is_declarator_flag_ref))
76 return -1;
77
78 dd_flags = SvIVX(*is_declarator_flag_ref);
79
80 return dd_flags;
81}
82
569ac469 83/* callback thingy */
84
85void dd_linestr_callback (pTHX_ char* type, char* name, char* s) {
86
87 char* linestr = SvPVX(PL_linestr);
88 int offset = s - linestr;
89
569ac469 90 dSP;
91
92 ENTER;
93 SAVETMPS;
94
95 PUSHMARK(SP);
96 XPUSHs(sv_2mortal(newSVpv(type, 0)));
97 XPUSHs(sv_2mortal(newSVpv(name, 0)));
98 XPUSHs(sv_2mortal(newSViv(offset)));
99 PUTBACK;
100
04a8a223 101 call_pv("Devel::Declare::linestr_callback", G_VOID|G_DISCARD);
569ac469 102
569ac469 103 FREETMPS;
104 LEAVE;
105}
106
107char* dd_get_linestr(pTHX) {
108 return SvPVX(PL_linestr);
109}
110
111void dd_set_linestr(pTHX_ char* new_value) {
112 int new_len = strlen(new_value);
113 char* old_linestr = SvPVX(PL_linestr);
114
115 SvGROW(PL_linestr, strlen(new_value));
116
117 if (SvPVX(PL_linestr) != old_linestr)
118 Perl_croak(aTHX_ "forced to realloc PL_linestr for line %s, bailing out before we crash harder", SvPVX(PL_linestr));
119
120 memcpy(SvPVX(PL_linestr), new_value, new_len+1);
121
122 SvCUR_set(PL_linestr, new_len);
123
124 PL_bufend = SvPVX(PL_linestr) + new_len;
125}
126
04a8a223 127char* dd_get_lex_stuff(pTHX) {
128 return SvPVX(PL_lex_stuff);
129}
130
131char* dd_clear_lex_stuff(pTHX) {
132 PL_lex_stuff = Nullsv;
133}
134
135char* dd_get_curstash_name(pTHX) {
136 return HvNAME(PL_curstash);
137}
138
923c07a8 139char* dd_move_past_token (pTHX_ char* s) {
140
141 /*
142 * buffer will be at the beginning of the declarator, -unless- the
143 * declarator is at EOL in which case it'll be the next useful line
144 * so we don't short-circuit out if we don't find the declarator
145 */
146
147 while (s < PL_bufend && isSPACE(*s)) s++;
148 if (memEQ(s, PL_tokenbuf, strlen(PL_tokenbuf)))
149 s += strlen(PL_tokenbuf);
150 return s;
151}
152
04a8a223 153int dd_toke_move_past_token (pTHX_ int offset) {
154 char* base_s = SvPVX(PL_linestr) + offset;
155 char* s = dd_move_past_token(aTHX_ base_s);
156 return s - base_s;
157}
158
923c07a8 159int dd_toke_scan_word(pTHX_ int offset, int handle_package) {
160 char tmpbuf[sizeof PL_tokenbuf];
161 char* base_s = SvPVX(PL_linestr) + offset;
162 STRLEN len;
163 char* s = scan_word(base_s, tmpbuf, sizeof tmpbuf, handle_package, &len);
164 return s - base_s;
165}
166
167int dd_toke_scan_str(pTHX_ int offset) {
168 char* base_s = SvPVX(PL_linestr) + offset;
169 char* s = scan_str(base_s, FALSE, FALSE);
170 return s - base_s;
171}
172
173int dd_toke_skipspace(pTHX_ int offset) {
174 char* base_s = SvPVX(PL_linestr) + offset;
175 char* s = skipspace(base_s);
176 return s - base_s;
177}
178
94caac6e 179/* replacement PL_check rv2cv entry */
180
bedac9ff 181STATIC OP *(*dd_old_ck_rv2cv)(pTHX_ OP *op);
182
94caac6e 183STATIC OP *dd_ck_rv2cv(pTHX_ OP *o) {
184 OP* kid;
185 char* s;
0ba8c7aa 186 char* save_s;
94caac6e 187 char tmpbuf[sizeof PL_tokenbuf];
0ba8c7aa 188 char found_name[sizeof PL_tokenbuf];
0f070758 189 char* found_proto = NULL, *found_traits = NULL;
0ba8c7aa 190 STRLEN len = 0;
0ba8c7aa 191 int dd_flags;
9026391e 192 char* cb_args[6];
53e3ab32 193 dSP; /* define stack pointer for later call stuff */
194 char* retstr;
195 STRLEN n_a; /* for POPpx */
94caac6e 196
197 o = dd_old_ck_rv2cv(aTHX_ o); /* let the original do its job */
198
199 if (in_declare) {
200 cb_args[0] = NULL;
dcca2c59 201#ifdef DD_DEBUG
202 printf("Deconstructing declare\n");
203 printf("PL_bufptr: %s\n", PL_bufptr);
204 printf("bufend at: %i\n", PL_bufend - PL_bufptr);
205 printf("linestr: %s\n", SvPVX(PL_linestr));
206 printf("linestr len: %i\n", PL_bufend - SvPVX(PL_linestr));
207#endif
94caac6e 208 call_argv("Devel::Declare::done_declare", G_VOID|G_DISCARD, cb_args);
0ba8c7aa 209 in_declare--;
dcca2c59 210#ifdef DD_DEBUG
211 printf("PL_bufptr: %s\n", PL_bufptr);
212 printf("bufend at: %i\n", PL_bufend - PL_bufptr);
213 printf("linestr: %s\n", SvPVX(PL_linestr));
214 printf("linestr len: %i\n", PL_bufend - SvPVX(PL_linestr));
215 printf("actual len: %i\n", strlen(PL_bufptr));
216#endif
94caac6e 217 return o;
218 }
219
220 kid = cUNOPo->op_first;
221
222 if (kid->op_type != OP_GV) /* not a GV so ignore */
223 return o;
224
225 if (PL_lex_state != LEX_NORMAL && PL_lex_state != LEX_INTERPNORMAL)
226 return o; /* not lexing? */
227
72f20f69 228 /* I was doing this, but the CONST wrap can't so it didn't gain anything
229 stash = GvSTASH(kGVOP_gv); */
94caac6e 230
c630715a 231#ifdef DD_DEBUG
72f20f69 232 printf("Checking GV %s -> %s\n", HvNAME(GvSTASH(kGVOP_gv)), GvNAME(kGVOP_gv));
c630715a 233#endif
94caac6e 234
72f20f69 235 dd_flags = dd_is_declarator(aTHX_ GvNAME(kGVOP_gv));
94caac6e 236
72f20f69 237 if (dd_flags == -1)
94caac6e 238 return o;
239
0ba8c7aa 240#ifdef DD_DEBUG
241 printf("dd_flags are: %i\n", dd_flags);
242#endif
243
94caac6e 244 s = PL_bufptr; /* copy the current buffer pointer */
245
c630715a 246 DD_DEBUG_S
247
248#ifdef DD_DEBUG
0ba8c7aa 249 printf("PL_tokenbuf: %s\n", PL_tokenbuf);
c630715a 250#endif
251
04a8a223 252 dd_linestr_callback(aTHX_ "rv2cv", GvNAME(kGVOP_gv), s);
94caac6e 253
53e3ab32 254 return o;
255}
256
bedac9ff 257STATIC OP *(*dd_old_ck_entereval)(pTHX_ OP *op);
258
b7505981 259OP* dd_pp_entereval(pTHX) {
260 dSP;
261 dPOPss;
262 STRLEN len;
263 const char* s;
264 if (SvPOK(sv)) {
53e3ab32 265#ifdef DD_DEBUG
b7505981 266 printf("mangling eval sv\n");
53e3ab32 267#endif
b7505981 268 if (SvREADONLY(sv))
269 sv = sv_2mortal(newSVsv(sv));
270 s = SvPVX(sv);
271 len = SvCUR(sv);
272 if (!len || s[len-1] != ';') {
273 if (!(SvFLAGS(sv) & SVs_TEMP))
274 sv = sv_2mortal(newSVsv(sv));
275 sv_catpvn(sv, "\n;", 2);
53e3ab32 276 }
b7505981 277 SvGROW(sv, 8192);
53e3ab32 278 }
b7505981 279 PUSHs(sv);
280 return PL_ppaddr[OP_ENTEREVAL](aTHX);
281}
53e3ab32 282
b7505981 283STATIC OP *dd_ck_entereval(pTHX_ OP *o) {
284 o = dd_old_ck_entereval(aTHX_ o); /* let the original do its job */
d8e65fc8 285 if (o->op_ppaddr == PL_ppaddr[OP_ENTEREVAL])
286 o->op_ppaddr = dd_pp_entereval;
94caac6e 287 return o;
288}
289
6a0bcf35 290static I32 dd_filter_realloc(pTHX_ int idx, SV *sv, int maxlen)
291{
292 const I32 count = FILTER_READ(idx+1, sv, maxlen);
293 SvGROW(sv, 8192); /* please try not to have a line longer than this :) */
294 /* filter_del(dd_filter_realloc); */
295 return count;
296}
297
bedac9ff 298STATIC OP *(*dd_old_ck_const)(pTHX_ OP*op);
299
300STATIC OP *dd_ck_const(pTHX_ OP *o) {
bedac9ff 301 int dd_flags;
302 char* s;
04a8a223 303 char* name;
bedac9ff 304
305 o = dd_old_ck_const(aTHX_ o); /* let the original do its job */
306
bedac9ff 307 /* don't try and look this up if it's not a string const */
308 if (!SvPOK(cSVOPo->op_sv))
309 return o;
310
04a8a223 311 name = SvPVX(cSVOPo->op_sv);
312
313 dd_flags = dd_is_declarator(aTHX_ name);
bedac9ff 314
a9fb5fb1 315 if (dd_flags == -1)
bedac9ff 316 return o;
317
bedac9ff 318 s = PL_bufptr;
319
04a8a223 320 dd_linestr_callback(aTHX_ "const", name, s);
bedac9ff 321
bedac9ff 322 return o;
323}
324
94caac6e 325static int initialized = 0;
326
327MODULE = Devel::Declare PACKAGE = Devel::Declare
328
329PROTOTYPES: DISABLE
330
331void
332setup()
333 CODE:
334 if (!initialized++) {
335 dd_old_ck_rv2cv = PL_check[OP_RV2CV];
336 PL_check[OP_RV2CV] = dd_ck_rv2cv;
b7505981 337 dd_old_ck_entereval = PL_check[OP_ENTEREVAL];
338 PL_check[OP_ENTEREVAL] = dd_ck_entereval;
bedac9ff 339 dd_old_ck_const = PL_check[OP_CONST];
340 PL_check[OP_CONST] = dd_ck_const;
94caac6e 341 }
6a0bcf35 342 filter_add(dd_filter_realloc, NULL);
569ac469 343
344char*
345get_linestr()
346 CODE:
347 RETVAL = dd_get_linestr(aTHX);
348 OUTPUT:
349 RETVAL
923c07a8 350
351void
352set_linestr(char* new_value)
353 CODE:
354 dd_set_linestr(aTHX_ new_value);
355
04a8a223 356char*
357get_lex_stuff()
358 CODE:
359 RETVAL = dd_get_lex_stuff(aTHX);
360 OUTPUT:
361 RETVAL
362
363void
364clear_lex_stuff()
365 CODE:
366 dd_clear_lex_stuff(aTHX);
367
368char*
369get_curstash_name()
370 CODE:
371 RETVAL = dd_get_curstash_name(aTHX);
372 OUTPUT:
373 RETVAL
374
923c07a8 375int
376toke_scan_word(int offset, int handle_package)
377 CODE:
378 RETVAL = dd_toke_scan_word(aTHX_ offset, handle_package);
379 OUTPUT:
380 RETVAL
381
382int
04a8a223 383toke_move_past_token(int offset);
384 CODE:
385 RETVAL = dd_toke_move_past_token(aTHX_ offset);
386 OUTPUT:
387 RETVAL
388
389int
923c07a8 390toke_scan_str(int offset);
391 CODE:
392 RETVAL = dd_toke_scan_str(aTHX_ offset);
393 OUTPUT:
394 RETVAL
395
396int
397toke_skipspace(int offset)
398 CODE:
399 RETVAL = dd_toke_skipspace(aTHX_ offset);
400 OUTPUT:
401 RETVAL
04a8a223 402
403int
404get_in_declare()
405 CODE:
406 RETVAL = in_declare;
407 OUTPUT:
408 RETVAL
409
410void
411set_in_declare(int value)
412 CODE:
413 in_declare = value;