From 00a57599ea916abf1ec08d6e0bec9254de24744c Mon Sep 17 00:00:00 2001
From: Aniket-Engg <aniketengg.225@gmail.com>
Date: Fri, 30 Sep 2022 14:06:45 +0530
Subject: [PATCH] clean code

---
 .../workspace/src/lib/remix-ui-workspace.tsx  | 28 ++++++++-----------
 .../src/templates/ozerc1155/index.ts          |  8 +++---
 .../src/templates/ozerc20/index.ts            |  8 +++---
 .../src/templates/ozerc721/index.ts           |  8 +++---
 4 files changed, 23 insertions(+), 29 deletions(-)

diff --git a/libs/remix-ui/workspace/src/lib/remix-ui-workspace.tsx b/libs/remix-ui/workspace/src/lib/remix-ui-workspace.tsx
index 0d6e7c3023..85367d0bc8 100644
--- a/libs/remix-ui/workspace/src/lib/remix-ui-workspace.tsx
+++ b/libs/remix-ui/workspace/src/lib/remix-ui-workspace.tsx
@@ -16,8 +16,6 @@ export function Workspace () {
   const [selectedWorkspace, setSelectedWorkspace] = useState<{ name: string, isGitRepo: boolean}>(null)
   const [showDropdown, setShowDropdown] = useState<boolean>(false)
   const displayOzCustomRef = useRef<HTMLDivElement>()
-  const ozFeatures = useRef({mintable: false, burnable: false, pausable: false})
-  const upgradeable = useRef()
   const mintableCheckboxRef = useRef()
   const burnableCheckboxRef = useRef()
   const pausableCheckboxRef = useRef()
@@ -118,10 +116,15 @@ export function Workspace () {
     // @ts-ignore: Object is possibly 'null'.
     const workspaceTemplateName = workspaceCreateTemplateInput.current.value || 'remixDefault'
     const initGitRepo = initGitRepoRef.current.checked
-    const features = ozFeatures.current
     const opts = {
-      upgradeable: upgradeable.current,
-      features
+      // @ts-ignore: Object is possibly 'null'.
+      mintable: mintableCheckboxRef.current.checked,
+      // @ts-ignore: Object is possibly 'null'.
+      burnable: burnableCheckboxRef.current.checked,
+      // @ts-ignore: Object is possibly 'null'.
+      pausable: pausableCheckboxRef.current.checked,
+      // @ts-ignore: Object is possibly 'null'.
+      upgradeable: transparentRadioRef.current.checked ? transparentRadioRef.current.value : ( uupsRadioRef.current.checked ? uupsRadioRef.current.value : false )
     }
 
     try {
@@ -160,8 +163,6 @@ export function Workspace () {
     // @ts-ignore
     if (workspaceCreateTemplateInput.current.value.startsWith('oz') && displayOzCustomRef && displayOzCustomRef.current) {
       displayOzCustomRef.current.style.display = 'block'
-      upgradeable.current = undefined
-      ozFeatures.current = {mintable: false, burnable: false, pausable: false}
       // @ts-ignore
       mintableCheckboxRef.current.checked = false
       // @ts-ignore
@@ -193,17 +194,10 @@ export function Workspace () {
   }
 
   const handleUpgradeability = (e) => {
-    // @ts-ignore
-    upgradeable.current = e.target.value
     // @ts-ignore
     workspaceCreateInput.current.value = `${workspaceCreateTemplateInput.current.value + '_upgradeable'}_${Date.now()}`
   }
 
-  const handleFeatures = (e) => {
-    // @ts-ignore
-    ozFeatures.current[e.target.value] = e.target.checked
-  }
-
   const createModalMessage = () => {
     return (
       <>
@@ -229,15 +223,15 @@ export function Workspace () {
           <label id="wsName" className="form-check-label d-block mb-1">Features</label>
           <div className="mb-2">
             <div className="d-flex ml-2 custom-control custom-checkbox">
-                <input className="custom-control-input" type="checkbox" name="feature" value="mintable" id="mintable" onChange={(e) => handleFeatures(e)} ref={mintableCheckboxRef} />
+                <input className="custom-control-input" type="checkbox" name="feature" value="mintable" id="mintable" ref={mintableCheckboxRef} />
                 <label className="form-check-label custom-control-label" htmlFor="mintable" data-id="featureTypeMintable" >Mintable</label>
             </div>
             <div className="d-flex ml-2 custom-control custom-checkbox">
-                <input className="custom-control-input" type="checkbox" name="feature" value="burnable" id="burnable" onChange={(e) => handleFeatures(e)} ref={burnableCheckboxRef} />
+                <input className="custom-control-input" type="checkbox" name="feature" value="burnable" id="burnable" ref={burnableCheckboxRef} />
                 <label className="form-check-label custom-control-label" htmlFor="burnable" data-id="featureTypeBurnable" >Burnable</label>
             </div>
             <div className="d-flex ml-2 custom-control custom-checkbox">
-                <input className="custom-control-input" type="checkbox" name="feature" value="pausable" id="pausable" onChange={(e) => handleFeatures(e)} ref={pausableCheckboxRef} />
+                <input className="custom-control-input" type="checkbox" name="feature" value="pausable" id="pausable" ref={pausableCheckboxRef} />
                 <label className="form-check-label custom-control-label" htmlFor="pausable" data-id="featureTypePausable" >Pausable</label>
             </div>
           </div>
diff --git a/libs/remix-ws-templates/src/templates/ozerc1155/index.ts b/libs/remix-ws-templates/src/templates/ozerc1155/index.ts
index 4164d91e76..df6a349189 100644
--- a/libs/remix-ws-templates/src/templates/ozerc1155/index.ts
+++ b/libs/remix-ws-templates/src/templates/ozerc1155/index.ts
@@ -1,10 +1,10 @@
 import { erc1155 } from '@openzeppelin/wizard';
 
 export default async (opts) => {
-    if (opts.features) {
-        erc1155.defaults.mintable = opts.features.mintable
-        erc1155.defaults.burnable = opts.features.burnable
-        erc1155.defaults.pausable = opts.features.pausable
+    if (opts) {
+        erc1155.defaults.mintable = opts.mintable
+        erc1155.defaults.burnable = opts.burnable
+        erc1155.defaults.pausable = opts.pausable
     }
 
     const filesObj = {
diff --git a/libs/remix-ws-templates/src/templates/ozerc20/index.ts b/libs/remix-ws-templates/src/templates/ozerc20/index.ts
index 5eca9475f8..529b11fc13 100644
--- a/libs/remix-ws-templates/src/templates/ozerc20/index.ts
+++ b/libs/remix-ws-templates/src/templates/ozerc20/index.ts
@@ -1,10 +1,10 @@
 import { erc20 } from '@openzeppelin/wizard';
 
 export default async (opts) => {
-    if (opts.features) {
-        erc20.defaults.mintable = opts.features.mintable
-        erc20.defaults.burnable = opts.features.burnable
-        erc20.defaults.pausable = opts.features.pausable
+    if (opts) {
+        erc20.defaults.mintable = opts.mintable
+        erc20.defaults.burnable = opts.burnable
+        erc20.defaults.pausable = opts.pausable
     }
     
     const filesObj = {
diff --git a/libs/remix-ws-templates/src/templates/ozerc721/index.ts b/libs/remix-ws-templates/src/templates/ozerc721/index.ts
index 5c7f01293d..12d4d2deb3 100644
--- a/libs/remix-ws-templates/src/templates/ozerc721/index.ts
+++ b/libs/remix-ws-templates/src/templates/ozerc721/index.ts
@@ -1,10 +1,10 @@
 import { erc721 } from '@openzeppelin/wizard';
 
 export default async (opts) => {
-    if (opts.features) {
-        erc721.defaults.mintable = opts.features.mintable
-        erc721.defaults.burnable = opts.features.burnable
-        erc721.defaults.pausable = opts.features.pausable
+    if (opts) {
+        erc721.defaults.mintable = opts.mintable
+        erc721.defaults.burnable = opts.burnable
+        erc721.defaults.pausable = opts.pausable
     }
 
     const filesObj = {