Permalink
Browse files

Fix hang on heredocuments with flush left terminators and indented co…

…ntents
  • Loading branch information...
sorear committed Feb 25, 2011
1 parent c7b4852 commit 857b7a1b89f835e7675f7ee63cb21ff1ff05ca90
Showing with 74 additions and 0 deletions.
  1. +74 −0 src/niecza
View
@@ -34,6 +34,80 @@ augment class RxOp::Sym { #OK exist
}
augment class NieczaActions {
+sub mkstringycat($/, *@strings) {
+ my @a;
+ for @strings -> $s {
+ my $i = ($s !~~ Op) ?? ::Op::StringLiteral.new(|node($/),
+ text => $s) !! $s;
+
+ # this *might* belong in an optimization pass
+ if @a && @a[*-1] ~~ ::Op::StringLiteral &&
+ $i ~~ ::Op::StringLiteral {
+ @a[*-1] = ::Op::StringLiteral.new(|node($/),
+ text => (@a[*-1].text ~ $i.text));
+ } else {
+ push @a, $i;
+ }
+ }
+ if @a == 0 {
+ return ::Op::StringLiteral.new(|node($/), text => "");
+ } elsif @a == 1 {
+ return (@a[0] ~~ ::Op::StringLiteral) ?? @a[0] !!
+ mkcall($/, '&prefix:<~>', @a[0]);
+ } else {
+ return mkcall($/, '&infix:<~>', @a);
+ }
+}
+method process_nibble($/, @bits, $prefix?) {
+ my @acc;
+ for @bits -> $n {
+ my $ast = $n.ast;
+
+ if $ast ~~ CClass {
+ $n.CURSOR.sorry("Cannot use a character class in a string");
+ $ast = "";
+ }
+
+ if $ast !~~ Op && defined($prefix) && $prefix ne "" {
+ $ast = $ast.split(/^^<before \h>[ $prefix || \h+ ]/).join("");
+ }
+
+ push @acc, $ast;
+ }
+
+ my $post = $/.CURSOR.postprocessor;
+ make mkstringycat($/, @acc);
+
+ if $post eq 'null' {
+ # already OK
+ }
+ # actually quotewords is a bit trickier than this...
+ elsif $post eq 'words' || $post eq 'quotewords' {
+ my $sl = $/.ast;
+ if !$sl.^isa(::Op::StringLiteral) {
+ make ::Op::CallMethod.new(|node($/), :name<words>, receiver => $sl);
+ }
+ else {
+ my @tok = $sl.text.words;
+ @tok = map { ::Op::StringLiteral.new(|node($/), text => $_) }, @tok;
+
+ make ((@tok == 1) ?? @tok[0] !!
+ ::Op::SimpleParcel.new(|node($/), items => @tok));
+ }
+ }
+ elsif $post eq 'path' {
+ # TODO could stand to be a lot fancier.
+ make ::Op::CallMethod(|node($/), receiver => $/.ast, :name<IO>);
+ }
+ elsif $post eq 'run' {
+ make mkcall($/, 'rungather', $/.ast);
+ }
+ else {
+ $/.CURSOR.sorry("Unhandled postprocessor $post");
+ }
+
+ $/.ast;
+}
method trait_mod:is ($/) {
my $trait = ~$<longname>;
my $noparm;

0 comments on commit 857b7a1

Please sign in to comment.