package Publican::XmlClean;

use strict;
use warnings;
use 5.008;
use Carp;
use XML::TreeBuilder 5.1;
use Config::Simple '-strict';
use Publican;
use File::Path;
use Term::ANSIColor qw(:constants);
use Publican::Builder;
use File::Inplace;
use DBI;
use Data::Dumper;
use File::Basename;

=head1 NAME

Publican::XmlClean - A module to reformat XML to Publican standards

=head1 SYNOPSIS

    use Publican::XmlClean;

    my $cleaner = Publican::XmlClean->new( { clean_id => 1 } );

    foreach my $xml_file ( sort(@xml_files) ) {
        $cleaner->process_file( { file => $xml_file, out_file => $xml_file } );
    }

  
=head1 DESCRIPTION

Publican::XmlClean tidies XML formatting and filters structure based on input rules.

=head1 INTERFACE 

=cut

my %UPDATED_IDS;
my %MAX_CONFORMANCE;

my %MAP_OUT = (
    section    => { block => 1, },
    refsection => { block => 1, no_id => 1 },
    chapter       => { block         => 1 },
    preface       => { block         => 1 },
    bibliography  => { block         => 1 },
    bibliodiv     => { block         => 1 },
    biblioentry   => { block         => 1 },
    othercredit   => { block         => 1 },
    legalnotice   => { block         => 1 },
    address       => { block         => 1, verbatim => 1 },
    book          => { block         => 1 },
    article       => { block         => 1 },
    part          => { block         => 1 },
    partintro     => { block         => 1 },
    para          => { block         => 1 },
    formalpara    => { block         => 1 },
    simpara       => { block         => 1 },
    itemizedlist  => { block         => 1 },
    orderedlist   => { block         => 1 },
    variablelist  => { block         => 1 },
    seglistitem   => { block         => 1 },
    segtitle      => { newline_after => 1 },
    seg           => { newline_after => 1 },
    segmentedlist => { block         => 1 },
    simplelist    => { block         => 1 },
    qandaset      => { block         => 1 },
    qandaentry    => { block         => 1 },
    question      => { block         => 1 },
    answer        => { block         => 1 },
    member        => { newline_after => 1 },
    remark        => { newline_after => 1 },
    userinput => {},
    listitem  => { block => 1, keep_id => 1 },
    title         => { newline_after => 1 },
    refentrytitle => { newline_after => 1 },
    refpurpose    => { newline_after => 1 },
    refname       => { newline_after => 1 },
    refnamediv    => { block         => 1, id_node => 'refname' },
    manvolnum     => { newline_after => 1 },
    street           => {},
    city             => {},
    state            => {},
    postcode         => {},
    country          => {},
    phone            => {},
    fax              => {},
    pob              => {},
    subtitle         => { newline_after => 1 },
    screen           => { block => 1, verbatim => 1 },
    programlisting   => { block => 1, verbatim => 1 },
    programlistingco => { block => 1, newline_after => 1 },
    xref        => { force_empty => 1 },
    footnoteref => { force_empty => 1 },
    important   => { block       => 1, no_id => 1 },
    note        => { block       => 1, no_id => 1 },
    warning     => { block       => 1, no_id => 1 },
    figure      => { block       => 1 },
    mediaobject => { block       => 1 },
    imageobject => { block       => 1 },
    imagedata     => {},
    'xi:include'  => { newline_after => 1 },
    'xi:fallback' => { newline_after => 1 },
    glossary          => { block         => 1 },
    glossentry        => { block         => 1, id_node => 'glossterm' },
    glossdiv          => { block         => 1 },
    glossdef          => { block         => 1 },
    glossterm         => { newline_after => 1 },
    glosssee          => { newline_after => 1 },
    glossseealso      => { newline_after => 1 },
    table             => { block         => 1 },
    informaltable     => { block         => 1 },
    thead             => { block         => 1 },
    tgroup            => { block         => 1 },
    tbody             => { block         => 1 },
    tr                => { block         => 1 },
    td                => { block         => 1 },
    row               => { block         => 1 },
    entry             => { newline_after => 1 },
    refentry          => { block         => 1 },
    refmeta           => { block         => 1 },
    refentryinfo      => { block         => 1, no_id => 1 },
    reference         => { block         => 1 },
    indexterm         => { block         => 1, mixed_mode => 1 },
    primary           => { newline_after => 1, mixed_mode => 1 },
    secondary         => { newline_after => 1, mixed_mode => 1 },
    tertiary          => { newline_after => 1, mixed_mode => 1 },
    bookinfo          => { block         => 1 },
    articleinfo       => { block         => 1 },
    abstract          => { block         => 1 },
    inlinemediaobject => { block         => 1 },
    publisher         => { block         => 1 },
    copyright         => { block         => 1 },
    authorgroup       => { block         => 1 },
    author            => { block         => 1 },
    editor            => { block         => 1 },
    corpauthor        => { block         => 1 },
    revision          => { block         => 1 },
    revhistory        => { block         => 1 },
    revdescription    => { block         => 1 },
    publishername     => { block         => 1 },
    affiliation       => { block         => 1 },
    year              => { newline_after => 1 },
    holder            => { newline_after => 1 },
    revnumber         => { newline_after => 1 },
    date              => { newline_after => 1 },
    honorific         => { newline_after => 1 },
    firstname         => { newline_after => 1 },
    surname           => { newline_after => 1 },
    email             => { newline_after => 1, mixed_mode => 1 },
    isbn              => { newline_after => 1 },
    issuenum          => { newline_after => 1 },
    edition           => { newline_after => 1 },
    pubdate           => { newline_after => 1 },
    productname       => { mixed_mode    => 1, newline_after => 1 },
    productnumber     => { newline_after => 1 },
    pubsnumber        => { newline_after => 1 },
    contrib           => { newline_after => 1 },
    shortaffil        => { newline_after => 1 },
    jobtitle          => { newline_after => 1 },
    orgname           => { newline_after => 1 },
    orgdiv            => { newline_after => 1 },
    citetitle      => {},
    trademark      => {},
    ulink          => {},
    firstterm      => {},
    menuchoice     => {},
    acronym        => {},
    abbrev         => {},
    command        => {},
    filename       => {},
    'index'        => {},
    application    => {},
    'package'      => {},
    guimenu        => {},
    sgmltag        => {},
    guilabel       => {},
    guibutton      => {},
    emphasis       => {},
    phrase         => {},
    replaceable    => {},
    computeroutput => {},
    guimenuitem    => {},
    textobject     => { block => 1 },
    varlistentry   => { block => 1, id_node => 'term' },
    term         => { newline_after => 1 },
    colspec      => { newline_after => 1 },
    areaspec     => { block         => 1 },
    areaset      => { block         => 1, keep_id => 1 },
    area         => { newline_after => 1, keep_id => 1 },
    calloutlist  => { block         => 1 },
    callout      => { block         => 1 },
    procedure    => { block         => 1 },
    appendix     => { block         => 1 },
    appendixinfo => { block         => 1 },
    cmdsynopsis  => { block         => 1 },
    arg          => { block         => 1 },
    group        => { block         => 1 },
    accel         => {},
    blockquote    => { block => 1 },
    classname     => {},
    code          => {},
    colophon      => { block => 1 },
    envar         => {},
    example       => { block => 1 },
    footnote      => { keep_id => 1 },
    guisubmenu    => {},
    interface     => {},
    keycap        => {},
    keycombo      => {},
    literal       => {},
    literallayout => { block => 1, verbatim => 1 },
    option        => {},
    parameter     => {},
    prompt        => {},
    property      => {},
    see              => { newline_after => 1, },
    seealso          => { newline_after => 1, },
    step             => { block         => 1, keep_id => 1 },
    substeps         => { block         => 1 },
    stepalternatives => { block         => 1 },
    systemitem    => {},
    wordasword    => {},
    citerefentry  => {},
    manvolnum     => {},
    function      => {},
    uri           => {},
    mousebutton   => {},
    hardware      => {},
    type          => {},
    methodname    => {},
    exceptionname => {},
    varname       => {},
    interfacename => {},
    othername     => { newline_after => 1 },
    '~comment'    => {},
    foreignphrase => {},
    chapterinfo   => { block => 1 },
    info          => { block => 1 },
    keywordset    => { block => 1 },
    keyword       => { newline_after => 1 },
    subjectset    => { block => 1 },
    subject       => { block => 1 },
    subjectterm   => { newline_after => 1 },
    superscript   => {},
    '~cdata'      => { verbatim => 1 },
);

