Merge pull request #6088 from ethereum/newaichat

Newaichat

Author: yann300

.circleci/config.yml

@@ -823,7 +823,7 @@ workflows:
               script: ["flaky.sh"]
               job: ["nogroup"]
               jobsize: ["1"]
-              parallelism: [5]
+              parallelism: [1]
   run_metamask_tests:
     when: << pipeline.parameters.run_metamask_tests >>
     jobs:

.gitignore

@@ -18,7 +18,9 @@ stats.json
 release
 .env
 .env.local
-
+.idea/
+.yalc/
+.yalc
 
 # compiled output
 /dist

apps/remix-ide-e2e/nightwatch-firefox.ts

@@ -35,7 +35,7 @@ module.exports = {
         'javascriptEnabled': true,
         'acceptSslCerts': true,
         'moz:firefoxOptions': {
-          args: ['-width=2560', '-height=1440']
+          args: ['-width=2560', '-height=1440', '--devtools']
         }
       }
     },

apps/remix-ide-e2e/src/commands/addFile.ts

@@ -3,8 +3,8 @@ import EventEmitter from 'events'
 
 class AddFile extends EventEmitter {
   command(this: NightwatchBrowser, name: string, content: NightwatchContractContent, readMeFile?:string): NightwatchBrowser {
-    if(!readMeFile)
-      readMeFile = 'README.txt' 
+    if (!readMeFile)
+      readMeFile = 'README.txt'
     this.api.perform((done) => {
       addFile(this.api, name, content, readMeFile, () => {
         done()
@@ -63,7 +63,7 @@ function addFile(browser: NightwatchBrowser, name: string, content: NightwatchCo
           })
           .setEditorValue(content.content)
           .getEditorValue((result) => {
-            if(result != content.content) {
+            if (result != content.content) {
               browser.setEditorValue(content.content)
             }
           })

apps/remix-ide-e2e/src/commands/assistantAddContext.ts

@@ -0,0 +1,52 @@
+import { NightwatchBrowser } from 'nightwatch'
+import EventEmitter from 'events'
+
+class AssistantAddCtx extends EventEmitter {
+  command(this: NightwatchBrowser, prompt: string): NightwatchBrowser {
+    this.api.perform((done) => {
+      selectCtx(this.api, prompt, () => {
+        done()
+        this.emit('complete')
+      })
+    })
+    return this
+  }
+}
+
+function selectCtx(browser: NightwatchBrowser, ctx: string, done: VoidFunction) {
+  browser
+    .waitForElementVisible('*[data-id="remix-ai-assistant"]')
+    .waitForElementVisible('*[data-id="composer-ai-add-context"]')
+    .click({
+      locateStrategy: 'xpath',
+      selector: '//*[@data-id="composer-ai-add-context"]'
+    })
+    .waitForElementVisible('*[data-id="currentFile-context-option"]')
+    .perform(async ()=> {
+      switch (ctx) {
+      case 'currentFile':
+        browser.click({
+          locateStrategy: 'xpath',
+          selector: '//*[@data-id="currentFile-context-option"]'
+        });
+        break;
+      case 'workspace':
+        browser.click({
+          locateStrategy: 'xpath',
+          selector: '//*[@data-id="workspace-context-option"]'
+        });
+        break;
+      case 'openedFiles':
+        browser.click({
+          locateStrategy: 'xpath',
+          selector: '//*[@data-id="allOpenedFiles-context-option"]'
+        });
+        break;
+      default:
+        break;
+      }
+    })
+  done()
+}
+
+module.exports = AssistantAddCtx;

apps/remix-ide-e2e/src/commands/assistantClearChat.ts

@@ -0,0 +1,25 @@
+import { NightwatchBrowser } from 'nightwatch'
+import EventEmitter from 'events'
+
+class AssistantClearChat extends EventEmitter {
+    command(this: NightwatchBrowser): NightwatchBrowser {
+        this.api.perform((done) => {
+            clearChat(this.api, () => {
+                done()
+                this.emit('complete')
+            })
+        })
+        return this
+    }
+}
+
+function clearChat(browser: NightwatchBrowser, done: VoidFunction) {
+    browser
+        .execute(function () {
+            (window as any).remixAIChat.current.clearChat();
+        }, [])
+
+    done()
+}
+
+module.exports = AssistantClearChat;

apps/remix-ide-e2e/src/commands/assistantGenerate.ts

@@ -0,0 +1,38 @@
+import { NightwatchBrowser } from 'nightwatch'
+import EventEmitter from 'events'
+
+class AssistantGenerate extends EventEmitter {
+  command(this: NightwatchBrowser, prompt: string, provider: string): NightwatchBrowser {
+    this.api.perform((done) => {
+      generate(this.api, prompt, provider, () => {
+        done()
+        this.emit('complete')
+      })
+    })
+    return this
+  }
+}
+
+function generate(browser: NightwatchBrowser, prompt: string, provider: string, done: VoidFunction) {
+
+  console.log('Generating with prompt:', prompt, 'and provider:', provider)
+  browser
+    .waitForElementPresent('*[data-id="remix-ai-assistant-ready"]')
+    .assistantSetProvider(provider)
+    .waitForElementPresent({
+      locateStrategy: 'xpath',
+      selector: "//*[@data-id='remix-ai-streaming' and @data-streaming='false']",
+    })
+
+    .execute(function (prompt) {
+      (window as any).remixAIChat.current.sendChat(`/generate ${prompt}`);
+    }, [prompt])
+    .waitForElementVisible({
+      locateStrategy: 'xpath',
+      selector: `//*[contains(@class,"chat-bubble") and contains(.,"/generate ${prompt}")]`
+    })
+
+  done()
+}
+
+module.exports = AssistantGenerate;

apps/remix-ide-e2e/src/commands/assistantSetProvider.ts

@@ -0,0 +1,36 @@
+import { NightwatchBrowser } from 'nightwatch'
+import EventEmitter from 'events'
+
+class SetAssistantProvider extends EventEmitter {
+  command(this: NightwatchBrowser, provider: string): NightwatchBrowser {
+    this.api.perform((done) => {
+      setAssistant( this.api, provider, () => {
+        done()
+        this.emit('complete')
+      })
+    })
+    return this
+  }
+}
+
+function setAssistant(browser: NightwatchBrowser, provider: string, done: VoidFunction) {
+  browser
+    .waitForElementPresent('*[data-id="remix-ai-assistant-ready"]')
+    .execute(function (provider) {
+      (window as any).remixAIChat.current.sendChat(`/setAssistant ${provider}`);
+    }, [provider])
+    .waitForElementVisible({
+      locateStrategy: 'xpath',
+      selector: `//*[contains(@class,'chat-bubble') and contains(.,'AI Provider set to')]`,
+      timeout: 50000,
+      abortOnFailure: true
+    })
+    .waitForElementPresent({
+      locateStrategy: 'xpath',
+      selector: "//*[@data-id='remix-ai-streaming' and @data-streaming='false']",
+    })
+    //.pause()
+    .perform(() => done())  
+}
+
+module.exports = SetAssistantProvider

apps/remix-ide-e2e/src/commands/assistantWorkspace.ts

@@ -0,0 +1,31 @@
+import { NightwatchBrowser } from 'nightwatch'
+import EventEmitter from 'events'
+
+class AssistantWorkspace extends EventEmitter {
+  command(this: NightwatchBrowser, prompt: string, provider: string): NightwatchBrowser {
+    this.api.perform((done) => {
+      workspaceGenerate(this.api, prompt, provider, () => {
+        done()
+        this.emit('complete')
+      })
+    })
+    return this
+  }
+}
+
+function workspaceGenerate(browser: NightwatchBrowser, prompt: string, provider: string, done: VoidFunction) {
+  browser
+    .waitForElementVisible('*[data-id="remix-ai-assistant"]')
+    .waitForElementPresent('*[data-id="remix-ai-assistant-ready"]')
+    .assistantSetProvider(provider)
+    .execute(function (prompt) {
+      (window as any).remixAIChat.current.sendChat(`/workspace ${prompt}`);
+    }, [prompt])
+    .waitForElementVisible({
+      locateStrategy: 'xpath',
+      selector: `//*[contains(@class,"chat-bubble") and contains(.,"/workspace ${prompt}")]`
+    })
+  done()
+}
+
+module.exports = AssistantWorkspace;

apps/remix-ide-e2e/src/commands/enableClipBoard.ts

@@ -2,18 +2,17 @@ import { NightwatchBrowser } from 'nightwatch'
 import EventEmitter from 'events'
 
 class EnableClipBoard extends EventEmitter {
-  command (this: NightwatchBrowser, remember:boolean, accept: boolean): NightwatchBrowser {
+  command (this: NightwatchBrowser,): NightwatchBrowser {
     const browser = this.api
-    
-    if(browser.browserName.indexOf('chrome') > -1){
+
+    if (browser.browserName.indexOf('chrome') > -1){
       const chromeBrowser = (browser as any).chrome
       chromeBrowser.setPermission('clipboard-read', 'granted')
       chromeBrowser.setPermission('clipboard-write', 'granted')
       // test it
       browser.executeAsyncScript(function (done) {
         navigator.clipboard.writeText('test').then(function () {
           navigator.clipboard.readText().then(function (text) {
-            console.log('Pasted content: ', text)
             done(text)
           }).catch(function (err) {
             console.error('Failed to read clipboard contents: ', err)