# Copyright (c) 2008 - OpenSLX GmbH
#
# This program is free software distributed under the GPL version 2.
# See http://openslx.org/COPYING
#
# If you have any feedback please consult http://openslx.org/feedback and
# send your suggestions, praise, or complaints to feedback@openslx.org
#
# General information about OpenSLX can be found at http://openslx.org/
# -----------------------------------------------------------------------------
# BootEnvironment::Preboot.pm
#    - provides general preboot implementation of the BootEnvironment API.
# -----------------------------------------------------------------------------
package OpenSLX::BootEnvironment::PBS;

use strict;
use warnings;

use base qw(OpenSLX::BootEnvironment::Base);

use OpenSLX::MakeInitRamFS::Engine::PBS;

use Clone qw(clone);
use File::Basename;
use File::Path;

use Data::Dumper;

use JSON;
use HTTP::Request::Common;
use LWP::UserAgent;

use OpenSLX::Basics;
use OpenSLX::ConfigDB qw(:support);
use OpenSLX::Utils;

sub initialize
{
    my $self   = shift;
    my $params = shift;
    
    return if !$self->SUPER::initialize($params);

    $self->{'original-path'} = "$openslxConfig{'public-path'}/pbs";
    $self->{'target-path'}   = "$openslxConfig{'public-path'}/pbs.new";

    $self->{'requires-default-client-config'} = 0;
        # we do not need a default.tgz since there's always an explicit client

    
    if (!$self->{'dry-run'}) {
        mkpath([$self->{'original-path'}]);
        rmtree($self->{'target-path'});
        mkpath("$self->{'target-path'}/client-config");
    }

    return 1;
}

sub writeBootloaderMenuFor
{
    my $self             = shift;
    my $client           = shift;
    my $externalClientID = shift;
    my $systemInfos      = shift || [];

    my $prebootSystemInfo
        = clone($self->_pickSystemWithNewestKernel($systemInfos));

    vlog(
        0, 
        _tr(
            "\nsend preboot information for client '%s' to pbs (%s)\n".
            "  (image templates provided based of %s) ...", 
            $client->{name}, $client->{attrs}->{preboot_server}, $prebootSystemInfo->{name}
        )
    );

    $self->_createPrebootStuff($client, $prebootSystemInfo);

    my $kernel      = "$self->{'target-path'}/imagebase/vmlinuz";
    my $initramfs   = "$self->{'target-path'}/imagebase/initramfs";
    
    my $kernel_md5 = qx/md5sum $kernel | awk '{print \$1}'/;
    my $initramfs_md5 = qx/md5sum $initramfs | awk '{print \$1}'/;

    my $data_json = to_json({
    	'slxinfo'   => qx/slxversion/,
    	'kernel'    => basename($prebootSystemInfo->{'kernel-file'}),
        'kernel_md5' => trim($kernel_md5), 
        'initramfs_md5' => trim($initramfs_md5), 
    	'systems'   => $systemInfos
    });
    my $ua = LWP::UserAgent->new;
    my $res = $ua->request(POST 'http://pbs.experimental.openslx.org/backend/system/sync', [data => $data_json]);
    
    if ($res->is_success) {
    	my $resData = from_json($res->content);
    	if ($resData->{'getKernel'} eq 'fresh') {
            $res = $ua->request(POST 'http://pbs.experimental.openslx.org/backend/system/addkernel', 
                ['kernel' => basename($prebootSystemInfo->{'kernel-file'}),
                 'kernelFile' => ["$self->{'target-path'}/imagebase/vmlinuz"],
                 'initramfsFile' => ["$self->{'target-path'}/imagebase/initramfs"],
                  ],
                'Content_Type' => 'form-data'
            );
            
            print Dumper($res->content);
    		
    	} else {
            if ($resData->{'getKernel'} eq 'update') {
                $res = $ua->request(POST 'http://pbs.experimental.openslx.org/backend/system/updatekernel', 
                    ['kernel' => basename($prebootSystemInfo->{'kernel-file'}),
                     'kernelFile' => ["$self->{'target-path'}/imagebase/vmlinuz"],
                      ],
                    'Content_Type' => 'form-data'
                );
                
                print Dumper($res->content);
            } else {
                # do nothing
            }
            if ($resData->{'getInitramfs'} eq 'update') {
	            $res = $ua->request(POST 'http://pbs.experimental.openslx.org/backend/system/updateinitramfs', 
	                ['kernel' => basename($prebootSystemInfo->{'kernel-file'}),
	                 'initramfsFile' => ["$self->{'target-path'}/imagebase/initramfs"],
	                  ],
	                'Content_Type' => 'form-data'
	            );
	            
	            print Dumper($res->content);
            } else {
                # do nothing
            }
    	}
    } else {
    	vlog(0, 'communication with pbs failed.. please check and rerun..');
    }

    return 1;
}

