From 95d6cc17b5b8164c425a512958cb820dfb6b9d01 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Bersier?= <loic.bersier1@gmail.com>
Date: Sat, 12 Oct 2024 21:14:22 +0200
Subject: [PATCH] Add embed and file size limit

---
 lib/HahaCommands/download.ex | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/lib/HahaCommands/download.ex b/lib/HahaCommands/download.ex
index 0459b4b..731c301 100644
--- a/lib/HahaCommands/download.ex
+++ b/lib/HahaCommands/download.ex
@@ -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