Supositware/Discord_Elixir_test
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Moved download to its own function

Browse source
This commit is contained in:
Loïc Bersier 2024-10-13 17:17:40 +02:00
parent 80e0f53e85
commit 23a884339d
2 changed files with 32 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
lib/HahaCommands/download.ex
View file

@ -26,31 +26,29 @@ defmodule HahaYes.Commands.Download do
url = Enum.at(args, 0) url = Enum.at(args, 0)
{:ok, loading} = Api.create_message(msg.channel_id, "Downloading...") {:ok, loading} = Api.create_message(msg.channel_id, "Downloading...")
with {:ok, output} <- HahaYes.Utility.download(url, "#{System.tmp_dir}/#{msg.id}") do
{:ok, file} = File.stat(output)
file_size =
file.size / 1000000.0
|> Decimal.from_float()
|> Decimal.round(2)
|> Decimal.to_float()
opt = ["-f", "bestvideo[height<=?480]+bestaudio/best", url, "-o", "#{System.tmp_dir}/#{msg.id}.%(ext)si", "--force-overwrites", "--playlist-reverse", "--no-playlist", "--remux-video=mp4/webm/mov", "--no-warnings"]; Api.delete_message(loading.channel_id, loading.id)
Api.delete_message(msg)
System.cmd("yt-dlp", opt) if file_size >= 25 do
Api.create_message(msg.channel_id, "File size is too big! (#{file_size})")
output = Enum.at(Path.wildcard("#{System.tmp_dir}/#{msg.id}.*"), 0) else
{:ok, file} = File.stat(output) embed =
file_size = %Nostrum.Struct.Embed{}
file.size / 1000000.0 |> put_color(431_948)
|> Decimal.from_float() |> put_author("Downloaded by #{msg.author.username} (#{file_size} MB)", url, "https://cdn.discordapp.com/avatars/#{msg.author.id}/#{msg.author.avatar}.webp")
|> Decimal.round(2) |> put_footer("You can get the original video by clicking on the \"Downloaded by #{msg.author.username}\" message!")
|> Decimal.to_float() Api.create_message(msg.channel_id, files: [output], embeds: [embed])
end
Api.delete_message(loading.channel_id, loading.id)
Api.delete_message(msg)
if file_size >= 25 do
Api.create_message(msg.channel_id, "File size is too big! (#{file_size})")
else else
embed = {:error, error} -> Api.create_message(msg.channel_id, "`#{error}`")
%Nostrum.Struct.Embed{}
|> put_color(431_948)
|> put_author("Downloaded by #{msg.author.username} (#{file_size} MB)", url, "https://cdn.discordapp.com/avatars/#{msg.author.id}/#{msg.author.avatar}.webp")
|> put_footer("You can get the original video by clicking on the \"Downloaded by #{msg.author.username}\" message!")
Api.create_message(msg.channel_id, files: [output], embeds: [embed])
end end
end end
end end

12
lib/haha_yes/utils.ex Normal file
View file

@ -0,0 +1,12 @@
defmodule HahaYes.Utility do
def download(url, output, format \\ "bestvideo[height<=?480]+bestaudio/best") do
opt = ["-f", format, url, "-o", "#{output}.%(ext)s", "--force-overwrites", "--playlist-reverse", "--no-playlist", "--remux-video=mp4/webm/mov", "--no-warnings"];
{error_output, status} = System.cmd("yt-dlp", opt, [stderr_to_stdout: true])
if status !== 0 do
{:error, error_output}
else
{:ok, Enum.at(Path.wildcard("#{output}.*"), 0)}
end
end
end