# IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html
#-------------------------------------------------------

=head1
  xCAT plugin package to handle getpostscript command

=cut

#-------------------------------------------------------
package xCAT_plugin::getpostscript;
use xCAT::Utils;
use xCAT::MsgUtils;
use xCAT::NodeRange;
1;

#-------------------------------------------------------

=head3  handled_commands

Return list of commands handled by this plugin

=cut

#-------------------------------------------------------

sub handled_commands
{
    return { 
               'getpostscript' => "getpostscript",
               'postage' => "getpostscript"
           };
}


#-------------------------------------------------------

=head3  process_request

  Process the command

=cut

#-------------------------------------------------------
sub process_request
{
    my $request  = shift;
    my $callback = shift;
    my $nodes    = $request->{node};
    my $command  = $request->{command}->[0];
    my $args     = $request->{arg};
    my $envs     = $request->{env};
    my $rsp;
    my $i = 1;
    my @nodes=@$nodes; 
    # do your processing here
    # return info
    
    if($command eq 'postage'){
        return postage($nodes, $callback);
    }

    my $client;
    if ($::XCATSITEVALS{nodeauthentication}) { #if requiring node authentication, this request will have a certificate associated with it, use it instead of name resolution
	unless (ref $request->{username}) { return; } #TODO: log an attempt without credentials? 
	$client = $request->{username}->[0];
    } else {
	    unless ($request->{'_xcat_clienthost'}->[0]) {
	      #ERROR? malformed request
	      return; #nothing to do here...
	    }
	    $client = $request->{'_xcat_clienthost'}->[0];
    }

    my $origclient = $client;
    if ($client) { ($client) = noderange($client) };
    unless ($client) { #Not able to do identify the host in question
       xCAT::MsgUtils->message("S","Received getpostscript from $origclient, which couldn't be correlated to a node (domain mismatch?)");
      return;
    }
    my $state;
    if ($request->{scripttype}) { $state = $request->{scripttype}->[0];}

    require xCAT::Postage;
    my $args = $request->{arg};
    my @scriptcontents;
    if( defined($args) && grep(/version2/, @$args)) {
        @scriptcontents = xCAT::Postage::makescript([$client],$state,$callback);  # the new method, use the template
    } else {
        #print "client:$client\n";
        @scriptcontents = xCAT::Postage::makescript($client,$state,$callback); # the original method
    }
    if (scalar(@scriptcontents)) {
       $rsp->{data} = \@scriptcontents;
    }
    $callback->($rsp);
}

sub postage {
    my $nodes = shift;
    my $callback = shift;
    require xCAT::Postage;
    foreach my $node (@$nodes){
        my @scriptcontents = xCAT::Postage::makescript($node,'postscripts',$callback);
        my $ps = 0;
        my $pbs = 0;
        foreach(@scriptcontents){
            chomp($_);
            if($_ =~ "postscripts-start-here"){
                $ps = 1;
                next;
              
            }
            if($_ =~ "postscripts-end-here"){
                $ps = 0;
                next;
            }
            if($_ =~ "postbootscripts-start-here"){
                $pbs = 1;
                next;
            }
            if($_ =~ "postbootscripts-end-here"){
                $pbs = 0;
                next;
            }
            if($ps eq 1){ 
                $callback->({info => ["$node: postscript: $_"]});
            }
            if($pbs eq 1){
                $callback->({info => ["$node: postbootscript: $_"]});
            }
        }
    }  
}