Fix driver display in order form.

[readifood.git] / lib / order.php

diff --git a/lib/order.php b/lib/order.php
index 12a34aa..a392f6e 100644 (file)
--- a/lib/order.php
+++ b/lib/order.php
@@ -235,7 +235,10 @@
     if ($order) {
       $q = new OrderStateQuery;
       $order_state = $q->filterByOrderId($order->getId())->orderByUpdated('desc')->limit(1)->findOne();
-      if ($order_state) $state = $order_state->getState();
+      if ($order_state) {
+        $state = $order_state->getState();
+        $driver_id = $order_state->getDriverId();
+      }
     }
     else $order = new Order;
 
@@ -501,6 +504,7 @@
 
       $driver_id = $order_state->getDriverId();
       if ($driver_id) $driver = get_contact_by_id($driver_id);
+      else $driver = null;
 
       /* XXX */
       $state = $order_state->getState();