#!/usr/bin/perl
#
# regen_perly.pl, DAPM 12-Feb-04
#
# Copyright (c) 2004, 2005, 2006, 2009, 2010, 2011 Larry Wall
#
# Given an input file perly.y, run bison on it and produce
# the following output files:
#
# perly.h       standard bison header file with minor doctoring of
#               #line directives plus adding a #ifdef PERL_CORE
#
# perly.tab     the parser table C definitions extracted from the bison output
#               plus an extra table generated by this script.
#
# perly.act     the action case statements extracted from the bison output
#
# Note that perly.c is *not* regenerated - this is now a static file which
# is not dependent on perly.y any more.
#
# If a filename of the form foo.y is given on the command line, then
# this is used instead as the basename for all the files mentioned
# above.
#
# Note that temporary files of the form perlytmp.h and perlytmp.c are
# created and then deleted during this process
#
# Note also that this script is intended to be run on a UNIX system;
# it may work elsewhere but no specific attempt has been made to make it
# portable.
use 5.006;
sub usage { die "usage: $0 [ -b bison_executable ] [ file.y ]\n" }
use warnings;
use strict;
our $Verbose;
BEGIN { require './regen/regen_lib.pl'; }
my $bison = 'bison';
if (@ARGV >= 2 and $ARGV[0] eq '-b') {
    shift;
    $bison = shift;
}
my $y_file = shift || 'perly.y';
usage unless @ARGV==0 && $y_file =~ /\.y$/;
(my $h_file    = $y_file) =~ s/\.y$/.h/;
(my $act_file  = $y_file) =~ s/\.y$/.act/;
(my $tab_file  = $y_file) =~ s/\.y$/.tab/;
(my $tmpc_file = $y_file) =~ s/\.y$/tmp.c/;
(my $tmph_file = $y_file) =~ s/\.y$/tmp.h/;
# the yytranslate[] table generated by bison is ASCII/EBCDIC sensitive
die "$0: must be run on an ASCII system\n" unless ord 'A' == 65;
# check for correct version number. The constraints are:
#  * must be >= 1.24 to avoid licensing issues.
#  * it must generate the yystos[] table. Version 1.28 doesn't generate
#    this; 1.35+ does
#  * Must produce output which is extractable by the regexes below
#  * Must produce the right values.
# These last two constraints  may well be met by earlier versions, but
# I simply haven't tested them yet. If it works for you, then modify
# the test below to allow that version too. DAPM Feb 04.
my $version = `$bison -V`;
unless ($version) { die <<EOF; }
Could not find a version of bison in your path. Please install bison.
EOF
# Don't change this to add new bison versions without testing that the generated
# files actually work :-) Win32 in particular may not like them. :-(
unless ($version =~ /\b(2\.[567]|3\.[0-7])\b/) { die <<EOF; }
You have the wrong version of bison in your path; currently versions
2.5-2.7 or 3.0-3.7 are known to work.  Try installing
    http://ftp.gnu.org/gnu/bison/bison-3.3.tar.gz
or similar.  Your bison identifies itself as:
$version
EOF
# bison's version number, not the entire string, is most useful later on.
$version = $1;
# creates $tmpc_file and $tmph_file
my_system("$bison -d -o $tmpc_file $y_file");
open my $ctmp_fh, '<', $tmpc_file or die "Can't open $tmpc_file: $!\n";
my $clines;
{ local $/; $clines = <$ctmp_fh>; }
die "failed to read $tmpc_file: length mismatch\n"
    unless length $clines == -s $tmpc_file;