=head2 new

Create a new Publican::XmlClean object.

=cut

sub new {
    my ( $this, $args ) = @_;
    my $class = ref($this) || $this;

    my $config = new Config::Simple();
    $config->syntax('simple');

    $config->param( 'lang', delete( $args->{lang} ) ) if ( $args->{lang} );
    $config->param( 'update_includes', delete( $args->{update_includes} ) )
        if ( $args->{update_includes} );
    $config->param( 'clean_id', ( delete( $args->{clean_id} ) ) || 0 );
    $config->param( 'unique_id', ( delete( $args->{unique_id} ) ) || 0 );
    $config->param( 'donotset_lang',
        ( delete( $args->{donotset_lang} ) ) || 0 );
    $config->param( 'distributed_set',
        ( delete( $args->{distributed_set} ) ) || 0 );
    $config->param( 'exclude_ent', ( delete( $args->{exclude_ent} ) ) || 0 );
    $config->param( 'cleaning',
        ( delete( $args->{cleaning} ) ) || $config->param('clean_id') );
    $config->param( 'id_attr', ( delete( $args->{id_attr} ) ) || 'id' );

    if ( %{$args} ) {
        croak(
            maketext(
                "unknown arguments: [_1]", join( ", ", keys %{$args} )
            )
        );
    }

    my $self = bless {}, $class;
    $self->{config} = $config;

    my $publican = Publican->new();
    $self->{publican} = $publican;

    return $self;
}

=head2  print_known_tags

Print a list of tags that have had their output QA'd.

=cut

