481 lines
17 KiB
Ruby
481 lines
17 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module DiscordInteractions
|
||
|
|
class YiffyAPI
|
||
|
|
attr_reader :raw
|
||
|
|
|
||
|
|
GREEN_TICK = APIKey::WebhookMethods::GREEN_TICK
|
||
|
|
RED_TICK = APIKey::WebhookMethods::RED_TICK
|
||
|
|
|
||
|
|
def initialize(raw)
|
||
|
|
@raw = raw
|
||
|
|
end
|
||
|
|
|
||
|
|
def [](key)
|
||
|
|
@raw[key.to_s]
|
||
|
|
end
|
||
|
|
|
||
|
|
# rubocop:disable Metrics/BlockNesting
|
||
|
|
# noinspection RubyCaseWithoutElseBlockInspection
|
||
|
|
def handle
|
||
|
|
return { type: DiscordConstants::InteractionResponseTypes::PONG } if raw["type"] == DiscordConstants::InteractionTypes::PING
|
||
|
|
|
||
|
|
if raw["guild_id"].nil?
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "My commands cannot be used in Direct Messages.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
CurrentUser.user = CurrentUser.user = APIUser.find_or_create_by(id: raw["member"]["user"]["id"])
|
||
|
|
CurrentUser.ip_addr = "0.0.0.0"
|
||
|
|
|
||
|
|
case raw["type"]
|
||
|
|
when DiscordConstants::InteractionTypes::APPLICATION_COMMAND
|
||
|
|
options = DiscordInteractions::InteractionOptions.new(raw["data"]["options"], raw["data"]["resolved"])
|
||
|
|
sub = options.get_subcommand
|
||
|
|
case raw["data"]["name"]
|
||
|
|
when "apikey"
|
||
|
|
keys = APIKey.for_owner(raw["member"]["user"]["id"])
|
||
|
|
case sub[0]
|
||
|
|
when "create"
|
||
|
|
if keys.length >= 3
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "You already have the maximum amount of api keys. Contact a developer if you believe an exception should be made.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::MODAL,
|
||
|
|
data: {
|
||
|
|
custom_id: "apikey-create",
|
||
|
|
components: [
|
||
|
|
{
|
||
|
|
type: DiscordConstants::ComponentTypes::ACTION_ROW,
|
||
|
|
components: [
|
||
|
|
{
|
||
|
|
custom_id: "apikey-create.name",
|
||
|
|
placeholder: "My Awesome Application",
|
||
|
|
min_length: 3,
|
||
|
|
max_length: 50,
|
||
|
|
label: "Name",
|
||
|
|
style: DiscordConstants::TextInputStyles::SHORT,
|
||
|
|
type: DiscordConstants::ComponentTypes::TEXT_INPUT,
|
||
|
|
},
|
||
|
|
],
|
||
|
|
},
|
||
|
|
{
|
||
|
|
type: DiscordConstants::ComponentTypes::ACTION_ROW,
|
||
|
|
components: [
|
||
|
|
{
|
||
|
|
custom_id: "apikey-create.usage",
|
||
|
|
placeholder: "Showing porn to my friends",
|
||
|
|
min_length: 5,
|
||
|
|
max_length: 400,
|
||
|
|
label: "Usage",
|
||
|
|
style: DiscordConstants::TextInputStyles::PARAGRAPH,
|
||
|
|
type: DiscordConstants::ComponentTypes::TEXT_INPUT,
|
||
|
|
},
|
||
|
|
],
|
||
|
|
},
|
||
|
|
],
|
||
|
|
title: "Create API Key",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
when "delete"
|
||
|
|
key = keys.find { |k| k.encoded == options.get_string("key", required: true) }
|
||
|
|
if key.nil? || key.owner_id.to_s != raw["member"]["user"]["id"]
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "Invalid key specified.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
if key.disabled?
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "This key has been disabled by a developer. To have this key deleted or removed, concat a developer.\n\nDisable Reason: **#{key.disabled_reason || '(None)'}**",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "Are you sure you want to delete the key **#{key.application_name}**? This action cannot be undone.",
|
||
|
|
components: [
|
||
|
|
{
|
||
|
|
type: DiscordConstants::ComponentTypes::ACTION_ROW,
|
||
|
|
components: [
|
||
|
|
{
|
||
|
|
custom_id: "apikey-delete-yes.#{key.encoded}.#{raw['member']['user']['id']}",
|
||
|
|
label: "Yes",
|
||
|
|
style: DiscordConstants::ButtonStyles::SUCCESS,
|
||
|
|
type: DiscordConstants::ComponentTypes::BUTTON,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
custom_id: "apikey-delete-no.#{raw['member']['user']['id']}",
|
||
|
|
label: "No",
|
||
|
|
style: DiscordConstants::ButtonStyles::DANGER,
|
||
|
|
type: DiscordConstants::ComponentTypes::BUTTON,
|
||
|
|
},
|
||
|
|
],
|
||
|
|
},
|
||
|
|
],
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
when "list"
|
||
|
|
if keys.empty?
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "You do not have any API keys.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
text = []
|
||
|
|
|
||
|
|
keys.each_with_index do |apikey, index|
|
||
|
|
text.push(
|
||
|
|
<<~TEXT,
|
||
|
|
#{index + 1}.)
|
||
|
|
- Key: ||#{apikey.key}||
|
||
|
|
- Application: `#{apikey.application_name}`
|
||
|
|
- Usage: `#{apikey.usage || 'None Specified'}`
|
||
|
|
- Active: #{apikey.active? ? GREEN_TICK : RED_TICK}
|
||
|
|
- Disabled: #{apikey.disabled? ? "#{GREEN_TICK} (Reason: #{apikey.disabled_reason || 'None Specified'})" : RED_TICK}
|
||
|
|
- Unlimited: #{apikey.unlimited? ? GREEN_TICK : RED_TICK}
|
||
|
|
- Services: #{apikey.access_string}
|
||
|
|
TEXT
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "We found the following api keys:\n\n#{text.join("\n\n")}",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
when "apidev"
|
||
|
|
unless Websites.config.yiffyapi_administrators.include?(raw["member"]["user"]["id"])
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "You are not allowed to use that.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
case sub[0]
|
||
|
|
when "list"
|
||
|
|
user_id = options.get_user("user", required: true)["id"]
|
||
|
|
keys = APIKey.for_owner(user_id)
|
||
|
|
|
||
|
|
if keys.empty?
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "That user does not have any API keys.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
text = []
|
||
|
|
|
||
|
|
keys.each_with_index do |apikey, index|
|
||
|
|
text.push(
|
||
|
|
<<~TEXT,
|
||
|
|
#{index + 1}.)
|
||
|
|
- Key: ||#{apikey.key}||
|
||
|
|
- Application: `#{apikey.application_name}`
|
||
|
|
- Usage: `#{apikey.usage || 'None Specified'}`
|
||
|
|
- Active: #{apikey.active? ? GREEN_TICK : RED_TICK}
|
||
|
|
- Disabled: #{apikey.disabled? ? "#{GREEN_TICK} (Reason: #{apikey.disabled_reason || 'None Specified'})" : RED_TICK}
|
||
|
|
- Unlimited: #{apikey.unlimited? ? GREEN_TICK : RED_TICK}
|
||
|
|
- Services: #{apikey.access_string}
|
||
|
|
TEXT
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "We found the following api keys for <@#{user_id}>:\n\n#{text.join("\n\n")}",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
when "disable"
|
||
|
|
key = options.get_string("key", required: true)
|
||
|
|
reason = options.get_string("reason") || "None Provided"
|
||
|
|
apikey = APIKey.find_by(key: key)
|
||
|
|
if apikey.nil?
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "I couldn't find that api key.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
if apikey.disabled?
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "That api key is already disabled.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
apikey.update(disabled: true, disabled_reason: reason)
|
||
|
|
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "Api key successfully disabled.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
when "enable"
|
||
|
|
key = options.get_string("key", required: true)
|
||
|
|
apikey = APIKey.find_by(key: key)
|
||
|
|
if apikey.nil?
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "I couldn't find that api key.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
unless apikey.disabled?
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "That api key is not disabled.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
apikey.update(disabled: false, disabled_reason: nil)
|
||
|
|
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "Api key successfully enabled.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
when "stats"
|
||
|
|
key = options.get_string("key")
|
||
|
|
ip = options.get_string("ip")
|
||
|
|
if key
|
||
|
|
apikey = APIKey.find_by(key: key)
|
||
|
|
if apikey.nil?
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "I couldn't find that api key.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
stats = APIKey.stats(ip: ip, key: key)
|
||
|
|
if key.present?
|
||
|
|
stats = flatten_hash(stats[:key])
|
||
|
|
text = "Stats for **#{apikey.application_name}** (||#{apikey.key}||)\n\n"
|
||
|
|
elsif ip.present?
|
||
|
|
stats = flatten_hash(stats[:ip])
|
||
|
|
text = "Stats for **#{ip}**\n\n"
|
||
|
|
else
|
||
|
|
stats = flatten_hash(stats[:global])
|
||
|
|
text = "Global stats\n\n"
|
||
|
|
end
|
||
|
|
|
||
|
|
stats.each do |name, value|
|
||
|
|
text += "**#{name}**: #{ActiveSupport::NumberHelper.number_to_delimited(value)}\n"
|
||
|
|
end
|
||
|
|
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: text,
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
end
|
||
|
|
when DiscordConstants::InteractionTypes::MESSAGE_COMPONENT
|
||
|
|
action, *other, user = raw["data"]["custom_id"].split(".")
|
||
|
|
if user != raw["member"]["user"]["id"]
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "That is not yours to play with.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
case action
|
||
|
|
when "apikey-delete-yes"
|
||
|
|
key = APIKey.for_owner(user).find { |k| k.encoded == other[0] }
|
||
|
|
if key.nil? || key.owner_id.to_s != user
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "Invalid key specified.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
key.destroy
|
||
|
|
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::UPDATE_MESSAGE,
|
||
|
|
data: {
|
||
|
|
content: "Key deleted.",
|
||
|
|
components: [],
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
when "apikey-delete-no"
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::UPDATE_MESSAGE,
|
||
|
|
data: {
|
||
|
|
content: "Cancelled.",
|
||
|
|
components: [],
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
when DiscordConstants::InteractionTypes::APPLICATION_COMMAND_AUTOCOMPLETE
|
||
|
|
options = DiscordInteractions::InteractionOptions.new(raw["data"]["options"], raw["data"]["resolved"])
|
||
|
|
sub, = options.get_subcommand
|
||
|
|
opt = options.get_focused_option(required: true)
|
||
|
|
case raw["data"]["name"]
|
||
|
|
when "apikey"
|
||
|
|
case sub
|
||
|
|
when "delete"
|
||
|
|
keys = APIKey.for_owner(raw["member"]["user"]["id"])
|
||
|
|
value = opt["value"]
|
||
|
|
keys = keys.filter { |k| k.application_name.start_with?(value) }
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::APPLICATION_COMMAND_AUTOCOMPLETE_RESULT,
|
||
|
|
data: {
|
||
|
|
choices: keys.map do |k|
|
||
|
|
{
|
||
|
|
name: k.application_name,
|
||
|
|
value: k.encoded,
|
||
|
|
}
|
||
|
|
end,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
end
|
||
|
|
when DiscordConstants::InteractionTypes::MODAL_SUBMIT
|
||
|
|
case raw["data"]["custom_id"]
|
||
|
|
when "apikey-create"
|
||
|
|
name = raw["data"]["components"][0]["components"][0]["value"]
|
||
|
|
usage = raw["data"]["components"][1]["components"][0]["value"]
|
||
|
|
if name.length < 3 || name.length > 50
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "Name must be between 3 and 50 characters.",
|
||
|
|
components: [],
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
if usage.length < 5 || usage.length > 400
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "Usage must be between 5 and 400 characters.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
key = APIKey.create(
|
||
|
|
owner_id: raw["member"]["user"]["id"],
|
||
|
|
application_name: name,
|
||
|
|
usage: usage,
|
||
|
|
)
|
||
|
|
|
||
|
|
if key.errors.present?
|
||
|
|
Rails.logger.error("Failed to create apikey for #{raw['member']['user']['id']}: #{key.errors.full_messages.join(', ')}")
|
||
|
|
return {
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "An error occurred while creating your API key.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
{
|
||
|
|
type: DiscordConstants::InteractionResponseTypes::CHANNEL_MESSAGE_WITH_SOURCE,
|
||
|
|
data: {
|
||
|
|
content: "Your API key: `#{key.key}`. Provide this in the `Authorization` header. You must still provide a unique user agent. If you have any issues, contact a developer.",
|
||
|
|
**common,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
# rubocop:enable Metrics/BlockNesting
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def flatten_hash(hash)
|
||
|
|
hash.each_with_object({}) do |(k, v), h|
|
||
|
|
if v.is_a?(Hash)
|
||
|
|
flatten_hash(v).map do |h_k, h_v|
|
||
|
|
h["#{k}.#{h_k}".to_sym] = h_v
|
||
|
|
end
|
||
|
|
else
|
||
|
|
h[k] = v
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def common
|
||
|
|
{
|
||
|
|
flags: DiscordConstants::MessageFlags::EPHEMERAL,
|
||
|
|
allowed_mentions: {
|
||
|
|
users: [],
|
||
|
|
roles: [],
|
||
|
|
everyone: [],
|
||
|
|
replied_user: false,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|