close $ctmp_fh;
my ($actlines, $tablines) = extract($clines);
our %tokens;
$tablines .= make_type_tab($y_file, $tablines);
my ($act_fh, $tab_fh, $h_fh) = map {
    open_new($_, '>', { by => $0, from => $y_file });
} $act_file, $tab_file, $h_file;
print $act_fh $actlines;
print $tab_fh $tablines;
unlink $tmpc_file;
# Wrap PERL_CORE round the symbol definitions. Also,  the
# C<#line 30 "perly.y"> confuses the Win32 resource compiler and the
# C<#line 188 "perlytmp.h"> gets picked up by make depend, so remove them.
open my $tmph_fh, '<', $tmph_file or die "Can't open $tmph_file: $!\n";
# add integer-encoded #def of the bison version
{
    $version =~ /^(\d+)\.(\d+)/
        or die "Can't handle bison version format: '$version'";
    my ($v1,$v2) = ($1,$2);
    die "Unexpectedly large bison version '$v1'"    if $v1 > 99;
    die "Unexpectedly large bison subversion '$v2'" if $v2 > 9999;
    printf $h_fh "#define PERL_BISON_VERSION %2d%04d\n\n", $v1, $v2;
}
my $endcore_done = 0;
my $gather_tokens = 0;
my $tokens;
while (<$tmph_fh>) {
    # bison 2.6 adds header guards, which break things because of where we
    # insert #ifdef PERL_CORE, so strip them because they aren't important
    next if /YY_PERLYTMP_H/;
    print $h_fh "#ifdef PERL_CORE\n" if $. == 1;
    if (!$endcore_done and /YYSTYPE_IS_DECLARED/) {
        print $h_fh <<h;
#ifdef PERL_IN_TOKE_C
static bool
S_is_opval_token(int type) {
    switch (type) {
h
        print $h_fh <<i for sort grep $tokens{$_} eq 'opval', keys %tokens;
    case $_:
i
        print $h_fh <<j;
        return 1;
    }
    return 0;
}
#endif /* PERL_IN_TOKE_C */
#endif /* PERL_CORE */
j
        $endcore_done = 1;
    }
    next if /^#line \d+ ".*"/;
    if (!$gather_tokens) {
        $gather_tokens = 1 if /^\s* enum \s* yytokentype \s* \{/x;
    }
    else {
        if (/^\# \s* endif/x) { # The #endif just after the end of the token enum
            $gather_tokens = 0;
            $_ .= "\n/* Tokens.  */\n$tokens";
        }
        else {
            my ($tok, $val) = /(\w+) \s* = \s* (\d+)/x;
            $tokens .= "#define $tok $val\n" if $tok;
        }
    }
    print $h_fh $_;
}
close $tmph_fh;
unlink $tmph_file;
foreach ($act_fh, $tab_fh, $h_fh) {
    read_only_bottom_close_and_rename($_, ['regen_perly.pl', $y_file]);
}
exit 0;
# extract the symbol kinds, tables and actions from the generated .c file
sub extract {
    my $clines = shift;
    my $tablines;
    my $actlines;
    # extract the symbol kind table if it exists
    $clines =~ m@
        (?:
            ^/\* \s* Symbol \s+ kind\. \s* \*/\n
        )?
        enum \s+ yysymbol_kind_t \s* \{
        .*?
        \} \s* ;\n
        typedef \s+ enum \s+ \w+ \s+ \w+ ; \n+
    @xms
        and $tablines .= $&;
    my $last_table = $version >= 3 ? 'yyr2' : 'yystos';
    $clines =~ m@
        (?:
            ^/* YYFINAL[^\n]+\n         #optional comment
        )?
        \# \s* define \s* YYFINAL       # first #define
        .*?                             # other defines + most tables
        $last_table\[\]\s*=             # start of last table
        .*?
        }\s*;                           # end of last table
    @xms
        or die "Can't extract tables from $tmpc_file\n";
    $tablines .= $&;
    # extract all the cases in the big action switch statement
    $clines =~ m@
        switch \s* \( \s* yyn \s* \) \s* { \s*
            ( .*?  default: \s* break; \s* )
        }
    @xms
        or die "Can't extract actions from $tmpc_file\n";
    $actlines = $1;
    # Remove extraneous comments from bison 2.4
    $actlines =~ s!\s* /\* \s* Line \s* \d+ \s* of \s* yacc\.c \s* \*/!!gx;
    # Remove extraneous comments from bison 3.x
    $actlines =~ s!\s* /\* \s* yacc\.c : \d+ \s* \*/!!gx;
    # C<#line 188 "perlytmp.c"> gets picked up by make depend, so remove them.
    $actlines =~ s/^#line \d+ "\Q$tmpc_file\E".*$//gm;
    # convert yyvsp[nnn] into ps[nnn].val
    $actlines =~ s/yyvsp\[(.*?)\]/ps[$1].val/g
        or die "Can't convert value stack name\n";
    return $actlines. "\n", $tablines. "\n";
}
# Generate a table, yy_type_tab[], that specifies for each token, what
# type of value it holds.
#
# Read the .y file and extract a list of all the token names and
# non-terminal names; then scan the string $tablines for the table yytname,
# which gives the token index of each token/non-terminal; then use this to
# create yy_type_tab.
#
# ie given (in perly.y),
#
#   %token <opval> A
#   %token <ival>  B
#   %type  <pval>  C
#   %type  <opval> D
#
# and (in $tablines),
#
#   yytname[] = { "A" "B", "C", "D", "E" };
#
# then return
#
#    typedef enum { toketype_ival, toketype_opval, toketype_pval } toketypes;
#
#    static const toketypes yy_type_tab[]
#          = { toketype_opval, toketype_ival, toketype_pval,
#                toketype_opval, toketype_ival }
#
# where "E" has the default type. The default type is determined
# by the __DEFAULT__ comment  next to the appropriate union member in
# perly.y
sub make_type_tab {
    my ($y_file, $tablines) = @_;
    my %just_tokens;
    my %tokens;
    my %types;
    my $default_token;
    open my $fh, '<', $y_file or die "Can't open $y_file: $!\n";
    while (<$fh>) {
        if (/(\$\d+)\s*=[^=]/) {
            warn "$y_file:$.: dangerous assignment to $1: $_";
        }
        if (/__DEFAULT__/) {
            m{(\w+) \s* ; \s* /\* \s* __DEFAULT__}x
                or die "$y_file: can't parse __DEFAULT__ line: $_";
            die "$y_file: duplicate __DEFAULT__ line: $_"
                    if defined $default_token;
            $default_token = $1;
            next;
        }
        next unless /^%(token|type)/;
        s/^%((token)|type)\s+<(\w+)>\s+//
            or die "$y_file: unparseable token/type line: $_";
        for (split ' ', $_) {
            $tokens{$_} = $3;
            if ($2) {
                $just_tokens{$_} = $3;
            }
        }
        $types{$3} = 1;
    }
    *tokens = \%just_tokens; # perly.h needs this
    die "$y_file: no __DEFAULT__ token defined\n" unless $default_token;
    $types{$default_token} = 1;
    $tablines =~ /^\Qstatic const char *const yytname[] =\E\n
            \{\n
            (.*?)
            ^};
            /xsm
        or die "Can't extract yytname[] from table string\n";
    my $fields = $1;
    $fields =~ s{"((?:[^"\\]|\\.)+)"}
                { "toketype_" .
                    (defined $tokens{$1} ? $tokens{$1} : $default_token)
                }ge;
    $fields =~ s/, \s* (?:0|YY_NULL|YY_NULLPTR) \s* $//x
        or die "make_type_tab: couldn't delete trailing ',0'\n";
    return 
          "\ntypedef enum {\n\t"
        . join(", ", map "toketype_$_", sort keys %types)
        . "\n} toketypes;\n\n"
        . "/* type of each token/terminal */\n"
        . "static const toketypes yy_type_tab[] =\n{\n"
        . $fields
        . "\n};\n";
}
sub my_system {
    if ($Verbose) {
        print "executing: @_\n";
    }
    system(@_);
    if ($? == -1) {
        die "failed to execute command '@_': $!\n";
    }
    elsif ($? & 127) {
        die sprintf "command '@_' died with signal %d\n",
            ($? & 127);
    }
    elsif ($? >> 8) {
        die sprintf "command '@_' exited with value %d\n", $? >> 8;
    }
}