#! /usr/bin/env php
<?php
// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2012 University of California
//
// BOINC is free software; you can redistribute it and/or modify it
// under the terms of the GNU Lesser General Public License
// as published by the Free Software Foundation,
// either version 3 of the License, or (at your option) any later version.
//
// BOINC is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with BOINC.  If not, see <http://www.gnu.org/licenses/>.

// Stage an input file: namely,
// - move or copy it to the download hierarchy
// - compute its md5
// - make a gzipped version if needed
//
// Usage (from project dir):
//
// bin/stage_file [--gzip] [--copy] file
//
// --gzip       Make a gzipped version of the file.
//              Use this if you specify <gzip> in the <file_info>
// --copy       Copy the file (default is to move it)

function error_exit($msg) {
    echo $msg;
    exit(1);
}

if (!file_exists("html/inc/dir_hier.inc")
    || !file_exists("config.xml")
) {
    error_exit("This script must be run in the project directory.\n");
}

require_once("html/inc/dir_hier.inc");
require_once("html/inc/util_basic.inc");

function usage() {
    error_exit("usage: stage_file [--gzip] [--copy] file
    --gzip      make a gzipped version of file for compressed download
                (use with <gzip/> in the input template)
    --copy      copy the file (default is to move it)
");
}

if ($argc < 2) usage();

$gzip = false;
$copy = false;

for ($i=1; $i<$argc-1; $i++) {
    switch($argv[$i]) {
    case "--gzip":
        $gzip = true;
        break;
    case "--copy":
        $copy = true;
        break;
    default:
        usage();
    }
}

$path = $argv[$argc-1];
if (!file_exists($path)) error_exit("no such file: $path\n");
$file = basename($path);

$fanout = parse_config(get_config(), "<uldl_dir_fanout>");
if (!$fanout) error_exit("can't find <uldl_dir_fanout> in config.xml");
$download_dir = parse_config(get_config(), "<download_dir>");
if (!$download_dir) error_exit("can't find <download_dir> in config.xml");

$dl_path = dir_hier_path($file, $download_dir, $fanout);
$dl_md5_path = "$dl_path.md5";

// compute the file's MD5
//
$md5 = md5_file($path);
$file_size = filesize($path);

// if file is already in download dir, make sure it's the same
//
if (file_exists($dl_path)) {
    if (file_exists($dl_md5_path)) {
        $x = file_get_contents($dl_md5_path);
        $x = explode(" ", $x);
        $dl_md5 = $x[0];
    } else {
        $dl_md5 = md5_file($dl_path);
    }
    if ($md5 != $dl_md5) {
        error_exit("There is already a file in your project's download directory
with that name, but different contents.
This is not allowed by BOINC, which requires that files are 'immutable'.
Please use a different file name.
");
    }
} else {
    // move or copy the file
    //
    if ($copy) {
        copy($path, $dl_path);
    } else {
        rename($path, $dl_path);
    }
    @unlink($dl_md5_path);
}

// make MD5 file if needed
//
if (!file_exists($dl_md5_path)) {
    $x = $md5." ".$file_size."\n";
    file_put_contents($dl_md5_path, $x);
}

// make gzipped version if needed
//
if ($gzip) {
    $dl_gzip_path = "$dl_path.gz";
    if (!file_exists($dl_gzip_path)) {
        $output = system("gzip -c $dl_path > $dl_gzip_path", $retval);
        if ($retval) {
            error_exit("failed to gzip file: $output\n");
        }
    }
}

?>