sub _createPrebootStuff
{
    my $self   = shift;
    my $client = shift;
    my $info   = shift;
    
    my $prebootClass = instantiateClass(
        "OpenSLX::BootEnvironment::Preboot::Base"
    );
    
    my $imagebase = "$self->{'target-path'}/imagebase";
    
    $prebootClass->initialize($self);
    $client->{attrs}->{boot_uri} = $client->{attrs}->{preboot_server};
    mkpath("$imagebase");
    $self->_makePBSInitRamFS($info, "$imagebase/initramfs", $client);

    my $kernelFile = $info->{'kernel-file'};
    my $kernelName = basename($kernelFile);
    slxsystem(qq{cp -p "$kernelFile" "$imagebase/vmlinuz"})
        unless $self->{'dry-run'};
    
    return 1;
}

sub _pickSystemWithNewestKernel
{
    my $self        = shift;
    my $systemInfos = shift;

    my $systemWithNewestKernel;
    my $newestKernelFileSortKey = '';
    foreach my $system (@$systemInfos) {
        next unless $system->{'kernel-file'} =~ m{
            (?:vmlinuz|x86)-(\d+)\.(\d+)\.(\d+)(?:\.(\d+))?-(\d+(?:\.\d+)?)
        }x;
        my $sortKey 
            = sprintf("%02d.%02d.%02d.%02d-%2.1f", $1, $2, $3, $4||0, $5);
        if ($newestKernelFileSortKey lt $sortKey) {
            $systemWithNewestKernel  = $system;
            $newestKernelFileSortKey = $sortKey;
        }
    }

    if (!defined $systemWithNewestKernel) {
        die _tr("unable to pick a system to be used for preboot!");
    }
    return $systemWithNewestKernel;
}

sub _makePBSInitRamFS
{
    my $self       = shift;
    my $info       = shift;
    my $initramfs  = shift;
    my $client     = shift;

    my $vendorOS = $info->{'vendor-os'};
    my $kernelFile = basename(followLink($info->{'kernel-file'}));

    my $attrs = clone($info->{attrs} || {});

    my $bootURI = $client->{attrs}->{boot_uri};
    if (!$bootURI) {
        die _tr("client $client->{name} needs an URI in attribute 'boot_uri' to be used for preboot!");
    }

    chomp(my $slxVersion = qx{slxversion});

    my $params = {
        'attrs'          => $attrs,
        'export-name'    => undef,
        'export-uri'     => undef,
        'initramfs'      => $initramfs,
        'kernel-params'  
            => [ split ' ', ($info->{attrs}->{kernel_params} || '') ],
        'kernel-version' => $kernelFile =~ m[-(.+)$] ? $1 : '',
        'plugins'        => '',
        'root-path'
            => "$openslxConfig{'private-path'}/stage1/$vendorOS->{name}",
        'slx-version'    => $slxVersion,
        'system-name'    => $info->{name},
        'preboot-id'     => $client->{name},
        'boot-uri'       => $bootURI,
    };

    # TODO: make debug-level an explicit attribute, it's used in many places!
    my $kernelParams = $info->{attrs}->{kernel_params} || '';
    if ($kernelParams =~ m{debug(?:=(\d+))?}) {
        my $debugLevel = defined $1 ? $1 : '1';
        $params->{'debug-level'} = $debugLevel;
    }

    my $makeInitRamFSEngine 
        = OpenSLX::MakeInitRamFS::Engine::PBS->new($params);
    $makeInitRamFSEngine->execute($self->{'dry-run'});

    # copy back kernel-params, as they might have been changed (by plugins)
    $info->{attrs}->{kernel_params} 
        = join ' ', $makeInitRamFSEngine->kernelParams();

    return;
}

1;