Sollet Chrome Extension v0 (project-serum#127)

* init

* ui changes

* injection

* upd

* adding connections page

* upd extension

* merge

* prettier

* fix

* fix

* fix

.gitignore

@@ -24,3 +24,6 @@ yarn-error.log*
 
 .idea
 .eslintcache
+
+# generate with `build:extension` script
+extension/build/*

extension/src/background.js

@@ -0,0 +1,70 @@
+const responseHandlers = new Map();
+
+function launchPopup(message, sender, sendResponse) {
+  const searchParams = new URLSearchParams();
+  searchParams.set('origin', sender.origin);
+  searchParams.set('network', message.data.params.network);
+  searchParams.set('request', JSON.stringify(message.data));
+
+  // TODO consolidate popup dimensions
+  chrome.windows.getLastFocused((focusedWindow) => {
+    chrome.windows.create({
+      url: 'index.html/#' + searchParams.toString(),
+      type: 'popup',
+      width: 375,
+      height: 600,
+      top: focusedWindow.top,
+      left: focusedWindow.left + (focusedWindow.width - 375),
+      setSelfAsOpener: true,
+      focused: true,
+    });
+  });
+
+  responseHandlers.set(message.data.id, sendResponse);
+}
+
+function handleConnect(message, sender, sendResponse) {
+  chrome.storage.local.get('connectedWallets', (result) => {
+    const connectedWallet = (result.connectedWallets || {})[sender.origin];
+    if (!connectedWallet) {
+      launchPopup(message, sender, sendResponse);
+    } else {
+      sendResponse({
+        method: 'connected',
+        params: {
+          publicKey: connectedWallet.publicKey,
+          autoApprove: connectedWallet.autoApprove,
+        },
+        id: message.data.id,
+      });
+    }
+  });
+}
+
+function handleDisconnect(message, sender, sendResponse) {
+  chrome.storage.local.get('connectedWallets', (result) => {
+    delete result.connectedWallets[sender.origin];
+    chrome.storage.local.set(
+      { connectedWallets: result.connectedWallets },
+      () => sendResponse({ method: 'disconnected', id: message.data.id }),
+    );
+  });
+}
+
+chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
+  if (message.channel === 'sollet_contentscript_background_channel') {
+    if (message.data.method === 'connect') {
+      handleConnect(message, sender, sendResponse);
+    } else if (message.data.method === 'disconnect') {
+      handleDisconnect(message, sender, sendResponse);
+    } else {
+      launchPopup(message, sender, sendResponse);
+    }
+    // keeps response channel open
+    return true;
+  } else if (message.channel === 'sollet_extension_background_channel') {
+    const responseHandler = responseHandlers.get(message.data.id);
+    responseHandlers.delete(message.data.id);
+    responseHandler(message.data);
+  }
+});

extension/src/contentscript.js

@@ -0,0 +1,24 @@
+const container = document.head || document.documentElement;
+const scriptTag = document.createElement('script');
+scriptTag.setAttribute('async', 'false');
+scriptTag.src = chrome.runtime.getURL('script.js');
+container.insertBefore(scriptTag, container.children[0]);
+container.removeChild(scriptTag);
+
+window.addEventListener('sollet_injected_script_message', (event) => {
+  chrome.runtime.sendMessage(
+    {
+      channel: 'sollet_contentscript_background_channel',
+      data: event.detail,
+    },
+    (response) => {
+      // Can return null response if window is killed
+      if (!response) {
+        return;
+      }
+      window.dispatchEvent(
+        new CustomEvent('sollet_contentscript_message', { detail: response }),
+      );
+    },
+  );
+});

extension/src/manifest.json

@@ -0,0 +1,35 @@
+{
+  "name": "Sollet",
+  "description": "Solana SPL Token Wallet",
+  "version": "0.1",
+  "browser_action": {
+    "default_popup": "index.html",
+    "default_title": "Open the popup"
+  },
+  "manifest_version": 2,
+  "icons": {
+    "16": "favicon.ico",
+    "192": "logo192.png",
+    "512": "logo512.png"
+  },
+  "background": {
+    "persistent": false,
+    "scripts": ["background.js"]
+  },
+  "permissions": [
+    "storage"
+  ],
+  "content_scripts": [
+    {
+      "matches": ["file://*/*", "http://*/*", "https://*/*"],
+      "js": [
+        "contentscript.js"
+      ],
+      "run_at": "document_start",
+      "all_frames": true
+    }
+  ],
+  "web_accessible_resources": ["script.js"],
+  "content_security_policy": "script-src 'self' 'sha256-ek+jXksbUr00x+EdLLqiv69t8hATh5rPjHVvVVGA9ms='; object-src 'self'"
+}
+