sub print_known_tags {
    my $self = shift();
    foreach my $key ( sort( keys(%MAP_OUT) ) ) {
        print("$key\n");
    }

    return;
}

=head2 prune_xml($node)

Remove unwanted nodes. i.e. 'profile' in DocBook speak.

=cut

sub prune_xml {
    my ( $self, $xml_doc ) = @_;

    my $original_tag = $xml_doc->root()->{'_tag'};
    my @prune_attrs
        = qw/arch audience condition conformance lang os revision revisionflag role security status userlevel vendor wordsize/;

    if ($xml_doc) {

        foreach my $attr (@prune_attrs) {
            my $cond = $self->{publican}->param($attr);
            $cond = $self->{config}->param($attr)
                if ( $attr eq 'lang' );    ## Always prune on language
            if ($cond) {
                $xml_doc->pos( $xml_doc->root() );
                $cond =~ s/;/|/g;
                my $regex = qr/(?:$cond)/;
                while (
                    my $node = $xml_doc->look_down(
                        sub {
                                   $_[0]->attr($attr)
                                && $_[0]->attr($attr) !~ /^$regex$/
                                && $_[0]->attr($attr) !~ /^$regex;/
                                && $_[0]->attr($attr) !~ /;$regex$/
                                && $_[0]->attr($attr) !~ /;$regex;/;
                        }
                    )
                    )
                {
                    croak(
                        maketext(
                            "FATAL ERROR: profiling would prune root node. Do NOT set attributes to prune on in a root node. Offending attibute: [_1]",
                            $attr
                        )
                    ) if ( $node->same_as( $xml_doc->root() ) );
                    $node->delete();
                }
            }
        }
    }

    return;
}

=head2 Clean_ID

Rename ID's and update xrefs.

If this node has a title as a child set it's ID else remove the ID

=cut

sub Clean_ID {
    my ( $self, $node ) = @_;
    my $my_id   = "";
    my $docname = $self->{publican}->param('docname');
    my $product = $self->{publican}->param('product');

    if ($node) {
        my $tag = $node->{'_tag'};

        # keep_id means keep the current ID without modification.
        if ( $MAP_OUT{$tag}->{keep_id} ) {
            $my_id = $node->attr( $self->{config}->param('id_attr') ) || "";
        }
        elsif ( !$MAP_OUT{$tag}->{no_id} ) {
## TODO this doesn;t work when teh title is in an info. <info><title>...
            foreach my $child ( $node->content_refs_list() ) {
                if ( ref $$child
                    && $$child->{'_tag'} eq
                    ( $MAP_OUT{$tag}->{id_node} || 'title' ) )
                {
                    $my_id = $$child->as_text;
                    $my_id =~ s/[- ]/_/g;
                    $my_id =~ s/[^a-zA-Z0-9\._]//g;
                    $my_id =~ s/_+/_/g;

                    # Must start with a letter!
                    if ( $my_id =~ /^\d/ ) {
                        $my_id = 'a' . $my_id;
                    }

                    if ( $node->parent() ) {
                        my $par_title = $node->parent()
                            ->look_up( sub { $_[0]->attr('id'); } );
                        if ( $my_id ne "" && $par_title ) {
                            my $my_p_id = $par_title->attr('id');

                            $my_p_id =~ s/^.*-//;
                            $my_id = "$my_p_id-$my_id";
                        }
                    }
                    last;
                }
            }
        }

        # prepend product & book name (to avoid problems in sets)
        # prepend tag type for translations BZ #427312
        # If $product is unset then we are dealing with common content, so don't do the above.
        if ( $my_id ne "" && $product ) {
            $my_id = "$product-$docname-$my_id";
            $my_id = substr( $tag, 0, 4 ) . "-$my_id";
        }

        $my_id = $node->attr( $self->{config}->param('id_attr') ) if ( !$my_id  || ($my_id eq "" ));


        if (   $node->attr( $self->{config}->param('id_attr') )
            && $node->attr( $self->{config}->param('id_attr') ) ne $my_id )
        {
            $UPDATED_IDS{ $node->attr( $self->{config}->param('id_attr') ) }
                = $my_id;
        }

        if ( !$my_id || ( $my_id eq "") ) {
            $my_id = undef;
        }

        $node->attr( $self->{config}->param('id_attr'), $my_id );
    }

    return;
}

=head2 print_xml

Print out utf8 XML files

Have to output xml/DTD header

=cut

