Add claiming of multipastes

Signed-off-by: Florian Pritz <bluewind@xinu.at>

2 files changed, 13 insertions, 0 deletions

diff --git a/application/models/mfile.php b/application/models/mfile.php
index 0ec27a817..51c865900 100644
--- a/application/models/mfile.php
+++ b/application/models/mfile.php
@@ -126,7 +126,9 @@ class Mfile extends CI_Model {
 
 		$this->db->set(array('user' => $userid ))
 			->where('id', $id)
+			->where('user', 0)
 			->update('files');
+		return $this->db->affected_rows();
 	}
 
 	// remove old/invalid/broken IDs
diff --git a/application/models/mmultipaste.php b/application/models/mmultipaste.php
index ed3b8e3a7..f80105ec9 100644
--- a/application/models/mmultipaste.php
+++ b/application/models/mmultipaste.php
@@ -88,6 +88,17 @@ class Mmultipaste extends CI_Model {
 		return true;
 	}
 
+	function adopt($id)
+	{
+		$userid = $this->muser->get_userid();
+
+		$this->db->set(array('user_id' => $userid ))
+			->where('url_id', $id)
+			->where('user_id', 0)
+			->update('multipaste');
+		return $this->db->affected_rows();
+	}
+
 	public function get_tarball_path($id)
 	{
 		return $this->config->item("upload_path")."/special/multipaste-tarballs/".substr(md5($id), 0, 3)."/$id.tar.gz";