first cut of XML::Tags
[catagits/Web-Simple.git] / lib / XML / Tags.pm
1 package XML::Tags;
2
3 use strict;
4 use warnings FATAL => 'all';
5
6 my $IN_SCOPE = 0;
7
8 sub import {
9   die "Can't import XML::Tags into a scope when already compiling one that uses it"
10     if $IN_SCOPE;
11   my ($class, @args) = @_;
12   my $opts = shift(@args) if ref($args[0]) eq 'HASH';
13   my $target = $class->_find_target(0, $opts);
14   my @tags = $class->_find_tags(@args);
15   my $unex = $class->_export_tags_into($target => @tags);
16   $class->_install_unexporter($unex);
17   $IN_SCOPE = 1;
18 }
19
20 sub _find_tags { shift; @_ }
21
22 sub _find_target {
23   my ($class, $extra_levels, $opts) = @_;
24   return $opts->{into} if defined($opts->{into});
25   my $level = ($opts->{into_level} || 1) + $extra_levels;
26   return (caller($level))[0];
27 }
28
29 {
30   my $setup;
31
32   sub _setup_glob_override {
33     return if $setup;
34     $setup = 1;
35     no warnings 'redefine';
36     *CORE::GLOBAL::glob = sub {
37       for ($_[0]) {
38         # unless it smells like </foo> or <foo bar="baz">
39         return CORE::glob($_[0]) unless (/^\/\w+$/ || /^\w+\s+\w+="/);
40       }
41       return '<'.$_[0].'>';
42     };
43   }
44 }
45
46 sub _export_tags_into {
47   my ($class, $into, @tags) = @_;
48   foreach my $tag (@tags) {
49     no strict 'refs';
50     tie *{"${into}::${tag}"}, 'XML::Tags::TIEHANDLE', "<${tag}>";
51   }
52   my $orig = \&CORE::GLOBAL::glob || sub { CORE::glob($_[0]) };
53   {
54     no warnings 'redefine';
55     *CORE::GLOBAL::glob = sub { '<'.$_[0].'>' };
56   }
57   return sub {
58     foreach my $tag (@tags) {
59       no strict 'refs';
60       delete ${"${into}::"}{$tag}
61     }
62     $IN_SCOPE = 0;
63   };
64 }
65
66 sub _install_unexporter {
67   my ($class, $unex) = @_;
68   $^H |= 0x120000; # localize %^H
69   $^H{'XML::Tags::Unex'} = bless($unex, 'XML::Tags::Unex');
70 }
71
72 package XML::Tags::TIEHANDLE;
73
74 sub TIEHANDLE { my $str = $_[1]; bless \$str, $_[0] }
75 sub READLINE { ${$_[0]} }
76
77 package XML::Tags::Unex;
78
79 sub DESTROY { local $@; eval { $_[0]->(); 1 } || warn "ARGH: $@" }
80
81 1;