From f1df4bbfc0f37df47b1750e9f7f89dda1cb42c1c Mon Sep 17 00:00:00 2001
From: Joseph Izang <josephizang@gmail.com>
Date: Sat, 10 Feb 2024 18:44:04 +0100
Subject: [PATCH] refactor vyper to work with context command.

---
 apps/vyper/src/app/app.tsx                    |  28 +++-
 .../src/app/components/CompilerButton.tsx     | 111 +--------------
 apps/vyper/src/app/utils/compiler.tsx         | 127 ++++++++++++++++++
 apps/vyper/src/app/utils/remix-client.tsx     |  23 +++-
 .../src/lib/vyper-compile-details.tsx         |  12 +-
 .../src/lib/vyperCompile.tsx                  |   1 -
 6 files changed, 186 insertions(+), 116 deletions(-)

diff --git a/apps/vyper/src/app/app.tsx b/apps/vyper/src/app/app.tsx
index ebcfdd3aac..0a20e9d462 100644
--- a/apps/vyper/src/app/app.tsx
+++ b/apps/vyper/src/app/app.tsx
@@ -28,7 +28,7 @@ interface OutputMap {
 
 const App = () => {
   const [contract, setContract] = useState<string>()
-  const [output, setOutput] = useState<any>({})
+  const [output, setOutput] = useState<any>(remixClient.compilerOutput)
   const [state, setState] = useState<AppState>({
     status: 'idle',
     environment: 'remote',
@@ -53,6 +53,30 @@ const App = () => {
     start()
   }, [])
 
+  useEffect(() => {
+    remixClient.eventEmitter.on('resetCompilerState', () => {
+      resetCompilerResultState()
+    })
+
+    return () => {
+      remixClient.eventEmitter.off('resetCompilerState', () => {
+        resetCompilerResultState()
+      })
+    }
+  }, [])
+
+  useEffect(() => {
+    remixClient.eventEmitter.on('setOutput', (payload) => {
+      setOutput(payload)
+    })
+
+    return () => {
+      remixClient.eventEmitter.off('setOutput', (payload) => {
+        setOutput(payload)
+      })
+    }
+  }, [])
+
   /** Update the environment state value */
   function setEnvironment(environment: 'local' | 'remote') {
     setState({...state, environment})
@@ -67,7 +91,7 @@ const App = () => {
   }
 
   function resetCompilerResultState() {
-    setOutput({})
+    setOutput(remixClient.compilerOutput)
   }
 
   return (
diff --git a/apps/vyper/src/app/components/CompilerButton.tsx b/apps/vyper/src/app/components/CompilerButton.tsx
index ad04d0ba04..f7d2331348 100644
--- a/apps/vyper/src/app/components/CompilerButton.tsx
+++ b/apps/vyper/src/app/components/CompilerButton.tsx
@@ -1,7 +1,6 @@
 import React, { Fragment, useState } from 'react'
-import {isVyper, compile, toStandardOutput, isCompilationError, remixClient, normalizeContractPath} from '../utils'
+import {isVyper, compile, toStandardOutput, isCompilationError, remixClient, normalizeContractPath, compileContract} from '../utils'
 import Button from 'react-bootstrap/Button'
-import _ from 'lodash'
 
 interface Props {
   compilerUrl: string
@@ -21,112 +20,14 @@ function CompilerButton({contract, setOutput, compilerUrl, resetCompilerState}:
   }
 
   /** Compile a Contract */
-  async function compileContract() {
-    resetCompilerState()
-    setLoadingSpinnerState(true)
-    try {
-      // await remixClient.discardHighlight()
-      let _contract: any
-      try {
-        _contract = await remixClient.getContract()
-      } catch (e: any) {
-        setOutput('', {status: 'failed', message: e.message})
-        return
-      }
-      remixClient.changeStatus({
-        key: 'loading',
-        type: 'info',
-        title: 'Compiling'
-      })
-      let output
-      try {
-        output = await compile(compilerUrl, _contract)
-      } catch (e: any) {
-        remixClient.changeStatus({
-          key: 'failed',
-          type: 'error',
-          title: e.message
-        })
-        return
-      }
-      const compileReturnType = () => {
-        const t: any = toStandardOutput(contract, output)
-        const temp = _.merge(t['contracts'][contract])
-        const normal = normalizeContractPath(contract)[2]
-        const abi = temp[normal]['abi']
-        const evm = _.merge(temp[normal]['evm'])
-        const dpb = evm.deployedBytecode
-        const runtimeBytecode = evm.bytecode
-        const methodIdentifiers = evm.methodIdentifiers
-
-        const result = {
-          contractName: normal,
-          abi: abi,
-          bytecode: dpb,
-          runtimeBytecode: runtimeBytecode,
-          ir: '',
-          methodIdentifiers: methodIdentifiers
-        }
-        return result
-      }
-
-      // ERROR
-      if (isCompilationError(output)) {
-        const line = output.line
-        if (line) {
-          const lineColumnPos = {
-            start: {line: line - 1, column: 10},
-            end: {line: line - 1, column: 10}
-          }
-          // remixClient.highlight(lineColumnPos as any, _contract.name, '#e0b4b4')
-        } else {
-          const regex = output?.message?.match(/line ((\d+):(\d+))+/g)
-          const errors = output?.message?.split(/line ((\d+):(\d+))+/g) // extract error message
-          if (regex) {
-            let errorIndex = 0
-            regex.map((errorLocation) => {
-              const location = errorLocation?.replace('line ', '').split(':')
-              let message = errors[errorIndex]
-              errorIndex = errorIndex + 4
-              if (message && message?.split('\n\n').length > 0) {
-                try {
-                  message = message?.split('\n\n')[message.split('\n\n').length - 1]
-                } catch (e) {}
-              }
-              if (location?.length > 0) {
-                const lineColumnPos = {
-                  start: {line: parseInt(location[0]) - 1, column: 10},
-                  end: {line: parseInt(location[0]) - 1, column: 10}
-                }
-                // remixClient.highlight(lineColumnPos as any, _contract.name, message)
-              }
-            })
-          }
-        }
-        throw new Error(output.message)
-      }
-      // SUCCESS
-      // remixClient.discardHighlight()
-      remixClient.changeStatus({
-        key: 'succeed',
-        type: 'success',
-        title: 'success'
-      })
-      const data = toStandardOutput(_contract.name, output)
-      remixClient.compilationFinish(_contract.name, _contract.content, data)
-      setOutput(_contract.name, compileReturnType())
-    } catch (err: any) {
-      remixClient.changeStatus({
-        key: 'failed',
-        type: 'error',
-        title: err.message
-      })
-    }
-  }
 
   return (
     <Fragment>
-      <button data-id="compile" onClick={compileContract} title={contract} className="btn btn-primary w-100 d-block btn-block text-break remixui_disabled mb-1 mt-3">
+      <button data-id="compile"
+        onClick={() => compileContract(contract, compilerUrl, setOutput)}
+        title={contract}
+        className="btn btn-primary w-100 d-block btn-block text-break remixui_disabled mb-1 mt-3"
+      >
         <div className="d-flex align-items-center justify-content-center fa-1x">
           <span className="fas fa-sync fa-pulse mr-1" />
           <div className="text-truncate overflow-hidden text-nowrap">
diff --git a/apps/vyper/src/app/utils/compiler.tsx b/apps/vyper/src/app/utils/compiler.tsx
index ed1cc80b50..3b624211ec 100644
--- a/apps/vyper/src/app/utils/compiler.tsx
+++ b/apps/vyper/src/app/utils/compiler.tsx
@@ -1,5 +1,8 @@
 import { ABIDescription} from '@remixproject/plugin-api'
 import axios from 'axios'
+import { remixClient } from './remix-client'
+import _ from 'lodash'
+
 
 export interface Contract {
   name: string
@@ -167,6 +170,130 @@ export function toStandardOutput(fileName: string, compilationResult: any): any
     }
   }
 }
+
+
+export async function compileContract(contract: string, compilerUrl: string, setOutput?: any) {
+  remixClient.eventEmitter.emit('resetCompilerState', {})
+
+  try {
+    // await remixClient.discardHighlight()
+    let _contract: any
+    try {
+      _contract = await remixClient.getContract()
+    } catch (e: any) {
+      if (setOutput === null || setOutput === undefined) {
+        const compileResult = {
+          status: 'failed',
+          message: e.message
+        }
+        const compileResultKey = ''
+        remixClient.eventEmitter.emit('setOutput', { compileResultKey, compileResult })
+      } else {
+        setOutput('', {status: 'failed', message: e.message})
+      }
+      return
+    }
+    remixClient.changeStatus({
+      key: 'loading',
+      type: 'info',
+      title: 'Compiling'
+    })
+    let output
+    try {
+      output = await compile(compilerUrl, _contract)
+    } catch (e: any) {
+      remixClient.changeStatus({
+        key: 'failed',
+        type: 'error',
+        title: e.message
+      })
+      return
+    }
+    const compileReturnType = () => {
+      const t: any = toStandardOutput(contract, output)
+      const temp = _.merge(t['contracts'][contract])
+      const normal = normalizeContractPath(contract)[2]
+      const abi = temp[normal]['abi']
+      const evm = _.merge(temp[normal]['evm'])
+      const dpb = evm.deployedBytecode
+      const runtimeBytecode = evm.bytecode
+      const methodIdentifiers = evm.methodIdentifiers
+
+      const result = {
+        contractName: normal,
+        abi: abi,
+        bytecode: dpb,
+        runtimeBytecode: runtimeBytecode,
+        ir: '',
+        methodIdentifiers: methodIdentifiers
+      }
+      return result
+    }
+
+    // ERROR
+    if (isCompilationError(output)) {
+      const line = output.line
+      if (line) {
+        const lineColumnPos = {
+          start: {line: line - 1, column: 10},
+          end: {line: line - 1, column: 10}
+        }
+        // remixClient.highlight(lineColumnPos as any, _contract.name, '#e0b4b4')
+      } else {
+        const regex = output?.message?.match(/line ((\d+):(\d+))+/g)
+        const errors = output?.message?.split(/line ((\d+):(\d+))+/g) // extract error message
+        if (regex) {
+          let errorIndex = 0
+          regex.map((errorLocation) => {
+            const location = errorLocation?.replace('line ', '').split(':')
+            let message = errors[errorIndex]
+            errorIndex = errorIndex + 4
+            if (message && message?.split('\n\n').length > 0) {
+              try {
+                message = message?.split('\n\n')[message.split('\n\n').length - 1]
+              } catch (e) {}
+            }
+            if (location?.length > 0) {
+              const lineColumnPos = {
+                start: {line: parseInt(location[0]) - 1, column: 10},
+                end: {line: parseInt(location[0]) - 1, column: 10}
+              }
+              // remixClient.highlight(lineColumnPos as any, _contract.name, message)
+            }
+          })
+        }
+      }
+      throw new Error(output.message)
+    }
+    // SUCCESS
+    // remixClient.discardHighlight()
+    remixClient.changeStatus({
+      key: 'succeed',
+      type: 'success',
+      title: 'success'
+    })
+
+    const data = toStandardOutput(_contract.name, output)
+    remixClient.compilationFinish(_contract.name, _contract.content, data)
+    if (setOutput === null || setOutput === undefined) {
+      const contractName = _contract['name']
+      const compileResult = compileReturnType()
+      remixClient.eventEmitter.emit('setOutput', { contractName, compileResult })
+    } else {
+      setOutput(_contract.name, compileReturnType())
+    }
+  } catch (err: any) {
+    remixClient.changeStatus({
+      key: 'failed',
+      type: 'error',
+      title: err.message
+    })
+  }
+}
+
+
+
+
 export type StandardOutput = {
   sources: {
     [fileName: string]: {
diff --git a/apps/vyper/src/app/utils/remix-client.tsx b/apps/vyper/src/app/utils/remix-client.tsx
index 96c7dd209f..4f0c45a09b 100644
--- a/apps/vyper/src/app/utils/remix-client.tsx
+++ b/apps/vyper/src/app/utils/remix-client.tsx
@@ -2,11 +2,22 @@ import {HighlightPosition, CompilationResult, RemixApi, customAction} from '@rem
 import {Api, Status} from '@remixproject/plugin-utils'
 import {createClient} from '@remixproject/plugin-webview'
 import {PluginClient} from '@remixproject/plugin'
-import {Contract} from './compiler'
+import {Contract, compileContract} from './compiler'
 import {ExampleContract} from '../components/VyperResult'
+import EventEmitter from 'events'
 
+
+export type VyperComplierAddress = 'https://vyper2.remixproject.org/' | 'http://localhost:8000/'
 export class RemixClient extends PluginClient {
   private client = createClient<Api, Readonly<RemixApi>>(this)
+  compilerUrl: VyperComplierAddress = 'https://vyper2.remixproject.org/'
+  compilerOutput: any
+  eventEmitter = new EventEmitter()
+
+  constructor() {
+    super()
+    this.compilerOutput = {}
+  }
 
   loaded() {
     return this.client.onload()
@@ -26,8 +37,16 @@ export class RemixClient extends PluginClient {
     })
   }
 
+  resetCompilerState() {
+    this.compilerOutput = {}
+    this.eventEmitter.emit('resetCompilerState', {})
+  }
+
   async vyperCompileCustomAction(action: customAction) {
-    console.log('vyperCompileCustomAction', action)
+    //read selected contract from file explorer and create contract type
+    const contract = await this.getContract()
+    //compile contract
+    await compileContract(contract.name, this.compilerUrl)
   }
 
   /** Load Ballot contract example into the file manager */
diff --git a/libs/remix-ui/vyper-compile-details/src/lib/vyper-compile-details.tsx b/libs/remix-ui/vyper-compile-details/src/lib/vyper-compile-details.tsx
index 63da76b412..8ef97958b1 100644
--- a/libs/remix-ui/vyper-compile-details/src/lib/vyper-compile-details.tsx
+++ b/libs/remix-ui/vyper-compile-details/src/lib/vyper-compile-details.tsx
@@ -10,12 +10,12 @@ interface RemixUiVyperCompileDetailsProps {
 }
 
 export function RemixUiVyperCompileDetails({ payload, theme, themeStyle }: RemixUiVyperCompileDetailsProps) {
-  const dpayload = Object.values(payload) as any ?? {}
-  const bcode = dpayload[0].bytecode ? dpayload[0].bytecode.object : ''
-  const runtimeBcode = dpayload[0].runtimeBytecode ? dpayload[0].runtimeBytecode.object : ''
-  const ir = dpayload[0].ir
-  const methodIdentifiers= dpayload[0].methodIdentifiers
-  const abi= dpayload[0].abi
+  const compileResult = payload['compileResult'] ?? {}
+  const bcode = compileResult.bytecode ? compileResult.bytecode.object : ''
+  const runtimeBcode = compileResult.runtimeBytecode ? compileResult.runtimeBytecode.object : ''
+  const ir = compileResult.ir
+  const methodIdentifiers= compileResult.methodIdentifiers
+  const abi= compileResult.abi
   return (
     <>
       <VyperCompile
diff --git a/libs/remix-ui/vyper-compile-details/src/lib/vyperCompile.tsx b/libs/remix-ui/vyper-compile-details/src/lib/vyperCompile.tsx
index 9123c606b0..fd1211111c 100644
--- a/libs/remix-ui/vyper-compile-details/src/lib/vyperCompile.tsx
+++ b/libs/remix-ui/vyper-compile-details/src/lib/vyperCompile.tsx
@@ -25,7 +25,6 @@ export interface VyperCompileProps {
 }
 
 export default function VyperCompile({result, theme, themeStyle}: VyperCompileProps) {
-
   const [active, setActive] = useState<keyof VyperCompilationResult>('abi')
   const tabContent = [
     {