stryan/stevefolder
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

set data prefix

Browse Source
This commit is contained in:
stryan 2022-08-11 14:07:12 -04:00
parent bc8f144d93
commit 4ab79e2893
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
main.rb
View File

@ -14,8 +14,13 @@ require './folder.rb'
logger = Logger.new(STDOUT)
logger.level = Logger::INFO
prefix = ENV["STEVEFOLDER_PREFIX"]
if prefix == ""
prefix = "./"
end
logger.info("starting with prefix #{prefix}")
config_obj = YAML::load_file( './config.yaml' )
config_obj = YAML::load_file( '#{prefix}config.yaml' )
logger.info("loaded config")
bot = Discordrb::Commands::CommandBot.new token: config_obj["bot_token"], prefix: "!folder"
folder = Folder.new
@ -72,8 +77,8 @@ bot.message(private: true) do |event|
if res != nil
if res.image? or res.video?
download = Down.download(file.url)
FileUtils.mv(download.path, "./spool/#{download.original_filename}")
folder.submit(Submission.new("./spool/#{download.original_filename}",true,sender))
FileUtils.mv(download.path, "#{prefix}spool/#{download.original_filename}")
folder.submit(Submission.new("#{prefix}spool/#{download.original_filename}",true,sender))
event.respond("Submission #{i} accepted")
i = i + 1
else
@ -89,8 +94,8 @@ bot.message(private: true) do |event|
if mime_guess != nil
if mime_guess.image? or mime_guess.video?
download = Down.download(file.url)
FileUtils.mv(download.path, "./spool/#{download.original_filename}")
folder.submit(Submission.new("./spool/#{download.original_filename}",true,sender))
FileUtils.mv(download.path, "#{prefix}spool/#{download.original_filename}")
folder.submit(Submission.new("#{prefix}spool/#{download.original_filename}",true,sender))
event.respond("Submission #{i} accepted")
updated = true
else