Add embed and file size limit
This commit is contained in:
parent
793c2f95fe
commit
95d6cc17b5
1 changed files with 23 additions and 3 deletions
|
@ -2,7 +2,7 @@ defmodule HahaYes.Commands.Download do
|
|||
@moduledoc """
|
||||
Download command
|
||||
"""
|
||||
|
||||
import Nostrum.Struct.Embed
|
||||
alias Nostrum.Api
|
||||
|
||||
@doc """
|
||||
|
@ -23,14 +23,34 @@ defmodule HahaYes.Commands.Download do
|
|||
Bot: <video file>
|
||||
"""
|
||||
def execute(msg, _ws_state, args) do
|
||||
url = Enum.at(args, 0)
|
||||
{:ok, loading} = Api.create_message(msg.channel_id, "Downloading...")
|
||||
|
||||
opt = ["-f", "bestvideo[height<=?480]+bestaudio/best", args, "-o", "#{System.tmp_dir}/#{msg.id}.%(ext)si", "--force-overwrites", "--playlist-reverse", "--no-playlist", "--remux-video=mp4/webm/mov", "--no-warnings"];
|
||||
|
||||
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"];
|
||||
|
||||
System.cmd("yt-dlp", opt)
|
||||
|
||||
output = Enum.at(Path.wildcard("#{System.tmp_dir}/#{msg.id}.*"), 0)
|
||||
{:ok, file} = File.stat(output)
|
||||
file_size =
|
||||
file.size / 1000000.0
|
||||
|> Decimal.from_float()
|
||||
|> Decimal.round(2)
|
||||
|> Decimal.to_float()
|
||||
|
||||
Api.delete_message(loading.channel_id, loading.id)
|
||||
Api.delete_message(msg)
|
||||
Api.create_message(msg.channel_id, files: [Enum.at(Path.wildcard("#{System.tmp_dir}/#{msg.id}.*"), 0)])
|
||||
|
||||
if file_size >= 25 do
|
||||
Api.create_message(msg.channel_id, "File size is too big! (#{file_size})")
|
||||
else
|
||||
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])
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
Loading…
Reference in a new issue