sub print_xml {
    my ( $self, $args ) = @_;

    my $xml_doc = delete( $args->{xml_doc} )
        || croak( maketext("xml_doc is a mandatory argument") );
    my $out_file = delete( $args->{out_file} )
        || croak( maketext("out_file is a mandatory argument") );

    if ( %{$args} ) {
        croak(
            maketext(
                "unknown arguments: [_1]", join( ", ", keys %{$args} )
            )
        );
    }

    my $dtdver    = $self->{publican}->param('dtdver');
    my $docname   = $self->{publican}->param('docname');
    my $lang      = $self->{config}->param('lang');
    my $main_file = $self->{publican}->param('mainfile');
    my $clean_id  = $self->{config}->param('clean_id');
    my $cleaning  = $self->{config}->param('cleaning');

    if ($xml_doc) {
        my $file = $out_file;
        my $path = '';

        # handle nested directories
        if ( $file =~ m|^(.*/xml)/(.*\/)[^\/]*\.xml| ) {
            $path = $2;
            mkpath("$1/$path") if ( !-d $path );
            $path =~ s|[^/]*/|\.\./|g;
        }

        $xml_doc->pos( $xml_doc->root() );
        if (   !$self->{config}->param('clean_id')
            && !$self->{config}->param('update_includes')
            && !$self->{config}->param('donotset_lang') )
        {
            if ( $dtdver !~ m/^5/ ) {
                $xml_doc->attr( 'lang', $lang );
            }
        }

        if ( $dtdver =~ m/^5/ ) {
            $xml_doc->attr( 'xmlns',       'http://docbook.org/ns/docbook' );
            $xml_doc->attr( 'xmlns:xlink', 'http://www.w3.org/1999/xlink' );
            $xml_doc->attr( 'version',     $dtdver );
            $xml_doc->attr( 'xml:lang',    $lang );
        }

        my $doctype;
        foreach my $node ( $xml_doc->look_down( '_tag', '~declaration' ) ) {
            $node->detach();
            unless ($docname) {

                if ( $node->attr('type') && $node->attr('type') eq 'DOCTYPE' )
                {
                    $doctype = q|<!DOCTYPE | . $node->attr('mytag');
                    $doctype .= q| PUBLIC "| . $node->attr('pid') . q|"|
                        if ( $node->attr('pid')
                        && $node->attr('pid') ne 'pid' );
                    $doctype .= q| "| . $node->attr('uri') . q|"|
                        if ( $node->attr('uri')
                        && $node->attr('uri') ne 'uri' );
                    $doctype .= qq| [\n|;
                }
                else {
                    $doctype
                        .= q|<!ENTITY |
                        . $node->attr('name') . q| "|
                        . $node->attr('value')
                        . qq|">\n|;
                }
            }

##             print(STDERR Dumper($node));
        }

        $doctype .= qq|]>\n| if ($doctype);

        my $type = $xml_doc->attr("_tag");
        $file =~ m|^(.*/xml/)|;
        my $text = $self->my_as_XML(
            { xml_doc => $xml_doc, path => ( $1 || './' ) } );
## BUGBUG revert to upstream as_XML?
##        my $text = $xml_doc->as_XML();
        #        $text =~ s/&#34;/"/g;
        #        $text =~ s/&#39;/'/g;
        #        $text =~ s/&quot;/"/g;
        #        $text =~ s/&apos;/'/g;

        $xml_doc->root()->delete();

        my $OUTDOC;
        open( $OUTDOC, ">:encoding(UTF-8)", "$out_file" )
            || croak(
            maketext( "Could not open [_1] for output!", $out_file ) );

        my $ent_file = undef;

        # Will be unset when processing common files outside books
        if ( $docname && !$self->{config}->param('exclude_ent') ) {
            my $xml_lang = $self->{publican}->param('xml_lang');
            if ( -e "$xml_lang/$main_file.ent" ) {
                $ent_file = "$main_file.ent";
                $ent_file = "$path$main_file.ent" if ($path);
            }
        }

        if ($doctype) {
            print( $OUTDOC $doctype );
        }
        else {
            print( $OUTDOC Publican::dtd_string(
                    {   tag      => $type,
                        dtdver   => $dtdver,
                        ent_file => $ent_file,
                        cleaning => $cleaning,
                    }
                )
            );
        }

        #debug_msg("is utf8: " . utf8::is_utf8($text) . "\n");
        print( $OUTDOC $text );
        close($OUTDOC);
    }

    return;
}

=head2 my_as_XML

Traverse tree and output xml as text. Overrides traverse ... evil stuff.

=cut

