<?php
/*=========================================================================
  Program:   CDash - Cross-Platform Dashboard System
  Module:    $Id$
  Language:  PHP
  Date:      $Date$
  Version:   $Revision$

  Copyright (c) Kitware, Inc. All rights reserved.
  See LICENSE or http://www.cdash.org/licensing/ for details.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE. See the above copyright notices for more information.
=========================================================================*/

// It is assumed that appropriate headers should be included before including this file
include_once 'models/testimage.php';
include_once 'models/testmeasurement.php';
include_once 'models/buildtestdiff.php';
include_once 'models/buildtest.php';
include_once 'models/label.php';
include_once 'models/testimage.php';
include_once 'models/image.php';
include_once 'models/constants.php';

/** Test */
class Test
{
    public $Id;
    public $Crc32;
    public $ProjectId;
    public $Name;
    public $Path;
    public $Command;
    public $Details;
    public $Output;
    public $CompressedOutput;

    public $Images;
    public $Labels;
    public $Measurements;

    public function __construct()
    {
        $this->Images = array();
        $this->Labels = array();
        $this->Measurements = array();
        $this->CompressedOutput = false;
    }

    public function AddMeasurement($measurement)
    {
        $measurement->TestId = $this->Id;
        $this->Measurements[] = $measurement;

        if ($measurement->Name == 'Label') {
            $label = new Label();
            $label->SetText($measurement->Value);
            $this->AddLabel($label);
        }
    }

    public function AddImage($image)
    {
        $this->Images[] = $image;
    }

    public function AddLabel($label)
    {
        $label->TestId = $this->Id;
        $this->Labels[] = $label;
    }

    /** Get the CRC32 */
    public function GetCrc32()
    {
        if (strlen($this->Crc32) > 0) {
            return $this->Crc32;
        }

        $command = pdo_real_escape_string($this->Command);
        $output = pdo_real_escape_string($this->Output);
        $name = pdo_real_escape_string($this->Name);
        $path = pdo_real_escape_string($this->Path);
        $details = pdo_real_escape_string($this->Details);

        // CRC32 is computed with the measurements name and type and value
        $buffer = $name . $path . $command . $output . $details;

        foreach ($this->Measurements as $measurement) {
            $buffer .= $measurement->Type . $measurement->Name . $measurement->Value;
        }
        $this->Crc32 = crc32($buffer);
        return $this->Crc32;
    }

    public function InsertLabelAssociations($buildid)
    {
        if ($this->Id && $buildid) {
            foreach ($this->Labels as $label) {
                $label->TestId = $this->Id;
                $label->TestBuildId = $buildid;
                $label->Insert();
            }
        } else {
            add_log('No Test::Id or buildid - cannot call $label->Insert...',
                'Test::InsertLabelAssociations', LOG_ERR,
                $this->ProjectId, $buildid,
                CDASH_OBJECT_TEST, $this->Id);
        }
    }

    /** Return if exists */
    public function Exists()
    {
        $name = pdo_real_escape_string($this->Name);
        $crc32 = $this->GetCrc32();
        $query = pdo_query('SELECT id FROM test WHERE projectid=' . qnum($this->ProjectId)
            . " AND name='" . $name . "'"
            . " AND crc32='" . $crc32 . "'");
        if (pdo_num_rows($query) > 0) {
            $query_array = pdo_fetch_array($query);
            $this->Id = $query_array['id'];
            return true;
        }
        return false;
    }

