1 files changed, 10 insertions, 3 deletions
diff --git a/muamba.db.inc b/muamba.db.inc
index ff682f5..773e65e 100644
--- a/muamba.db.inc
+++ b/muamba.db.inc
@@ -82,16 +82,23 @@ function muamba_get_transactions($uid, $type = 'sent', $status = NULL) {
 /**
  * Get a single transaction.
  *
- * @param $mid
+ * @param $id
  *   Transaction id.
  *
+ * @param $index
+ *   Index (mid or thread_id).
+ *
  * @return
  *   Transaction data.
  */
-function muamba_get_transaction($mid) {
+function muamba_get_transaction($id, $index = 'mid') {
+  if ($index != 'mid' && $index != 'thread_id') {
+    return;
+  }
+
   $query = db_select('muamba', 'm');
   $query->fields('m', array('mid', 'nid', 'uid', 'owner', 'status', 'thread_id'));
-  $query->condition('m.mid', (int) $mid, '=');
+  $query->condition('m.'. $index, (int) $id, '=');
 
   $rows    = array();
   $results = $query->execute()->fetchAll();