From af3a7323ce10825ac7c45474290dafce266f497a Mon Sep 17 00:00:00 2001
From: Thomas <tschneider.ac@gmail.com>
Date: Fri, 1 May 2020 16:40:07 +0200
Subject: [PATCH] Support polls

---
 src/Controller/FediPlanController.php | 28 +++++++++++++++++++++++++--
 src/Services/Mastodon_api.php         |  8 ++++++++
 src/SocialEntity/PollOption.php       |  8 ++++----
 templates/fediplan/schedule.html.twig |  4 ++--
 4 files changed, 40 insertions(+), 8 deletions(-)

diff --git a/src/Controller/FediPlanController.php b/src/Controller/FediPlanController.php
index d6a00c8..cbd3c24 100644
--- a/src/Controller/FediPlanController.php
+++ b/src/Controller/FediPlanController.php
@@ -17,7 +17,6 @@ use App\SocialEntity\MastodonAccount;
 use App\SocialEntity\PollOption;
 use DateTime;
 use DateTimeZone;
-use Doctrine\Common\Collections\ArrayCollection;
 use Symfony\Bundle\FrameworkBundle\Controller\AbstractController;
 use Symfony\Component\Form\FormError;
 use Symfony\Component\HttpFoundation\JsonResponse;
@@ -204,6 +203,30 @@ class FediPlanController extends AbstractController
             }
             $params['sensitive'] = ($data->getSensitive() == null || !$data->getSensitive())?false:true;
 
+            $pollOptions = $data->getPollOptions();
+            $pollExpiresAt = $data->getPollExpiresAt();
+            $isPollMultiple = $data->isPollMultiple();
+            if( count($pollOptions) > 0){
+                $count_correct_values = 0;
+                foreach($pollOptions as $po) {
+                    /** @var $po PollOption */
+                    if( $po->getTitle() != null && strlen(trim($po->getTitle())) > 0){
+                        $count_correct_values++;
+                    }
+                }
+
+                if( $count_correct_values > 1 ){
+                    $params['poll']['options'] = [];
+                    foreach($pollOptions as $po) {
+                        /** @var $po PollOption */
+                        if( $po->getTitle() != null && strlen(trim($po->getTitle())) > 0){
+                            $params['poll']['options'][] = trim($po->getTitle());
+                        }
+                    }
+                    $params['poll']['expires_in'] = $pollExpiresAt;
+                    $params['poll']['multiple'] = $isPollMultiple;
+                }
+            }
             try {
                 $date = new DateTime( $data->getScheduledAt()->format("Y-m-d H:i"), new DateTimeZone($data->getTimeZone()) );
                 $date->setTimezone(  new DateTimeZone("UTC"));
@@ -212,13 +235,14 @@ class FediPlanController extends AbstractController
             try {
                 $response = $mastodon_api->post_statuses($params);
             } catch (\ErrorException $e) {}
+
+
             $session = $request->getSession();
             if( isset($response['error']) ){
                 $session->getFlashBag()->add(
                     'Error',
                     $response['error_message']
                 );
-                $form->get('content')->addError(new FormError( $response['error_message']));
             }else{
                 unset($compose);
                 unset($form);
diff --git a/src/Services/Mastodon_api.php b/src/Services/Mastodon_api.php
index c727fab..df3a832 100644
--- a/src/Services/Mastodon_api.php
+++ b/src/Services/Mastodon_api.php
@@ -220,7 +220,15 @@ class Mastodon_api {
 
             unset($parameters['body']['media_ids']);
         }
+        if(isset($parameters['body']['poll']['options'])) {
+            $url .= strpos($url, '?')  !== false ? '' : '?';
+            foreach ($parameters['body']['poll']['options'] as $key => $value){
+                $url .= 'poll[options][]=' . $value . '&';
+            }
+            $url = substr($url, 0, -1);
 
+            unset($parameters['body']['poll']['options']);
+        }
         //Special treatment for filtering notifications
         if( isset($parameters['body']['exclude_types[]']) && $parameters['method'] == "GET" ){
             $url .="?";
diff --git a/src/SocialEntity/PollOption.php b/src/SocialEntity/PollOption.php
index 19e8e3a..271ea82 100644
--- a/src/SocialEntity/PollOption.php
+++ b/src/SocialEntity/PollOption.php
@@ -14,7 +14,7 @@ class PollOption
     /**
      * @return string
      */
-    public function getTitle(): string
+    public function getTitle(): ?string
     {
         return $this->title;
     }
@@ -22,7 +22,7 @@ class PollOption
     /**
      * @param string $title
      */
-    public function setTitle(string $title): void
+    public function setTitle(?string $title): void
     {
         $this->title = $title;
     }
@@ -30,7 +30,7 @@ class PollOption
     /**
      * @return int
      */
-    public function getVotesCount(): int
+    public function getVotesCount(): ?int
     {
         return $this->votes_count;
     }
@@ -38,7 +38,7 @@ class PollOption
     /**
      * @param int $votes_count
      */
-    public function setVotesCount(int $votes_count): void
+    public function setVotesCount(?int $votes_count): void
     {
         $this->votes_count = $votes_count;
     }
diff --git a/templates/fediplan/schedule.html.twig b/templates/fediplan/schedule.html.twig
index 21b902e..11874b0 100644
--- a/templates/fediplan/schedule.html.twig
+++ b/templates/fediplan/schedule.html.twig
@@ -10,13 +10,13 @@
         {% for message in messages %}
             {% if type == 'Error' %}
                 <div class="row">
-                    <div class="alert alert-danger col-md-6" style="text-align: center">
+                    <div class="alert alert-danger col-md-12" style="text-align: center">
                         {{ message }}
                     </div>
                 </div>
             {% else %}
                 <div class="row">
-                    <div class="alert alert-success col-md-6" style="text-align: center">
+                    <div class="alert alert-success col-md-12" style="text-align: center">
                         {{ message }}
                     </div>
                 </div>