From 3429cf99c8b64e19973882d7ecb9bb7f2f13ffca Mon Sep 17 00:00:00 2001
From: Quin Lynch <49576606+quinchs@users.noreply.github.com>
Date: Fri, 14 Jan 2022 07:55:36 -0400
Subject: [PATCH] Add AddRow and WithRows to ComponentBuilder (#2038)

---
 .../MessageComponents/ComponentBuilder.cs     | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/src/Discord.Net.Core/Entities/Interactions/MessageComponents/ComponentBuilder.cs b/src/Discord.Net.Core/Entities/Interactions/MessageComponents/ComponentBuilder.cs
index 4461a4205..b086535f7 100644
--- a/src/Discord.Net.Core/Entities/Interactions/MessageComponents/ComponentBuilder.cs
+++ b/src/Discord.Net.Core/Entities/Interactions/MessageComponents/ComponentBuilder.cs
@@ -238,6 +238,38 @@ namespace Discord
             return this;
         }
 
+        /// <summary>
+        ///     Adds a row to this component builder.
+        /// </summary>
+        /// <param name="row">The row to add.</param>
+        /// <exception cref="IndexOutOfRangeException">The component builder contains the max amount of rows defined as <see cref="MaxActionRowCount"/>.</exception>
+        /// <returns>The current builder.</returns>
+        public ComponentBuilder AddRow(ActionRowBuilder row)
+        {
+            _actionRows ??= new();
+
+            if (_actionRows.Count >= MaxActionRowCount)
+                throw new IndexOutOfRangeException("The max amount of rows has been reached");
+
+            ActionRows.Add(row);
+            return this;
+        }
+
+        /// <summary>
+        ///     Sets the rows of this component builder to a specified collection.
+        /// </summary>
+        /// <param name="rows">The rows to set.</param>
+        /// <exception cref="IndexOutOfRangeException">The collection contains more rows then is allowed by discord.</exception>
+        /// <returns>The current builder.</returns>
+        public ComponentBuilder WithRows(IEnumerable<ActionRowBuilder> rows)
+        {
+            if (rows.Count() > MaxActionRowCount)
+                throw new IndexOutOfRangeException($"Cannot have more than {MaxActionRowCount} rows");
+
+            _actionRows = new List<ActionRowBuilder>(rows);
+            return this;
+        }
+
         /// <summary>
         ///     Builds this builder into a <see cref="MessageComponent"/> used to send your components.
         /// </summary>