Discord_Elixir_test/lib/HahaCommands/download.ex

63 lines
2 KiB
Elixir

defmodule HahaYes.Commands.Download do
@moduledoc """
Download command
"""
import Nostrum.Struct.Embed
alias Nostrum.Api
alias HahaYes.Utility
@doc """
Download the video sent by the user with yt-dlp.
## Parameters
- url: String that represents the URL to download a video from.
## Examples
User: h3h3 download https://x.com/i/status/1845119662402511072
Bot: <video file>
User: h3h3 download https://www.youtube.com/watch?v=ryS7TS_J7KA
Bot: <video file>
User: h3h3 download https://cdn.discordapp.com/attachments/790701936394633216/1295042668929355816/1295042650113704087.mp4?ex=670d35f9&is=670be479&hm=bf61358b9b6f8c94a14be4e7e3650799ef733331e84d87254b71a098646c4bc2&
Bot: <video file>
User: h3h3 download invalid
Bot: `[generic] Extracting URL: invalid
ERROR: [generic] 'invalid' is not a valid URL. Set --default-search "ytsearch" (or run yt-dlp "ytsearch:invalid" ) to search YouTube`
"""
def execute(msg, _ws_state, args) do
url = Enum.at(args, 0)
{:ok, loading} = Api.create_message(msg.channel_id, "⚙️ Processing...")
with {:ok, output} <- Utility.do_download(url, "#{System.tmp_dir()}/#{msg.id}", loading),
{:ok, output} <- Utility.do_compression(output, loading),
{:ok, file_size} <- Utility.validate_filesize(output) do
embed =
%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])
# Delete original message and loading message
Api.delete_message(loading.channel_id, loading.id)
Api.delete_message(msg)
else
{:error, error} -> Api.create_message(msg.channel_id, "`#{error}`")
end
end
end