/*
 * Guillaume Cottenceau (gc@mandrakesoft.com)
 *
 * Copyright 2000 MandrakeSoft
 *
 * This software may be freely redistributed under the terms of the GNU
 * public license.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */

/*
 * Portions from Erik Troan (ewt@redhat.com)
 *
 * Copyright 1996 Red Hat Software 
 *
 */

#include <stdlib.h>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <sys/mount.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>
#include "stage1.h"
#include "frontend.h"
#include "modules.h"
#include "probing.h"
#include "log.h"
#include "mount.h"
#include "lomount.h"
#include "automatic.h"
#include "disk.h"
#include "partition.h"

static char * disk_extract_list_directory(char * direct)
{
	char ** full = list_directory(direct);
	char tmp[2000] = "";
	int i;
	for (i=0; i<5 ; i++) {
		if (!full || !*full)
			break;
		strcat(tmp, *full);
		strcat(tmp, "\n");
		full++;
	}
	return strdup(tmp);
}

static enum return_type try_with_device(char *dev_name)
{
	char * questions_location[] = { "Directory or ISO image", NULL };
	char * questions_location_auto[] = { "directory", NULL };
	static char ** answers_location = NULL;
	char location_full[500];

#ifndef MANDRAKE_MOVE
	char * disk_own_mount = "/tmp/hdimage";
#else
	char * disk_own_mount = SLASH_LOCATION "/tmp/hdimage";
#endif
        char * loopdev = NULL;

	char * parts[50];
	char * parts_comments[50];
	struct stat statbuf;
	enum return_type results;
	char * choice;
        
        if (list_partitions(dev_name, parts, parts_comments)) {
		stg1_error_message("Could not read partitions information.");
		return RETURN_ERROR;
        }

        /* uglyness to allow auto starting with devfs */
        if (!IS_AUTOMATIC || streq((choice = get_auto_value("partition")), "")) {
                if (parts[0] == NULL) {
                        stg1_error_message("No partitions found.");
                        return RETURN_ERROR;
                }

                results = ask_from_list_comments_auto("Please choose the partition where " DISTRIB_NAME " is copied.",
                                                      parts, parts_comments, &choice, "partition", parts);
                if (results != RETURN_OK)
                        return results;
        }
#ifdef MANDRAKE_MOVE
	mkdir (SLASH_LOCATION "/tmp", 0755);
#endif
	
        if (try_mount(choice, disk_own_mount)) {
		stg1_error_message("I can't find a valid filesystem (tried: ext2, vfat, reiserfs).");
		return try_with_device(dev_name);
	}

	if (ask_from_entries_auto("Please enter the directory (or ISO image file) containing the " DISTRIB_NAME " Distribution.",
				  questions_location, &answers_location, 24, questions_location_auto, NULL) != RETURN_OK) {
		umount(disk_own_mount);
		return try_with_device(dev_name);
	}

	strcpy(location_full, disk_own_mount);
	strcat(location_full, "/");
	strcat(location_full, answers_location[0]);

	if (access(location_full, R_OK)) {
		stg1_error_message("Directory or ISO image file could not be found on partition.\n"
			      "Here's a short extract of the files in the root of the partition:\n"
			      "%s", disk_extract_list_directory(disk_own_mount));
		umount(disk_own_mount);
		return try_with_device(dev_name);
	}

	unlink(IMAGE_LOCATION);

