Finalize download object request

Benjamin Renard
1 parent 554f1fae
Showing 4 changed files with 145 additions and 83 deletions Show diff stats
src/InputOutput/IHMImpl/DownloadObject/IHMInputOutputDownloadObjectClass.php
src/InputOutput/IHMImpl/IHMInputOutputClass.php
src/Request/DownloadObjectRequestClass.php
src/Request/DownloadObjectRequestDataClass.php
@@ -7,7 +7,30 @@
  */
 class IHMInputOutputDownloadObjectClass implements InputOutputInterface
 {
+    /*
+	 * @brief Get working dir name
+	 */
+	protected function getWorkingDirName()
+	{
+		return $this->requestDirPrefix."DownloadData_";
+	}
+
+	/*
+	 * @brief Get full working dir path
+	*/
+	protected function getWorkingPath()
+	{
+		return IHMConfigClass::getRequestPath().$this->getWorkingDirName().'/';
+	}
+
     public function getInputData($input, $function, $requestId = ""){
+        if (is_dir(IHMConfigClass::getDownloadTmpPath())) {
+			foreach (glob(IHMConfigClass::getDownloadTmpPath().'*.zip') as $filename)  unlink($filename);
+			foreach (glob(IHMConfigClass::getDownloadTmpPath().'*.tar.gz') as $filename)  unlink($filename);
+		}
+        else {
+            mkdir(IHMConfigClass::getDownloadTmpPath());
+        }
         $this->downloadObjectData = new DownloadObjectRequestDataClass();
@@ -15,35 +38,53 @@ class IHMInputOutputDownloadObjectClass implements InputOutputInterface
         {
             // Set type
             $this->downloadObjectData->setType(DownloadObjectTypeEnumClass::ASTROIMG);
-
-            // Set tmp directory
-            $this->downloadObjectData->setTmpDir(IHMConfigClass::getRequestPath());
-
-            // Set compression format
-            if($input->compression == DownloadObjectFormatEnumClass::ZIP)
-                $this->downloadObjectData->setCompressionFormat(DownloadObjectFormatEnumClass::ZIP);
-
+            
+            
              // Set list of links to download   
             if(sizeof($input->list) > 0){
-                $this->downloadObjectData->setList($input->list);
+                foreach ($input->list as $file) {
+                    $this->downloadObjectData->addFile($file->name, $file->url);
+                }
             }
         }
+        else {
+            throw new Exception('Unknown download data type.');
+        }
+
+        $outputFileName = "".$this->downloadObjectData->getType()."_".time();
+
+        // Set working dir
+        $this->downloadObjectData->setWorkingPath($this->getWorkingPath());
+
+        // Set compression format
+        switch ($input->compression) {
+            case DownloadObjectFormatEnumClass::ZIP:
+                $this->downloadObjectData->setCompressionFormat(DownloadObjectFormatEnumClass::ZIP);
+                $extension = ".zip";
+                break;
+            default:
+                throw new Exception('Unknown compression format type.');
+        } 
+        
+
+        // Set output
+        $outputPath = IHMConfigClass::getDownloadTmpPath() . $outputFileName . $extension;
+        $this->downloadObjectData->setOutputPath($outputPath);
+
         return $this->downloadObjectData;
     }
     public function getOutput($data)
 	{
-        $result = NULL;
-        if($data->getResult() != NULL){
-            if($data->getType() == DownloadObjectTypeEnumClass::ASTROIMG && $data->getCompression() == DownloadObjectFormatEnumClass::ZIP){
-
-                $result = array(
-                    "success" => true,
-                    "download" => $data->getResult()
-                );
-            }
-        }
-        return $result;
+        if (!$data->getSuccess())
+            return array(
+                'success' => false,
+                'message' => $data->getLastErrorMessage());
+
+        return array(
+            'success' => true,
+            'download' => str_replace(IHM_SRC_DIR, '', $data->getOutputPath()),
+        );
     }
 }
 \ No newline at end of file
@@ -71,10 +71,10 @@ class IHMInputOutputClass implements InputOutputInterface
 				$this->inputOutput = new IHMInputOutputParamsGeneratorClass();
 				$requestId = "ParamGen";
 				break;