sub my_as_XML {
    my ( $self, $args ) = @_;

    my $xml_doc = delete( $args->{xml_doc} )
        || croak( maketext("'xml_doc' is a mandatory argument") );
    my $path = delete( $args->{path} )
        || croak( maketext("'path' is a mandatory argument") );

    # based on as_HTML
    my $tree              = $xml_doc->root();
    my @xml               = ();
    my $empty_element_map = $tree->_empty_element_map;

    my %banned_tags = ();
    foreach my $btag (
        split( /,/, ( $self->{publican}->param('banned_tags') || "" ) ) )
    {
        $banned_tags{$btag} = 1;
    }

    my %banned_attrs = ();
    foreach my $battr (
        split( /,/, ( $self->{publican}->param('banned_attrs') || "" ) ) )
    {
        $banned_attrs{$battr} = 1;
    }

    my $clean_id = $self->{config}->param('clean_id');
    my $unique_id  = $self->{config}->param('unique_id');
    my $lang     = $self->{config}->param('lang');

    # This flags tags that use  /> instead of end tags IF they are empty.
    $empty_element_map->{xref}         = 1;
    $empty_element_map->{footnoteref}  = 1;
    $empty_element_map->{'index'}      = 1;
    $empty_element_map->{'xi:include'} = 1;
    $empty_element_map->{ulink}        = 1;
    $empty_element_map->{imagedata}    = 1;
    $empty_element_map->{area}         = 1;

    my $depth  = 0;
    my $indent = "\t";

    my ( $tag, $node, $start );    # per-iteration scratch

    # $_[0] = node
    # $_[1] = startflag
    # $_[2] = depth
    # $_[3] = parent
    # $_[4] = text node index

    $tree->traverse(
        sub {
            ( $node, $start ) = @_;
            if ( ref $node ) {    # it's an element
                                  # delete internal attrs
                $node->attr( 'depth', undef );
                $node->attr( 'name',  undef );

                $tag = $node->{'_tag'};

                #print(STDERR "tag: $tag\n");

                if ($start) {     # on the way in
                    if ( $banned_tags{$tag} ) {
                        croak(
                            maketext(
                                "ERROR: Banned tag ([_1]) detected. Discuss this with your brands owners if you think this is in error.",
                                $tag
                                )
                                . "\n"
                        );
                    }

                    foreach my $attr ( keys(%banned_attrs) ) {
                        if ( $node->attr($attr) ) {
                            croak(
                                maketext(
                                    "ERROR: Banned attribute ([_1]) detected. Discuss this with your brands owners if you think this is in error.",
                                    $attr
                                    )
                                    . "\n\n"
                            );
                        }
                    }

                    if ($clean_id) {
                        $self->Clean_ID($node);
                    }

                    if (( $MAP_OUT{$tag}->{'newline'} )
                        && (   ( not defined $MAP_OUT{$tag}->{mixed_mode} )
                            || ( not $MAP_OUT{$tag}->{mixed_mode} )
                            || (not $node->look_up( '_tag', qr/para|entry/ ) )
                        )
                        )
                    {
                        push( @xml, "\n", $indent x $depth );
                    }

                    if ( $MAP_OUT{$tag}->{verbatim} && $tag ne '~cdata' ) {
                        push( @xml, "\n" );
                    }
                    elsif ( $MAP_OUT{$tag}->{block} ) {

                   # Check to make sure the block is starting on it's own line
                   # If not add a new line and indent
                        if (( $xml[$#xml] && $xml[$#xml] =~ /\S/ )
                            && (( not defined $MAP_OUT{$tag}->{mixed_mode} )
                                || ( not $MAP_OUT{$tag}->{mixed_mode} )
                                || (not $node->look_up( '_tag',
                                        qr/para|entry/ ) )
                            )
                            )
                        {
                            push( @xml, "\n", $indent x $depth );
                        }
                        $depth++;
                    }

                    if ( $tag eq 'imagedata' || $tag eq 'graphic' ) {
                        $node->attr('fileref') =~ m/(...)$/;
                        my $format = uc($1);
                        if ( !$node->attr('format') && $format ) {
                            $node->attr( 'format', $format );
                        }

                        my $img_file = "$path" . $node->attr('fileref');
                        $img_file
                            = $self->{publican}->param('xml_lang') . "/"
                            . $img_file
                            if ($clean_id || $unique_id);
                        if ( -f $img_file ) {

                            #nop
                        }
                        elsif ( $img_file !~ /Common_Content/ ) {
                            logger(
                                "\t"
                                    . maketext(
                                    "WARNING: Image missing: [_1]",
                                    $img_file )
                                    . "\n",
                                RED
                            );
                        }

                  # when building distrubuted sets, we need to prepend the
                  # books name to the image path to prevent image name clashes
                        my $preptxt
                            = $self->{publican}->param('img_dir') . '/'
                            . $self->{publican}->param('docname');

                        if (   $self->{config}->param('distributed_set')
                            && $node->attr('fileref') !~ /^$preptxt/ )
                        {
                            $node->attr( 'fileref',
                                "$preptxt/" . $node->attr('fileref') );
                        }
                    }

                    if ( $tag eq '~cdata' ) {
                        push( @xml, '<![CDATA[' );
                    }
                    elsif ( $empty_element_map->{$tag}
                        and !@{ $node->content_array_ref() } )
                    {
                        push( @xml, $node->starttag_XML( undef, 1 ) );
                        if ($MAP_OUT{$tag}->{newline_after}
                            && (( not defined $MAP_OUT{$tag}->{mixed_mode} )
                                || ( not $MAP_OUT{$tag}->{mixed_mode} )
                                || (not $node->look_up( '_tag',
                                        qr/para|entry/ ) )
                            )
                            )
                        {
                            push( @xml, "\n", $indent x $depth );
                        }
                    }
                    else {
                        push( @xml, $node->starttag_XML(undef) );
                    }

                    if ( $MAP_OUT{$tag}->{block} ) {
                        if (   $node->parent()
                            && $MAP_OUT{ $node->parent()->{'_tag'} }
                            ->{'line_wrap'} )
                        {
                            push( @xml, "\n" );
                        }
                        elsif (
                            ( not $MAP_OUT{$tag}->{verbatim} )
                            && (( not defined $MAP_OUT{$tag}->{mixed_mode} )
                                || ( not $MAP_OUT{$tag}->{mixed_mode} )
                                || (not $node->look_up( '_tag',
                                        qr/para|entry/ ) )
                            )
                            )
                        {
                            push( @xml, "\n", $indent x $depth );
                        }
                    }
                }
                else {    # on the way out
                    if ( $MAP_OUT{$tag}->{block} ) {

                        # remove empty lines
                        if ( $xml[$#xml] =~ /^[\t ]*$/s ) {
                            pop(@xml);
                            if ( $xml[$#xml] =~ /^[\t ]*$/s ) {
                                pop(@xml);
                            }
                        }

                        # remove trailing space
                        if ( $xml[$#xml] =~ /[\t ]*$/ )    # ||
                        {
                            $xml[$#xml] =~ s/[\t ]*$//;
                        }

                        if ( $MAP_OUT{$tag}->{block} ) {
                            if ( $MAP_OUT{$tag}->{verbatim} ) {
## BZ #604465 don't add trailing newline.
##                                push( @xml, "\n" );
                            }
                            elsif (( defined $MAP_OUT{$tag}->{mixed_mode} )
                                && ( $MAP_OUT{$tag}->{mixed_mode} )
                                && ($node->look_up( '_tag', qr/para|entry/ ) )
                                )
                            {
                                $depth--;
                            }
                            elsif ($node->parent()
                                && $MAP_OUT{ $node->parent()->{'_tag'} }
                                ->{'line_wrap'} )
                            {
                                $depth--;
                                push( @xml, "\n" );
                            }
                            else {
                                $depth--;
                                push( @xml, "\n", $indent x $depth );
                            }
                        }
                    }

                    unless ( $empty_element_map->{$tag}
                        and !@{ $node->content_array_ref() } )
                    {
                        if ( $tag eq '~cdata' ) {
                            push( @xml, qq|]]>| );
                        }
                        else {
                            push( @xml, $node->endtag_XML() );
                        }
                    }    # otherwise it will have been an <... /> tag.

                    if (( $MAP_OUT{$tag}->{newline_after} )
                        && (   ( not defined $MAP_OUT{$tag}->{mixed_mode} )
                            || ( not $MAP_OUT{$tag}->{mixed_mode} )
                            || (not $node->look_up( '_tag', qr/para|entry/ ) )
                        )
                        )
                    {
                        push( @xml, "\n", $indent x $depth );
                    }

                    if (( $MAP_OUT{$tag}->{block} )
                        && (   ( not defined $MAP_OUT{$tag}->{mixed_mode} )
                            || ( not $MAP_OUT{$tag}->{mixed_mode} )
                            || (not $node->look_up( '_tag', qr/para|entry/ ) )
                        )
                        )
                    {
                        push( @xml, "\n", $indent x $depth );
                    }
                }
            }
            else {    # it's just text
                my $parent = $_[3];

                # Remove extra space from non-verbatim tags
                if ( $parent
                    && !( $MAP_OUT{ $parent->{'_tag'} }->{verbatim} ) )
                {

                  # Don't out put empty tags
                  # BZ #453067 but spaces between inline tags should be output
                    if ( $node !~ /^[\t ]*$/ || $node !~ /\n/ ) {

                        # Truncate leading space
                        $node =~ s/[\n\r\f\t ]+/ /g;

                        if ( $MAP_OUT{ $parent->{'_tag'} }->{block} ) {

                     # for the first child, remove leading space and indent it
                            if ( $_[4] == 0 ) {
                                $node =~ s/^ //g;
                            }
                        }

                        $tree->_xml_escape($node);
                        $node =~ s/&#34;/"/g;
                        $node =~ s/&#39;/'/g;
                        $node =~ s/&quot;/"/g;
                        $node =~ s/&apos;/'/g;

                        push( @xml, $node );
                    }
                }
                else {    # Verbatim
                    if ( $parent->{'_tag'} ne '~cdata' ) {
                        $tree->_xml_escape($node);
                        $node =~ s/&#34;/"/g;
                        $node =~ s/&#39;/'/g;
                        $node =~ s/&quot;/"/g;
                        $node =~ s/&apos;/'/g;

                    }

                    push( @xml, $node );
                }
            }
            1;    # keep traversing
        }
    );

    return ( join( '', @xml, "\n" ) );
}

=head2 validate_tables

Ensure Tables comply to requirements not enforceable in XML validation.

1. tgroup attribute cols must match the number of entries in every row.

=cut

sub validate_tables {
    my ( $self, $xml_doc ) = @_;

    if ($xml_doc) {
        $xml_doc->pos( $xml_doc->root() );

        foreach my $node ( $xml_doc->look_down( "_tag", "tgroup" ) ) {

            # TODO this should report the line number
            # until then it try's to determine the Tables title or id
            my $table = $node->look_up( "_tag", qr/table|informaltable/ );
            if ( !$table ) {
                logger(
                    maketext(
                        "WARNING: table validation failed. Could not determine table for tgroup, column numbers cannot be validated"
                    )
                );
                next;
            }
            my $title = $table->look_down( "_tag", "title" );
            if ($title) {
                $title = $title->as_text();
            }
            else {
                $title = ( $table->attr('id') || "Can't identify table" );
            }
            my $cols = $node->attr('cols')
                || croak maketext(
                "*ERROR: Fatal Table Error* Table ([_1]) contains invalid data\nAttribute cols is mandatory for tgroup",
                $title
                ) . "\n";

            foreach my $row ( $node->look_down( "_tag", "row" ) ) {
                my @entries = $row->look_down( "_tag", qr/(?:entry|entrytbl)/,
                    sub { $_[0]->depth() == ( $row->depth() + 1 ) } );
## BUGBUG this is incomplete, it should check colwidths
                if ( @entries > $cols ) {
                    croak maketext(
                        "*ERROR: Fatal Table Error* Table ([_1]) contains invalid data\nAttribute cols ([_2]) does not match number of entry elements ([_3])",
                        $title, $cols, scalar @entries
                    ) . "\n";
                }
            }
        }
    }

    return;
}

=head2 process_file

Create XML::TreeBuilder object and perform operations.

=cut

sub process_file {
    my ( $self, $args ) = @_;

    my $file = delete( $args->{file} )
        || croak( maketext("file is a mandatory argument") );
    my $out_file = delete( $args->{out_file} )
        || croak( maketext("out_file is a mandatory argument") );

    if ( %{$args} ) {
        croak(
            maketext(
                "unknown arguments: [_1]", join( ", ", keys %{$args} )
            )
        );
    }

    logger(   "\t"
            . maketext( "Processing file [_1] -> [_2]", $file, $out_file )
            . "\n" );

    my $clean_id        = $self->{config}->param('clean_id');
    my $update_includes = $self->{config}->param('update_includes');
    my $xml_lang        = $self->{publican}->param('xml_lang');

    my $xml_doc = XML::TreeBuilder->new(
        { 'NoExpand' => "1", 'ErrorContext' => "2" } );
    $xml_doc->store_comments(1);
    $xml_doc->store_pis(1);
    $xml_doc->store_cdata(1);
    $xml_doc->store_declarations(1);
##debug_msg("here 1");
    eval { $xml_doc->parse_file($file); };

    croak( maketext( "Can't open file '[_1]' [_2]", $file, $@ ) ) if ($@);
##debug_msg("here 2");

    $self->validate_tables($xml_doc);

    if ($update_includes) {
        $self->update_include($xml_doc);
    }
    elsif ( !$clean_id ) {
        $self->prune_xml($xml_doc);
    }

    $self->print_xml( { xml_doc => $xml_doc, out_file => $out_file } );

    if ($clean_id) {

        # Update links in xml
        foreach my $xml_file ( dir_list( $xml_lang, '*.xml' ) ) {
            my $editor = new File::Inplace( file => $xml_file );
            my $perm   = ( stat $editor->{infh} )[2] & 07777;
            my $uid    = ( stat $editor->{infh} )[4];
            my $gid    = ( stat $editor->{infh} )[5];

            while ( my ($line) = $editor->next_line ) {
                foreach my $key ( keys(%UPDATED_IDS) ) {
                    $line =~ s/linkend="$key"/linkend="$UPDATED_IDS{$key}"/g;
                }
                $editor->replace_line($line);
            }

            $editor->commit;
            $editor = undef;
            chmod( $perm | 0600, $xml_file ) || logger(
                maketext(
                    "WARNING: Could not reset file permissions for [_1] because [_2]",
                    $xml_file,
                    $!
                )
            );
            chown( $uid, $gid, $xml_file ) || logger(
                maketext(
                    "WARNING: Could not reset file ownership for [_1] because [_2]",
                    $xml_file,
                    $!
                )
            );
        }

        # update links in PO files
        foreach my $dir ( split( /,/, get_all_langs() ) ) {
            next if ( $dir eq $xml_lang );
            foreach my $po_file ( dir_list( $dir, '*.po' ) ) {
                my $editor = new File::Inplace( file => $po_file );
                while ( my ($line) = $editor->next_line ) {
                    foreach my $key ( keys(%UPDATED_IDS) ) {

                        # all of string on one line
                        $line =~ s/=\\"$key\\"/=\\"$UPDATED_IDS{$key}\\"/g;

                        # tail of string line wrapped
                        $line =~ s/=\\"$key"/=\\"$UPDATED_IDS{$key}"/g;

                        # string line wrapped after '='
                        $line =~ s/\\"$key\\"/\\"$UPDATED_IDS{$key}\\"/g;
                    }
                    $editor->replace_line($line);
                }

                $editor->commit;
            }
        }

# clear out changes ... might be better to save them up and do a single pass...
        %UPDATED_IDS = ();
    }

    return;
}

=head2 set_unique_ids

Set unique ids for every nodes which have id

=cut

sub set_unique_ids {
    my ( $self, $args ) = @_;

    my $xml_lang = $self->{publican}->param('xml_lang');

    # create database to track section id changes
    my $db_file = "$xml_lang/max_unique_id.db";
    $self->{dbh}
        = DBI->connect( "dbi:SQLite:dbname=$db_file", "", "",
        { RaiseError => 1 } )
        || croak( maketext($DBI::errstr) );
    $self->create_db();

    my $sql = "select max_unique_id from max_unique_id";
    my $sth = $self->{dbh}->prepare($sql);
    $sth->execute();
    my $result = $sth->fetchrow_hashref();

    my $unique_id = $result->{max_unique_id} || 0;
    $sth->finish();

    my @xml_files;
    @xml_files = dir_list( $xml_lang, '*.xml' );

    my $conformance = 0;

    foreach my $xml_file ( sort(@xml_files) ) {
        next if ( $xml_file =~ m{/extras/} );

        my $xml_doc = XML::TreeBuilder->new(
            { 'NoExpand' => "1", 'ErrorContext' => "2" } );
        $xml_doc->store_comments(1);
        $xml_doc->store_pis(1);
        $xml_doc->store_cdata(1);
        $xml_doc->store_declarations(1);

        $xml_doc->parse_file($xml_file);

        # based on as_HTML
        my $tree              = $xml_doc->root();
        my $empty_element_map = $tree->_empty_element_map;

        # This flags tags that use  /> instead of end tags IF they are empty.
        $empty_element_map->{xref}         = 1;
        $empty_element_map->{footnoteref}  = 1;
        $empty_element_map->{'index'}      = 1;
        $empty_element_map->{'xi:include'} = 1;
        $empty_element_map->{ulink}        = 1;
        $empty_element_map->{imagedata}    = 1;
        $empty_element_map->{area}         = 1;

        my ( $tag, $node, $start );    # per-iteration scratch

        # $_[0] = node
        # $_[1] = startflag
        # $_[2] = depth
        # $_[3] = parent
        # $_[4] = text node index

        $tree->traverse(
            sub {
                ( $node, $start ) = @_;
                if ( ref $node ) {
                    $tag = $node->{'_tag'};

                    if (   $start
                        && $node->attr( $self->{config}->param('id_attr') )
                        && !$node->attr('conformance') )
                    {    # on the way in
                        $node->attr( 'conformance', ++$unique_id );
                    }
                }
                1;       # keep traversing
            }
        );

       $self->print_xml( { xml_doc => $tree, out_file => $xml_file } );
    }

    eval {
        my $update_sql = <<SQL;
            UPDATE max_unique_id 
                SET max_unique_id = $unique_id
              WHERE
                id  = 1
SQL
        $self->{dbh}->do($update_sql);
    };

    if ($@) {
        $self->{dbh}->rollback();
        $self->{dbh}->disconnect();
        croak( maketext("Error setting max id: $@") );
    }

    $self->{dbh}->disconnect();

    return;
}

=head2 create_db

Create a database to track the max unique id

=cut

sub create_db {
    my $self = shift;
    my $sql;

    eval {
        my $check_table_sql = <<SQL;
          SELECT name 
            FROM sqlite_master 
           WHERE name='max_unique_id'
SQL

        my $create_table_sql = <<SQL;
          CREATE TABLE max_unique_id (
            id             INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
            max_unique_id  INTEGER NOT NULL DEFAULT 0
        )
SQL

        my $table = $self->{dbh}->selectrow_hashref($check_table_sql);

        return if ( defined $table->{name} );

        $self->{dbh}->do($create_table_sql);

        # create first row and set value to 0
        my $insert_sql = <<SQL;
             INSERT INTO max_unique_id 
               (max_unique_id) 
               VALUES (0)
SQL

        $self->{dbh}->do($insert_sql);

    };

    if ($@) {
        $self->{dbh}->rollback();
        croak( maketext("Failed to create table: $@") );
    }

    return;
}

1;    # Magic true value required at end of module

=head1 DIAGNOSTICS

=over

=item C<< unknown args %s >>

All subs with named parameters will return this error when unexpected named arguments are provided.

=item C<< %s is a required argument >>

Any sub with a mandatory parameter will return this error if the parameter is undef.

=item C<< Could not open %s for output! >>

The named file could not be opened.

=item C<< Can't calculate image size of %s >>

Images are automatically scaled if thy are to wide, this check could not be
performed due to either access permissions or file weirdness.

=back

=head1 CONFIGURATION AND ENVIRONMENT

Publican::XmlClean requires no configuration files or environment variables.


=head1 DEPENDENCIES

Carp
version
XML::TreeBuilder
Text::Wrap
Config::Simple
Publican
File::Path
Term::ANSIColor
Cwd

=head1 INCOMPATIBILITIES

None reported.


=head1 BUGS AND LIMITATIONS

No bugs have been reported.

Please report any bugs or feature requests to
C<publican-list@redhat.com>, or through the web interface at
L<https://bugzilla.redhat.com/bugzilla/enter_bug.cgi?product=Publican&component=publican>.

=head1 AUTHOR

Jeff Fearn  C<< <jfearn@redhat.com> >>