	if (!stat(location_full, &statbuf) && !S_ISDIR(statbuf.st_mode)) {
		log_message("%s exists and is not a directory, assuming this is an ISO image", location_full);
		if (lomount(location_full, IMAGE_LOCATION, &loopdev, 0)) {
			stg1_error_message("Could not mount file %s as an ISO image of the " DISTRIB_NAME " Distribution.", answers_location[0]);
			umount(disk_own_mount);
			return try_with_device(dev_name);
		}
	} else
		symlink(location_full, IMAGE_LOCATION);

#ifndef MANDRAKE_MOVE
	if (IS_SPECIAL_STAGE2 || ramdisk_possible()) {
		/* RAMDISK install */
		if (access(IMAGE_LOCATION RAMDISK_LOCATION, R_OK)) {
			stg1_error_message("I can't find the " DISTRIB_NAME " Distribution in the specified directory. "
				      "(I need the subdirectory " RAMDISK_LOCATION ")\n"
				      "Here's a short extract of the files in the directory:\n"
				      "%s", disk_extract_list_directory(IMAGE_LOCATION));
			del_loop(loopdev);
			umount(disk_own_mount);
			return try_with_device(dev_name);
		}
		if (load_ramdisk() != RETURN_OK) {
			stg1_error_message("Could not load program into memory.");
			del_loop(loopdev);
			umount(disk_own_mount);
			return try_with_device(dev_name);
		}
	} else {
#endif
		/* LIVE install */
#ifdef MANDRAKE_MOVE
		if (access(IMAGE_LOCATION "/live_tree.clp", R_OK)) {
#else
		if (access(IMAGE_LOCATION LIVE_LOCATION, R_OK)) {
#endif
			stg1_error_message("I can't find the " DISTRIB_NAME " Distribution in the specified directory. "
				      "(I need the subdirectory " LIVE_LOCATION ")\n"
				      "Here's a short extract of the files in the directory:\n"
				      "%s", disk_extract_list_directory(IMAGE_LOCATION));
			del_loop(loopdev);
			umount(disk_own_mount);
			return try_with_device(dev_name);
		}
#ifndef MANDRAKE_MOVE
		char p;
		if (readlink(IMAGE_LOCATION LIVE_LOCATION "/usr/bin/runinstall2", &p, 1) != 1) {
			stg1_error_message("The " DISTRIB_NAME " Distribution seems to be copied on a Windows partition. "
				      "You need more memory to perform an installation from a Windows partition. "
				      "Another solution if to copy the " DISTRIB_NAME " Distribution on a Linux partition.");
			del_loop(loopdev);
			umount(disk_own_mount);
			return try_with_device(dev_name);
		}
		log_message("found the " DISTRIB_NAME " Installation, good news!");
	}
#endif

	if (IS_RESCUE) {
                del_loop(loopdev);
		umount(disk_own_mount);
	}

	method_name = strdup("disk");
	return RETURN_OK;
}

enum return_type disk_prepare(void)
{
	char ** medias, ** medias_models;
	char * choice;
	int i;
	enum return_type results;

        int count = get_disks(&medias, &medias_models);

	if (count == 0) {
		stg1_error_message("No DISK drive found.");
		i = ask_insmod(SCSI_ADAPTERS);
		if (i == RETURN_BACK)
			return RETURN_BACK;
		return disk_prepare();
	}

	if (count == 1) {
		results = try_with_device(*medias);
		if (results == RETURN_OK)
			return RETURN_OK;
		i = ask_insmod(SCSI_ADAPTERS);
		if (i == RETURN_BACK)
			return RETURN_BACK;
		return disk_prepare();
	}

	results = ask_from_list_comments_auto("Please choose the DISK drive on which you copied the " DISTRIB_NAME " Distribution.",
					      medias, medias_models, &choice, "disk", medias);

	if (results != RETURN_OK)
		return results;

	results = try_with_device(choice);
	if (results == RETURN_OK)
		return RETURN_OK;
	i = ask_insmod(SCSI_ADAPTERS);
	if (i == RETURN_BACK)
		return RETURN_BACK;
	return disk_prepare();
}

int
process_recovery(void)
{
	char ** medias, ** medias_models;
        int count, i;

        log_message("trying the automatic recovery of oem installs");

        count = get_disks(&medias, &medias_models);

        for (i=0; i<count; i++) {
                char * parts[50];
                char * parts_comments[50];
                char ** part, ** comment;
                log_message("examining disk %s (%s)", medias[i], medias_models[i]);

                if (list_partitions(medias[i], parts, parts_comments)) {
                        log_message("could not read partitions information, bailing out");
                        return 0;
                }

                part = parts;
                comment = parts_comments;
                while (part && *part) {
                        char * disk_own_mount = "/tmp/hdimage";
                        log_message("examining partition %s (%s)", *part, *comment);
                        if (try_mount(*part, disk_own_mount))
                                log_message("couldn't mount it");
                        else {
                                FILE *f;
                                char buf[500];
                                char location[500];
                                strcpy(location, disk_own_mount);
                                strcat(location, "/VERSION");
                                if (!(f = fopen(location, "rb")) || !fgets(buf, sizeof(buf), f)) {
                                        log_perror("could not fopen or fgets VERSION");
                                        goto examine_next_part;
                                }
                                fclose(f);
                                if (!strstr(buf, VERSION)) {
                                        log_message("mismatching VERSION contents");
                                        goto examine_next_part;
                                }
                                strcpy(location, disk_own_mount);
                                strcat(location, "/Mandrake/base");
                                if (access(location, R_OK)) {
                                        log_message("Mandrake/base is not here");
                                        goto examine_next_part;
                                }

                                log_message("going on with a recovery on disk %s partition %s", medias[i], *part);

                                symlink(disk_own_mount, IMAGE_LOCATION);
                                if (ramdisk_possible())
                                        load_ramdisk(); /* if load of ramdisk failed, try to continue in live */
                                
                                method_name = strdup("disk");
                                return 1;
                        }

                examine_next_part:
                        umount(disk_own_mount);
                        part++;
                        comment++;
                }
        }

        return 0;
}