From ade8fddea8e317ab91f9e57ca138f9f9f5cc7d8c Mon Sep 17 00:00:00 2001
From: Janusz Dobrowolski <janusz@frontaccounting.eu>
Date: Sun, 18 Jul 2010 16:38:08 +0000
Subject: [PATCH] Improved key_in_foreign table function to check in multiply
 tables.

---
 admin/db/company_db.inc | 33 +++++++++++++++++++++++++++------
 1 file changed, 27 insertions(+), 6 deletions(-)

diff --git a/admin/db/company_db.inc b/admin/db/company_db.inc
index cf20d52b..b3988454 100644
--- a/admin/db/company_db.inc
+++ b/admin/db/company_db.inc
@@ -149,15 +149,36 @@ function get_payment_terms_all($show_inactive)
 	if (!$show_inactive) $sql .= " WHERE !inactive";
 	return db_query($sql,"could not get payment terms");
 }
-
-function key_in_foreign_table($id, $table, $key, $escaped=false)
+/*
+	Return number of records in tables, where some foreign key $id is used.
+	$id - searched key value
+	$tables - array of table names (without prefix); when table name is used as a key, then
+		value is name of foreign key field. For numeric keys $stdkey field name is used.
+	$stdkey - standard name of foreign key.
+*/
+function key_in_foreign_table($id, $tables, $stdkey, $escaped=false)
 {
 	if (!$escaped)
 		$id = db_escape($id);
-	$sql= "SELECT COUNT(*) FROM ".TB_PREF."$table WHERE $key = $id";
-	$result = db_query($sql,"check $table relations failed");
-	$myrow = db_fetch_row($result);
-	return ($myrow[0] > 0); 
+
+	if (!is_array($tables))
+		$tables = array($tables);
+
+	$sqls = array();
+	foreach ($tables as $tbl => $key) {
+		if (is_numeric($tbl)) {
+			$tbl = $key;
+			$key = $stdkey;
+		}
+		$sqls[] = "(SELECT COUNT(*) as cnt FROM ".TB_PREF."$tbl WHERE $key=$id)\n";
+	}
+
+	$sql = "SELECT sum(cnt) FROM (". implode(' UNION ', $sqls).") as counts";
+
+	$result = db_query($sql, "check relations for ".implode(',',$tables)." failed");
+	$count =  db_fetch($result);
+
+	return $count[0];
 }
 
 ?>
\ No newline at end of file
-- 
2.30.2