From 790c52a896487802d204135b9c0fb5f9cd43f9ea Mon Sep 17 00:00:00 2001
From: =?utf8?q?Rapha=C3=ABl=20Barrois?= <raphael.barrois@polytechnique.org>
Date: Mon, 22 Mar 2010 11:30:22 +0100
Subject: [PATCH] Fix ProfileAddresses filtering
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

Signed-off-by: Raphaël Barrois <raphael.barrois@polytechnique.org>
---
 include/profilefields.inc.php | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/include/profilefields.inc.php b/include/profilefields.inc.php
index 0bb11ba..dda573d 100644
--- a/include/profilefields.inc.php
+++ b/include/profilefields.inc.php
@@ -249,6 +249,8 @@ class Address
     public $administrativeArea;
     public $country;
 
+    public $comment;
+
     private $phones = array();
 
     /** Fields are:
@@ -259,6 +261,7 @@ class Address
         foreach ($data as $key => $val) {
             $this->$key = $val;
         }
+        $this->flags = new PlFlagSet($this->flags);
     }
 
     public function addPhone(Phone &$phone)
@@ -273,9 +276,9 @@ class Address
         return $this->phones;
     }
 
-    public function hasFlags($flags)
+    public function hasFlag($flag)
     {
-        return $flags & $this->flags;
+        return $this->flags->hasFlag($flag);
     }
 }
 // }}}
@@ -548,7 +551,17 @@ class ProfileAddresses extends ProfileField
         $res = array();
         $nb = 0;
         foreach ($this->addresses as $addr) {
-            if ($addr->hasFlags($flags)) {
+            if (
+                ($flags & Profile::ADDRESS_ALL)
+                ||
+                (($flags & Profile::ADDRESS_MAIN) && $addr->hasFlag('current'))
+                ||
+                (($flags & Profile::ADDRESS_POSTAL) && $addr->hasFlag('mail'))
+                ||
+                (($flags & Profile::ADDRESS_PERSO) && $addr->link_type == 'home')
+                ||
+                (($flags & Profile::ADDRESS_PRO) && $addr->link_type == 'job')
+            ) {
                 $res[] = $addr;
                 $nb++;
             }
@@ -563,7 +576,7 @@ class ProfileAddresses extends ProfileField
     {
         $data = XDB::iterator('SELECT  pa.id, pa.pid, pa.flags, pa.type AS link_type,
                                        IF(pa.type = \'home\', pid, jobid) AS link_id,
-                                       pa.text, pa.postalCode, pa.latitude, pa.longitude,
+                                       pa.text, pa.postalCode, pa.latitude, pa.longitude, pa.comment,
                                        gl.name AS locality, gas.name AS subAdministrativeArea,
                                        ga.name AS administrativeArea, gc.countryFR AS country
                                  FROM  profile_addresses AS pa
-- 
2.1.4