    // Save in the database
    public function Insert()
    {
        if ($this->Exists()) {
            return true;
        }

        include 'config/config.php';
        $command = pdo_real_escape_string($this->Command);

        $name = pdo_real_escape_string($this->Name);
        $path = pdo_real_escape_string($this->Path);
        $details = pdo_real_escape_string($this->Details);

        $id = '';
        $idvalue = '';
        if ($this->Id) {
            $id = 'id,';
            $idvalue = "'" . $this->Id . "',";
        }

        if ($this->CompressedOutput) {
            if ($CDASH_DB_TYPE == 'pgsql') {
                $output = $this->Output;
            } else {
                $output = base64_decode($this->Output);
            }
        } elseif ($CDASH_USE_COMPRESSION) {
            $output = gzcompress($this->Output);
            if ($output === false) {
                $output = $this->Output;
            } else {
                if ($CDASH_DB_TYPE == 'pgsql') {
                    if (strlen($this->Output) < 2000) {
                        // compression doesn't help for small chunk

                        $output = $this->Output;
                    }
                    $output = base64_encode($output);
                }
            }
        } else {
            $output = $this->Output;
        }

        // We check for mysql that the
        if ($CDASH_DB_TYPE == '' || $CDASH_DB_TYPE == 'mysql') {
            $query = pdo_query("SHOW VARIABLES LIKE 'max_allowed_packet'");
            $query_array = pdo_fetch_array($query);
            $max = $query_array[1];
            if (strlen($this->Output) > $max) {
                add_log('Output is bigger than max_allowed_packet', 'Test::Insert', LOG_ERR, $this->ProjectId);
                // We cannot truncate the output because it is compressed (too complicated)
            }
        }

        $pdo = get_link_identifier()->getPdo();
        if ($this->Id) {
            $stmt = $pdo->prepare('
                INSERT INTO test (id, projectid, crc32, name, path, command, details, output)
                VALUES (:id, :projectid, :crc32, :name, :path, :command, :details, :output)');
            $stmt->bindParam(':id', $this->Id);
            $stmt->bindParam(':projectid', $this->ProjectId);
            $stmt->bindParam(':crc32', $this->Crc32);
            $stmt->bindParam(':name', $name);
            $stmt->bindParam(':path', $path);
            $stmt->bindParam(':command', $command);
            $stmt->bindParam(':details', $details);
            $stmt->bindParam(':output', $output, PDO::PARAM_LOB);
            $success = $stmt->execute();
        } else {
            $stmt = $pdo->prepare('
                INSERT INTO test (projectid, crc32, name, path, command, details, output)
                VALUES (:projectid, :crc32, :name, :path, :command, :details, :output)');
            $stmt->bindParam(':projectid', $this->ProjectId);
            $stmt->bindParam(':crc32', $this->Crc32);
            $stmt->bindParam(':name', $name);
            $stmt->bindParam(':path', $path);
            $stmt->bindParam(':command', $command);
            $stmt->bindParam(':details', $details);
            $stmt->bindParam(':output', $output, PDO::PARAM_LOB);
            $success = $stmt->execute();
            $this->Id = pdo_insert_id('test');
        }

        if (!$success) {
            add_last_sql_error(
                    "Cannot insert test: $name into the database", $this->ProjectId);
            return false;
        }

        // Add the measurements
        foreach ($this->Measurements as $measurement) {
            $measurement->TestId = $this->Id;
            $measurement->Insert();
        }

        // Add the images
        foreach ($this->Images as $image) {
            // Decode the data
            $imgStr = base64_decode($image->Data);
            $img = imagecreatefromstring($imgStr);
            ob_start();
            switch ($image->Extension) {
                case 'image/jpg':
                    imagejpeg($img);
                    break;
                case 'image/jpeg':
                    imagejpeg($img);
                    break;
                case 'image/gif':
                    imagegif($img);
                    break;
                case 'image/png':
                    imagepng($img);
                    break;
                default:
                    echo "Unknown image type: $type";
                    return;
            }
            $imageVariable = ob_get_contents();
            ob_end_clean();

            $image->Data = $imageVariable;
            $image->Checksum = crc32($imageVariable);
            $image->Save();

            $testImage = new TestImage();
            $testImage->Id = $image->Id;
            $testImage->TestId = $this->Id;
            $testImage->Role = $image->Name;
            $testImage->Insert();
        }
        return true;
    }
}