quote all extra libs in FatNode, in case they contain spaces
[scpubgit/Object-Remote.git] / lib / Object / Remote / FatNode.pm
1 package Object::Remote::FatNode;
2
3 use strictures 1;
4 use Config;
5 use B qw(perlstring);
6
7 my @exclude_mods = qw(XSLoader.pm DynaLoader.pm);
8 #used by t/watchdog_fatnode
9 our $INHIBIT_RUN_NODE = 0;
10
11 sub stripspace {
12   my ($text) = @_;
13   $text =~ /^(\s+)/ && $text =~ s/^$1//mg;
14   $text;
15 }
16
17 my %maybe_libs = map +($_ => 1), grep defined, (values %Config, '.');
18
19 my @extra_libs = grep not(ref($_) or $maybe_libs{$_}), @INC;
20
21 my $extra_libs = join '', map qq[  -I"$_"\n], @extra_libs;
22 my $perl_quote = $^O eq "MSWin32" ? q["] : q['];
23
24 my $command = qq(
25   $^X
26   $extra_libs
27   -mObject::Remote
28   -mObject::Remote::Connector::STDIO
29   -mFuture
30   -mMRO::Compat
31   -mClass::C3
32   -mClass::C3::next
33   -mAlgorithm::C3
34   -mObject::Remote::ModuleLoader
35   -mObject::Remote::Node
36   -mMethod::Generate::BuildAll
37   -mMethod::Generate::DemolishAll
38   -mJSON::PP
39   -e $perl_quote print join qq[\\n], \%INC $perl_quote
40 );
41
42 $command =~ s/\n/ /g;
43
44 chomp(my @inc = qx($command));
45
46 my %exclude = map { $_ => 1 } @exclude_mods;
47 my %mods = reverse @inc;
48 my %file_names = @inc;
49
50 foreach(keys(%mods)) {
51   if ($exclude{ $mods{$_} }) {
52     delete($mods{$_});
53   }
54 }
55
56 my @non_core_non_arch = ( $file_names{'Devel/GlobalDestruction.pm'} );
57 push @non_core_non_arch, grep +(
58   not (
59     #some of the config variables can be empty which will eval as a matching regex
60     $Config{privlibexp} ne '' && /^\Q$Config{privlibexp}/
61       or $Config{archlibexp} ne '' && /^\Q$Config{archlibexp}/
62       or $Config{vendorarchexp} ne '' && /^\Q$Config{vendorarchexp}/
63       or $Config{sitearchexp} ne '' && /^\Q$Config{sitearchexp}/
64   )
65 ), grep !/\Q$Config{archname}/, grep !/\Q$Config{myarchname}/, keys %mods;
66
67 my @core_non_arch = grep +(
68   $Config{privlibexp} ne '' && /^\Q$Config{privlibexp}/
69   and not($Config{archlibexp} ne '' && /^\Q$Config{archlibexp}/
70     or /\Q$Config{archname}/ or /\Q$Config{myarchname}/)
71 ), keys %mods;
72
73 my $start = stripspace <<'END_START';
74   # This chunk of stuff was generated by Object::Remote::FatNode. To find
75   # the original file's code, look for the end of this BEGIN block or the
76   # string 'FATPACK'
77   BEGIN {
78   my (%fatpacked,%fatpacked_extra);
79 END_START
80
81 $start .= 'my %exclude = map { $_ => 1 } (\'' . join("','", @exclude_mods) . "');\n";
82
83 my $end = stripspace <<'END_END';
84   s/^  //mg for values %fatpacked, values %fatpacked_extra;
85
86   sub load_from_hash {
87     if (my $fat = $_[0]->{$_[1]}) {
88       if ($exclude{$_[1]}) {
89         warn "Will not pre-load '$_[1]'";
90         return undef;
91       }
92
93       #warn "Handling $_[1]";
94       open my $fh, '<', \$fat;
95       return $fh;
96     }
97
98     #Uncomment this to find brokenness
99     #warn "Missing $_[1]";
100     return;
101   }
102
103   unshift @INC, sub { load_from_hash(\%fatpacked, $_[1]) };
104   push @INC, sub { load_from_hash(\%fatpacked_extra, $_[1]) };
105
106   } # END OF FATPACK CODE
107
108   use strictures 1;
109   use Object::Remote::Node;
110
111   unless ($Object::Remote::FatNode::INHIBIT_RUN_NODE) {
112     Object::Remote::Node->run(watchdog_timeout => $WATCHDOG_TIMEOUT);
113   }
114
115 END_END
116
117 my %files = map +($mods{$_} => scalar do { local (@ARGV, $/) = ($_); <> }),
118               @non_core_non_arch, @core_non_arch;
119
120 sub generate_fatpack_hash {
121   my ($hash_name, $orig) = @_;
122   (my $stub = $orig) =~ s/\.pm$//;
123   my $name = uc join '_', split '/', $stub;
124   my $data = $files{$orig} or die $orig; $data =~ s/^/  /mg;
125   $data .= "\n" unless $data =~ m/\n$/;
126   my $ret = '$'.$hash_name.'{'.perlstring($orig).qq!} = <<'${name}';\n!
127     .qq!${data}${name}\n!;
128 #  warn $ret;
129   return $ret;
130 }
131
132 my @segments = (
133   map(generate_fatpack_hash('fatpacked', $_), sort map $mods{$_}, @non_core_non_arch),
134   map(generate_fatpack_hash('fatpacked_extra', $_), sort map $mods{$_}, @core_non_arch),
135 );
136
137 #print STDERR Dumper(\@segments);
138 our $DATA = join "\n", $start, @segments, $end;
139
140 1;