fix args in returnto

[quix0rs-gnu-social.git] / classes / Profile.php

diff --git a/classes/Profile.php b/classes/Profile.php
index 5a678eb54294887018238b3fcfdee6b2fa289c7c..b7bd2a2d9293260525fae5f53958e40e5ed8417e 100644 (file)
--- a/classes/Profile.php
+++ b/classes/Profile.php
@@ -72,4 +72,65 @@ class Profile extends DB_DataObject
                        return NULL;
                }
        }
+
+       function setOriginal($source) {
+
+               $info = @getimagesize($source);
+
+               if (!$info) {
+                       return NULL;
+               }
+
+               $filename = common_avatar_filename($this->id,
+                                                                                  image_type_to_extension($info[2]),
+                                                                                  NULL, common_timestamp());
+               $filepath = common_avatar_path($filename);
+
+               copy($source, $filepath);
+
+               $avatar = new Avatar();
+
+               $avatar->profile_id = $this->id;
+               $avatar->width = $info[0];
+               $avatar->height = $info[1];
+               $avatar->mediatype = image_type_to_mime_type($info[2]);
+               $avatar->filename = $filename;
+               $avatar->original = true;
+               $avatar->url = common_avatar_url($filename);
+               $avatar->created = DB_DataObject_Cast::dateTime(); # current time
+
+               # XXX: start a transaction here
+
+               if (!$this->delete_avatars()) {
+                       @unlink($filepath);
+                       return NULL;
+               }
+
+               if (!$avatar->insert()) {
+                       @unlink($filepath);
+                       return NULL;
+               }
+
+               foreach (array(AVATAR_PROFILE_SIZE, AVATAR_STREAM_SIZE, AVATAR_MINI_SIZE) as $size) {
+                       # We don't do a scaled one if original is our scaled size
+                       if (!($avatar->width == $size && $avatar->height == $size)) {
+                               $s = $avatar->scale($size);
+                               if (!$s) {
+                                       return NULL;
+                               }
+                       }
+               }
+
+               return $avatar;
+       }
+
+       function delete_avatars() {
+               $avatar = DB_DataObject::factory('avatar');
+               $avatar->profile_id = $this->id;
+               $avatar->find();
+               while ($avatar->fetch()) {
+                       $avatar->delete();
+               }
+               return true;
+       }
 }