From 886319d88bff9dadf6bc89e5675bd2c4e4a36302 Mon Sep 17 00:00:00 2001
From: Shish <shish@shishnet.org>
Date: Sat, 4 Feb 2023 18:00:23 +0000
Subject: [PATCH] numeric score API

---
 ext/numeric_score/main.php  | 82 ++++++++++++++++++++++++++++++++-----
 ext/numeric_score/theme.php |  6 +--
 2 files changed, 75 insertions(+), 13 deletions(-)

diff --git a/ext/numeric_score/main.php b/ext/numeric_score/main.php
index 6c76469c..74cf9f12 100644
--- a/ext/numeric_score/main.php
+++ b/ext/numeric_score/main.php
@@ -4,6 +4,76 @@ declare(strict_types=1);
 
 namespace Shimmie2;
 
+use GQLA\Type;
+use GQLA\Field;
+use GQLA\Mutation;
+
+#[Type(name: "NumericScoreVote")]
+class NumericScoreVote
+{
+    public int $image_id;
+    public int $user_id;
+
+    #[Field]
+    public int $score;
+
+    #[Field]
+    public function post(): Image
+    {
+        return Image::by_id($this->image_id);
+    }
+
+    #[Field]
+    public function user(): User
+    {
+        return User::by_id($this->user_id);
+    }
+
+    #[Field(extends: "Post")]
+    public static function score(Image $post): int
+    {
+        global $database;
+        if ($post->score ?? null) {
+            return $post->score;
+        }
+        return $database->get_one(
+            "SELECT sum(score) FROM numeric_score_votes WHERE image_id=:image_id",
+            ['image_id'=>$post->id]
+        ) ?? 0;
+    }
+
+    #[Field(extends: "Post", type: "[NumericScoreVote!]!")]
+    public static function votes(Image $post): array
+    {
+        global $database;
+        $rows = $database->get_all(
+            "SELECT * FROM numeric_score_votes WHERE image_id=:image_id",
+            ['image_id'=>$post->id]
+        );
+        $votes = [];
+        foreach ($rows as $row) {
+            $nsv = new NumericScoreVote();
+            $nsv->image_id = $row["image_id"];
+            $nsv->user_id = $row["user_id"];
+            $nsv->score = $row["score"];
+            $votes[] = $nsv;
+        }
+        return $votes;
+    }
+
+    #[Mutation]
+    public static function create_vote(int $post_id, int $score): bool
+    {
+        global $user;
+        if (!$user->is_anonymous()) {
+            assert($score == 0 || $score == -1 || $score == 1);
+            send_event(new NumericScoreSetEvent($post_id, $user, $score));
+            return true;
+        }
+        return false;
+    }
+}
+
 class NumericScoreSetEvent extends Event
 {
     public int $image_id;
@@ -72,16 +142,8 @@ class NumericScore extends Extension
         } elseif ($event->page_matches("numeric_score_vote") && $user->check_auth_token()) {
             if (!$user->is_anonymous()) {
                 $image_id = int_escape($_POST['image_id']);
-                $char = $_POST['vote'];
-                $score = null;
-                if ($char == "up") {
-                    $score = 1;
-                } elseif ($char == "null") {
-                    $score = 0;
-                } elseif ($char == "down") {
-                    $score = -1;
-                }
-                if (!is_null($score) && $image_id>0) {
+                $score = int_escape($_POST['vote']);
+                if (($score == -1 || $score == 0 || $score == 1) && $image_id>0) {
                     send_event(new NumericScoreSetEvent($image_id, $user, $score));
                 }
                 $page->set_mode(PageMode::REDIRECT);
diff --git a/ext/numeric_score/theme.php b/ext/numeric_score/theme.php
index 94a9215c..6cbda2db 100644
--- a/ext/numeric_score/theme.php
+++ b/ext/numeric_score/theme.php
@@ -21,21 +21,21 @@ class NumericScoreTheme extends Themelet
 			<p><form action='".make_link("numeric_score_vote")."' method='POST'>
 			".$user->get_auth_html()."
 			<input type='hidden' name='image_id' value='$i_image_id'>
-			<input type='hidden' name='vote' value='up'>
+			<input type='hidden' name='vote' value='1'>
 			<input type='submit' value='Vote Up'>
 			</form>
 
 			<form action='".make_link("numeric_score_vote")."' method='POST'>
 			".$user->get_auth_html()."
 			<input type='hidden' name='image_id' value='$i_image_id'>
-			<input type='hidden' name='vote' value='null'>
+			<input type='hidden' name='vote' value='0'>
 			<input type='submit' value='Remove Vote'>
 			</form>
 
 			<form action='".make_link("numeric_score_vote")."' method='POST'>
 			".$user->get_auth_html()."
 			<input type='hidden' name='image_id' value='$i_image_id'>
-			<input type='hidden' name='vote' value='down'>
+			<input type='hidden' name='vote' value='-1'>
 			<input type='submit' value='Vote Down'>
 			</form>
 		";