Improved export functionality (erikdubbelboer#120)

Added support for filter and transform parameters while exporting bulk databases

Author: fratuz610

export.php

@@ -3,26 +3,30 @@
 require_once 'includes/common.inc.php';
 
 
-
-
 // Export to redis-cli commands
-function export_redis($key) {
+function export_redis($key, $filter = false, $transform = false) {
+
   global $redis;
 
   $type = $redis->type($key);
 
-
+  // we rename the keys as necessary
+  if($filter !== false && $transform !== false)
+    $outputKey = str_replace($filter, $transform, $key);
+  else
+    $outputKey = $key;
+  
   // String
   if ($type == 'string') {
-    echo 'SET "',addslashes($key),'" "',addslashes($redis->get($key)),'"',PHP_EOL;
+    echo 'SET "',addslashes($outputKey),'" "',addslashes($redis->get($key)),'"',PHP_EOL;
   }
 
   // Hash
   else if ($type == 'hash') {
     $values = $redis->hGetAll($key);
 
     foreach ($values as $k => $v) {
-      echo 'HSET "',addslashes($key),'" "',addslashes($k),'" "',addslashes($v),'"',PHP_EOL;
+      echo 'HSET "',addslashes($outputKey),'" "',addslashes($k),'" "',addslashes($v),'"',PHP_EOL;
     }
   }
 
@@ -31,7 +35,7 @@ function export_redis($key) {
     $size = $redis->lLen($key);
 
     for ($i = 0; $i < $size; ++$i) {
-      echo 'RPUSH "',addslashes($key),'" "',addslashes($redis->lIndex($key, $i)),'"',PHP_EOL;
+      echo 'RPUSH "',addslashes($outputKey),'" "',addslashes($redis->lIndex($key, $i)),'"',PHP_EOL;
     }
   }
 
@@ -40,7 +44,7 @@ function export_redis($key) {
     $values = $redis->sMembers($key);
 
     foreach ($values as $v) {
-      echo 'SADD "',addslashes($key),'" "',addslashes($v),'"',PHP_EOL;
+      echo 'SADD "',addslashes($outputKey),'" "',addslashes($v),'"',PHP_EOL;
     }
   }
 
@@ -51,7 +55,7 @@ function export_redis($key) {
     foreach ($values as $v) {
       $s = $redis->zScore($key, $v);
 
-      echo 'ZADD "',addslashes($key),'" ',$s,' "',addslashes($v),'"',PHP_EOL;
+      echo 'ZADD "',addslashes($outputKey),'" ',$s,' "',addslashes($v),'"',PHP_EOL;
     }
   }
 }
@@ -64,7 +68,6 @@ function export_json($key) {
 
   $type = $redis->type($key);
 
-
   // String
   if ($type == 'string') {
     $value = $redis->get($key);
@@ -116,9 +119,12 @@ function export_json($key) {
   header('Content-type: '.$ct.'; charset=utf-8');
   header('Content-Disposition: inline; filename="export.'.$ext.'"');
 
+  $filter = !empty($_POST['filter']) ? trim($_POST['filter']) : false;
+  $transform = !empty($_POST['transform']) ? trim($_POST['transform']) : false;
 
   // JSON
   if ($_POST['type'] == 'json') {
+    
     // Single key
     if (isset($_GET['key'])) {
       echo json_encode(export_json($_GET['key']));
@@ -127,7 +133,18 @@ function export_json($key) {
       $vals = array();
 
       foreach ($keys as $key) {
-        $vals[$key] = export_json($key);
+
+        // if we have a filter and no match, nothing to do
+        if($filter !== false && stripos($key, $filter) === false)
+          continue;
+        
+        // we rename the keys as necessary
+        if($filter !== false && $transform !== false)
+          $outputKey = str_replace($filter, $transform, $key);
+        else
+          $outputKey = $key;
+        
+        $vals[$outputKey] = export_json($key);
       }
 
       echo json_encode($vals);
@@ -136,14 +153,20 @@ function export_json($key) {
 
   // Redis Commands
   else {
+
     // Single key
     if (isset($_GET['key'])) {
       export_redis($_GET['key']);
     } else { // All keys
       $keys = $redis->keys('*');
 
       foreach ($keys as $key) {
-        export_redis($key);
+
+        // if we have a filter and no match, we skip
+        if($filter !== false && stripos($key, $filter) === false)
+          continue;
+        
+        export_redis($key, $filter, $transform);
       }
     }
   }
@@ -171,7 +194,19 @@ function export_json($key) {
 <option value="redis" <?php echo (isset($_GET['type']) && ($_GET['type'] == 'redis')) ? 'selected="selected"' : ''?>>Redis</option>
 <option value="json"  <?php echo (isset($_GET['type']) && ($_GET['type'] == 'json' )) ? 'selected="selected"' : ''?>>JSON</option>
 </select>
-</p>
+</p> 
+
+<?php if (!isset($_GET['key'])): ?>
+  <p>
+  <label for="filter">Filter:</label>
+  <input type="text" name="filter" />
+  </p>
+
+  <p>
+  <label for="transform">Tranform:</label>
+  <input type="text" name="transform" />
+  </p>
+<?php endif; ?>
 
 <p>
 <input type="submit" class="button" value="Export">