From 1064277cd9864219e9a80db9d42ad2a25248149c Mon Sep 17 00:00:00 2001
From: Maciej Lenartowicz <maciej.lenartowicz@nic.cz>
Date: Fri, 29 Nov 2019 16:01:53 +0100
Subject: [PATCH] Allow adding classes to DownloadButton

---
 src/bootstrap/DownloadButton.js | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/bootstrap/DownloadButton.js b/src/bootstrap/DownloadButton.js
index 8b3d043..13d1de1 100644
--- a/src/bootstrap/DownloadButton.js
+++ b/src/bootstrap/DownloadButton.js
@@ -10,12 +10,25 @@ import PropTypes from "prop-types";
 
 DownloadButton.propTypes = {
     href: PropTypes.string.isRequired,
+    className: PropTypes.string,
     children: PropTypes.oneOfType([
         PropTypes.arrayOf(PropTypes.node),
         PropTypes.node,
     ]),
 };
 
-export function DownloadButton({ href, children }) {
-    return <a href={href} className="btn btn-primary" download>{children}</a>;
+DownloadButton.defaultProps = {
+    className: "btn-primary",
+};
+
+export function DownloadButton({ href, className, children }) {
+    return (
+        <a
+            href={href}
+            className={`btn ${className}`.trim()}
+            download
+        >
+            {children}
+        </a>
+    );
 }