/*
 * Copyright (c) 2004-2010 The Trustees of Indiana University.
 *                         All rights reserved.
 *
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 */

#include "opal_config.h"

#include <string.h>
#include <sys/wait.h>
#if HAVE_SYS_TYPES_H
#include <sys/types.h>
#endif
#if HAVE_UNISTD_H
#include <unistd.h>
#endif
#ifdef HAVE_FCNTL_H
#include <fcntl.h>
#endif  /* HAVE_FCNTL_H */
#ifdef HAVE_SYS_STAT_H
#include <sys/stat.h>
#endif

#include "opal/mca/mca.h"
#include "opal/mca/base/base.h"
#include "opal/include/opal/constants.h"
#include "opal/util/os_dirpath.h"
#include "opal/util/output.h"
#include "opal/util/argv.h"

#include "opal/mca/compress/compress.h"
#include "opal/mca/compress/base/base.h"

/******************
 * Local Function Defs
 ******************/

/******************
 * Object stuff
 ******************/

int opal_compress_base_tar_create(char ** target)
{
    int exit_status = OPAL_SUCCESS;
    char *cmd = NULL;
    char *tar_target = NULL;
    char **argv = NULL;
    pid_t child_pid = 0;
    int status = 0;

    asprintf(&tar_target, "%s.tar", *target);

    child_pid = fork();
    if( 0 == child_pid ) { /* Child */
        asprintf(&cmd, "tar -cf %s %s", tar_target, *target);

        argv = opal_argv_split(cmd, ' ');
        status = execvp(argv[0], argv);

        opal_output(0, "compress:base: Tar:: Failed to exec child [%s] status = %d\n", cmd, status);
        exit(OPAL_ERROR);
    }
    else if(0 < child_pid) {
        waitpid(child_pid, &status, 0);

        if( !WIFEXITED(status) ) {
            exit_status = OPAL_ERROR;
            goto cleanup;
        }

        free(*target);
        *target = strdup(tar_target);
    }
    else {
        exit_status = OPAL_ERROR;
        goto cleanup;
    }

 cleanup:
    if( NULL != cmd ) {
        free(cmd);
        cmd = NULL;
    }
    if( NULL != tar_target ) {
        free(tar_target);
        tar_target = NULL;
    }

    return exit_status;
}

int opal_compress_base_tar_extract(char ** target)
{
    int exit_status = OPAL_SUCCESS;
    char *cmd = NULL;
    char **argv = NULL;
    pid_t child_pid = 0;
    int status = 0;

    child_pid = fork();
    if( 0 == child_pid ) { /* Child */
        asprintf(&cmd, "tar -xf %s", *target);

        argv = opal_argv_split(cmd, ' ');
        status = execvp(argv[0], argv);

        opal_output(0, "compress:base: Tar:: Failed to exec child [%s] status = %d\n", cmd, status);
        exit(OPAL_ERROR);
    }
    else if(0 < child_pid) {
        waitpid(child_pid, &status, 0);

        if( !WIFEXITED(status) ) {
            exit_status = OPAL_ERROR;
            goto cleanup;
        }

        /* Strip off the '.tar' */
        (*target)[strlen(*target)-4] = '\0';
    }
    else {
        exit_status = OPAL_ERROR;
        goto cleanup;
    }

 cleanup:
    if( NULL != cmd ) {
        free(cmd);
        cmd = NULL;
    }

    return exit_status;
}

/******************
 * Local Functions
 ******************/