#-------------------------------------------------------------------
# WebGUI is Copyright 2001-2006 Plain Black Corporation.
#-------------------------------------------------------------------
# Please read the legal notices (docs/legal.txt) and the license
# (docs/license.txt) that came with this distribution before using
# this software.
#-------------------------------------------------------------------
# http://www.plainblack.com                     info@plainblack.com
#-------------------------------------------------------------------

use lib "../../lib";
use strict;
use Getopt::Long;
use WebGUI::Session;
use WebGUI::Storage;
use WebGUI::Asset;


my $toVersion = "0.0.0"; # make this match what version you're going to
my $quiet; # this line required


my $session = start(); # this line required

# upgrade functions go here

finish($session); # this line required


##-------------------------------------------------
#sub exampleFunction {
#	my $session = shift;
#	print "\tWe're doing some stuff here that you should know about.\n" unless ($quiet);
#	# and here's our code
#}


# --------------- DO NOT EDIT BELOW THIS LINE --------------------------------

#----------------------------------------------------------------------------
# Add a package to the import node
sub addPackage {
    my $session     = shift;
    my $file        = shift;

    # Make a storage location for the package
    my $storage     = WebGUI::Storage->createTemp( $session );
    $storage->addFileFromFilesystem( $file );

    # Import the package into the import node
    WebGUI::Asset->getImportNode($session)->importPackage( $storage );

    # Make the package not a package anymore
}

#----------------------------------------------------------------------------
# Add a template from a file
sub addTemplate {
    my $session     = shift;
    my $file        = shift;
    my $newFolder   = shift;

    open(FILE,"<",$file);
    my $first = 1;
    my $create = 0;
    my $head = 0;
    my %properties = (className=>"WebGUI::Asset::Template");
    while (my $line = <FILE>) {
        if ($first) {
            $line =~ m/^\#(.*)$/;
            $properties{id} = $1;
            $first = 0;
        } 
        elsif ($line =~ m/^\#create$/) {
            $create = 1;
        } 
        elsif ($line =~ m/^\#(.*):(.*)$/) {
            $properties{$1} = $2;
        } 
        elsif ($line =~ m/^~~~$/) {
            $head = 1;
        } 
        elsif ($head) {
            $properties{headBlock} .= $line;
        } 
        else {
            $properties{template} .= $line;	
        }
    }
    close(FILE);
    if ($create) {
        $$newFolder = createNewTemplatesFolder(WebGUI::Asset->getImportNode($session)) 
            unless (defined $$newFolder);
        my $template = $$newFolder->addChild(\%properties, $properties{id});
    } 
    else {
        my $template = WebGUI::Asset->new($session,$properties{id}, "WebGUI::Asset::Template");
        if (defined $template) {
            my $newRevision = $template->addRevision(\%properties);
        }
    }
}

#-------------------------------------------------
sub start {
	my $configFile;
	$|=1; #disable output buffering
	GetOptions(
    		'configFile=s'=>\$configFile,
        	'quiet'=>\$quiet
	);
	my $session = WebGUI::Session->open("../..",$configFile);
	$session->user({userId=>3});
	my $versionTag = WebGUI::VersionTag->getWorking($session);
	$versionTag->set({name=>"Upgrade to ".$toVersion});
	$session->db->write("insert into webguiVersion values (".$session->db->quote($toVersion).",'upgrade',".$session->datetime->time().")");
	updateTemplates($session);
	return $session;
}

#-------------------------------------------------
sub finish {
	my $session = shift;
	my $versionTag = WebGUI::VersionTag->getWorking($session);
	$versionTag->commit;
	$session->close();
}

#-------------------------------------------------
sub updateTemplates {
    my $session = shift;
    return undef unless (-d "templates-".$toVersion);
    print "\tUpdating templates.\n" unless ($quiet);
    opendir(DIR,"templates-".$toVersion);
    my @files = readdir(DIR);
    closedir(DIR);
    my $newFolder = undef;
    foreach my $file (@files) {
        next unless ($file =~ /\.(tmpl|wgpkg)$/);
        my $type    = $1;
        $file       = "templates-" . $toVersion . "/" . $file;
        if ($type eq "tmpl") {
            addTemplate( $session, $file, \$newFolder );
        }
        elsif ($type eq "wgpkg") {
            addPackage( $session, $file );
        }
    }
}

#-------------------------------------------------
sub createNewTemplatesFolder {
	my $importNode = shift;
	my $newFolder = $importNode->addChild({
		className=>"WebGUI::Asset::Wobject::Folder",
		title => $toVersion." New Templates",
		menuTitle => $toVersion." New Templates",
		url=> $toVersion."_new_templates",
		groupIdView=>"12"
		});
	return $newFolder;
}