Enhancing request/accept

1 files changed, 13 insertions, 6 deletions

diff --git a/muamba.db.inc b/muamba.db.inc
index e8e1c70..224132f 100644
--- a/muamba.db.inc
+++ b/muamba.db.inc
@@ -82,17 +82,22 @@ function muamba_get_transactions($uid, $type = 'sent', $status = NULL) {
 /**
  * Get a single transaction.
  *
- * @param $mid
- *   Transaction id.
+ * @param $data
+ *   Transaction id or node object.
  *
  * @return
  *   Transaction data.
  */
-function muamba_get_transaction($mid) {
-  $mid = (int) $mid;
+function muamba_get_transaction($data) {
   $query = db_select('muamba', 'm');
   $query->fields('m', array('mid', 'nid', 'uid', 'owner', 'status', 'thread_id'));
-  $query->condition('m.mid', $mid, '=');
+
+  if (is_object($data)) {
+    $query->condition('m.nid', $data->nid, '=');
+  }
+  else {
+    $query->condition('m.mid', (int) $data, '=');
+  }
 
   $rows    = array();
   $results = $query->execute()->fetchAll();
@@ -102,5 +107,7 @@ function muamba_get_transaction($mid) {
     $rows[] = array_map('check_plain', (array) $entry);
   }
 
-  return $rows[0];
+  if (isset($rows[0])) {
+    return $rows[0];
+  }
 }