-                        case FunctionTypeEnumClass::PARAMSINFOGEN :
-                                $this->inputOutput = new IHMInputOutputParamsInfoGeneratorClass();
-                                $requestId = "ParamInfoGen";
-                                break;
+            case FunctionTypeEnumClass::PARAMSINFOGEN :
+                $this->inputOutput = new IHMInputOutputParamsInfoGeneratorClass();
+                $requestId = "ParamInfoGen";
+                break;
 			case FunctionTypeEnumClass::ACTION :
 				if (!$input->action->multiplot) {
 					$this->inputOutput = new IHMInputOutputParamsPlotClass();
@@ -14,38 +14,48 @@ class DownloadObjectRequestClass extends RequestAbstractClass
     public function init(){ 
         if (!isset($this->requestData))
 			return false;
+        if (!is_dir($this->requestData->getWorkingPath())) {
+            mkdir($this->requestData->getWorkingPath());
+        }
         return true;
     }
     public function run(){ 
-
-        // Create a temporary directory to store the files
-
-        $tempDir = $this->requestData->getTmpDir().'/curl_files_'. uniqid();
-        mkdir($tempDir);
-
         // Array to hold the paths of the downloaded files
         $filePaths = array();
-        $fileList = $this->requestData->getList();
-        foreach($fileList as $fileData){
+        $files = $this->requestData->getFiles();
+        foreach($files as $fileName => $fileUrl){
+            $result = $this->downloadFile($fileName, $fileUrl, $this->requestData->getWorkingPath());
+            if (!$result['success']) {
+                $this->cleanup($filePaths);
+                $this->requestData->setLastErrorMessage($result['message']);
+                $this->requestData->setSuccess(false);
+                return false;
+            }
+            $filePaths[] = $result['filePath'];
+        }
+
-            $filePaths[] = $this->curlFile($fileData, $tempDir);
+        $result = $this->zipFiles($filePaths, $this->requestData->getOutputPath());
+        if (!$result['success']) {
+            $this->cleanup($filePaths);
+            $this->requestData->setLastErrorMessage($result['message']);
+            $this->requestData->setSuccess(false);
+
+            return false;
         }
-        $zipFilePath = $filePaths[0]; // $this->zipFiles($filePaths, $tempDir); // fonctionne pas pour le moment.
-        $basePath = '/home/amda_admin/AMDA/AMDA_IHM/';
+        $this->cleanup($filePaths);
-        $zipFilePath = str_replace($basePath, '', $zipFilePath);
-        $this->requestData->setResult($zipFilePath);
         $this->requestData->setSuccess(true);
-        return  $this->requestData->getSuccess();
+        return  true;
     }
-    public function curlFile($fileData, $tempDir){
+    public function downloadFile($fileName, $fileUrl, $workingDir){
          // Initialize cURL session
-        $ch = curl_init($fileData->url);
+        $ch = curl_init($fileUrl);
         // Set cURL options
         curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
         curl_setopt($ch, CURLOPT_FOLLOWLOCATION, true);
@@ -55,47 +65,58 @@ class DownloadObjectRequestClass extends RequestAbstractClass
          // Check for cURL errors
          if (curl_errno($ch)) {
-            echo 'cURL error: ' . curl_error($ch);
             curl_close($ch);
-            return false;
+            return array(
+                "success" => FALSE,
+                "message" => "Cannot download $fileUrl",
+            );
         }
-         // Close cURL session
-         curl_close($ch);
-        // Define the file path
-        $filePath = $tempDir . '/' . basename($fileData->name);
+        // Close cURL session
+        curl_close($ch);
+        
+        // Write file in working dir
+        $filePath = $workingDir . '/' . $fileName;
         file_put_contents($filePath, $content);
-        // Store the file path
-        return $filePath;
+        return array(
+            "success" => TRUE,
+            "filePath" => $filePath,
+        );
     }
-    public function zipFiles($filePaths, $tempDir){
+    public function zipFiles($filePaths, $outputPath){
         // Create a zip file
+        $zip = new ZipArchive();
-    $zipFilePath = IHMConfigClass::getRequestPath().'/curl_files_'.uniqid() . '.zip';
-    $zip = new ZipArchive();
-    
-    if ($zip->open($zipFilePath, ZipArchive::CREATE) !== TRUE) {
-        exit("Cannot open <$zipFilePath>\n");
-    }
+        if ($zip->open($outputPath, ZipArchive::CREATE) !== TRUE) {
+            return array(
+                "success" => FALSE,
+                "message" => "Cannot create output archive",
+            );
+        }
-    // Add files to the zip archive
-    foreach ($filePaths as $filePath) {
-        $zip->addFile($filePath, basename($filePath));
-    }
+        // Add files to the zip archive
+        foreach ($filePaths as $filePath) {
+            $zip->addFile($filePath, basename($filePath));
+        }
-    // Close the zip archive
-    $zip->close();
+        // Close the zip archive
+        $zip->close();
-    // Clean up the temporary files
-    foreach ($filePaths as $filePath) {
-        unlink($filePath);
+        // Return the path of the zip file
+        return array(
+            "success" => TRUE,
+            "zipfilepath" => $outputPath,
+        );
     }
-    rmdir($tempDir);
-    // Return the path of the zip file
-    return $zipFilePath;
+    public function cleanup($filePaths) {
+        // Clean up the temporary files
+        foreach ($filePaths as $filePath) {
+            if (file_exists($filePath))
+                unlink($filePath);
+        }
     }
 }
 \ No newline at end of file
@@ -30,11 +30,11 @@ abstract class DownloadObjectFormatEnumClass
  */
 class DownloadObjectRequestDataClass extends RequestDataClass
 {
-    private $type    = DownloadObjectTypeEnumClass::UNKNOWN;
-    private $list = NULL;
+    private $type        = DownloadObjectTypeEnumClass::UNKNOWN;
+    private $files       = array();
     private $compression = DownloadObjectFormatEnumClass::ZIP;
-	private $tmpDir = "";
-	private $result = NULL;
+	private $workingPath = "";
+	private $outputPath  = "";
     public function getType()
@@ -47,34 +47,34 @@ class DownloadObjectRequestDataClass extends RequestDataClass
 		$this->type = $type;
 	}
-    public function getList()
+    public function getFiles()
 	{
-		return $this->list;
+		return $this->files;
 	}
-	public function setList($list)
+	public function addFile($name, $url)
 	{
-		$this->list = $list;
+		$this->files[$name] = $url;
 	}
-	public function getTmpDir()
+	public function getWorkingPath()
 	{
-		return $this->tmpDir;
+		return $this->workingPath;
 	}
-	public function setTmpDir($tmpDir)
+	public function setWorkingPath($workingPath)
 	{
-		$this->tmpDir = $tmpDir;
+		$this->workingPath = $workingPath;
 	}
-	public function getResult()
+	public function getOutputPath()
 	{
-		return $this->result;
+		return $this->outputPath;
 	}
-	public function setResult($result)
+	public function setOutputPath($outputPath)
 	{
-		$this->result = $result;
+		$this->outputPath = $outputPath;
 	}
     public function getCompressionFormat()