#!/usr/bin/perl -w
##############################################################################
#
# Alex Burger - Oct 28th, 2004
#
# Purpose: Modify .html files to add a header and footer for use
# on the Net-SNMP web site.
#
# Can also be used to change the 'section' variable
# for use in the menu system.
#
# Notes: A backup of each file is made to *.old.
#
# Any DOS newlines are removed from the destination file.
#
# Permissions are maintained.
#
##############################################################################
#
use File::Copy;
use File::stat;
use Getopt::Long;
my $tidy_options = '-f /dev/null -m -i -asxhtml -wrap 130 -quiet --quote-nbsp n';
my $pattern = '';
my $section = '';
my $tidy = 0;
my $body = 0;
my $help = 0;
my @files = ();
GetOptions ('pattern=s' => \$pattern,
'section=s' => \$section,
'tidy' => \$tidy,
'body' => \$body,
'help' => \$help);
if ($help == 1)
{
$USAGE = qq/
Usage:
add-header-footer [<options>] file1 file2 file3 ...
Options:
--section= Menu section
--tidy Run tidy on input file before processing (turns on --body)
--body Remove everything before <body> and after <\/body>
--help Display this message
Examples:
add-header-footer.pl --section=tutorial --body cat.html dog.html mouse.html
find . -name '*.html' | add-header-footer.pl --section=tutorial --body
/;
print $USAGE;
exit 0;
}
if ($ARGV[0]) {
# Files listed on command line
foreach my $arg (@ARGV) {
chomp $arg;
push @files, $arg;
#print "$arg\n";
}
}
else {
# No arguments, so accept STDIN
while (<STDIN>) {
chomp;
push @files, $_;
#print "$_\n";
}
}
if (! (@files) ) {
exit 0;
}
#print "@files";
foreach my $file (@files) {
chomp $file;
print "Processing file: $file\n";
# Grab current permissions
my $sb = stat($file);
my $stat_permissions = sprintf ("%04o", $sb->mode & 07777);
my $stat_uid = $sb->uid;
my $stat_gid = $sb->gid;
my @old_file = ();
my @new_file = ();
my $body_count = 0;
# Backup old file
if (! (copy ("$file", "$file.old"))) {
print "Could not backup existing file $file to $file.new. Aborting.\n";
next;
}
# Set permissions on old file to match original file
chmod oct($stat_permissions), "$file.old";
chown $stat_uid, $stat_uid, "$file.old";
if ($tidy == 1) {
$body = 1; # Enable body, as tidy will add it in.
my $tidy_command = "tidy $tidy_options $file";
`$tidy_command`;
}
if (open (I, "<$file")) {
# Load entire file
while (<I>) {
s/\015//g; # Remove any DOS newlines
chomp;
push (@old_file, $_);
}
}
else {
print "Could not open file $file. Aborting\n";
next;
}
if (!@old_file) {
print "Empty file. Skipping\n";
next;
}
# Remove empty lines at start
while (1) {
if ($old_file[0] eq "") {
splice (@old_file, 0, 1);
}
else {
last;
}
}
# Remove empty lines at end
while (1) {
if ($old_file[$#old_file] eq "") {
splice (@old_file, -1, 1);
}
else {
last;
}
}
if ($body == 1) {
# Count the number of <body lines
for (my $i = 0; $i <= $#old_file; $i++) {
if ($old_file[$i] =~ /<body/) {
$body_count++;
next;
}
}
# Remove anything before and including <body
while ($body_count > 0) {
while (! ($old_file[0] =~ /<body/)) {
splice (@old_file, 0, 1);
}
splice (@old_file, 0, 1); # <body line
$body_count--;
}
}
# Start to build new file in memory with header
push (@new_file, "<!--#set var=\"section\" value=\"$section\" -->\n");
push (@new_file, '<!--#include virtual="/page-top.html" -->' . "\n");
push (@new_file, '<!-- CONTENT START -->' . "\n");
# Add in old file, skipping existing header and footer and stopping at <body/>
for (my $i = 0; $i <= $#old_file; $i++) {
if (!(defined($old_file[$i]))) { next; }
if ($body == 1 && ($old_file[$i] =~ /<\/body>/)) { last; }
elsif ($old_file[$i] =~ /<!--#set var="section" value=/) { next; }
elsif ($old_file[$i] =~ /<!--#include virtual="\/page-top.html" -->/) { next; }
elsif ($old_file[$i] =~ /<!-- CONTENT START -->/) { next; }
elsif ($old_file[$i] =~ /<!-- CONTENT END -->/) { next; }
elsif ($old_file[$i] =~ /<!--#include virtual="\/page-bottom.html" -->/) { next; }
push (@new_file, $old_file[$i] . "\n");
}
# Finish to building new file in memory with footer
push (@new_file, '<!-- CONTENT END -->' . "\n");
push (@new_file, '<!--#include virtual="/page-bottom.html" -->' . "\n");
# Save new file
if (open (O, ">$file")) {
for (my $i = 0; $i <= $#new_file; $i++) {
print O "$new_file[$i]";
}
print O "\n";
close O;
# Set permissions
chmod oct($stat_permissions), $file;
chown $stat_uid, $stat_uid, $file;
}
else {
print "Could not create new file: $file.new\n"
}
close I;
}