Allow save to module

- No breaking changes - Upload file to module
Andrewsuares · Jan 21, 2023 · a88ef1d · a88ef1d
1 parent de11061
commit a88ef1d
Showing 1 changed file with 26 additions and 18 deletions.
diff --git a/engine/tg_helpers/file_helper.php b/engine/tg_helpers/file_helper.php
@@ -2,7 +2,12 @@
 class File_helper {
 
     public function upload($config) {
-        extract($config);
+
+        //declare all inbound variables
+        $destination = $config['destination'] ?? NULL;
+        $target_module = $config['target_module'] ?? segment(1);
+        $upload_to_module = $config['upload_to_module'] ?? false;
+        $make_rand_name = $config['make_rand_name'] ?? false;
 
         if (!isset($destination)) {
             die('ERROR: upload requires inclusion of \'destination\' property.  Check documentation for details.');
@@ -11,18 +16,18 @@ public function upload($config) {
         $userfile = array_keys($_FILES)[0];
         $target_file = $_FILES[$userfile];
 
-        if(!isset($make_rand_name)) {
+        if (!isset($make_rand_name)) {
             $make_rand_name = false;
         }
 
         //init $new_file_name variable (the name of the uploaded file)
-        if($make_rand_name == true) {
+        if ($make_rand_name == true) {
             $file_name_without_extension = strtolower(make_rand_str(10));
 
             //add file extension onto rand file name
             $file_info = return_file_info($target_file['name']);
             $file_extension = $file_info['file_extension'];
-            $new_file_name = $file_name_without_extension.$file_extension;
+            $new_file_name = $file_name_without_extension . $file_extension;
         } else {
             //get the file name and extension
             $file_info = return_file_info($target_file['name']);
@@ -32,47 +37,50 @@ public function upload($config) {
             //remove dangerous characters from the file name
             $file_name = url_title($file_name);
             $file_name_without_extension = str_replace('-', '_', $file_name);
-            $new_file_name = $file_name_without_extension.$file_extension;
+            $new_file_name = $file_name_without_extension . $file_extension;
         }
 
-        $target_destination = '../public/'.$destination;
+        //set the target destination directory
+        if ($upload_to_module == true) {
+            $target_destination = '../modules/' . $target_module . '/assets/' . $destination;
+        } else {
+            //add code here to deal with external URLs (AWS, Google Drive, OneDrive, etc...)
+            $target_destination = $destination;
+        }
 
         try {
             //make sure the destination folder exists
             if (!is_dir($target_destination)) {
                 $error_msg = 'Invalid directory';
-                if (strlen($target_destination)>0) {
-                    $error_msg.= ': \''.$target_destination.'\' (string '.strlen($target_destination).')';
+                if (strlen($target_destination) > 0) {
+                    $error_msg .= ': \'' . $target_destination . '\' (string ' . strlen($target_destination) . ')';
                 }
                 throw new Exception($error_msg);
             }
 
             //upload the temp file to the destination
-            $new_file_path = $target_destination.'/'.$new_file_name;
+            $new_file_path = $target_destination . '/' . $new_file_name;
 
             $i = 2;
-            while(file_exists($new_file_path)) {
-                $new_file_name = $file_name_without_extension.'_'.$i.$file_extension;
-                $new_file_path = $target_destination.'/'.$new_file_name;
+            while (file_exists($new_file_path)) {
+                $new_file_name = $file_name_without_extension . '_' . $i . $file_extension;
+                $new_file_path = $target_destination . '/' . $new_file_name;
                 $i++;
             }
 
-            move_uploaded_file($target_file['tmp_name'], $new_file_path);            
-        
+            move_uploaded_file($target_file['tmp_name'], $new_file_path);
+
             //create an array to store file information
             $file_info = array();
             $file_info['file_name'] = $new_file_name;
             $file_info['file_path'] = $new_file_path;
             $file_info['file_type'] = $target_file['type'];
             $file_info['file_size'] = $target_file['size'];
             return $file_info;
-
         } catch (Exception $e) {
             // Code to handle the exception
             echo "An exception occurred: " . $e->getMessage();
             die();
         }
-
     }
-
-}
+}