james/fembooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

PM count

Browse Source
This commit is contained in:
Shish 2012-03-02 17:41:25 +00:00
parent 1ff0703a91
commit b6b4122952
1 changed files with 23 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
contrib/pm/main.php
View File

@ -77,14 +77,14 @@ class PrivMsg extends Extension {
} }
} }
/* public function onUserBlockBuilding(UserBlockBuildingEvent $event) {
public function onUserBlockBuilding(UserBlockBuilding $event) {
global $user; global $user;
if(!$user->is_anonymous()) { if(!$user->is_anonymous()) {
$event->add_link("Private Messages", make_link("pm")); $count = $this->count_pms($user);
$h_count = $count > 0 ? " ($count)" : "";
$event->add_link("Private Messages", make_link("user"));
} }
} }
*/
public function onUserPageBuilding(UserPageBuildingEvent $event) { public function onUserPageBuilding(UserPageBuildingEvent $event) {
global $page, $user; global $page, $user;
@ -113,6 +113,7 @@ class PrivMsg extends Extension {
else if(($pm["to_id"] == $user->id) || $user->is_admin()) { else if(($pm["to_id"] == $user->id) || $user->is_admin()) {
$from_user = User::by_id(int_escape($pm["from_id"])); $from_user = User::by_id(int_escape($pm["from_id"]));
$database->get_row("UPDATE private_message SET is_read='Y' WHERE id = :id", array("id" => $pm_id)); $database->get_row("UPDATE private_message SET is_read='Y' WHERE id = :id", array("id" => $pm_id));
$database->cache->delete("pm-count-{$user->id}");
$this->theme->display_message($page, $from_user, $user, new PM($pm)); $this->theme->display_message($page, $from_user, $user, new PM($pm));
} }
else { else {
@ -128,6 +129,7 @@ class PrivMsg extends Extension {
} }
else if(($pm["to_id"] == $user->id) || $user->is_admin()) { else if(($pm["to_id"] == $user->id) || $user->is_admin()) {
$database->execute("DELETE FROM private_message WHERE id = :id", array("id" => $pm_id)); $database->execute("DELETE FROM private_message WHERE id = :id", array("id" => $pm_id));
$database->cache->delete("pm-count-{$user->id}");
log_info("pm", "Deleted PM #$pm_id"); log_info("pm", "Deleted PM #$pm_id");
$page->set_mode("redirect"); $page->set_mode("redirect");
$page->set_redirect($_SERVER["HTTP_REFERER"]); $page->set_redirect($_SERVER["HTTP_REFERER"]);
@ -163,6 +165,7 @@ class PrivMsg extends Extension {
array("fromid" => $event->pm->from_id, "fromip" => $event->pm->from_ip, array("fromid" => $event->pm->from_id, "fromip" => $event->pm->from_ip,
"toid" => $event->pm->to_id, "subject" => $event->pm->subject, "message" => $event->pm->message) "toid" => $event->pm->to_id, "subject" => $event->pm->subject, "message" => $event->pm->message)
); );
$database->cache->delete("pm-count-{$event->pm->to_id}");
log_info("pm", "Sent PM to User #{$event->pm->to_id}"); log_info("pm", "Sent PM to User #{$event->pm->to_id}");
} }
@ -182,5 +185,21 @@ class PrivMsg extends Extension {
} }
return $pms; return $pms;
} }
private function count_pms(User $user) {
global $database;
$count = $database->cache->get("pm-count-{$user->id}");
if(is_null($count)) {
$count = $database->get_one("
SELECT count(*)
FROM private_message
WHERE to_id = :toid
AND is_read = :is_read
", array("toid" => $user->id, "is_read" => "N"));
$database->cache->set("pm-count-{$user->id}", $count, 60);
}
return $count;
}
} }
?> ?>