From 5943a2ebbfcaac7d1acde6417aa2080e92c22a33 Mon Sep 17 00:00:00 2001
From: Joseph Izang <josephizang@gmail.com>
Date: Wed, 5 Apr 2023 04:42:55 +0100
Subject: [PATCH] add FE context menu to generate docs. fix styles in doc-gen
 ui

---
 apps/doc-gen/src/app/App.tsx          | 8 +++++---
 apps/doc-gen/src/app/docgen-client.ts | 9 ++++++---
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/apps/doc-gen/src/app/App.tsx b/apps/doc-gen/src/app/App.tsx
index de6cc7737b..27de6ec2f5 100644
--- a/apps/doc-gen/src/app/App.tsx
+++ b/apps/doc-gen/src/app/App.tsx
@@ -29,9 +29,11 @@ const App = () => {
 
   return (
     <div className="p-3">
-      <h3>Compile a solidity contract in order to build documentation as markdown.</h3>
-      {fileName && <h6>File: {fileName}</h6>}
-      {hasBuild && <button className="btn btn-primary btn-block mt-4 rounded" onClick={() => client.generateDocs()}>Generate doc</button>}
+      <h5 className="h-5 mb-3">Compile a solidity contract in order to build documentation as markdown or right click on a contract and click on "Generate Documentation".</h5>
+      {fileName && <div className="border-bottom border-top px-2 py-3 justify-center align-items-center d-flex">
+        <h6>File: {fileName}</h6>
+      </div>}
+      {hasBuild && <button className="btn btn-primary btn-block mt-4" onClick={() => client.generateDocs()}>Generate Documentation</button>}
     </div>
   )
 }
diff --git a/apps/doc-gen/src/app/docgen-client.ts b/apps/doc-gen/src/app/docgen-client.ts
index 802b25ecdb..8cc908a5ba 100644
--- a/apps/doc-gen/src/app/docgen-client.ts
+++ b/apps/doc-gen/src/app/docgen-client.ts
@@ -56,8 +56,8 @@ export class DocGenClient extends PluginClient {
   }
 
   async generateDocsCustomAction(action: customAction) {
-    console.log({ action })
-    // await this.generateDocs()
+    await this.call('solidity', 'compile', action.path[0])
+    await this.generateDocs()
   }
 
   async docgen(builds: Build[], userConfig?: Config): Promise<void> {
@@ -86,6 +86,9 @@ export class DocGenClient extends PluginClient {
   }
 
   async generateDocs() {
-    this.docgen([this.build])
+    this.eventEmitter.on('compilationFinished', async (build: Build, fileName: string) => {
+      await this.docgen([build])
+    })
+    this.eventEmitter.off('compilationFinished', () => console.log('remove compilationFinished listener'))
   }
 }