Skip to content

Commit

Permalink
Seperate out steal function into its own file.
Browse files Browse the repository at this point in the history
  • Loading branch information
Droid00000 committed Nov 14, 2024
1 parent 2839e55 commit 2b38fe7
Show file tree
Hide file tree
Showing 5 changed files with 42 additions and 29 deletions.
12 changes: 9 additions & 3 deletions src/frost/admin/settings.rb
Original file line number Diff line number Diff line change
Expand Up @@ -10,11 +10,17 @@
def settings(type, server)
case type
when :archiver
archiver_records(server: server, type: :check) ? "**Archive Channel:** <##{archiver_records(server: server, type: :get)}>" : '**Enabled:** No'
archiver_records(server: server,
type: :check) ? "**Archive Channel:** <##{archiver_records(server: server,
type: :get)}>" : '**Enabled:** No'
when :booster
booster_records(server: server, type: :enabled) ? "**Hoist Role:** <@&#{booster_records(server: server, type: :hoist_role)}>" : '**Enabled:** No'
booster_records(server: server,
type: :enabled) ? "**Hoist Role:** <@&#{booster_records(server: server,
type: :hoist_role)}>" : '**Enabled:** No'
when :events
event_records(server: server, type: :enabled) ? "**Roles:** #{event_records(server: server, type: :get_roles)}" : '**Enabled:** No'
event_records(server: server,
type: :enabled) ? "**Roles:** #{event_records(server: server,
type: :get_roles)}" : '**Enabled:** No'
end
end

Expand Down
4 changes: 3 additions & 1 deletion src/frost/data/schema.rb
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,9 @@ def booster_records(server: nil, user: nil, role: nil, type: nil)
when :get_role
POSTGRES[:Server_Boosters].where(server_id: server, user_id: user).select(:role_id).map(:role_id)&.join.to_i
when :delete_role
POSTGRES[:Server_Boosters].where(server_id: server, role_id: role).delete if !POSTGRES[:Server_Boosters].where(server_id: server, role_id: role).empty?
POSTGRES[:Server_Boosters].where(server_id: server, role_id: role).delete if !POSTGRES[:Server_Boosters].where(
server_id: server, role_id: role
).empty?
when :enabled
!POSTGRES[:Booster_Settings].where(server_id: server).select(:enabled).map(:enabled).empty?
when :disable
Expand Down
25 changes: 0 additions & 25 deletions src/frost/emojis/emoji.rb
Original file line number Diff line number Diff line change
Expand Up @@ -24,28 +24,3 @@ def create_emoji(data)
end
end
end

def steal_emojis(data)
unless data.target.emoji?
data.edit_response(content: RESPONSE[55])
return
end

unless data.bot.profile.on(data.server).permission?(:manage_emojis)
data.edit_response(content: RESPONSE[61])
return
end

if data.server.emoji_limit?
data.edit_response(content: RESPONSE[58])
return
end

data.target.emoji.each do |emoji|
break if data.server.emoji_limit?(emoji)

data.server.add_emoji(emoji.name, emoji.file)
end

data.edit_response(content: "#{RESPONSE[57]} **#{data.target.emoji.count}**")
end
1 change: 1 addition & 0 deletions src/frost/emojis/handler.rb
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

require_relative 'button'
require_relative 'emoji'
require_relative 'steal'

module EmojiCommands
extend Discordrb::EventContainer
Expand Down
29 changes: 29 additions & 0 deletions src/frost/emojis/steal.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
# frozen_string_literal: true

require 'discordrb'
require 'data/constants'

def steal_emojis(data)
unless data.target.emoji?
data.edit_response(content: RESPONSE[55])
return
end

unless data.bot.profile.on(data.server).permission?(:manage_emojis)
data.edit_response(content: RESPONSE[61])
return
end

if data.server.emoji_limit?
data.edit_response(content: RESPONSE[58])
return
end

data.target.emoji.each do |emoji|
break if data.server.emoji_limit?(emoji)

data.server.add_emoji(emoji.name, emoji.file)
end

data.edit_response(content: "#{RESPONSE[57]} **#{data.target.emoji.count}**")
end

0 comments on commit 2b38fe7

Please sign in to comment.