From 4c0c1c40a8d88561b8ddf4a717e0b8014a8a6eff Mon Sep 17 00:00:00 2001
From: localhost_frssoft <userquicker@disroot.org>
Date: Fri, 10 Nov 2023 18:00:13 +0300
Subject: [PATCH] update README and hardcode poll options to 20

---
 README                  |  4 ++++
 mastodon/mastodon.go    |  9 +++++++++
 mastodon/status.go      | 13 +++++++++++++
 renderer/renderer.go    | 10 ++++++++++
 service/service.go      | 11 ++++++++++-
 service/transport.go    | 16 +++++++++++++++-
 templates/postform.tmpl | 15 +++++++++++++++
 7 files changed, 76 insertions(+), 2 deletions(-)

diff --git a/README b/README
index f921586..14261d7 100644
--- a/README
+++ b/README
@@ -12,6 +12,10 @@ Changes (localhost_custom fork):
 - visible edited post time
 - visible quoted post (status in status)
 - visible profile banner in spoiler
+- add schedule status
+- add language input form
+- add expiry status
+- add support for send poll (hardcoded to 20 options)
 - hide boosts in spoiler
 - hide NSFW content and attachments in spoiler
 - some micro visual changes
diff --git a/mastodon/mastodon.go b/mastodon/mastodon.go
index 9bd00f2..04afeb4 100644
--- a/mastodon/mastodon.go
+++ b/mastodon/mastodon.go
@@ -213,6 +213,15 @@ type Toot struct {
 	Language    string   `json:"language"`
 	ExpiresIn   int      `json:"expires_in"`
 	ScheduledAt string   `json:"scheduled_at"`
+	Poll        TootPoll `json:"poll"`
+}
+
+// TootPoll is struct to poll in post status.
+type TootPoll struct {
+	ExpiresIn  int      `json:"expires_in"`
+	HideTotals bool     `json:"hide_totals"`
+	Multiple   bool     `json:"multiple"`
+	Options    []string `json:"options"`
 }
 
 // Mention hold information for mention.