extension/src/script.js

@@ -0,0 +1,15 @@
+window.sollet = {
+  postMessage: (message) => {
+    const listener = (event) => {
+      if (event.detail.id === message.id) {
+        window.removeEventListener('sollet_contentscript_message', listener);
+        window.postMessage(event.detail);
+      }
+    };
+    window.addEventListener('sollet_contentscript_message', listener);
+
+    window.dispatchEvent(
+      new CustomEvent('sollet_injected_script_message', { detail: message }),
+    );
+  },
+};

package.json

@@ -37,14 +37,21 @@
     "predeploy": "git pull --ff-only && yarn && yarn build",
     "deploy": "gh-pages -d build",
     "fix": "run-s fix:*",
-    "fix:prettier": "prettier \"src/**/*.js\" --write",
+    "fix:prettier": "prettier \"src/**/*.js\" \"extension/src/*.js\" --write",
     "start": "react-scripts start",
     "build": "react-scripts build",
+    "build:extension": "yarn build && cp -a ./build/. ./extension/build/ && yarn build:extension-scripts",
+    "build:extension-scripts": "cp -a ./extension/src/. ./extension/build/.",
     "test": "react-scripts test",
     "eject": "react-scripts eject"
   },
   "eslintConfig": {
-    "extends": "react-app"
+    "env": {
+      "browser": true,
+      "es6": true,
+      "webextensions": true
+    },
+    "extends": ["react-app"]
   },
   "jest": {
     "transformIgnorePatterns": [

src/App.js

@@ -10,10 +10,14 @@ import NavigationFrame from './components/NavigationFrame';
 import { ConnectionProvider } from './utils/connection';
 import WalletPage from './pages/WalletPage';
 import { useWallet, WalletProvider } from './utils/wallet';
+import { ConnectedWalletsProvider } from './utils/connected-wallets';
 import LoadingIndicator from './components/LoadingIndicator';
 import { SnackbarProvider } from 'notistack';
 import PopupPage from './pages/PopupPage';
 import LoginPage from './pages/LoginPage';
+import ConnectionsPage from './pages/ConnectionsPage';
+import { isExtension } from './utils/utils';
+import { PageProvider, usePage } from './utils/page';
 
 export default function App() {
   // TODO: add toggle for dark mode
@@ -25,6 +29,8 @@ export default function App() {
           type: prefersDarkMode ? 'dark' : 'light',
           primary: blue,
         },
+        // TODO consolidate popup dimensions
+        ext: '450',
       }),
     [prefersDarkMode],
   );
@@ -34,20 +40,30 @@ export default function App() {
     return null;
   }
 
+  let appElement = (
+    <NavigationFrame>
+      <Suspense fallback={<LoadingIndicator />}>
+        <PageContents />
+      </Suspense>
+    </NavigationFrame>
+  );
+
+  if (isExtension) {
+    appElement = (
+      <ConnectedWalletsProvider>
+        <PageProvider>{appElement}</PageProvider>
+      </ConnectedWalletsProvider>
+    );
+  }
+
   return (
     <Suspense fallback={<LoadingIndicator />}>
       <ThemeProvider theme={theme}>
         <CssBaseline />
 
         <ConnectionProvider>
           <SnackbarProvider maxSnack={5} autoHideDuration={8000}>
-            <WalletProvider>
-              <NavigationFrame>
-                <Suspense fallback={<LoadingIndicator />}>
-                  <PageContents />
-                </Suspense>
-              </NavigationFrame>
-            </WalletProvider>
+            <WalletProvider>{appElement}</WalletProvider>
           </SnackbarProvider>
         </ConnectionProvider>
       </ThemeProvider>
@@ -57,11 +73,16 @@ export default function App() {
 
 function PageContents() {
   const wallet = useWallet();
+  const [page] = usePage();
   if (!wallet) {
     return <LoginPage />;
   }
   if (window.opener) {
     return <PopupPage opener={window.opener} />;
   }
-  return <WalletPage />;
+  if (page === 'wallet') {
+    return <WalletPage />;
+  } else if (page === 'connections') {
+    return <ConnectionsPage />;
+  }
 }