[GIT-1] Updated reference links to support custom ids

jwage · Jun 8, 2010 · 6e94009 · 6e94009
1 parent 72009d2
commit 6e94009
Showing 1 changed file with 34 additions and 6 deletions.
diff --git a/mongodbadmin.php b/mongodbadmin.php
@@ -43,10 +43,10 @@
  * @param string $document 
  * @return string $preview
  */
-function renderDocumentPreview($document)
+function renderDocumentPreview($mongo, $document)
 {
   $document = prepareMongoDBDocumentForEdit($document);
-  $preview = linkDocumentReferences($document);
+  $preview = linkDocumentReferences($mongo, $document);
   $preview = print_r($preview, true);
   return $preview;
 }
@@ -58,13 +58,26 @@ function renderDocumentPreview($document)
  * @param array $document
  * @return array $document
  */
-function linkDocumentReferences($document)
+function linkDocumentReferences($mongo, $document)
 {
   foreach ($document as $key => $value) {
     if (is_array($value)) {
       if (isset($value['$ref'])) {
+        $collection = $mongo->selectDB($_REQUEST['db'])->selectCollection($value['$ref']);
+        $id = $value['$id'];
+
+        $ref = $collection->findOne(array('_id' => new MongoId($value['$id'])));
+        if (!$ref) {
+          $ref = $collection->findOne(array('_id' => $value['$id']));
+        }
+
         $document[$key]['$ref'] = '<a href="'.$_SERVER['PHP_SELF'].'?db='.$_REQUEST['db'].'&collection='.$value['$ref'].'">'.$document[$key]['$ref'].'</a>';
-        $document[$key]['$id'] = '<a href="'.$_SERVER['PHP_SELF'].'?db='.$_REQUEST['db'].'&collection='.$value['$ref'].'&id='.$value['$id'].'">'.$document[$key]['$id'].'</a>';
+
+        if ($ref['_id'] instanceof MongoId) {
+          $document[$key]['$id'] = '<a href="'.$_SERVER['PHP_SELF'].'?db='.$_REQUEST['db'].'&collection='.$value['$ref'].'&id='.$value['$id'].'">'.$document[$key]['$id'].'</a>';
+        } else {
+          $document[$key]['$id'] = '<a href="'.$_SERVER['PHP_SELF'].'?db='.$_REQUEST['db'].'&collection='.$value['$ref'].'&id='.$value['$id'].'&custom_id=1">'.$document[$key]['$id'].'</a>';
+        }
       } else {
         $document[$key] = linkDocumentReferences($value);
       }
@@ -136,6 +149,21 @@ function prepareMongoDBDocumentForEdit($value)
   return $prepared;
 }
 
+function findMongoDbDocument($id, $custom_id = false)
+{
+    global $mongo;
+
+    $collection = $mongo->selectDB($_REQUEST['db'])->selectCollection($_REQUEST['collection']);
+
+    if (isset($_REQUEST['custom_id'])) {
+        $document =$collection->findOne(array('_id' => $_REQUEST['id']));
+    } else {
+        $document = $collection->findOne(array('_id' => new MongoId($_REQUEST['id'])));
+    }
+
+    return $document;
+}
+
 // Actions
 try {
   // DELETE DB
@@ -523,7 +551,7 @@ function prepareMongoDBDocumentForEdit($value)
     $collection = $mongo->selectDB($_REQUEST['db'])->selectCollection($_REQUEST['collection']);
 
     if (isset($_REQUEST['custom_id'])) {
-      $document =$collection->findOne(array('_id' => $_REQUEST['id']));
+      $document = $collection->findOne(array('_id' => $_REQUEST['id']));
     } else {
       $document = $collection->findOne(array('_id' => new MongoId($_REQUEST['id'])));
     }
@@ -535,7 +563,7 @@ function prepareMongoDBDocumentForEdit($value)
       <input type="hidden" name="<?php echo $k ?>" value="<?php echo $v ?>" />
     <?php endforeach; ?>
 
-    <pre><code><?php echo renderDocumentPreview($document) ?></code></pre>
+    <pre><code><?php echo renderDocumentPreview($mongo, $document) ?></code></pre>
 
     <?php $prepared = prepareMongoDBDocumentForEdit($document) ?>