diff --git a/mastodon/status.go b/mastodon/status.go
index f56ba21..064ffd0 100644
--- a/mastodon/status.go
+++ b/mastodon/status.go
@@ -13,6 +13,7 @@ import (
 	"encoding/json"
 	"path"
 	"strings"
+	"strconv"
 )
 
 type StatusPleroma struct {
@@ -367,7 +368,19 @@ func (c *Client) PostStatus(ctx context.Context, toot *Toot) (*Status, error) {
 	if toot.ScheduledAt != "" {
 		params.Set("scheduled_at", toot.ScheduledAt)
 	}
+	if len(toot.Poll.Options) > 2 {
+		for _, option := range toot.Poll.Options {
+			params.Add("poll[options][]", string(option))
+		}
+		params.Set("poll[expires_in]", strconv.Itoa(toot.Poll.ExpiresIn))
+		if toot.Poll.Multiple {
+			params.Set("poll[multiple]", "true")
+		}
+		if toot.Poll.HideTotals {
+			params.Set("poll[hide_totals]", "true")
+		}
 
+	}
 	var status Status
 	if toot.Edit != "" {
 		err := c.doAPI(ctx, http.MethodPut, fmt.Sprintf("/api/v1/statuses/%s", toot.Edit), params, &status, nil)
diff --git a/renderer/renderer.go b/renderer/renderer.go
index 470490e..34046ba 100644
--- a/renderer/renderer.go
+++ b/renderer/renderer.go
@@ -7,6 +7,7 @@ import (
 	"strconv"
 	"strings"
 	"time"
+	"fmt"
 
 	"bloat/mastodon"
 )
@@ -67,6 +68,14 @@ func emojiFilter(content string, emojis []mastodon.Emoji) string {
 	return strings.NewReplacer(replacements...).Replace(content)
 }
 
+func generatePollOptions() string {
+	var pollbuilder string
+	for i := 0; i < 20; i++ {
+		pollbuilder = pollbuilder + `<div><input id="` + fmt.Sprintf("poll-option-%d", i) + `" name="` + fmt.Sprintf("poll-option-%d", i) + `"></div>`
+	}
+	return pollbuilder
+}
+
 var quoteRE = regexp.MustCompile("(?mU)(^|> *|\n)(&gt;.*)(<br|$)")
 
 func statusContentFilter(content string, emojis []mastodon.Emoji, mentions []mastodon.Mention) string {
@@ -158,6 +167,7 @@ func NewRenderer(templateGlobPattern string) (r *renderer, err error) {
 	t, err = t.Funcs(template.FuncMap{
 		"EmojiFilter":             emojiFilter,
 		"Allowed_emoji_page":             allowed_emoji_page,
+		"GeneratePollOptions":     generatePollOptions,
 		"StatusContentFilter":     statusContentFilter,
 		"DisplayInteractionCount": displayInteractionCount,
 		"TimeSince":               timeSince,
diff --git a/service/service.go b/service/service.go
index cfbab9c..b4cfff2 100644
--- a/service/service.go
+++ b/service/service.go
@@ -978,7 +978,8 @@ func (s *service) Signout(c *client) (err error) {
 
 func (s *service) Post(c *client, content string, replyToID string,
 	format string, visibility string, isNSFW bool, spoilerText string,
-	files []*multipart.FileHeader, edit string, language string, expiresIn int, scheduledAt string) (id string, err error) {
+	files []*multipart.FileHeader, edit string, language string, expiresIn int, scheduledAt string, 
+	pollOptions []string, pollExpiresIn int, pollHideTotals bool, pollMultiple bool) (id string, err error) {
 
 	var mediaIDs []string
 	for _, f := range files {
@@ -991,6 +992,13 @@ func (s *service) Post(c *client, content string, replyToID string,
 
 	expiresIn = expiresIn * 3600
 
+	pollTweet := mastodon.TootPoll{
+		ExpiresIn: pollExpiresIn,
+		Options: pollOptions,
+		Multiple: pollMultiple,
+		HideTotals: pollHideTotals,
+	}
+
 	tweet := &mastodon.Toot{
 		SpoilerText: spoilerText,
 		Status:      content,
@@ -1003,6 +1011,7 @@ func (s *service) Post(c *client, content string, replyToID string,
 		Language: language,
 		ExpiresIn: expiresIn, // pleroma compatible
 		ScheduledAt: scheduledAt,
+		Poll: pollTweet,
 	}
 
 	st, err := c.PostStatus(c.ctx, tweet)
diff --git a/service/transport.go b/service/transport.go
index 41809f8..97ce653 100644
--- a/service/transport.go
+++ b/service/transport.go
@@ -327,8 +327,22 @@ func NewHandler(s *service, verbose bool, staticDir string) http.Handler {
 			}
 			scheduledAt = string(scheduled.UTC().Format(time.RFC3339))
 		}
+		var pollOptions []string
+		for i := 0; i < 16; i++ {
+			v := c.r.FormValue(fmt.Sprintf("poll-option-%d", i))
+			if len(v) == 0 {
+				continue
+			}
+			pollOptions = append(pollOptions, v)
+		}
+		pollExpiresIn, err := strconv.Atoi(c.r.FormValue("poll-expires-in"))
+		if err != nil {
+			return err
+		}
+		pollHideTotals := c.r.FormValue("poll-hide-totals") == "true"
+		pollMultiple := c.r.FormValue("poll-is-multiple") == "true"
 
-		id, err := s.Post(c, content, replyToID, format, visibility, isNSFW, spoilerText, files, edit, language, expiresIn, scheduledAt)
+		id, err := s.Post(c, content, replyToID, format, visibility, isNSFW, spoilerText, files, edit, language, expiresIn, scheduledAt, pollOptions, pollExpiresIn, pollHideTotals, pollMultiple)
 		if err != nil {
 			return err
 		}
diff --git a/templates/postform.tmpl b/templates/postform.tmpl
index 6b49f6c..9ec6926 100644
--- a/templates/postform.tmpl
+++ b/templates/postform.tmpl
@@ -51,6 +51,21 @@
 	<input id="lang-code" name="lang-code" placeholder="lang" title="Post language (ISO 639) [en, ru, etc..] Default: none" size="4" value="{{if .ReplyContext}}{{.ReplyContext.ReplyLanguage}}{{end}}">
 	<input type="number" id="expires-in" name="expires-in" title="Post autodeleted after hour(s)" min="0" value="0" size="4">
 	<input type="datetime-local" id="scheduled" name="scheduled" step=300 title="Schedule your status (timezone UTC+0)">
+	<details><summary>Poll</summary>
+		<div>
+		<input type="number" id="poll-expires-in" name="poll-expires-in" min="1" value="300">
+		<label for="poll-expires-in"> Expires in (secs) </label>
+		</div>
+		<div>
+		<input type="checkbox" id="poll-hide-totals" name="poll-hide-totals" value="true">
+		<label for="poll-hide-totals"> Hide vote counts? </label>
+		</div>
+		<div>
+		<input type="checkbox" id="poll-is-multiple" name="poll-is-multiple" value="true">
+		<label for="poll-is_multiple "> Allow multiple choice? </label>
+		</div>
+		<div>{{GeneratePollOptions | Raw}}</div>
+	</details>
 </form>
 {{end}}