<?php

function lab_migration_download_lab_pdf()
{
  $lab_id = arg(2);
  _latex_copy_script_file();
  $full_lab = arg(3);
  if ($full_lab == "1")
  	_latex_generate_files($lab_id, TRUE);
  else
  	_latex_generate_files($lab_id, FALSE);
}

/* full lab parameter is for gennerating the entire lab pdf including all unapproved solutions */
function _latex_generate_files($lab_id, $full_lab = FALSE)
{
  $root_path = lab_migration_path();
  $dir_path = $root_path . "latex/";

  $lab_filedata = "";
  $solution_provider_filedata = "";
  $latex_filedata = "";
  $latex_dep_filedata = "";

  $depedency_list = array();

  $eol = "\n";
  $sep = "#";

  $proposal_q = db_query("SELECT * FROM {lab_migration_proposal} WHERE id = %d", $lab_id);
  $proposal_data = db_fetch_object($proposal_q);
  if (!$proposal_data)
  {
    drupal_set_message('Invalid lab specified.', 'error');
    drupal_goto('');
  }
  if ($proposal_data->approval_status == 0)
  {
    drupal_set_message('Lab proposal is still in pending review.', 'error');
    drupal_goto('');
  }

	$category_data = '';
	switch ($proposal_data->category)
	{
	case 0: $category_data = 'Not Selected'; break;
	case 1: $category_data = 'Fluid Mechanics'; break;
	case 2: $category_data = 'Control Theory & Control Systems'; break;
	case 3: $category_data = 'Chemical Engineering'; break;
	case 4: $category_data = 'Thermodynamics'; break;
	case 5: $category_data = 'Mechanical Engineering'; break;
	case 6: $category_data = 'Signal Processing'; break;
	case 7: $category_data = 'Digital Communications'; break;
	case 8: $category_data = 'Electrical Technology'; break;
	case 9: $category_data = 'Mathematics & Pure Science'; break;
	case 10: $category_data = 'Analog Electronics'; break;
	case 11: $category_data = 'Digital Electronics'; break;
	case 12: $category_data = 'Computer Programming'; break;
	case 13: $category_data = 'Others'; break;
	default: $category_data = 'Unknown'; break;
	}

  $lab_filedata = $proposal_data->lab_title . $sep . $proposal_data->name_title . $sep . $proposal_data->name . $sep . $proposal_data->department . $sep . $proposal_data->university . $sep . $category_data . $eol;

  /* regenerate lab if full lab selected */
  if ($full_lab)
  	lab_migration_del_lab_pdf($proposal_data->id);

  /* check if lab already generated */
  if (file_exists($dir_path . "lab_" . $proposal_data->id . ".pdf"))
  {
		/* download zip file */
		header('Content-Type: application/pdf');
		header('Content-disposition: attachment; filename="' . $proposal_data->lab_title . '.pdf"');
		header('Content-Length: ' . filesize($dir_path . "lab_" . $proposal_data->id . ".pdf"));
		readfile($dir_path . "lab_" . $proposal_data->id . ".pdf");
		return;
  }

  $solution_provider_user = user_load($proposal_data->solution_provider_uid);
  if (!$solution_provider_user)
  {
    drupal_set_message('Could not fetch solution provider information for the lab specified.', 'error');
  }
  $solution_provider_filedata .= $proposal_data->solution_provider_name_title . $sep . $proposal_data->solution_provider_name . $sep . $proposal_data->solution_provider_department . $sep . $proposal_data->solution_provider_university . $eol;

  $experiment_q = db_query("SELECT * FROM {lab_migration_experiment} WHERE proposal_id = %d ORDER BY number DESC", $proposal_data->id);
  while ($experiment_data = db_fetch_object($experiment_q))
  {
  	if ($full_lab)
  		$solution_q = db_query("SELECT * FROM {lab_migration_solution} WHERE experiment_id = %d ORDER BY code_number DESC", $experiment_data->id);
  	else
    	$solution_q = db_query("SELECT * FROM {lab_migration_solution} WHERE experiment_id = %d AND approval_status = 1 ORDER BY code_number DESC", $experiment_data->id);
    while ($solution_data = db_fetch_object($solution_q))
    {
      $solution_files_q = db_query("SELECT * FROM {lab_migration_solution_files} WHERE solution_id = %d", $solution_data->id);
      while ($solution_files_data = db_fetch_object($solution_files_q))
      {
        $latex_filedata .= $experiment_data->number . $sep;
        $latex_filedata .= $experiment_data->title . $sep;
        $latex_filedata .= $solution_data->code_number . $sep;
        $latex_filedata .= $solution_data->caption . $sep;
        $latex_filedata .= $solution_files_data->filename . $sep;
        $latex_filedata .= $solution_files_data->filepath . $sep;
        $latex_filedata .= $solution_files_data->filetype . $sep;
        $latex_filedata .= $sep;
        $latex_filedata .= $solution_files_data->id;
        $latex_filedata .= $eol;
      }
      $dependency_files_q = db_query("SELECT * FROM {lab_migration_solution_dependency} WHERE solution_id = %d", $solution_data->id);
      while ($dependency_files_data = db_fetch_object($dependency_files_q))
      {
        $dependency_q = db_query("SELECT * FROM {lab_migration_dependency_files} WHERE id = %d LIMIT 1", $dependency_files_data->dependency_id);
        if ($dependency_data = db_fetch_object($dependency_q))
        {
		if(substr($dependency_data->filename, -3) != "wav") {
		  $latex_filedata .= $experiment_data->number . $sep;
		  $latex_filedata .= $experiment_data->title . $sep;
		  $latex_filedata .= $solution_data->code_number . $sep;
		  $latex_filedata .= $solution_data->caption . $sep;
		  $latex_filedata .= $dependency_data->filename . $sep;
		  $latex_filedata .= $dependency_data->filepath . $sep;
		  $latex_filedata .= 'D' . $sep;
		  $latex_filedata .= $dependency_data->caption . $sep;
		  $latex_filedata .= $dependency_data->id;
		  $latex_filedata .= $eol;

		  $depedency_list[$dependency_data->id] = "D";
		}
        }
      }
    }
  }

  foreach ($depedency_list as $row => $data) {
    $dependency_q = db_query("SELECT * FROM {lab_migration_dependency_files} WHERE id = %d LIMIT 1", $row);
    if ($dependency_data = db_fetch_object($dependency_q))
    {
      $latex_dep_filedata .= $dependency_data->filename . $sep;
      $latex_dep_filedata .= $dependency_data->filepath . $sep;
      $latex_dep_filedata .= $dependency_data->caption . $sep;
      $latex_dep_filedata .= $dependency_data->id;
      $latex_dep_filedata .= $eol;
    }
  }

  /**************************** WRITE TO FILES ********************************/
  $download_filename = $proposal_data->lab;
  $lab_filename = "tmp_" . $proposal_data->id . "_lab.txt";
  $solution_provider_filename = "tmp_" . $proposal_data->id . "_solution_provider.txt";
  $latex_filename = "tmp_" . $proposal_data->id . "_data.txt";
  $latex_dep_filename = "tmp_" . $proposal_data->id . "_dep_data.txt";
  $pdf_filename = "lab_" . $proposal_data->id . ".pdf";

  $fb = fopen($dir_path . $lab_filename, 'w');
  fwrite($fb, $lab_filedata);
  fclose($fb);

  $fc = fopen($dir_path . $solution_provider_filename, 'w');
  fwrite($fc, $solution_provider_filedata);
  fclose($fc);

  $fl = fopen($dir_path . $latex_filename, 'w');
  fwrite($fl, $latex_filedata);
  fclose($fl);

  $fd = fopen($dir_path . $latex_dep_filename, 'w');
  fwrite($fd, $latex_dep_filedata);
  fclose($fd);

  if (_latex_run_script($lab_filename, $solution_provider_filename, $latex_filename, $latex_dep_filename, $pdf_filename))
  {
		/* download zip file */
		header('Content-Type: application/pdf');
		header('Content-disposition: attachment; filename="' . $proposal_data->lab_title . '_' . $proposal_data->name_title . '_' . $proposal_data->name . '.pdf"');
		header('Content-Length: ' . filesize($dir_path . $pdf_filename));
		readfile($dir_path . $pdf_filename);
	} else {
		drupal_set_message("Error occurred when generating the PDF version of the Book.", 'error');
	}

  /*********************** DELETING TEMPORARY FILES ***************************/
  /* regenerate lab if full lab selected */
  if ($full_lab)
  	lab_migration_del_lab_pdf($proposal_data->id);
}

function _latex_copy_script_file()
{
  exec("cp ./" . drupal_get_path('module', 'lab_migration') . "/latex/* " . lab_migration_path() . "latex");
  exec("chmod u+x ./uploads/latex/*.sh");
}

function _latex_run_script($lab_filename, $solution_provider_filename, $latex_filename, $latex_dep_filename, $pdf_filename)
{
  $root_path = lab_migration_path();
  $ret = 0;
  chdir("lab_migration_uploads");
  chdir("latex");

  $sh_command = "./pdf_creator.sh " . $lab_filename . " " . $solution_provider_filename . " " . $latex_filename . " " . $latex_dep_filename;
  exec($sh_command);
  exec("mv TEX_final.pdf " . $pdf_filename);

  if ($ret == 0)
    return TRUE;
  else
    return FALSE;
}

function lab_migration_delete_lab_pdf()
{
	$lab_id = arg(3);
	lab_migration_del_lab_pdf($lab_id);
	drupal_set_message(t('Lab schedule for regeneration.'), 'status');
	drupal_goto('lab_migration/code_approval/bulk');
	return;
}