source: release-kits/lirk2/wrapper/mywrapper.c@ 15980

#include <iostream>
#include <fstream>
#include <iomanip>

using namespace std;

#include "mywrapper.h"

bool extract_bundled_file( const char[], int, char*, bool );

int main(int argc, char** argv) {

    string tempdir = (string)argv[0] + ".tmp"; //temporary directory where we will store extracted files
    string jarfile = tempdir + "/greenstone2.jar"; //where we will store the jar file
    string search4jfile = tempdir + "/search4j"; //where we will store the jar file
    string javafile = tempdir + "/@java.installer@"; //where we will store the java tar file
    bool succeeded = false;

    //create the temp folder
    cout << "Creating temp directory..." << endl;
    succeeded = ( 0 == system( ("mkdir " + tempdir).c_str() ) );
    if ( !succeeded ) {
        cout << "Failed to write to create the temp directory '" << tempdir << "'" << endl;
        cout << "Check that it does not already exist and try again" << endl;
        cout << "Exiting" << endl;
        return 1;
    }

    //extract files
    cout << "Extracting installer jar..." << endl;
    succeeded = extract_bundled_file( greenstone2jar, sizeof(greenstone2jar), (char*)jarfile.c_str(), false);

    cout << "Extracting search4j tool..." << endl;
    succeeded = extract_bundled_file( search4j, sizeof(search4j), (char*)search4jfile.c_str(), true ) && succeeded;

    #ifdef java_is_bundled
    cout << "Extracting bundled java..." << endl;
    succeeded = extract_bundled_file( java, sizeof(java), (char*)javafile.c_str(), true ) && succeeded;
    #endif
        
    if ( !succeeded ) {
        cout << "Failed to extract one or more resources" << endl;
        cout << "This installer may not have sufficient file permissions to write it to disk" << endl;
        cout << "Or, the files may be corrupt or missing from this executable" << endl;
        cout << "Exiting" << endl;
        return 1;
    }

    //change to the temp directory
    chdir( tempdir.c_str() );

    #ifdef java_is_bundled
    succeeded = (system( "tar -xzf @java.installer@" ) == 0);
    if ( !succeeded ) {
        cout << "Failed to extract the bundled java archive to the temp directory" << endl;
        cout << "You need the tar program on your PATH" << endl;
        cout << "Exiting" << endl;
        return 1;
    }
    #endif

    //check if an appropriate java is found
    bool jvmFound = (system( "./search4j -m @java.min.version@ -h ./@java.extracted@" ) == 0);

    //if the jvm was not found, report not found
    if ( !jvmFound ) {

        //did not find a good java
        cout << "Greenstone requires java @java.min.version@ or greater." << endl;
        
        //tell them if java is absent or just too old
        if ( system( "./search4j -h ./@java.extracted@" ) == 0 ) {
            cout << "You have java, but it is too old." << endl;
        } else {
            cout << "Could not find java on your system." << endl;
        }

        cout << "Install java (@java.min.version@ or greater) and set JAVA_HOME or JRE_HOME, and try again" << endl;
        #ifndef java_is_bundled
        cout << "Or, download a greentsone3 installer with bundled java and use that instead of this one" << endl;
        #endif

    //if we have found it, launch the installer
    } else {

        cout << "Launching Installer ..." << endl;
        int launch_exit_code = 0;
        launch_exit_code = system("./search4j -m @java.min.version@ -l greenstone2.jar -h ./@java.extracted@");

        //report how it went
        if ( launch_exit_code == 0 ) {
            cout << "Setup complete" << endl;
        } else {
            cout << "The installer exited with an error" << endl;
            cout << "Greenstone may not be correctly installed" << endl;
        }

    }

    //change back to the original directory
    chdir("..");

    //delete the temp files
    cout << "Deleting the temp directory" << endl;
    system( ("rm -rf " + tempdir).c_str() );

    return 0;

}

bool extract_bundled_file( const char data[], int size, char* filename, bool make_executable) {

    if ( size == 0 ) return false;

    //delete the file if it exists
    remove( filename );

    //open the file
    fstream binary_file( filename, ios::out|ios::binary );
    if ( !binary_file.good() ) {
        binary_file.close();
        return false;
    }

    //write the file
    binary_file.write( data, size );
    if ( !binary_file.good() ) {
        binary_file.close();
        return false;
    }

    //close the file
    binary_file.close();
    if ( !binary_file.good() ) {
        binary_file.close();
        return false;
    }

    if ( make_executable ) {
        string command = "chmod a+x " + string(filename);
        system( command.c_str() );
    }

    return true;

}