Changeset 4445 for trunk/include/ws_functions.inc.php

Timestamp:

Dec 7, 2009, 11:30:02 PM (15 years ago)

Author:

plg

Message:

merge r4444 from branch 2.0 to trunk

feature 1312 added: pwg.images.setInfo can replace multiple values propreties
(tags/categories). The primary condition is to provide "tag_ids" and/or
"categories" input parameters, ie pwg.images.set won't create emptiness. You
can reduce the set of tags/categories but not remove all tags/categories.

bug 1277 fixed: with a stronger algorithm for adding/replacing categories, we
now avoid to recreate an existing image_category association.

When a remote client calls pwg.images.setInfo, Piwigo returns an error 500 if:

1. the "categories" parameter is malformed (no numeric id inside)
2. one of the input categories does not exist at database level.

File:

• trunk/include/ws_functions.inc.php (modified) (6 diffs)

trunk/include/ws_functions.inc.php

@@ -1731 +1731 @@
     ws_add_image_category_relations(
       $params['image_id'],
-      $params['categories']
+      $params['categories'],
+      $params['replace_mode']
       );
   }
@@ -1739 +1740 @@
   {
     include_once(PHPWG_ROOT_PATH.'admin/include/functions.php');
-    add_tags(
-      explode(',', $params['tag_ids']),
-      array($params['image_id'])
-      );
+
+    $tag_ids = explode(',', $params['tag_ids']);
+
+    if ($params['replace_mode'])
+    {
+      set_tags(
+        $tag_ids,
+        $params['image_id']
+        );
+    }
+    else
+    {
+      add_tags(
+        $tag_ids,
+        array($params['image_id'])
+        );
+    }
   }
 
@@ -1748 +1762 @@
 }
 
-function ws_add_image_category_relations($image_id, $categories_string)
+function ws_add_image_category_relations($image_id, $categories_string, $replace_mode=false)
 {
   // let's add links between the image and the categories
@@ -1766 +1780 @@
     @list($cat_id, $rank) = explode(',', $token);
 
+    if (!preg_match('/^\d+$/', $cat_id))
+    {
+      continue;
+    }
+
     array_push($cat_ids, $cat_id);
 
@@ -1782 +1801 @@
   $cat_ids = array_unique($cat_ids);
 
-  if (count($cat_ids) > 0)
-  {
-    if ($search_current_ranks)
+  if (count($cat_ids) == 0)
+  {
+    new PwgError(
+      500,
+      '[ws_add_image_category_relations] there is no category defined in "'.$categories_string.'"'
+      );
+    exit();
+  }
+  
+  $query = '
+SELECT
+    id
+  FROM '.CATEGORIES_TABLE.'
+  WHERE id IN ('.implode(',', $cat_ids).')
+;';
+  $db_cat_ids = array_from_query($query, 'id');
+
+  $unknown_cat_ids = array_diff($cat_ids, $db_cat_ids);
+  if (count($unknown_cat_ids) != 0)
+  {
+    new PwgError(
+      500,
+      '[ws_add_image_category_relations] the following categories are unknown: '.implode(', ', $unknown_cat_ids)
+      );
+    exit();
+  }
+  
+  $to_update_cat_ids = array();
+    
+  // in case of replace mode, we first check the existing associations
+  $query = '
+SELECT
+    category_id
+  FROM '.IMAGE_CATEGORY_TABLE.'
+  WHERE image_id = '.$image_id.'
+;';
+  $existing_cat_ids = array_from_query($query, 'category_id');
+
+  if ($replace_mode)
+  {
+    $to_remove_cat_ids = array_diff($existing_cat_ids, $cat_ids);
+    if (count($to_remove_cat_ids) > 0)
     {
       $query = '
+DELETE
+  FROM '.IMAGE_CATEGORY_TABLE.'
+  WHERE image_id = '.$image_id.'
+    AND category_id IN ('.implode(', ', $to_remove_cat_ids).')
+;';
+      pwg_query($query);
+      update_category($to_remove_cat_ids);
+    }
+  }
+  
+  $new_cat_ids = array_diff($cat_ids, $existing_cat_ids);
+  if (count($new_cat_ids) == 0)
+  {
+    return true;
+  }
+    
+  if ($search_current_ranks)
+  {
+    $query = '
 SELECT
     category_id,
@@ -1792 +1869 @@
   FROM '.IMAGE_CATEGORY_TABLE.'
   WHERE rank IS NOT NULL
-    AND category_id IN ('.implode(',', $cat_ids).')
+    AND category_id IN ('.implode(',', $new_cat_ids).')
   GROUP BY category_id
 ;';
-      $current_rank_of = simple_hash_from_query(
-        $query,
-        'category_id',
-        'max_rank'
-        );
-
-      foreach ($cat_ids as $cat_id)
+    $current_rank_of = simple_hash_from_query(
+      $query,
+      'category_id',
+      'max_rank'
+      );
+
+    foreach ($new_cat_ids as $cat_id)
+    {
+      if (!isset($current_rank_of[$cat_id]))
       {
-        if (!isset($current_rank_of[$cat_id]))
-        {
-          $current_rank_of[$cat_id] = 0;
-        }
-
-        if ('auto' == $rank_on_category[$cat_id])
-        {
-          $rank_on_category[$cat_id] = $current_rank_of[$cat_id] + 1;
-        }
+        $current_rank_of[$cat_id] = 0;
       }
-    }
-
-    $inserts = array();
-
-    foreach ($cat_ids as $cat_id)
-    {
-      array_push(
-        $inserts,
-        array(
-          'image_id' => $image_id,
-          'category_id' => $cat_id,
-          'rank' => $rank_on_category[$cat_id],
-          )
-        );
-    }
-
-    include_once(PHPWG_ROOT_PATH.'admin/include/functions.php');
-    mass_inserts(
-      IMAGE_CATEGORY_TABLE,
-      array_keys($inserts[0]),
-      $inserts
-      );
-
-    update_category($cat_ids);
-  }
+      
+      if ('auto' == $rank_on_category[$cat_id])
+      {
+        $rank_on_category[$cat_id] = $current_rank_of[$cat_id] + 1;
+      }
+    }
+  }
+  
+  $inserts = array();
+  
+  foreach ($new_cat_ids as $cat_id)
+  {
+    array_push(
+      $inserts,
+      array(
+        'image_id' => $image_id,
+        'category_id' => $cat_id,
+        'rank' => $rank_on_category[$cat_id],
+        )
+      );
+  }
+  
+  include_once(PHPWG_ROOT_PATH.'admin/include/functions.php');
+  mass_inserts(
+    IMAGE_CATEGORY_TABLE,
+    array_keys($inserts[0]),
+    $inserts
+    );
+  
+  update_category($new_cat_ids);
 }