1 files changed, 52 insertions, 0 deletions

diff --git a/engine/lib/database.php b/engine/lib/database.php
index c09b07d57..420cb6f4c 100644
--- a/engine/lib/database.php
+++ b/engine/lib/database.php
@@ -216,6 +216,38 @@
 	}
 	
 	/**
+	 * @class SetQueryComponent Set query.
+	 * Represents an update set query.
+	 * @author Marcus Povey
+	 */
+	class SetQueryComponent extends QueryComponent
+	{
+		/**
+		 * Construct a setting query
+		 *
+		 * @param string $table The table to modify
+		 * @param string $field The field to modify
+		 * @param mixed $value The value to set it to
+		 */
+		function __construct($table, $field, $value)
+		{
+			global $CONFIG;
+			
+		 	$this->table = $CONFIG->dbprefix . sanitise_string($table);
+		 	$this->field = sanitise_string($field);
+		 	if (is_numeric($value))
+		 		$this->value = (int)$value;
+		 	else
+		 		$this->value = "'".sanitise_string($value)."'";
+		}
+		
+		function __toString() 
+		{
+			return "{$this->table}.{$this->field}={$this->value}";
+		}
+	}
+	
+	/**
 	 * @class WhereQueryComponent
 	 * A component of a where query.
 	 * @author Marcus Povey
@@ -417,6 +449,9 @@
 		/// Join tables
 		private $joins;
 		
+		/// Set values 
+		private $sets;
+		
 		/// Where query
 		private $where;
 		
@@ -438,6 +473,7 @@
 			$this->tables = array();
 			$this->joins = array();
 			$this->where = array();
+			$this->sets = array();
 			
 			$this->setQueryType(new SelectQueryTypeQueryComponent());
 		}
@@ -454,6 +490,8 @@
 		
 		public function addJoin(JoinQueryComponent $component) { $this->joins[] = $component; }
 		
+		public function addSet(SetQueryComponent $component) { $this->sets[] = $component; }
+		
 		public function setQueryType(QueryTypeQueryComponent $component) { $this->query_type = $component; }
 		
 		public function setOrder(OrderQueryComponent $component) { $this->order = $component; }
@@ -513,6 +551,20 @@
 					}
 				}
 				
+				// Setting values
+				if (
+					($this->query_type->query_type == 'update') || 
+					($this->query_type->query_type == 'insert')
+				) 
+				{
+					$sql .= "set ";
+					
+					foreach ($this->sets as $set)
+						$sql .= "$set, ";
+						
+					$sql = trim($sql, ", ") . " ";
+				}
+				
 				// Where
 				if (!empty($this->where))
 				{