From 1e6289872d2fd09dd19e48d8faa2e3e1f243a5ba Mon Sep 17 00:00:00 2001
From: fred <fred@mslsr.com>
Date: Wed, 26 Oct 2022 11:41:33 +0200
Subject: [PATCH] add support for pinning and unpinning

messages get pinned in the same channel as the message was reacted to
---
 bot.rb | 56 ++++++++++++++++++++++++++++++++++++++------------------
 1 file changed, 38 insertions(+), 18 deletions(-)

diff --git a/bot.rb b/bot.rb
index 21022b9..21970ba 100644
--- a/bot.rb
+++ b/bot.rb
@@ -3,32 +3,52 @@ require 'discordrb'
 
 bot = Discordrb::Bot.new token: ENV['DISCORD_TOKEN']
 
-EMOJI = '🔖'
+BOOKMARK_EMOJI = '🔖'
+PIN_EMOJI = '📌'
+
 TRIM_MESSAGE_LENGTH = 500
 
+def get_bot_token
+  "Bot #{ENV['DISCORD_TOKEN']}"
+end
+
 bot.reaction_add do |event|
-  next unless event.emoji.name == EMOJI
-  # extract data of the message's author
-  message_author = event.message.author
-  message_author_pfp = message_author.avatar_url 'png'
-  message_author_nickname = message_author.display_name
-  message_author_distinct = message_author.distinct
+  next unless [BOOKMARK_EMOJI, PIN_EMOJI].include? event.emoji.name
 
-  message_author_fname = "#{message_author_nickname} (#{message_author_distinct})" # this is the 'author' field of the embed
+  if(event.emoji.name == BOOKMARK_EMOJI)
+    # extract data of the message's author
+    message_author = event.message.author
+    message_author_pfp = message_author.avatar_url 'png'
+    message_author_nickname = message_author.display_name
+    message_author_distinct = message_author.distinct
 
-  trimmed_msg = "#{event.message.content[0..TRIM_MESSAGE_LENGTH]}#{'...' if event.message.content.length > TRIM_MESSAGE_LENGTH}" # trim the string to TRIM_MESSAGE_LENGTH chars max to not make the embed too huge
-  original_msg_link = event.message.link
+    message_author_fname = "#{message_author_nickname} (#{message_author_distinct})" # this is the 'author' field of the embed
 
-  pm = event.user.pm
+    trimmed_msg = "#{event.message.content[0..TRIM_MESSAGE_LENGTH]}#{'...' if event.message.content.length > TRIM_MESSAGE_LENGTH}" # trim the string to TRIM_MESSAGE_LENGTH chars max to not make the embed too huge
+    original_msg_link = event.message.link
 
-  pm.send_embed 'Bookmark created!' do |embed|
-    embed.title = 'Go to message'
-    embed.url = original_msg_link
-    embed.author = Discordrb::Webhooks::EmbedAuthor.new(icon_url: message_author_pfp, name: message_author_fname)
-    embed << Discordrb::Webhooks::EmbedField.new(name: 'Message (trimmed)', value: trimmed_msg)
-    embed.timestamp = event.message.timestamp
+    pm = event.user.pm
+
+    pm.send_embed 'Bookmark created!' do |embed|
+      embed.title = 'Go to message'
+      embed.url = original_msg_link
+      embed.author = Discordrb::Webhooks::EmbedAuthor.new(icon_url: message_author_pfp, name: message_author_fname)
+      embed << Discordrb::Webhooks::EmbedField.new(name: 'Message (trimmed)', value: trimmed_msg)
+      embed.timestamp = event.message.timestamp
+    end
   end
+
+  if(event.emoji.name == PIN_EMOJI)
+    Discordrb::API::Channel.pin_message(get_bot_token(), event.channel.id, event.message.id)
+  end
+
   nil # `event` is of type Respondable, so the return line of the block needs to be nil to prevent the bot sending something in the chhannel the reaction occured.
 end
 
-bot.run
\ No newline at end of file
+bot.reaction_remove do |event|
+  next unless event.emoji.name == PIN_EMOJI
+  Discordrb::API::Channel.unpin_message(get_bot_token(), event.channel.id, event.message.id)
+  nil
+end
+
+bot.run