From 741c609daef922d87cb64ef6ebb7d6bf1f5e8e4f Mon Sep 17 00:00:00 2001
From: Joseph Izang <josephizang@gmail.com>
Date: Mon, 20 Feb 2023 10:44:25 +0100
Subject: [PATCH] add components for sol2uml

---
 .../src/lib/components/UmlDownload.tsx        | 19 +++++++++++++++++++
 .../src/lib/solidity-uml-gen.tsx              |  8 ++------
 2 files changed, 21 insertions(+), 6 deletions(-)
 create mode 100644 libs/remix-ui/solidity-uml-gen/src/lib/components/UmlDownload.tsx

diff --git a/libs/remix-ui/solidity-uml-gen/src/lib/components/UmlDownload.tsx b/libs/remix-ui/solidity-uml-gen/src/lib/components/UmlDownload.tsx
new file mode 100644
index 0000000000..d4e258da99
--- /dev/null
+++ b/libs/remix-ui/solidity-uml-gen/src/lib/components/UmlDownload.tsx
@@ -0,0 +1,19 @@
+import React, { Fragment } from 'react'
+import { Dropdown } from 'react-bootstrap'
+
+export default function UmlDownload() {
+  return(
+    <Fragment>
+      <Dropdown>
+        <Dropdown.Toggle className="badge badge-info remixui_no-shadow p-2 rounded-circle mr-2">
+        <i className="far fa-arrow-to-bottom align-item-center d-flex justify-content-center"></i>
+        </Dropdown.Toggle>
+        <Dropdown.Menu>
+          <Dropdown.Item>Download as PNG</Dropdown.Item>
+          <Dropdown.Divider />
+          <Dropdown.Item>Download as PDF</Dropdown.Item>
+        </Dropdown.Menu>
+      </Dropdown>
+    </Fragment>
+  )
+}
\ No newline at end of file
diff --git a/libs/remix-ui/solidity-uml-gen/src/lib/solidity-uml-gen.tsx b/libs/remix-ui/solidity-uml-gen/src/lib/solidity-uml-gen.tsx
index 036fb57ca8..89108f32a8 100644
--- a/libs/remix-ui/solidity-uml-gen/src/lib/solidity-uml-gen.tsx
+++ b/libs/remix-ui/solidity-uml-gen/src/lib/solidity-uml-gen.tsx
@@ -1,6 +1,7 @@
 import React, { Fragment, useEffect, useState } from 'react'
 import { TransformComponent, TransformWrapper } from 'react-zoom-pan-pinch'
 import { ThemeSummary } from '../types'
+import UmlDownload from './components/UmlDownload'
 import './css/solidity-uml-gen.css'
 export interface RemixUiSolidityUmlGenProps {
   updatedSvg?: string
@@ -47,12 +48,7 @@ export function RemixUiSolidityUmlGen ({ updatedSvg, loading }: RemixUiSolidityU
           style={{ zIndex: 3, top: "10", right: "2em" }}
         >
           <div className="py-2 px-2 d-flex justify-content-center align-items-center">
-            <button
-              className="btn btn-outline-info d-none rounded-circle mr-2"
-              onClick={() => resetTransform()}
-            >
-              <i className="far fa-arrow-to-bottom align-item-center d-flex justify-content-center"></i>
-            </button>
+            <UmlDownload />
             <button
               className="badge badge-info remixui_no-shadow p-2 rounded-circle mr-2"
               onClick={() => zoomIn()}